init

main.go

@@ -0,0 +1,60 @@
+// 主程序入口
+package main
+
+import (
+	"context"
+	"flag"
+	"fmt"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"s5/pkg/config"
+	"s5/pkg/logger"
+	"s5/pkg/proxy"
+)
+
+func main() {
+	// 解析命令行参数
+	configFile := flag.String("config", "config.yaml", "配置文件路径")
+	flag.Parse()
+
+	// 加载配置文件
+	cfg, err := config.Load(*configFile)
+	if err != nil {
+		panic(err)
+	}
+
+	// 初始化日志系统
+	if err := logger.Init(cfg.Log); err != nil {
+		panic(err)
+	}
+	defer logger.Sync()
+
+	log := logger.Get()
+	log.Info("代理服务器启动中...")
+
+	// 创建并配置代理服务器
+	server := proxy.NewServer(proxy.ServerConfig{
+		Socks5Addr: fmt.Sprintf("%s:%d", cfg.Server.BindAddress, cfg.Server.Socks5Port),
+		HttpAddr:   fmt.Sprintf("%s:%d", cfg.Server.BindAddress, cfg.Server.HttpPort),
+		BufferSize: cfg.Proxy.BufferSize,
+	}, cfg.Proxy.Upstream)
+
+	// 创建上下文和取消函数
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	// 处理系统信号
+	sigCh := make(chan os.Signal, 1)
+	signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
+	go func() {
+		<-sigCh
+		cancel()
+	}()
+
+	// 启动服务器
+	if err := server.Start(ctx); err != nil {
+		log.Fatalf("服务器启动失败: %v", err)
+	}
+}