diff options
Diffstat (limited to 'src/sysExec.cpp')
| -rw-r--r-- | src/sysExec.cpp | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/src/sysExec.cpp b/src/sysExec.cpp new file mode 100644 index 0000000..9a397a2 --- /dev/null +++ b/src/sysExec.cpp @@ -0,0 +1,163 @@ +/* + * anytun + * + * The secure anycast tunneling protocol (satp) defines a protocol used + * for communication between any combination of unicast and anycast + * tunnel endpoints. It has less protocol overhead than IPSec in Tunnel + * mode and allows tunneling of every ETHER TYPE protocol (e.g. + * ethernet, ip, arp ...). satp directly includes cryptography and + * message authentication based on the methodes used by SRTP. It is + * intended to deliver a generic, scaleable and secure solution for + * tunneling and relaying of packets of any protocol. + * + * + * Copyright (C) 2007-2008 Othmar Gsenger, Erwin Nindl, + * Christian Pointner <satp@wirdorange.org> + * + * This file is part of Anytun. + * + * Anytun is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License version 3 as + * published by the Free Software Foundation. + * + * Anytun is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with anytun. If not, see <http://www.gnu.org/licenses/>. + */ + +#include <boost/bind.hpp> +#include <boost/thread.hpp> + +#include "datatypes.h" +#include "sysExec.h" +#include "log.h" +#include "anytunError.h" + +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <errno.h> +#include <sys/wait.h> +#include <sys/select.h> +#include <stdlib.h> +#include <string.h> + +void anytun_exec(std::string const& script) +{ + anytun_exec(script, StringVector(), StringList()); +} + +void anytun_exec(std::string const& script, StringVector const& args) +{ + anytun_exec(script, args, StringList()); +} + +void anytun_exec(std::string const& script, StringList const& env) +{ + anytun_exec(script, StringVector(), env); +} + +void anytun_exec(std::string const& script, StringVector const& args, StringList const& env) +{ + int pipefd[2]; + if(pipe(pipefd) == -1) { + cLog.msg(Log::PRIO_ERROR) << "executing script '" << script << "' pipe() error: " << AnytunErrno(errno); + return; + } + + pid_t pid; + pid = fork(); + if(pid == -1) { + cLog.msg(Log::PRIO_ERROR) << "executing script '" << script << "' fork() error: " << AnytunErrno(errno); + return; + } + + if(pid) { + close(pipefd[1]); + boost::thread(boost::bind(waitForScript, script, pid, pipefd[0])); + return; + } + +// child code + int fd; + for (fd=getdtablesize();fd>=0;--fd) // close all file descriptors + if(fd != pipefd[1]) close(fd); + + fd = open("/dev/null",O_RDWR); // stdin + if(fd == -1) + cLog.msg(Log::PRIO_WARNING) << "can't open stdin"; + else { + if(dup(fd) == -1) // stdout + cLog.msg(Log::PRIO_WARNING) << "can't open stdout"; + if(dup(fd) == -1) // stderr + cLog.msg(Log::PRIO_WARNING) << "can't open stderr"; + } + + char** argv = new char*[args.size() + 2]; + argv[0] = new char[script.size() + 1]; + std::strcpy(argv[0], script.c_str()); + for(unsigned int i=0; i<args.size(); ++i) { + argv[i+1] = new char[args[i].size() + 1]; + std::strcpy(argv[i+1], args[i].c_str()); + } + argv[args.size() + 1] = new char[1]; + argv[args.size() + 1][0] = NULL; + + char** evp; + if(env.size()) { + evp = new char*[env.size() + 1]; + unsigned int i = 0; + for(StringList::const_iterator it = env.begin(); it != env.end(); ++it) { + evp[i] = new char[it->size() + 1]; + std::strcpy(evp[i], it->c_str()); + ++i; + } + evp[env.size()] = new char[1]; + evp[env.size()][0] = NULL; + } else { + evp = new char*[1]; + evp[0] = new char[1]; + evp[0][0] = NULL; + } + + execve(script.c_str(), argv, evp); + // if execve returns, an error occurred, but logging doesn't work + // because we closed all file descriptors, so just write errno to + // pipe and call exit + int err = errno; + int ret = write(pipefd[1], (void*)(&err), sizeof(err)); + if(ret != sizeof(errno)) + exit(-2); + exit(-1); +} + +void waitForScript(std::string const& script, pid_t pid, int pipefd) +{ + int status = 0; + waitpid(pid, &status, 0); + + fd_set rfds; + FD_ZERO(&rfds); + FD_SET(pipefd, &rfds); + struct timeval tv = { 0 , 0 }; + if(select(pipefd+1, &rfds, NULL, NULL, &tv) == 1) { + int err = 0; + if(read(pipefd, (void*)(&err), sizeof(err)) >= static_cast<int>(sizeof(err))) { + cLog.msg(Log::PRIO_NOTICE) << "script '" << script << "' exec() error: " << AnytunErrno(err); + close(pipefd); + return; + } + } + if(WIFEXITED(status)) + cLog.msg(Log::PRIO_NOTICE) << "script '" << script << "' returned " << WEXITSTATUS(status); + else if(WIFSIGNALED(status)) + cLog.msg(Log::PRIO_NOTICE) << "script '" << script << "' terminated after signal " << WTERMSIG(status); + else + cLog.msg(Log::PRIO_ERROR) << "executing script '" << script << "': unkown error"; + + close(pipefd); +} |