67 lines
1.4 KiB
Lua
67 lines
1.4 KiB
Lua
-- 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
|
|
}
|