Respond pings with pongs.

[cascardo/f2fchat.git] / friend.c

diff --git a/friend.c b/friend.c
index fd79643..03d306d 100644 (file)
--- a/friend.c
+++ b/friend.c
@@ -27,11 +27,18 @@
 #include <stdio.h>
 #include "message.h"
 
+enum {
+       STATE_OFFLINE,
+       STATE_PINGED,
+       STATE_ONLINE,
+};
+
 struct friend {
        char *name;
        char *address;
        uint16_t port;
        GInetSocketAddress *saddr;
+       int state;
 };
 
 static GSocket *usock;
@@ -40,14 +47,22 @@ int sock_init(void)
 {
        GSocketAddress *address;
        GInetAddress *any_addr;
+       GError *error;
+       int err = 0;
        any_addr = g_inet_address_new_any(G_SOCKET_FAMILY_IPV6);
        usock = g_socket_new(G_SOCKET_FAMILY_IPV6, G_SOCKET_TYPE_DATAGRAM, G_SOCKET_PROTOCOL_UDP, NULL);
        address = g_inet_socket_address_new(any_addr, 17078);
-       g_socket_bind(usock, address, TRUE, NULL);
-       g_object_unref(any_addr);
+       if (!g_socket_bind(usock, address, TRUE, &error)) {
+               err = error->code;
+               g_error_free(error);
+       }
        g_object_unref(address);
-       message_init(usock);
-       return 0;
+       g_object_unref(any_addr);
+       if (!err)
+               message_init(usock);
+       else
+               g_object_unref(usock);
+       return err;
 }
 
 int friend_send_message(struct friend *friend, char *buffer, size_t len)
@@ -56,6 +71,22 @@ int friend_send_message(struct friend *friend, char *buffer, size_t len)
        return 0;
 }
 
+void friend_timeout(struct friend *friend)
+{
+       if (friend->state == STATE_PINGED) {
+               friend->state = STATE_OFFLINE;
+       }
+}
+
+void friend_got_message(struct friend *friend, char *buffer, size_t len)
+{
+       if (len >= 4 && !strncmp(buffer, "PING", 4)) {
+               pong(friend);
+       } else if (len >= 4 && !strncmp(buffer, "PONG", 4)) {
+               friend->state = STATE_ONLINE;
+       }
+}
+
 struct cache {
        GList *friends;
 };
@@ -115,6 +146,7 @@ int cache_add_friend(struct cache *cache, char *name, char *address, uint16_t po
        g_object_unref(addr);
        cache->friends = g_list_append(cache->friends, friend);
        ping(friend);
+       friend->state = STATE_PINGED;
        return 0;
 }