feature: allowed some ngx APIs to be used in worker thread.

Author: kingluo

README.markdown

@@ -9124,7 +9124,35 @@ ngx.run_worker_thread
 
 Wrap the [nginx worker thread](http://nginx.org/en/docs/dev/development_guide.html#threads) to execute lua function. The caller coroutine would yield until the function returns.
 
-Note that no ngx_lua API can be used in the `function_name` function of the `module` module since it is invoked in a separate thread.
+Only the following ngx_lua APIs could be used in `function_name` function of the `module` module:
+
+* `ngx.encode_base64`
+* `ngx.decode_base64`
+
+* `ngx.hmac_sha1`
+* `ngx.encode_args`
+* `ngx.decode_args`
+* `ngx.quote_sql_str`
+
+* `ngx.re.match`
+* `ngx.re.find`
+* `ngx.re.gmatch`
+* `ngx.re.sub`
+* `ngx.re.gsub`
+
+* `ngx.crc32_short`
+* `ngx.crc32_long`
+* `ngx.hmac_sha1`
+* `ngx.md5_bin`
+* `ngx.md5`
+
+* `ngx.config.subsystem`
+* `ngx.config.debug`
+* `ngx.config.prefix`
+* `ngx.config.nginx_version`
+* `ngx.config.nginx_configure`
+* `ngx.config.ngx_lua_version`
+
 
 The first argument `threadpool` specifies the Nginx thread pool name defined by [thread_pool](https://nginx.org/en/docs/ngx_core_module.html#thread_pool).
 

doc/HttpLuaModule.wiki

@@ -7820,7 +7820,35 @@ This API was first enabled in the <code>v0.6.0</code> release.
 
 Wrap the [http://nginx.org/en/docs/dev/development_guide.html#threads nginx worker thread] to execute lua function. The caller coroutine would yield until the function returns.
 
-Note that no ngx_lua API can be used in the `function_name` function of the `module` module since it is invoked in a separate thread.
+Only the following ngx_lua APIs could be used in `function_name` function of the `module` module:
+
+* `ngx.encode_base64`
+* `ngx.decode_base64`
+
+* `ngx.hmac_sha1`
+* `ngx.encode_args`
+* `ngx.decode_args`
+* `ngx.quote_sql_str`
+
+* `ngx.re.match`
+* `ngx.re.find`
+* `ngx.re.gmatch`
+* `ngx.re.sub`
+* `ngx.re.gsub`
+
+* `ngx.crc32_short`
+* `ngx.crc32_long`
+* `ngx.hmac_sha1`
+* `ngx.md5_bin`
+* `ngx.md5`
+
+* `ngx.config.subsystem`
+* `ngx.config.debug`
+* `ngx.config.prefix`
+* `ngx.config.nginx_version`
+* `ngx.config.nginx_configure`
+* `ngx.config.ngx_lua_version`
+
 
 The first argument `threadpool` specifies the Nginx thread pool name defined by [thread_pool](https://nginx.org/en/docs/ngx_core_module.html#thread_pool).
 

src/ngx_http_lua_worker_thread.c

@@ -14,6 +14,8 @@
 
 #include "ngx_http_lua_worker_thread.h"
 #include "ngx_http_lua_util.h"
+#include "ngx_http_lua_string.h"
+#include "ngx_http_lua_config.h"
 
 
 #if (NGX_THREADS)
@@ -127,8 +129,36 @@ ngx_http_lua_get_task_ctx(lua_State *L, ngx_http_request_t *r)
         /* pop path, cpath and "package" table from L */
         lua_pop(L, 3);
 
-        /* pop the "package" table */
-        lua_pop(vm, 1);
+        /* inject API from C */
+        lua_newtable(L);    /* ngx.* */
+        ngx_http_lua_inject_string_api(vm);
+        ngx_http_lua_inject_config_api(vm);
+        lua_setglobal(vm, "ngx");
+
+        /* inject API via ffi */
+        lua_getglobal(vm, "require");
+        lua_pushstring(vm, "resty.core.regex");
+        if (lua_pcall(vm, 1, 0, 0) != 0) {
+            lua_close(vm);
+            ngx_free(ctx);
+            return NULL;
+        }
+
+        lua_getglobal(vm, "require");
+        lua_pushstring(vm, "resty.core.hash");
+        if (lua_pcall(vm, 1, 0, 0) != 0) {
+            lua_close(vm);
+            ngx_free(ctx);
+            return NULL;
+        }
+
+        lua_getglobal(vm, "require");
+        lua_pushstring(vm, "resty.core.base64");
+        if (lua_pcall(vm, 1, 0, 0) != 0) {
+            lua_close(vm);
+            ngx_free(ctx);
+            return NULL;
+        }
 
     } else {
         ctx = ctxpool->next;