win32 support for sysExec, sysExec.hpp arch-specific files introduced

1 files changed, 160 insertions, 0 deletions

diff --git a/src/win32/sysExec.hpp b/src/win32/sysExec.hpp
new file mode 100644
index 0000000..d04ed40
--- /dev/null
+++ b/src/win32/sysExec.hpp
@@ -0,0 +1,160 @@
+/*

+ *  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-2009 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 as published by

+ *  the Free Software Foundation, either version 3 of the License, or

+ *  any later version.

+ *

+ *  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/>.

+ */

+#pragma once

+#ifndef ANYTUN_sysexec_hpp_INCLUDED

+#define ANYTUN_sysexec_hpp_INCLUDED

+

+#include <algorithm>

+#include <iostream> // todo remove

+#include <windows.h>

+

+SysExec::~SysExec()

+{

+  if(!closed_) {

+    CloseHandle(process_info_.hProcess);

+    CloseHandle(process_info_.hThread);

+  }

+}

+

+STARTUPINFOA getStartupInfo() {

+  STARTUPINFOA startup_info;

+  startup_info.cb = sizeof(STARTUPINFOA);

+  GetStartupInfoA(&startup_info);

+

+  //startup_info.dwFlags = STARTF_USESTDHANDLES;

+  //startup_info.hStdInput = CreateFile("NUL", GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, 0, 0, 0); // INVALID_HANDLE_VALUE;

+  //startup_info.hStdOutput = CreateFile("NUL", GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, 0, 0, 0); // INVALID_HANDLE_VALUE;

+  //startup_info.hStdError = CreateFile("NUL", GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, 0, 0, 0); // INVALID_HANDLE_VALUE;

+  startup_info.dwFlags |= STARTF_USESHOWWINDOW;

+  startup_info.wShowWindow = SW_HIDE;

+  

+  return startup_info;

+}

+

+char const * const BATCH_FILE_EXTS[] = { ".bat", ".cmd" };

+int const BATCH_FILE_EXTS_COUNT = sizeof(BATCH_FILE_EXTS) / sizeof(BATCH_FILE_EXTS[0]);

+

+bool endsWith(std::string const& string, std::string const& suffix) {

+  return string.find(suffix, string.size() - suffix.size()) != std::string::npos;

+}

+

+void SysExec::doExec(std::string const& script, StringVector const& args, StringList const& env)

+{

+  std::vector<char> arguments;

+  

+  bool isBatchFile = false;

+  for(int i = 0; i < BATCH_FILE_EXTS_COUNT; ++i) {

+    if(endsWith(script, BATCH_FILE_EXTS[i])) {

+      isBatchFile = true;

+      break;

+    }

+  }

+

+  if(isBatchFile) {

+    std::string const BATCH_INTERPRETER = "cmd /c \"";

+    arguments.insert(arguments.end(), BATCH_INTERPRETER.begin(), BATCH_INTERPRETER.end());

+  }

+  arguments.push_back('\"');

+  arguments.insert(arguments.end(), script.begin(), script.end());

+  arguments.push_back('\"');

+  arguments.push_back(' ');

+  

+  for(StringVector::const_iterator it = args.begin(); it != args.end(); ++it) {

+    arguments.push_back('\"');

+    arguments.insert(arguments.end(), it->begin(), it->end());

+    arguments.push_back('\"');

+    arguments.push_back(' ');

+  }

+

+  if(isBatchFile) {

+    arguments.push_back('\"');

+  }

+  arguments.push_back(0);

+  

+  STARTUPINFOA startup_info = getStartupInfo();

+

+  std::map<std::string, std::string> envDict;

+  for(StringList::const_iterator it = env.begin(); it != env.begin(); ++it) {

+    size_t delimiter_pos = it->find('=');

+    envDict.insert(std::make_pair(it->substr(0, delimiter_pos), it->substr(delimiter_pos + 1)));

+  }

+  std::vector<char> env;

+  for(std::map<std::string, std::string>::iterator it = envDict.begin(); it != envDict.end(); ++it) {

+    env.insert(env.end(), it->first.begin(), it->first.end());

+    env.push_back(0);

+  }

+  env.push_back(0);

+

+  if(!CreateProcessA(NULL,

+                   &arguments[0],

+                   NULL,

+                   NULL,

+                   false,

+                   NULL,

+                   &env[0],

+                   NULL,

+                   &startup_info,

+                   &process_info_

+                   ))

+  {

+    cLog.msg(Log::PRIO_ERROR) << "executing script '" << script << "' CreateProcess() error: " << GetLastError();

+    return;

+  }

+}

+

+int SysExec::waitForScript()

+{

+  DWORD result = WaitForSingleObject(process_info_.hProcess, INFINITE);

+  assert(WAIT_OBJECT_0 == result); // WAIT_FAILED, WAIT_TIMEOUT ... ???

+  bool success = GetExitCodeProcess(process_info_.hProcess, &return_code_) != 0;

+  assert(true == success); // false -> HU?

+

+  CloseHandle(process_info_.hProcess);

+  CloseHandle(process_info_.hThread);

+  closed_ = true;

+

+  return static_cast<int>(return_code_);

+}

+

+void SysExec::waitAndDestroy(SysExec*& s)

+{

+  if(!s)

+    return;

+

+  s->waitForScript();

+  cLog.msg(Log::PRIO_NOTICE) << "script '" << s->script_ << "' returned " << s->return_code_;

+

+  delete(s);

+  s = NULL;

+}

+

+#endif // ANYTUN_sysexec_hpp_INCLUDED