10#include "lldb/Host/Config.h"
14#include "llvm/Config/llvm-config.h"
15#include "llvm/Support/WindowsError.h"
31static DWORD
ToTimeout(std::optional<MainLoopWindows::TimePoint> point) {
32 using namespace std::chrono;
37 nanoseconds dur = (std::max)(*point - steady_clock::now(), nanoseconds(0));
38 return ceil<milliseconds>(dur).count();
45 explicit PipeEvent(HANDLE handle)
46 : IOEvent(CreateEventW(NULL, TRUE,
48 m_handle(handle), m_ready(CreateEventW(NULL, TRUE,
50 assert(m_event && m_ready);
51 m_monitor_thread = std::thread(&PipeEvent::Monitor,
this);
54 ~PipeEvent()
override {
55 if (m_monitor_thread.joinable()) {
60 CancelIoEx(m_handle, NULL);
61 }
while (WaitForSingleObject(m_monitor_thread.native_handle(), 1) ==
63 m_monitor_thread.join();
69 void WillPoll()
override {
70 if (WaitForSingleObject(m_event, 0) != WAIT_TIMEOUT) {
75 if (WaitForSingleObject(m_ready, 0) != WAIT_TIMEOUT) {
83 void Disarm()
override { ResetEvent(m_event); }
89 WaitForSingleObject(m_ready, INFINITE);
95 ZeroMemory(&ov,
sizeof(ov));
100 ReadFile(m_handle, buf, 0, &bytes_read, &ov);
101 DWORD bytes_available = 0;
102 DWORD err = GetLastError();
103 if (!success && err == ERROR_IO_PENDING) {
104 success = GetOverlappedResult(m_handle, &ov, &bytes_read,
106 err = GetLastError();
110 PeekNamedPipe(m_handle, NULL, 0, NULL, &bytes_available, NULL);
111 err = GetLastError();
114 if (bytes_available == 0) {
118 }
else if (err == ERROR_NO_DATA) {
122 }
else if (err == ERROR_OPERATION_ABORTED) {
134 WaitForSingleObject(m_ready, INFINITE);
135 }
while (!m_stopped);
141 std::thread m_monitor_thread;
142 std::atomic<bool> m_stopped =
false;
147 explicit SocketEvent(SOCKET socket)
148 : IOEvent(WSACreateEvent()), m_socket(socket) {
149 assert(m_event != WSA_INVALID_EVENT);
152 ~SocketEvent()
override { WSACloseEvent(m_event); }
154 void WillPoll()
override {
156 WSAEventSelect(m_socket, m_event, FD_READ | FD_ACCEPT | FD_CLOSE);
161 void DidPoll()
override {
162 int result = WSAEventSelect(m_socket, WSA_INVALID_EVENT, 0);
167 void Disarm()
override { WSAResetEvent(m_event); }
182 assert(result == TRUE);
187 std::vector<HANDLE> events;
190 fd_info.event->WillPoll();
191 events.push_back(fd_info.event->GetHandle());
196 WSAWaitForMultipleEvents(events.size(), events.data(), FALSE,
200 fd_info.event->DidPoll();
202 if (result >= WSA_WAIT_EVENT_0 && result < WSA_WAIT_EVENT_0 + events.size())
203 return result - WSA_WAIT_EVENT_0;
206 if (result == WSA_WAIT_TIMEOUT)
207 return events.size() - 1;
209 return llvm::createStringError(llvm::inconvertibleErrorCode(),
210 "WSAWaitForMultipleEvents failed");
216 if (!object_sp || !object_sp->IsValid()) {
226 "File descriptor %p already monitored.", waitable_handle);
232 std::make_unique<SocketEvent>(
233 reinterpret_cast<SOCKET
>(waitable_handle)),
236 DWORD file_type = GetFileType(waitable_handle);
237 if (file_type != FILE_TYPE_PIPE) {
243 m_read_fds[waitable_handle] = {std::make_unique<PipeEvent>(waitable_handle),
262 llvm::Expected<size_t> signaled_event =
Poll();
267 auto &KV = *std::next(
m_read_fds.begin(), *signaled_event);
268 KV.second.event->Disarm();
269 KV.second.callback(*
this);
static llvm::raw_ostream & error(Stream &strm)
static DWORD ToTimeout(std::optional< MainLoopWindows::TimePoint > point)
static const WaitableHandle kInvalidHandleValue
lldb::file_t WaitableHandle
std::unique_ptr< ReadHandle > ReadHandleUP
std::optional< TimePoint > GetNextWakeupTime()
ReadHandleUP CreateReadHandle(const lldb::IOObjectSP &object_sp)
std::function< void(MainLoopBase &)> Callback
void Interrupt() override
llvm::Expected< size_t > Poll()
~MainLoopWindows() override
llvm::DenseMap< IOObject::WaitableHandle, FdInfo > m_read_fds
ReadHandleUP RegisterReadObject(const lldb::IOObjectSP &object_sp, const Callback &callback, Status &error) override
void UnregisterReadObject(IOObject::WaitableHandle handle) override
static Status FromErrorStringWithFormat(const char *format,...) __attribute__((format(printf
static Status FromErrorString(const char *str)
static Status FromError(llvm::Error error)
Avoid using this in new code. Migrate APIs to llvm::Expected instead.
#define UNUSED_IF_ASSERT_DISABLED(x)
A class that represents a running process on the host machine.
std::shared_ptr< lldb_private::IOObject > IOObjectSP