iocp.hpp

#pragma once
 
#include <coroio/base.hpp>
#include <coroio/socket.hpp>
#include <coroio/poller.hpp>
#include <coroio/arena.hpp>
 
namespace NNet {
 
class TIOCp: public TPollerBase {
public:
    using TSocket = NNet::TPollerDrivenSocket<TIOCp>;
    using TFileHandle = NNet::TPollerDrivenFileHandle<TIOCp>;
 
    TIOCp();
    ~TIOCp();
    void Read(int fd, void* buf, int size, std::coroutine_handle<> handle);
    void Write(int fd, const void* buf, int size, std::coroutine_handle<> handle);
    void Recv(int fd, void* buf, int size, std::coroutine_handle<> handle);
    void Send(int fd, const void* buf, int size, std::coroutine_handle<> handle);
    void Accept(int fd, struct sockaddr* addr, socklen_t* len, std::coroutine_handle<> handle);
    void Connect(int fd, const sockaddr* addr, socklen_t len, std::coroutine_handle<> handle);
    void Cancel(int fd);
    void Register(int fd);
    int Result();
    void Poll();
 
private:
    struct TIO {
        OVERLAPPED overlapped;
        THandle handle;
        struct sockaddr* addr = nullptr; // for accept
        socklen_t* len = nullptr; // for accept
        int sock = -1; // for accept
 
        TIO() {
            memset(&overlapped, 0, sizeof(overlapped));
        }
    };
 
    long GetTimeoutMs();
    TIO* NewTIO();
    void FreeTIO(TIO*);
 
    HANDLE Port_;
 
    // Allocator to avoid dynamic memory allocation for each IOCP event structure.
    TArenaAllocator<TIO> Allocator_;
    std::vector<OVERLAPPED_ENTRY> Entries_;
    std::queue<int> Results_;
};
 
}