2 * Copyright (C) 2013 Thadeu Lima de Souza Cascardo <cascardo@cascardo.info>
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.
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.
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.
22 #include <sys/socket.h>
23 #include <netinet/in.h>
24 #include <arpa/inet.h>
41 GInetSocketAddress *saddr;
45 static GSocket *usock;
49 GSocketAddress *address;
50 GInetAddress *any_addr;
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)) {
60 g_object_unref(address);
61 g_object_unref(any_addr);
65 g_object_unref(usock);
69 int friend_send_message(struct friend *friend, char *buffer, size_t len)
71 g_socket_send_to(usock, G_SOCKET_ADDRESS(friend->saddr), buffer, len, NULL, NULL);
75 void friend_timeout(struct friend *friend)
77 if (friend->state == STATE_PINGED) {
78 friend->state = STATE_OFFLINE;
82 void friend_got_message(struct friend *friend, char *buffer, size_t len)
84 if (len >= 4 && !strncmp(buffer, "PING", 4)) {
85 friend->state = STATE_ONLINE;
87 } else if (len >= 4 && !strncmp(buffer, "PONG", 4)) {
88 friend->state = STATE_ONLINE;
92 static void friend_list(gchar **args, GSocketAddress *address);
93 static void friend_add(gchar **args, GSocketAddress *address);
95 struct menu_item cmds[] = {
96 { "list", friend_list },
97 { "add", friend_add },
100 void friend_cmd(gchar **args, GSocketAddress *address)
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);
112 void friend_init(void)
114 struct menu_item *mi;
115 mi = g_malloc(sizeof(*mi));
117 mi->func = friend_cmd;
125 static struct cache *ucache;
127 static void friend_list(gchar **args, GSocketAddress *address)
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);
137 g_socket_send_to(usock, address, buffer, 0, NULL, NULL);
140 static void friend_add(gchar **args, GSocketAddress *address)
146 if (args[2] == NULL || args[3] == NULL || args[4] == NULL)
152 cache_add_friend(ucache, name, addr, port);
155 struct friend *friend_get_by_address(GInetAddress *address, uint16_t port)
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)
167 int create_cache(struct cache **cache)
169 ucache = *cache = g_slice_new0(struct cache);
170 (*cache)->friends = NULL;
174 static void destroy_friend(gpointer data)
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);
183 int destroy_cache(struct cache *cache)
186 g_list_free_full(cache->friends, destroy_friend);
187 g_slice_free(struct cache, cache);
190 char * friend_get_name(struct friend *friend)
195 int cache_add_friend(struct cache *cache, char *name, char *address, uint16_t port)
197 struct friend *friend;
199 friend = g_slice_new0(struct friend);
200 friend->name = g_strdup(name);
201 friend->address = g_strdup(address);
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);
208 friend->state = STATE_PINGED;
212 int load_cache(struct cache *cache, char *fname)
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++) {
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);
232 g_key_file_free(file);
236 int store_cache(struct cache *cache, char *fname)
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);
250 contents = g_key_file_to_data(file, &len, NULL);
251 g_file_set_contents(fname, contents, len, NULL);
253 g_key_file_free(file);