diff --git a/app/dns/lua.go b/app/dns/lua.go index deb96b394..f74f197d1 100644 --- a/app/dns/lua.go +++ b/app/dns/lua.go @@ -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) } diff --git a/app/dns/lua_test.go b/app/dns/lua_test.go index 0b59dd4e4..0648c0305 100644 --- a/app/dns/lua_test.go +++ b/app/dns/lua_test.go @@ -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) diff --git a/app/dns/script.go b/app/dns/script.go index 6391a1fd9..a8e308612 100644 --- a/app/dns/script.go +++ b/app/dns/script.go @@ -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 {