test: fix build without ring PMD
[dpdk.git] / app / test / process.h
index cb8973e..c3b3780 100644 (file)
@@ -1,42 +1,22 @@
-/*-
- *   BSD LICENSE
- *
- *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
- *   All rights reserved.
- *
- *   Redistribution and use in source and binary forms, with or without
- *   modification, are permitted provided that the following conditions
- *   are met:
- *
- *     * Redistributions of source code must retain the above copyright
- *       notice, this list of conditions and the following disclaimer.
- *     * Redistributions in binary form must reproduce the above copyright
- *       notice, this list of conditions and the following disclaimer in
- *       the documentation and/or other materials provided with the
- *       distribution.
- *     * Neither the name of Intel Corporation nor the names of its
- *       contributors may be used to endorse or promote products derived
- *       from this software without specific prior written permission.
- *
- *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+/* SPDX-License-Identifier: BSD-3-Clause
+ * Copyright(c) 2010-2014 Intel Corporation
  */
 
 #ifndef _PROCESS_H_
 #define _PROCESS_H_
 
-#ifndef RTE_EXEC_ENV_BAREMETAL
+#include <errno.h>  /* errno */
+#include <limits.h> /* PATH_MAX */
+#include <libgen.h> /* basename et al */
+#include <stdlib.h> /* NULL */
+#include <string.h> /* strerror */
+#include <unistd.h> /* readlink */
+#include <dirent.h>
+#include <sys/wait.h>
 
-#ifdef RTE_EXEC_ENV_BSDAPP
+#include <rte_string_fns.h> /* strlcpy */
+
+#ifdef RTE_EXEC_ENV_FREEBSD
 #define self "curproc"
 #define exe "file"
 #else
 #define exe "exe"
 #endif
 
+#ifdef RTE_LIBRTE_PDUMP
+#ifdef RTE_LIBRTE_RING_PMD
+#include <pthread.h>
+extern void *send_pkts(void *empty);
+extern uint16_t flag_for_send_pkts;
+#endif
+#endif
+
 /*
  * launches a second copy of the test process using the given argv parameters,
  * which should include argv[0] as the process name. To identify in the
@@ -54,13 +42,14 @@ static inline int
 process_dup(const char *const argv[], int numargs, const char *env_value)
 {
        int num;
-#ifdef RTE_LIBRTE_XEN_DOM0
-       char *argv_cpy[numargs + 2];
-#else
        char *argv_cpy[numargs + 1];
-#endif
-       int i, fd, status;
+       int i, status;
        char path[32];
+#ifdef RTE_LIBRTE_PDUMP
+#ifdef RTE_LIBRTE_RING_PMD
+       pthread_t thread;
+#endif
+#endif
 
        pid_t pid = fork();
        if (pid < 0)
@@ -69,22 +58,53 @@ process_dup(const char *const argv[], int numargs, const char *env_value)
                /* make a copy of the arguments to be passed to exec */
                for (i = 0; i < numargs; i++)
                        argv_cpy[i] = strdup(argv[i]);
-       #ifdef RTE_LIBRTE_XEN_DOM0
-               argv_cpy[i] = strdup("--xen-dom0");
-               argv_cpy[i + 1] = NULL;
-               num = numargs + 1;
-       #else
                argv_cpy[i] = NULL;
                num = numargs;
-       #endif
 
-               /* close all open file descriptors, check /proc/self/fd to only
-                * call close on open fds. Exclude fds 0, 1 and 2*/
-               for (fd = getdtablesize(); fd > 2; fd-- ) {
-                       snprintf(path, sizeof(path), "/proc/" exe "/fd/%d", fd);
-                       if (access(path, F_OK) == 0)
+#ifdef RTE_EXEC_ENV_LINUX
+               {
+                       const char *procdir = "/proc/" self "/fd/";
+                       struct dirent *dirent;
+                       char *endptr;
+                       int fd, fdir;
+                       DIR *dir;
+
+                       /* close all open file descriptors, check /proc/self/fd
+                        * to only call close on open fds. Exclude fds 0, 1 and
+                        * 2
+                        */
+                       dir = opendir(procdir);
+                       if (dir == NULL) {
+                               rte_panic("Error opening %s: %s\n", procdir,
+                                               strerror(errno));
+                       }
+
+                       fdir = dirfd(dir);
+                       if (fdir < 0) {
+                               status = errno;
+                               closedir(dir);
+                               rte_panic("Error %d obtaining fd for dir %s: %s\n",
+                                               fdir, procdir,
+                                               strerror(status));
+                       }
+
+                       while ((dirent = readdir(dir)) != NULL) {
+                               errno = 0;
+                               fd = strtol(dirent->d_name, &endptr, 10);
+                               if (errno != 0 || endptr[0] != '\0') {
+                                       printf("Error converting name fd %d %s:\n",
+                                               fd, dirent->d_name);
+                                       continue;
+                               }
+
+                               if (fd == fdir || fd <= 2)
+                                       continue;
+
                                close(fd);
+                       }
+                       closedir(dir);
                }
+#endif
                printf("Running binary with argv[]:");
                for (i = 0; i < num; i++)
                        printf("'%s' ", argv_cpy[i]);
@@ -93,15 +113,59 @@ process_dup(const char *const argv[], int numargs, const char *env_value)
                /* set the environment variable */
                if (setenv(RECURSIVE_ENV_VAR, env_value, 1) != 0)
                        rte_panic("Cannot export environment variable\n");
-               if (execv("/proc/" self "/" exe, argv_cpy) < 0)
-                       rte_panic("Cannot exec\n");
+
+               strlcpy(path, "/proc/" self "/" exe, sizeof(path));
+               if (execv(path, argv_cpy) < 0) {
+                       if (errno == ENOENT) {
+                               printf("Could not find '%s', is procfs mounted?\n",
+                                               path);
+                       }
+                       rte_panic("Cannot exec: %s\n", strerror(errno));
+               }
        }
        /* parent process does a wait */
+#ifdef RTE_LIBRTE_PDUMP
+#ifdef RTE_LIBRTE_RING_PMD
+       if ((strcmp(env_value, "run_pdump_server_tests") == 0))
+               pthread_create(&thread, NULL, &send_pkts, NULL);
+#endif
+#endif
+
        while (wait(&status) != pid)
-                       ;
+               ;
+#ifdef RTE_LIBRTE_PDUMP
+#ifdef RTE_LIBRTE_RING_PMD
+       if ((strcmp(env_value, "run_pdump_server_tests") == 0)) {
+               flag_for_send_pkts = 0;
+               pthread_join(thread, NULL);
+       }
+#endif
+#endif
        return status;
 }
 
-#endif /* not baremetal */
+/* FreeBSD doesn't support file prefixes, so force compile failures for any
+ * tests attempting to use this function on FreeBSD.
+ */
+#ifdef RTE_EXEC_ENV_LINUX
+static char *
+get_current_prefix(char *prefix, int size)
+{
+       char path[PATH_MAX] = {0};
+       char buf[PATH_MAX] = {0};
+
+       /* get file for config (fd is always 3) */
+       snprintf(path, sizeof(path), "/proc/self/fd/%d", 3);
+
+       /* return NULL on error */
+       if (readlink(path, buf, sizeof(buf)) == -1)
+               return NULL;
+
+       /* get the prefix */
+       snprintf(prefix, size, "%s", basename(dirname(buf)));
+
+       return prefix;
+}
+#endif
 
 #endif /* _PROCESS_H_ */