feat: 支持使用 flag 绑定网卡

tools/eth.go

@@ -1,7 +1,6 @@
 package tools
 
 import (
-	"github.com/Mmx233/BitSrunLoginGo/internal/global"
 	log "github.com/sirupsen/logrus"
 	"net"
 	"regexp"
@@ -13,38 +12,50 @@ type Eth struct {
 	Addr net.Addr
 }
 
-func GetInterfaceAddr() ([]Eth, error) {
+// ConvertInterface 当没有 ipv4 地址时 eth 可能为 nil
+func ConvertInterface(eth net.Interface) (*Eth, error) {
+	addresses, e := eth.Addrs()
+	if e != nil {
+		return nil, e
+	}
+	for _, addr := range addresses {
+		if strings.Contains(addr.String(), ".") {
+			var ip *net.TCPAddr
+			ip, e = net.ResolveTCPAddr("tcp", strings.Split(addr.String(), "/")[0]+":0")
+			if e != nil {
+				log.Warnln(eth.Name+" ip解析失败：", e)
+				continue
+			}
+			return &Eth{
+				Name: eth.Name,
+				Addr: ip,
+			}, nil
+		}
+	}
+	return nil, nil
+}
+
+func GetInterfaceAddr(regexpStr string) ([]Eth, error) {
 	var result []Eth
 
 	interfaces, e := net.Interfaces()
 	if e != nil {
 		return nil, e
 	}
-	reg, e := regexp.Compile(global.Config.Settings.Basic.Interfaces)
+	reg, e := regexp.Compile(regexpStr)
 	if e != nil {
 		log.Fatalln("interfaces设置异常，无法解析: ", e)
 	}
 	for _, eth := range interfaces {
 		if reg.Match([]byte(eth.Name)) {
-			addrs, e := eth.Addrs()
+			cEth, e := ConvertInterface(eth)
 			if e != nil {
 				log.Warnln(eth.Name+" 网卡地址获取失败: ", e)
 				continue
 			}
-			for _, addr := range addrs {
-				if strings.Contains(addr.String(), ".") {
-					var ip *net.TCPAddr
-					ip, e = net.ResolveTCPAddr("tcp", strings.Split(addr.String(), "/")[0]+":0")
-					if e != nil {
-						log.Warnln(eth.Name+" ip解析失败：", e)
-						continue
-					}
-					result = append(result, Eth{
-						Name: eth.Name,
-						Addr: ip,
-					})
-					break
-				}
+
+			if cEth != nil {
+				result = append(result, *cEth)
 			}
 		} else {
 			log.Debugf("网卡 %s 不匹配", eth.Name)