improve: 为 linux 添加可靠的网卡绑定

tools/http.go

@@ -2,8 +2,6 @@ package tools
 
 import (
 	"github.com/Mmx233/BitSrunLoginGo/internal/global"
-	"github.com/Mmx233/tool"
-	"net"
 	"net/http"
 )
 
@@ -13,40 +11,25 @@ type Http struct {
 
 var HttpPack *Http
 
-var httpTools map[net.Addr]*Http
+var httpTools map[string]*Http
 
 func init() {
 	if global.Config.Settings.Basic.Interfaces == "" {
 		HttpPack = genHttpPack(nil)
 	} else {
-		httpTools = make(map[net.Addr]*Http, 0)
+		httpTools = make(map[string]*Http, 0)
 	}
 }
 
-func HttpPackSelect(addr net.Addr) *Http {
+func HttpPackSelect(eth *Eth) *Http {
 	if HttpPack != nil {
 		return HttpPack
 	}
-	if addrHttp, ok := httpTools[addr]; ok {
+	if addrHttp, ok := httpTools[eth.Name]; ok {
 		return addrHttp
 	} else {
-		addrHttp = genHttpPack(addr)
-		httpTools[addr] = addrHttp
+		addrHttp = genHttpPack(eth)
+		httpTools[eth.Name] = addrHttp
 		return addrHttp
 	}
 }
-
-func genHttpPack(addr net.Addr) *Http {
-	tr := tool.GenHttpTransport(&tool.HttpTransportOptions{
-		Timeout:           global.Timeout,
-		LocalAddr:         addr,
-		SkipSslCertVerify: global.Config.Settings.Basic.SkipCertVerify,
-	})
-	tr.Proxy = http.ProxyFromEnvironment
-	return &Http{
-		Client: tool.GenHttpClient(&tool.HttpClientOptions{
-			Transport: tr,
-			Timeout:   global.Timeout,
-		}),
-	}
-}

tools/http_default.go

@@ -0,0 +1,29 @@
+//go:build !linux
+
+package tools
+
+import (
+	"github.com/Mmx233/BitSrunLoginGo/internal/global"
+	"github.com/Mmx233/tool"
+	"net"
+	"net/http"
+)
+
+func genHttpPack(eth *Eth) *Http {
+	var addr net.Addr
+	if eth != nil {
+		addr = eth.Addr
+	}
+	tr := tool.GenHttpTransport(&tool.HttpTransportOptions{
+		Timeout:           global.Timeout,
+		LocalAddr:         addr,
+		SkipSslCertVerify: global.Config.Settings.Basic.SkipCertVerify,
+	})
+	tr.Proxy = http.ProxyFromEnvironment
+	return &Http{
+		Client: tool.GenHttpClient(&tool.HttpClientOptions{
+			Transport: tr,
+			Timeout:   global.Timeout,
+		}),
+	}
+}

tools/http_linux.go

@@ -0,0 +1,45 @@
+package tools
+
+import (
+	"github.com/Mmx233/BitSrunLoginGo/internal/global"
+	"github.com/Mmx233/tool"
+	"net"
+	"net/http"
+	"syscall"
+)
+
+func genHttpPack(eth *Eth) *Http {
+	dialer := net.Dialer{
+		Timeout: global.Timeout,
+	}
+	if eth != nil {
+		dialer.LocalAddr = eth.Addr
+		ethName := eth.Name
+		dialer.Control = func(network string, address string, c syscall.RawConn) error {
+			var operr error
+			fn := func(fd uintptr) {
+				operr = syscall.SetsockoptString(int(fd), syscall.SOL_SOCKET, syscall.SO_BINDTODEVICE, ethName)
+			}
+			if err := c.Control(fn); err != nil {
+				return err
+			}
+			if operr != nil {
+				return operr
+			}
+			return nil
+		}
+	}
+
+	tr := &http.Transport{
+		DialContext:         dialer.DialContext,
+		TLSHandshakeTimeout: global.Timeout,
+		IdleConnTimeout:     global.Timeout,
+	}
+	tr.Proxy = http.ProxyFromEnvironment
+	return &Http{
+		Client: tool.GenHttpClient(&tool.HttpClientOptions{
+			Transport: tr,
+			Timeout:   global.Timeout,
+		}),
+	}
+}