GoTunnel/cmd/client/main.go

package main

import (
	"flag"
	"log"
	"os"
	"path/filepath"
	"time"

	"github.com/gotunnel/internal/client/tunnel"
	"github.com/gotunnel/pkg/crypto"
	"github.com/gotunnel/pkg/plugin"
	"github.com/gotunnel/pkg/plugin/builtin"
	"github.com/gotunnel/pkg/plugin/sign"
)

func main() {
	server := flag.String("s", "", "server address (ip:port)")
	token := flag.String("t", "", "auth token")
	id := flag.String("id", "", "client id (optional, auto-assigned if empty)")
	noTLS := flag.Bool("no-tls", false, "disable TLS")
	skipVerify := flag.Bool("skip-verify", false, "skip TLS certificate verification (insecure)")
	flag.Parse()

	if *server == "" || *token == "" {
		log.Fatal("Usage: client -s <server:port> -t <token> [-id <client_id>] [-no-tls] [-skip-verify]")
	}

	client := tunnel.NewClient(*server, *token, *id)

	// TLS 默认启用，使用 TOFU 验证
	if !*noTLS {
		client.TLSEnabled = true
		// 获取数据目录
		home, _ := os.UserHomeDir()
		dataDir := filepath.Join(home, ".gotunnel")
		client.TLSConfig = crypto.ClientTLSConfigWithTOFU(*server, dataDir, *skipVerify)
		if *skipVerify {
			log.Printf("[Client] TLS enabled (certificate verification DISABLED - insecure)")
		} else {
			log.Printf("[Client] TLS enabled with TOFU certificate verification")
		}
	}

	// 初始化安全配置
	initSecurityConfig()

	// 初始化插件系统
	registry := plugin.NewRegistry()
	for _, h := range builtin.GetClientPlugins() {
		if err := registry.RegisterClient(h); err != nil {
			log.Fatalf("[Plugin] Register error: %v", err)
		}
	}
	client.SetPluginRegistry(registry)
	log.Printf("[Plugin] Registered %d plugins", len(builtin.GetClientPlugins()))

	client.Run()
}

// 官方安全配置 URL（与服务端保持一致）
const (
	officialRevocationURL = "https://git.92coco.cn:8443/flik/GoTunnel-Plugins/raw/branch/main/security/revocation.json"
	officialKeyListURL    = "https://git.92coco.cn:8443/flik/GoTunnel-Plugins/raw/branch/main/security/keys.json"
)

// initSecurityConfig 初始化安全配置
func initSecurityConfig() {
	// 配置撤销列表
	sign.SetRevocationConfig(sign.RevocationConfig{
		RemoteURL:       officialRevocationURL,
		FetchInterval:   1 * time.Hour,
		RequestTimeout:  10 * time.Second,
		VerifySignature: true,
	})

	// 配置公钥列表
	sign.SetKeyListConfig(sign.KeyListConfig{
		RemoteURL:      officialKeyListURL,
		FetchInterval:  24 * time.Hour,
		RequestTimeout: 10 * time.Second,
	})

	// 启动后台刷新
	stopCh := make(chan struct{})
	go sign.StartRevocationRefresher(stopCh)

	// 立即拉取一次
	if err := sign.FetchRemoteKeyList(); err != nil {
		log.Printf("[Security] Fetch key list failed: %v", err)
	}

	log.Printf("[Security] Initialized")
}