u

netstack/setup_windows.go

@@ -0,0 +1,189 @@
+//go:build windows
+
+package netstack
+
+import (
+	"crypto/tls"
+	"fmt"
+	"gogo"
+	"goodlink/winipcfg"
+	"io"
+	"net"
+	"net/http"
+	"net/netip"
+	"time"
+
+	"github.com/quic-go/quic-go"
+	"gvisor.dev/gvisor/pkg/tcpip"
+	"gvisor.dev/gvisor/pkg/tcpip/network/ipv4"
+	"gvisor.dev/gvisor/pkg/tcpip/stack"
+	"gvisor.dev/gvisor/pkg/tcpip/transport/tcp"
+)
+
+func SetWinTunIP(wintunEP *Device, ip string, mask int, ip2r string, mask2r int) error {
+	var ipf netip.Prefix
+	var err error
+
+	// 将LUID转换为Windows网络接口标识
+	// LUID: 本地唯一接口标识符（Local Unique Identifier）
+	// 用于唯一标识网络接口，通常用于网络配置和管理
+	link := winipcfg.LUID((*wintunEP).(*TUN).GetNt().LUID())
+
+	// 将IP地址和掩码组合为CIDR格式（如192.168.1.1/24）
+	if ipf, err = netip.ParsePrefix(fmt.Sprintf("%s/%d", ip, mask)); err != nil {
+		return err
+	}
+
+	// 设置接口的主IP地址
+	if err = link.SetIPAddresses([]netip.Prefix{ipf}); err != nil {
+		return err
+	}
+
+	if ipf, err = netip.ParsePrefix(fmt.Sprintf("%s/%d", ip2r, mask2r)); err != nil {
+		return err
+	}
+
+	// 配置路由规则参数
+	routeData := &winipcfg.RouteData{
+		Destination: ipf,                     // 目标网络（CIDR格式）
+		NextHop:     netip.MustParseAddr(ip), // 下一跳地址（本机IP）
+		Metric:      0,                       // 路由优先级（数值越小优先级越高）
+	}
+
+	// 应用路由配置到网络接口
+	if err = link.SetRoutes([]*winipcfg.RouteData{
+		routeData,
+	}); err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func SetNetStackIP(s *stack.Stack, nicID tcpip.NICID, ip string, mask int, ip2r, mask2r string) error {
+	// 设置IP地址
+	// 配置网络接口的IPv4地址为192.168.3.3/24
+	protocolAddr := tcpip.ProtocolAddress{
+		Protocol: ipv4.ProtocolNumber,
+		AddressWithPrefix: tcpip.AddressWithPrefix{
+			Address:   tcpip.AddrFromSlice(net.ParseIP(ip).To4()), // 设置IP地址
+			PrefixLen: mask,                                       // 设置子网掩码长度
+		},
+	}
+	// 将IP地址添加到网络接口，设置为静态配置的主端点
+	if err := s.AddProtocolAddress(nicID, protocolAddr, stack.AddressProperties{
+		PEB:        stack.CanBePrimaryEndpoint, // 允许作为主端点
+		ConfigType: stack.AddressConfigStatic,  // 使用静态配置
+	}); err != nil {
+		return fmt.Errorf("AddProtocolAddress failed: %v", err)
+	}
+
+	// 设置路由
+	// 配置默认路由，将所有192.168.3.0/24网段的流量转发到该接口
+	subnet, err := tcpip.NewSubnet(
+		tcpip.AddrFromSlice(net.ParseIP(ip2r).To4()),   // 设置目标网段
+		tcpip.MaskFromBytes(net.ParseIP(mask2r).To4()), // 设置子网掩码
+	)
+	if err != nil {
+		return fmt.Errorf("NewSubnet failed: %v", err)
+	}
+	// 设置路由表，将指定网段的流量转发到NIC ID 1
+	s.SetRouteTable([]tcpip.Route{
+		{
+			Destination: subnet, // 目标网段
+			NIC:         nicID,  // 使用的网络接口
+		},
+	})
+
+	// 设置网络接口为混杂模式
+	// 允许接收所有网络数据包，用于网络数据包的转发
+	s.SetPromiscuousMode(nicID, true)
+
+	return nil
+}
+
+var (
+	init_stack_suss = false
+	netstack_stack  *stack.Stack
+)
+
+func InitWintunDll() error {
+	if !gogo.Utils().FileExist("wintun.dll") {
+		var res []byte
+		var err error
+		var resp *http.Response
+
+		client := &http.Client{
+			Transport: &http.Transport{
+				TLSClientConfig: &tls.Config{
+					InsecureSkipVerify: true, // 跳过证书验证
+				},
+			},
+			Timeout: 3 * time.Second,
+		}
+		if resp, err = client.Get("https://gitee.com/konyshe/goodlink_conf/raw/master/wintun.dll"); err != nil {
+			return err
+		}
+		defer resp.Body.Close()
+
+		if res, err = io.ReadAll(resp.Body); err != nil {
+			return err
+		}
+
+		gogo.Utils().FileAppend("wintun.dll", res)
+	}
+
+	return nil
+}
+
+// setupNetstack 初始化并配置网络栈
+// 该函数负责创建协议栈、设置网络接口、配置IP地址和路由表
+// 返回:
+//   - *stack.Stack: 配置好的网络栈实例
+//   - error: 初始化过程中的错误信息
+func Start() error {
+	InitWintunDll()
+
+	if init_stack_suss {
+		return nil
+	}
+
+	//Sudo()
+
+	// 创建协议栈
+	// 创建新的协议栈实例，配置网络层和传输层协议
+	netstack_stack = stack.New(stack.Options{
+		NetworkProtocols:   []stack.NetworkProtocolFactory{ipv4.NewProtocol},  // 配置IPv4协议
+		TransportProtocols: []stack.TransportProtocolFactory{tcp.NewProtocol}, // 配置TCP协议
+	})
+
+	// 使用Open函数创建TUN设备，设备名称为"GoodLink"，MTU为0表示使用系统默认值
+	wintunEP, err := Open("GoodLink", 1400) //因为要加自定义的头，防止超出1500，造成不必要的性能损耗
+	if err != nil {
+		return fmt.Errorf("请管理员权限运行")
+	}
+
+	SetWinTunIP(&wintunEP, "192.17.19.1", 32, "192.17.19.1", 32)
+
+	// 创建网络接口
+	// 将TUN设备注册到协议栈中，使用NIC ID 1
+	nicID := tcpip.NICID(1)
+	if err := netstack_stack.CreateNIC(nicID, wintunEP); err != nil {
+		return fmt.Errorf("设备注册: %v", err)
+	}
+
+	SetNetStackIP(netstack_stack, nicID, "192.17.19.1", 32, "192.17.19.0", "255.255.255.0")
+
+	init_stack_suss = true
+
+	return nil
+}
+
+func SetForWarder(stun_quic_conn quic.Connection) {
+	// 设置TCP协议处理器
+	netstack_stack.SetTransportProtocolHandler(tcp.ProtocolNumber, NewTcpForwarder(netstack_stack, stun_quic_conn).HandlePacket)
+}
+
+func GetRemoteIP() string {
+	return "192.17.19.1"
+}