-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.go
48 lines (35 loc) · 737 Bytes
/
main.go
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
package main
import (
"flag"
"github.com/lartie/tcp-proxy/proxy"
"github.com/lartie/tcp-proxy/utils"
"sync"
)
var (
env = flag.String("env", "dev", ".env.{environment} file")
)
func init() {
flag.Parse()
utils.LoadEnv(*env)
}
func main() {
ipt := loadIPTable()
listenPorts := loadListenPorts()
wg := sync.WaitGroup{}
wg.Add(1)
p := proxy.NewDefaultTCPProxy(&wg)
for _, port := range listenPorts {
go p.ListenAndProxy(port, ipt)
}
wg.Wait()
}
func loadIPTable() proxy.IPTable {
ipt := make(proxy.IPTable)
utils.FillStructFromConfig("IP_TABLE_CONFIG", &ipt)
return ipt
}
func loadListenPorts() proxy.Ports {
var ports proxy.Ports
utils.FillStructFromConfig("LISTEN_PORTS_CONFIG", &ports)
return ports
}