84 lines
1.6 KiB
C++
84 lines
1.6 KiB
C++
#include <stdio.h>
|
|
|
|
#include <WinSock2.h>
|
|
#include <ws2tcpip.h>
|
|
#include <iphlpapi.h>
|
|
|
|
#include <memory>
|
|
|
|
struct WSA {
|
|
WSADATA wsa;
|
|
|
|
explicit WSA(WORD version = MAKEWORD(2, 2));
|
|
~WSA() noexcept;
|
|
};
|
|
|
|
struct AddrInfoPtr {
|
|
struct addrinfo *info;
|
|
|
|
AddrInfoPtr() noexcept;
|
|
~AddrInfoPtr() noexcept;
|
|
|
|
addrinfo** operator&() noexcept;
|
|
addrinfo& operator*() noexcept;
|
|
addrinfo* operator->() noexcept;
|
|
const addrinfo& operator*() const noexcept;
|
|
|
|
operator addrinfo*() noexcept;
|
|
};
|
|
|
|
struct AddrInfo {
|
|
struct addrinfo info;
|
|
|
|
AddrInfo() noexcept;
|
|
~AddrInfo() noexcept = default;
|
|
|
|
addrinfo* operator->() noexcept;
|
|
addrinfo* operator&() noexcept;
|
|
|
|
operator const addrinfo&() const noexcept;
|
|
operator addrinfo&() noexcept;
|
|
|
|
AddrInfoPtr get_addrinfo(PCSTR node, PCSTR service);
|
|
};
|
|
|
|
struct Socket {
|
|
enum AddrAction {
|
|
NO_ACTION,
|
|
CONNECT,
|
|
BIND,
|
|
};
|
|
|
|
SOCKET s;
|
|
struct sockaddr addr;
|
|
|
|
explicit Socket(SOCKET init = INVALID_SOCKET) noexcept;
|
|
~Socket() noexcept;
|
|
|
|
operator SOCKET() const noexcept;
|
|
|
|
explicit operator bool() const noexcept;
|
|
bool is_valid() const noexcept;
|
|
|
|
void from_addrinfo(const struct addrinfo &info, AddrAction action = NO_ACTION);
|
|
|
|
Socket accept_client() const noexcept;
|
|
|
|
void stop_send() const;
|
|
|
|
int send_data(const Buffer &data) const;
|
|
int recv_data(Buffer &data) const;
|
|
};
|
|
|
|
struct Buffer {
|
|
std::unique_ptr<char[]> data;
|
|
int size;
|
|
|
|
Buffer(int len);
|
|
|
|
char* get();
|
|
const char* get() const;
|
|
|
|
int len() const;
|
|
};
|