mirror of https://github.com/oxen-io/lokinet
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.4 KiB
C++
58 lines
1.4 KiB
C++
#include "exec.hpp"
|
|
#include "exception.hpp"
|
|
#include <llarp/util/logging.hpp>
|
|
|
|
#include <array>
|
|
|
|
namespace llarp::win32
|
|
{
|
|
namespace
|
|
{
|
|
auto logcat = log::Cat("win32:exec");
|
|
|
|
/// get the directory for system32 which contains all the executables we use
|
|
std::string
|
|
SystemExeDir()
|
|
{
|
|
std::array<char, MAX_PATH + 1> path{};
|
|
|
|
if (GetSystemDirectoryA(path.data(), path.size()))
|
|
return path.data();
|
|
|
|
return "C:\\Windows\\system32";
|
|
}
|
|
|
|
} // namespace
|
|
|
|
void
|
|
Exec(std::string exe, std::string args)
|
|
{
|
|
OneShotExec{exe, args};
|
|
}
|
|
|
|
OneShotExec::OneShotExec(std::string cmd, std::chrono::milliseconds timeout)
|
|
: _si{}, _pi{}, _timeout{static_cast<DWORD>(timeout.count())}
|
|
{
|
|
log::info(logcat, "exec: {}", cmd);
|
|
if (not CreateProcessA(
|
|
nullptr, cmd.data(), nullptr, nullptr, false, 0, nullptr, nullptr, &_si, &_pi))
|
|
throw win32::error(GetLastError(), "failed to execute subprocess");
|
|
}
|
|
|
|
OneShotExec::~OneShotExec()
|
|
{
|
|
WaitForSingleObject(_pi.hProcess, _timeout);
|
|
CloseHandle(_pi.hProcess);
|
|
CloseHandle(_pi.hThread);
|
|
}
|
|
|
|
OneShotExec::OneShotExec(std::string cmd, std::string args, std::chrono::milliseconds timeout)
|
|
: OneShotExec{fmt::format("{}\\{} {}", SystemExeDir(), cmd, args), timeout}
|
|
{}
|
|
|
|
DeferExec::~DeferExec()
|
|
{
|
|
OneShotExec{std::move(_exe), std::move(_args), std::move(_timeout)};
|
|
}
|
|
} // namespace llarp::win32
|