mirror of
http://bgp.hk.skcks.cn:10088/github.com/oneclickvirt/ecs
synced 2026-04-20 21:01:12 +08:00
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package tests
|
||
|
||
import (
|
||
"fmt"
|
||
"os"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/oneclickvirt/UnlockTests/executor"
|
||
bgptools "github.com/oneclickvirt/backtrace/bgptools"
|
||
backtrace "github.com/oneclickvirt/backtrace/bk"
|
||
. "github.com/oneclickvirt/defaultset"
|
||
)
|
||
|
||
type IpInfo struct {
|
||
Ip string `json:"ip"`
|
||
City string `json:"city"`
|
||
Region string `json:"region"`
|
||
Country string `json:"country"`
|
||
Org string `json:"org"`
|
||
}
|
||
|
||
type ConcurrentResults struct {
|
||
bgpResult string
|
||
backtraceResult string
|
||
bgpError error
|
||
// backtraceError error
|
||
}
|
||
|
||
var IPV4, IPV6 string
|
||
|
||
func UpstreamsCheck(language string) {
|
||
// 添加panic恢复机制
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
if language == "zh" {
|
||
fmt.Println("\n上游检测出现错误，已跳过")
|
||
} else {
|
||
fmt.Println("\nUpstream check failed, skipped")
|
||
}
|
||
fmt.Fprintf(os.Stderr, "[WARN] Upstream check panic: %v\n", r)
|
||
}
|
||
}()
|
||
|
||
results := ConcurrentResults{}
|
||
var wg sync.WaitGroup
|
||
if IPV4 != "" {
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
fmt.Fprintf(os.Stderr, "[WARN] BGP info panic: %v\n", r)
|
||
}
|
||
}()
|
||
for i := 0; i < 2; i++ {
|
||
result, err := bgptools.GetPoPInfo(IPV4)
|
||
results.bgpError = err
|
||
if err == nil && result.Result != "" {
|
||
results.bgpResult = result.Result
|
||
return
|
||
}
|
||
if i == 0 {
|
||
time.Sleep(3 * time.Second)
|
||
}
|
||
}
|
||
}()
|
||
}
|
||
wg.Add(1)
|
||
go func() {
|
||
defer wg.Done()
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
fmt.Fprintf(os.Stderr, "[WARN] Backtrace panic: %v\n", r)
|
||
}
|
||
}()
|
||
result := backtrace.BackTrace(executor.IPV6)
|
||
results.backtraceResult = result
|
||
}()
|
||
wg.Wait()
|
||
if results.bgpResult != "" {
|
||
fmt.Print(results.bgpResult)
|
||
}
|
||
if results.backtraceResult != "" {
|
||
fmt.Printf("%s\n", results.backtraceResult)
|
||
}
|
||
if language == "zh" {
|
||
fmt.Println(Yellow("准确线路自行查看详细路由，本测试结果仅作参考"))
|
||
fmt.Println(Yellow("同一目标地址多个线路时，检测可能已越过汇聚层，除第一个线路外，后续信息可能无效"))
|
||
} else {
|
||
fmt.Println(Yellow("For accurate routing, check the detailed routes yourself. This result is for reference only."))
|
||
fmt.Println(Yellow("When multiple routes share the same destination, detection may have passed the aggregation layer; only the first route is reliable."))
|
||
}
|
||
}
|