1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
package main
import (
_ "embed"
"github.com/yuin/gopher-lua"
"os"
"os/signal"
"runtime/pprof"
"syscall"
"time"
)
var lastTime = time.Now()
var signalChannel chan os.Signal
var serializeVer uint8 = 29
var protoVer uint16 = 40
//go:embed builtin/luax/init.lua
var builtinLuaX string
//go:embed builtin/vector.lua
var builtinVector string
//go:embed builtin/escapes.lua
var builtinEscapes string
//go:embed builtin/client.lua
var builtinClient string
//go:embed builtin/base64.lua
var builtinBase64 string
var builtinFiles = []string{
builtinLuaX,
builtinVector,
builtinEscapes,
builtinClient,
builtinBase64,
}
var hydraFuncs = map[string]lua.LGFunction{
"client": l_client,
"map": l_map,
"dtime": l_dtime,
"poll": l_poll,
"close": l_close,
}
func l_dtime(l *lua.LState) int {
l.Push(lua.LNumber(time.Since(lastTime).Seconds()))
lastTime = time.Now()
return 1
}
func l_poll(l *lua.LState) int {
return doPoll(l, getClients(l))
}
func l_close(l *lua.LState) int {
for _, client := range getClients(l) {
client.closeConn()
}
return 0
}
func main() {
if len(os.Args) < 2 {
panic("missing filename")
}
signalChannel = make(chan os.Signal, 1)
signal.Notify(signalChannel, os.Interrupt, syscall.SIGTERM, syscall.SIGHUP)
go func() {
ch := make(chan os.Signal, 1)
signal.Notify(ch, syscall.SIGUSR1)
for {
<-ch
pprof.Lookup("goroutine").WriteTo(os.Stdout, 1)
}
}()
l := lua.NewState()
defer l.Close()
arg := l.NewTable()
for i, a := range os.Args {
l.RawSetInt(arg, i-1, lua.LString(a))
}
l.SetGlobal("arg", arg)
hydra := l.SetFuncs(l.NewTable(), hydraFuncs)
l.SetField(hydra, "BS", lua.LNumber(10.0))
l.SetField(hydra, "serialize_ver", lua.LNumber(serializeVer))
l.SetField(hydra, "proto_ver", lua.LNumber(protoVer))
l.SetGlobal("hydra", hydra)
l.SetField(l.NewTypeMetatable("hydra.client"), "__index", l.NewFunction(l_client_index))
l.SetField(l.NewTypeMetatable("hydra.map"), "__index", l.SetFuncs(l.NewTable(), mapFuncs))
l.SetField(l.NewTypeMetatable("hydra.comp.auth"), "__index", l.SetFuncs(l.NewTable(), compAuthFuncs))
l.SetField(l.NewTypeMetatable("hydra.comp.map"), "__index", l.SetFuncs(l.NewTable(), compMapFuncs))
l.SetField(l.NewTypeMetatable("hydra.comp.pkts"), "__index", l.SetFuncs(l.NewTable(), compPktsFuncs))
for _, str := range builtinFiles {
if err := l.DoString(str); err != nil {
panic(err)
}
}
if err := l.DoFile(os.Args[1]); err != nil {
panic(err)
}
}
|
