mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-28 05:46:36 +08:00
ryan
aa4faddade
Result: {"status":"keep","total_issues":8,"golint_canonicalheader":0,"golint_errname":0,"golint_errorlint":1,"golint_exhaustive":0,"golint_forcetypeassert":0,"golint_gosec":0,"golint_intrange":0,"golint_modernize":3,"golint_nilnil":3,"golint_perfsprint":0,"golint_prealloc":0,"golint_recvcheck":1,"golint_usestdlibvars":0,"golint_wastedassign":0,"golint_total":8,"golint_test_testifylint":0,"golint_test_thelper":0,"golint_test_usetesting":0,"golint_test_total":0,"golint_vetx_total":0,"eslint_problems":0,"eslint_errors":0,"eslint_warnings":0,"tsc_errors":0,"vitest_failed":0,"vitest_total":116,"measure_s":81}
95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
// Copyright 2026 Arctel.net
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package nginx
|
|
|
|
import (
|
|
"testing"
|
|
|
|
lua "github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
func TestPowRuntimePassesConfigKeyToInternalChallenge(t *testing.T) {
|
|
state := lua.NewState()
|
|
defer state.Close()
|
|
|
|
if err := state.DoString(`
|
|
package.preload["pow.policy"] = function()
|
|
return {
|
|
match_any = function() return false end,
|
|
has_entries = function() return false end,
|
|
}
|
|
end
|
|
package.preload["cjson.safe"] = function()
|
|
return { encode = function() return "{}" end }
|
|
end
|
|
|
|
local config_values = {}
|
|
local config_dict = {}
|
|
function config_dict:set(key, value) config_values[key] = value return true end
|
|
function config_dict:get(key) return config_values[key] end
|
|
|
|
local sessions = {}
|
|
function sessions:get() return nil end
|
|
function sessions:set() return true end
|
|
|
|
ngx = {
|
|
var = {
|
|
host = "pow.example.com",
|
|
uri = "/protected",
|
|
scheme = "https",
|
|
remote_addr = "192.0.2.1",
|
|
http_user_agent = "test",
|
|
request_id = "request-1",
|
|
},
|
|
ctx = {},
|
|
header = {},
|
|
shared = {
|
|
openflare_pow_sessions = sessions,
|
|
openflare_pow_config = config_dict,
|
|
},
|
|
req = {},
|
|
now = function() return 1 end,
|
|
}
|
|
function ngx.req.set_uri_args(args) captured_uri_args = args end
|
|
function ngx.exec(uri, args)
|
|
captured_exec_uri = uri
|
|
captured_exec_args = args
|
|
end
|
|
`); err != nil {
|
|
t.Fatalf("prepare Lua runtime: %v", err)
|
|
}
|
|
|
|
chunk, err := state.LoadString(openRestyPowRuntimeLua)
|
|
if err != nil {
|
|
t.Fatalf("load PoW runtime: %v", err)
|
|
}
|
|
if err := state.CallByParam(lua.P{Fn: chunk, NRet: 1, Protect: true}); err != nil {
|
|
t.Fatalf("initialize PoW runtime: %v", err)
|
|
}
|
|
runtimeModule := state.Get(-1)
|
|
state.Pop(1)
|
|
|
|
evaluate := state.GetField(runtimeModule, "evaluate")
|
|
config := state.NewTable()
|
|
config.RawSetString("challenge_ttl", lua.LNumber(300))
|
|
if err := state.CallByParam(lua.P{Fn: evaluate, NRet: 1, Protect: true}, config); err != nil {
|
|
t.Fatalf("evaluate PoW node: %v", err)
|
|
}
|
|
state.Pop(1)
|
|
|
|
if got := state.GetGlobal("captured_exec_uri").String(); got != "/.within.website/x/cmd/anubis/api/make-challenge" {
|
|
t.Fatalf("unexpected internal challenge URI: %q", got)
|
|
}
|
|
execArgs, ok := state.GetGlobal("captured_exec_args").(*lua.LTable)
|
|
if !ok {
|
|
t.Fatal("expected ngx.exec to receive explicit challenge arguments")
|
|
}
|
|
if got := execArgs.RawGetString("openflare_pow_config_key").String(); got != "_request_config:request-1" {
|
|
t.Fatalf("unexpected PoW config key: %q", got)
|
|
}
|
|
if state.GetGlobal("captured_uri_args") != execArgs {
|
|
t.Fatal("expected URI arguments and internal redirect arguments to use the same table")
|
|
}
|
|
}
|