app/test/process.h

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #ifndef _PROCESS_H_
  35 #define _PROCESS_H_
  36
  37 #ifndef RTE_EXEC_ENV_BAREMETAL
  38
  39 #ifdef RTE_EXEC_ENV_BSDAPP
  40 #define self "curproc"
  41 #define exe "file"
  42 #else
  43 #define self "self"
  44 #define exe "exe"
  45 #endif
  46
  47 /*
  48  * launches a second copy of the test process using the given argv parameters,
  49  * which should include argv[0] as the process name. To identify in the
  50  * subprocess the source of the call, the env_value parameter is set in the
  51  * environment as $RTE_TEST
  52  */
  53 static inline int
  54 process_dup(const char *const argv[], int numargs, const char *env_value)
  55 {
  56         int num;
  57 #ifdef RTE_LIBRTE_XEN_DOM0
  58         char *argv_cpy[numargs + 2];
  59 #else
  60         char *argv_cpy[numargs + 1];
  61 #endif
  62         int i, fd, status;
  63         char path[32];
  64
  65         pid_t pid = fork();
  66         if (pid < 0)
  67                 return -1;
  68         else if (pid == 0) {
  69                 /* make a copy of the arguments to be passed to exec */
  70                 for (i = 0; i < numargs; i++)
  71                         argv_cpy[i] = strdup(argv[i]);
  72         #ifdef RTE_LIBRTE_XEN_DOM0
  73                 argv_cpy[i] = strdup("--xen-dom0");
  74                 argv_cpy[i + 1] = NULL;
  75                 num = numargs + 1;
  76         #else
  77                 argv_cpy[i] = NULL;
  78                 num = numargs;
  79         #endif
  80
  81                 /* close all open file descriptors, check /proc/self/fd to only
  82                  * call close on open fds. Exclude fds 0, 1 and 2*/
  83                 for (fd = getdtablesize(); fd > 2; fd-- ) {
  84                         snprintf(path, sizeof(path), "/proc/" exe "/fd/%d", fd);
  85                         if (access(path, F_OK) == 0)
  86                                 close(fd);
  87                 }
  88                 printf("Running binary with argv[]:");
  89                 for (i = 0; i < num; i++)
  90                         printf("'%s' ", argv_cpy[i]);
  91                 printf("\n");
  92
  93                 /* set the environment variable */
  94                 if (setenv(RECURSIVE_ENV_VAR, env_value, 1) != 0)
  95                         rte_panic("Cannot export environment variable\n");
  96                 if (execv("/proc/" self "/" exe, argv_cpy) < 0)
  97                         rte_panic("Cannot exec\n");
  98         }
  99         /* parent process does a wait */
 100         while (wait(&status) != pid)
 101                         ;
 102         return status;
 103 }
 104
 105 #endif /* not baremetal */
 106
 107 #endif /* _PROCESS_H_ */