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
  30 struct friend {
  31         char *name;
  32         char *address;
  33         uint16_t port;
  34         GInetSocketAddress *saddr;
  35 };
  36
  37 static GSocket *usock;
  38
  39 int sock_init(void)
  40 {
  41         GSocketAddress *address;
  42         GInetAddress *any_addr;
  43         any_addr = g_inet_address_new_any(G_SOCKET_FAMILY_IPV6);
  44         usock = g_socket_new(G_SOCKET_FAMILY_IPV6, G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP, NULL);
  45         address = g_inet_socket_address_new(any_addr, 17078);
  46         g_socket_bind(usock, address, TRUE, NULL);
  47         g_object_unref(any_addr);
  48         g_object_unref(address);
  49         message_init(usock);
  50         return 0;
  51 }
  52
  53 int friend_send_message(struct friend *friend, char *buffer, size_t len)
  54 {
  55         g_socket_send_to(usock, G_SOCKET_ADDRESS(friend->saddr), buffer, len, NULL, NULL);
  56         return 0;
  57 }
  58
  59 struct cache {
  60         GList *friends;
  61 };
  62
  63 int create_cache(struct cache **cache)
  64 {
  65         *cache = g_slice_new0(struct cache);
  66         (*cache)->friends = NULL;
  67         return 0;
  68 }
  69
  70 static void destroy_friend(gpointer data)
  71 {
  72         struct friend *friend = data;
  73         g_free(friend->name);
  74         g_free(friend->address);
  75         g_object_unref(friend->saddr);
  76         g_slice_free(struct friend, friend);
  77 }
  78
  79 int destroy_cache(struct cache *cache)
  80 {
  81         if (cache->friends)
  82                 g_list_free_full(cache->friends, destroy_friend);
  83         g_slice_free(struct cache, cache);
  84 }
  85
  86 int cache_add_friend(struct cache *cache, char *name, char *address, uint16_t port)
  87 {
  88         struct friend *friend;
  89         GInetAddress *addr;
  90         friend = g_slice_new0(struct friend);
  91         friend->name = g_strdup(name);
  92         friend->address = g_strdup(address);
  93         friend->port = port;
  94         addr = g_inet_address_new_from_string(address);
  95         friend->saddr = G_INET_SOCKET_ADDRESS(g_inet_socket_address_new(addr, friend->port));
  96         g_object_unref(addr);
  97         cache->friends = g_list_append(cache->friends, friend);
  98         return 0;
  99 }
 100
 101 int load_cache(struct cache *cache, char *fname)
 102 {
 103         GKeyFile *file;
 104         gchar **groups;
 105         gchar **group;
 106         file = g_key_file_new();
 107         g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL);
 108         groups = g_key_file_get_groups(file, NULL);
 109         for (group = groups; *group != NULL; group++) {
 110                 gchar *name;
 111                 gchar *address;
 112                 uint16_t port;
 113                 name = g_key_file_get_value(file, *group, "name", NULL);
 114                 address = g_key_file_get_value(file, *group, "address", NULL);
 115                 port = g_key_file_get_integer(file, *group, "port", NULL);
 116                 cache_add_friend(cache, name, address, port);
 117                 g_free(name);
 118                 g_free(address);
 119         }
 120         g_strfreev(groups);
 121         g_key_file_free(file);
 122         return 0;
 123 }
 124
 125 int store_cache(struct cache *cache, char *fname)
 126 {
 127         GKeyFile *file;
 128         GList *f;
 129         gchar *contents;
 130         gssize len;
 131         file = g_key_file_new();
 132         g_key_file_load_from_file(file, fname, G_KEY_FILE_KEEP_COMMENTS | G_KEY_FILE_KEEP_TRANSLATIONS, NULL);
 133         for (f = g_list_first(cache->friends); f != NULL; f = g_list_next(f)) {
 134                 struct friend *friend = f->data;
 135                 g_key_file_set_value(file, friend->name, "name", friend->name);
 136                 g_key_file_set_value(file, friend->name, "address", friend->address);
 137                 g_key_file_set_integer(file, friend->name, "port", friend->port);
 138         }
 139         contents = g_key_file_to_data(file, &len, NULL);
 140         g_file_set_contents(fname, contents, len, NULL);
 141         g_free(contents);
 142         g_key_file_free(file);
 143         return 0;
 144 }