mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
tsosunchia
c6eb9bbd2e
交互式变基操作正在进行中；至 cbc511f
最后完成的命令（1 条命令被执行）：
pick 356e3db 修复ft时个别地方未应用色彩参数的问题
接下来要执行的命令（剩余 1 条命令）：
pick 9ee7c3f 修正在ECMP网络下的表现 之前ECMP网络可能会导致一次trace的结果来自不同flow（按icmp id区分的策略下），目前id不再携带TTL信息，因此一次trace的id将保持一致。 要提交的变更： 修改： trace/icmp_ipv4.go 修改： trace/icmp_ipv6.go
您在执行将分支 'main' 变基到 'cbc511f' 的操作。
要提交的变更：
修改： fast_trace/fast_trace.go
422 lines
12 KiB
Go
422 lines
12 KiB
Go
package fastTrace
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"github.com/fatih/color"
|
|
"github.com/nxtrace/NTrace-core/ipgeo"
|
|
"github.com/nxtrace/NTrace-core/printer"
|
|
"github.com/nxtrace/NTrace-core/trace"
|
|
"github.com/nxtrace/NTrace-core/tracelog"
|
|
"github.com/nxtrace/NTrace-core/util"
|
|
"github.com/nxtrace/NTrace-core/wshandle"
|
|
"log"
|
|
"net"
|
|
"os"
|
|
"os/signal"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type FastTracer struct {
|
|
TracerouteMethod trace.Method
|
|
ParamsFastTrace ParamsFastTrace
|
|
}
|
|
|
|
type ParamsFastTrace struct {
|
|
SrcDev string
|
|
SrcAddr string
|
|
BeginHop int
|
|
MaxHops int
|
|
RDns bool
|
|
AlwaysWaitRDNS bool
|
|
Lang string
|
|
PktSize int
|
|
Timeout time.Duration
|
|
File string
|
|
}
|
|
|
|
type IpListElement struct {
|
|
Ip string
|
|
Desc string
|
|
Version4 bool // true for IPv4, false for IPv6
|
|
}
|
|
|
|
var oe = false
|
|
|
|
func (f *FastTracer) tracert(location string, ispCollection ISPCollection) {
|
|
fmt.Fprintf(color.Output, "%s\n", color.New(color.FgYellow, color.Bold).Sprintf("『%s %s 』", location, ispCollection.ISPName))
|
|
fmt.Printf("traceroute to %s, %d hops max, %d byte packets\n", ispCollection.IP, f.ParamsFastTrace.MaxHops, f.ParamsFastTrace.PktSize)
|
|
|
|
ip, err := util.DomainLookUp(ispCollection.IP, "4", "", true)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
var conf = trace.Config{
|
|
BeginHop: f.ParamsFastTrace.BeginHop,
|
|
DestIP: ip,
|
|
DestPort: 80,
|
|
MaxHops: f.ParamsFastTrace.MaxHops,
|
|
NumMeasurements: 3,
|
|
ParallelRequests: 18,
|
|
RDns: f.ParamsFastTrace.RDns,
|
|
AlwaysWaitRDNS: f.ParamsFastTrace.AlwaysWaitRDNS,
|
|
PacketInterval: 100,
|
|
TTLInterval: 500,
|
|
IPGeoSource: ipgeo.GetSource("LeoMoeAPI"),
|
|
Timeout: f.ParamsFastTrace.Timeout,
|
|
SrcAddr: f.ParamsFastTrace.SrcAddr,
|
|
PktSize: f.ParamsFastTrace.PktSize,
|
|
Lang: f.ParamsFastTrace.Lang,
|
|
}
|
|
|
|
if oe {
|
|
fp, err := os.OpenFile("/tmp/trace.log", os.O_CREATE|os.O_APPEND|os.O_RDWR, os.ModePerm)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer func(fp *os.File) {
|
|
err := fp.Close()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}(fp)
|
|
|
|
log.SetOutput(fp)
|
|
log.SetFlags(0)
|
|
log.Printf("『%s %s 』\n", location, ispCollection.ISPName)
|
|
log.Printf("traceroute to %s, %d hops max, %d byte packets\n", ispCollection.IP, f.ParamsFastTrace.MaxHops, f.ParamsFastTrace.PktSize)
|
|
conf.RealtimePrinter = tracelog.RealtimePrinter
|
|
} else {
|
|
conf.RealtimePrinter = printer.RealtimePrinter
|
|
}
|
|
|
|
_, err = trace.Traceroute(f.TracerouteMethod, conf)
|
|
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
fmt.Println()
|
|
}
|
|
|
|
func FastTest(tm bool, outEnable bool, paramsFastTrace ParamsFastTrace) {
|
|
// tm means tcp mode
|
|
var c string
|
|
oe = outEnable
|
|
|
|
if paramsFastTrace.File != "" {
|
|
testFile(paramsFastTrace, tm)
|
|
return
|
|
}
|
|
|
|
fmt.Println("Hi，欢迎使用 Fast Trace 功能，请注意 Fast Trace 功能只适合新手使用\n因为国内网络复杂，我们设置的测试目标有限，建议普通用户自测以获得更加精准的路由情况")
|
|
fmt.Println("请您选择要测试的 IP 类型\n1. IPv4\n2. IPv6")
|
|
fmt.Print("请选择选项：")
|
|
_, err := fmt.Scanln(&c)
|
|
if err != nil {
|
|
c = "1"
|
|
}
|
|
if c == "2" {
|
|
if paramsFastTrace.SrcDev != "" {
|
|
dev, _ := net.InterfaceByName(paramsFastTrace.SrcDev)
|
|
if addrs, err := dev.Addrs(); err == nil {
|
|
for _, addr := range addrs {
|
|
if (addr.(*net.IPNet).IP.To4() == nil) == true {
|
|
paramsFastTrace.SrcAddr = addr.(*net.IPNet).IP.String()
|
|
// 检查是否是内网IP
|
|
if !(net.ParseIP(paramsFastTrace.SrcAddr).IsPrivate() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLoopback() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLinkLocalUnicast() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLinkLocalMulticast()) {
|
|
// 若不是则跳出
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
FastTestv6(tm, outEnable, paramsFastTrace)
|
|
return
|
|
}
|
|
if paramsFastTrace.SrcDev != "" {
|
|
dev, _ := net.InterfaceByName(paramsFastTrace.SrcDev)
|
|
if addrs, err := dev.Addrs(); err == nil {
|
|
for _, addr := range addrs {
|
|
if (addr.(*net.IPNet).IP.To4() == nil) == false {
|
|
paramsFastTrace.SrcAddr = addr.(*net.IPNet).IP.String()
|
|
// 检查是否是内网IP
|
|
if !(net.ParseIP(paramsFastTrace.SrcAddr).IsPrivate() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLoopback() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLinkLocalUnicast() ||
|
|
net.ParseIP(paramsFastTrace.SrcAddr).IsLinkLocalMulticast()) {
|
|
// 若不是则跳出
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
fmt.Println("您想测试哪些ISP的路由？\n1. 国内四网\n2. 电信\n3. 联通\n4. 移动\n5. 教育网\n6. 全部")
|
|
fmt.Print("请选择选项：")
|
|
_, err = fmt.Scanln(&c)
|
|
if err != nil {
|
|
c = "1"
|
|
}
|
|
|
|
ft := FastTracer{
|
|
ParamsFastTrace: paramsFastTrace,
|
|
}
|
|
|
|
// 建立 WebSocket 连接
|
|
w := wshandle.New()
|
|
w.Interrupt = make(chan os.Signal, 1)
|
|
signal.Notify(w.Interrupt, os.Interrupt)
|
|
defer func() {
|
|
w.Conn.Close()
|
|
}()
|
|
|
|
if !tm {
|
|
ft.TracerouteMethod = trace.ICMPTrace
|
|
fmt.Println("您将默认使用ICMP协议进行路由跟踪，如果您想使用TCP SYN进行路由跟踪，可以加入 -T 参数")
|
|
} else {
|
|
ft.TracerouteMethod = trace.TCPTrace
|
|
}
|
|
|
|
switch c {
|
|
case "1":
|
|
ft.testFast()
|
|
case "2":
|
|
ft.testCT()
|
|
case "3":
|
|
ft.testCU()
|
|
case "4":
|
|
ft.testCM()
|
|
case "5":
|
|
ft.testEDU()
|
|
case "6":
|
|
ft.testAll()
|
|
default:
|
|
ft.testFast()
|
|
}
|
|
}
|
|
|
|
func testFile(paramsFastTrace ParamsFastTrace, tm bool) {
|
|
// 建立 WebSocket 连接
|
|
w := wshandle.New()
|
|
w.Interrupt = make(chan os.Signal, 1)
|
|
signal.Notify(w.Interrupt, os.Interrupt)
|
|
defer func() {
|
|
w.Conn.Close()
|
|
}()
|
|
|
|
var tracerouteMethod trace.Method
|
|
if !tm {
|
|
tracerouteMethod = trace.ICMPTrace
|
|
fmt.Println("您将默认使用ICMP协议进行路由跟踪，如果您想使用TCP SYN进行路由跟踪，可以加入 -T 参数")
|
|
} else {
|
|
tracerouteMethod = trace.TCPTrace
|
|
}
|
|
|
|
filePath := paramsFastTrace.File
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
fmt.Println("Error opening file:", err)
|
|
return
|
|
}
|
|
defer func(file *os.File) {
|
|
err := file.Close()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}(file)
|
|
var ipList []IpListElement
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
line := scanner.Text()
|
|
parts := strings.SplitN(line, " ", 2)
|
|
|
|
var ip, desc string
|
|
if len(parts) == 2 {
|
|
ip = parts[0]
|
|
desc = parts[1]
|
|
} else if len(parts) == 1 {
|
|
ip = parts[0]
|
|
desc = ip // Set the description to the IP if no description is provided
|
|
} else {
|
|
fmt.Printf("Ignoring invalid line: %s\n", line)
|
|
continue
|
|
}
|
|
|
|
parsedIP := net.ParseIP(ip)
|
|
if parsedIP == nil {
|
|
netIp, err := util.DomainLookUp(ip, "all", "", true)
|
|
if err != nil {
|
|
fmt.Printf("Ignoring invalid IP: %s\n", ip)
|
|
continue
|
|
}
|
|
if len(parts) == 1 {
|
|
desc = ip
|
|
}
|
|
ip = netIp.String()
|
|
}
|
|
|
|
ipElem := IpListElement{
|
|
Ip: ip,
|
|
Desc: desc,
|
|
Version4: strings.Contains(ip, "."),
|
|
}
|
|
|
|
ipList = append(ipList, ipElem)
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
fmt.Println("Error reading file:", err)
|
|
}
|
|
|
|
for _, ip := range ipList {
|
|
fmt.Fprintf(color.Output, "%s\n",
|
|
color.New(color.FgYellow, color.Bold).Sprint("『 "+ip.Desc+"』"),
|
|
)
|
|
if util.EnableHidDstIP == "" {
|
|
fmt.Printf("traceroute to %s, %d hops max, %d bytes packets\n", ip.Ip, paramsFastTrace.MaxHops, paramsFastTrace.PktSize)
|
|
} else {
|
|
fmt.Printf("traceroute to %s, %d hops max, %d bytes packets\n", util.HideIPPart(ip.Ip), paramsFastTrace.MaxHops, paramsFastTrace.PktSize)
|
|
}
|
|
var srcAddr string
|
|
if ip.Version4 {
|
|
if paramsFastTrace.SrcDev != "" {
|
|
dev, _ := net.InterfaceByName(paramsFastTrace.SrcDev)
|
|
if addrs, err := dev.Addrs(); err == nil {
|
|
for _, addr := range addrs {
|
|
if (addr.(*net.IPNet).IP.To4() == nil) == false {
|
|
srcAddr = addr.(*net.IPNet).IP.String()
|
|
// 检查是否是内网IP
|
|
if !(net.ParseIP(srcAddr).IsPrivate() ||
|
|
net.ParseIP(srcAddr).IsLoopback() ||
|
|
net.ParseIP(srcAddr).IsLinkLocalUnicast() ||
|
|
net.ParseIP(srcAddr).IsLinkLocalMulticast()) {
|
|
// 若不是则跳出
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if paramsFastTrace.SrcDev != "" {
|
|
dev, _ := net.InterfaceByName(paramsFastTrace.SrcDev)
|
|
if addrs, err := dev.Addrs(); err == nil {
|
|
for _, addr := range addrs {
|
|
if (addr.(*net.IPNet).IP.To4() == nil) == true {
|
|
srcAddr = addr.(*net.IPNet).IP.String()
|
|
// 检查是否是内网IP
|
|
if !(net.ParseIP(srcAddr).IsPrivate() ||
|
|
net.ParseIP(srcAddr).IsLoopback() ||
|
|
net.ParseIP(srcAddr).IsLinkLocalUnicast() ||
|
|
net.ParseIP(srcAddr).IsLinkLocalMulticast()) {
|
|
// 若不是则跳出
|
|
break
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
var conf = trace.Config{
|
|
BeginHop: paramsFastTrace.BeginHop,
|
|
DestIP: net.ParseIP(ip.Ip),
|
|
DestPort: 80,
|
|
MaxHops: paramsFastTrace.MaxHops,
|
|
NumMeasurements: 3,
|
|
ParallelRequests: 18,
|
|
RDns: paramsFastTrace.RDns,
|
|
AlwaysWaitRDNS: paramsFastTrace.AlwaysWaitRDNS,
|
|
PacketInterval: 100,
|
|
TTLInterval: 500,
|
|
IPGeoSource: ipgeo.GetSource("LeoMoeAPI"),
|
|
Timeout: paramsFastTrace.Timeout,
|
|
SrcAddr: srcAddr,
|
|
PktSize: paramsFastTrace.PktSize,
|
|
Lang: paramsFastTrace.Lang,
|
|
}
|
|
|
|
if oe {
|
|
fp, err := os.OpenFile("/tmp/trace.log", os.O_CREATE|os.O_APPEND|os.O_RDWR, os.ModePerm)
|
|
if err != nil {
|
|
return
|
|
}
|
|
log.SetOutput(fp)
|
|
log.SetFlags(0)
|
|
log.Printf("『%s』\n", ip.Desc)
|
|
log.Printf("traceroute to %s, %d hops max, %d byte packets\n", ip.Ip, paramsFastTrace.MaxHops, paramsFastTrace.PktSize)
|
|
conf.RealtimePrinter = tracelog.RealtimePrinter
|
|
err = fp.Close()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
} else {
|
|
conf.RealtimePrinter = printer.RealtimePrinter
|
|
}
|
|
|
|
_, err := trace.Traceroute(tracerouteMethod, conf)
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
}
|
|
fmt.Println()
|
|
}
|
|
|
|
}
|
|
|
|
func (f *FastTracer) testAll() {
|
|
f.testCT()
|
|
println()
|
|
f.testCU()
|
|
println()
|
|
f.testCM()
|
|
println()
|
|
f.testEDU()
|
|
}
|
|
|
|
func (f *FastTracer) testCT() {
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CT163)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CT163)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CTCN2)
|
|
f.tracert(TestIPsCollection.Hangzhou.Location, TestIPsCollection.Hangzhou.CT163)
|
|
f.tracert(TestIPsCollection.Guangzhou.Location, TestIPsCollection.Guangzhou.CT163)
|
|
}
|
|
|
|
func (f *FastTracer) testCU() {
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CU169)
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CU9929)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CU169)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CU9929)
|
|
f.tracert(TestIPsCollection.Hangzhou.Location, TestIPsCollection.Hangzhou.CU169)
|
|
f.tracert(TestIPsCollection.Guangzhou.Location, TestIPsCollection.Guangzhou.CU169)
|
|
}
|
|
|
|
func (f *FastTracer) testCM() {
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CM)
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CMIN2)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CM)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.CMIN2)
|
|
f.tracert(TestIPsCollection.Hangzhou.Location, TestIPsCollection.Hangzhou.CM)
|
|
f.tracert(TestIPsCollection.Guangzhou.Location, TestIPsCollection.Guangzhou.CM)
|
|
}
|
|
|
|
func (f *FastTracer) testEDU() {
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.EDU)
|
|
f.tracert(TestIPsCollection.Shanghai.Location, TestIPsCollection.Shanghai.EDU)
|
|
f.tracert(TestIPsCollection.Hangzhou.Location, TestIPsCollection.Hangzhou.EDU)
|
|
// 科技网暂时算在EDU里面，等拿到了足够多的数据再分离出去，单独用于测试
|
|
f.tracert(TestIPsCollection.Hefei.Location, TestIPsCollection.Hefei.CST)
|
|
}
|
|
|
|
func (f *FastTracer) testFast() {
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CT163)
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CU169)
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.CM)
|
|
f.tracert(TestIPsCollection.Beijing.Location, TestIPsCollection.Beijing.EDU)
|
|
}
|