mirror of
https://github.com/mod-playerbots/azerothcore-wotlk.git
synced 2026-01-15 01:59:09 +00:00
feat(Core/DBUpdater): implement db auto update (#6576)
* feat(Core/DBUpdater): implement db auto update * 1 * 2 * 3 * Some minor improvements * add find bin for mysql 8.0 * lic Co-authored-by: Kitzunu <24550914+Kitzunu@users.noreply.github.com>
This commit is contained in:
Kargatum
257
src/common/Utilities/StartProcess.cpp
Normal file
257
src/common/Utilities/StartProcess.cpp
Normal file
@@ -0,0 +1,257 @@
|
||||
/*
|
||||
* Copyright (C) 2016+ AzerothCore <www.azerothcore.org>, released under GNU AGPL v3 license: https://github.com/azerothcore/azerothcore-wotlk/blob/master/LICENSE-AGPL3
|
||||
* Copyright (C) 2021+ WarheadCore <https://github.com/WarheadCore>
|
||||
*/
|
||||
|
||||
#include "StartProcess.h"
|
||||
#include "Errors.h"
|
||||
#include "Log.h"
|
||||
#include "Optional.h"
|
||||
#include "Util.h"
|
||||
#include <boost/algorithm/string/join.hpp>
|
||||
#include <boost/iostreams/copy.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::iostreams;
|
||||
|
||||
namespace Acore
|
||||
{
|
||||
template<typename T>
|
||||
class ACLogSink
|
||||
{
|
||||
T callback_;
|
||||
|
||||
public:
|
||||
typedef char char_type;
|
||||
typedef sink_tag category;
|
||||
|
||||
// Requires a callback type which has a void(std::string) signature
|
||||
ACLogSink(T callback)
|
||||
: callback_(std::move(callback)) { }
|
||||
|
||||
std::streamsize write(char const* str, std::streamsize size)
|
||||
{
|
||||
std::string consoleStr(str, size);
|
||||
std::string utf8;
|
||||
if (consoleToUtf8(consoleStr, utf8))
|
||||
callback_(utf8);
|
||||
return size;
|
||||
}
|
||||
};
|
||||
|
||||
template<typename T>
|
||||
auto MakeACLogSink(T&& callback)
|
||||
-> ACLogSink<typename std::decay<T>::type>
|
||||
{
|
||||
return { std::forward<T>(callback) };
|
||||
}
|
||||
|
||||
template<typename T>
|
||||
static int CreateChildProcess(T waiter, std::string const& executable,
|
||||
std::vector<std::string> const& argsVector,
|
||||
std::string const& logger, std::string const& input,
|
||||
bool secure)
|
||||
{
|
||||
ipstream outStream;
|
||||
ipstream errStream;
|
||||
|
||||
if (!secure)
|
||||
{
|
||||
LOG_TRACE(logger, "Starting process \"%s\" with arguments: \"%s\".",
|
||||
executable.c_str(), boost::algorithm::join(argsVector, " ").c_str());
|
||||
}
|
||||
|
||||
// prepare file with only read permission (boost process opens with read_write)
|
||||
std::shared_ptr<FILE> inputFile(!input.empty() ? fopen(input.c_str(), "rb") : nullptr, [](FILE* ptr)
|
||||
{
|
||||
if (ptr != nullptr)
|
||||
fclose(ptr);
|
||||
});
|
||||
|
||||
// Start the child process
|
||||
child c = [&]()
|
||||
{
|
||||
if (inputFile)
|
||||
{
|
||||
// With binding stdin
|
||||
return child{
|
||||
exe = boost::filesystem::absolute(executable).string(),
|
||||
args = argsVector,
|
||||
env = environment(boost::this_process::environment()),
|
||||
std_in = inputFile.get(),
|
||||
std_out = outStream,
|
||||
std_err = errStream
|
||||
};
|
||||
}
|
||||
else
|
||||
{
|
||||
// Without binding stdin
|
||||
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
|
||||
};
|
||||
}
|
||||
}();
|
||||
|
||||
auto outInfo = MakeACLogSink([&](std::string const& msg)
|
||||
{
|
||||
LOG_INFO(logger, "%s", msg.c_str());
|
||||
});
|
||||
|
||||
auto outError = MakeACLogSink([&](std::string const& msg)
|
||||
{
|
||||
LOG_ERROR(logger, "%s", msg.c_str());
|
||||
});
|
||||
|
||||
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.
|
||||
int const result = waiter(c);
|
||||
|
||||
if (!secure)
|
||||
{
|
||||
LOG_TRACE(logger, ">> Process \"%s\" finished with return value %i.",
|
||||
executable.c_str(), result);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
int StartProcess(std::string const& executable, std::vector<std::string> const& args,
|
||||
std::string const& logger, std::string input_file, bool secure)
|
||||
{
|
||||
return CreateChildProcess([](child& c) -> int
|
||||
{
|
||||
try
|
||||
{
|
||||
c.wait();
|
||||
return c.exit_code();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
return EXIT_FAILURE;
|
||||
}
|
||||
}, executable, args, logger, input_file, secure);
|
||||
}
|
||||
|
||||
class AsyncProcessResultImplementation
|
||||
: public AsyncProcessResult
|
||||
{
|
||||
std::string const executable;
|
||||
std::vector<std::string> const args;
|
||||
std::string const logger;
|
||||
std::string const input_file;
|
||||
bool const is_secure;
|
||||
|
||||
std::atomic<bool> was_terminated;
|
||||
|
||||
// Workaround for missing move support in boost < 1.57
|
||||
Optional<std::shared_ptr<std::future<int>>> result;
|
||||
Optional<std::reference_wrapper<child>> my_child;
|
||||
|
||||
public:
|
||||
explicit AsyncProcessResultImplementation(std::string executable_, std::vector<std::string> args_,
|
||||
std::string logger_, std::string input_file_,
|
||||
bool secure)
|
||||
: executable(std::move(executable_)), args(std::move(args_)),
|
||||
logger(std::move(logger_)), input_file(input_file_),
|
||||
is_secure(secure), was_terminated(false) { }
|
||||
|
||||
AsyncProcessResultImplementation(AsyncProcessResultImplementation const&) = delete;
|
||||
AsyncProcessResultImplementation& operator= (AsyncProcessResultImplementation const&) = delete;
|
||||
AsyncProcessResultImplementation(AsyncProcessResultImplementation&&) = delete;
|
||||
AsyncProcessResultImplementation& operator= (AsyncProcessResultImplementation&&) = delete;
|
||||
|
||||
int StartProcess()
|
||||
{
|
||||
ASSERT(!my_child, "Process started already!");
|
||||
|
||||
return CreateChildProcess([&](child& c) -> int
|
||||
{
|
||||
int result;
|
||||
my_child = std::reference_wrapper<child>(c);
|
||||
|
||||
try
|
||||
{
|
||||
c.wait();
|
||||
result = c.exit_code();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
result = EXIT_FAILURE;
|
||||
}
|
||||
|
||||
my_child.reset();
|
||||
return was_terminated ? EXIT_FAILURE : result;
|
||||
|
||||
}, executable, args, logger, input_file, is_secure);
|
||||
}
|
||||
|
||||
void SetFuture(std::future<int> result_)
|
||||
{
|
||||
result = std::make_shared<std::future<int>>(std::move(result_));
|
||||
}
|
||||
|
||||
/// Returns the future which contains the result of the process
|
||||
/// as soon it is finished.
|
||||
std::future<int>& GetFutureResult() override
|
||||
{
|
||||
ASSERT(*result, "The process wasn't started!");
|
||||
return **result;
|
||||
}
|
||||
|
||||
/// Tries to terminate the process
|
||||
void Terminate() override
|
||||
{
|
||||
if (my_child)
|
||||
{
|
||||
was_terminated = true;
|
||||
try
|
||||
{
|
||||
my_child->get().terminate();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
// Do nothing
|
||||
}
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
std::shared_ptr<AsyncProcessResult>
|
||||
StartAsyncProcess(std::string executable, std::vector<std::string> args,
|
||||
std::string logger, std::string input_file, bool secure)
|
||||
{
|
||||
auto handle = std::make_shared<AsyncProcessResultImplementation>(
|
||||
std::move(executable), std::move(args), std::move(logger), std::move(input_file), secure);
|
||||
|
||||
handle->SetFuture(std::async(std::launch::async, [handle] { return handle->StartProcess(); }));
|
||||
return handle;
|
||||
}
|
||||
|
||||
std::string SearchExecutableInPath(std::string const& filename)
|
||||
{
|
||||
try
|
||||
{
|
||||
return search_path(filename).string();
|
||||
}
|
||||
catch (...)
|
||||
{
|
||||
return "";
|
||||
}
|
||||
}
|
||||
|
||||
} // namespace Acore
|
||||
Reference in New Issue
Block a user