EVOLUTION-NINJA

Edit File: proxysslverify.lua

-- Copyright (C) Yichun Zhang (agentzh)

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

local ffi = require "ffi"
local C = ffi.C
local ffi_gc = ffi.gc
local ffi_str = ffi.string
local get_request = base.get_request
local error = error
local errmsg = base.get_errmsg_ptr()

local FFI_OK = base.FFI_OK
local FFI_ERROR = base.FFI_ERROR
local subsystem = ngx.config.subsystem
local ngx_phase = ngx.get_phase

local ngx_lua_ffi_proxy_ssl_set_verify_result
local ngx_lua_ffi_proxy_ssl_get_verify_result
local ngx_lua_ffi_proxy_ssl_get_verify_cert
local ngx_lua_ffi_proxy_ssl_free_verify_cert

if subsystem == 'http' then
    ffi.cdef[[
    int ngx_http_lua_ffi_proxy_ssl_set_verify_result(ngx_http_request_t *r,
        int verify_result, char **err);

int ngx_http_lua_ffi_proxy_ssl_get_verify_result(ngx_http_request_t *r,
        char **err);

void *ngx_http_lua_ffi_proxy_ssl_get_verify_cert(ngx_http_request_t *r,
        char **err);

void ngx_http_lua_ffi_proxy_ssl_free_verify_cert(void *cdata);
    ]]

ngx_lua_ffi_proxy_ssl_set_verify_result =
        C.ngx_http_lua_ffi_proxy_ssl_set_verify_result
    ngx_lua_ffi_proxy_ssl_get_verify_result =
        C.ngx_http_lua_ffi_proxy_ssl_get_verify_result
    ngx_lua_ffi_proxy_ssl_get_verify_cert =
        C.ngx_http_lua_ffi_proxy_ssl_get_verify_cert
    ngx_lua_ffi_proxy_ssl_free_verify_cert =
        C.ngx_http_lua_ffi_proxy_ssl_free_verify_cert

elseif subsystem == 'stream' then
    ffi.cdef[[
    int ngx_stream_lua_ffi_proxy_ssl_set_verify_result(
        ngx_stream_lua_request_t *r, int verify_result, char **err);

int ngx_stream_lua_ffi_proxy_ssl_get_verify_result(
        ngx_stream_lua_request_t *r, char **err);

void *ngx_stream_lua_ffi_proxy_ssl_get_verify_cert(
        ngx_stream_lua_request_t *r, char **err);

void ngx_stream_lua_ffi_proxy_ssl_free_verify_cert(void *cdata);
    ]]

ngx_lua_ffi_proxy_ssl_set_verify_result =
        C.ngx_stream_lua_ffi_proxy_ssl_set_verify_result
    ngx_lua_ffi_proxy_ssl_get_verify_result =
        C.ngx_stream_lua_ffi_proxy_ssl_get_verify_result
    ngx_lua_ffi_proxy_ssl_get_verify_cert =
        C.ngx_stream_lua_ffi_proxy_ssl_get_verify_cert
    ngx_lua_ffi_proxy_ssl_free_verify_cert =
        C.ngx_stream_lua_ffi_proxy_ssl_free_verify_cert
end

local _M = { version = base.version }

-- return ok, err
function _M.set_verify_result(verify_result)
    local r = get_request()
    if not r then
        error("no request found")
    end

if ngx_phase() ~= "proxy_ssl_verify" then
        error("API disabled in the current context")
    end

local rc = ngx_lua_ffi_proxy_ssl_set_verify_result(r, verify_result, errmsg)
    if rc == FFI_OK then
        return true
    end

return nil, ffi_str(errmsg[0])
end

-- return verify_result, err
function _M.get_verify_result()
    local r = get_request()
    if not r then
        error("no request found")
    end

if ngx_phase() ~= "proxy_ssl_verify" then
        error("API disabled in the current context")
    end

local rc = ngx_lua_ffi_proxy_ssl_get_verify_result(r, errmsg)
    if rc == FFI_ERROR then
        return nil, ffi_str(errmsg[0])
    end

return rc
end

-- return cert, err
function _M.get_verify_cert()
    local r = get_request()
    if not r then
        error("no request found")
    end

if ngx_phase() ~= "proxy_ssl_verify" then
        error("API disabled in the current context")
    end

local cert = ngx_lua_ffi_proxy_ssl_get_verify_cert(r, errmsg)
    if cert ~= nil then
        return ffi_gc(cert, ngx_lua_ffi_proxy_ssl_free_verify_cert)
    end

return nil, ffi_str(errmsg[0])
end

return _M