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

읽기/쓰기 socket 비동기 구현 완료

Browse Source
This commit is contained in:
HappyTanuki
2025-06-05 02:12:50 +09:00
parent c0e0279e5c
commit 6428db9d31
3 changed files with 38 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
impl/socket/iocp.cpp
View File

@@ -23,10 +23,19 @@ void IOCP::registerSocket(IOCPPASSINDATA* data) {
(HANDLE)data->socket->sock, completionPort_, data->socket->sock, 0);
if (returnData == 0) completionPort_ = returnData;
data->event = IOCPEVENT::READ;
IOCPPASSINDATA* recv_data = new IOCPPASSINDATA(data->bufsize);
recv_data->event = IOCPEVENT::READ;
recv_data->socket = data->socket;
DWORD recvbytes = 0, flags = 0;
::WSARecv(data->socket->sock, &data->wsabuf, 1, &recvbytes, &flags,
&data->overlapped, NULL);
int result = ::WSARecv(recv_data->socket->sock, &recv_data->wsabuf, 1,
&recvbytes, &flags, &recv_data->overlapped, NULL);
if (result == SOCKET_ERROR) {
int err = ::WSAGetLastError();
if (err != WSA_IO_PENDING) {
spdlog::error("WSARecv failed: {}", err);
// 반드시 여기서 리턴하거나 처리해야 합니다.
}
}
#endif
}
@@ -131,9 +140,9 @@ void IOCP::packet_sender_(SOCKET sock) {
int data_len = 0;
if (proto_ == SessionProtocol::TLS || proto_ == SessionProtocol::QUIC) {
int ret = ::SSL_write(front->ssl, front->wsabuf.buf, front->wsabuf.len);
int ret = ::SSL_write(front->ssl.get(), front->wsabuf.buf, front->wsabuf.len);
if (ret <= 0) {
int err = ::SSL_get_error(front->ssl, ret);
int err = ::SSL_get_error(front->ssl.get(), ret);
if (err == SSL_ERROR_WANT_READ || err == SSL_ERROR_WANT_WRITE) {
queue->push_front(front);
break;
@@ -143,7 +152,7 @@ void IOCP::packet_sender_(SOCKET sock) {
break;
}
while ((data_len = ::BIO_read(front->wbio, buf.data(), buf.size())) > 0) {
while ((data_len = ::BIO_read(::SSL_get_wbio(front->ssl.get()), buf.data(), buf.size())) > 0) {
wsabuf.buf = buf.data();
wsabuf.len = data_len;