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
  21 #include <unistd.h>
  22 #include <sys/prctl.h>
  23 #include <assert.h>
  24 #include <fcntl.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <signal.h>
  28 #include <sys/types.h>
  29 #include "common_handlers.h"
  30 #include "tcpdump.h"
  31
  32 pid_t tcpdump_start(char *interface) {
  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         pid_t tcpdump_pid = fork();
  40
  41         if(tcpdump_pid == 0) {
  42                 prctl(PR_SET_PDEATHSIG, SIGHUP);
  43                 close(pipes[1]);
  44                 // Open log file for TCP Dump
  45                 int fd = open(TCPDUMP_LOG_FILE, O_WRONLY | O_CREAT | O_TRUNC, 0644);
  46                 assert(fd != -1);
  47                 close(STDOUT_FILENO);
  48                 assert(dup2(fd, STDOUT_FILENO) != -1);
  49
  50                 // Launch TCPDump with port numbers of sleepy, gateway & relay
  51                 execvp("/usr/sbin/tcpdump", argv);
  52                 perror("execvp ");
  53                 exit(1);
  54         } else {
  55                 close(pipes[0]);
  56         }
  57
  58         return tcpdump_pid;
  59 }
  60
  61 void tcpdump_stop(pid_t tcpdump_pid) {
  62         PRINT_TEST_CASE_MSG("\n\x1b[32mStopping TCP Dump.\x1b[0m\n");
  63         assert(!kill(tcpdump_pid, SIGTERM));
  64 }