5 "github.com/yuin/gopher-lua"
12 var lastTime = time.Now()
13 var signalChannel chan os.Signal
15 var serializeVer uint8 = 28
16 var protoVer uint16 = 39
18 //go:embed builtin/luax/init.lua
19 var builtinLuaX string
21 //go:embed builtin/vector.lua
22 var builtinVector string
24 //go:embed builtin/escapes.lua
25 var builtinEscapes string
27 //go:embed builtin/client.lua
28 var builtinClient string
30 //go:embed builtin/base64.lua
31 var builtinBase64 string
33 var builtinFiles = []string{
41 var hydraFuncs = map[string]lua.LGFunction{
49 func l_dtime(l *lua.LState) int {
50 l.Push(lua.LNumber(time.Since(lastTime).Seconds()))
55 func l_poll(l *lua.LState) int {
56 return doPoll(l, getClients(l))
59 func l_close(l *lua.LState) int {
60 for _, client := range getClients(l) {
69 panic("missing filename")
72 signalChannel = make(chan os.Signal, 1)
73 signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM, syscall.SIGHUP)
79 for i, a := range os.Args {
80 l.RawSetInt(arg, i-1, lua.LString(a))
82 l.SetGlobal("arg", arg)
84 hydra := l.SetFuncs(l.NewTable(), hydraFuncs)
85 l.SetField(hydra, "BS", lua.LNumber(10.0))
86 l.SetField(hydra, "serialize_ver", lua.LNumber(serializeVer))
87 l.SetField(hydra, "proto_ver", lua.LNumber(protoVer))
88 l.SetGlobal("hydra", hydra)
90 l.SetField(l.NewTypeMetatable("hydra.client"), "__index", l.NewFunction(l_client_index))
91 l.SetField(l.NewTypeMetatable("hydra.map"), "__index", l.SetFuncs(l.NewTable(), mapFuncs))
93 l.SetField(l.NewTypeMetatable("hydra.comp.auth"), "__index", l.SetFuncs(l.NewTable(), compAuthFuncs))
94 l.SetField(l.NewTypeMetatable("hydra.comp.map"), "__index", l.SetFuncs(l.NewTable(), compMapFuncs))
95 l.SetField(l.NewTypeMetatable("hydra.comp.pkts"), "__index", l.SetFuncs(l.NewTable(), compPktsFuncs))
97 for _, str := range builtinFiles {
98 if err := l.DoString(str); err != nil {
103 if err := l.DoFile(os.Args[1]); err != nil {