gio/tests/gsubprocess-testprog.c

   1 #include <gio/gio.h>
   2 #include <string.h>
   3 #include <stdio.h>
   4 #include <stdlib.h>
   5 #include <errno.h>
   6 #ifdef G_OS_UNIX
   7 #include <unistd.h>
   8 #include <gio/gunixinputstream.h>
   9 #include <gio/gunixoutputstream.h>
  10 #else
  11 #include <io.h>
  12 #endif
  13
  14 static GOptionEntry options[] = {
  15   {NULL}
  16 };
  17
  18 static void
  19 write_all (int           fd,
  20            const guint8* buf,
  21            gsize         len)
  22 {
  23   while (len > 0)
  24     {
  25       gssize bytes_written = write (fd, buf, len);
  26       int errsv = errno;
  27       if (bytes_written < 0)
  28         g_error ("Failed to write to fd %d: %s",
  29                  fd, g_strerror (errsv));
  30       buf += bytes_written;
  31       len -= bytes_written;
  32     }
  33 }
  34
  35 static int
  36 echo_mode (int argc,
  37            char **argv)
  38 {
  39   int i;
  40
  41   for (i = 2; i < argc; i++)
  42     {
  43       write_all (1, (guint8*)argv[i], strlen (argv[i]));
  44       write_all (1, (guint8*)"\n", 1);
  45     }
  46
  47   return 0;
  48 }
  49
  50 static int
  51 echo_stdout_and_stderr_mode (int argc,
  52                              char **argv)
  53 {
  54   int i;
  55
  56   for (i = 2; i < argc; i++)
  57     {
  58       write_all (1, (guint8*)argv[i], strlen (argv[i]));
  59       write_all (1, (guint8*)"\n", 1);
  60       write_all (2, (guint8*)argv[i], strlen (argv[i]));
  61       write_all (2, (guint8*)"\n", 1);
  62     }
  63
  64   return 0;
  65 }
  66
  67 static int
  68 cat_mode (int argc,
  69           char **argv)
  70 {
  71   GIOChannel *chan_stdin;
  72   GIOChannel *chan_stdout;
  73   GIOStatus status;
  74   char buf[1024];
  75   gsize bytes_read, bytes_written;
  76   GError *local_error = NULL;
  77   GError **error = &local_error;
  78
  79   chan_stdin = g_io_channel_unix_new (0);
  80   g_io_channel_set_encoding (chan_stdin, NULL, error);
  81   g_assert_no_error (local_error);
  82   chan_stdout = g_io_channel_unix_new (1);
  83   g_io_channel_set_encoding (chan_stdout, NULL, error);
  84   g_assert_no_error (local_error);
  85
  86   while (TRUE)
  87     {
  88       do
  89         status = g_io_channel_read_chars (chan_stdin, buf, sizeof (buf),
  90                                           &bytes_read, error);
  91       while (status == G_IO_STATUS_AGAIN);
  92
  93       if (status == G_IO_STATUS_EOF || status == G_IO_STATUS_ERROR)
  94         break;
  95
  96       do
  97         status = g_io_channel_write_chars (chan_stdout, buf, bytes_read,
  98                                            &bytes_written, error);
  99       while (status == G_IO_STATUS_AGAIN);
 100
 101       if (status == G_IO_STATUS_EOF || status == G_IO_STATUS_ERROR)
 102         break;
 103     }
 104
 105   g_io_channel_unref (chan_stdin);
 106   g_io_channel_unref (chan_stdout);
 107
 108   if (local_error)
 109     {
 110       g_printerr ("I/O error: %s\n", local_error->message);
 111       g_clear_error (&local_error);
 112       return 1;
 113     }
 114   return 0;
 115 }
 116
 117 static gint
 118 sleep_forever_mode (int argc,
 119                     char **argv)
 120 {
 121   GMainLoop *loop;
 122
 123   loop = g_main_loop_new (NULL, TRUE);
 124   g_main_loop_run (loop);
 125
 126   return 0;
 127 }
 128
 129 static int
 130 write_to_fds (int argc, char **argv)
 131 {
 132   int i;
 133
 134   for (i = 2; i < argc; i++)
 135     {
 136       int fd = atoi (argv[i]);
 137       FILE *f = fdopen (fd, "w");
 138       const char buf[] = "hello world\n";
 139       size_t bytes_written;
 140
 141       g_assert (f != NULL);
 142
 143       bytes_written = fwrite (buf, 1, sizeof (buf), f);
 144       g_assert (bytes_written == sizeof (buf));
 145
 146       if (fclose (f) == -1)
 147         g_assert_not_reached ();
 148     }
 149
 150   return 0;
 151 }
 152
 153 static int
 154 env_mode (int argc, char **argv)
 155 {
 156   char **env;
 157   int i;
 158
 159   env = g_get_environ ();
 160
 161   for (i = 0; env[i]; i++)
 162     g_print ("%s\n", env[i]);
 163
 164   g_strfreev (env);
 165
 166   return 0;
 167 }
 168
 169 static int
 170 cwd_mode (int argc, char **argv)
 171 {
 172   char *cwd;
 173
 174   cwd = g_get_current_dir ();
 175   g_print ("%s\n", cwd);
 176   g_free (cwd);
 177
 178   return 0;
 179 }
 180
 181 static int
 182 printenv_mode (int argc, char **argv)
 183 {
 184   gint i;
 185
 186   for (i = 2; i < argc; i++)
 187     {
 188       const gchar *value = g_getenv (argv[i]);
 189
 190       if (value != NULL)
 191         g_print ("%s=%s\n", argv[i], value);
 192     }
 193
 194   return 0;
 195 }
 196
 197 int
 198 main (int argc, char **argv)
 199 {
 200   GOptionContext *context;
 201   GError *error = NULL;
 202   const char *mode;
 203
 204   context = g_option_context_new ("MODE - Test GSubprocess stuff");
 205   g_option_context_add_main_entries (context, options, NULL);
 206   if (!g_option_context_parse (context, &argc, &argv, &error))
 207     {
 208       g_printerr ("%s: %s\n", argv[0], error->message);
 209       return 1;
 210     }
 211
 212   if (argc < 2)
 213     {
 214       g_printerr ("MODE argument required\n");
 215       return 1;
 216     }
 217
 218   mode = argv[1];
 219   if (strcmp (mode, "noop") == 0)
 220     return 0;
 221   else if (strcmp (mode, "exit1") == 0)
 222     return 1;
 223   else if (strcmp (mode, "assert-argv0") == 0)
 224     {
 225       if (strcmp (argv[0], "moocow") == 0)
 226         return 0;
 227       g_printerr ("argv0=%s != moocow\n", argv[0]);
 228       return 1;
 229     }
 230   else if (strcmp (mode, "echo") == 0)
 231     return echo_mode (argc, argv);
 232   else if (strcmp (mode, "echo-stdout-and-stderr") == 0)
 233     return echo_stdout_and_stderr_mode (argc, argv);
 234   else if (strcmp (mode, "cat") == 0)
 235     return cat_mode (argc, argv);
 236   else if (strcmp (mode, "sleep-forever") == 0)
 237     return sleep_forever_mode (argc, argv);
 238   else if (strcmp (mode, "write-to-fds") == 0)
 239     return write_to_fds (argc, argv);
 240   else if (strcmp (mode, "env") == 0)
 241     return env_mode (argc, argv);
 242   else if (strcmp (mode, "cwd") == 0)
 243     return cwd_mode (argc, argv);
 244   else if (strcmp (mode, "printenv") == 0)
 245     return printenv_mode (argc, argv);
 246   else
 247     {
 248       g_printerr ("Unknown MODE %s\n", argv[1]);
 249       return 1;
 250     }
 251
 252   return TRUE;
 253 }