增加sql的waf功能

resty/core/exit.lua

@@ -0,0 +1,66 @@
+-- Copyright (C) Yichun Zhang (agentzh)
+
+
+local ffi = require "ffi"
+local base = require "resty.core.base"
+
+
+local C = ffi.C
+local ffi_string = ffi.string
+local ngx = ngx
+local error = error
+local get_string_buf = base.get_string_buf
+local get_size_ptr = base.get_size_ptr
+local get_request = base.get_request
+local co_yield = coroutine._yield
+local subsystem = ngx.config.subsystem
+
+
+local ngx_lua_ffi_exit
+
+
+if subsystem == "http" then
+    ffi.cdef[[
+    int ngx_http_lua_ffi_exit(ngx_http_request_t *r, int status,
+                               unsigned char *err, size_t *errlen);
+    ]]
+
+    ngx_lua_ffi_exit = C.ngx_http_lua_ffi_exit
+
+elseif subsystem == "stream" then
+    ffi.cdef[[
+    int ngx_stream_lua_ffi_exit(ngx_stream_lua_request_t *r, int status,
+                                unsigned char *err, size_t *errlen);
+    ]]
+
+    ngx_lua_ffi_exit = C.ngx_stream_lua_ffi_exit
+end
+
+
+local ERR_BUF_SIZE = 128
+local FFI_DONE = base.FFI_DONE
+
+
+ngx.exit = function (rc)
+    local err = get_string_buf(ERR_BUF_SIZE)
+    local errlen = get_size_ptr()
+    local r = get_request()
+    if r == nil then
+        error("no request found")
+    end
+    errlen[0] = ERR_BUF_SIZE
+    rc = ngx_lua_ffi_exit(r, rc, err, errlen)
+    if rc == 0 then
+        -- print("yielding...")
+        return co_yield()
+    end
+    if rc == FFI_DONE then
+        return
+    end
+    error(ffi_string(err, errlen[0]), 2)
+end
+
+
+return {
+    version = base.version
+}