src/posix/sysExec.hpp

   1 /*
   2  *  anytun
   3  *
   4  *  The secure anycast tunneling protocol (satp) defines a protocol used
   5  *  for communication between any combination of unicast and anycast
   6  *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
   7  *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
   8  *  ethernet, ip, arp ...). satp directly includes cryptography and
   9  *  message authentication based on the methods used by SRTP.  It is
  10  *  intended to deliver a generic, scaleable and secure solution for
  11  *  tunneling and relaying of packets of any protocol.
  12  *
  13  *
  14  *  Copyright (C) 2007-2014 Markus Grüneis, Othmar Gsenger, Erwin Nindl,
  15  *                          Christian Pointner <satp@wirdorange.org>
  16  *
  17  *  This file is part of Anytun.
  18  *
  19  *  Anytun is free software: you can redistribute it and/or modify
  20  *  it under the terms of the GNU General Public License as published by
  21  *  the Free Software Foundation, either version 3 of the License, or
  22  *  any later version.
  23  *
  24  *  Anytun is distributed in the hope that it will be useful,
  25  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  26  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  27  *  GNU General Public License for more details.
  28  *
  29  *  You should have received a copy of the GNU General Public License
  30  *  along with Anytun.  If not, see <http://www.gnu.org/licenses/>.
  31  *
  32  *  In addition, as a special exception, the copyright holders give
  33  *  permission to link the code of portions of this program with the
  34  *  OpenSSL library under certain conditions as described in each
  35  *  individual source file, and distribute linked combinations
  36  *  including the two.
  37  *  You must obey the GNU General Public License in all respects
  38  *  for all of the code used other than OpenSSL.  If you modify
  39  *  file(s) with this exception, you may extend this exception to your
  40  *  version of the file(s), but you are not obligated to do so.  If you
  41  *  do not wish to do so, delete this exception statement from your
  42  *  version.  If you delete this exception statement from all source
  43  *  files in the program, then also delete it here.
  44  */
  45
  46 #pragma once
  47 #ifndef ANYTUN_sysexec_hpp_INCLUDED
  48 #define ANYTUN_sysexec_hpp_INCLUDED
  49
  50 #include <sys/types.h>
  51 #include <sys/stat.h>
  52 #include <fcntl.h>
  53 #include <errno.h>
  54 #include <sys/wait.h>
  55 #include <sys/select.h>
  56 #include <stdlib.h>
  57 #include <string.h>
  58 #include <cstring>
  59
  60 SysExec::~SysExec()
  61 {
  62   if(!closed_) {
  63     close(pipefd_);
  64   }
  65 }
  66
  67
  68 template<class T>
  69 char** dupSysStringArray(T const& array)
  70 {
  71   char** new_array;
  72   new_array = static_cast<char**>(malloc((array.size() + 1)*sizeof(char*)));
  73   if(!new_array) {
  74     return NULL;
  75   }
  76
  77   unsigned int i = 0;
  78   for(typename T::const_iterator it = array.begin(); it != array.end(); ++it) {
  79     new_array[i] = strdup(it->c_str());
  80     if(!new_array) {
  81       while(i--) {
  82         free(new_array[i]);
  83       }
  84       free(new_array);
  85       return NULL;
  86     }
  87     ++i;
  88   }
  89   new_array[array.size()] = NULL;
  90   return new_array;
  91 }
  92
  93 void freeSysStringArray(char** array)
  94 {
  95   if(!array) {
  96     return;
  97   }
  98
  99   for(int i=0; array[i] ; ++i) {
 100     free(array[i]);
 101   }
 102
 103   free(array);
 104 }
 105
 106 void SysExec::doExec(StringVector args, StringList env)
 107 {
 108   int pipefd[2];
 109   if(pipe(pipefd) == -1) {
 110     cLog.msg(Log::PRIO_ERROR) << "executing script '" << script_ << "' pipe() error: " << AnytunErrno(errno);
 111     return;
 112   }
 113
 114   pid_ = fork();
 115   if(pid_ == -1) {
 116     cLog.msg(Log::PRIO_ERROR) << "executing script '" << script_ << "' fork() error: " << AnytunErrno(errno);
 117     return;
 118   }
 119
 120   if(pid_) {
 121     close(pipefd[1]);
 122     pipefd_=pipefd[0];
 123     // parent exits here, call waitForScript to cleanup up zombie
 124     return;
 125   }
 126   // child code, exec the script
 127   int fd;
 128   for(fd=getdtablesize(); fd>=0; --fd) // close all file descriptors
 129     if(fd != pipefd[1]) { close(fd); }
 130
 131   fd = open("/dev/null",O_RDWR);        // stdin
 132   if(fd == -1) {
 133     cLog.msg(Log::PRIO_WARNING) << "can't open stdin";
 134   } else {
 135     if(dup(fd) == -1) { // stdout
 136       cLog.msg(Log::PRIO_WARNING) << "can't open stdout";
 137     }
 138     if(dup(fd) == -1) { // stderr
 139       cLog.msg(Log::PRIO_WARNING) << "can't open stderr";
 140     }
 141   }
 142
 143   args.insert(args.begin(), script_);
 144   char** argv = dupSysStringArray(args);
 145   char** evp = dupSysStringArray(env);
 146
 147   execve(script_.c_str(), argv, evp);
 148   // if execve returns, an error occurred, but logging doesn't work
 149   // because we closed all file descriptors, so just write errno to
 150   // pipe and call exit
 151
 152   freeSysStringArray(argv);
 153   freeSysStringArray(evp);
 154
 155   int err = errno;
 156   int ret = write(pipefd[1], (void*)(&err), sizeof(err));
 157   if(ret != sizeof(errno)) {
 158     exit(-2);
 159   }
 160   exit(-1);
 161 }
 162
 163 int SysExec::waitForScript()
 164 {
 165   int status = 0;
 166   waitpid(pid_, &status, 0);
 167
 168   fd_set rfds;
 169   FD_ZERO(&rfds);
 170   FD_SET(pipefd_, &rfds);
 171   struct timeval tv = { 0 , 0 };
 172   if(select(pipefd_+1, &rfds, NULL, NULL, &tv) == 1) {
 173     int err = 0;
 174     if(read(pipefd_, (void*)(&err), sizeof(err)) >= static_cast<int>(sizeof(err))) {
 175       cLog.msg(Log::PRIO_ERROR) << "script '" << script_ << "' exec() error: " << AnytunErrno(err);
 176       close(pipefd_);
 177       return_code_ = -1;
 178       return return_code_;
 179     }
 180   }
 181
 182   close(pipefd_);
 183   closed_ = true;
 184   return_code_ = status;
 185
 186   return return_code_;
 187 }
 188
 189 void SysExec::waitAndDestroy(SysExec*& s)
 190 {
 191   if(!s) {
 192     return;
 193   }
 194
 195   s->waitForScript();
 196   if(WIFEXITED(s->return_code_)) {
 197     cLog.msg(Log::PRIO_NOTICE) << "script '" << s->script_ << "' returned " << WEXITSTATUS(s->return_code_);
 198   } else if(WIFSIGNALED(s->return_code_)) {
 199     cLog.msg(Log::PRIO_NOTICE) << "script '" << s->script_ << "' terminated after signal " << WTERMSIG(s->return_code_);
 200   } else if(WIFSTOPPED(s->return_code_)) {
 201     cLog.msg(Log::PRIO_NOTICE) << "script '" << s->script_ << "' stopped after signal " << WSTOPSIG(s->return_code_);
 202   } else if(WIFCONTINUED(s->return_code_)) {
 203     cLog.msg(Log::PRIO_NOTICE) << "script '" << s->script_ << "' continued after SIGCONT";
 204   }
 205
 206   delete(s);
 207   s = NULL;
 208 }
 209
 210 #endif // ANYTUN_sysexec_hpp_INCLUDED