common: Windows 'spawn' support for tx and block notifications

This commit is contained in:
xiphon 2018-10-01 06:39:37 +00:00
parent 83d8f03c23
commit d5541e44fe
4 changed files with 65 additions and 11 deletions

View file

@ -34,13 +34,13 @@ set(common_sources
dns_utils.cpp dns_utils.cpp
download.cpp download.cpp
error.cpp error.cpp
exec.cpp
expect.cpp expect.cpp
util.cpp util.cpp
i18n.cpp i18n.cpp
notify.cpp notify.cpp
password.cpp password.cpp
perf_timer.cpp perf_timer.cpp
spawn.cpp
threadpool.cpp threadpool.cpp
updates.cpp updates.cpp
aligned.c) aligned.c)
@ -60,7 +60,6 @@ set(common_private_headers
dns_utils.h dns_utils.h
download.h download.h
error.h error.h
exec.h
expect.h expect.h
http_connection.h http_connection.h
int-util.h int-util.h
@ -74,6 +73,7 @@ set(common_private_headers
i18n.h i18n.h
password.h password.h
perf_timer.h perf_timer.h
spawn.h
stack_trace.h stack_trace.h
threadpool.h threadpool.h
updates.h updates.h

View file

@ -29,12 +29,17 @@
#include <boost/algorithm/string.hpp> #include <boost/algorithm/string.hpp>
#include "misc_log_ex.h" #include "misc_log_ex.h"
#include "file_io_utils.h" #include "file_io_utils.h"
#include "exec.h" #include "spawn.h"
#include "notify.h" #include "notify.h"
namespace tools namespace tools
{ {
/*
TODO:
- Improve tokenization to handle paths containing whitespaces, quotes, etc.
- Windows unicode support (implies implementing unicode command line parsing code)
*/
Notify::Notify(const char *spec) Notify::Notify(const char *spec)
{ {
CHECK_AND_ASSERT_THROW_MES(spec, "Null spec"); CHECK_AND_ASSERT_THROW_MES(spec, "Null spec");
@ -51,11 +56,7 @@ int Notify::notify(const char *parameter)
for (std::string &s: margs) for (std::string &s: margs)
boost::replace_all(s, "%s", parameter); boost::replace_all(s, "%s", parameter);
char **cargs = (char**)alloca(sizeof(char*) * (margs.size() + 1)); return tools::spawn(filename.c_str(), margs, false);
for (size_t n = 0; n < margs.size(); ++n)
cargs[n] = (char*)margs[n].c_str();
cargs[margs.size()] = NULL;
return tools::exec(filename.c_str(), cargs, false);
} }
} }

View file

@ -29,16 +29,68 @@
#include <errno.h> #include <errno.h>
#include <unistd.h> #include <unistd.h>
#include <sys/types.h> #include <sys/types.h>
#ifdef _WIN32
#include <boost/algorithm/string/join.hpp>
#include <boost/scope_exit.hpp>
#include <windows.h>
#else
#include <sys/wait.h> #include <sys/wait.h>
#endif
#include "misc_log_ex.h" #include "misc_log_ex.h"
#include "exec.h" #include "spawn.h"
namespace tools namespace tools
{ {
int exec(const char *filename, char * const argv[], bool wait) int spawn(const char *filename, const std::vector<std::string>& args, bool wait)
{ {
#ifdef _WIN32
std::string joined = boost::algorithm::join(args, " ");
char *commandLine = !joined.empty() ? &joined[0] : nullptr;
STARTUPINFOA si = {};
si.cb = sizeof(si);
PROCESS_INFORMATION pi;
if (!CreateProcessA(filename, commandLine, nullptr, nullptr, false, 0, nullptr, nullptr, &si, &pi))
{
MERROR("CreateProcess failed. Error code " << GetLastError());
return -1;
}
BOOST_SCOPE_EXIT(&pi)
{
CloseHandle(pi.hThread);
CloseHandle(pi.hProcess);
}
BOOST_SCOPE_EXIT_END
if (!wait)
{
return 0;
}
DWORD result = WaitForSingleObject(pi.hProcess, INFINITE);
if (result != WAIT_OBJECT_0)
{
MERROR("WaitForSingleObject failed. Result " << result << ", error code " << GetLastError());
return -1;
}
DWORD exitCode;
if (!GetExitCodeProcess(pi.hProcess, &exitCode))
{
MERROR("GetExitCodeProcess failed. Error code " << GetLastError());
return -1;
}
MINFO("Child exited with " << exitCode);
return static_cast<int>(exitCode);
#else
char **argv = (char**)alloca(sizeof(char*) * (args.size() + 1));
for (size_t n = 0; n < args.size(); ++n)
argv[n] = (char*)args[n].c_str();
argv[args.size()] = NULL;
pid_t pid = fork(); pid_t pid = fork();
if (pid < 0) if (pid < 0)
{ {
@ -83,6 +135,7 @@ int exec(const char *filename, char * const argv[], bool wait)
} }
MERROR("Secret passage found"); MERROR("Secret passage found");
return -1; return -1;
#endif
} }
} }

View file

@ -31,6 +31,6 @@
namespace tools namespace tools
{ {
int exec(const char *filename, char * const argv[], bool wait); int spawn(const char *filename, const std::vector<std::string>& args, bool wait);
} }