mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
119 lines
2.7 KiB
Go
119 lines
2.7 KiB
Go
package printer
|
||
|
||
import (
|
||
"fmt"
|
||
"net"
|
||
"strings"
|
||
|
||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||
"github.com/xgadget-lab/nexttrace/methods"
|
||
)
|
||
|
||
var dataOrigin string
|
||
|
||
func TraceroutePrinter(ip net.IP, res map[uint16][]methods.TracerouteHop, dataOrigin string, rdnsenable bool) {
|
||
for hi := uint16(1); hi < 30; hi++ {
|
||
fmt.Print(hi)
|
||
for _, v := range res[hi] {
|
||
hopPrinter(v, rdnsenable)
|
||
if v.Address != nil && ip.String() == v.Address.String() {
|
||
hi = 31
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
func hopPrinter(v2 methods.TracerouteHop, rdnsenable bool) {
|
||
if v2.Address == nil {
|
||
fmt.Println("\t*")
|
||
} else {
|
||
var iPGeoData *ipgeo.IPGeoData
|
||
var err error
|
||
|
||
ipStr := v2.Address.String()
|
||
|
||
// 判断 err 返回，并且在CLI终端提示错误
|
||
switch dataOrigin {
|
||
case "LeoMoeAPI":
|
||
iPGeoData, err = ipgeo.LeoIP(ipStr)
|
||
case "IP.SB":
|
||
iPGeoData, err = ipgeo.IPSB(ipStr)
|
||
case "IPInfo":
|
||
iPGeoData, err = ipgeo.IPInfo(ipStr)
|
||
case "IPInsight":
|
||
iPGeoData, err = ipgeo.IPInSight(ipStr)
|
||
default:
|
||
iPGeoData, err = ipgeo.LeoIP(ipStr)
|
||
}
|
||
|
||
geo := ""
|
||
if err != nil {
|
||
geo = fmt.Sprint("Error: ", err)
|
||
} else {
|
||
geo = formatIpGeoData(ipStr, iPGeoData)
|
||
}
|
||
|
||
txt := "\t"
|
||
|
||
if rdnsenable {
|
||
ptr, err := net.LookupAddr(ipStr)
|
||
if err != nil {
|
||
txt += fmt.Sprint(ipStr, " ", fmt.Sprintf("%.2f", v2.RTT.Seconds()*1000), "ms ", geo)
|
||
} else {
|
||
txt += fmt.Sprint(ptr[0], " (", ipStr, ") ", fmt.Sprintf("%.2f", v2.RTT.Seconds()*1000), "ms ", geo)
|
||
}
|
||
} else {
|
||
txt += fmt.Sprint(ipStr, " ", fmt.Sprintf("%.2f", v2.RTT.Seconds()*1000), "ms ", geo)
|
||
}
|
||
|
||
fmt.Println(txt)
|
||
}
|
||
}
|
||
|
||
func formatIpGeoData(ip string, data *ipgeo.IPGeoData) string {
|
||
var res = make([]string, 0, 10)
|
||
|
||
if data.Asnumber == "" {
|
||
res = append(res, "*")
|
||
} else {
|
||
res = append(res, "AS"+data.Asnumber)
|
||
}
|
||
|
||
// TODO: 判断阿里云和腾讯云内网，数据不足，有待进一步完善
|
||
if strings.HasPrefix(ip, "9.") {
|
||
res = append(res, "局域网", "腾讯云")
|
||
} else if strings.HasPrefix(ip, "11.") {
|
||
res = append(res, "局域网", "阿里云")
|
||
} else if data.Country == "" {
|
||
res = append(res, "局域网")
|
||
} else {
|
||
// 有些IP的归属信息为空，这个时候将ISP的信息填入
|
||
if data.Owner == "" {
|
||
data.Owner = data.Isp
|
||
}
|
||
if data.District != "" {
|
||
data.City = data.City + ", " + data.District
|
||
}
|
||
if data.Prov == "" && data.City == "" {
|
||
// anyCast或是骨干网数据不应该有国家信息
|
||
data.Owner = data.Owner + ", " + data.Owner
|
||
} else {
|
||
// 非骨干网正常填入IP的国家信息数据
|
||
res = append(res, data.Country)
|
||
}
|
||
|
||
if data.Prov != "" {
|
||
res = append(res, data.Prov)
|
||
}
|
||
if data.City != "" {
|
||
res = append(res, data.City)
|
||
}
|
||
|
||
if data.Owner != "" {
|
||
res = append(res, data.Owner)
|
||
}
|
||
}
|
||
|
||
return strings.Join(res, ", ")
|
||
}
|