mirror of
https://git.suyu.dev/suyu/suyu.git
synced 2024-12-30 12:20:58 +01:00
345 lines
14 KiB
C++
345 lines
14 KiB
C++
|
// Copyright (c) 2016 Klemens D. Morgenstern
|
||
|
//
|
||
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
||
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
||
|
|
||
|
#ifndef BOOST_PROCESS_EXTENSIONS_HPP_
|
||
|
#define BOOST_PROCESS_EXTENSIONS_HPP_
|
||
|
|
||
|
#include <boost/process/detail/handler.hpp>
|
||
|
#include <boost/process/detail/used_handles.hpp>
|
||
|
#include <memory>
|
||
|
|
||
|
#if defined(BOOST_WINDOWS_API)
|
||
|
#include <boost/process/detail/windows/executor.hpp>
|
||
|
#include <boost/process/detail/windows/async_handler.hpp>
|
||
|
#include <boost/process/detail/windows/asio_fwd.hpp>
|
||
|
#else
|
||
|
#include <boost/process/detail/posix/executor.hpp>
|
||
|
#include <boost/process/detail/posix/async_handler.hpp>
|
||
|
#include <boost/process/detail/posix/asio_fwd.hpp>
|
||
|
#endif
|
||
|
|
||
|
|
||
|
/** \file boost/process/extend.hpp
|
||
|
*
|
||
|
* This header which provides the types and functions provided for custom extensions.
|
||
|
*
|
||
|
* \xmlonly
|
||
|
Please refer to the <link linkend="boost_process.extend">tutorial</link> for more details.
|
||
|
\endxmlonly
|
||
|
*/
|
||
|
|
||
|
|
||
|
namespace boost {
|
||
|
namespace process {
|
||
|
namespace detail {
|
||
|
template<typename Tuple>
|
||
|
inline asio::io_context& get_io_context(const Tuple & tup);
|
||
|
}
|
||
|
|
||
|
|
||
|
///Namespace for extensions \attention This is experimental.
|
||
|
namespace extend {
|
||
|
|
||
|
#if defined(BOOST_WINDOWS_API)
|
||
|
|
||
|
template<typename Char, typename Sequence>
|
||
|
using windows_executor = ::boost::process::detail::windows::executor<Char, Sequence>;
|
||
|
template<typename Sequence>
|
||
|
struct posix_executor;
|
||
|
|
||
|
#elif defined(BOOST_POSIX_API)
|
||
|
|
||
|
template<typename Sequence>
|
||
|
using posix_executor = ::boost::process::detail::posix::executor<Sequence>;
|
||
|
template<typename Char, typename Sequence>
|
||
|
struct windows_executor;
|
||
|
|
||
|
#endif
|
||
|
|
||
|
using ::boost::process::detail::handler;
|
||
|
using ::boost::process::detail::api::require_io_context;
|
||
|
using ::boost::process::detail::api::async_handler;
|
||
|
using ::boost::process::detail::get_io_context;
|
||
|
using ::boost::process::detail::get_last_error;
|
||
|
using ::boost::process::detail::throw_last_error;
|
||
|
using ::boost::process::detail::uses_handles;
|
||
|
using ::boost::process::detail::foreach_used_handle;
|
||
|
using ::boost::process::detail::get_used_handles;
|
||
|
|
||
|
///This handler is invoked before the process in launched, to setup parameters. The required signature is `void(Exec &)`, where `Exec` is a template parameter.
|
||
|
constexpr boost::process::detail::make_handler_t<boost::process::detail::on_setup_> on_setup;
|
||
|
///This handler is invoked if an error occured. The required signature is `void(auto & exec, const std::error_code&)`, where `Exec` is a template parameter.
|
||
|
constexpr boost::process::detail::make_handler_t<boost::process::detail::on_error_> on_error;
|
||
|
///This handler is invoked if launching the process has succeeded. The required signature is `void(auto & exec)`, where `Exec` is a template parameter.
|
||
|
constexpr boost::process::detail::make_handler_t<boost::process::detail::on_success_> on_success;
|
||
|
|
||
|
#if defined(BOOST_POSIX_API) || defined(BOOST_PROCESS_DOXYGEN)
|
||
|
///This handler is invoked if the fork failed. The required signature is `void(auto & exec)`, where `Exec` is a template parameter. \note Only available on posix.
|
||
|
constexpr ::boost::process::detail::make_handler_t<::boost::process::detail::posix::on_fork_error_ > on_fork_error;
|
||
|
///This handler is invoked if the fork succeeded. The required signature is `void(Exec &)`, where `Exec` is a template parameter. \note Only available on posix.
|
||
|
constexpr ::boost::process::detail::make_handler_t<::boost::process::detail::posix::on_exec_setup_ > on_exec_setup;
|
||
|
///This handler is invoked if the exec call errored. The required signature is `void(auto & exec)`, where `Exec` is a template parameter. \note Only available on posix.
|
||
|
constexpr ::boost::process::detail::make_handler_t<::boost::process::detail::posix::on_exec_error_ > on_exec_error;
|
||
|
#endif
|
||
|
|
||
|
#if defined(BOOST_PROCESS_DOXYGEN)
|
||
|
///Helper function to get the last error code system-independent
|
||
|
inline std::error_code get_last_error();
|
||
|
|
||
|
///Helper function to get and throw the last system error.
|
||
|
/// \throws boost::process::process_error
|
||
|
/// \param msg A message to add to the error code.
|
||
|
inline void throw_last_error(const std::string & msg);
|
||
|
///\overload void throw_last_error(const std::string & msg)
|
||
|
inline void throw_last_error();
|
||
|
|
||
|
|
||
|
/** This function gets the io_context from the initializer sequence.
|
||
|
*
|
||
|
* \attention Yields a compile-time error if no `io_context` is provided.
|
||
|
* \param seq The Sequence of the initializer.
|
||
|
*/
|
||
|
template<typename Sequence>
|
||
|
inline asio::io_context& get_io_context(const Sequence & seq);
|
||
|
|
||
|
/** This class is the base for every initializer, to be used for extensions.
|
||
|
*
|
||
|
* The usage is done through compile-time polymorphism, so that the required
|
||
|
* functions can be overloaded.
|
||
|
*
|
||
|
* \note None of the function need to be `const`.
|
||
|
*
|
||
|
*/
|
||
|
struct handler
|
||
|
{
|
||
|
///This function is invoked before the process launch. \note It is not required to be const.
|
||
|
template <class Executor>
|
||
|
void on_setup(Executor&) const {}
|
||
|
|
||
|
/** This function is invoked if an error occured while trying to launch the process.
|
||
|
* \note It is not required to be const.
|
||
|
*/
|
||
|
template <class Executor>
|
||
|
void on_error(Executor&, const std::error_code &) const {}
|
||
|
|
||
|
/** This function is invoked if the process was successfully launched.
|
||
|
* \note It is not required to be const.
|
||
|
*/
|
||
|
template <class Executor>
|
||
|
void on_success(Executor&) const {}
|
||
|
|
||
|
/**This function is invoked if an error occured during the call of `fork`.
|
||
|
* \note This function will only be called on posix.
|
||
|
*/
|
||
|
template<typename Executor>
|
||
|
void on_fork_error (Executor &, const std::error_code&) const {}
|
||
|
|
||
|
/**This function is invoked if the call of `fork` was successful, before
|
||
|
* calling `execve`.
|
||
|
* \note This function will only be called on posix.
|
||
|
* \attention It will be invoked from the new process.
|
||
|
*/
|
||
|
template<typename Executor>
|
||
|
void on_exec_setup (Executor &) const {}
|
||
|
|
||
|
/**This function is invoked if the call of `execve` failed.
|
||
|
* \note This function will only be called on posix.
|
||
|
* \attention It will be invoked from the new process.
|
||
|
*/
|
||
|
template<typename Executor>
|
||
|
void on_exec_error (Executor &, const std::error_code&) const {}
|
||
|
|
||
|
};
|
||
|
|
||
|
|
||
|
/** Inheriting the class will tell the launching process that an `io_context` is
|
||
|
* needed. This should always be used when \ref get_io_context is used.
|
||
|
*
|
||
|
*/
|
||
|
struct require_io_context {};
|
||
|
/** Inheriting this class will tell the launching function, that an event handler
|
||
|
* shall be invoked when the process exits. This automatically does also inherit
|
||
|
* \ref require_io_context.
|
||
|
*
|
||
|
* You must add the following function to your implementation:
|
||
|
*
|
||
|
\code{.cpp}
|
||
|
template<typename Executor>
|
||
|
std::function<void(int, const std::error_code&)> on_exit_handler(Executor & exec)
|
||
|
{
|
||
|
auto handler_ = this->handler;
|
||
|
return [handler_](int exit_code, const std::error_code & ec)
|
||
|
{
|
||
|
handler_(static_cast<int>(exit_code), ec);
|
||
|
};
|
||
|
|
||
|
}
|
||
|
\endcode
|
||
|
|
||
|
The callback will be obtained by calling this function on setup and it will be
|
||
|
invoked when the process exits.
|
||
|
|
||
|
*
|
||
|
* \warning Cannot be used with \ref boost::process::spawn
|
||
|
*/
|
||
|
struct async_handler : handler, require_io_context
|
||
|
{
|
||
|
|
||
|
};
|
||
|
|
||
|
///The posix executor type.
|
||
|
/** This type represents the posix executor and can be used for overloading in a custom handler.
|
||
|
* \note It is an alias for the implementation on posix, and a forward-declaration on windows.
|
||
|
*
|
||
|
* \tparam Sequence The used initializer-sequence, it is fulfills the boost.fusion [sequence](http://www.boost.org/doc/libs/master/libs/fusion/doc/html/fusion/sequence.html) concept.
|
||
|
|
||
|
|
||
|
\xmlonly
|
||
|
As information for extension development, here is the structure of the process launching (in pseudo-code and uml)
|
||
|
<xi:include href="posix_pseudocode.xml" xmlns:xi="http://www.w3.org/2001/XInclude"/>
|
||
|
|
||
|
<mediaobject>
|
||
|
<caption>
|
||
|
<para>The sequence if when no error occurs.</para>
|
||
|
</caption>
|
||
|
<imageobject>
|
||
|
<imagedata fileref="boost_process/posix_success.svg"/>
|
||
|
</imageobject>
|
||
|
</mediaobject>
|
||
|
|
||
|
<mediaobject>
|
||
|
<caption>
|
||
|
<para>The sequence if the execution fails.</para>
|
||
|
</caption>
|
||
|
<imageobject>
|
||
|
<imagedata fileref="boost_process/posix_exec_err.svg"/>
|
||
|
</imageobject>
|
||
|
</mediaobject>
|
||
|
|
||
|
<mediaobject>
|
||
|
<caption>
|
||
|
<para>The sequence if the fork fails.</para>
|
||
|
</caption>
|
||
|
<imageobject>
|
||
|
<imagedata fileref="boost_process/posix_fork_err.svg"/>
|
||
|
</imageobject>
|
||
|
</mediaobject>
|
||
|
|
||
|
\endxmlonly
|
||
|
|
||
|
|
||
|
\note Error handling if execve fails is done through a pipe, unless \ref ignore_error is used.
|
||
|
|
||
|
*/
|
||
|
template<typename Sequence>
|
||
|
struct posix_executor
|
||
|
{
|
||
|
///A reference to the actual initializer-sequence
|
||
|
Sequence & seq;
|
||
|
///A pointer to the name of the executable.
|
||
|
const char * exe = nullptr;
|
||
|
///A pointer to the argument-vector.
|
||
|
char *const* cmd_line = nullptr;
|
||
|
///A pointer to the environment variables, as default it is set to [environ](http://pubs.opengroup.org/onlinepubs/009695399/basedefs/xbd_chap08.html)
|
||
|
char **env = ::environ;
|
||
|
///The pid of the process - it will be -1 before invoking [fork](http://pubs.opengroup.org/onlinepubs/009695399/functions/fork.html), and after forking either 0 for the new process or a positive value if in the current process. */
|
||
|
pid_t pid = -1;
|
||
|
///This shared-pointer holds the exit code. It's done this way, so it can be shared between an `asio::io_context` and \ref child.
|
||
|
std::shared_ptr<std::atomic<int>> exit_status = std::make_shared<std::atomic<int>>(still_active);
|
||
|
|
||
|
///This function returns a const reference to the error state of the executor.
|
||
|
const std::error_code & error() const;
|
||
|
|
||
|
///This function can be used to report an error to the executor. This will be handled according to the configuration of the executor, i.e. it
|
||
|
/// might throw an exception. \note This is the required way to handle errors in initializers.
|
||
|
void set_error(const std::error_code &ec, const std::string &msg);
|
||
|
///\overload void set_error(const std::error_code &ec, const std::string &msg);
|
||
|
void set_error(const std::error_code &ec, const char* msg);
|
||
|
};
|
||
|
|
||
|
///The windows executor type.
|
||
|
/** This type represents the posix executor and can be used for overloading in a custom handler.
|
||
|
*
|
||
|
* \note It is an alias for the implementation on posix, and a forward-declaration on windows.
|
||
|
* \tparam Sequence The used initializer-sequence, it is fulfills the boost.fusion [sequence](http://www.boost.org/doc/libs/master/libs/fusion/doc/html/fusion/sequence.html) concept.
|
||
|
* \tparam Char The used char-type, either `char` or `wchar_t`.
|
||
|
*
|
||
|
|
||
|
\xmlonly
|
||
|
As information for extension development, here is the structure of the process launching (in pseudo-code and uml)<xi:include href="windows_pseudocode.xml" xmlns:xi="http://www.w3.org/2001/XInclude"/>
|
||
|
<mediaobject>
|
||
|
<caption>
|
||
|
<para>The sequence for windows process creation.</para>
|
||
|
</caption>
|
||
|
<imageobject>
|
||
|
<imagedata fileref="boost_process/windows_exec.svg"/>
|
||
|
</imageobject>
|
||
|
</mediaobject>
|
||
|
\endxmlonly
|
||
|
|
||
|
*/
|
||
|
|
||
|
template<typename Char, typename Sequence>
|
||
|
struct windows_executor
|
||
|
{
|
||
|
///A reference to the actual initializer-sequence
|
||
|
Sequence & seq;
|
||
|
|
||
|
///A pointer to the name of the executable. It's null by default.
|
||
|
const Char * exe = nullptr;
|
||
|
///A pointer to the argument-vector. Must be set by some initializer.
|
||
|
char Char* cmd_line = nullptr;
|
||
|
///A pointer to the environment variables. It's null by default.
|
||
|
char Char* env = nullptr;
|
||
|
///A pointer to the working directory. It's null by default.
|
||
|
const Char * work_dir = nullptr;
|
||
|
|
||
|
///A pointer to the process-attributes of type [SECURITY_ATTRIBUTES](https://msdn.microsoft.com/en-us/library/windows/desktop/aa379560.aspx). It's null by default.
|
||
|
::boost::detail::winapi::LPSECURITY_ATTRIBUTES_ proc_attrs = nullptr;
|
||
|
///A pointer to the thread-attributes of type [SECURITY_ATTRIBUTES](https://msdn.microsoft.com/en-us/library/windows/desktop/aa379560.aspx). It' null by default.
|
||
|
::boost::detail::winapi::LPSECURITY_ATTRIBUTES_ thread_attrs = nullptr;
|
||
|
///A logical bool value setting whether handles shall be inherited or not.
|
||
|
::boost::detail::winapi::BOOL_ inherit_handles = false;
|
||
|
|
||
|
///The element holding the process-information after process creation. The type is [PROCESS_INFORMATION](https://msdn.microsoft.com/en-us/library/windows/desktop/ms684873.aspx)
|
||
|
::boost::detail::winapi::PROCESS_INFORMATION_ proc_info{nullptr, nullptr, 0,0};
|
||
|
|
||
|
|
||
|
///This shared-pointer holds the exit code. It's done this way, so it can be shared between an `asio::io_context` and \ref child.
|
||
|
std::shared_ptr<std::atomic<int>> exit_status = std::make_shared<std::atomic<int>>(still_active);
|
||
|
|
||
|
///This function returns a const reference to the error state of the executor.
|
||
|
const std::error_code & error() const;
|
||
|
|
||
|
///This function can be used to report an error to the executor. This will be handled according to the configuration of the executor, i.e. it
|
||
|
/// might throw an exception. \note This is the required way to handle errors in initializers.
|
||
|
void set_error(const std::error_code &ec, const std::string &msg);
|
||
|
///\overload void set_error(const std::error_code &ec, const std::string &msg);
|
||
|
void set_error(const std::error_code &ec, const char* msg);
|
||
|
|
||
|
///The creation flags of the process
|
||
|
::boost::detail::winapi::DWORD_ creation_flags;
|
||
|
///The type of the [startup-info](https://msdn.microsoft.com/en-us/library/windows/desktop/ms686331.aspx), depending on the char-type.
|
||
|
typedef typename detail::startup_info<Char>::type startup_info_t;
|
||
|
///The type of the [extended startup-info](https://msdn.microsoft.com/de-de/library/windows/desktop/ms686329.aspx), depending the char-type; only defined with winapi-version equal or higher than 6.
|
||
|
typedef typename detail::startup_info_ex<Char>::type startup_info_ex_t;
|
||
|
///This function switches the information, so that the extended structure is used. \note It's only defined with winapi-version equal or higher than 6.
|
||
|
void set_startup_info_ex();
|
||
|
///This element is an instance or a reference (if \ref startup_info_ex exists) to the [startup-info](https://msdn.microsoft.com/en-us/library/windows/desktop/ms686331.aspx) for the process.
|
||
|
startup_info_t startup_info;
|
||
|
///This element is the instance of the [extended startup-info](https://msdn.microsoft.com/de-de/library/windows/desktop/ms686329.aspx). It is only available with a winapi-version equal or highter than 6.
|
||
|
startup_info_ex_t startup_info_ex;
|
||
|
};
|
||
|
|
||
|
|
||
|
|
||
|
#endif
|
||
|
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#endif
|