improve: 使用标准项目结构
This commit is contained in:
61
tools/eth.go
Normal file
61
tools/eth.go
Normal file
@@ -0,0 +1,61 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"github.com/Mmx233/BitSrunLoginGo/internal/global"
|
||||
log "github.com/sirupsen/logrus"
|
||||
"net"
|
||||
"regexp"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type Eth struct {
|
||||
Name string
|
||||
Addr net.Addr
|
||||
}
|
||||
|
||||
func GetInterfaceAddr() ([]Eth, error) {
|
||||
var result []Eth
|
||||
|
||||
interfaces, e := net.Interfaces()
|
||||
if e != nil {
|
||||
return nil, e
|
||||
}
|
||||
reg, e := regexp.Compile(global.Config.Settings.Basic.Interfaces)
|
||||
if e != nil {
|
||||
log.Fatalln("interfaces设置异常,无法解析: ", e)
|
||||
}
|
||||
for _, eth := range interfaces {
|
||||
if reg.Match([]byte(eth.Name)) {
|
||||
addrs, e := eth.Addrs()
|
||||
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
|
||||
}
|
||||
}
|
||||
} else {
|
||||
log.Debugf("网卡 %s 不匹配", eth.Name)
|
||||
}
|
||||
}
|
||||
|
||||
log.Debugln("有效匹配网卡:", result)
|
||||
|
||||
if len(result) == 0 {
|
||||
log.Warnln("没有扫描到有效匹配网卡")
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
52
tools/http.go
Normal file
52
tools/http.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package tools
|
||||
|
||||
import (
|
||||
"github.com/Mmx233/BitSrunLoginGo/internal/global"
|
||||
"github.com/Mmx233/tool"
|
||||
"net"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type Http struct {
|
||||
Client *http.Client
|
||||
}
|
||||
|
||||
var HttpPack *Http
|
||||
|
||||
var httpTools map[net.Addr]*Http
|
||||
|
||||
func init() {
|
||||
if global.Config.Settings.Basic.Interfaces == "" {
|
||||
HttpPack = genHttpPack(nil)
|
||||
} else {
|
||||
httpTools = make(map[net.Addr]*Http, 0)
|
||||
}
|
||||
}
|
||||
|
||||
func HttpPackSelect(addr net.Addr) *Http {
|
||||
if HttpPack != nil {
|
||||
return HttpPack
|
||||
}
|
||||
if addrHttp, ok := httpTools[addr]; ok {
|
||||
return addrHttp
|
||||
} else {
|
||||
addrHttp = genHttpPack(addr)
|
||||
httpTools[addr] = 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,
|
||||
}),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user