tests/gobject/timeloop-closure.c

   1 #undef G_DISABLE_ASSERT
   2 #undef G_LOG_DOMAIN
   3
   4 #include <errno.h>
   5 #include <stdlib.h>
   6 #include <unistd.h>
   7 #include <stdio.h>
   8 #include <sys/time.h>
   9 #include <sys/resource.h>
  10
  11 #include <glib.h>
  12 #include <glib-object.h>
  13
  14 static int n_children = 3;
  15 static int n_active_children;
  16 static int n_iters = 10000;
  17 static GMainLoop *loop;
  18
  19 static void
  20 io_pipe (GIOChannel **channels)
  21 {
  22   int fds[2];
  23
  24   if (pipe(fds) < 0)
  25     {
  26       fprintf (stderr, "Cannot create pipe %s\n", g_strerror (errno));
  27       exit (1);
  28     }
  29
  30   channels[0] = g_io_channel_unix_new (fds[0]);
  31   channels[1] = g_io_channel_unix_new (fds[1]);
  32 }
  33
  34 static gboolean
  35 read_all (GIOChannel *channel, char *buf, int len)
  36 {
  37   gsize bytes_read = 0;
  38   gsize count;
  39   GIOError err;
  40
  41   while (bytes_read < len)
  42     {
  43       err = g_io_channel_read (channel, buf + bytes_read, len - bytes_read, &count);
  44       if (err)
  45         {
  46           if (err != G_IO_ERROR_AGAIN)
  47             return FALSE;
  48         }
  49       else if (count == 0)
  50         return FALSE;
  51
  52       bytes_read += count;
  53     }
  54
  55   return TRUE;
  56 }
  57
  58 static gboolean
  59 write_all (GIOChannel *channel, char *buf, int len)
  60 {
  61   gsize bytes_written = 0;
  62   gsize count;
  63   GIOError err;
  64
  65   while (bytes_written < len)
  66     {
  67       err = g_io_channel_write (channel, buf + bytes_written, len - bytes_written, &count);
  68       if (err && err != G_IO_ERROR_AGAIN)
  69         return FALSE;
  70
  71       bytes_written += count;
  72     }
  73
  74   return TRUE;
  75 }
  76
  77 static void
  78 run_child (GIOChannel *in_channel, GIOChannel *out_channel)
  79 {
  80   int i;
  81   int val = 1;
  82   GTimer *timer = g_timer_new();
  83
  84   for (i = 0; i < n_iters; i++)
  85     {
  86       write_all (out_channel, (char *)&val, sizeof (val));
  87       read_all (in_channel, (char *)&val, sizeof (val));
  88     }
  89
  90   val = 0;
  91   write_all (out_channel, (char *)&val, sizeof (val));
  92
  93   val = g_timer_elapsed (timer, NULL) * 1000;
  94
  95   write_all (out_channel, (char *)&val, sizeof (val));
  96   g_timer_destroy (timer);
  97
  98   exit (0);
  99 }
 100
 101 static gboolean
 102 input_callback (GIOChannel   *source,
 103                 GIOCondition  condition,
 104                 gpointer      data)
 105 {
 106   int val;
 107   GIOChannel *dest = (GIOChannel *)data;
 108
 109   if (!read_all (source, (char *)&val, sizeof(val)))
 110     {
 111       fprintf (stderr, "Unexpected EOF\n");
 112       exit (1);
 113     }
 114
 115   if (val)
 116     {
 117       write_all (dest, (char *)&val, sizeof(val));
 118
 119       return TRUE;
 120     }
 121   else
 122     {
 123       g_io_channel_close (source);
 124       g_io_channel_close (dest);
 125
 126       n_active_children--;
 127       if (n_active_children == 0)
 128         g_main_loop_quit (loop);
 129
 130       return FALSE;
 131     }
 132 }
 133
 134 static void
 135 create_child (void)
 136 {
 137   int pid;
 138   GIOChannel *in_channels[2];
 139   GIOChannel *out_channels[2];
 140   GSource *source;
 141
 142   io_pipe (in_channels);
 143   io_pipe (out_channels);
 144
 145   pid = fork ();
 146
 147   if (pid > 0)                  /* Parent */
 148     {
 149       g_io_channel_close (in_channels[0]);
 150       g_io_channel_close (out_channels[1]);
 151
 152       source = g_io_create_watch (out_channels[0], G_IO_IN | G_IO_HUP);
 153       g_source_set_closure (source,
 154                             g_cclosure_new (G_CALLBACK (input_callback), in_channels[1],
 155                                             (GClosureNotify)g_io_channel_unref));
 156       g_source_attach (source, NULL);
 157       g_source_unref (source);
 158
 159       g_io_channel_unref (in_channels[0]);
 160       g_io_channel_unref (out_channels[0]);
 161       g_io_channel_unref (out_channels[1]);
 162
 163     }
 164   else if (pid == 0)            /* Child */
 165     {
 166       g_io_channel_close (in_channels[1]);
 167       g_io_channel_close (out_channels[0]);
 168
 169       setsid ();
 170
 171       run_child (in_channels[0], out_channels[1]);
 172     }
 173   else                          /* Error */
 174     {
 175       fprintf (stderr, "Cannot fork: %s\n", g_strerror (errno));
 176       exit (1);
 177     }
 178 }
 179
 180 static double
 181 difftimeval (struct timeval *old, struct timeval *new)
 182 {
 183   return
 184     (new->tv_sec - old->tv_sec) * 1000. + (new->tv_usec - old->tv_usec) / 1000;
 185 }
 186
 187 int
 188 main (int argc, char **argv)
 189 {
 190   int i;
 191   struct rusage old_usage;
 192   struct rusage new_usage;
 193
 194   if (argc > 1)
 195     n_children = atoi(argv[1]);
 196
 197   if (argc > 2)
 198     n_iters = atoi(argv[2]);
 199
 200   printf ("Children: %d     Iters: %d\n", n_children, n_iters);
 201
 202   n_active_children = n_children;
 203   for (i = 0; i < n_children; i++)
 204     create_child ();
 205
 206   getrusage (RUSAGE_SELF, &old_usage);
 207   loop = g_main_loop_new (NULL, FALSE);
 208   g_main_loop_run (loop);
 209   getrusage (RUSAGE_SELF, &new_usage);
 210
 211   printf ("Elapsed user: %g\n",
 212           difftimeval (&old_usage.ru_utime, &new_usage.ru_utime));
 213   printf ("Elapsed system: %g\n",
 214           difftimeval (&old_usage.ru_stime, &new_usage.ru_stime));
 215   printf ("Elapsed total: %g\n",
 216           difftimeval (&old_usage.ru_utime, &new_usage.ru_utime) +
 217           difftimeval (&old_usage.ru_stime, &new_usage.ru_stime));
 218   printf ("total / iteration: %g\n",
 219           (difftimeval (&old_usage.ru_utime, &new_usage.ru_utime) +
 220            difftimeval (&old_usage.ru_stime, &new_usage.ru_stime)) /
 221           (n_iters * n_children));
 222
 223   g_main_loop_unref (loop);
 224
 225   return 0;
 226 }