X-Git-Url: http://git.cascardo.eti.br/?p=cascardo%2Ff2fchat.git;a=blobdiff_plain;f=friend.c;h=60569c1e9cb645ad10d3e429a7a924c36a21ccec;hp=dff03e265f4529332597f0e3188670571e159034;hb=HEAD;hpb=966597cb2c97b429ed9de6202b466981ad80cbba diff --git a/friend.c b/friend.c index dff03e2..60569c1 100644 --- a/friend.c +++ b/friend.c @@ -23,50 +23,150 @@ #include #include #include +#include +#include +#include "message.h" +#include "menu.h" -static int connect_friend(struct sockaddr **saddr, char *address, char *port) -{ - struct addrinfo *addresses; - struct addrinfo *addr; - struct addrinfo hint; - int r; - 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, port, &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; - char *port; - struct sockaddr *saddr; + uint16_t port; + GInetSocketAddress *saddr; + int state; +}; + +static GSocket *usock; + +int sock_init(void) +{ + 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) +{ + 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)) { + friend->state = STATE_ONLINE; + pong(friend); + } else if (len >= 4 && !strncmp(buffer, "PONG", 4)) { + friend->state = STATE_ONLINE; + } +} + +static void friend_list(gchar **args, GSocketAddress *address); +static void friend_add(gchar **args, GSocketAddress *address); + +struct menu_item cmds[] = { + { "list", friend_list }, + { "add", friend_add }, }; +void friend_cmd(gchar **args, GSocketAddress *address) +{ + int i; + if (args[1] == NULL) + return; + for (i = 0; i < sizeof(cmds)/sizeof(cmds[0]); i++) { + if (!strcmp(args[1], cmds[i].cmd)) { + cmds[i].func(args, address); + } + } +} + +void friend_init(void) +{ + struct menu_item *mi; + mi = g_malloc(sizeof(*mi)); + mi->cmd = "friend"; + mi->func = friend_cmd; + menu_add(mi); +} + struct cache { GList *friends; }; +static struct cache *ucache; + +static void friend_list(gchar **args, GSocketAddress *address) +{ + char *buffer; + GList *l; + for (l = g_list_first(ucache->friends); l != NULL; l = g_list_next(l)) { + struct friend *friend = l->data; + buffer = g_strdup_printf("%s\n", friend->name); + g_socket_send_to(usock, address, buffer, strlen(buffer), NULL, NULL); + g_free(buffer); + } + g_socket_send_to(usock, address, buffer, 0, NULL, NULL); +} + +static void friend_add(gchar **args, GSocketAddress *address) +{ + char *name; + char *addr; + char *sport; + uint16_t port; + if (args[2] == NULL || args[3] == NULL || args[4] == NULL) + return; + name = args[2]; + addr = args[3]; + sport = args[4]; + port = atoi(sport); + cache_add_friend(ucache, name, addr, port); +} + +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; } @@ -76,8 +176,7 @@ static void destroy_friend(gpointer data) struct friend *friend = data; g_free(friend->name); g_free(friend->address); - g_free(friend->port); - g_free(friend->saddr); + g_object_unref(friend->saddr); g_slice_free(struct friend, friend); } @@ -88,15 +187,25 @@ int destroy_cache(struct cache *cache) g_slice_free(struct cache, cache); } -int cache_add_friend(struct cache *cache, char *name, char *address, char *port) +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 = g_strdup(port); - connect_friend(&friend->saddr, friend->address, friend->port); - g_list_append(cache->friends, friend); + friend->port = 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; } @@ -111,14 +220,13 @@ int load_cache(struct cache *cache, char *fname) for (group = groups; *group != NULL; group++) { gchar *name; gchar *address; - gchar *port; + uint16_t port; name = g_key_file_get_value(file, *group, "name", NULL); address = g_key_file_get_value(file, *group, "address", NULL); - port = g_key_file_get_value(file, *group, "port", NULL); + port = g_key_file_get_integer(file, *group, "port", NULL); cache_add_friend(cache, name, address, port); g_free(name); g_free(address); - g_free(port); } g_strfreev(groups); g_key_file_free(file); @@ -129,14 +237,19 @@ int store_cache(struct cache *cache, char *fname) { GKeyFile *file; GList *f; + gchar *contents; + gssize len; file = g_key_file_new(); g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL); for (f = g_list_first(cache->friends); f != NULL; f = g_list_next(f)) { struct friend *friend = f->data; g_key_file_set_value(file, friend->name, "name", friend->name); g_key_file_set_value(file, friend->name, "address", friend->address); - g_key_file_set_value(file, friend->name, "port", friend->port); + g_key_file_set_integer(file, friend->name, "port", friend->port); } + contents = g_key_file_to_data(file, &len, NULL); + g_file_set_contents(fname, contents, len, NULL); + g_free(contents); g_key_file_free(file); return 0; }