X-Git-Url: http://git.cascardo.eti.br/?p=cascardo%2Ff2fchat.git;a=blobdiff_plain;f=friend.c;h=60569c1e9cb645ad10d3e429a7a924c36a21ccec;hp=753a5881a15b71175f1f3e3e5b0a9fa19754160c;hb=HEAD;hpb=e6f19319684b4b1755b397833b229c9a5970724a

diff --git a/friend.c b/friend.c
index 753a588..60569c1 100644
--- a/friend.c
+++ b/friend.c
@@ -17,143 +17,239 @@
  */
 
 #include "friend.h"
-#include <stdio.h>
-#include <errno.h>
-#include <stdlib.h>
 #include <string.h>
+#include <glib.h>
 #include <sys/socket.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
 #include <netdb.h>
+#include <errno.h>
+#include <stdio.h>
+#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 = 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 {
-	int nfriends;
-	struct friend *friends;
+	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 = malloc(sizeof(*cache));
-	if (!*cache)
-		return -errno;
-	(*cache)->nfriends = 0;
+	ucache = *cache = g_slice_new0(struct cache);
 	(*cache)->friends = NULL;
 	return 0;
 }
 
+static void destroy_friend(gpointer data)
+{
+	struct friend *friend = data;
+	g_free(friend->name);
+	g_free(friend->address);
+	g_object_unref(friend->saddr);
+	g_slice_free(struct friend, friend);
+}
+
 int destroy_cache(struct cache *cache)
 {
 	if (cache->friends)
-		free(cache->friends);
-	free(cache);
-}
-
-int cache_add_friend(struct cache *cache, char *friend, char *address, char *port)
-{
-	struct friend *tfriend;
-	if (!cache->friends) {
-		cache->friends = malloc(sizeof(struct friend));
-		cache->nfriends = 1;
-	} else {
-		struct friend *new_friends;
-		cache->nfriends++;
-		new_friends = realloc(cache->friends, cache->nfriends * sizeof(struct friend));
-		if (!new_friends) {
-			cache->nfriends--;
-			return -errno;
-		}
-		cache->friends = new_friends;
-	}
-	tfriend = &cache->friends[cache->nfriends - 1];
-	tfriend->name = friend;
-	tfriend->address = address;
-	tfriend->port = port;
-	connect_friend(&tfriend->saddr, tfriend->address, tfriend->port);
+		g_list_free_full(cache->friends, destroy_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;
+	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;
 }
 
 int load_cache(struct cache *cache, char *fname)
 {
-	FILE *file;
-	int err = 0;
-	char *buffer = NULL;
-	size_t len = 0;
-	int r;
-	file = fopen(fname, "r");
-	if (!file)
-		return -errno;
-	while ((r = getline(&buffer, &len, file)) > 0) {
-		char *name;
-		char *address;
-		char *port;
-		char *end;
-		name = buffer;
-		address = name;
-		while (*++address != '\t');
-		*address++ = '\0';
-		port = address;
-		while (*++port != '\t');
-		*port++ = '\0';
-		end = port;
-		while (*++end != '\n');
-		*end = '\0';
-		fprintf(file, "%s\t%s\t%s\n", name, address, port);
-		cache_add_friend(cache, strdup(name), strdup(address), strdup(port));
+	GKeyFile *file;
+	gchar **groups;
+	gchar **group;
+	file = g_key_file_new();
+	g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL);
+	groups = g_key_file_get_groups(file, NULL);
+	for (group = groups; *group != NULL; group++) {
+		gchar *name;
+		gchar *address;
+		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_integer(file, *group, "port", NULL);
+		cache_add_friend(cache, name, address, port);
+		g_free(name);
+		g_free(address);
 	}
-out:
-	fclose(file);
-	return err;
+	g_strfreev(groups);
+	g_key_file_free(file);
+	return 0;
 }
 
 int store_cache(struct cache *cache, char *fname)
 {
-	FILE *file;
-	int err = 0;
-	int i;
-	file = fopen(fname, "w");
-	if (!file)
-		return -errno;
-	for (i = 0; i < cache->nfriends; i++) {
-		struct friend *friend = &cache->friends[i];
-		fprintf(file, "%s\t%s\t%s\n", friend->name, friend->address, friend->port);
+	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_integer(file, friend->name, "port", friend->port);
 	}
-out:
-	fclose(file);
-	return err;
+	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;
 }