*/
#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;
}
projects / cascardo / f2fchat.git / blobdiff