v2/windows/as_user_launcher.hpp

0001 //
0002 // boost/process/v2/windows/default_launcher.hpp
0003 // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
0004 //
0005 // Copyright (c) 2022 Klemens D. Morgenstern (klemens dot morgenstern at gmx dot net)
0006 //
0007 // Distributed under the Boost Software License, Version 1.0. (See accompanying
0008 // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
0009 //
0010
0011 #ifndef BOOST_PROCESS_V2_WINDOWS_AS_USER_LAUNCHER_HPP
0012 #define BOOST_PROCESS_V2_WINDOWS_AS_USER_LAUNCHER_HPP
0013
0014 #include <boost/process/v2/detail/config.hpp>
0015 #include <boost/process/v2/windows/default_launcher.hpp>
0016
0017 BOOST_PROCESS_V2_BEGIN_NAMESPACE
0018 namespace windows
0019 {
0020
0021 /// A windows launcher using CreateProcessAsUser instead of CreateProcess
0022 struct as_user_launcher : default_launcher
0023 {
0024   /// The token to be used in CreateProcessAsUser.
0025   HANDLE token;
0026   as_user_launcher(HANDLE token = INVALID_HANDLE_VALUE) : token(token) {}
0027
0028   template<typename ExecutionContext, typename Args, typename ... Inits>
0029   auto operator()(ExecutionContext & context,
0030                   const typename std::enable_if<std::is_convertible<
0031                              ExecutionContext&, net::execution_context&>::value,
0032                              filesystem::path >::type & executable,
0033                   Args && args,
0034                   Inits && ... inits ) -> basic_process<typename ExecutionContext::executor_type>
0035   {
0036       error_code ec;
0037       auto proc =  (*this)(context, ec, executable, std::forward<Args>(args), std::forward<Inits>(inits)...);
0038
0039       if (ec)
0040           v2::detail::throw_error(ec, "as_user_launcher");
0041
0042       return proc;
0043   }
0044
0045
0046   template<typename ExecutionContext, typename Args, typename ... Inits>
0047   auto operator()(ExecutionContext & context,
0048                      error_code & ec,
0049                      const typename std::enable_if<std::is_convertible<
0050                              ExecutionContext&, net::execution_context&>::value,
0051                              filesystem::path >::type & executable,
0052                      Args && args,
0053                      Inits && ... inits ) -> basic_process<typename ExecutionContext::executor_type>
0054   {
0055       return (*this)(context.get_executor(), ec, executable, std::forward<Args>(args), std::forward<Inits>(inits)...);
0056   }
0057
0058   template<typename Executor, typename Args, typename ... Inits>
0059   auto operator()(Executor exec,
0060                      const typename std::enable_if<
0061                              net::execution::is_executor<Executor>::value ||
0062                              net::is_executor<Executor>::value,
0063                              filesystem::path >::type & executable,
0064                      Args && args,
0065                      Inits && ... inits ) -> basic_process<Executor>
0066   {
0067       error_code ec;
0068       auto proc = (*this)(std::move(exec), ec, executable, std::forward<Args>(args), std::forward<Inits>(inits)...);
0069
0070       if (ec)
0071           detail::throw_error(ec, "as_user_launcher");
0072
0073       return proc;
0074   }
0075
0076   template<typename Executor, typename Args, typename ... Inits>
0077   auto operator()(Executor exec,
0078                   error_code & ec,
0079                   const typename std::enable_if<
0080                       net::execution::is_executor<Executor>::value ||
0081                       net::is_executor<Executor>::value,
0082                         filesystem::path >::type & executable,
0083                   Args && args,
0084                   Inits && ... inits ) -> basic_process<Executor>
0085   {
0086     auto command_line = this->build_command_line(executable, args);
0087
0088     ec = detail::on_setup(*this, executable, command_line, inits...);
0089     if (ec)
0090     {
0091       detail::on_error(*this, executable, command_line, ec, inits...);
0092       return basic_process<Executor>(exec);
0093     }
0094
0095     if (!inherited_handles.empty())
0096     {
0097       set_handle_list(ec);
0098       if (ec)
0099         return basic_process<Executor>(exec);
0100     }
0101
0102     auto ok = ::CreateProcessAsUserW(
0103         token,
0104         executable.empty() ? nullptr : executable.c_str(),
0105         command_line.empty() ? nullptr : &command_line.front(),
0106         process_attributes,
0107         thread_attributes,
0108         inherited_handles.empty() ? FALSE : TRUE,
0109         creation_flags,
0110         environment,
0111         current_directory.empty() ? nullptr : current_directory.c_str(),
0112         &startup_info.StartupInfo,
0113         &process_information);
0114
0115
0116     if (ok == 0)
0117     {
0118       BOOST_PROCESS_V2_ASSIGN_LAST_ERROR(ec);
0119       detail::on_error(*this, executable, command_line, ec, inits...);
0120
0121       if (process_information.hProcess != INVALID_HANDLE_VALUE)
0122         ::CloseHandle(process_information.hProcess);
0123       if (process_information.hThread != INVALID_HANDLE_VALUE)
0124         ::CloseHandle(process_information.hThread);
0125
0126       return basic_process<Executor>(exec);
0127     } else
0128     {
0129       detail::on_success(*this, executable, command_line, inits...);
0130
0131       if (process_information.hThread != INVALID_HANDLE_VALUE)
0132         ::CloseHandle(process_information.hThread);
0133
0134       return basic_process<Executor>(exec,
0135                                      this->process_information.dwProcessId,
0136                                      this->process_information.hProcess);
0137     }
0138   }
0139 };
0140
0141 }
0142 BOOST_PROCESS_V2_END_NAMESPACE
0143
0144 #endif //  BOOST_PROCESS_V2_WINDOWS_AS_USER_LAUNCHER_HPP