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         if (len <= 0) {
  52                 goto out;
  53         }
  54         buffer = g_malloc(len);
  55         len = g_socket_receive_from(gusock, &address, buffer, len, NULL, NULL);
  56         iaddress = g_inet_socket_address_get_address(G_INET_SOCKET_ADDRESS(address));
  57         if (g_inet_address_get_is_loopback(iaddress)) {
  58                 command(buffer, len);
  59         } else {
  60                 struct friend *friend;
  61                 uint16_t port = g_inet_socket_address_get_port(G_INET_SOCKET_ADDRESS(address));
  62                 friend = friend_get_by_address(iaddress, port);
  63                 if (friend) {
  64                         printf("got message from %s\n", friend_get_name(friend));
  65                         friend_got_message(friend, buffer, len);
  66                 } else {
  67                         printf("could not find friend from address %s\n", g_inet_address_to_string(iaddress));
  68                 }
  69         }
  70         g_object_unref(address);
  71         g_free(buffer);
  72 out:
  73         return TRUE;
  74 }
  75
  76 int message_init(GSocket *sock)
  77 {
  78         uchannel = g_io_channel_unix_new(g_socket_get_fd(sock));
  79         g_io_add_watch(uchannel, G_IO_IN, message_incoming, NULL);
  80         return 0;
  81 }
  82
  83 int ping(struct friend *friend)
  84 {
  85         int err;
  86         char ping[5] = "PING";
  87         err = friend_send_message(friend, ping, 4);
  88         if (!err)
  89                 g_timeout_add(2000, ping_timeout, friend);
  90         return err;
  91 }
  92
  93 int pong(struct friend *friend)
  94 {
  95         char pong[5] = "PONG";
  96         return friend_send_message(friend, pong, 4);
  97 }