整理

include/boost/process/v2/detail/process_handle_windows.hpp

@@ -0,0 +1,326 @@
+// Copyright (c) 2022 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_V2_DETAIL_PROCESS_HANDLE_WINDOWS_HPP
+#define BOOST_PROCESS_V2_DETAIL_PROCESS_HANDLE_WINDOWS_HPP
+
+#include <boost/process/v2/detail/config.hpp>
+
+#include <boost/process/v2/exit_code.hpp>
+#include <boost/process/v2/pid.hpp>
+#include <boost/process/v2/detail/throw_error.hpp>
+
+#if defined(BOOST_PROCESS_V2_STANDALONE)
+#include <asio/any_io_executor.hpp>
+#include <asio/compose.hpp>
+#include <asio/windows/basic_object_handle.hpp>
+#else
+#include <boost/asio/any_io_executor.hpp>
+#include <boost/asio/compose.hpp>
+#include <boost/asio/windows/basic_object_handle.hpp>
+#endif
+
+BOOST_PROCESS_V2_BEGIN_NAMESPACE
+
+namespace detail
+{
+  
+BOOST_PROCESS_V2_DECL void get_exit_code_( void * handle, native_exit_code_type & exit_code, error_code & ec);
+BOOST_PROCESS_V2_DECL void * open_process_(pid_type pid);
+BOOST_PROCESS_V2_DECL void terminate_if_running_(void * handle);
+BOOST_PROCESS_V2_DECL bool check_handle_(void* handle, error_code & ec);
+BOOST_PROCESS_V2_DECL bool check_pid_(pid_type pid_, error_code & ec);
+BOOST_PROCESS_V2_DECL void interrupt_(pid_type pid_, error_code & ec);
+BOOST_PROCESS_V2_DECL void suspend_(void * handle, error_code & ec);
+BOOST_PROCESS_V2_DECL void resume_(void * handle, error_code & ec);
+BOOST_PROCESS_V2_DECL void terminate_(void * handle, error_code & ec, native_exit_code_type & exit_code);
+BOOST_PROCESS_V2_DECL void request_exit_(pid_type pid_, error_code & ec);
+BOOST_PROCESS_V2_DECL void check_running_(void* handle, error_code & ec, native_exit_code_type & exit_status);
+
+template<typename Executor = net::any_io_executor>
+struct basic_process_handle_win
+{
+    typedef net::windows::basic_object_handle<Executor> handle_type;
+    typedef typename handle_type::native_handle_type native_handle_type;
+
+    typedef Executor executor_type;
+
+    executor_type get_executor()
+    { return handle_.get_executor(); }
+
+    /// Rebinds the process_handle to another executor.
+    template<typename Executor1>
+    struct rebind_executor
+    {
+        /// The socket type when rebound to the specified executor.
+        typedef basic_process_handle_win<Executor1> other;
+    };
+
+    template<typename ExecutionContext>
+    basic_process_handle_win(ExecutionContext &context,
+                         typename std::enable_if<
+                                 std::is_convertible<ExecutionContext &,
+                                         net::execution_context &>::value
+                         >::type = 0)
+            : pid_(0), handle_(context)
+    {
+    }
+
+    basic_process_handle_win(Executor executor)
+            : pid_(0), handle_(executor)
+    {
+    }
+
+    basic_process_handle_win(Executor executor, pid_type pid)
+            : pid_(pid), handle_(executor, detail::open_process_(pid))
+    {
+    }
+
+    basic_process_handle_win(Executor executor, pid_type pid, native_handle_type process_handle)
+            : pid_(pid), handle_(executor, process_handle)
+    {
+    }
+
+
+    template<typename Executor1>
+    basic_process_handle_win(basic_process_handle_win<Executor1> && other)
+            : pid_(other.pid_), handle_(std::move(other.handle_))
+    {
+      other.pid_ = static_cast<DWORD>(-1);
+    }
+
+    basic_process_handle_win(basic_process_handle_win && other)
+        :  pid_(other.pid_), handle_(std::move(other.handle_))
+    {
+      other.pid_ = static_cast<DWORD>(-1);
+    }
+
+    basic_process_handle_win& operator=(basic_process_handle_win && other)
+    {
+        pid_ = other.pid_;
+        handle_ = std::move(other.handle_);
+        other.pid_ = static_cast<DWORD>(-1);
+        return *this;
+    }
+
+    template<typename Executor1>
+    basic_process_handle_win& operator=(basic_process_handle_win<Executor1> && other)
+    {
+        pid_ = other.pid_;
+        handle_ = std::move(other.handle_);
+        other.pid_ = static_cast<DWORD>(-1);
+        return *this;
+    }
+
+    ~basic_process_handle_win()
+    {
+        if (handle_.is_open())
+        {
+            error_code ec;
+            handle_.close(ec);
+        }
+    }
+
+    native_handle_type native_handle()
+    { return handle_.native_handle(); }
+
+    pid_type id() const
+    { return pid_; }
+
+    void terminate_if_running(error_code &)
+    {
+        detail::terminate_if_running_(handle_.native_handle());
+    }
+
+    void terminate_if_running()
+    {
+        detail::terminate_if_running_(handle_.native_handle());
+    }
+
+    void wait(native_exit_code_type &exit_status, error_code &ec)
+    {
+        if (!detail::check_handle_(handle_.native_handle(), ec))
+            return;
+
+        handle_.wait(ec);
+        if (!ec)
+            detail::get_exit_code_(handle_.native_handle(), exit_status, ec);
+    }
+
+
+    void wait(native_exit_code_type &exit_status)
+    {
+        error_code ec;
+        wait(exit_status, ec);
+        if (ec)
+            detail::throw_error(ec, "wait(pid)");
+    }
+
+    void interrupt(error_code &ec)
+    {
+        if (!detail::check_pid_(pid_, ec))
+            return;
+
+        detail::interrupt_(pid_, ec);
+    }
+
+    void interrupt()
+    {
+        error_code ec;
+        interrupt(ec);
+        if (ec)
+            detail::throw_error(ec, "interrupt");
+    }
+
+    void request_exit(error_code &ec)
+    {
+        if (!detail::check_pid_(pid_, ec))
+            return;
+        detail::request_exit_(pid_, ec);
+    }
+
+    void request_exit()
+    {
+        error_code ec;
+        request_exit(ec);
+        if (ec)
+            detail::throw_error(ec, "request_exit");
+    }
+
+    void suspend(error_code &ec)
+    {
+        detail::suspend_(handle_.native_handle(), ec);
+    }
+
+    void suspend()
+    {
+        error_code ec;
+        suspend(ec);
+        if (ec)
+            detail::throw_error(ec, "suspend");
+    }
+
+    void resume(error_code &ec)
+    {
+        detail::resume_(handle_.native_handle(), ec);
+    }
+
+    void resume()
+    {
+        error_code ec;
+        resume(ec);
+        if (ec)
+            detail::throw_error(ec, "resume");
+    }
+
+    void terminate(native_exit_code_type &exit_status, error_code &ec)
+    {
+        if (!detail::check_handle_(handle_.native_handle(), ec))
+            return;
+
+        detail::terminate_(handle_.native_handle(), ec, exit_status);
+        if (!ec)
+            wait(exit_status, ec);
+
+    }
+
+    void terminate(native_exit_code_type &exit_status)
+    {
+        error_code ec;
+        terminate(exit_status, ec);
+        if (ec)
+            detail::throw_error(ec, "terminate");
+    }
+
+    bool running(native_exit_code_type &exit_code, error_code & ec)
+    {
+        if (!detail::check_handle_(handle_.native_handle(), ec))
+            return false;
+
+        native_exit_code_type code;
+        //single value, not needed in the winapi.
+        detail::check_running_(handle_.native_handle(), ec, code);
+        if (ec)
+            return false;
+
+        if (process_is_running(code))
+            return true;
+        else
+            exit_code = code;
+
+        return false;
+    }
+
+    bool running(native_exit_code_type &exit_code)
+    {
+        error_code ec;
+        bool res = running(exit_code, ec);
+        if (ec)
+            detail::throw_error(ec, "is_running");
+        return res;
+    }
+
+    bool is_open() const
+    {
+        return handle_.is_open();
+    }
+
+    template<typename>
+    friend struct basic_process_handle_win;
+  private:
+    pid_type pid_;
+    handle_type handle_;
+
+    struct async_wait_op_
+    {
+        handle_type &handle;
+        native_exit_code_type & exit_code;
+        template<typename Self>
+        void operator()(Self &&self)
+        {
+
+            self.reset_cancellation_state(asio::enable_total_cancellation());
+            auto sl = self.get_cancellation_state().slot();
+            auto & h = handle;
+            if (sl.is_connected())
+                sl.assign(
+                    [&h](asio::cancellation_type ct)
+                    {
+                      error_code ec;
+                      h.cancel(ec);
+                    });
+            handle.async_wait(std::move(self));
+        }
+
+        template<typename Self>
+        void operator()(Self &&self, error_code ec)
+        {
+            if (ec == asio::error::operation_aborted && !self.get_cancellation_state().cancelled())
+              return handle.async_wait(std::move(self));
+
+            if (!ec && process_is_running(exit_code)) // exit_code could be set by another call to wait.
+                detail::get_exit_code_(handle.native_handle(), exit_code, ec);
+            std::move(self).complete(ec);
+        }
+    };
+ public:
+    template<BOOST_PROCESS_V2_COMPLETION_TOKEN_FOR(void(error_code))
+             WaitHandler = net::default_completion_token_t<executor_type>>
+    auto async_wait(native_exit_code_type & exit_code,
+                    WaitHandler &&handler = net::default_completion_token_t<executor_type>())
+        -> decltype(net::async_compose<WaitHandler, void(error_code)>(
+                    async_wait_op_{handle_, exit_code}, handler, handle_))
+    {
+        return net::async_compose<WaitHandler, void(error_code)>(
+                async_wait_op_{handle_, exit_code}, handler, handle_
+        );
+    }
+};
+
+}
+
+BOOST_PROCESS_V2_END_NAMESPACE
+
+
+#endif //BOOST_PROCESS_V2_DETAIL_PROCESS_HANDLE_WINDOWS_HPP