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

This commit is contained in:
Mmx233
2023-06-01 10:02:53 +08:00
parent bcb840560c
commit b0b0f2d6a7
8 changed files with 84 additions and 38 deletions

45
tools/http_linux.go Normal file
View File

@@ -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,
}),
}
}