wip: try to implement socket in init_by_lua* phase.

Author: jiahao

Diff for: src/ngx_http_lua_common.h

@@ -141,6 +141,7 @@ typedef struct {
 #define NGX_HTTP_LUA_CONTEXT_SSL_SESS_FETCH     0x1000
 #define NGX_HTTP_LUA_CONTEXT_EXIT_WORKER        0x2000
 #define NGX_HTTP_LUA_CONTEXT_SSL_CLIENT_HELLO   0x4000
+#define NGX_HTTP_LUA_CONTEXT_INIT               0x8000
 
 
 #define NGX_HTTP_LUA_FFI_NO_REQ_CTX         -100

Diff for: src/ngx_http_lua_event.h

@@ -35,19 +35,63 @@ extern int ngx_http_lua_event_inited;
 static ngx_inline ngx_int_t
 ngx_http_lua_init_event(ngx_cycle_t *cycle)
 {
-    void              ***ccf;
-    ngx_event_conf_t    *ecf;
+    ngx_module_t        *module;
+    ngx_module_t       **modules;
+    ngx_event_module_t  *event_module;
+    ngx_uint_t           i;
+    ngx_connection_t    *c, *next;
+    ngx_event_t         *rev, *wev;
+
+    module = NULL;
+
+#if (nginx_version >= 1009011)
+    modules = cycle->modules;
+#else
+    modules = ngx_modules;
+#endif
+
+    for (i = 0; modules[i]; i++) {
+        if (modules[i]->type != NGX_EVENT_MODULE) {
+            continue;
+        }
+
+        event_module = modules[i]->ctx;
+        if (ngx_strcmp(event_module->name->data, "event_core") == 0)
+        {
+            continue;
+        }
 
-    ccf = ngx_get_conf(cycle->conf_ctx, ngx_events_module);
-    if (ccf == NULL) {
+        module = modules[i];
+        break;
+    }
+
+    if (module == NULL) {
+        ngx_log_error(NGX_LOG_EMERG, cycle->log, 0, "no events module found");
         return NGX_ERROR;
     }
 
-    ecf = (*ccf)[ngx_event_core_module.ctx_index];
+    event_module = module->ctx;
+
+    /* FIXME: should init event_module actions here, like:
+     * ```
+     * if (event_module->actions.init(cycle, ngx_timer_resolution) != NGX_OK) {
+     * ```
+     *
+     * but `epcf` is not initial here before
+     * ```
+     *  static ngx_int_t
+     *  ngx_epoll_init(ngx_cycle_t *cycle, ngx_msec_t timer)
+     *  {
+     *      ngx_epoll_conf_t  *epcf;
+     *
+     *      // Segmentation fault below
+     *      epcf = ngx_event_get_conf(cycle->conf_ctx, ngx_epoll_module);
+     * ```
+     */
 
 #if (NGX_HAVE_EPOLL) && !(NGX_TEST_BUILD_EPOLL)
 
-    if (ngx_strcmp(ecf->name, "epoll") == 0) {
+    if (ngx_strcmp(event_module->name->data, "epoll") == 0) {
         ngx_http_lua_event_actions = ngx_http_lua_epoll;
 
     } else
@@ -56,7 +100,7 @@ ngx_http_lua_init_event(ngx_cycle_t *cycle)
 
 #if (NGX_HAVE_POLL)
 
-    if (ngx_strcmp(ecf->name, "poll") == 0) {
+    if (ngx_strcmp(event_module->name->data, "poll") == 0) {
         ngx_http_lua_event_actions = ngx_http_lua_poll;
 
     } else
@@ -65,7 +109,7 @@ ngx_http_lua_init_event(ngx_cycle_t *cycle)
 
 #if (NGX_HAVE_KQUEUE)
 
-    if (ngx_strcmp(ecf->name, "kqueue") == 0) {
+    if (ngx_strcmp(event_module->name->data, "kqueue") == 0) {
         ngx_http_lua_event_actions = ngx_http_lua_kqueue;
 
     } else
@@ -76,6 +120,55 @@ ngx_http_lua_init_event(ngx_cycle_t *cycle)
         return NGX_ERROR;
     }
 
+    cycle->connection_n = 128;
+    cycle->connections =
+        ngx_alloc(sizeof(ngx_connection_t) * cycle->connection_n, cycle->log);
+    if (cycle->connections == NULL) {
+        return NGX_ERROR;
+    }
+
+    c = cycle->connections;
+
+    cycle->read_events = ngx_alloc(sizeof(ngx_event_t) * cycle->connection_n,
+                                   cycle->log);
+    if (cycle->read_events == NULL) {
+        return NGX_ERROR;
+    }
+
+    rev = cycle->read_events;
+    for (i = 0; i < cycle->connection_n; i++) {
+        rev[i].closed = 1;
+        rev[i].instance = 1;
+    }
+
+    cycle->write_events = ngx_alloc(sizeof(ngx_event_t) * cycle->connection_n,
+                                    cycle->log);
+    if (cycle->write_events == NULL) {
+        return NGX_ERROR;
+    }
+
+    wev = cycle->write_events;
+    for (i = 0; i < cycle->connection_n; i++) {
+        wev[i].closed = 1;
+    }
+
+    i = cycle->connection_n;
+    next = NULL;
+
+    do {
+        i--;
+
+        c[i].data = next;
+        c[i].read = &cycle->read_events[i];
+        c[i].write = &cycle->write_events[i];
+        c[i].fd = (ngx_socket_t) -1;
+
+        next = &c[i];
+    } while (i);
+
+    cycle->free_connections = next;
+    cycle->free_connection_n = cycle->connection_n;
+
     return ngx_http_lua_event_actions.init_event(cycle);
 }
 

Diff for: src/ngx_http_lua_initworkerby.c

@@ -13,13 +13,8 @@
 #include "ngx_http_lua_initworkerby.h"
 #include "ngx_http_lua_util.h"
 #include "ngx_http_lua_pipe.h"
-#include "ngx_http_lua_event.h"
 
 
-int ngx_http_lua_event_inited = 0;
-
-ngx_http_lua_event_actions_t  ngx_http_lua_event_actions;
-
 static u_char *ngx_http_lua_log_init_worker_error(ngx_log_t *log,
     u_char *buf, size_t len);
 
@@ -51,10 +46,6 @@ ngx_http_lua_init_worker(ngx_cycle_t *cycle)
         return NGX_OK;
     }
 
-    if (ngx_http_lua_init_event(cycle) == NGX_OK) {
-        ngx_http_lua_event_inited = 1;
-    }
-
     /* lmcf != NULL && lmcf->lua != NULL */
 
 #if !(NGX_WIN32)