flvx/go-gost/x/internal/util/port/port.go

package port

import (
	"fmt"
	"os/exec"
	"strconv"
	"strings"
	"time"
)

func ForceClosePortConnections(addr string) (err error) {
	defer func() {
		if r := recover(); r != nil {
			fmt.Printf("⚠️ ForceClosePortConnections panic recovered: %v\n", r)
			err = nil // 永远返回 nil
		}
	}()

	if addr == "" {
		fmt.Println("⚠️ 地址为空")
		return nil
	}

	_, portStr, err := splitHostPort(addr)
	if err != nil {
		fmt.Printf("⚠️ 地址解析失败: %v\n", err)
		return nil
	}

	port, err := strconv.Atoi(portStr)
	if err != nil {
		fmt.Printf("⚠️ 端口非法: %v\n", err)
		return nil
	}

	cmd := exec.Command("tcpkill", "-i", "any", "port", fmt.Sprintf("%d", port))
	if err := cmd.Start(); err != nil {
		fmt.Printf("⚠️ 启动 tcpkill 失败: %v\n", err)
		return nil
	}

	go func() {
		defer func() {
			if r := recover(); r != nil {
				fmt.Printf("⚠️ tcpkill goroutine panic recovered: %v\n", r)
			}
		}()
		time.Sleep(2 * time.Second)
		if cmd.Process != nil {
			if err := cmd.Process.Kill(); err != nil {
				fmt.Printf("⚠️ 终止 tcpkill 失败: %v\n", err)
			}
		}
	}()

	fmt.Printf("✅ 正在断开端口 %d 上的所有连接...\n", port)
	return nil
}

func splitHostPort(addr string) (string, string, error) {
	if !strings.Contains(addr, ":") {
		return "", "", fmt.Errorf("missing port in address")
	}
	return netSplitHostPortCompat(addr)
}

func netSplitHostPortCompat(addr string) (string, string, error) {
	lastColon := strings.LastIndex(addr, ":")
	if lastColon < 0 {
		return "", "", fmt.Errorf("missing port")
	}
	return addr[:lastColon], addr[lastColon+1:], nil
}