1 files changed, 137 insertions, 0 deletions
diff --git a/src/event/modules/ngx_iocp_module.c b/src/event/modules/ngx_iocp_module.c
new file mode 100644
index 000000000..f474c4c61
--- /dev/null
+++ b/src/event/modules/ngx_iocp_module.c
@@ -0,0 +1,137 @@
+
+#include <ngx_config.h>
+
+#include <ngx_core.h>
+#include <ngx_log.h>
+#include <ngx_errno.h>
+#include <ngx_time.h>
+#include <ngx_connection.h>
+#include <ngx_event.h>
+#include <ngx_event_timer.h>
+
+#include <ngx_iocp_module.h>
+
+
+int ngx_iocp_threads = 0;;
+
+
+static HANDLE        iocp;
+static ngx_event_t  *timer_queue;
+
+
+int ngx_iocp_init(int max_connections, ngx_log_t *log)
+{
+    iocp = CreateIoCompletionPort(INVALID_HANDLE_VALUE,
+                                  NULL, 0, ngx_iocp_threads);
+
+    if (iocp == NULL) {
+        ngx_log_error(NGX_LOG_EMERG, log, ngx_errno,
+                      "CreateIoCompletionPort() failed");
+        return NGX_ERROR;
+    }
+
+    timer_queue = ngx_event_init_timer(log);
+    if (timer_queue == NULL) {
+        return NGX_ERROR;
+    }
+
+    ngx_event_actions.process = ngx_iocp_process_events;
+
+    ngx_event_flags = NGX_HAVE_AIO_EVENT|NGX_HAVE_IOCP_EVENT;
+
+    return NGX_OK;
+}
+
+
+int ngx_iocp_add_event(ngx_event_t *ev)
+{
+    ngx_connection_t  *c;
+
+    c = (ngx_connection_t *) ev->data;
+
+    ngx_log_debug(ev->log, "iocp: %d, %08x:%08x" _ c->fd _ ev _ &ev->ovlp);
+
+    if (CreateIoCompletionPort((HANDLE) c->fd, iocp, (DWORD) ev, 0) == NULL) {
+        ngx_log_error(NGX_LOG_ALERT, ev->log, ngx_errno,
+                      "CreateIoCompletionPort() failed");
+        return NGX_ERROR;
+    }
+
+    return NGX_OK;
+}
+
+
+int ngx_iocp_process_events(ngx_log_t *log)
+{
+    int                rc;
+    size_t             bytes;
+    ngx_err_t          err;
+    ngx_msec_t         timer, delta;
+    ngx_event_t       *ev, *e;
+    ngx_event_ovlp_t  *ovlp;
+
+    ngx_log_debug(log, "iocp");
+
+    timer = ngx_event_find_timer();
+
+    if (timer) {
+        delta = ngx_msec();
+
+    } else {
+        timer = INFINITE;
+        delta = 0;
+    }
+
+    ngx_log_debug(log, "iocp timer: %d" _ timer);
+
+#if 1
+    rc = GetQueuedCompletionStatus(iocp, &bytes, (LPDWORD) &e,
+                                   (LPOVERLAPPED *) &ovlp, timer);
+    ngx_log_debug(log, "iocp: %d, %d:%08x:%08x" _ rc _ bytes _ e _ ovlp);
+    if (rc == 0) {
+#else
+    if (GetQueuedCompletionStatus(iocp, &bytes, (LPDWORD) &e,
+                                  (LPOVERLAPPED *) &ovlp, timer) == 0) {
+#endif
+        err = ngx_errno;
+
+        if (ovlp == NULL) {
+            if (err != WAIT_TIMEOUT) {
+                ngx_log_error(NGX_LOG_ALERT, log, err,
+                              "GetQueuedCompletionStatus() failed");
+
+                return NGX_ERROR;
+            }
+
+        } else {
+            ovlp->error = err;
+        }
+    }
+
+    if (timer != INFINITE) {
+        delta = ngx_msec() - delta;
+    }
+
+    if (ovlp) {
+        ev = ovlp->event;
+
+ngx_log_debug(log, "iocp ev: %08x" _ ev);
+
+        if (ev == e) {
+            ev->ready = 1;
+            ev->available = bytes;
+        }
+
+ngx_log_debug(log, "iocp ev: %08x" _ ev->event_handler);
+
+        if (ev->event_handler(ev) == NGX_ERROR) {
+            ev->close_handler(ev);
+        }
+    }
+
+    if (timer != INFINITE) {
+        ngx_event_expire_timers(delta);
+    }
+
+    return NGX_OK;
+}