116 lines
3.0 KiB
C++
116 lines
3.0 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
|
|
//
|
|
// 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/WineventPoll.h"
|
|
|
|
char disable_linker_warning_about_empty_file_wineventpoll_cpp TD_UNUSED;
|
|
|
|
#ifdef TD_POLL_WINEVENT
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/misc.h"
|
|
#include "td/utils/port/Fd.h"
|
|
#include "td/utils/port/PollBase.h"
|
|
#include "td/utils/port/sleep.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
namespace detail {
|
|
IOCP::~IOCP() {
|
|
clear();
|
|
}
|
|
|
|
void IOCP::loop() {
|
|
IOCP::Guard guard(this);
|
|
while (true) {
|
|
DWORD bytes = 0;
|
|
ULONG_PTR key = 0;
|
|
OVERLAPPED *overlapped = nullptr;
|
|
bool ok = GetQueuedCompletionStatus(iocp_handle_.io_handle(), &bytes, &key, &overlapped, 1000);
|
|
if (bytes || key || overlapped) {
|
|
LOG(ERROR) << "Got iocp " << bytes << " " << key << " " << overlapped;
|
|
}
|
|
if (ok) {
|
|
auto callback = reinterpret_cast<IOCP::Callback *>(key);
|
|
if (callback == nullptr) {
|
|
LOG(ERROR) << "Interrupt IOCP loop";
|
|
return;
|
|
}
|
|
callback->on_iocp(bytes, overlapped);
|
|
} else {
|
|
if (overlapped != nullptr) {
|
|
auto error = OS_ERROR("from iocp");
|
|
auto callback = reinterpret_cast<IOCP::Callback *>(key);
|
|
CHECK(callback != nullptr);
|
|
callback->on_iocp(std::move(error), overlapped);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void IOCP::interrupt_loop() {
|
|
post(0, nullptr, nullptr);
|
|
}
|
|
|
|
void IOCP::init() {
|
|
CHECK(!iocp_handle_);
|
|
auto res = CreateIoCompletionPort(INVALID_HANDLE_VALUE, nullptr, 0, 0);
|
|
if (res == nullptr) {
|
|
auto error = OS_ERROR("Iocp creation failed");
|
|
LOG(FATAL) << error;
|
|
}
|
|
iocp_handle_ = NativeFd(res);
|
|
}
|
|
|
|
void IOCP::clear() {
|
|
iocp_handle_.close();
|
|
}
|
|
|
|
void IOCP::subscribe(const NativeFd &native_fd, Callback *callback) {
|
|
CHECK(iocp_handle_);
|
|
auto iocp_handle =
|
|
CreateIoCompletionPort(native_fd.io_handle(), iocp_handle_.io_handle(), reinterpret_cast<ULONG_PTR>(callback), 0);
|
|
if (iocp_handle == INVALID_HANDLE_VALUE) {
|
|
auto error = OS_ERROR("CreateIoCompletionPort");
|
|
LOG(FATAL) << error;
|
|
}
|
|
CHECK(iocp_handle == iocp_handle_.io_handle()) << iocp_handle << " " << iocp_handle_.io_handle();
|
|
}
|
|
|
|
void IOCP::post(size_t size, Callback *callback, OVERLAPPED *overlapped) {
|
|
PostQueuedCompletionStatus(iocp_handle_.io_handle(), DWORD(size), reinterpret_cast<ULONG_PTR>(callback), overlapped);
|
|
}
|
|
|
|
void WineventPoll::init() {
|
|
}
|
|
|
|
void WineventPoll::clear() {
|
|
}
|
|
|
|
void WineventPoll::subscribe(PollableFd fd, PollFlags flags) {
|
|
fd.release_as_list_node();
|
|
}
|
|
|
|
void WineventPoll::unsubscribe(PollableFdRef fd) {
|
|
fd.lock();
|
|
}
|
|
|
|
void WineventPoll::unsubscribe_before_close(PollableFdRef fd) {
|
|
unsubscribe(std::move(fd));
|
|
}
|
|
|
|
void WineventPoll::run(int timeout_ms) {
|
|
UNREACHABLE();
|
|
}
|
|
|
|
} // namespace detail
|
|
} // namespace td
|
|
|
|
#endif
|