mirror of
https://github.com/XTLS/Xray-core.git
synced 2026-09-30 21:16:03 +00:00
93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package log
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
lua "github.com/yuin/gopher-lua"
|
|
)
|
|
|
|
type luaLogHandler struct {
|
|
messages []Message
|
|
}
|
|
|
|
func (h *luaLogHandler) Handle(msg Message) {
|
|
h.messages = append(h.messages, msg)
|
|
}
|
|
|
|
func TestLuaLog(t *testing.T) {
|
|
logHandler.RLock()
|
|
previous := logHandler.Handler
|
|
logHandler.RUnlock()
|
|
t.Cleanup(func() { RegisterHandler(previous) })
|
|
handler := &luaLogHandler{}
|
|
RegisterHandler(handler)
|
|
|
|
L := lua.NewState()
|
|
defer L.Close()
|
|
RegisterLua(L)
|
|
nativeError := L.NewUserData()
|
|
nativeError.Value = fmt.Errorf("lookup failed: %w", errors.New("upstream timeout"))
|
|
L.SetGlobal("nativeError", nativeError)
|
|
path := filepath.Join(t.TempDir(), "logging.lua")
|
|
if err := os.WriteFile(path, []byte(`
|
|
local log = require("xray.log")
|
|
assert(log == require("xray.log"))
|
|
log.Debug("query: ", "example.com")
|
|
log.Info("count=", 42, ", enabled=", true, ", value=", nil)
|
|
log.Warning(setmetatable({}, {
|
|
__tostring = function() return "fallback" end
|
|
}))
|
|
assert(select("#", log.Error("failed")) == 0)
|
|
log.Error("DNS failed: ", nativeError)
|
|
log.Warning(nativeError)
|
|
local ok, err = pcall(function() error("Lua failure", 0) end)
|
|
assert(not ok)
|
|
log.Error(err)
|
|
function logHook()
|
|
log.Info("hook")
|
|
end
|
|
`), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := L.DoFile(path); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := L.DoString(`
|
|
logHook()
|
|
require("xray.log").Info("anonymous")
|
|
`); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
want := []struct {
|
|
severity Severity
|
|
message string
|
|
}{
|
|
{Severity_Debug, "[Debug] logging.lua: query: example.com"},
|
|
{Severity_Info, "[Info] logging.lua: count=42, enabled=true, value=nil"},
|
|
{Severity_Warning, "[Warning] logging.lua: fallback"},
|
|
{Severity_Error, "[Error] logging.lua: failed"},
|
|
{Severity_Error, "[Error] logging.lua: DNS failed: lookup failed: upstream timeout"},
|
|
{Severity_Warning, "[Warning] logging.lua: lookup failed: upstream timeout"},
|
|
{Severity_Error, "[Error] logging.lua: Lua failure"},
|
|
{Severity_Info, "[Info] logging.lua: hook"},
|
|
{Severity_Info, "[Info] <string>: anonymous"},
|
|
}
|
|
if len(handler.messages) != len(want) {
|
|
t.Fatalf("logged %d messages, want %d", len(handler.messages), len(want))
|
|
}
|
|
for i, expected := range want {
|
|
msg, ok := handler.messages[i].(*GeneralMessage)
|
|
if !ok {
|
|
t.Fatalf("message %d has type %T, want *GeneralMessage", i, handler.messages[i])
|
|
}
|
|
if msg.Severity != expected.severity || msg.String() != expected.message {
|
|
t.Errorf("message %d = %q with severity %v, want %q with severity %v", i, msg.String(), msg.Severity, expected.message, expected.severity)
|
|
}
|
|
}
|
|
}
|