169 lines
3.4 KiB
Go
169 lines
3.4 KiB
Go
package config
|
||
|
||
import (
|
||
"agent-wdd/log"
|
||
"encoding/json"
|
||
"io"
|
||
"net/http"
|
||
"time"
|
||
)
|
||
|
||
// 能够联网,就大于这个数字 外网9 国内7 不能联网1 未知0
|
||
const InternetBaseLine = 1
|
||
|
||
// 定义响应数据结构体
|
||
type IPInfo struct {
|
||
IP string `json:"ip"`
|
||
City string `json:"city"`
|
||
Region string `json:"region"`
|
||
Country string `json:"country"`
|
||
Loc string `json:"loc"`
|
||
Org string `json:"org"`
|
||
Postal string `json:"postal"`
|
||
Timezone string `json:"timezone"`
|
||
}
|
||
|
||
// Gather 获取网络相关的信息
|
||
func (network *Network) Gather() {
|
||
|
||
log.Info("Gathering INFO => NETWORK !")
|
||
// 能够联网
|
||
network.Internet = CanConnectInternet()
|
||
|
||
// 获取公网的相关信息
|
||
pub := PublicInfo{}
|
||
pub.GetPublicInfo()
|
||
|
||
network.Public = pub
|
||
|
||
//获取本机网卡相关的内容
|
||
}
|
||
|
||
// CanConnectInternet 判定主机能够上网
|
||
func CanConnectInternet() int {
|
||
|
||
// 初始化 ConfigCache
|
||
if ConfigCache == nil {
|
||
ConfigCache = &Config{
|
||
TimeStamp: "",
|
||
ModifiedTimes: 0,
|
||
Agent: Agent{
|
||
OS: OS{},
|
||
Network: Network{},
|
||
CPU: CPU{},
|
||
Mem: Memory{},
|
||
Swap: Swap{},
|
||
Disks: []Disk{},
|
||
},
|
||
}
|
||
}
|
||
|
||
// 读取 config 文件，判定能否上网
|
||
internetCode := ConfigCache.Agent.Network.Internet
|
||
|
||
if internetCode == 0 {
|
||
// 没有相关的信息,需要重新判定
|
||
internetCode = judgeCanConnectInternet()
|
||
|
||
// 持久化保存
|
||
ConfigCache.Agent.Network.Internet = internetCode
|
||
SaveConfig()
|
||
}
|
||
|
||
return internetCode
|
||
}
|
||
|
||
// judgeCanConnectInternet 请求网址判定主机能否上网 请求 www.google.com 如果请求正常 返回9 如果超时三秒 请求baidu.com，如果没有错误返回7 如果错误返回1
|
||
func judgeCanConnectInternet() int {
|
||
client := http.Client{
|
||
Timeout: 3 * time.Second,
|
||
}
|
||
|
||
_, err := client.Get("https://www.google.com")
|
||
if err == nil {
|
||
return 9
|
||
}
|
||
|
||
_, err = client.Get("https://www.baidu.com")
|
||
if err == nil {
|
||
return 7
|
||
}
|
||
|
||
return 1
|
||
}
|
||
|
||
// GetPublicInfo 获取服务器的公网信息
|
||
func (p PublicInfo) GetPublicInfo() {
|
||
|
||
if CanConnectInternet() == InternetBaseLine {
|
||
// 无法联网, 假信息
|
||
|
||
fakePublicInfo := PublicInfo{
|
||
IPv4: "1.1.1.1",
|
||
IPv6: "2400::1",
|
||
Country: "CN",
|
||
City: "Shanghai",
|
||
ASN: "Wdd Inc",
|
||
}
|
||
|
||
ConfigCache.Agent.Network.Public = fakePublicInfo
|
||
|
||
SaveConfig()
|
||
return
|
||
}
|
||
|
||
// 可以联网
|
||
// 创建带有超时的HTTP客户端
|
||
client := &http.Client{
|
||
Timeout: 3 * time.Second,
|
||
}
|
||
|
||
// 创建新的请求对象
|
||
req, err := http.NewRequest("GET", "https://ipinfo.io", nil)
|
||
if err != nil {
|
||
log.Error("创建请求失败: %v", err)
|
||
}
|
||
|
||
// 设置请求头
|
||
req.Header.Add("Authorization", "Bearer 6ecb0db9bd8f19")
|
||
req.Header.Add("Accept", "application/json")
|
||
|
||
// 发送请求
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
log.Error("请求PublicInfo失败: %v", err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 检查响应状态码
|
||
if resp.StatusCode != http.StatusOK {
|
||
log.Error("非200状态码: %d", resp.StatusCode)
|
||
}
|
||
|
||
// 读取响应体
|
||
body, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
log.Error("读取响应失败: %v", err)
|
||
}
|
||
|
||
// 解析JSON数据
|
||
var info IPInfo
|
||
if err := json.Unmarshal(body, &info); err != nil {
|
||
log.Error("JSON解析失败: %v", err)
|
||
}
|
||
|
||
// 打印解析结果
|
||
log.Info("IP信息:\n%+v\n", info)
|
||
|
||
// 保存信息
|
||
p.IPv4 = info.IP
|
||
p.ASN = info.Org
|
||
p.Country = info.Country
|
||
p.City = info.City
|
||
|
||
ConfigCache.Agent.Network.Public = p
|
||
|
||
SaveConfig()
|
||
|
||
}
|