friend.c

   1 /*
   2  *  Copyright (C) 2013  Thadeu Lima de Souza Cascardo <cascardo@cascardo.info>
   3  *
   4  *  This program is free software; you can redistribute it and/or modify
   5  *  it under the terms of the GNU General Public License as published by
   6  *  the Free Software Foundation; either version 3 of the License, or
   7  *  (at your option) any later version.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License along
  15  *  with this program; if not, write to the Free Software Foundation, Inc.,
  16  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include "friend.h"
  20 #include <string.h>
  21 #include <glib.h>
  22 #include <sys/socket.h>
  23 #include <netinet/in.h>
  24 #include <arpa/inet.h>
  25 #include <netdb.h>
  26 #include <errno.h>
  27 #include <stdio.h>
  28 #include "message.h"
  29 #include "menu.h"
  30
  31 enum {
  32         STATE_OFFLINE,
  33         STATE_PINGED,
  34         STATE_ONLINE,
  35 };
  36
  37 struct friend {
  38         char *name;
  39         char *address;
  40         uint16_t port;
  41         GInetSocketAddress *saddr;
  42         int state;
  43 };
  44
  45 static GSocket *usock;
  46
  47 int sock_init(void)
  48 {
  49         GSocketAddress *address;
  50         GInetAddress *any_addr;
  51         GError *error;
  52         int err = 0;
  53         any_addr = g_inet_address_new_any(G_SOCKET_FAMILY_IPV6);
  54         usock = g_socket_new(G_SOCKET_FAMILY_IPV6, G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP, NULL);
  55         address = g_inet_socket_address_new(any_addr, 17078);
  56         if (!g_socket_bind(usock, address, TRUE, &error)) {
  57                 err = error->code;
  58                 g_error_free(error);
  59         }
  60         g_object_unref(address);
  61         g_object_unref(any_addr);
  62         if (!err)
  63                 message_init(usock);
  64         else
  65                 g_object_unref(usock);
  66         return err;
  67 }
  68
  69 int friend_send_message(struct friend *friend, char *buffer, size_t len)
  70 {
  71         g_socket_send_to(usock, G_SOCKET_ADDRESS(friend->saddr), buffer, len, NULL, NULL);
  72         return 0;
  73 }
  74
  75 void friend_timeout(struct friend *friend)
  76 {
  77         if (friend->state == STATE_PINGED) {
  78                 friend->state = STATE_OFFLINE;
  79         }
  80 }
  81
  82 void friend_got_message(struct friend *friend, char *buffer, size_t len)
  83 {
  84         if (len >= 4 && !strncmp(buffer, "PING", 4)) {
  85                 friend->state = STATE_ONLINE;
  86                 pong(friend);
  87         } else if (len >= 4 && !strncmp(buffer, "PONG", 4)) {
  88                 friend->state = STATE_ONLINE;
  89         }
  90 }
  91
  92 static void friend_list(gchar **args, GSocketAddress *address);
  93 static void friend_add(gchar **args, GSocketAddress *address);
  94
  95 struct menu_item cmds[] = {
  96         { "list", friend_list },
  97         { "add", friend_add },
  98 };
  99
 100 void friend_cmd(gchar **args, GSocketAddress *address)
 101 {
 102         int i;
 103         if (args[1] == NULL)
 104                 return;
 105         for (i = 0; i < sizeof(cmds)/sizeof(cmds[0]); i++) {
 106                 if (!strcmp(args[1], cmds[i].cmd)) {
 107                         cmds[i].func(args, address);
 108                 }
 109         }
 110 }
 111
 112 void friend_init(void)
 113 {
 114         struct menu_item *mi;
 115         mi = g_malloc(sizeof(*mi));
 116         mi->cmd = "friend";
 117         mi->func = friend_cmd;
 118         menu_add(mi);
 119 }
 120
 121 struct cache {
 122         GList *friends;
 123 };
 124
 125 static struct cache *ucache;
 126
 127 static void friend_list(gchar **args, GSocketAddress *address)
 128 {
 129         char *buffer;
 130         GList *l;
 131         for (l = g_list_first(ucache->friends); l != NULL; l = g_list_next(l)) {
 132                 struct friend *friend = l->data;
 133                 buffer = g_strdup_printf("%s\n", friend->name);
 134                 g_socket_send_to(usock, address, buffer, strlen(buffer), NULL, NULL);
 135                 g_free(buffer);
 136         }
 137         g_socket_send_to(usock, address, buffer, 0, NULL, NULL);
 138 }
 139
 140 static void friend_add(gchar **args, GSocketAddress *address)
 141 {
 142         char *name;
 143         char *addr;
 144         char *sport;
 145         uint16_t port;
 146         if (args[2] == NULL || args[3] == NULL || args[4] == NULL)
 147                 return;
 148         name = args[2];
 149         addr = args[3];
 150         sport = args[4];
 151         port = atoi(sport);
 152         cache_add_friend(ucache, name, addr, port);
 153 }
 154
 155 struct friend *friend_get_by_address(GInetAddress *address, uint16_t port)
 156 {
 157         GList *l;
 158         for (l = g_list_first(ucache->friends); l != NULL; l = g_list_next(l)) {
 159                 struct friend *friend = l->data;
 160                 if (g_inet_address_equal(g_inet_socket_address_get_address(friend->saddr), address) &&
 161                     friend->port == port)
 162                         return friend;
 163         }
 164         return NULL;
 165 }
 166
 167 int create_cache(struct cache **cache)
 168 {
 169         ucache = *cache = g_slice_new0(struct cache);
 170         (*cache)->friends = NULL;
 171         return 0;
 172 }
 173
 174 static void destroy_friend(gpointer data)
 175 {
 176         struct friend *friend = data;
 177         g_free(friend->name);
 178         g_free(friend->address);
 179         g_object_unref(friend->saddr);
 180         g_slice_free(struct friend, friend);
 181 }
 182
 183 int destroy_cache(struct cache *cache)
 184 {
 185         if (cache->friends)
 186                 g_list_free_full(cache->friends, destroy_friend);
 187         g_slice_free(struct cache, cache);
 188 }
 189
 190 char * friend_get_name(struct friend *friend)
 191 {
 192         return friend->name;
 193 }
 194
 195 int cache_add_friend(struct cache *cache, char *name, char *address, uint16_t port)
 196 {
 197         struct friend *friend;
 198         GInetAddress *addr;
 199         friend = g_slice_new0(struct friend);
 200         friend->name = g_strdup(name);
 201         friend->address = g_strdup(address);
 202         friend->port = port;
 203         addr = g_inet_address_new_from_string(address);
 204         friend->saddr = G_INET_SOCKET_ADDRESS(g_inet_socket_address_new(addr, friend->port));
 205         g_object_unref(addr);
 206         cache->friends = g_list_append(cache->friends, friend);
 207         ping(friend);
 208         friend->state = STATE_PINGED;
 209         return 0;
 210 }
 211
 212 int load_cache(struct cache *cache, char *fname)
 213 {
 214         GKeyFile *file;
 215         gchar **groups;
 216         gchar **group;
 217         file = g_key_file_new();
 218         g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL);
 219         groups = g_key_file_get_groups(file, NULL);
 220         for (group = groups; *group != NULL; group++) {
 221                 gchar *name;
 222                 gchar *address;
 223                 uint16_t port;
 224                 name = g_key_file_get_value(file, *group, "name", NULL);
 225                 address = g_key_file_get_value(file, *group, "address", NULL);
 226                 port = g_key_file_get_integer(file, *group, "port", NULL);
 227                 cache_add_friend(cache, name, address, port);
 228                 g_free(name);
 229                 g_free(address);
 230         }
 231         g_strfreev(groups);
 232         g_key_file_free(file);
 233         return 0;
 234 }
 235
 236 int store_cache(struct cache *cache, char *fname)
 237 {
 238         GKeyFile *file;
 239         GList *f;
 240         gchar *contents;
 241         gssize len;
 242         file = g_key_file_new();
 243         g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL);
 244         for (f = g_list_first(cache->friends); f != NULL; f = g_list_next(f)) {
 245                 struct friend *friend = f->data;
 246                 g_key_file_set_value(file, friend->name, "name", friend->name);
 247                 g_key_file_set_value(file, friend->name, "address", friend->address);
 248                 g_key_file_set_integer(file, friend->name, "port", friend->port);
 249         }
 250         contents = g_key_file_to_data(file, &len, NULL);
 251         g_file_set_contents(fname, contents, len, NULL);
 252         g_free(contents);
 253         g_key_file_free(file);
 254         return 0;
 255 }