116 lines
3.0 KiB
Lua
116 lines
3.0 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 type = type
|
|
local error = error
|
|
local tostring = tostring
|
|
local get_string_buf = base.get_string_buf
|
|
local subsystem = ngx.config.subsystem
|
|
|
|
|
|
local ngx_lua_ffi_escape_uri
|
|
local ngx_lua_ffi_unescape_uri
|
|
local ngx_lua_ffi_uri_escaped_length
|
|
|
|
local NGX_ESCAPE_URI = 0
|
|
local NGX_ESCAPE_URI_COMPONENT = 2
|
|
local NGX_ESCAPE_MAIL_AUTH = 6
|
|
|
|
|
|
if subsystem == "http" then
|
|
ffi.cdef[[
|
|
size_t ngx_http_lua_ffi_uri_escaped_length(const unsigned char *src,
|
|
size_t len, int type);
|
|
|
|
void ngx_http_lua_ffi_escape_uri(const unsigned char *src, size_t len,
|
|
unsigned char *dst, int type);
|
|
|
|
size_t ngx_http_lua_ffi_unescape_uri(const unsigned char *src,
|
|
size_t len, unsigned char *dst);
|
|
]]
|
|
|
|
ngx_lua_ffi_escape_uri = C.ngx_http_lua_ffi_escape_uri
|
|
ngx_lua_ffi_unescape_uri = C.ngx_http_lua_ffi_unescape_uri
|
|
ngx_lua_ffi_uri_escaped_length = C.ngx_http_lua_ffi_uri_escaped_length
|
|
|
|
elseif subsystem == "stream" then
|
|
ffi.cdef[[
|
|
size_t ngx_stream_lua_ffi_uri_escaped_length(const unsigned char *src,
|
|
size_t len, int type);
|
|
|
|
void ngx_stream_lua_ffi_escape_uri(const unsigned char *src, size_t len,
|
|
unsigned char *dst, int type);
|
|
|
|
size_t ngx_stream_lua_ffi_unescape_uri(const unsigned char *src,
|
|
size_t len, unsigned char *dst);
|
|
]]
|
|
|
|
ngx_lua_ffi_escape_uri = C.ngx_stream_lua_ffi_escape_uri
|
|
ngx_lua_ffi_unescape_uri = C.ngx_stream_lua_ffi_unescape_uri
|
|
ngx_lua_ffi_uri_escaped_length = C.ngx_stream_lua_ffi_uri_escaped_length
|
|
end
|
|
|
|
|
|
ngx.escape_uri = function (s, esc_type)
|
|
if type(s) ~= 'string' then
|
|
if not s then
|
|
s = ''
|
|
|
|
else
|
|
s = tostring(s)
|
|
end
|
|
end
|
|
|
|
if esc_type == nil then
|
|
esc_type = NGX_ESCAPE_URI_COMPONENT
|
|
|
|
else
|
|
if type(esc_type) ~= 'number' then
|
|
error("\"type\" is not a number", 3)
|
|
end
|
|
|
|
if esc_type < NGX_ESCAPE_URI or esc_type > NGX_ESCAPE_MAIL_AUTH then
|
|
error("\"type\" " .. esc_type .. " out of range", 3)
|
|
end
|
|
end
|
|
|
|
local slen = #s
|
|
local dlen = ngx_lua_ffi_uri_escaped_length(s, slen, esc_type)
|
|
|
|
-- print("dlen: ", tonumber(dlen))
|
|
if dlen == slen then
|
|
return s
|
|
end
|
|
local dst = get_string_buf(dlen)
|
|
ngx_lua_ffi_escape_uri(s, slen, dst, esc_type)
|
|
return ffi_string(dst, dlen)
|
|
end
|
|
|
|
|
|
ngx.unescape_uri = function (s)
|
|
if type(s) ~= 'string' then
|
|
if not s then
|
|
s = ''
|
|
else
|
|
s = tostring(s)
|
|
end
|
|
end
|
|
local slen = #s
|
|
local dlen = slen
|
|
local dst = get_string_buf(dlen)
|
|
dlen = ngx_lua_ffi_unescape_uri(s, slen, dst)
|
|
return ffi_string(dst, dlen)
|
|
end
|
|
|
|
|
|
return {
|
|
version = base.version,
|
|
}
|