#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 = 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;
}
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);
}
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;
}
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);
{
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;
}