DTLCP 实战：基于 gotlcp 实现 GM/T 0128-2023 数据报传输层密码协议

package main

import (
	"fmt"
	"net"
	"os"
	"time"

	"github.com/Trisia/gotlcp"
)

func main() {
	// 加载双证书
	signCert, err := gotlcp.LoadX509KeyPair("sign.crt", "sign.key")
	if err != nil {
		fmt.Printf("加载签名证书失败: %v\n", err)
		os.Exit(1)
	}
	encCert, err := gotlcp.LoadX509KeyPair("enc.crt", "enc.key")
	if err != nil {
		fmt.Printf("加载加密证书失败: %v\n", err)
		os.Exit(1)
	}

	// DTLCP 配置
	config := &gotlcp.Config{
		Certificates: []gotlcp.Certificate{
			signCert, // 签名证书在前
			encCert,  // 加密证书在后
		},
		PMTU: 1400, // 默认路径最大传输单元
	}

	// 监听 UDP
	addr, _ := net.ResolveUDPAddr("udp", ":8443")
	conn, err := net.ListenUDP("udp", addr)
	if err != nil {
		fmt.Printf("监听失败: %v\n", err)
		os.Exit(1)
	}
	defer conn.Close()

	fmt.Println("DTLCP 服务端已启动，监听 :8443")

	// 接受连接
	for {
		// DTLCP 通过 Accept 处理 HelloVerifyRequest 握手
		dtlsConn, err := gotlcp.Accept(conn, config)
		if err != nil {
			fmt.Printf("接受连接失败: %v\n", err)
			continue
		}

		go handleConnection(dtlsConn)
	}
}

func handleConnection(conn *gotlcp.Conn) {
	defer conn.Close()

	// 读取客户端数据
	buf := make([]byte, 4096)
	n, err := conn.Read(buf)
	if err != nil {
		fmt.Printf("读取失败: %v\n", err)
		return
	}

	fmt.Printf("收到 [%s]: %s\n", conn.RemoteAddr(), string(buf[:n]))

	// 回复
	reply := fmt.Sprintf("DTLCP 服务端确认: 已收到 %d 字节", n)
	_, err = conn.Write([]byte(reply))
	if err != nil {
		fmt.Printf("写入失败: %v\n", err)
		return
	}
}

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/Trisia/gotlcp"
)

func main() {
	// 客户端配置（跳过证书验证用于测试）
	config := &gotlcp.Config{
		InsecureSkipVerify: true, // 生产环境必须验证证书链
		PMTU:               1400,
		InitialTimeout:     time.Second,
		MaxTimeout:         64 * time.Second,
	}

	// 连接服务端
	serverAddr := "127.0.0.1:8443"
	udpAddr, err := net.ResolveUDPAddr("udp", serverAddr)
	if err != nil {
		fmt.Printf("解析地址失败: %v\n", err)
		os.Exit(1)
	}

	udpConn, err := net.ListenUDP("udp", nil)
	if err != nil {
		fmt.Printf("创建 UDP 连接失败: %v\n", err)
		os.Exit(1)
	}

	conn, err := gotlcp.Client(udpConn, udpAddr, config)
	if err != nil {
		fmt.Printf("DTLCP 握手失败: %v\n", err)
		os.Exit(1)
	}
	defer conn.Close()

	fmt.Println("DTLCP 握手成功，开始通信...")

	// 发送数据
	message := "Hello DTLCP! 这是国密加密传输测试。"
	_, err = conn.Write([]byte(message))
	if err != nil {
		fmt.Printf("发送失败: %v\n", err)
		os.Exit(1)
	}

	// 读取响应
	buf := make([]byte, 4096)
	n, err := conn.Read(buf)
	if err != nil {
		fmt.Printf("读取响应失败: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("服务端响应: %s\n", string(buf[:n]))
}