1#@!#!123s

: / opt / imunify360-webshield / lualib / resty / core /
Filename : ndk.lua
back
-- Copyright (C) by OpenResty Inc.


local ffi = require 'ffi'
local base = require "resty.core.base"
base.allows_subsystem('http')


local C = ffi.C
local ffi_cast = ffi.cast
local ffi_new = ffi.new
local ffi_str = ffi.string
local FFI_OK = base.FFI_OK
local new_tab = base.new_tab
local get_string_buf = base.get_string_buf
local get_request = base.get_request
local setmetatable = setmetatable
local type = type
local tostring = tostring
local error = error


local _M = {
    version = base.version
}


ffi.cdef[[
typedef void * ndk_set_var_value_pt;

int ngx_http_lua_ffi_ndk_lookup_directive(const unsigned char *var_data,
    size_t var_len, ndk_set_var_value_pt *func);
int ngx_http_lua_ffi_ndk_set_var_get(ngx_http_request_t *r,
    ndk_set_var_value_pt func, const unsigned char *arg_data, size_t arg_len,
    ngx_http_lua_ffi_str_t *value);
]]


local func_p = ffi_new("void*[1]")
local ffi_str_size = ffi.sizeof("ngx_http_lua_ffi_str_t")
local ffi_str_type = ffi.typeof("ngx_http_lua_ffi_str_t*")


local function ndk_set_var_get(self, var)
    if type(var) ~= "string" then
        var = tostring(var)
    end

    if C.ngx_http_lua_ffi_ndk_lookup_directive(var, #var, func_p) ~= FFI_OK then
        error('ndk.set_var: directive "' .. var
              .. '" not found or does not use ndk_set_var_value', 2)
    end

    local func = func_p[0]

    return function (arg)
        local r = get_request()
        if not r then
            error("no request found")
        end

        if type(arg) ~= "string" then
            arg = tostring(arg)
        end

        local buf = get_string_buf(ffi_str_size)
        local value = ffi_cast(ffi_str_type, buf)
        local rc = C.ngx_http_lua_ffi_ndk_set_var_get(r, func, arg, #arg, value)
        if rc ~= FFI_OK then
            error("calling directive " .. var .. " failed with code " .. rc, 2)
        end

        return ffi_str(value.data, value.len)
    end
end


local function ndk_set_var_set()
    error("not allowed", 2)
end


if ndk then
    local mt = new_tab(0, 2)
    mt.__newindex = ndk_set_var_set
    mt.__index = ndk_set_var_get

    ndk.set_var = setmetatable(new_tab(0, 0), mt)
end


return _M