gio/tests/unix-fd.c

   1 #include <gio/gio.h>
   2 #include <gio/gnetworking.h>
   3 #include <gio/gunixfdmessage.h>
   4 #include <gio/gunixsocketaddress.h>
   5 #ifdef G_OS_UNIX
   6 #include <glib-unix.h>
   7 #include <unistd.h>
   8 #endif
   9 #include <glib/gstdio.h>
  10 #include <fcntl.h>
  11 #include <string.h>
  12 #ifdef G_OS_WIN32
  13 #include <io.h>
  14 #endif
  15
  16 /* ensures that no FDs are left open at the end */
  17 static void
  18 check_fd_list (const gint *fd_list)
  19 {
  20   gint i;
  21
  22   for (i = 0; i < 40; i++)
  23     {
  24       int my_fd;
  25
  26       my_fd = dup (0);
  27       g_assert_cmpint (fd_list[i], ==, my_fd);
  28     }
  29
  30   for (i = 0; i < 40; i++)
  31     g_close (fd_list[i], NULL);
  32 }
  33
  34 static void
  35 create_fd_list (gint *fd_list)
  36 {
  37   gint i;
  38
  39   for (i = 0; i < 40; i++)
  40     {
  41       fd_list[i] = dup (0);
  42       g_assert_cmpint (fd_list[i], >, 0);
  43     }
  44
  45   for (i = 0; i < 40; i++)
  46     g_close (fd_list[i], NULL);
  47 }
  48
  49 static void
  50 test_fd_list (void)
  51 {
  52   GError *err = NULL;
  53   GUnixFDList *list;
  54   const gint *peek;
  55   gint *stolen;
  56   gint fd_list[40];
  57   gint sv[3];
  58   gint s;
  59
  60   create_fd_list (fd_list);
  61   sv[2] = -1;
  62 #ifdef G_OS_WIN32
  63   s = _pipe (sv, 4096, _O_NOINHERIT | _O_BINARY);
  64   g_assert_cmpint (s, ==, 0);
  65 #else
  66   g_unix_open_pipe (sv, O_CLOEXEC, &err);
  67   g_assert_no_error (err);
  68 #endif
  69   list = g_unix_fd_list_new_from_array (sv, -1);
  70   peek = g_unix_fd_list_peek_fds (list, &s);
  71   g_assert_cmpint (s, ==, 2);
  72   g_assert_cmpint (peek[0], ==, sv[0]);
  73   g_assert_cmpint (peek[1], ==, sv[1]);
  74
  75   s = g_unix_fd_list_get (list, 0, &err);
  76   g_assert_no_error (err);
  77   g_close (s, &err);
  78   g_assert_no_error (err);
  79   s = g_unix_fd_list_get (list, 1, &err);
  80   g_assert_no_error (err);
  81   g_close (s, &err);
  82   g_assert_no_error (err);
  83
  84   s = g_unix_fd_list_append (list, sv[0], &err);
  85   g_assert_no_error (err);
  86   g_assert_cmpint (s, >=, 0);
  87   stolen = g_unix_fd_list_steal_fds (list, &s);
  88   g_assert_cmpint (s, ==, 3);
  89   g_assert_cmpint (stolen[0], ==, sv[0]);
  90   g_assert_cmpint (stolen[1], ==, sv[1]);
  91   g_assert_cmpint (stolen[2], >=, 0);
  92   g_close (stolen[0], &err);
  93   g_assert_no_error (err);
  94   g_close (stolen[1], &err);
  95   g_assert_no_error (err);
  96   g_close (stolen[2], &err);
  97   g_assert_no_error (err);
  98   g_free (stolen);
  99
 100   g_object_unref (list);
 101   check_fd_list (fd_list);
 102 }
 103
 104 static void
 105 test_scm (void)
 106 {
 107 #ifndef G_OS_WIN32
 108   GError *err = NULL;
 109   GUnixFDMessage *message;
 110   GUnixFDMessage **mv;
 111   GUnixFDList *list, *l2;
 112   GSocket *sockets[2];
 113   GSocketAddress *addr;
 114   const gint *peek;
 115   char buffer[1024];
 116   gint fd_list[40];
 117   GOutputVector ov;
 118   GInputVector iv;
 119   gint *stolen;
 120   gint sv[3];
 121   gint flags;
 122   gint nm;
 123   gint s, i;
 124   gchar *path;
 125   GByteArray *array;
 126   gboolean abstract;
 127   GUnixSocketAddressType type;
 128
 129   create_fd_list (fd_list);
 130
 131   s = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
 132   g_assert_cmpint (s, ==, 0);
 133   sv[2] = -1;
 134
 135   list = g_unix_fd_list_new_from_array (sv, -1);
 136   message = G_UNIX_FD_MESSAGE (g_unix_fd_message_new_with_fd_list (list));
 137   g_object_unref (list);
 138
 139   g_assert (g_unix_fd_message_get_fd_list (message) == list);
 140   g_object_get (message, "fd-list", &l2, NULL);
 141   g_assert (l2 == list);
 142   g_assert_cmpint (g_unix_fd_list_get_length (list), ==, 2);
 143
 144   peek = g_unix_fd_list_peek_fds (list, &s);
 145   g_assert_cmpint (s, ==, 2);
 146   stolen = g_unix_fd_message_steal_fds (message, &s);
 147   g_assert_cmpint (s, ==, 2);
 148   g_assert (stolen == peek);
 149
 150   g_assert_cmpint (stolen[0], ==, sv[0]);
 151   g_assert_cmpint (stolen[1], ==, sv[1]);
 152   g_assert_cmpint (stolen[2], ==, sv[2]);
 153   g_free (stolen);
 154
 155   g_unix_fd_message_append_fd (message, sv[0], &err);
 156   g_assert_no_error (err);
 157   g_close (sv[0], &err);
 158   g_assert_no_error (err);
 159   g_unix_fd_message_append_fd (message, sv[1], &err);
 160   g_assert_no_error (err);
 161   g_close (sv[1], &err);
 162   g_assert_no_error (err);
 163
 164   for (i = 0; i < 3; i++)
 165     {
 166       s = g_unix_fd_list_get (list, 0, &err);
 167       g_assert_no_error (err);
 168       g_close (s, &err);
 169       g_assert_no_error (err);
 170       s = g_unix_fd_list_get (list, 1, &err);
 171       g_assert_no_error (err);
 172       g_close (s, &err);
 173       g_assert_no_error (err);
 174     }
 175
 176   g_object_unref (message);
 177   g_object_unref (list);
 178
 179   message = G_UNIX_FD_MESSAGE (g_unix_fd_message_new ());
 180   list = g_unix_fd_message_get_fd_list (message);
 181   s = pipe (sv);
 182   g_assert_cmpint (s, ==, 0);
 183
 184   s = g_unix_fd_list_append (list, sv[0], &err);
 185   g_assert_no_error (err);
 186   g_assert_cmpint (s, >=, 0);
 187   s = g_unix_fd_list_append (list, sv[1], &err);
 188   g_assert_no_error (err);
 189   g_assert_cmpint (s, >=, 0);
 190
 191   g_close (sv[0], &err);
 192   g_assert_no_error (err);
 193   g_close (sv[1], &err);
 194   g_assert_no_error (err);
 195   s = g_unix_fd_list_get (list, 0, &err);
 196   g_assert_no_error (err);
 197   g_close (s, &err);
 198   g_assert_no_error (err);
 199   s = g_unix_fd_list_get (list, 1, &err);
 200   g_assert_no_error (err);
 201   g_close (s, &err);
 202   g_assert_no_error (err);
 203
 204   s = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
 205   g_assert_cmpint (s, ==, 0);
 206
 207   sockets[0] = g_socket_new_from_fd (sv[0], &err);
 208   g_assert_no_error (err);
 209   g_assert (G_IS_SOCKET (sockets[0]));
 210   sockets[1] = g_socket_new_from_fd (sv[1], &err);
 211   g_assert_no_error (err);
 212   g_assert (G_IS_SOCKET (sockets[1]));
 213
 214   addr = g_socket_get_local_address (sockets[0], &err);
 215   g_assert_no_error (err);
 216   g_assert (G_IS_UNIX_SOCKET_ADDRESS (addr));
 217   g_assert_cmpint (g_unix_socket_address_get_address_type (G_UNIX_SOCKET_ADDRESS (addr)), ==, G_UNIX_SOCKET_ADDRESS_ANONYMOUS);
 218   g_assert_cmpint (g_unix_socket_address_get_path_len (G_UNIX_SOCKET_ADDRESS (addr)), ==, 0);
 219
 220 G_GNUC_BEGIN_IGNORE_DEPRECATIONS
 221   g_assert (!g_unix_socket_address_get_is_abstract (G_UNIX_SOCKET_ADDRESS (addr)));
 222 G_GNUC_END_IGNORE_DEPRECATIONS
 223
 224   g_object_get (addr,
 225                 "path", &path,
 226                 "path-as-array", &array,
 227                 "abstract", &abstract,
 228                 "address-type", &type,
 229                 NULL);
 230   g_assert_cmpstr (path, ==, "");
 231   g_assert_cmpint (array->len, ==, 0);
 232   g_assert (!abstract);
 233   g_assert (type == G_UNIX_SOCKET_ADDRESS_ANONYMOUS);
 234   g_free (path);
 235   g_byte_array_free (array, TRUE);
 236
 237   g_object_unref (addr);
 238
 239   buffer[0] = 0xff;
 240   ov.buffer = buffer;
 241   ov.size = 1;
 242   s = g_socket_send_message (sockets[0], NULL, &ov, 1,
 243                              (GSocketControlMessage **) &message,
 244                              1, 0, NULL, &err);
 245   g_assert_no_error (err);
 246   g_assert_cmpint (s, ==, 1);
 247   g_object_unref (message);
 248
 249   message = NULL;
 250
 251   flags = 0;
 252   iv.buffer = buffer;
 253   iv.size = 1;
 254   s = g_socket_receive_message (sockets[1], NULL, &iv, 1,
 255                                 (GSocketControlMessage ***) &mv,
 256                                 &nm, &flags, NULL, &err);
 257   g_assert_no_error (err);
 258   g_assert_cmpint (s, ==, 1);
 259   g_object_unref (sockets[0]);
 260   g_object_unref (sockets[1]);
 261
 262   g_assert_cmpint (nm, ==, 1);
 263   message = mv[0];
 264   g_free (mv);
 265
 266   g_assert (G_IS_UNIX_FD_MESSAGE (message));
 267   list = g_object_ref (g_unix_fd_message_get_fd_list (message));
 268   g_object_unref (message);
 269
 270   peek = g_unix_fd_list_peek_fds (list, &s);
 271   g_assert_cmpint (s, ==, 2);
 272   sv[0] = g_unix_fd_list_get (list, 1, &err);
 273   g_assert_no_error (err);
 274
 275   strcpy (buffer, "failure to say failure to say 'i love gnome-panel!'.");
 276   s = write (sv[0], buffer, strlen (buffer) + 1);
 277   g_assert_cmpint (s, ==, strlen (buffer) + 1);
 278
 279   g_close (sv[0], NULL);
 280   memset (buffer, 0xff, sizeof buffer);
 281
 282   s = read (peek[0], buffer, sizeof buffer);
 283   g_assert_cmpint (s, ==, 53);
 284   g_assert_cmpstr (buffer, ==,
 285                    "failure to say failure to say 'i love gnome-panel!'.");
 286
 287   g_object_unref (list);
 288
 289   check_fd_list (fd_list);
 290 #else
 291   g_test_skip ("FD SCM support doesn’t exist on Windows");
 292 #endif
 293 }
 294
 295 int
 296 main (int argc, char **argv)
 297 {
 298   g_test_init (&argc, &argv, NULL);
 299
 300   g_test_add_func ("/unix-fd/fd-list", test_fd_list);
 301   g_test_add_func ("/unix-fd/scm", test_scm);
 302
 303   return g_test_run();
 304 }