GSocket is done.

[cascardo/gnio.git] / test / test-server.c

diff --git a/test/test-server.c b/test/test-server.c
index 8121cc0..71c7bb1 100644 (file)
--- a/test/test-server.c
+++ b/test/test-server.c
@@ -3,8 +3,8 @@
 #include <gnio/ginetsocketaddress.h>
 #include <gnio/ginet4address.h>
 #include <glib.h>
-#include <glib/gprintf.h>
 #include <sys/socket.h>
+#include <errno.h>
 
 GMainLoop *loop;
 
@@ -42,7 +42,7 @@ accept_callback (GSocket *socket, GAsyncResult *result, gpointer data)
        if (!address)
                g_error (error->message);
 
-       g_printf ("got a new connection from %s:%d\n", g_inet_address_to_string (g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address))), g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)));
+       g_print ("got a new connection from %s:%d\n", g_inet_address_to_string (g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address))), g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)));
 
        g_idle_add (accept_source, (gpointer) socket);
 }
@@ -62,26 +62,55 @@ int main (int argc, char *argv[])
 
        socket = g_socket_new (G_SOCKET_DOMAIN_INET, G_SOCKET_TYPE_STREAM, NULL, NULL);
 
-       if (!g_socket_bind (socket, G_SOCKET_ADDRESS (g_inet_socket_address_new (G_INET_ADDRESS (g_inet4_address_from_string ("127.0.0.1")), 31882)), &error))
+       if (!g_socket_bind (socket, G_SOCKET_ADDRESS (g_inet_socket_address_new (G_INET_ADDRESS (g_inet4_address_from_string ("127.0.0.1")), 31882)), &error)) {
                g_error (error->message);
+               return 0;
+       }
 
        g_socket_listen (socket, 10);
 
-       g_printf ("listening on port 31882...\n");
+       g_print ("listening on port 31882...\n");
 
-       new_socket = g_socket_accept (socket, NULL);
+       new_socket = g_socket_accept (socket, &error);
 
-       if (!new_socket)
+       if (!new_socket) {
                g_error (error->message);
+               return 0;
+       }
 
        address = g_socket_get_remote_address (new_socket, &error);
 
-       if (!address)
+       if (!address) {
                g_error (error->message);
+               return 0;
+       }
+
+       g_print ("got a new connection from %s:%d\n", g_inet_address_to_string (g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address))), g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)));
+
+       while (TRUE) {
+               gchar buffer[128] = { };
+               gssize size;
+
+               if ((size = g_socket_receive (new_socket, buffer, 128, &error)) < 0) {
+                       g_error (error->message);
+                       return 0;
+               }
+
+               if (size == 0)
+                       break;
+
+               g_print ("received %d bytes of data: %s\n", size, buffer);
+
+               if ((size = g_socket_send (new_socket, buffer, size, &error)) < 0) {
+                       g_error (error->message);
+                       return 0;
+               }
 
-       g_printf ("got a new connection from %s:%d\n", g_inet_address_to_string (g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address))), g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (address)));
+               if (size == 0)
+                       break;
+       }
 
-       g_main_loop_run (loop);
+       g_print ("connection closed\n");
 
        return 0;
 }