app/test/process.h

   1 /* SPDX-License-Identifier: BSD-3-Clause
   2  * Copyright(c) 2010-2014 Intel Corporation
   3  */
   4
   5 #ifndef _PROCESS_H_
   6 #define _PROCESS_H_
   7
   8 #include <errno.h>  /* errno */
   9 #include <limits.h> /* PATH_MAX */
  10 #include <libgen.h> /* basename et al */
  11 #include <stdlib.h> /* NULL */
  12 #include <string.h> /* strerror */
  13 #include <unistd.h> /* readlink */
  14 #include <sys/wait.h>
  15
  16 #include <rte_string_fns.h> /* strlcpy */
  17
  18 #ifdef RTE_EXEC_ENV_FREEBSD
  19 #define self "curproc"
  20 #define exe "file"
  21 #else
  22 #define self "self"
  23 #define exe "exe"
  24 #endif
  25
  26 #ifdef RTE_LIBRTE_PDUMP
  27 #include <pthread.h>
  28 extern void *send_pkts(void *empty);
  29 extern uint16_t flag_for_send_pkts;
  30 #endif
  31
  32 /*
  33  * launches a second copy of the test process using the given argv parameters,
  34  * which should include argv[0] as the process name. To identify in the
  35  * subprocess the source of the call, the env_value parameter is set in the
  36  * environment as $RTE_TEST
  37  */
  38 static inline int
  39 process_dup(const char *const argv[], int numargs, const char *env_value)
  40 {
  41         int num;
  42         char *argv_cpy[numargs + 1];
  43         int i, fd, status;
  44         char path[32];
  45 #ifdef RTE_LIBRTE_PDUMP
  46         pthread_t thread;
  47 #endif
  48
  49         pid_t pid = fork();
  50         if (pid < 0)
  51                 return -1;
  52         else if (pid == 0) {
  53                 /* make a copy of the arguments to be passed to exec */
  54                 for (i = 0; i < numargs; i++)
  55                         argv_cpy[i] = strdup(argv[i]);
  56                 argv_cpy[i] = NULL;
  57                 num = numargs;
  58
  59                 /* close all open file descriptors, check /proc/self/fd to only
  60                  * call close on open fds. Exclude fds 0, 1 and 2*/
  61                 for (fd = getdtablesize(); fd > 2; fd-- ) {
  62                         snprintf(path, sizeof(path), "/proc/" exe "/fd/%d", fd);
  63                         if (access(path, F_OK) == 0)
  64                                 close(fd);
  65                 }
  66                 printf("Running binary with argv[]:");
  67                 for (i = 0; i < num; i++)
  68                         printf("'%s' ", argv_cpy[i]);
  69                 printf("\n");
  70
  71                 /* set the environment variable */
  72                 if (setenv(RECURSIVE_ENV_VAR, env_value, 1) != 0)
  73                         rte_panic("Cannot export environment variable\n");
  74
  75                 strlcpy(path, "/proc/" self "/" exe, sizeof(path));
  76                 if (execv(path, argv_cpy) < 0) {
  77                         if (errno == ENOENT) {
  78                                 printf("Could not find '%s', is procfs mounted?\n",
  79                                                 path);
  80                         }
  81                         rte_panic("Cannot exec: %s\n", strerror(errno));
  82                 }
  83         }
  84         /* parent process does a wait */
  85 #ifdef RTE_LIBRTE_PDUMP
  86         if ((strcmp(env_value, "run_pdump_server_tests") == 0))
  87                 pthread_create(&thread, NULL, &send_pkts, NULL);
  88 #endif
  89
  90         while (wait(&status) != pid)
  91                 ;
  92 #ifdef RTE_LIBRTE_PDUMP
  93         if ((strcmp(env_value, "run_pdump_server_tests") == 0)) {
  94                 flag_for_send_pkts = 0;
  95                 pthread_join(thread, NULL);
  96         }
  97 #endif
  98         return status;
  99 }
 100
 101 /* FreeBSD doesn't support file prefixes, so force compile failures for any
 102  * tests attempting to use this function on FreeBSD.
 103  */
 104 #ifdef RTE_EXEC_ENV_LINUX
 105 static char *
 106 get_current_prefix(char *prefix, int size)
 107 {
 108         char path[PATH_MAX] = {0};
 109         char buf[PATH_MAX] = {0};
 110
 111         /* get file for config (fd is always 3) */
 112         snprintf(path, sizeof(path), "/proc/self/fd/%d", 3);
 113
 114         /* return NULL on error */
 115         if (readlink(path, buf, sizeof(buf)) == -1)
 116                 return NULL;
 117
 118         /* get the prefix */
 119         snprintf(prefix, size, "%s", basename(dirname(buf)));
 120
 121         return prefix;
 122 }
 123 #endif
 124
 125 #endif /* _PROCESS_H_ */