spawn-test.c 9.09 KB
Newer Older
Havoc Pennington's avatar
Havoc Pennington committed
1 2 3 4 5 6
/* GLIB - Library of useful routines for C programming
 * Copyright (C) 1995-1997  Peter Mattis, Spencer Kimball and Josh MacDonald
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
7
 * version 2.1 of the License, or (at your option) any later version.
Havoc Pennington's avatar
Havoc Pennington committed
8 9 10 11 12 13 14
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
15
 * License along with this library; if not, see <http://www.gnu.org/licenses/>.
Havoc Pennington's avatar
Havoc Pennington committed
16 17 18 19 20 21 22 23 24
 */

/*
 * Modified by the GLib Team and others 1997-2000.  See the AUTHORS
 * file for a list of people on the GLib Team.  See the ChangeLog
 * files for a list of changes.  These files are distributed with
 * GLib at ftp://ftp.gtk.org/pub/gtk/. 
 */

25
#undef G_DISABLE_ASSERT
Havoc Pennington's avatar
Havoc Pennington committed
26 27 28
#undef G_LOG_DOMAIN

#include <glib.h>
29
#include <glib/gstdio.h>
Havoc Pennington's avatar
Havoc Pennington committed
30 31 32 33
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

34 35
#ifdef G_OS_WIN32
#include <fcntl.h>
Hans Breuer's avatar
Hans Breuer committed
36
#include <io.h>
37
#define pipe(fds) _pipe(fds, 4096, _O_BINARY)
38 39
#endif

Havoc Pennington's avatar
Havoc Pennington committed
40 41

static void
42
run_tests (const gchar* argv0)
Havoc Pennington's avatar
Havoc Pennington committed
43
{
44
  GError *err = NULL;
Havoc Pennington's avatar
Havoc Pennington committed
45
  gchar *output = NULL;
46
  gchar *erroutput = NULL;
47 48
  gchar *dirname = g_path_get_dirname (argv0);
#ifdef G_OS_WIN32
49 50
  int pipedown[2], pipeup[2];
  gchar **argv = 0;
51 52 53
  gchar spawn_binary[1000] = {0};
  gchar full_cmdline[1000] = {0};
  g_snprintf (spawn_binary, sizeof (spawn_binary), "%s\\spawn-test-win32-gui.exe", dirname);
54
#endif
55
  g_free (dirname);
Havoc Pennington's avatar
Havoc Pennington committed
56 57 58 59 60 61 62 63
  
  err = NULL;
  if (!g_spawn_command_line_sync ("nonexistent_application foo 'bar baz' blah blah",
                                  NULL, NULL, NULL,
                                  &err))
    {
      g_error_free (err);
    }
64 65 66 67 68
  else
    {
      g_warning ("no error for sync spawn of nonexistent application");
      exit (1);
    }
Havoc Pennington's avatar
Havoc Pennington committed
69 70 71 72 73 74 75

  err = NULL;
  if (!g_spawn_command_line_async ("nonexistent_application foo bar baz \"blah blah\"",
                                   &err))
    {
      g_error_free (err);
    }
76 77 78 79 80
  else
    {
      g_warning ("no error for async spawn of nonexistent application");
      exit (1);
    }
Havoc Pennington's avatar
Havoc Pennington committed
81 82

  err = NULL;
83
#ifdef G_OS_UNIX
Havoc Pennington's avatar
Havoc Pennington committed
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
  if (!g_spawn_command_line_sync ("/bin/sh -c 'echo hello'",
                                  &output, NULL, NULL,
                                  &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  else
    {
      g_assert (output != NULL);
      
      if (strcmp (output, "hello\n") != 0)
        {
          printf ("output was '%s', should have been 'hello'\n",
                  output);

          exit (1);
        }

      g_free (output);
105
      output = NULL;
Havoc Pennington's avatar
Havoc Pennington committed
106
    }
107 108 109 110 111 112 113
#endif
  /* Running sort synchronously, collecting its output. 'sort' command is selected
   * because it is non-builtin command on both unix and win32 with well-defined stdout behaviour.
   */
  g_file_set_contents ("spawn-test-created-file.txt", "line first\nline 2\nline last\n", -1, &err);
  g_assert_no_error(err);
  if (!g_spawn_command_line_sync ("sort spawn-test-created-file.txt",
114
                                  &output, &erroutput, NULL,
115 116 117 118 119 120 121 122 123
                                  &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  else
    {
      g_assert (output != NULL);
124
      g_assert (erroutput != NULL);
125
      
126
      if (strstr (output, "\nline first") == 0)
127
        {
128
          printf ("output was '%s', should have contained 'line first' in second line\n",
129 130 131 132
                  output);

          exit (1);
        }
133 134 135 136 137 138 139 140 141 142 143
      if (erroutput[0] != '\0')
	{
	  printf ("error output was '%s', should have been empty\n",
		  erroutput);
	  exit (1);
	}

      g_free (output);
      output = NULL;
      g_free (erroutput);
      erroutput = NULL;
144
      g_unlink ("spawn-test-created-file.txt");
145 146
    }

147 148 149 150 151 152 153 154 155 156 157
  if (!g_spawn_command_line_sync ("sort non-existing-file.txt",
                                  NULL, &erroutput, NULL,
                                  &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  else
    {
      g_assert (erroutput != NULL);
158

159 160 161 162 163 164 165 166
      if (erroutput[0] == '\0')
        {
          printf ("erroutput was empty, expected contain error message about non-existing-file.txt\n");
          exit (1);
        }
      g_free (erroutput);
      erroutput = NULL;
    }
167

168 169 170 171 172 173
#ifdef G_OS_WIN32
  printf ("Running spawn-test-win32-gui in various ways.\n");

  printf ("First asynchronously (without wait).\n");
  g_snprintf (full_cmdline, sizeof (full_cmdline), "'%s' 1", spawn_binary);
  if (!g_spawn_command_line_async (full_cmdline, &err))
174 175 176 177 178 179
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }

180
  printf ("Now synchronously, collecting its output.\n");
181 182
  g_snprintf (full_cmdline, sizeof (full_cmdline), "'%s' 2", spawn_binary);
  if (!g_spawn_command_line_sync (full_cmdline,
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
				  &output, &erroutput, NULL,
				  &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  else
    {
      g_assert (output != NULL);
      g_assert (erroutput != NULL);
      
      if (strcmp (output, "This is stdout\r\n") != 0)
        {
          printf ("output was '%s', should have been 'This is stdout'\n",
                  g_strescape (output, NULL));

          exit (1);
        }
      if (strcmp (erroutput, "This is stderr\r\n") != 0)
	{
	  printf ("error output was '%s', should have been 'This is stderr'\n",
		  g_strescape (erroutput, NULL));
	  exit (1);
	}
208 209

      g_free (output);
210
      output = NULL;
211
      g_free (erroutput);
212
      erroutput = NULL;
213 214
    }

215
  printf ("Now with G_SPAWN_FILE_AND_ARGV_ZERO.\n");
216 217
  g_snprintf (full_cmdline, sizeof (full_cmdline), "'%s' this-should-be-argv-zero print_argv0", spawn_binary);
  if (!g_shell_parse_argv (full_cmdline, NULL, &argv, &err))
218 219 220 221 222 223
    {
      fprintf (stderr, "Error parsing command line? %s\n", err->message);
      g_error_free (err);
      exit (1);
    }

224
  if (!g_spawn_sync (NULL, argv, NULL,
225
		      G_SPAWN_FILE_AND_ARGV_ZERO,
226
		      NULL, NULL, &output, NULL, NULL,
227 228 229 230 231 232
		      &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
233 234 235 236 237 238 239 240 241 242
  else
    {
      if (strcmp (output, "this-should-be-argv-zero") != 0)
	{
	  printf ("output was '%s', should have been 'this-should-be-argv-zero'\n", output);
	  exit (1);
	}
      g_free (output);
      output = NULL;
    }
243 244

  printf ("Now talking to it through pipes.\n");
245 246 247 248 249 250

  if (pipe (pipedown) < 0 ||
      pipe (pipeup) < 0)
    {
      fprintf (stderr, "Could not create pipes\n");
      exit (1);
251 252
    }

253 254
  g_snprintf (full_cmdline, sizeof (full_cmdline), "'%s' pipes %d %d", spawn_binary, pipedown[0], pipeup[1]);
  if (!g_shell_parse_argv (full_cmdline,
255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279
                           NULL, &argv,
                           &err))
    {
      fprintf (stderr, "Error parsing command line? %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  
  if (!g_spawn_async (NULL, argv, NULL,
		      G_SPAWN_LEAVE_DESCRIPTORS_OPEN |
		      G_SPAWN_DO_NOT_REAP_CHILD,
		      NULL, NULL, NULL,
		      &err))
    {
      fprintf (stderr, "Error: %s\n", err->message);
      g_error_free (err);
      exit (1);
    }
  else
    {
      int k, n;
      char buf[100];

      if ((k = read (pipeup[0], &n, sizeof (n))) != sizeof (n))
	{
280
	  int errsv = errno;
281
	  if (k == -1)
282
	    fprintf (stderr, "Read error: %s\n", g_strerror (errsv));
283 284
	  else
	    fprintf (stderr, "Wanted to read %d bytes, got %d\n",
285
		     (int)sizeof (n), k);
286 287 288 289 290
	  exit (1);
	}

      if ((k = read (pipeup[0], buf, n)) != n)
	{
291
	  int errsv = errno;
292
	  if (k == -1)
293
	    fprintf (stderr, "Read error: %s\n", g_strerror (errsv));
294 295 296 297 298 299 300 301 302 303
	  else
	    fprintf (stderr, "Wanted to read %d bytes, got %d\n",
		     n, k);
	  exit (1);
	}

      n = strlen ("Bye then");
      if (write (pipedown[1], &n, sizeof (n)) == -1 ||
	  write (pipedown[1], "Bye then", n) == -1)
	{
304 305
	  int errsv = errno;
	  fprintf (stderr, "Write error: %s\n", g_strerror (errsv));
306 307 308 309 310
	  exit (1);
	}

      if ((k = read (pipeup[0], &n, sizeof (n))) != sizeof (n))
	{
311
	  int errsv = errno;
312
	  if (k == -1)
313
	    fprintf (stderr, "Read error: %s\n", g_strerror (errsv));
314 315
	  else
	    fprintf (stderr, "Wanted to read %d bytes, got %d\n",
316 317 318 319 320 321
		     (int)sizeof (n), k);
	  exit (1);
	}
      if (n != strlen ("See ya"))
	{
	  printf ("child wrote %d bytes, expected %d", n, (int) strlen ("See ya"));
322 323 324 325 326
	  exit (1);
	}

      if ((k = read (pipeup[0], buf, n)) != n)
	{
327
	  int errsv = errno;
328
	  if (k == -1)
329
	    fprintf (stderr, "Read error: %s\n", g_strerror (errsv));
330 331 332 333 334
	  else
	    fprintf (stderr, "Wanted to read %d bytes, got %d\n",
		     n, k);
	  exit (1);
	}
335 336 337 338 339 340
      buf[n] = '\0';
      if (strcmp (buf, "See ya") != 0)
	{
	  printf ("output was '%s', should have been 'See ya'\n", buf);
	  exit (1);
	}
341
    }
342
#endif
Havoc Pennington's avatar
Havoc Pennington committed
343 344 345 346 347 348
}

int
main (int   argc,
      char *argv[])
{
349
  run_tests (argv[0]);
Havoc Pennington's avatar
Havoc Pennington committed
350 351 352
  
  return 0;
}