읽기/쓰기 일단 완료

impl/socket/iocp.cpp

@@ -11,6 +11,29 @@ IOCP::IOCP() {
 
 IOCP::~IOCP() { destruct(); }
 
+void IOCP::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 IOCP::destruct() {
 #ifdef __linux__
 
@@ -40,7 +63,7 @@ void IOCP::registerSocket(IOCPPASSINDATA* data) {
 #endif
 }
 
-int IOCP::recv(IOCPPASSINDATA* data) {
+int IOCP::recv(IOCPPASSINDATA* data) { //읽은 바이트수가 무조건 100임? 왜..?
   SOCKET sock = data->socket->sock;
   std::lock_guard lock(*GetRecvQueueMutex_(sock));
   auto queue = GetRecvQueue_(sock);
@@ -88,6 +111,73 @@ int IOCP::send(SOCKET sock, std::vector<IOCPPASSINDATA*>* data) {
   return 0;
 }
 
+int IOCP::GetRecvedPacketCount(SOCKET sock) {
+  std::lock_guard lock(socket_mod_mutex_);
+  auto queue = GetRecvQueue_(sock);
+  return queue->size();
+}
+
+void IOCP::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;
+  std::lock_guard lock(*GetRecvQueueMutex_(sock));
+  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->event = IOCPEVENT::READ;
+    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);
+}
+
 std::shared_ptr<std::list<IOCPPASSINDATA*>> IOCP::GetSendQueue_(SOCKET sock) {
   std::lock_guard lock(socket_mod_mutex_);
   if (send_queue_.find(sock) == send_queue_.end()) {