posix/tst-spawn.c

   1 /* Tests for spawn.
   2    Copyright (C) 2000 Free Software Foundation, Inc.
   3    This file is part of the GNU C Library.
   4    Contributed by Ulrich Drepper <drepper@cygnus.com>, 2000.
   5
   6    The GNU C Library is free software; you can redistribute it and/or
   7    modify it under the terms of the GNU Lesser General Public
   8    License as published by the Free Software Foundation; either
   9    version 2.1 of the License, or (at your option) any later version.
  10
  11    The GNU C Library is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14    Lesser General Public License for more details.
  15
  16    You should have received a copy of the GNU Lesser General Public
  17    License along with the GNU C Library; if not, write to the Free
  18    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
  19    02111-1307 USA.  */
  20
  21 #include <errno.h>
  22 #include <error.h>
  23 #include <fcntl.h>
  24 #include <spawn.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27 #include <unistd.h>
  28 #include <wait.h>
  29 #include <sys/param.h>
  30
  31
  32 /* Nonzero if the program gets called via `exec'.  */
  33 static int restart;
  34
  35
  36 #define CMDLINE_OPTIONS \
  37   { "restart", no_argument, &restart, 1 },
  38
  39 /* Prototype for our test function.  */
  40 extern void do_prepare (int argc, char *argv[]);
  41 extern int do_test (int argc, char *argv[]);
  42
  43 /* We have a preparation function.  */
  44 #define PREPARE do_prepare
  45
  46 #include "../test-skeleton.c"
  47
  48
  49 /* Name of the temporary files.  */
  50 static char *name1;
  51 static char *name2;
  52 static char *name3;
  53
  54 /* The contents of our files.  */
  55 static const char fd1string[] = "This file should get closed";
  56 static const char fd2string[] = "This file should stay opened";
  57 static const char fd3string[] = "This file will be opened";
  58
  59
  60 /* We have a preparation function.  */
  61 void
  62 do_prepare (int argc, char *argv[])
  63 {
  64    char name_len;
  65
  66    name_len = strlen (test_dir);
  67    name1 = (char *) malloc (name_len + sizeof ("/spawnXXXXXX"));
  68    mempcpy (mempcpy (name1, test_dir, name_len),
  69             "/spawnXXXXXX", sizeof ("/spawnXXXXXX"));
  70    add_temp_file (name1);
  71
  72    name2 = (char *) malloc (name_len + sizeof ("/spawnXXXXXX"));
  73    mempcpy (mempcpy (name2, test_dir, name_len),
  74             "/spawnXXXXXX", sizeof ("/spawnXXXXXX"));
  75    add_temp_file (name2);
  76
  77    name3 = (char *) malloc (name_len + sizeof ("/spawnXXXXXX"));
  78    mempcpy (mempcpy (name3, test_dir, name_len),
  79             "/spawnXXXXXX", sizeof ("/spawnXXXXXX"));
  80    add_temp_file (name3);
  81 }
  82
  83
  84 static int
  85 handle_restart (const char *fd1s, const char *fd2s, const char *fd3s,
  86                 const char *fd4s, const char *name)
  87 {
  88   char buf[100];
  89   int fd1;
  90   int fd2;
  91   int fd3;
  92   int fd4;
  93
  94   /* First get the descriptors.  */
  95   fd1 = atol (fd1s);
  96   fd2 = atol (fd2s);
  97   fd3 = atol (fd3s);
  98   fd4 = atol (fd4s);
  99
 100   /* Sanity check.  */
 101   if (fd1 == fd2)
 102     error (EXIT_FAILURE, 0, "value of fd1 and fd2 is the same");
 103   if (fd1 == fd3)
 104     error (EXIT_FAILURE, 0, "value of fd1 and fd3 is the same");
 105   if (fd1 == fd4)
 106     error (EXIT_FAILURE, 0, "value of fd1 and fd4 is the same");
 107   if (fd2 == fd3)
 108     error (EXIT_FAILURE, 0, "value of fd2 and fd3 is the same");
 109   if (fd2 == fd4)
 110     error (EXIT_FAILURE, 0, "value of fd2 and fd4 is the same");
 111   if (fd3 == fd4)
 112     error (EXIT_FAILURE, 0, "value of fd3 and fd4 is the same");
 113
 114   /* First the easy part: read from the file descriptor which is
 115      supposed to be open.  */
 116   if (lseek (fd2, 0, SEEK_CUR) != strlen (fd2string))
 117     error (EXIT_FAILURE, errno, "file 2 not in right position");
 118   /* The duped descriptor must have the same position.  */
 119   if (lseek (fd4, 0, SEEK_CUR) != strlen (fd2string))
 120     error (EXIT_FAILURE, errno, "file 4 not in right position");
 121   if (lseek (fd2, 0, SEEK_SET) != 0)
 122     error (EXIT_FAILURE, 0, "cannot reset position in file 2");
 123   if (lseek (fd4, 0, SEEK_CUR) != 0)
 124     error (EXIT_FAILURE, errno, "file 4 not set back, too");
 125   if (read (fd2, buf, sizeof buf) != strlen (fd2string))
 126     error (EXIT_FAILURE, 0, "cannot read file 2");
 127   if (memcmp (fd2string, buf, strlen (fd2string)) != 0)
 128     error (EXIT_FAILURE, 0, "file 2 does not match");
 129
 130   /* Now read from the third file.  */
 131   if (read (fd3, buf, sizeof buf) != strlen (fd3string))
 132     error (EXIT_FAILURE, 0, "cannot read file 3");
 133   if (memcmp (fd3string, buf, strlen (fd3string)) != 0)
 134     error (EXIT_FAILURE, 0, "file 3 does not match");
 135   /* Try to write to the file.  This should not be allowed.  */
 136   if (write (fd3, "boo!", 4) != -1 || errno != EBADF)
 137     error (EXIT_FAILURE, 0, "file 3 is writable");
 138
 139   /* Now try to read the first file.  First make sure it is not opened.  */
 140   if (lseek (fd1, 0, SEEK_CUR) != (off_t) -1 || errno != EBADF)
 141     error (EXIT_FAILURE, 0, "file 1 (%d) is not closed", fd1);
 142
 143   /* Now open the file and read it.  */
 144   fd1 = open (name, O_RDONLY);
 145   if (fd1 == -1)
 146     error (EXIT_FAILURE, errno,
 147            "cannot open first file \"%s\" for verification", name);
 148
 149   if (read (fd1, buf, sizeof buf) != strlen (fd1string))
 150     error (EXIT_FAILURE, errno, "cannot read file 1");
 151   if (memcmp (fd1string, buf, strlen (fd1string)) != 0)
 152     error (EXIT_FAILURE, 0, "file 1 does not match");
 153
 154   return 0;
 155 }
 156
 157
 158 int
 159 do_test (int argc, char *argv[])
 160 {
 161   pid_t pid;
 162   int fd1;
 163   int fd2;
 164   int fd3;
 165   int fd4;
 166   int status;
 167   posix_spawn_file_actions_t actions;
 168   char fd1name[18];
 169   char fd2name[18];
 170   char fd3name[18];
 171   char fd4name[18];
 172   char *spargv[12];
 173
 174   /* We must have
 175      - four parameters left of called initially
 176        + path for ld.so
 177        + "--library-path"
 178        + the library path
 179        + the application name
 180      - five parameters left if called through re-execution
 181        + file descriptor number which is supposed to be closed
 182        + the open file descriptor
 183        + the newly opened file descriptor
 184        + thhe duped second descriptor
 185        + the name of the closed descriptor
 186   */
 187   if (argc != (restart ? 6 : 5))
 188     error (EXIT_FAILURE, 0, "wrong number of arguments (%d)", argc);
 189
 190   if (restart)
 191     return handle_restart (argv[1], argv[2], argv[3], argv[4], argv[5]);
 192
 193   /* Prepare the test.  We are creating two files: one which file descriptor
 194      will be marked with FD_CLOEXEC, another which is not.  */
 195
 196    /* Open our test files.   */
 197    fd1 = mkstemp (name1);
 198    if (fd1 == -1)
 199      error (EXIT_FAILURE, errno, "cannot open test file `%s'", name1);
 200    fd2 = mkstemp (name2);
 201    if (fd2 == -1)
 202      error (EXIT_FAILURE, errno, "cannot open test file `%s'", name2);
 203    fd3 = mkstemp (name3);
 204    if (fd3 == -1)
 205      error (EXIT_FAILURE, errno, "cannot open test file `%s'", name3);
 206
 207    /* Write something in the files.  */
 208    if (write (fd1, fd1string, strlen (fd1string)) != strlen (fd1string))
 209      error (EXIT_FAILURE, errno, "cannot write to first file");
 210    if (write (fd2, fd2string, strlen (fd2string)) != strlen (fd2string))
 211      error (EXIT_FAILURE, errno, "cannot write to second file");
 212    if (write (fd3, fd3string, strlen (fd3string)) != strlen (fd3string))
 213      error (EXIT_FAILURE, errno, "cannot write to third file");
 214
 215    /* Close the third file.  It'll be opened by `spawn'.  */
 216    close (fd3);
 217
 218    /* Tell `spawn' what to do.  */
 219    if (posix_spawn_file_actions_init (&actions) != 0)
 220      error (EXIT_FAILURE, errno, "posix_spawn_file_actions_init");
 221    /* Close `fd1'.  */
 222    if (posix_spawn_file_actions_addclose (&actions, fd1) != 0)
 223      error (EXIT_FAILURE, errno, "posix_spawn_file_actions_addclose");
 224    /* We want to open the third file.  */
 225    if (posix_spawn_file_actions_addopen (&actions, fd3, name3,
 226                                          O_RDONLY, 0666) != 0)
 227      error (EXIT_FAILURE, errno, "posix_spawn_file_actions_addopen");
 228    /* We dup the second descriptor.  */
 229    fd4 = MAX (2, MAX (fd1, MAX (fd2, fd3))) + 1;
 230    if (posix_spawn_file_actions_adddup2 (&actions, fd2, fd4) != 0)
 231      error (EXIT_FAILURE, errno, "posix_spawn_file_actions_adddup2");
 232
 233    /* Now spawn the process.  */
 234    snprintf (fd1name, sizeof fd1name, "%d", fd1);
 235    snprintf (fd2name, sizeof fd2name, "%d", fd2);
 236    snprintf (fd3name, sizeof fd3name, "%d", fd3);
 237    snprintf (fd4name, sizeof fd4name, "%d", fd4);
 238
 239    spargv[0] = argv[1];
 240    spargv[1] = argv[2];
 241    spargv[2] = argv[3];
 242    spargv[3] = argv[4];
 243    spargv[4] = (char *) "--direct";
 244    spargv[5] = (char *) "--restart";
 245    spargv[6] = fd1name;
 246    spargv[7] = fd2name;
 247    spargv[8] = fd3name;
 248    spargv[9] = fd4name;
 249    spargv[10] = name1;
 250    spargv[11] = NULL;
 251
 252    if (posix_spawn (&pid, argv[1], &actions, NULL, spargv, environ) != 0)
 253      error (EXIT_FAILURE, errno, "posix_spawn");
 254
 255    /* Cleanup.  */
 256    if (posix_spawn_file_actions_destroy (&actions) != 0)
 257      error (EXIT_FAILURE, errno, "posix_spawn_file_actions_destroy");
 258
 259   /* Wait for the child.  */
 260   if (waitpid (pid, &status, 0) != pid)
 261     error (EXIT_FAILURE, errno, "wrong child");
 262
 263   if (WTERMSIG (status) != 0)
 264     error (EXIT_FAILURE, 0, "Child terminated incorrectly");
 265   status = WEXITSTATUS (status);
 266
 267   /* Remove the test files.  */
 268   unlink (name1);
 269   unlink (name2);
 270   unlink (name3);
 271
 272   return status;
 273 }