signal_set_service.ipp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826
  1. //
  2. // detail/impl/signal_set_service.ipp
  3. // ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
  4. //
  5. // Copyright (c) 2003-2023 Christopher M. Kohlhoff (chris at kohlhoff dot com)
  6. //
  7. // Distributed under the Boost Software License, Version 1.0. (See accompanying
  8. // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
  9. //
  10. #ifndef ASIO_DETAIL_IMPL_SIGNAL_SET_SERVICE_IPP
  11. #define ASIO_DETAIL_IMPL_SIGNAL_SET_SERVICE_IPP
  12. #if defined(_MSC_VER) && (_MSC_VER >= 1200)
  13. # pragma once
  14. #endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
  15. #include "asio/detail/config.hpp"
  16. #include <cstring>
  17. #include <stdexcept>
  18. #include "asio/detail/signal_blocker.hpp"
  19. #include "asio/detail/signal_set_service.hpp"
  20. #include "asio/detail/static_mutex.hpp"
  21. #include "asio/detail/throw_exception.hpp"
  22. #if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  23. # include "asio/detail/io_uring_service.hpp"
  24. #else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  25. # include "asio/detail/reactor.hpp"
  26. #endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  27. #include "asio/detail/push_options.hpp"
  28. namespace asio {
  29. namespace detail {
  30. struct signal_state
  31. {
  32. // Mutex used for protecting global state.
  33. static_mutex mutex_;
  34. // The read end of the pipe used for signal notifications.
  35. int read_descriptor_;
  36. // The write end of the pipe used for signal notifications.
  37. int write_descriptor_;
  38. // Whether the signal state has been prepared for a fork.
  39. bool fork_prepared_;
  40. // The head of a linked list of all signal_set_service instances.
  41. class signal_set_service* service_list_;
  42. // A count of the number of objects that are registered for each signal.
  43. std::size_t registration_count_[max_signal_number];
  44. // The flags used for each registered signal.
  45. signal_set_base::flags_t flags_[max_signal_number];
  46. };
  47. signal_state* get_signal_state()
  48. {
  49. static signal_state state = {
  50. ASIO_STATIC_MUTEX_INIT, -1, -1, false, 0,
  51. { 0 }, { signal_set_base::flags_t() } };
  52. return &state;
  53. }
  54. void asio_signal_handler(int signal_number)
  55. {
  56. #if defined(ASIO_WINDOWS) \
  57. || defined(ASIO_WINDOWS_RUNTIME) \
  58. || defined(__CYGWIN__)
  59. signal_set_service::deliver_signal(signal_number);
  60. #else // defined(ASIO_WINDOWS)
  61. // || defined(ASIO_WINDOWS_RUNTIME)
  62. // || defined(__CYGWIN__)
  63. int saved_errno = errno;
  64. signal_state* state = get_signal_state();
  65. signed_size_type result = ::write(state->write_descriptor_,
  66. &signal_number, sizeof(signal_number));
  67. (void)result;
  68. errno = saved_errno;
  69. #endif // defined(ASIO_WINDOWS)
  70. // || defined(ASIO_WINDOWS_RUNTIME)
  71. // || defined(__CYGWIN__)
  72. #if defined(ASIO_HAS_SIGNAL) && !defined(ASIO_HAS_SIGACTION)
  73. ::signal(signal_number, asio_signal_handler);
  74. #endif // defined(ASIO_HAS_SIGNAL) && !defined(ASIO_HAS_SIGACTION)
  75. }
  76. #if !defined(ASIO_WINDOWS) \
  77. && !defined(ASIO_WINDOWS_RUNTIME) \
  78. && !defined(__CYGWIN__)
  79. class signal_set_service::pipe_read_op :
  80. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  81. public io_uring_operation
  82. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  83. public reactor_op
  84. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  85. {
  86. public:
  87. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  88. pipe_read_op()
  89. : io_uring_operation(asio::error_code(), &pipe_read_op::do_prepare,
  90. &pipe_read_op::do_perform, pipe_read_op::do_complete)
  91. {
  92. }
  93. static void do_prepare(io_uring_operation*, ::io_uring_sqe* sqe)
  94. {
  95. signal_state* state = get_signal_state();
  96. int fd = state->read_descriptor_;
  97. ::io_uring_prep_poll_add(sqe, fd, POLLIN);
  98. }
  99. static bool do_perform(io_uring_operation*, bool)
  100. {
  101. signal_state* state = get_signal_state();
  102. int fd = state->read_descriptor_;
  103. int signal_number = 0;
  104. while (::read(fd, &signal_number, sizeof(int)) == sizeof(int))
  105. if (signal_number >= 0 && signal_number < max_signal_number)
  106. signal_set_service::deliver_signal(signal_number);
  107. return false;
  108. }
  109. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  110. pipe_read_op()
  111. : reactor_op(asio::error_code(),
  112. &pipe_read_op::do_perform, pipe_read_op::do_complete)
  113. {
  114. }
  115. static status do_perform(reactor_op*)
  116. {
  117. signal_state* state = get_signal_state();
  118. int fd = state->read_descriptor_;
  119. int signal_number = 0;
  120. while (::read(fd, &signal_number, sizeof(int)) == sizeof(int))
  121. if (signal_number >= 0 && signal_number < max_signal_number)
  122. signal_set_service::deliver_signal(signal_number);
  123. return not_done;
  124. }
  125. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  126. static void do_complete(void* /*owner*/, operation* base,
  127. const asio::error_code& /*ec*/,
  128. std::size_t /*bytes_transferred*/)
  129. {
  130. pipe_read_op* o(static_cast<pipe_read_op*>(base));
  131. delete o;
  132. }
  133. };
  134. #endif // !defined(ASIO_WINDOWS)
  135. // && !defined(ASIO_WINDOWS_RUNTIME)
  136. // && !defined(__CYGWIN__)
  137. signal_set_service::signal_set_service(execution_context& context)
  138. : execution_context_service_base<signal_set_service>(context),
  139. scheduler_(asio::use_service<scheduler_impl>(context)),
  140. #if !defined(ASIO_WINDOWS) \
  141. && !defined(ASIO_WINDOWS_RUNTIME) \
  142. && !defined(__CYGWIN__)
  143. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  144. io_uring_service_(asio::use_service<io_uring_service>(context)),
  145. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  146. reactor_(asio::use_service<reactor>(context)),
  147. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  148. #endif // !defined(ASIO_WINDOWS)
  149. // && !defined(ASIO_WINDOWS_RUNTIME)
  150. // && !defined(__CYGWIN__)
  151. next_(0),
  152. prev_(0)
  153. {
  154. get_signal_state()->mutex_.init();
  155. #if !defined(ASIO_WINDOWS) \
  156. && !defined(ASIO_WINDOWS_RUNTIME) \
  157. && !defined(__CYGWIN__)
  158. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  159. io_uring_service_.init_task();
  160. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  161. reactor_.init_task();
  162. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  163. #endif // !defined(ASIO_WINDOWS)
  164. // && !defined(ASIO_WINDOWS_RUNTIME)
  165. // && !defined(__CYGWIN__)
  166. for (int i = 0; i < max_signal_number; ++i)
  167. registrations_[i] = 0;
  168. add_service(this);
  169. }
  170. signal_set_service::~signal_set_service()
  171. {
  172. remove_service(this);
  173. }
  174. void signal_set_service::shutdown()
  175. {
  176. remove_service(this);
  177. op_queue<operation> ops;
  178. for (int i = 0; i < max_signal_number; ++i)
  179. {
  180. registration* reg = registrations_[i];
  181. while (reg)
  182. {
  183. ops.push(*reg->queue_);
  184. reg = reg->next_in_table_;
  185. }
  186. }
  187. scheduler_.abandon_operations(ops);
  188. }
  189. void signal_set_service::notify_fork(execution_context::fork_event fork_ev)
  190. {
  191. #if !defined(ASIO_WINDOWS) \
  192. && !defined(ASIO_WINDOWS_RUNTIME) \
  193. && !defined(__CYGWIN__)
  194. signal_state* state = get_signal_state();
  195. static_mutex::scoped_lock lock(state->mutex_);
  196. switch (fork_ev)
  197. {
  198. case execution_context::fork_prepare:
  199. {
  200. int read_descriptor = state->read_descriptor_;
  201. state->fork_prepared_ = true;
  202. lock.unlock();
  203. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  204. (void)read_descriptor;
  205. io_uring_service_.deregister_io_object(io_object_data_);
  206. io_uring_service_.cleanup_io_object(io_object_data_);
  207. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  208. reactor_.deregister_internal_descriptor(read_descriptor, reactor_data_);
  209. reactor_.cleanup_descriptor_data(reactor_data_);
  210. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  211. }
  212. break;
  213. case execution_context::fork_parent:
  214. if (state->fork_prepared_)
  215. {
  216. int read_descriptor = state->read_descriptor_;
  217. state->fork_prepared_ = false;
  218. lock.unlock();
  219. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  220. (void)read_descriptor;
  221. io_uring_service_.register_internal_io_object(io_object_data_,
  222. io_uring_service::read_op, new pipe_read_op);
  223. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  224. reactor_.register_internal_descriptor(reactor::read_op,
  225. read_descriptor, reactor_data_, new pipe_read_op);
  226. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  227. }
  228. break;
  229. case execution_context::fork_child:
  230. if (state->fork_prepared_)
  231. {
  232. asio::detail::signal_blocker blocker;
  233. close_descriptors();
  234. open_descriptors();
  235. int read_descriptor = state->read_descriptor_;
  236. state->fork_prepared_ = false;
  237. lock.unlock();
  238. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  239. (void)read_descriptor;
  240. io_uring_service_.register_internal_io_object(io_object_data_,
  241. io_uring_service::read_op, new pipe_read_op);
  242. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  243. reactor_.register_internal_descriptor(reactor::read_op,
  244. read_descriptor, reactor_data_, new pipe_read_op);
  245. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  246. }
  247. break;
  248. default:
  249. break;
  250. }
  251. #else // !defined(ASIO_WINDOWS)
  252. // && !defined(ASIO_WINDOWS_RUNTIME)
  253. // && !defined(__CYGWIN__)
  254. (void)fork_ev;
  255. #endif // !defined(ASIO_WINDOWS)
  256. // && !defined(ASIO_WINDOWS_RUNTIME)
  257. // && !defined(__CYGWIN__)
  258. }
  259. void signal_set_service::construct(
  260. signal_set_service::implementation_type& impl)
  261. {
  262. impl.signals_ = 0;
  263. }
  264. void signal_set_service::destroy(
  265. signal_set_service::implementation_type& impl)
  266. {
  267. asio::error_code ignored_ec;
  268. clear(impl, ignored_ec);
  269. cancel(impl, ignored_ec);
  270. }
  271. asio::error_code signal_set_service::add(
  272. signal_set_service::implementation_type& impl, int signal_number,
  273. signal_set_base::flags_t f, asio::error_code& ec)
  274. {
  275. // Check that the signal number is valid.
  276. if (signal_number < 0 || signal_number >= max_signal_number)
  277. {
  278. ec = asio::error::invalid_argument;
  279. return ec;
  280. }
  281. // Check that the specified flags are supported.
  282. #if !defined(ASIO_HAS_SIGACTION)
  283. if (f != signal_set_base::flags::dont_care)
  284. {
  285. ec = asio::error::operation_not_supported;
  286. return ec;
  287. }
  288. #endif // !defined(ASIO_HAS_SIGACTION)
  289. signal_state* state = get_signal_state();
  290. static_mutex::scoped_lock lock(state->mutex_);
  291. // Find the appropriate place to insert the registration.
  292. registration** insertion_point = &impl.signals_;
  293. registration* next = impl.signals_;
  294. while (next && next->signal_number_ < signal_number)
  295. {
  296. insertion_point = &next->next_in_set_;
  297. next = next->next_in_set_;
  298. }
  299. // Only do something if the signal is not already registered.
  300. if (next == 0 || next->signal_number_ != signal_number)
  301. {
  302. registration* new_registration = new registration;
  303. #if defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  304. // Register for the signal if we're the first.
  305. if (state->registration_count_[signal_number] == 0)
  306. {
  307. # if defined(ASIO_HAS_SIGACTION)
  308. using namespace std; // For memset.
  309. struct sigaction sa;
  310. memset(&sa, 0, sizeof(sa));
  311. sa.sa_handler = asio_signal_handler;
  312. sigfillset(&sa.sa_mask);
  313. if (f != signal_set_base::flags::dont_care)
  314. sa.sa_flags = static_cast<int>(f);
  315. if (::sigaction(signal_number, &sa, 0) == -1)
  316. # else // defined(ASIO_HAS_SIGACTION)
  317. if (::signal(signal_number, asio_signal_handler) == SIG_ERR)
  318. # endif // defined(ASIO_HAS_SIGACTION)
  319. {
  320. # if defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  321. ec = asio::error::invalid_argument;
  322. # else // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  323. ec = asio::error_code(errno,
  324. asio::error::get_system_category());
  325. # endif // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  326. delete new_registration;
  327. return ec;
  328. }
  329. # if defined(ASIO_HAS_SIGACTION)
  330. state->flags_[signal_number] = f;
  331. # endif // defined(ASIO_HAS_SIGACTION)
  332. }
  333. # if defined(ASIO_HAS_SIGACTION)
  334. // Otherwise check to see if the flags have changed.
  335. else if (f != signal_set_base::flags::dont_care)
  336. {
  337. if (f != state->flags_[signal_number])
  338. {
  339. using namespace std; // For memset.
  340. if (state->flags_[signal_number] != signal_set_base::flags::dont_care)
  341. {
  342. ec = asio::error::invalid_argument;
  343. delete new_registration;
  344. return ec;
  345. }
  346. struct sigaction sa;
  347. memset(&sa, 0, sizeof(sa));
  348. sa.sa_handler = asio_signal_handler;
  349. sigfillset(&sa.sa_mask);
  350. sa.sa_flags = static_cast<int>(f);
  351. if (::sigaction(signal_number, &sa, 0) == -1)
  352. {
  353. ec = asio::error_code(errno,
  354. asio::error::get_system_category());
  355. delete new_registration;
  356. return ec;
  357. }
  358. state->flags_[signal_number] = f;
  359. }
  360. }
  361. # endif // defined(ASIO_HAS_SIGACTION)
  362. #endif // defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  363. // Record the new registration in the set.
  364. new_registration->signal_number_ = signal_number;
  365. new_registration->queue_ = &impl.queue_;
  366. new_registration->next_in_set_ = next;
  367. *insertion_point = new_registration;
  368. // Insert registration into the registration table.
  369. new_registration->next_in_table_ = registrations_[signal_number];
  370. if (registrations_[signal_number])
  371. registrations_[signal_number]->prev_in_table_ = new_registration;
  372. registrations_[signal_number] = new_registration;
  373. ++state->registration_count_[signal_number];
  374. }
  375. ec = asio::error_code();
  376. return ec;
  377. }
  378. asio::error_code signal_set_service::remove(
  379. signal_set_service::implementation_type& impl,
  380. int signal_number, asio::error_code& ec)
  381. {
  382. // Check that the signal number is valid.
  383. if (signal_number < 0 || signal_number >= max_signal_number)
  384. {
  385. ec = asio::error::invalid_argument;
  386. return ec;
  387. }
  388. signal_state* state = get_signal_state();
  389. static_mutex::scoped_lock lock(state->mutex_);
  390. // Find the signal number in the list of registrations.
  391. registration** deletion_point = &impl.signals_;
  392. registration* reg = impl.signals_;
  393. while (reg && reg->signal_number_ < signal_number)
  394. {
  395. deletion_point = &reg->next_in_set_;
  396. reg = reg->next_in_set_;
  397. }
  398. if (reg != 0 && reg->signal_number_ == signal_number)
  399. {
  400. #if defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  401. // Set signal handler back to the default if we're the last.
  402. if (state->registration_count_[signal_number] == 1)
  403. {
  404. # if defined(ASIO_HAS_SIGACTION)
  405. using namespace std; // For memset.
  406. struct sigaction sa;
  407. memset(&sa, 0, sizeof(sa));
  408. sa.sa_handler = SIG_DFL;
  409. if (::sigaction(signal_number, &sa, 0) == -1)
  410. # else // defined(ASIO_HAS_SIGACTION)
  411. if (::signal(signal_number, SIG_DFL) == SIG_ERR)
  412. # endif // defined(ASIO_HAS_SIGACTION)
  413. {
  414. # if defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  415. ec = asio::error::invalid_argument;
  416. # else // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  417. ec = asio::error_code(errno,
  418. asio::error::get_system_category());
  419. # endif // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  420. return ec;
  421. }
  422. # if defined(ASIO_HAS_SIGACTION)
  423. state->flags_[signal_number] = signal_set_base::flags_t();
  424. # endif // defined(ASIO_HAS_SIGACTION)
  425. }
  426. #endif // defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  427. // Remove the registration from the set.
  428. *deletion_point = reg->next_in_set_;
  429. // Remove the registration from the registration table.
  430. if (registrations_[signal_number] == reg)
  431. registrations_[signal_number] = reg->next_in_table_;
  432. if (reg->prev_in_table_)
  433. reg->prev_in_table_->next_in_table_ = reg->next_in_table_;
  434. if (reg->next_in_table_)
  435. reg->next_in_table_->prev_in_table_ = reg->prev_in_table_;
  436. --state->registration_count_[signal_number];
  437. delete reg;
  438. }
  439. ec = asio::error_code();
  440. return ec;
  441. }
  442. asio::error_code signal_set_service::clear(
  443. signal_set_service::implementation_type& impl,
  444. asio::error_code& ec)
  445. {
  446. signal_state* state = get_signal_state();
  447. static_mutex::scoped_lock lock(state->mutex_);
  448. while (registration* reg = impl.signals_)
  449. {
  450. #if defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  451. // Set signal handler back to the default if we're the last.
  452. if (state->registration_count_[reg->signal_number_] == 1)
  453. {
  454. # if defined(ASIO_HAS_SIGACTION)
  455. using namespace std; // For memset.
  456. struct sigaction sa;
  457. memset(&sa, 0, sizeof(sa));
  458. sa.sa_handler = SIG_DFL;
  459. if (::sigaction(reg->signal_number_, &sa, 0) == -1)
  460. # else // defined(ASIO_HAS_SIGACTION)
  461. if (::signal(reg->signal_number_, SIG_DFL) == SIG_ERR)
  462. # endif // defined(ASIO_HAS_SIGACTION)
  463. {
  464. # if defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  465. ec = asio::error::invalid_argument;
  466. # else // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  467. ec = asio::error_code(errno,
  468. asio::error::get_system_category());
  469. # endif // defined(ASIO_WINDOWS) || defined(__CYGWIN__)
  470. return ec;
  471. }
  472. # if defined(ASIO_HAS_SIGACTION)
  473. state->flags_[reg->signal_number_] = signal_set_base::flags_t();
  474. # endif // defined(ASIO_HAS_SIGACTION)
  475. }
  476. #endif // defined(ASIO_HAS_SIGNAL) || defined(ASIO_HAS_SIGACTION)
  477. // Remove the registration from the registration table.
  478. if (registrations_[reg->signal_number_] == reg)
  479. registrations_[reg->signal_number_] = reg->next_in_table_;
  480. if (reg->prev_in_table_)
  481. reg->prev_in_table_->next_in_table_ = reg->next_in_table_;
  482. if (reg->next_in_table_)
  483. reg->next_in_table_->prev_in_table_ = reg->prev_in_table_;
  484. --state->registration_count_[reg->signal_number_];
  485. impl.signals_ = reg->next_in_set_;
  486. delete reg;
  487. }
  488. ec = asio::error_code();
  489. return ec;
  490. }
  491. asio::error_code signal_set_service::cancel(
  492. signal_set_service::implementation_type& impl,
  493. asio::error_code& ec)
  494. {
  495. ASIO_HANDLER_OPERATION((scheduler_.context(),
  496. "signal_set", &impl, 0, "cancel"));
  497. op_queue<operation> ops;
  498. {
  499. signal_state* state = get_signal_state();
  500. static_mutex::scoped_lock lock(state->mutex_);
  501. while (signal_op* op = impl.queue_.front())
  502. {
  503. op->ec_ = asio::error::operation_aborted;
  504. impl.queue_.pop();
  505. ops.push(op);
  506. }
  507. }
  508. scheduler_.post_deferred_completions(ops);
  509. ec = asio::error_code();
  510. return ec;
  511. }
  512. void signal_set_service::cancel_ops_by_key(
  513. signal_set_service::implementation_type& impl, void* cancellation_key)
  514. {
  515. op_queue<operation> ops;
  516. {
  517. op_queue<signal_op> other_ops;
  518. signal_state* state = get_signal_state();
  519. static_mutex::scoped_lock lock(state->mutex_);
  520. while (signal_op* op = impl.queue_.front())
  521. {
  522. impl.queue_.pop();
  523. if (op->cancellation_key_ == cancellation_key)
  524. {
  525. op->ec_ = asio::error::operation_aborted;
  526. ops.push(op);
  527. }
  528. else
  529. other_ops.push(op);
  530. }
  531. impl.queue_.push(other_ops);
  532. }
  533. scheduler_.post_deferred_completions(ops);
  534. }
  535. void signal_set_service::deliver_signal(int signal_number)
  536. {
  537. signal_state* state = get_signal_state();
  538. static_mutex::scoped_lock lock(state->mutex_);
  539. signal_set_service* service = state->service_list_;
  540. while (service)
  541. {
  542. op_queue<operation> ops;
  543. registration* reg = service->registrations_[signal_number];
  544. while (reg)
  545. {
  546. if (reg->queue_->empty())
  547. {
  548. ++reg->undelivered_;
  549. }
  550. else
  551. {
  552. while (signal_op* op = reg->queue_->front())
  553. {
  554. op->signal_number_ = signal_number;
  555. reg->queue_->pop();
  556. ops.push(op);
  557. }
  558. }
  559. reg = reg->next_in_table_;
  560. }
  561. service->scheduler_.post_deferred_completions(ops);
  562. service = service->next_;
  563. }
  564. }
  565. void signal_set_service::add_service(signal_set_service* service)
  566. {
  567. signal_state* state = get_signal_state();
  568. static_mutex::scoped_lock lock(state->mutex_);
  569. #if !defined(ASIO_WINDOWS) && !defined(__CYGWIN__)
  570. // If this is the first service to be created, open a new pipe.
  571. if (state->service_list_ == 0)
  572. open_descriptors();
  573. #endif // !defined(ASIO_WINDOWS) && !defined(__CYGWIN__)
  574. // If a scheduler_ object is thread-unsafe then it must be the only
  575. // scheduler used to create signal_set objects.
  576. if (state->service_list_ != 0)
  577. {
  578. if (!ASIO_CONCURRENCY_HINT_IS_LOCKING(SCHEDULER,
  579. service->scheduler_.concurrency_hint())
  580. || !ASIO_CONCURRENCY_HINT_IS_LOCKING(SCHEDULER,
  581. state->service_list_->scheduler_.concurrency_hint()))
  582. {
  583. std::logic_error ex(
  584. "Thread-unsafe execution context objects require "
  585. "exclusive access to signal handling.");
  586. asio::detail::throw_exception(ex);
  587. }
  588. }
  589. // Insert service into linked list of all services.
  590. service->next_ = state->service_list_;
  591. service->prev_ = 0;
  592. if (state->service_list_)
  593. state->service_list_->prev_ = service;
  594. state->service_list_ = service;
  595. #if !defined(ASIO_WINDOWS) \
  596. && !defined(ASIO_WINDOWS_RUNTIME) \
  597. && !defined(__CYGWIN__)
  598. // Register for pipe readiness notifications.
  599. int read_descriptor = state->read_descriptor_;
  600. lock.unlock();
  601. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  602. (void)read_descriptor;
  603. service->io_uring_service_.register_internal_io_object(
  604. service->io_object_data_, io_uring_service::read_op, new pipe_read_op);
  605. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  606. service->reactor_.register_internal_descriptor(reactor::read_op,
  607. read_descriptor, service->reactor_data_, new pipe_read_op);
  608. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  609. #endif // !defined(ASIO_WINDOWS)
  610. // && !defined(ASIO_WINDOWS_RUNTIME)
  611. // && !defined(__CYGWIN__)
  612. }
  613. void signal_set_service::remove_service(signal_set_service* service)
  614. {
  615. signal_state* state = get_signal_state();
  616. static_mutex::scoped_lock lock(state->mutex_);
  617. if (service->next_ || service->prev_ || state->service_list_ == service)
  618. {
  619. #if !defined(ASIO_WINDOWS) \
  620. && !defined(ASIO_WINDOWS_RUNTIME) \
  621. && !defined(__CYGWIN__)
  622. // Disable the pipe readiness notifications.
  623. int read_descriptor = state->read_descriptor_;
  624. lock.unlock();
  625. # if defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  626. (void)read_descriptor;
  627. service->io_uring_service_.deregister_io_object(service->io_object_data_);
  628. service->io_uring_service_.cleanup_io_object(service->io_object_data_);
  629. lock.lock();
  630. # else // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  631. service->reactor_.deregister_internal_descriptor(
  632. read_descriptor, service->reactor_data_);
  633. service->reactor_.cleanup_descriptor_data(service->reactor_data_);
  634. lock.lock();
  635. # endif // defined(ASIO_HAS_IO_URING_AS_DEFAULT)
  636. #endif // !defined(ASIO_WINDOWS)
  637. // && !defined(ASIO_WINDOWS_RUNTIME)
  638. // && !defined(__CYGWIN__)
  639. // Remove service from linked list of all services.
  640. if (state->service_list_ == service)
  641. state->service_list_ = service->next_;
  642. if (service->prev_)
  643. service->prev_->next_ = service->next_;
  644. if (service->next_)
  645. service->next_->prev_= service->prev_;
  646. service->next_ = 0;
  647. service->prev_ = 0;
  648. #if !defined(ASIO_WINDOWS) && !defined(__CYGWIN__)
  649. // If this is the last service to be removed, close the pipe.
  650. if (state->service_list_ == 0)
  651. close_descriptors();
  652. #endif // !defined(ASIO_WINDOWS) && !defined(__CYGWIN__)
  653. }
  654. }
  655. void signal_set_service::open_descriptors()
  656. {
  657. #if !defined(ASIO_WINDOWS) \
  658. && !defined(ASIO_WINDOWS_RUNTIME) \
  659. && !defined(__CYGWIN__)
  660. signal_state* state = get_signal_state();
  661. int pipe_fds[2];
  662. if (::pipe(pipe_fds) == 0)
  663. {
  664. state->read_descriptor_ = pipe_fds[0];
  665. ::fcntl(state->read_descriptor_, F_SETFL, O_NONBLOCK);
  666. state->write_descriptor_ = pipe_fds[1];
  667. ::fcntl(state->write_descriptor_, F_SETFL, O_NONBLOCK);
  668. #if defined(FD_CLOEXEC)
  669. ::fcntl(state->read_descriptor_, F_SETFD, FD_CLOEXEC);
  670. ::fcntl(state->write_descriptor_, F_SETFD, FD_CLOEXEC);
  671. #endif // defined(FD_CLOEXEC)
  672. }
  673. else
  674. {
  675. asio::error_code ec(errno,
  676. asio::error::get_system_category());
  677. asio::detail::throw_error(ec, "signal_set_service pipe");
  678. }
  679. #endif // !defined(ASIO_WINDOWS)
  680. // && !defined(ASIO_WINDOWS_RUNTIME)
  681. // && !defined(__CYGWIN__)
  682. }
  683. void signal_set_service::close_descriptors()
  684. {
  685. #if !defined(ASIO_WINDOWS) \
  686. && !defined(ASIO_WINDOWS_RUNTIME) \
  687. && !defined(__CYGWIN__)
  688. signal_state* state = get_signal_state();
  689. if (state->read_descriptor_ != -1)
  690. ::close(state->read_descriptor_);
  691. state->read_descriptor_ = -1;
  692. if (state->write_descriptor_ != -1)
  693. ::close(state->write_descriptor_);
  694. state->write_descriptor_ = -1;
  695. #endif // !defined(ASIO_WINDOWS)
  696. // && !defined(ASIO_WINDOWS_RUNTIME)
  697. // && !defined(__CYGWIN__)
  698. }
  699. void signal_set_service::start_wait_op(
  700. signal_set_service::implementation_type& impl, signal_op* op)
  701. {
  702. scheduler_.work_started();
  703. signal_state* state = get_signal_state();
  704. static_mutex::scoped_lock lock(state->mutex_);
  705. registration* reg = impl.signals_;
  706. while (reg)
  707. {
  708. if (reg->undelivered_ > 0)
  709. {
  710. --reg->undelivered_;
  711. op->signal_number_ = reg->signal_number_;
  712. scheduler_.post_deferred_completion(op);
  713. return;
  714. }
  715. reg = reg->next_in_set_;
  716. }
  717. impl.queue_.push(op);
  718. }
  719. } // namespace detail
  720. } // namespace asio
  721. #include "asio/detail/pop_options.hpp"
  722. #endif // ASIO_DETAIL_IMPL_SIGNAL_SET_SERVICE_IPP