gio/tests/socket-server.c

   1 #include <gio/gio.h>
   2 #include <gio/gunixsocketaddress.h>
   3 #include <glib.h>
   4 #include <stdlib.h>
   5 #include <string.h>
   6
   7 #include "socket-common.c"
   8
   9 GMainLoop *loop;
  10
  11 int port = 7777;
  12 gboolean verbose = FALSE;
  13 gboolean dont_reuse_address = FALSE;
  14 gboolean non_blocking = FALSE;
  15 gboolean use_udp = FALSE;
  16 gboolean use_source = FALSE;
  17 int cancel_timeout = 0;
  18 gboolean unix_socket = FALSE;
  19
  20 static GOptionEntry cmd_entries[] = {
  21   {"port", 'p', 0, G_OPTION_ARG_INT, &port,
  22    "Local port to bind to", NULL},
  23   {"cancel", 'c', 0, G_OPTION_ARG_INT, &cancel_timeout,
  24    "Cancel any op after the specified amount of seconds", NULL},
  25   {"udp", 'u', 0, G_OPTION_ARG_NONE, &use_udp,
  26    "Use udp instead of tcp", NULL},
  27   {"verbose", 'v', 0, G_OPTION_ARG_NONE, &verbose,
  28    "Be verbose", NULL},
  29   {"no-reuse", 0, 0, G_OPTION_ARG_NONE, &dont_reuse_address,
  30    "Don't SOADDRREUSE", NULL},
  31   {"non-blocking", 'n', 0, G_OPTION_ARG_NONE, &non_blocking,
  32    "Enable non-blocking i/o", NULL},
  33   {"use-source", 's', 0, G_OPTION_ARG_NONE, &use_source,
  34    "Use GSource to wait for non-blocking i/o", NULL},
  35   {"unix", 'U', 0, G_OPTION_ARG_NONE, &unix_socket,
  36    "Use a unix socket instead of IP", NULL},
  37   {NULL}
  38 };
  39
  40 static gboolean
  41 source_ready (gpointer data,
  42               GIOCondition condition)
  43 {
  44   g_main_loop_quit (loop);
  45   return FALSE;
  46 }
  47
  48 static void
  49 ensure_condition (GSocket *socket,
  50                   const char *where,
  51                   GCancellable *cancellable,
  52                   GIOCondition condition)
  53 {
  54   GError *error = NULL;
  55   GSource *source;
  56
  57   if (!non_blocking)
  58     return;
  59
  60   if (use_source)
  61     {
  62       source = g_socket_create_source (socket,
  63                                        condition,
  64                                        cancellable);
  65       g_source_set_callback (source,
  66                              (GSourceFunc) source_ready,
  67                              NULL, NULL);
  68       g_source_attach (source, NULL);
  69       g_source_unref (source);
  70       g_main_loop_run (loop);
  71     }
  72   else
  73     {
  74       if (!g_socket_condition_wait (socket, condition, cancellable, &error))
  75         {
  76           g_printerr ("condition wait error for %s: %s\n",
  77                       where,
  78                       error->message);
  79           exit (1);
  80         }
  81     }
  82 }
  83
  84 static gpointer
  85 cancel_thread (gpointer data)
  86 {
  87   GCancellable *cancellable = data;
  88
  89   g_usleep (1000*1000*cancel_timeout);
  90   g_print ("Cancelling\n");
  91   g_cancellable_cancel (cancellable);
  92   return NULL;
  93 }
  94
  95 int
  96 main (int argc,
  97       char *argv[])
  98 {
  99   GSocket *socket, *new_socket, *recv_socket;
 100   GSocketAddress *src_address;
 101   GSocketAddress *address;
 102   GSocketType socket_type;
 103   GSocketFamily socket_family;
 104   GError *error = NULL;
 105   GOptionContext *context;
 106   GCancellable *cancellable;
 107   char *display_addr;
 108
 109   g_thread_init (NULL);
 110
 111   g_type_init ();
 112
 113   context = g_option_context_new (" - Test GSocket server stuff");
 114   g_option_context_add_main_entries (context, cmd_entries, NULL);
 115   if (!g_option_context_parse (context, &argc, &argv, &error))
 116     {
 117       g_printerr ("%s: %s\n", argv[0], error->message);
 118       return 1;
 119     }
 120
 121   if (unix_socket && argc != 2)
 122     {
 123       g_printerr ("%s: %s\n", argv[0], "Need to specify unix socket name");
 124       return 1;
 125     }
 126
 127   if (cancel_timeout)
 128     {
 129       cancellable = g_cancellable_new ();
 130       g_thread_create (cancel_thread, cancellable, FALSE, NULL);
 131     }
 132   else
 133     {
 134       cancellable = NULL;
 135     }
 136
 137   loop = g_main_loop_new (NULL, FALSE);
 138
 139   if (use_udp)
 140     socket_type = G_SOCKET_TYPE_DATAGRAM;
 141   else
 142     socket_type = G_SOCKET_TYPE_STREAM;
 143
 144   if (unix_socket)
 145     socket_family = G_SOCKET_FAMILY_UNIX;
 146   else
 147     socket_family = G_SOCKET_FAMILY_IPV4;
 148
 149   socket = g_socket_new (socket_family, socket_type, 0, &error);
 150
 151   if (socket == NULL)
 152     {
 153       g_printerr ("%s: %s\n", argv[0], error->message);
 154       return 1;
 155     }
 156
 157   if (non_blocking)
 158     g_socket_set_blocking (socket, FALSE);
 159
 160   if (unix_socket)
 161     {
 162       src_address = socket_address_from_string (argv[1]);
 163       if (src_address == NULL)
 164         {
 165           g_printerr ("%s: Could not parse '%s' as unix socket name\n", argv[0], argv[1]);
 166           return 1;
 167         }
 168     }
 169   else
 170     {
 171       src_address = g_inet_socket_address_new (g_inet_address_new_any (G_SOCKET_FAMILY_IPV4), port);
 172     }
 173
 174   if (!g_socket_bind (socket, src_address, !dont_reuse_address, &error))
 175     {
 176       g_printerr ("Can't bind socket: %s\n", error->message);
 177       return 1;
 178     }
 179   g_object_unref (src_address);
 180
 181   if (!use_udp)
 182     {
 183       if (!g_socket_listen (socket, &error))
 184         {
 185           g_printerr ("Can't listen on socket: %s\n", error->message);
 186           return 1;
 187         }
 188
 189       address = g_socket_get_local_address (socket, &error);
 190       if (!address)
 191         {
 192           g_printerr ("Error getting local address: %s\n",
 193                       error->message);
 194           return 1;
 195         }
 196       display_addr = socket_address_to_string (address);
 197       g_print ("listening on %s...\n", display_addr);
 198       g_free (display_addr);
 199
 200       ensure_condition (socket, "accept", cancellable, G_IO_IN);
 201       new_socket = g_socket_accept (socket, cancellable, &error);
 202       if (!new_socket)
 203         {
 204           g_printerr ("Error accepting socket: %s\n",
 205                       error->message);
 206           return 1;
 207         }
 208
 209       if (non_blocking)
 210         g_socket_set_blocking (new_socket, FALSE);
 211
 212       address = g_socket_get_remote_address (new_socket, &error);
 213       if (!address)
 214         {
 215           g_printerr ("Error getting remote address: %s\n",
 216                       error->message);
 217           return 1;
 218         }
 219
 220       display_addr = socket_address_to_string (address);
 221       g_print ("got a new connection from %s\n", display_addr);
 222       g_free(display_addr);
 223       g_object_unref (address);
 224
 225       recv_socket = new_socket;
 226     }
 227   else
 228     {
 229       recv_socket = socket;
 230       new_socket = NULL;
 231     }
 232
 233
 234   while (TRUE)
 235     {
 236       gchar buffer[4096] = { };
 237       gssize size;
 238       gsize to_send;
 239
 240       ensure_condition (recv_socket, "receive", cancellable, G_IO_IN);
 241       if (use_udp)
 242         size = g_socket_receive_from (recv_socket, &address,
 243                                       buffer, sizeof buffer,
 244                                       cancellable, &error);
 245       else
 246         size = g_socket_receive (recv_socket, buffer, sizeof buffer,
 247                                  cancellable, &error);
 248
 249       if (size < 0)
 250         {
 251           g_printerr ("Error receiving from socket: %s\n",
 252                       error->message);
 253           return 1;
 254         }
 255
 256       if (size == 0)
 257         break;
 258
 259       g_print ("received %" G_GSSIZE_FORMAT " bytes of data", size);
 260       if (use_udp)
 261         g_print (" from %s", socket_address_to_string (address));
 262       g_print ("\n");
 263
 264       if (verbose)
 265         g_print ("-------------------------\n"
 266                  "%.*s\n"
 267                  "-------------------------\n",
 268                  (int)size, buffer);
 269
 270       to_send = size;
 271
 272       while (to_send > 0)
 273         {
 274           ensure_condition (recv_socket, "send", cancellable, G_IO_OUT);
 275           if (use_udp)
 276             size = g_socket_send_to (recv_socket, address,
 277                                      buffer, to_send, cancellable, &error);
 278           else
 279             size = g_socket_send (recv_socket, buffer, to_send,
 280                                   cancellable, &error);
 281
 282           if (size < 0)
 283             {
 284               if (g_error_matches (error,
 285                                    G_IO_ERROR,
 286                                    G_IO_ERROR_WOULD_BLOCK))
 287                 {
 288                   g_print ("socket send would block, handling\n");
 289                   g_error_free (error);
 290                   error = NULL;
 291                   continue;
 292                 }
 293               else
 294                 {
 295                   g_printerr ("Error sending to socket: %s\n",
 296                               error->message);
 297                   return 1;
 298                 }
 299             }
 300
 301           g_print ("sent %" G_GSSIZE_FORMAT " bytes of data\n", size);
 302
 303           if (size == 0)
 304             {
 305               g_printerr ("Unexpected short write\n");
 306               return 1;
 307             }
 308
 309           to_send -= size;
 310         }
 311     }
 312
 313   g_print ("connection closed\n");
 314
 315   if (new_socket)
 316     {
 317       if (!g_socket_close (new_socket, &error))
 318         {
 319           g_printerr ("Error closing connection socket: %s\n",
 320                       error->message);
 321           return 1;
 322         }
 323
 324       g_object_unref (G_OBJECT (new_socket));
 325     }
 326
 327   if (!g_socket_close (socket, &error))
 328     {
 329       g_printerr ("Error closing master socket: %s\n",
 330                   error->message);
 331       return 1;
 332     }
 333
 334   g_object_unref (G_OBJECT (socket));
 335
 336   return 0;
 337 }