mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
refactor: udp trace
This commit is contained in:
zhshch2002
6
go.mod
6
go.mod
@@ -10,10 +10,12 @@ require (
|
||||
require (
|
||||
github.com/mattn/go-colorable v0.1.9 // indirect
|
||||
github.com/mattn/go-isatty v0.0.14 // indirect
|
||||
github.com/panjf2000/ants/v2 v2.5.0 // indirect
|
||||
golang.org/x/sync v0.0.0-20220513210516-0976fa681c29 // indirect
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/davecgh/go-spew v1.1.0 // indirect
|
||||
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||
github.com/fatih/color v1.13.0
|
||||
github.com/pmezard/go-difflib v1.0.0 // indirect
|
||||
github.com/rodaine/table v1.0.1
|
||||
@@ -22,5 +24,5 @@ require (
|
||||
github.com/tidwall/match v1.1.1 // indirect
|
||||
github.com/tidwall/pretty v1.2.0 // indirect
|
||||
golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c // indirect
|
||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b // indirect
|
||||
)
|
||||
|
||||
6
go.sum
6
go.sum
@@ -1,5 +1,6 @@
|
||||
github.com/davecgh/go-spew v1.1.0 h1:ZDRjVQ15GmhC3fiQ8ni8+OwkZQO4DARzQgrnXU1Liz8=
|
||||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/fatih/color v1.13.0 h1:8LOYc1KYPPmyKMuN8QV2DNRWNbLo6LZ0iLs8+mlH53w=
|
||||
github.com/fatih/color v1.13.0/go.mod h1:kLAiJbzzSOZDVNGyDpeOxJ47H46qBXwg5ILebYFFOfk=
|
||||
github.com/google/gopacket v1.1.19 h1:ves8RnFZPGiFnTS0uPQStjwru6uO6h+nlr9j6fL7kF8=
|
||||
@@ -11,6 +12,8 @@ github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9
|
||||
github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94=
|
||||
github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0=
|
||||
github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
|
||||
github.com/panjf2000/ants/v2 v2.5.0 h1:1rWGWSnxCsQBga+nQbA4/iY6VMeNoOIAM0ZWh9u3q2Q=
|
||||
github.com/panjf2000/ants/v2 v2.5.0/go.mod h1:cU93usDlihJZ5CfRGNDYsiBYvoilLvBF5Qp/BT2GNRE=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/rodaine/table v1.0.1 h1:U/VwCnUxlVYxw8+NJiLIuCxA/xa6jL38MY3FYysVWWQ=
|
||||
@@ -34,6 +37,8 @@ golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLL
|
||||
golang.org/x/net v0.0.0-20220425223048-2871e0cb64e4 h1:HVyaeDAYux4pnY+D/SiwmLOR36ewZ4iGQIIrtnuCjFA=
|
||||
golang.org/x/net v0.0.0-20220425223048-2871e0cb64e4/go.mod h1:CfG3xpIq0wQ8r1q4Su4UZFWDARRcnwPjda9FqA0JpMk=
|
||||
golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220513210516-0976fa681c29 h1:w8s32wxx3sY+OjLlv9qltkLU5yvJzxjjgiHWLjdIcw4=
|
||||
golang.org/x/sync v0.0.0-20220513210516-0976fa681c29/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
|
||||
@@ -48,3 +53,4 @@ gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
|
||||
@@ -11,3 +11,16 @@ type IPGeoData struct {
|
||||
}
|
||||
|
||||
type Source = func(ip string) (*IPGeoData, error)
|
||||
|
||||
func GetSource(s string) Source {
|
||||
switch s {
|
||||
case "LeoMoeAPI":
|
||||
return LeoIP
|
||||
case "IP.SB":
|
||||
return IPSB
|
||||
case "IPInsight":
|
||||
return IPInSight
|
||||
default:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
112
main.go
112
main.go
@@ -3,14 +3,13 @@ package main
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||||
"github.com/xgadget-lab/nexttrace/printer"
|
||||
"github.com/xgadget-lab/nexttrace/trace"
|
||||
"github.com/xgadget-lab/nexttrace/util"
|
||||
"log"
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/xgadget-lab/nexttrace/methods"
|
||||
"github.com/xgadget-lab/nexttrace/methods/tcp"
|
||||
"github.com/xgadget-lab/nexttrace/methods/udp"
|
||||
"github.com/xgadget-lab/nexttrace/util"
|
||||
"github.com/xgadget-lab/nexttrace/util/printer"
|
||||
)
|
||||
|
||||
var tcpSYNFlag = flag.Bool("T", false, "Use TCP SYN for tracerouting (default port is 80 in TCP, 53 in UDP)")
|
||||
@@ -22,61 +21,6 @@ var dataOrigin = flag.String("d", "LeoMoeAPI", "Choose IP Geograph Data Provider
|
||||
var displayMode = flag.String("displayMode", "table", "Choose The Display Mode [table, classic]")
|
||||
var rdnsenable = flag.Bool("rdns", false, "Set whether rDNS will be display")
|
||||
|
||||
func main() {
|
||||
printer.PrintCopyRight()
|
||||
domain := flagApply()
|
||||
ip := util.DomainLookUp(domain)
|
||||
printer.PrintTraceRouteNav(ip, domain, *dataOrigin)
|
||||
|
||||
if *tcpSYNFlag {
|
||||
tcpTraceroute := tcp.New(ip, methods.TracerouteConfig{
|
||||
MaxHops: uint16(*maxHops),
|
||||
NumMeasurements: uint16(*numMeasurements),
|
||||
ParallelRequests: uint16(*parallelRequests),
|
||||
Port: *port,
|
||||
Timeout: time.Second / 2,
|
||||
})
|
||||
res, err := tcpTraceroute.Start()
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("请赋予 sudo (root) 权限运行本程序")
|
||||
} else {
|
||||
util.Printer(&util.PrinterConfig{
|
||||
IP: ip,
|
||||
DisplayMode: *displayMode,
|
||||
DataOrigin: *dataOrigin,
|
||||
Rdnsenable: *rdnsenable,
|
||||
Results: *res,
|
||||
})
|
||||
}
|
||||
|
||||
} else {
|
||||
if *port == 80 {
|
||||
*port = 53
|
||||
}
|
||||
udpTraceroute := udp.New(ip, true, methods.TracerouteConfig{
|
||||
MaxHops: uint16(*maxHops),
|
||||
NumMeasurements: uint16(*numMeasurements),
|
||||
ParallelRequests: uint16(*parallelRequests),
|
||||
Port: *port,
|
||||
Timeout: 2 * time.Second,
|
||||
})
|
||||
res, err := udpTraceroute.Start()
|
||||
|
||||
if err != nil {
|
||||
fmt.Println("请赋予 sudo (root) 权限运行本程序")
|
||||
} else {
|
||||
util.Printer(&util.PrinterConfig{
|
||||
IP: ip,
|
||||
DisplayMode: *displayMode,
|
||||
DataOrigin: *dataOrigin,
|
||||
Rdnsenable: *rdnsenable,
|
||||
Results: *res,
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func flagApply() string {
|
||||
flag.Parse()
|
||||
ipArg := flag.Args()
|
||||
@@ -86,3 +30,49 @@ func flagApply() string {
|
||||
}
|
||||
return ipArg[0]
|
||||
}
|
||||
|
||||
func main() {
|
||||
if os.Getuid() != 0 {
|
||||
log.Fatalln("Traceroute requires root/sudo privileges.")
|
||||
}
|
||||
|
||||
domain := flagApply()
|
||||
ip := util.DomainLookUp(domain)
|
||||
printer.PrintTraceRouteNav(ip, domain, *dataOrigin)
|
||||
|
||||
var m trace.Method = ""
|
||||
if *tcpSYNFlag {
|
||||
m = trace.TCPTrace
|
||||
} else {
|
||||
m = trace.UDPTrace
|
||||
}
|
||||
|
||||
if !*tcpSYNFlag && *port == 80 {
|
||||
*port = 53
|
||||
}
|
||||
|
||||
var conf = trace.Config{
|
||||
DestIP: ip,
|
||||
DestPort: *port,
|
||||
MaxHops: *maxHops,
|
||||
NumMeasurements: *numMeasurements,
|
||||
ParallelRequests: *parallelRequests,
|
||||
RDns: *rdnsenable,
|
||||
IPGeoSource: ipgeo.GetSource(*dataOrigin),
|
||||
Timeout: 2 * time.Second,
|
||||
|
||||
//Quic: false,
|
||||
}
|
||||
|
||||
res, err := trace.Traceroute(m, conf)
|
||||
|
||||
if err != nil {
|
||||
log.Fatalln(err)
|
||||
}
|
||||
|
||||
if *displayMode == "table" {
|
||||
printer.TracerouteTablePrinter(res)
|
||||
} else {
|
||||
printer.TraceroutePrinter(res)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,8 +1,6 @@
|
||||
package methods
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
@@ -24,37 +22,6 @@ type TracerouteConfig struct {
|
||||
Timeout time.Duration
|
||||
}
|
||||
|
||||
func GetIPHeaderLength(data []byte) (int, error) {
|
||||
if len(data) < 1 {
|
||||
return 0, errors.New("received invalid IP header")
|
||||
}
|
||||
return int((data[0] & 0x0F) * 4), nil
|
||||
}
|
||||
|
||||
func GetICMPResponsePayload(data []byte) ([]byte, error) {
|
||||
length, err := GetIPHeaderLength(data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(data) < length {
|
||||
return nil, errors.New("length of packet too short")
|
||||
}
|
||||
|
||||
return data[length:], nil
|
||||
}
|
||||
|
||||
func GetUDPSrcPort(data []byte) uint16 {
|
||||
srcPortBytes := data[:2]
|
||||
srcPort := binary.BigEndian.Uint16(srcPortBytes)
|
||||
return srcPort
|
||||
}
|
||||
|
||||
func GetTCPSeq(data []byte) uint32 {
|
||||
seqBytes := data[4:8]
|
||||
return binary.BigEndian.Uint32(seqBytes)
|
||||
}
|
||||
|
||||
func ReduceFinalResult(preliminary map[uint16][]TracerouteHop, maxHops uint16, destIP net.IP) map[uint16][]TracerouteHop {
|
||||
// reduce the results to remove all hops after the first encounter to final destination
|
||||
finalResults := map[uint16][]TracerouteHop{}
|
||||
|
||||
@@ -133,11 +133,11 @@ func (tr *Traceroute) addToResult(ttl uint16, hop methods.TracerouteHop) {
|
||||
}
|
||||
|
||||
func (tr *Traceroute) handleICMPMessage(msg listener_channel.ReceivedMessage, data []byte) {
|
||||
header, err := methods.GetICMPResponsePayload(data)
|
||||
header, err := util.GetICMPResponsePayload(data)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
sequenceNumber := methods.GetTCPSeq(header)
|
||||
sequenceNumber := util.GetTCPSeq(header)
|
||||
val, ok := tr.results.inflightRequests.LoadAndDelete(sequenceNumber)
|
||||
if !ok {
|
||||
return
|
||||
|
||||
@@ -1,311 +0,0 @@
|
||||
package udp
|
||||
|
||||
import (
|
||||
"log"
|
||||
"math/rand"
|
||||
"net"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"github.com/google/gopacket"
|
||||
"github.com/google/gopacket/layers"
|
||||
"github.com/xgadget-lab/nexttrace/listener_channel"
|
||||
"github.com/xgadget-lab/nexttrace/methods"
|
||||
"github.com/xgadget-lab/nexttrace/methods/quic"
|
||||
"github.com/xgadget-lab/nexttrace/parallel_limiter"
|
||||
"github.com/xgadget-lab/nexttrace/signal"
|
||||
"github.com/xgadget-lab/nexttrace/taskgroup"
|
||||
"github.com/xgadget-lab/nexttrace/util"
|
||||
"golang.org/x/net/context"
|
||||
"golang.org/x/net/icmp"
|
||||
"golang.org/x/net/ipv4"
|
||||
)
|
||||
|
||||
type inflightData struct {
|
||||
icmpMsg chan<- net.Addr
|
||||
}
|
||||
|
||||
type opConfig struct {
|
||||
quic bool
|
||||
destIP net.IP
|
||||
wg *taskgroup.TaskGroup
|
||||
|
||||
icmpConn net.PacketConn
|
||||
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
}
|
||||
|
||||
type results struct {
|
||||
inflightRequests sync.Map
|
||||
|
||||
results map[uint16][]methods.TracerouteHop
|
||||
resultsMu sync.Mutex
|
||||
err error
|
||||
|
||||
concurrentRequests *parallel_limiter.ParallelLimiter
|
||||
reachedFinalHop *signal.Signal
|
||||
}
|
||||
|
||||
type Traceroute struct {
|
||||
trcrtConfig methods.TracerouteConfig
|
||||
opConfig opConfig
|
||||
results results
|
||||
}
|
||||
|
||||
func New(destIP net.IP, quic bool, config methods.TracerouteConfig) *Traceroute {
|
||||
return &Traceroute{
|
||||
opConfig: opConfig{
|
||||
quic: quic,
|
||||
destIP: destIP,
|
||||
},
|
||||
trcrtConfig: config,
|
||||
}
|
||||
}
|
||||
|
||||
func (tr *Traceroute) Start() (*map[uint16][]methods.TracerouteHop, error) {
|
||||
tr.opConfig.ctx, tr.opConfig.cancel = context.WithCancel(context.Background())
|
||||
|
||||
tr.results = results{
|
||||
inflightRequests: sync.Map{},
|
||||
concurrentRequests: parallel_limiter.New(int(tr.trcrtConfig.ParallelRequests)),
|
||||
results: map[uint16][]methods.TracerouteHop{},
|
||||
reachedFinalHop: signal.New(),
|
||||
}
|
||||
|
||||
var err error
|
||||
tr.opConfig.icmpConn, err = icmp.ListenPacket("ip4:icmp", "0.0.0.0")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return tr.start()
|
||||
}
|
||||
|
||||
func (tr *Traceroute) addToResult(ttl uint16, hop methods.TracerouteHop) {
|
||||
tr.results.resultsMu.Lock()
|
||||
defer tr.results.resultsMu.Unlock()
|
||||
if tr.results.results[ttl] == nil {
|
||||
tr.results.results[ttl] = []methods.TracerouteHop{}
|
||||
}
|
||||
|
||||
tr.results.results[ttl] = append(tr.results.results[ttl], hop)
|
||||
}
|
||||
|
||||
func (tr *Traceroute) getUDPConn(try int) (net.IP, int, net.PacketConn) {
|
||||
srcIP, _ := util.LocalIPPort(tr.opConfig.destIP)
|
||||
|
||||
var ipString string
|
||||
|
||||
if srcIP == nil {
|
||||
ipString = ""
|
||||
} else {
|
||||
ipString = srcIP.String()
|
||||
}
|
||||
|
||||
udpConn, err := net.ListenPacket("udp", ipString+":0")
|
||||
if err != nil {
|
||||
if try > 3 {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return tr.getUDPConn(try + 1)
|
||||
}
|
||||
|
||||
return srcIP, udpConn.LocalAddr().(*net.UDPAddr).Port, udpConn
|
||||
}
|
||||
|
||||
func (tr *Traceroute) sendMessage(ttl uint16) {
|
||||
srcIP, srcPort, udpConn := tr.getUDPConn(0)
|
||||
|
||||
var payload []byte
|
||||
if tr.opConfig.quic {
|
||||
payload = quic.GenerateWithRandomIds()
|
||||
} else {
|
||||
ipHeader := &layers.IPv4{
|
||||
SrcIP: srcIP,
|
||||
DstIP: tr.opConfig.destIP,
|
||||
Protocol: layers.IPProtocolTCP,
|
||||
TTL: uint8(ttl),
|
||||
}
|
||||
|
||||
udpHeader := &layers.UDP{
|
||||
SrcPort: layers.UDPPort(srcPort),
|
||||
DstPort: layers.UDPPort(tr.trcrtConfig.Port),
|
||||
}
|
||||
_ = udpHeader.SetNetworkLayerForChecksum(ipHeader)
|
||||
buf := gopacket.NewSerializeBuffer()
|
||||
opts := gopacket.SerializeOptions{
|
||||
ComputeChecksums: true,
|
||||
FixLengths: true,
|
||||
}
|
||||
if err := gopacket.SerializeLayers(buf, opts, udpHeader, gopacket.Payload("HAJSFJHKAJSHFKJHAJKFHKASHKFHHKAFKHFAHSJK")); err != nil {
|
||||
tr.results.err = err
|
||||
tr.opConfig.cancel()
|
||||
return
|
||||
}
|
||||
|
||||
payload = buf.Bytes()
|
||||
}
|
||||
|
||||
err := ipv4.NewPacketConn(udpConn).SetTTL(int(ttl))
|
||||
if err != nil {
|
||||
tr.results.err = err
|
||||
tr.opConfig.cancel()
|
||||
return
|
||||
}
|
||||
|
||||
icmpMsg := make(chan net.Addr, 1)
|
||||
udpMsg := make(chan net.Addr, 1)
|
||||
|
||||
start := time.Now()
|
||||
if _, err := udpConn.WriteTo(payload, &net.UDPAddr{IP: tr.opConfig.destIP, Port: tr.trcrtConfig.Port}); err != nil {
|
||||
tr.results.err = err
|
||||
tr.opConfig.cancel()
|
||||
return
|
||||
}
|
||||
|
||||
inflight := inflightData{
|
||||
icmpMsg: icmpMsg,
|
||||
}
|
||||
|
||||
tr.results.inflightRequests.Store(uint16(srcPort), inflight)
|
||||
|
||||
go func() {
|
||||
reply := make([]byte, 1500)
|
||||
_, peer, err := udpConn.ReadFrom(reply)
|
||||
if err != nil {
|
||||
// probably because we closed the connection
|
||||
return
|
||||
}
|
||||
udpMsg <- peer
|
||||
}()
|
||||
|
||||
select {
|
||||
case peer := <-icmpMsg:
|
||||
rtt := time.Since(start)
|
||||
if peer.(*net.IPAddr).IP.Equal(tr.opConfig.destIP) {
|
||||
tr.results.reachedFinalHop.Signal()
|
||||
}
|
||||
tr.addToResult(ttl, methods.TracerouteHop{
|
||||
Success: true,
|
||||
Address: peer,
|
||||
TTL: ttl,
|
||||
RTT: &rtt,
|
||||
})
|
||||
case peer := <-udpMsg:
|
||||
rtt := time.Since(start)
|
||||
ip := peer.(*net.UDPAddr).IP
|
||||
if ip.Equal(tr.opConfig.destIP) {
|
||||
tr.results.reachedFinalHop.Signal()
|
||||
}
|
||||
tr.addToResult(ttl, methods.TracerouteHop{
|
||||
Success: true,
|
||||
Address: &net.IPAddr{IP: ip},
|
||||
TTL: ttl,
|
||||
RTT: &rtt,
|
||||
})
|
||||
case <-time.After(tr.trcrtConfig.Timeout):
|
||||
tr.addToResult(ttl, methods.TracerouteHop{
|
||||
Success: false,
|
||||
Address: nil,
|
||||
TTL: ttl,
|
||||
RTT: nil,
|
||||
})
|
||||
}
|
||||
|
||||
tr.results.inflightRequests.Delete(uint16(srcPort))
|
||||
udpConn.Close()
|
||||
tr.results.concurrentRequests.Finished()
|
||||
tr.opConfig.wg.Done()
|
||||
}
|
||||
|
||||
func (tr *Traceroute) handleICMPMessage(msg listener_channel.ReceivedMessage, data []byte) {
|
||||
header, err := methods.GetICMPResponsePayload(data)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
srcPort := methods.GetUDPSrcPort(header)
|
||||
val, ok := tr.results.inflightRequests.LoadAndDelete(srcPort)
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
request := val.(inflightData)
|
||||
request.icmpMsg <- msg.Peer
|
||||
}
|
||||
|
||||
func (tr *Traceroute) icmpListener() {
|
||||
lc := listener_channel.New(tr.opConfig.icmpConn)
|
||||
|
||||
defer lc.Stop()
|
||||
|
||||
go lc.Start()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-tr.opConfig.ctx.Done():
|
||||
return
|
||||
case msg := <-lc.Messages:
|
||||
if msg.N == nil {
|
||||
continue
|
||||
}
|
||||
rm, err := icmp.ParseMessage(1, msg.Msg[:*msg.N])
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
continue
|
||||
}
|
||||
switch rm.Type {
|
||||
case ipv4.ICMPTypeTimeExceeded:
|
||||
body := rm.Body.(*icmp.TimeExceeded).Data
|
||||
tr.handleICMPMessage(msg, body)
|
||||
case ipv4.ICMPTypeDestinationUnreachable:
|
||||
body := rm.Body.(*icmp.DstUnreach).Data
|
||||
tr.handleICMPMessage(msg, body)
|
||||
default:
|
||||
log.Println("received icmp message of unknown type", rm.Type)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (tr *Traceroute) sendLoop() {
|
||||
rand.Seed(time.Now().UTC().UnixNano())
|
||||
|
||||
for ttl := uint16(1); ttl <= tr.trcrtConfig.MaxHops; ttl++ {
|
||||
select {
|
||||
case <-tr.results.reachedFinalHop.Chan():
|
||||
return
|
||||
default:
|
||||
}
|
||||
for i := 0; i < int(tr.trcrtConfig.NumMeasurements); i++ {
|
||||
select {
|
||||
case <-tr.opConfig.ctx.Done():
|
||||
return
|
||||
case <-tr.results.concurrentRequests.Start():
|
||||
tr.opConfig.wg.Add()
|
||||
go tr.sendMessage(ttl)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (tr *Traceroute) start() (*map[uint16][]methods.TracerouteHop, error) {
|
||||
go tr.icmpListener()
|
||||
|
||||
wg := taskgroup.New()
|
||||
tr.opConfig.wg = wg
|
||||
|
||||
tr.sendLoop()
|
||||
|
||||
wg.Wait()
|
||||
|
||||
tr.opConfig.cancel()
|
||||
tr.opConfig.icmpConn.Close()
|
||||
|
||||
if tr.results.err != nil {
|
||||
return nil, tr.results.err
|
||||
}
|
||||
|
||||
result := methods.ReduceFinalResult(tr.results.results, tr.trcrtConfig.MaxHops, tr.opConfig.destIP)
|
||||
|
||||
return &result, tr.results.err
|
||||
}
|
||||
79
printer/printer.go
Normal file
79
printer/printer.go
Normal file
@@ -0,0 +1,79 @@
|
||||
package printer
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/xgadget-lab/nexttrace/trace"
|
||||
"strings"
|
||||
|
||||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||||
)
|
||||
|
||||
var dataOrigin string
|
||||
|
||||
func TraceroutePrinter(res *trace.Result) {
|
||||
for i, hop := range res.Hops {
|
||||
fmt.Print(i + 1)
|
||||
for _, h := range hop {
|
||||
hopPrinter(h)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func hopPrinter(h trace.Hop) {
|
||||
if h.Address == nil {
|
||||
fmt.Println("\t*")
|
||||
} else {
|
||||
txt := "\t"
|
||||
|
||||
if h.Hostname == "" {
|
||||
txt += fmt.Sprint(h.Address, " ", fmt.Sprintf("%.2f", h.RTT.Seconds()*1000), "ms")
|
||||
} else {
|
||||
txt += fmt.Sprint(h.Hostname, " (", h.Address, ") ", fmt.Sprintf("%.2f", h.RTT.Seconds()*1000), "ms")
|
||||
}
|
||||
|
||||
if h.Geo != nil {
|
||||
txt += " " + formatIpGeoData(h.Address.String(), h.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: 判断阿里云和腾讯云内网，数据不足，有待进一步完善
|
||||
// TODO: 移动IDC判断到Hop.fetchIPData函数，减少API调用
|
||||
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 {
|
||||
if data.Owner == "" {
|
||||
data.Owner = data.Isp
|
||||
}
|
||||
if data.District != "" {
|
||||
data.City = data.City + ", " + data.District
|
||||
}
|
||||
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, ", ")
|
||||
}
|
||||
109
printer/tableprinter.go
Normal file
109
printer/tableprinter.go
Normal file
@@ -0,0 +1,109 @@
|
||||
package printer
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/xgadget-lab/nexttrace/trace"
|
||||
"strings"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/rodaine/table"
|
||||
)
|
||||
|
||||
type rowData struct {
|
||||
Hop string
|
||||
IP string
|
||||
Latency string
|
||||
Asnumber string
|
||||
Country string
|
||||
Prov string
|
||||
City string
|
||||
District string
|
||||
Owner string
|
||||
}
|
||||
|
||||
func TracerouteTablePrinter(res *trace.Result) {
|
||||
// 初始化表格
|
||||
tbl := New()
|
||||
for _, hop := range res.Hops {
|
||||
for k, h := range hop {
|
||||
data := tableDataGenerator(h)
|
||||
if k > 0 {
|
||||
data.Hop = ""
|
||||
}
|
||||
tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, data.Country, data.Prov, data.City, data.Owner)
|
||||
}
|
||||
}
|
||||
// 打印表格
|
||||
tbl.Print()
|
||||
}
|
||||
|
||||
func New() table.Table {
|
||||
// 初始化表格
|
||||
headerFmt := color.New(color.FgGreen, color.Underline).SprintfFunc()
|
||||
columnFmt := color.New(color.FgYellow).SprintfFunc()
|
||||
|
||||
tbl := table.New("Hop", "IP", "Lantency", "ASN", "Country", "Province", "City", "Owner")
|
||||
tbl.WithHeaderFormatter(headerFmt).WithFirstColumnFormatter(columnFmt)
|
||||
return tbl
|
||||
}
|
||||
|
||||
func tableDataGenerator(h trace.Hop) *rowData {
|
||||
if h.Address == nil {
|
||||
return &rowData{
|
||||
Hop: fmt.Sprint(h.TTL),
|
||||
IP: "*",
|
||||
}
|
||||
} else {
|
||||
lantency := fmt.Sprintf("%.2fms", h.RTT.Seconds()*1000)
|
||||
IP := h.Address.String()
|
||||
|
||||
if strings.HasPrefix(IP, "9.") {
|
||||
return &rowData{
|
||||
Hop: fmt.Sprint(h.TTL),
|
||||
IP: IP,
|
||||
Latency: lantency,
|
||||
Country: "局域网",
|
||||
Owner: "腾讯云",
|
||||
}
|
||||
} else if strings.HasPrefix(IP, "11.") {
|
||||
return &rowData{
|
||||
Hop: fmt.Sprint(h.TTL),
|
||||
IP: IP,
|
||||
Latency: lantency,
|
||||
Country: "局域网",
|
||||
Owner: "阿里云",
|
||||
}
|
||||
}
|
||||
|
||||
if h.Hostname != "" {
|
||||
IP = fmt.Sprint(h.Hostname, " (", IP, ") ")
|
||||
}
|
||||
|
||||
r := &rowData{
|
||||
Hop: fmt.Sprint(h.TTL),
|
||||
IP: IP,
|
||||
Latency: lantency,
|
||||
Asnumber: h.Geo.Asnumber,
|
||||
Country: h.Geo.Country,
|
||||
Prov: h.Geo.Prov,
|
||||
City: h.Geo.City,
|
||||
District: h.Geo.District,
|
||||
Owner: h.Geo.Owner,
|
||||
}
|
||||
|
||||
if h.Geo == nil {
|
||||
return r
|
||||
}
|
||||
|
||||
if h.Geo.Owner == "" {
|
||||
h.Geo.Owner = h.Geo.Isp
|
||||
}
|
||||
r.Asnumber = h.Geo.Asnumber
|
||||
r.Country = h.Geo.Country
|
||||
r.Prov = h.Geo.Prov
|
||||
r.City = h.Geo.City
|
||||
r.District = h.Geo.District
|
||||
r.Owner = h.Geo.Owner
|
||||
return r
|
||||
}
|
||||
}
|
||||
55
trace/packet_listener.go
Normal file
55
trace/packet_listener.go
Normal file
@@ -0,0 +1,55 @@
|
||||
package trace
|
||||
|
||||
import (
|
||||
"golang.org/x/net/context"
|
||||
"net"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ReceivedMessage struct {
|
||||
N *int
|
||||
Peer net.Addr
|
||||
Msg []byte
|
||||
Err error
|
||||
}
|
||||
|
||||
type PacketListener struct {
|
||||
ctx context.Context
|
||||
Conn net.PacketConn
|
||||
Messages chan ReceivedMessage
|
||||
}
|
||||
|
||||
func NewPacketListener(conn net.PacketConn, ctx context.Context) *PacketListener {
|
||||
results := make(chan ReceivedMessage, 50)
|
||||
|
||||
return &PacketListener{Conn: conn, ctx: ctx, Messages: results}
|
||||
}
|
||||
|
||||
func (l *PacketListener) Start() {
|
||||
for {
|
||||
select {
|
||||
case <-l.ctx.Done():
|
||||
return
|
||||
default:
|
||||
}
|
||||
|
||||
reply := make([]byte, 1500)
|
||||
err := l.Conn.SetReadDeadline(time.Now().Add(2 * time.Second))
|
||||
if err != nil {
|
||||
l.Messages <- ReceivedMessage{Err: err}
|
||||
continue
|
||||
}
|
||||
|
||||
n, peer, err := l.Conn.ReadFrom(reply)
|
||||
if err != nil {
|
||||
l.Messages <- ReceivedMessage{Err: err}
|
||||
continue
|
||||
}
|
||||
l.Messages <- ReceivedMessage{
|
||||
N: &n,
|
||||
Peer: peer,
|
||||
Err: nil,
|
||||
Msg: reply,
|
||||
}
|
||||
}
|
||||
}
|
||||
File diff suppressed because one or more lines are too long
105
trace/trace.go
Normal file
105
trace/trace.go
Normal file
@@ -0,0 +1,105 @@
|
||||
package trace
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||||
"net"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrInvalidMethod = errors.New("invalid method")
|
||||
ErrTracerouteExecuted = errors.New("traceroute already executed")
|
||||
ErrHopLimitTimeout = errors.New("hop timeout")
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
MaxHops int
|
||||
NumMeasurements int
|
||||
ParallelRequests int
|
||||
Timeout time.Duration
|
||||
DestIP net.IP
|
||||
DestPort int
|
||||
Quic bool
|
||||
IPGeoSource ipgeo.Source
|
||||
RDns bool
|
||||
}
|
||||
|
||||
type Method string
|
||||
|
||||
const (
|
||||
UDPTrace Method = "udp"
|
||||
TCPTrace Method = "tcp"
|
||||
)
|
||||
|
||||
type Tracer interface {
|
||||
Execute() (*Result, error)
|
||||
}
|
||||
|
||||
func Traceroute(method Method, config Config) (*Result, error) {
|
||||
var tracer Tracer
|
||||
|
||||
if config.MaxHops == 0 {
|
||||
config.MaxHops = 30
|
||||
}
|
||||
if config.NumMeasurements == 0 {
|
||||
config.NumMeasurements = 3
|
||||
}
|
||||
if config.ParallelRequests == 0 {
|
||||
config.ParallelRequests = config.NumMeasurements * 5
|
||||
}
|
||||
|
||||
switch method {
|
||||
case UDPTrace:
|
||||
tracer = &UDPTracer{Config: config}
|
||||
default:
|
||||
return &Result{}, ErrInvalidMethod
|
||||
}
|
||||
return tracer.Execute()
|
||||
}
|
||||
|
||||
type Result struct {
|
||||
Hops [][]Hop
|
||||
lock sync.Mutex
|
||||
}
|
||||
|
||||
func (s *Result) add(hop Hop) {
|
||||
s.lock.Lock()
|
||||
defer s.lock.Unlock()
|
||||
k := hop.TTL - 1
|
||||
for len(s.Hops) < hop.TTL {
|
||||
s.Hops = append(s.Hops, make([]Hop, 0))
|
||||
}
|
||||
s.Hops[k] = append(s.Hops[k], hop)
|
||||
|
||||
}
|
||||
|
||||
func (s *Result) reduce(final int) {
|
||||
if final > 0 && final < len(s.Hops) {
|
||||
s.Hops = s.Hops[:final]
|
||||
}
|
||||
}
|
||||
|
||||
type Hop struct {
|
||||
Success bool
|
||||
Address net.Addr
|
||||
Hostname string
|
||||
TTL int
|
||||
RTT time.Duration
|
||||
Error error
|
||||
Geo *ipgeo.IPGeoData
|
||||
}
|
||||
|
||||
func (h *Hop) fetchIPData(c Config) (err error) {
|
||||
if c.RDns && h.Hostname == "" {
|
||||
ptr, err := net.LookupAddr(h.Address.String())
|
||||
if err == nil && len(ptr) > 0 {
|
||||
h.Hostname = ptr[0]
|
||||
}
|
||||
}
|
||||
if c.IPGeoSource != nil && h.Geo == nil {
|
||||
h.Geo, err = c.IPGeoSource(h.Address.String())
|
||||
}
|
||||
return
|
||||
}
|
||||
256
trace/udp.go
Normal file
256
trace/udp.go
Normal file
@@ -0,0 +1,256 @@
|
||||
package trace
|
||||
|
||||
import (
|
||||
"github.com/google/gopacket"
|
||||
"github.com/google/gopacket/layers"
|
||||
"github.com/xgadget-lab/nexttrace/util"
|
||||
"golang.org/x/net/context"
|
||||
"golang.org/x/net/icmp"
|
||||
"golang.org/x/net/ipv4"
|
||||
"golang.org/x/sync/semaphore"
|
||||
"log"
|
||||
"net"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type UDPTracer struct {
|
||||
Config
|
||||
wg sync.WaitGroup
|
||||
res Result
|
||||
ctx context.Context
|
||||
inflightRequest map[int]chan Hop
|
||||
inflightRequestLock sync.Mutex
|
||||
|
||||
icmp net.PacketConn
|
||||
|
||||
final int
|
||||
finalLock sync.Mutex
|
||||
|
||||
sem *semaphore.Weighted
|
||||
}
|
||||
|
||||
func (t *UDPTracer) Execute() (*Result, error) {
|
||||
if len(t.res.Hops) > 0 {
|
||||
return &t.res, ErrTracerouteExecuted
|
||||
}
|
||||
|
||||
var err error
|
||||
t.icmp, err = icmp.ListenPacket("ip4:icmp", "0.0.0.0")
|
||||
if err != nil {
|
||||
return &t.res, err
|
||||
}
|
||||
defer t.icmp.Close()
|
||||
|
||||
var cancel context.CancelFunc
|
||||
t.ctx, cancel = context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
t.inflightRequest = make(map[int]chan Hop)
|
||||
t.final = -1
|
||||
|
||||
go t.listenICMP()
|
||||
|
||||
t.sem = semaphore.NewWeighted(int64(t.ParallelRequests))
|
||||
for ttl := 1; ttl <= t.MaxHops; ttl++ {
|
||||
for i := 0; i < t.NumMeasurements; i++ {
|
||||
t.wg.Add(1)
|
||||
go t.send(ttl)
|
||||
}
|
||||
}
|
||||
|
||||
t.wg.Wait()
|
||||
t.res.reduce(t.final)
|
||||
|
||||
return &t.res, nil
|
||||
}
|
||||
|
||||
func (t *UDPTracer) listenICMP() {
|
||||
lc := NewPacketListener(t.icmp, t.ctx)
|
||||
go lc.Start()
|
||||
for {
|
||||
select {
|
||||
case <-t.ctx.Done():
|
||||
return
|
||||
case msg := <-lc.Messages:
|
||||
if msg.N == nil {
|
||||
continue
|
||||
}
|
||||
rm, err := icmp.ParseMessage(1, msg.Msg[:*msg.N])
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
continue
|
||||
}
|
||||
switch rm.Type {
|
||||
case ipv4.ICMPTypeTimeExceeded:
|
||||
t.handleICMPMessage(msg, rm.Body.(*icmp.TimeExceeded).Data)
|
||||
case ipv4.ICMPTypeDestinationUnreachable:
|
||||
t.handleICMPMessage(msg, rm.Body.(*icmp.DstUnreach).Data)
|
||||
default:
|
||||
log.Println("received icmp message of unknown type", rm.Type)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func (t *UDPTracer) handleICMPMessage(msg ReceivedMessage, data []byte) {
|
||||
header, err := util.GetICMPResponsePayload(data)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
srcPort := util.GetUDPSrcPort(header)
|
||||
t.inflightRequestLock.Lock()
|
||||
defer t.inflightRequestLock.Unlock()
|
||||
ch, ok := t.inflightRequest[int(srcPort)]
|
||||
if !ok {
|
||||
return
|
||||
}
|
||||
ch <- Hop{
|
||||
Success: true,
|
||||
Address: msg.Peer,
|
||||
}
|
||||
}
|
||||
|
||||
func (t *UDPTracer) getUDPConn(try int) (net.IP, int, net.PacketConn) {
|
||||
srcIP, _ := util.LocalIPPort(t.DestIP)
|
||||
|
||||
var ipString string
|
||||
if srcIP == nil {
|
||||
ipString = ""
|
||||
} else {
|
||||
ipString = srcIP.String()
|
||||
}
|
||||
|
||||
udpConn, err := net.ListenPacket("udp", ipString+":0")
|
||||
if err != nil {
|
||||
if try > 3 {
|
||||
log.Fatal(err)
|
||||
}
|
||||
return t.getUDPConn(try + 1)
|
||||
}
|
||||
|
||||
return srcIP, udpConn.LocalAddr().(*net.UDPAddr).Port, udpConn
|
||||
}
|
||||
|
||||
func (t *UDPTracer) send(ttl int) error {
|
||||
err := t.sem.Acquire(context.Background(), 1)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer t.sem.Release(1)
|
||||
|
||||
defer t.wg.Done()
|
||||
if t.final != -1 && ttl > t.final {
|
||||
return nil
|
||||
}
|
||||
|
||||
srcIP, srcPort, udpConn := t.getUDPConn(0)
|
||||
|
||||
var payload []byte
|
||||
if t.Quic {
|
||||
payload = GenerateQuicPayloadWithRandomIds()
|
||||
} else {
|
||||
ipHeader := &layers.IPv4{
|
||||
SrcIP: srcIP,
|
||||
DstIP: t.DestIP,
|
||||
Protocol: layers.IPProtocolTCP,
|
||||
TTL: uint8(ttl),
|
||||
}
|
||||
|
||||
udpHeader := &layers.UDP{
|
||||
SrcPort: layers.UDPPort(srcPort),
|
||||
DstPort: layers.UDPPort(t.DestPort),
|
||||
}
|
||||
_ = udpHeader.SetNetworkLayerForChecksum(ipHeader)
|
||||
buf := gopacket.NewSerializeBuffer()
|
||||
opts := gopacket.SerializeOptions{
|
||||
ComputeChecksums: true,
|
||||
FixLengths: true,
|
||||
}
|
||||
if err := gopacket.SerializeLayers(buf, opts, udpHeader, gopacket.Payload("HAJSFJHKAJSHFKJHAJKFHKASHKFHHKAFKHFAHSJK")); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
payload = buf.Bytes()
|
||||
}
|
||||
|
||||
err = ipv4.NewPacketConn(udpConn).SetTTL(ttl)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
start := time.Now()
|
||||
if _, err := udpConn.WriteTo(payload, &net.UDPAddr{IP: t.DestIP, Port: t.DestPort}); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
t.inflightRequestLock.Lock()
|
||||
hopCh := make(chan Hop)
|
||||
t.inflightRequest[srcPort] = hopCh
|
||||
t.inflightRequestLock.Unlock()
|
||||
defer func() {
|
||||
t.inflightRequestLock.Lock()
|
||||
close(hopCh)
|
||||
delete(t.inflightRequest, srcPort)
|
||||
t.inflightRequestLock.Unlock()
|
||||
}()
|
||||
|
||||
go func() {
|
||||
reply := make([]byte, 1500)
|
||||
_, peer, err := udpConn.ReadFrom(reply)
|
||||
if err != nil {
|
||||
// probably because we closed the connection
|
||||
return
|
||||
}
|
||||
hopCh <- Hop{
|
||||
Success: true,
|
||||
Address: peer,
|
||||
}
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-t.ctx.Done():
|
||||
return nil
|
||||
case h := <-hopCh:
|
||||
rtt := time.Since(start)
|
||||
if t.final != -1 && ttl > t.final {
|
||||
return nil
|
||||
}
|
||||
|
||||
if addr, ok := h.Address.(*net.IPAddr); ok && addr.IP.Equal(t.DestIP) {
|
||||
t.finalLock.Lock()
|
||||
if t.final == -1 || ttl < t.final {
|
||||
t.final = ttl
|
||||
}
|
||||
t.finalLock.Unlock()
|
||||
} else if addr, ok := h.Address.(*net.UDPAddr); ok && addr.IP.Equal(t.DestIP) {
|
||||
t.finalLock.Lock()
|
||||
if t.final == -1 || ttl < t.final {
|
||||
t.final = ttl
|
||||
}
|
||||
t.finalLock.Unlock()
|
||||
}
|
||||
|
||||
h.TTL = ttl
|
||||
h.RTT = rtt
|
||||
|
||||
h.fetchIPData(t.Config)
|
||||
|
||||
t.res.add(h)
|
||||
|
||||
case <-time.After(t.Timeout):
|
||||
if t.final != -1 && ttl > t.final {
|
||||
return nil
|
||||
}
|
||||
|
||||
t.res.add(Hop{
|
||||
Success: false,
|
||||
Address: nil,
|
||||
TTL: ttl,
|
||||
RTT: 0,
|
||||
Error: ErrHopLimitTimeout,
|
||||
})
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
@@ -1,39 +0,0 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"net"
|
||||
|
||||
"github.com/xgadget-lab/nexttrace/methods"
|
||||
"github.com/xgadget-lab/nexttrace/util/printer"
|
||||
)
|
||||
|
||||
type IPGeoData struct {
|
||||
Asnumber string `json:"asnumber"`
|
||||
Country string `json:"country"`
|
||||
Prov string `json:"prov"`
|
||||
City string `json:"city"`
|
||||
District string `json:"district"`
|
||||
Owner string `json:"owner"`
|
||||
Isp string `json:"isp"`
|
||||
}
|
||||
|
||||
type PrinterConfig struct {
|
||||
IP net.IP
|
||||
DataOrigin string
|
||||
DisplayMode string
|
||||
Rdnsenable bool
|
||||
Results map[uint16][]methods.TracerouteHop
|
||||
}
|
||||
|
||||
func Printer(config *PrinterConfig) {
|
||||
switch config.DisplayMode {
|
||||
case "table":
|
||||
printer.TracerouteTablePrinter(config.IP, config.Results, config.DataOrigin, config.Rdnsenable)
|
||||
case "classic":
|
||||
printer.TraceroutePrinter(config.IP, config.Results, config.DataOrigin, config.Rdnsenable)
|
||||
case "json":
|
||||
//TracerouteJSONPrinter(config.Results, config.DataOrigin)
|
||||
default:
|
||||
printer.TraceroutePrinter(config.IP, config.Results, config.DataOrigin, config.Rdnsenable)
|
||||
}
|
||||
}
|
||||
@@ -1,109 +0,0 @@
|
||||
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 {
|
||||
if data.Owner == "" {
|
||||
data.Owner = data.Isp
|
||||
}
|
||||
if data.District != "" {
|
||||
data.City = data.City + ", " + data.District
|
||||
}
|
||||
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, ", ")
|
||||
}
|
||||
@@ -1,137 +0,0 @@
|
||||
package printer
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net"
|
||||
"strings"
|
||||
|
||||
"github.com/fatih/color"
|
||||
"github.com/rodaine/table"
|
||||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||||
"github.com/xgadget-lab/nexttrace/methods"
|
||||
)
|
||||
|
||||
type rowData struct {
|
||||
Hop int64
|
||||
IP string
|
||||
Latency string
|
||||
Asnumber string
|
||||
Country string
|
||||
Prov string
|
||||
City string
|
||||
District string
|
||||
Owner string
|
||||
}
|
||||
|
||||
func TracerouteTablePrinter(ip net.IP, res map[uint16][]methods.TracerouteHop, dataOrigin string, rdnsenable bool) {
|
||||
// 初始化表格
|
||||
tbl := New()
|
||||
for hi := uint16(1); hi < 30; hi++ {
|
||||
for _, v := range res[hi] {
|
||||
data := tableDataGenerator(v, rdnsenable)
|
||||
tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, data.Country, data.Prov, data.City, data.Owner)
|
||||
if v.Address != nil && ip.String() == v.Address.String() {
|
||||
hi = 31
|
||||
}
|
||||
}
|
||||
}
|
||||
// 打印表格
|
||||
tbl.Print()
|
||||
}
|
||||
|
||||
func New() table.Table {
|
||||
// 初始化表格
|
||||
headerFmt := color.New(color.FgGreen, color.Underline).SprintfFunc()
|
||||
columnFmt := color.New(color.FgYellow).SprintfFunc()
|
||||
|
||||
tbl := table.New("Hop", "IP", "Lantency", "ASN", "Country", "Province", "City", "Owner")
|
||||
tbl.WithHeaderFormatter(headerFmt).WithFirstColumnFormatter(columnFmt)
|
||||
return tbl
|
||||
}
|
||||
|
||||
func tableDataGenerator(v2 methods.TracerouteHop, rdnsenable bool) *rowData {
|
||||
if v2.Address == nil {
|
||||
return &rowData{
|
||||
Hop: int64(v2.TTL),
|
||||
IP: "*",
|
||||
}
|
||||
} else {
|
||||
// 初始化变量
|
||||
var iPGeoData *ipgeo.IPGeoData
|
||||
var err error
|
||||
var lantency, IP string
|
||||
|
||||
ipStr := v2.Address.String()
|
||||
|
||||
if strings.HasPrefix(ipStr, "9.") {
|
||||
return &rowData{
|
||||
Hop: int64(v2.TTL),
|
||||
IP: ipStr,
|
||||
Latency: lantency,
|
||||
Country: "局域网",
|
||||
Owner: "腾讯云",
|
||||
}
|
||||
} else if strings.HasPrefix(ipStr, "11.") {
|
||||
return &rowData{
|
||||
Hop: int64(v2.TTL),
|
||||
IP: ipStr,
|
||||
Latency: lantency,
|
||||
Country: "局域网",
|
||||
Owner: "阿里云",
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: 判断 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)
|
||||
}
|
||||
|
||||
if rdnsenable {
|
||||
ptr, err_LookupAddr := net.LookupAddr(ipStr)
|
||||
if err_LookupAddr != nil {
|
||||
IP = fmt.Sprint(ipStr)
|
||||
} else {
|
||||
IP = fmt.Sprint(ptr[0], " (", ipStr, ") ")
|
||||
}
|
||||
} else {
|
||||
IP = fmt.Sprint(ipStr)
|
||||
}
|
||||
|
||||
lantency = fmt.Sprintf("%.2fms", v2.RTT.Seconds()*1000)
|
||||
|
||||
if iPGeoData.Owner == "" {
|
||||
iPGeoData.Owner = iPGeoData.Isp
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
fmt.Print("Error: ", err)
|
||||
return &rowData{
|
||||
Hop: int64(v2.TTL),
|
||||
IP: IP,
|
||||
Latency: lantency,
|
||||
}
|
||||
} else {
|
||||
|
||||
return &rowData{
|
||||
Hop: int64(v2.TTL),
|
||||
IP: IP,
|
||||
Latency: lantency,
|
||||
Asnumber: iPGeoData.Asnumber,
|
||||
Country: iPGeoData.Country,
|
||||
Prov: iPGeoData.Prov,
|
||||
City: iPGeoData.City,
|
||||
District: iPGeoData.District,
|
||||
Owner: iPGeoData.Owner,
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
37
util/trace.go
Normal file
37
util/trace.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package util
|
||||
|
||||
import (
|
||||
"encoding/binary"
|
||||
"errors"
|
||||
)
|
||||
|
||||
func GetIPHeaderLength(data []byte) (int, error) {
|
||||
if len(data) < 1 {
|
||||
return 0, errors.New("received invalid IP header")
|
||||
}
|
||||
return int((data[0] & 0x0F) * 4), nil
|
||||
}
|
||||
|
||||
func GetICMPResponsePayload(data []byte) ([]byte, error) {
|
||||
length, err := GetIPHeaderLength(data)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(data) < length {
|
||||
return nil, errors.New("length of packet too short")
|
||||
}
|
||||
|
||||
return data[length:], nil
|
||||
}
|
||||
|
||||
func GetUDPSrcPort(data []byte) uint16 {
|
||||
srcPortBytes := data[:2]
|
||||
srcPort := binary.BigEndian.Uint16(srcPortBytes)
|
||||
return srcPort
|
||||
}
|
||||
|
||||
func GetTCPSeq(data []byte) uint32 {
|
||||
seqBytes := data[4:8]
|
||||
return binary.BigEndian.Uint32(seqBytes)
|
||||
}
|
||||
Reference in New Issue
Block a user