HappyTanuki/Np_Term
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 5 Wiki Activity

읽기/쓰기 일단 완료

Browse Source
This commit is contained in:
HappyTanuki
2025-06-05 03:00:29 +09:00
parent 6428db9d31
commit 5fdae769e2
3 changed files with 128 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
include/socket/iocp.h
View File

@@ -60,7 +60,7 @@ struct IOCPPASSINDATA {
std::memset(&overlapped, 0, sizeof(overlapped));
event = IOCPEVENT::QUIT;
socket = nullptr;
ssl = std::make_shared<SSL>(::SSL_new(ctx), ::SSL_free);
ssl = std::shared_ptr<SSL>(::SSL_new(ctx), ::SSL_free);
::SSL_set_bio(ssl.get(), ::BIO_new(::BIO_s_mem()),
::BIO_new(::BIO_s_mem()));
transferredbytes = 0;
@@ -118,100 +118,21 @@ class IOCP {
IOCP();
~IOCP();
void init(utils::ThreadPool* __IOCPThread, SessionProtocol proto) {
IOCPThread_ = __IOCPThread;
proto_ = proto;
#ifdef _WIN32
completionPort_ =
::CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
if (completionPort_ == NULL) {
spdlog::critical("CreateIoCompletionPort()");
std::exit(EXIT_FAILURE);
}
int tCount = __IOCPThread->threadCount;
spdlog::info("Resizing threadpool size to: {}", tCount * 2);
__IOCPThread->respawnWorker(tCount * 2);
for (int i = 0; i < tCount; i++)
IOCPThread_->enqueueJob(
[this](utils::ThreadPool* th, std::uint8_t __) { iocpWatcher_(th); },
0);
#endif
}
void init(utils::ThreadPool* __IOCPThread, SessionProtocol proto);
void destruct();
void registerSocket(IOCPPASSINDATA* data);
int recv(IOCPPASSINDATA* data);
// data는 한 가지 소켓에 보내는 패킷만 담아야 합니다
int send(SOCKET sock, std::vector<IOCPPASSINDATA*>* data);
int recv(IOCPPASSINDATA* data);
int GetRecvedPacketCount(SOCKET sock);
private:
#ifdef _WIN32
void iocpWatcher_(utils::ThreadPool* IOCPThread) {
IOCPPASSINDATA* data;
SOCKET sock;
DWORD cbTransfrred;
int jitter = jitterDist_(gen_);
int retVal = GetQueuedCompletionStatus(completionPort_, &cbTransfrred,
(PULONG_PTR)&sock,
(LPOVERLAPPED*)&data, 1000 + jitter);
if (retVal == 0 || cbTransfrred == 0) {
DWORD lasterror = GetLastError();
if (lasterror == WAIT_TIMEOUT) {
IOCPThread->enqueueJob([this](utils::ThreadPool* th,
std::uint8_t __) { iocpWatcher_(th); },
0);
return;
}
data->event = IOCPEVENT::QUIT;
spdlog::debug("Disconnected. [{}]",
(std::string)(data->socket->remoteAddr));
delete data;
} else {
data->transferredbytes = cbTransfrred;
}
std::vector<char> buf(16384); // SSL_read최대 반환 크기
int red_data = 0;
auto queue_list = GetRecvQueue_(data->socket->sock);
if (data->event == IOCPEVENT::READ) {
if (proto_ == SessionProtocol::TLS || proto_ == SessionProtocol::QUIC) {
::BIO_write(::SSL_get_rbio(data->ssl.get()), data->wsabuf.buf,
cbTransfrred);
while ((red_data =
::SSL_read(data->ssl.get(), buf.data(), buf.size())) > 0) {
queue_list->emplace_back(std::make_pair(
std::vector<char>(buf.begin(), buf.begin() + red_data), 0));
}
} else {
::memcpy(buf.data(), data->wsabuf.buf, data->transferredbytes);
queue_list->emplace_back(std::make_pair(
std::vector<char>(buf.begin(),
buf.begin() + data->transferredbytes),
0));
}
DWORD recvbytes = 0, flags = 0;
IOCPPASSINDATA* recv_data = new IOCPPASSINDATA(data->bufsize);
recv_data->socket = data->socket;
delete data;
::WSARecv(recv_data->socket->sock, &recv_data->wsabuf, 1, &recvbytes,
&flags, &recv_data->overlapped, NULL);
} else { // WRITE 시, 무시한다.
delete data;
}
IOCPThread->enqueueJob(
[this](utils::ThreadPool* th, std::uint8_t __) { iocpWatcher_(th); },
0);
}
void iocpWatcher_(utils::ThreadPool* IOCPThread);
#elif __linux__
#endif