使用nginx + lua脚本 + redis进行token鉴权

时间:2024-04-09 19:41:14

1.问题描述:

最近老大交给我一个任务,使用nginx + lua脚本 + redis 来对从客户端发来的下载请求进行token的鉴权。该下载请求为一条带有token信息的URL,假设URL下载请求所带的token也会同步存入redis中,现在是为了防止别人伪造URL请求,需要验证redis中是否存在该token。

2.整体思路:

  • 1.首先从客户端传过来一个带token的下载请求。
  • 2.nginx提供代理服务,利用location的规则来拦截发来的下载请求。
  • 3.匹配好的下载请求交由lua脚本处理。
  • 4.在lua脚本连接redis服务器,验证token是否存在。
  • 5.token存在则验证通过,交由nginx服务器进行反向代理服务;否则返回报错信息。

3.下面是token鉴权的时序图

使用nginx + lua脚本 + redis进行token鉴权

4.相关代码

4.1 nginx.conf中的代理集群
 upstream download{
    server  127.0.0.1:8303;
 }
4.2 location的拦截
下载请求:http://localhost:8082/download/files/04a9b52ec74d46829b4b2296fcadb99a/data?token=b17efb43-292e-4cc9-ac5d-0b46bce059c5
#匹配下载请求前缀, 进行token鉴权
location ~ /download/files/.*?/data {
    default_type 'text/html';
    rewrite_by_lua_file  lua-resty-redis/lib/resty/token.lua;
    proxy_pass  http://download;
 }
4.3 进行鉴权的脚本token.lua
local var = ngx.var
local arg = ngx.req.get_uri_args()
local cjson = require("cjson")
--在table的agr中获取token, token字符串组合成的key值
if arg ~= nil and arg.token ~= nil and arg.token ~= '' then
  token = "download:files:" .. arg.token 
end

--连接redis
local redis = require "resty.redis"
local red = redis:new()

red:set_timeout(1000) -- 1 sec

local ok, err = red:connect("127.0.0.1", 6379)
if not ok then
    ngx.say("failed to connect: ", err, "<br/>")
    return
end

-- 请注意这里 auth 的调用过程 这是redis设置密码的
local count
count, err = red:get_reused_times()
if 0 == count then
    ok, err = red:auth(password)
    if not ok then
         ngx.say(cjson.encode({code = 500,message = err}))
        return
    end
elseif err then
    	ngx.say("failed to get reused times: ", err, "<br/>")
    return
end

--redis中若 key 存在返回 1 ,否则返回 0 。
local res, errs = red:exists(token)

if res == 0 then
	local loginfailobj = {code = 500,message = "token is wrong"}
    local loginfailjson = cjson.encode(loginfailobj)
    ngx.say(loginfailjson)
end

在浏览器地址栏中输入模拟的URL请求:http://localhost:8082/download/files/04a9b52ec74d46829b4b2296fcadb99a/data?token=b17efb43-292e-4cc9-ac5d-0b46bce059c5
至此,大功告成。