git.meshlink.io Git - meshlink/blob - test/blackbox/common/tcpdump.c

   1 /*
   2     tcpdump.c -- Implementation of Black Box Test Execution for meshlink
   3
   4     Copyright (C) 2018  Guus Sliepen <guus@meshlink.io>
   5
   6     This program is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     This program 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
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License along
  17     with this program; if not, write to the Free Software Foundation, Inc.,
  18     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 */
  20 #include <unistd.h>
  21 #include <sys/prctl.h>
  22 #include <assert.h>
  23 #include <fcntl.h>
  24 #include <stdio.h>
  25 #include <signal.h>
  26 #include <sys/types.h>
  27 #include "common_handlers.h"
  28 #include "containers.h"
  29 #include "tcpdump.h"
  30
  31 pid_t tcpdump_start(char *interface) {
  32         pid_t tcpdump_pid;
  33         char *argv[] = { "tcpdump", "-i", interface, NULL };
  34         // child process have a pipe to the parent process when parent process terminates SIGPIPE kills the tcpdump
  35         int pipes[2];
  36         assert(pipe(pipes) != -1);
  37         PRINT_TEST_CASE_MSG("\x1b[32mLaunching TCP Dump ..\x1b[0m\n");
  38
  39         if((tcpdump_pid = fork()) == 0) {
  40                 prctl(PR_SET_PDEATHSIG, SIGHUP);
  41                 close(pipes[1]);
  42                 // Open log file for TCP Dump
  43                 int fd = open(TCPDUMP_LOG_FILE, O_WRONLY | O_CREAT | O_TRUNC, 0644);
  44                 assert(fd != -1);
  45                 close(STDOUT_FILENO);
  46                 assert(dup2(fd, STDOUT_FILENO) != -1);
  47
  48                 // Launch TCPDump with port numbers of sleepy, gateway & relay
  49                 int ret = execvp("/usr/sbin/tcpdump", argv);
  50                 perror("execvp ");
  51                 assert(ret != -1);
  52         } else {
  53                 close(pipes[0]);
  54                 return tcpdump_pid;
  55         }
  56 }
  57
  58 void tcpdump_stop(pid_t tcpdump_pid) {
  59         PRINT_TEST_CASE_MSG("\n\x1b[32mStopping TCP Dump.\x1b[0m\n");
  60         assert(!kill(tcpdump_pid, SIGTERM));
  61 }