wait_for_exit.hpp 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  1. // Copyright (c) 2006, 2007 Julio M. Merino Vidal
  2. // Copyright (c) 2008 Ilya Sokolov, Boris Schaeling
  3. // Copyright (c) 2009 Boris Schaeling
  4. // Copyright (c) 2010 Felipe Tanus, Boris Schaeling
  5. // Copyright (c) 2011, 2012 Jeff Flinn, Boris Schaeling
  6. // Copyright (c) 2016 Klemens D. Morgenstern
  7. //
  8. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  9. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  10. #ifndef BOOST_PROCESS_WINDOWS_WAIT_FOR_EXIT_HPP
  11. #define BOOST_PROCESS_WINDOWS_WAIT_FOR_EXIT_HPP
  12. #include <boost/process/detail/config.hpp>
  13. #include <system_error>
  14. #include <boost/winapi/synchronization.hpp>
  15. #include <boost/winapi/process.hpp>
  16. #include <boost/process/detail/windows/child_handle.hpp>
  17. #include <chrono>
  18. namespace boost { namespace process { namespace detail { namespace windows {
  19. inline void wait(child_handle &p, int & exit_code, std::error_code &ec) noexcept
  20. {
  21. ::boost::winapi::DWORD_ _exit_code = 1;
  22. if (::boost::winapi::WaitForSingleObject(p.process_handle(),
  23. ::boost::winapi::infinite) == ::boost::winapi::wait_failed)
  24. ec = std::error_code(
  25. ::boost::winapi::GetLastError(),
  26. std::system_category());
  27. else if (!::boost::winapi::GetExitCodeProcess(p.process_handle(), &_exit_code))
  28. ec = std::error_code(
  29. ::boost::winapi::GetLastError(),
  30. std::system_category());
  31. else
  32. ec.clear();
  33. ::boost::winapi::CloseHandle(p.proc_info.hProcess);
  34. p.proc_info.hProcess = ::boost::winapi::INVALID_HANDLE_VALUE_;
  35. exit_code = static_cast<int>(_exit_code);
  36. }
  37. inline void wait(child_handle &p, int & exit_code)
  38. {
  39. std::error_code ec;
  40. wait(p, exit_code, ec);
  41. boost::process::detail::throw_error(ec, "wait error");
  42. }
  43. template< class Clock, class Duration >
  44. inline bool wait_until(
  45. child_handle &p,
  46. int & exit_code,
  47. const std::chrono::time_point<Clock, Duration>& timeout_time,
  48. std::error_code &ec) noexcept
  49. {
  50. std::chrono::milliseconds ms =
  51. std::chrono::duration_cast<std::chrono::milliseconds>(
  52. timeout_time - Clock::now());
  53. ::boost::winapi::DWORD_ wait_code;
  54. wait_code = ::boost::winapi::WaitForSingleObject(p.process_handle(),
  55. static_cast<::boost::winapi::DWORD_>(ms.count()));
  56. if (wait_code == ::boost::winapi::wait_failed)
  57. ec = std::error_code(
  58. ::boost::winapi::GetLastError(),
  59. std::system_category());
  60. else if (wait_code == ::boost::winapi::wait_timeout)
  61. return false;
  62. ::boost::winapi::DWORD_ _exit_code;
  63. if (!::boost::winapi::GetExitCodeProcess(p.process_handle(), &_exit_code))
  64. ec = std::error_code(
  65. ::boost::winapi::GetLastError(),
  66. std::system_category());
  67. else
  68. ec.clear();
  69. exit_code = static_cast<int>(_exit_code);
  70. ::boost::winapi::CloseHandle(p.proc_info.hProcess);
  71. p.proc_info.hProcess = ::boost::winapi::INVALID_HANDLE_VALUE_;
  72. return true;
  73. }
  74. template< class Clock, class Duration >
  75. inline bool wait_until(
  76. child_handle &p,
  77. int & exit_code,
  78. const std::chrono::time_point<Clock, Duration>& timeout_time)
  79. {
  80. std::error_code ec;
  81. bool b = wait_until(p, exit_code, timeout_time, ec);
  82. boost::process::detail::throw_error(ec, "wait_until error");
  83. return b;
  84. }
  85. template< class Rep, class Period >
  86. inline bool wait_for(
  87. child_handle &p,
  88. int & exit_code,
  89. const std::chrono::duration<Rep, Period>& rel_time,
  90. std::error_code &ec) noexcept
  91. {
  92. return wait_until(p, exit_code, std::chrono::steady_clock::now() + rel_time, ec);
  93. }
  94. template< class Rep, class Period >
  95. inline bool wait_for(
  96. child_handle &p,
  97. int & exit_code,
  98. const std::chrono::duration<Rep, Period>& rel_time)
  99. {
  100. std::error_code ec;
  101. bool b = wait_for(p, exit_code, rel_time, ec);
  102. boost::process::detail::throw_error(ec, "wait_for error");
  103. return b;
  104. }
  105. }}}}
  106. #endif