message.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 "message.h"
  20 #include "friend.h"
  21 #include <glib.h>
  22 #include <gio/gio.h>
  23
  24 #include <stdio.h>
  25
  26 gboolean ping_timeout(gpointer data)
  27 {
  28         struct friend *friend = data;
  29         friend_timeout(friend);
  30         return G_SOURCE_REMOVE;
  31 }
  32
  33 static GIOChannel *uchannel;
  34 static GSocket *gusock;
  35
  36 static void command(char *buffer, size_t len)
  37 {
  38         printf("message from loopback: %d %.*s\n", len, len, buffer);
  39 }
  40
  41 gboolean message_incoming(GIOChannel *channel, GIOCondition cond, gpointer data)
  42 {
  43         size_t len;
  44         char *buffer;
  45         GSocketAddress *address;
  46         GInetAddress *iaddress;
  47         if (!gusock) {
  48                 gusock = g_socket_new_from_fd(g_io_channel_unix_get_fd(channel), NULL);
  49         }
  50         len = g_socket_get_available_bytes(gusock);
  51         buffer = g_malloc(len);
  52         len = g_socket_receive_from(gusock, &address, buffer, len, NULL, NULL);
  53         iaddress = g_inet_socket_address_get_address(G_INET_SOCKET_ADDRESS(address));
  54         if (g_inet_address_get_is_loopback(iaddress)) {
  55                 command(buffer, len);
  56         } else {
  57                 struct friend *friend;
  58                 uint16_t port = g_inet_socket_address_get_port(G_INET_SOCKET_ADDRESS(address));
  59                 friend = friend_get_by_address(iaddress, port);
  60                 if (friend) {
  61                         printf("got message from %s\n", friend_get_name(friend));
  62                         friend_got_message(friend, buffer, len);
  63                 } else {
  64                         printf("could not find friend from address %s\n", g_inet_address_to_string(iaddress));
  65                 }
  66         }
  67         g_object_unref(address);
  68         g_free(buffer);
  69         return TRUE;
  70 }
  71
  72 int message_init(GSocket *sock)
  73 {
  74         uchannel = g_io_channel_unix_new(g_socket_get_fd(sock));
  75         g_io_add_watch(uchannel, G_IO_IN, message_incoming, NULL);
  76         return 0;
  77 }
  78
  79 int ping(struct friend *friend)
  80 {
  81         int err;
  82         char ping[5] = "PING";
  83         err = friend_send_message(friend, ping, 4);
  84         if (!err)
  85                 g_timeout_add(2000, ping_timeout, friend);
  86         return err;
  87 }
  88
  89 int pong(struct friend *friend)
  90 {
  91         char pong[5] = "PONG";
  92         return friend_send_message(friend, pong, 4);
  93 }