48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"flag"
|
|||
|
|
"fmt"
|
|||
|
|
"os"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 主循环，为每个端口映射启动一个代理
|
|||
|
|
func Loop(config Config) {
|
|||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|||
|
|
|
|||
|
|
// 启动 TCP 代理
|
|||
|
|
for tcp_listenAddr, tcp_targetAddr := range config.Global.LT {
|
|||
|
|
tcp_wg.Add(1)
|
|||
|
|
go StartTcpProxy(tcp_listenAddr, tcp_targetAddr)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 启动 UDP 代理
|
|||
|
|
for udp_listenAddr, udp_targetAddr := range config.Global.LU {
|
|||
|
|
udp_wg.Add(1)
|
|||
|
|
go StartUdpProxy(ctx, udp_listenAddr, udp_targetAddr)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 监听退出信号，触发 `cancel()` 让所有 Goroutine 退出
|
|||
|
|
WaitForExit(cancel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func main() {
|
|||
|
|
daemon := flag.Bool("d", false, "守护进程模式") // 解析命令行参数，是否以守护进程模式运行
|
|||
|
|
config := flag.String("c", "4to6.conf", "指定配置文件") // 解析命令行参数，是否以守护进程模式运行
|
|||
|
|
flag.Parse()
|
|||
|
|
|
|||
|
|
if *daemon {
|
|||
|
|
Daemon() // 如果设置了-d参数，则进入守护进程模式
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
LT, err := parseConfig(*config)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Fprintf(os.Stderr, "读取配置错误: %v\n", err)
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
Loop(LT) // 启动所有代理服务
|
|||
|
|
|
|||
|
|
}
|