#include <stdio.h>
#include "message.h"
-static int connect_friend(struct sockaddr **saddr, char *address, uint16_t port)
-{
- struct addrinfo *addresses;
- struct addrinfo *addr;
- struct addrinfo hint;
- int r;
- char sport[6];
- snprintf(sport, sizeof(sport), "%us", port);
- memset(&hint, 0, sizeof(hint));
- hint.ai_family = AF_UNSPEC;
- hint.ai_socktype = SOCK_STREAM;
- hint.ai_protocol = IPPROTO_TCP;
- hint.ai_flags = AI_ADDRCONFIG;
- r = getaddrinfo(address, sport, &hint, &addresses);
- if (r) {
- return r;
- }
- if (addresses != NULL) {
- *saddr = g_malloc(addresses->ai_addrlen);
- if (!*saddr) {
- r = -1;
- } else {
- memcpy(*saddr, addresses->ai_addr, addresses->ai_addrlen);
- }
- }
- freeaddrinfo(addresses);
- if (r == -1)
- return EAI_SYSTEM;
- return 0;
-}
+enum {
+ STATE_OFFLINE,
+ STATE_PINGED,
+ STATE_ONLINE,
+};
struct friend {
char *name;
char *address;
uint16_t port;
- struct sockaddr *saddr;
+ GInetSocketAddress *saddr;
+ int state;
};
-static int usock;
+static GSocket *usock;
int sock_init(void)
{
- struct sockaddr_in6 sa;
- memset(&sa, 0, sizeof(sa));
- sa.sin6_family = AF_INET6;
- sa.sin6_port = htons(17078);
- memcpy((void *) &sa.sin6_addr, (void *) &in6addr_any, sizeof(in6addr_any));
- usock = socket(AF_INET6, SOCK_DGRAM, 0);
- bind(usock, (struct sockaddr *) &sa, sizeof(sa));
- message_init(usock);
- return 0;
+ GSocketAddress *address;
+ GInetAddress *any_addr;
+ GError *error;
+ int err = 0;
+ any_addr = g_inet_address_new_any(G_SOCKET_FAMILY_IPV6);
+ usock = g_socket_new(G_SOCKET_FAMILY_IPV6, G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP, NULL);
+ address = g_inet_socket_address_new(any_addr, 17078);
+ if (!g_socket_bind(usock, address, TRUE, &error)) {
+ err = error->code;
+ g_error_free(error);
+ }
+ g_object_unref(address);
+ g_object_unref(any_addr);
+ if (!err)
+ message_init(usock);
+ else
+ g_object_unref(usock);
+ return err;
}
int friend_send_message(struct friend *friend, char *buffer, size_t len)
{
- socklen_t sl;
- if (friend->saddr->sa_family == AF_INET)
- sl = sizeof(struct sockaddr_in);
- else if (friend->saddr->sa_family == AF_INET6)
- sl = sizeof(struct sockaddr_in6);
- else
- return -EAFNOSUPPORT;
- sendto(usock, buffer, len, 0, friend->saddr, sl);
+ g_socket_send_to(usock, G_SOCKET_ADDRESS(friend->saddr), buffer, len, NULL, NULL);
return 0;
}
+void friend_timeout(struct friend *friend)
+{
+ if (friend->state == STATE_PINGED) {
+ friend->state = STATE_OFFLINE;
+ }
+}
+
+void friend_got_message(struct friend *friend, char *buffer, size_t len)
+{
+ if (len >= 4 && !strncmp(buffer, "PING", 4)) {
+ pong(friend);
+ } else if (len >= 4 && !strncmp(buffer, "PONG", 4)) {
+ friend->state = STATE_ONLINE;
+ }
+}
+
struct cache {
GList *friends;
};
+static struct cache *ucache;
+
+struct friend *friend_get_by_address(GInetAddress *address, uint16_t port)
+{
+ GList *l;
+ for (l = g_list_first(ucache->friends); l != NULL; l = g_list_next(l)) {
+ struct friend *friend = l->data;
+ if (g_inet_address_equal(g_inet_socket_address_get_address(friend->saddr), address) &&
+ friend->port == port)
+ return friend;
+ }
+ return NULL;
+}
+
int create_cache(struct cache **cache)
{
- *cache = g_slice_new0(struct cache);
+ ucache = *cache = g_slice_new0(struct cache);
(*cache)->friends = NULL;
return 0;
}
struct friend *friend = data;
g_free(friend->name);
g_free(friend->address);
- g_free(friend->saddr);
+ g_object_unref(friend->saddr);
g_slice_free(struct friend, friend);
}
g_slice_free(struct cache, cache);
}
+char * friend_get_name(struct friend *friend)
+{
+ return friend->name;
+}
+
int cache_add_friend(struct cache *cache, char *name, char *address, uint16_t port)
{
struct friend *friend;
+ GInetAddress *addr;
friend = g_slice_new0(struct friend);
friend->name = g_strdup(name);
friend->address = g_strdup(address);
friend->port = port;
- connect_friend(&friend->saddr, friend->address, friend->port);
+ addr = g_inet_address_new_from_string(address);
+ friend->saddr = G_INET_SOCKET_ADDRESS(g_inet_socket_address_new(addr, friend->port));
+ g_object_unref(addr);
cache->friends = g_list_append(cache->friends, friend);
+ ping(friend);
+ friend->state = STATE_PINGED;
return 0;
}