refactor(dns): make server-specific Lua registration private

This commit is contained in:
Meo597
2026-10-04 05:41:34 +08:00
parent 399563b6d9
commit 745526f14c
3 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -37,8 +37,8 @@ func RegisterLua(L *lua.LState, client featureDNS.Client) {
registerLua(L, servers, client)
}
// RegisterLua makes xray.dns available to DNS scripts.
func (s *DNS) RegisterLua(L *lua.LState) {
// registerLua makes xray.dns available to DNS scripts.
func (s *DNS) registerLua(L *lua.LState) {
registerLua(L, luaServers(s), nil)
}
+2 -2
View File
@@ -154,7 +154,7 @@ func TestLuaDNSServerQuery(t *testing.T) {
option := featureDNS.IPOption{IPv4Enable: true}
ips := []net.IP{net.ParseIP("127.0.0.1"), net.ParseIP("8.8.8.8")}
server := &DNS{clients: []*Client{{server: &benchmarkLuaNameServer{ips: ips}, ipOption: &option, timeoutMs: time.Second}}}
server.RegisterLua(L)
server.registerLua(L)
if err := L.DoString(`
local server = require("xray.dns").Servers[1]
local matcher = require("xray.geodata").BuildIPMatcher("127.0.0.0/8")
@@ -254,7 +254,7 @@ func BenchmarkLuaDNSHookCall(b *testing.B) {
server := &DNS{clients: []*Client{client}}
L := lua.NewState()
defer L.Close()
server.RegisterLua(L)
server.registerLua(L)
if err := L.DoString(`
local server = require("xray.dns").Servers[1]
function HandleDNSQuery(domain, ipv4, ipv6, fake)
+1 -1
View File
@@ -30,7 +30,7 @@ func newScriptEngine(path string, server *DNS) (*scriptEngine, error) {
func(L *lua.LState) {
geodata.RegisterLua(L)
log.RegisterLua(L)
server.RegisterLua(L)
server.registerLua(L)
},
func(L *lua.LState) error {
if L.GetGlobal("HandleDNSQuery").Type() != lua.LTFunction {