gio/tests/socket.c

   1 /* GLib testing framework examples and tests
   2  *
   3  * Copyright (C) 2008-2010 Red Hat, Inc.
   4  *
   5  * This library is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public
   7  * License as published by the Free Software Foundation; either
   8  * version 2 of the License, or (at your option) any later version.
   9  *
  10  * This library is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General
  16  * Public License along with this library; if not, write to the
  17  * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
  18  * Boston, MA 02111-1307, USA.
  19  *
  20  * Author: David Zeuthen <davidz@redhat.com>
  21  */
  22
  23 #include <gio/gio.h>
  24
  25 #ifdef G_OS_UNIX
  26 #include <errno.h>
  27 #include <sys/types.h>
  28 #include <sys/socket.h>
  29 #include <sys/wait.h>
  30 #include <string.h>
  31 #include <stdlib.h>
  32 #include <gio/gunixconnection.h>
  33 #endif
  34
  35 #ifdef G_OS_UNIX
  36 static void
  37 test_unix_from_fd (void)
  38 {
  39   gint fd;
  40   GError *error;
  41   GSocket *s;
  42
  43   fd = socket (AF_UNIX, SOCK_STREAM, 0);
  44   g_assert_cmpint (fd, !=, -1);
  45
  46   error = NULL;
  47   s = g_socket_new_from_fd (fd, &error);
  48   g_assert_no_error (error);
  49   g_assert_cmpint (g_socket_get_family (s), ==, G_SOCKET_FAMILY_UNIX);
  50   g_assert_cmpint (g_socket_get_socket_type (s), ==, G_SOCKET_TYPE_STREAM);
  51   g_assert_cmpint (g_socket_get_protocol (s), ==, G_SOCKET_PROTOCOL_DEFAULT);
  52   g_object_unref (s);
  53 }
  54
  55 static void
  56 test_unix_connection (void)
  57 {
  58   gint fd;
  59   GError *error;
  60   GSocket *s;
  61   GSocketConnection *c;
  62
  63   fd = socket (AF_UNIX, SOCK_STREAM, 0);
  64   g_assert_cmpint (fd, !=, -1);
  65
  66   error = NULL;
  67   s = g_socket_new_from_fd (fd, &error);
  68   g_assert_no_error (error);
  69   c = g_socket_connection_factory_create_connection (s);
  70   g_assert (G_IS_UNIX_CONNECTION (c));
  71   g_object_unref (c);
  72   g_object_unref (s);
  73 }
  74
  75 static GSocketConnection *
  76 create_connection_for_fd (int fd)
  77 {
  78   GError *err = NULL;
  79   GSocket *socket;
  80   GSocketConnection *connection;
  81
  82   socket = g_socket_new_from_fd (fd, &err);
  83   g_assert_no_error (err);
  84   g_assert (G_IS_SOCKET (socket));
  85   connection = g_socket_connection_factory_create_connection (socket);
  86   g_assert (G_IS_UNIX_CONNECTION (connection));
  87   g_object_unref (socket);
  88   return connection;
  89 }
  90
  91 #define TEST_DATA "failure to say failure to say 'i love gnome-panel!'."
  92
  93 static void
  94 test_unix_connection_ancillary_data (void)
  95 {
  96   GError *err = NULL;
  97   gint pv[2], sv[3];
  98   gint status, fd, len;
  99   char buffer[1024];
 100   pid_t pid;
 101
 102   status = pipe (pv);
 103   g_assert_cmpint (status, ==, 0);
 104
 105   status = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
 106   g_assert_cmpint (status, ==, 0);
 107
 108   pid = fork ();
 109   g_assert_cmpint (pid, >=, 0);
 110
 111   /* Child: close its copy of the write end of the pipe, receive it
 112    * again from the parent over the socket, and write some text to it.
 113    *
 114    * Parent: send the write end of the pipe (still open for the
 115    * parent) over the socket, close it, and read some text from the
 116    * read end of the pipe.
 117    */
 118   if (pid == 0)
 119     {
 120       GSocketConnection *connection;
 121
 122       close (sv[1]);
 123       connection = create_connection_for_fd (sv[0]);
 124
 125       status = close (pv[1]);
 126       g_assert_cmpint (status, ==, 0);
 127
 128       err = NULL;
 129       fd = g_unix_connection_receive_fd (G_UNIX_CONNECTION (connection), NULL,
 130                                          &err);
 131       g_assert_no_error (err);
 132       g_assert_cmpint (fd, >, -1);
 133       g_object_unref (connection);
 134
 135       do
 136         len = write (fd, TEST_DATA, sizeof (TEST_DATA));
 137       while (len == -1 && errno == EINTR);
 138       g_assert_cmpint (len, ==, sizeof (TEST_DATA));
 139       exit (0);
 140     }
 141   else
 142     {
 143       GSocketConnection *connection;
 144
 145       close (sv[0]);
 146       connection = create_connection_for_fd (sv[1]);
 147
 148       err = NULL;
 149       g_unix_connection_send_fd (G_UNIX_CONNECTION (connection), pv[1], NULL,
 150                                  &err);
 151       g_assert_no_error (err);
 152       g_object_unref (connection);
 153
 154       status = close (pv[1]);
 155       g_assert_cmpint (status, ==, 0);
 156
 157       memset (buffer, 0xff, sizeof buffer);
 158       do
 159         len = read (pv[0], buffer, sizeof buffer);
 160       while (len == -1 && errno == EINTR);
 161
 162       g_assert_cmpint (len, ==, sizeof (TEST_DATA));
 163       g_assert_cmpstr (buffer, ==, TEST_DATA);
 164
 165       waitpid (pid, &status, 0);
 166       g_assert (WIFEXITED (status));
 167       g_assert_cmpint (WEXITSTATUS (status), ==, 0);
 168     }
 169
 170   /* TODO: add test for g_unix_connection_send_credentials() and
 171    * g_unix_connection_receive_credentials().
 172    */
 173 }
 174 #endif /* G_OS_UNIX */
 175
 176 int
 177 main (int   argc,
 178       char *argv[])
 179 {
 180   g_type_init ();
 181   g_test_init (&argc, &argv, NULL);
 182
 183 #ifdef G_OS_UNIX
 184   g_test_add_func ("/socket/unix-from-fd", test_unix_from_fd);
 185   g_test_add_func ("/socket/unix-connection", test_unix_connection);
 186   g_test_add_func ("/socket/unix-connection-ancillary-data", test_unix_connection_ancillary_data);
 187 #endif
 188
 189   return g_test_run();
 190 }
 191