mirror of
https://github.com/TrinityCore/TrinityCore.git
synced 2026-01-15 23:20:36 +01:00
Dep: Switch to boost process instead of old standalone version
(cherry picked from commit 524d167398)
This commit is contained in:
@@ -24,10 +24,15 @@
|
||||
|
||||
#include <boost/algorithm/string/join.hpp>
|
||||
#include <boost/iostreams/copy.hpp>
|
||||
#include <boost/process.hpp>
|
||||
#include <boost/process/args.hpp>
|
||||
#include <boost/process/child.hpp>
|
||||
#include <boost/process/env.hpp>
|
||||
#include <boost/process/exe.hpp>
|
||||
#include <boost/process/io.hpp>
|
||||
#include <boost/process/pipe.hpp>
|
||||
#include <boost/process/search_path.hpp>
|
||||
|
||||
using namespace boost::process;
|
||||
using namespace boost::process::initializers;
|
||||
using namespace boost::iostreams;
|
||||
|
||||
namespace Trinity
|
||||
@@ -48,7 +53,10 @@ public:
|
||||
|
||||
std::streamsize write(char const* str, std::streamsize size)
|
||||
{
|
||||
callback_(std::string(str, size));
|
||||
std::string consoleStr(str, size);
|
||||
std::string utf8;
|
||||
if (consoleToUtf8(consoleStr, utf8))
|
||||
callback_(utf8);
|
||||
return size;
|
||||
}
|
||||
};
|
||||
@@ -62,62 +70,60 @@ auto MakeTCLogSink(T&& callback)
|
||||
|
||||
template<typename T>
|
||||
static int CreateChildProcess(T waiter, std::string const& executable,
|
||||
std::vector<std::string> const& args,
|
||||
std::vector<std::string> const& argsVector,
|
||||
std::string const& logger, std::string const& input,
|
||||
bool secure)
|
||||
{
|
||||
auto outPipe = create_pipe();
|
||||
auto errPipe = create_pipe();
|
||||
|
||||
Optional<file_descriptor_source> inputSource;
|
||||
ipstream outStream;
|
||||
ipstream errStream;
|
||||
|
||||
if (!secure)
|
||||
{
|
||||
TC_LOG_TRACE(logger, "Starting process \"%s\" with arguments: \"%s\".",
|
||||
executable.c_str(), boost::algorithm::join(args, " ").c_str());
|
||||
executable.c_str(), boost::algorithm::join(argsVector, " ").c_str());
|
||||
}
|
||||
|
||||
// Start the child process
|
||||
child c = [&]
|
||||
child c = [&]()
|
||||
{
|
||||
if (!input.empty())
|
||||
{
|
||||
inputSource = file_descriptor_source(input);
|
||||
|
||||
// With binding stdin
|
||||
return execute(run_exe(boost::filesystem::absolute(executable)),
|
||||
set_args(args),
|
||||
inherit_env(),
|
||||
bind_stdin(*inputSource),
|
||||
bind_stdout(file_descriptor_sink(outPipe.sink, close_handle)),
|
||||
bind_stderr(file_descriptor_sink(errPipe.sink, close_handle)));
|
||||
return child{
|
||||
exe = boost::filesystem::absolute(executable).string(),
|
||||
args = argsVector,
|
||||
env = environment(boost::this_process::environment()),
|
||||
std_in = input,
|
||||
std_out = outStream,
|
||||
std_err = errStream
|
||||
};
|
||||
}
|
||||
else
|
||||
{
|
||||
// Without binding stdin
|
||||
return execute(run_exe(boost::filesystem::absolute(executable)),
|
||||
set_args(args),
|
||||
inherit_env(),
|
||||
bind_stdout(file_descriptor_sink(outPipe.sink, close_handle)),
|
||||
bind_stderr(file_descriptor_sink(errPipe.sink, close_handle)));
|
||||
return child{
|
||||
exe = boost::filesystem::absolute(executable).string(),
|
||||
args = argsVector,
|
||||
env = environment(boost::this_process::environment()),
|
||||
std_in = boost::process::close,
|
||||
std_out = outStream,
|
||||
std_err = errStream
|
||||
};
|
||||
}
|
||||
}();
|
||||
|
||||
file_descriptor_source outFd(outPipe.source, close_handle);
|
||||
file_descriptor_source errFd(errPipe.source, close_handle);
|
||||
|
||||
auto outInfo = MakeTCLogSink([&](std::string msg)
|
||||
auto outInfo = MakeTCLogSink([&](std::string const& msg)
|
||||
{
|
||||
TC_LOG_INFO(logger, "%s", msg.c_str());
|
||||
});
|
||||
|
||||
auto outError = MakeTCLogSink([&](std::string msg)
|
||||
auto outError = MakeTCLogSink([&](std::string const& msg)
|
||||
{
|
||||
TC_LOG_ERROR(logger, "%s", msg.c_str());
|
||||
});
|
||||
|
||||
copy(outFd, outInfo);
|
||||
copy(errFd, outError);
|
||||
copy(outStream, outInfo);
|
||||
copy(errStream, outError);
|
||||
|
||||
// Call the waiter in the current scope to prevent
|
||||
// the streams from closing too early on leaving the scope.
|
||||
@@ -129,9 +135,6 @@ static int CreateChildProcess(T waiter, std::string const& executable,
|
||||
executable.c_str(), result);
|
||||
}
|
||||
|
||||
if (inputSource)
|
||||
inputSource->close();
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@@ -142,7 +145,8 @@ int StartProcess(std::string const& executable, std::vector<std::string> const&
|
||||
{
|
||||
try
|
||||
{
|
||||
return wait_for_exit(c);
|
||||
c.wait();
|
||||
return c.exit_code();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
@@ -190,7 +194,8 @@ public:
|
||||
|
||||
try
|
||||
{
|
||||
result = wait_for_exit(c);
|
||||
c.wait();
|
||||
result = c.exit_code();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
@@ -224,7 +229,7 @@ public:
|
||||
was_terminated = true;
|
||||
try
|
||||
{
|
||||
terminate(my_child->get());
|
||||
my_child->get().terminate();
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
@@ -249,7 +254,7 @@ std::string SearchExecutableInPath(std::string const& filename)
|
||||
{
|
||||
try
|
||||
{
|
||||
return search_path(filename);
|
||||
return search_path(filename).string();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user