// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024 // // 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) // #include "td/utils/port/detail/EventFdBsd.h" char disable_linker_warning_about_empty_file_event_fd_bsd_cpp TD_UNUSED; #ifdef TD_EVENTFD_BSD #include "td/utils/logging.h" #include "td/utils/port/detail/NativeFd.h" #include "td/utils/port/detail/skip_eintr.h" #include "td/utils/port/PollFlags.h" #include "td/utils/port/SocketFd.h" #include "td/utils/Slice.h" #include #include #include #include #include namespace td { namespace detail { // TODO: it is extremely non optimal on Darwin. kqueue events should be used instead void EventFdBsd::init() { int fds[2]; int err = socketpair(AF_UNIX, SOCK_STREAM, 0, fds); auto socketpair_errno = errno; #if TD_CYGWIN // it looks like CYGWIN bug int max_retries = 1000000; while (err == -1 && socketpair_errno == EADDRINUSE && max_retries-- > 0) { err = socketpair(AF_UNIX, SOCK_STREAM, 0, fds); socketpair_errno = errno; } // LOG_IF(ERROR, max_retries < 1000000) << max_retries; #endif LOG_IF(FATAL, err == -1) << Status::PosixError(socketpair_errno, "socketpair failed"); auto fd_a = NativeFd(fds[0]); auto fd_b = NativeFd(fds[1]); fd_a.set_is_blocking_unsafe(false).ensure(); fd_b.set_is_blocking_unsafe(false).ensure(); in_ = SocketFd::from_native_fd(std::move(fd_a)).move_as_ok(); out_ = SocketFd::from_native_fd(std::move(fd_b)).move_as_ok(); } bool EventFdBsd::empty() { return in_.empty(); } void EventFdBsd::close() { in_.close(); out_.close(); } Status EventFdBsd::get_pending_error() { return Status::OK(); } PollableFdInfo &EventFdBsd::get_poll_info() { return out_.get_poll_info(); } void EventFdBsd::release() { int value = 1; auto result = in_.write(Slice(reinterpret_cast(&value), sizeof(value))); if (result.is_error()) { LOG(FATAL) << "EventFdBsd write failed: " << result.error(); } size_t size = result.ok(); if (size != sizeof(value)) { LOG(FATAL) << "EventFdBsd write returned " << size << " instead of " << sizeof(value); } } void EventFdBsd::acquire() { sync_with_poll(out_); out_.get_poll_info().add_flags(PollFlags::Read()); while (can_read_local(out_)) { uint8 value[1024]; auto result = out_.read(MutableSlice(value, sizeof(value))); if (result.is_error()) { LOG(FATAL) << "EventFdBsd read failed:" << result.error(); } } } void EventFdBsd::wait(int timeout_ms) { detail::skip_eintr_timeout( [this](int timeout_ms) { pollfd fd; fd.fd = get_poll_info().native_fd().fd(); fd.events = POLLIN; return poll(&fd, 1, timeout_ms); }, timeout_ms); } } // namespace detail } // namespace td #endif