mirror of
https://github.com/nxtrace/NTrace-core.git
synced 2025-08-12 06:26:39 +00:00
130 lines
3.3 KiB
Go
130 lines
3.3 KiB
Go
package main
|
||
|
||
import (
|
||
"flag"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/xgadget-lab/nexttrace/ipgeo"
|
||
"github.com/xgadget-lab/nexttrace/printer"
|
||
"github.com/xgadget-lab/nexttrace/reporter"
|
||
"github.com/xgadget-lab/nexttrace/trace"
|
||
"github.com/xgadget-lab/nexttrace/util"
|
||
)
|
||
|
||
var fSet = flag.NewFlagSet("", flag.ExitOnError)
|
||
var tcpSYNFlag = fSet.Bool("T", false, "Use TCP SYN for tracerouting (default port is 80)")
|
||
var udpPackageFlag = fSet.Bool("U", false, "Use UDP Package for tracerouting (default port is 53 in UDP)")
|
||
var port = flag.Int("p", 80, "Set SYN Traceroute Port")
|
||
var numMeasurements = fSet.Int("q", 3, "Set the number of probes per each hop.")
|
||
var parallelRequests = fSet.Int("r", 18, "Set ParallelRequests number. It should be 1 when there is a multi-routing.")
|
||
var maxHops = fSet.Int("m", 30, "Set the max number of hops (max TTL to be reached).")
|
||
var dataOrigin = fSet.String("d", "LeoMoeAPI", "Choose IP Geograph Data Provider [LeoMoeAPI, IP.SB, IPInfo, IPInsight, IPAPI.com]")
|
||
var rdnsenable = fSet.Bool("rdns", false, "Set whether rDNS will be display")
|
||
var routePath = fSet.Bool("report", false, "Route Path")
|
||
var realtimePrint = fSet.Bool("realtime", false, "Output trace results in runtime")
|
||
var tablePrint = fSet.Bool("table", false, "Output trace results as table")
|
||
var ver = fSet.Bool("V", false, "Check Version")
|
||
|
||
func printArgHelp() {
|
||
fmt.Println("\nArgs Error\nUsage : 'nexttrace [option...] HOSTNAME' or 'nexttrace HOSTNAME [option...]'\nOPTIONS: [-VTU] [-d DATAORIGIN.STR ] [ -m TTL ] [ -p PORT ] [ -q PROBES.COUNT ] [ -r PARALLELREQUESTS.COUNT ] [-rdns] [ -realtime | -table ] -report")
|
||
fSet.PrintDefaults()
|
||
os.Exit(2)
|
||
}
|
||
|
||
func flagApply() string {
|
||
printer.Version()
|
||
|
||
target := ""
|
||
if len(os.Args) < 2 {
|
||
printArgHelp()
|
||
}
|
||
if !strings.HasPrefix(os.Args[1], "-") {
|
||
target = os.Args[1]
|
||
fSet.Parse(os.Args[2:])
|
||
} else {
|
||
fSet.Parse(os.Args[1:])
|
||
target = fSet.Arg(0)
|
||
}
|
||
if *ver {
|
||
os.Exit(0)
|
||
}
|
||
if target == "" {
|
||
printArgHelp()
|
||
}
|
||
return target
|
||
}
|
||
|
||
func main() {
|
||
|
||
domain := flagApply()
|
||
|
||
if os.Getuid() != 0 {
|
||
log.Fatalln("Traceroute requires root/sudo privileges.")
|
||
}
|
||
|
||
ip := util.DomainLookUp(domain)
|
||
printer.PrintTraceRouteNav(ip, domain, *dataOrigin)
|
||
|
||
var m trace.Method = ""
|
||
|
||
switch {
|
||
case *tcpSYNFlag:
|
||
m = trace.TCPTrace
|
||
case *udpPackageFlag:
|
||
m = trace.UDPTrace
|
||
default:
|
||
m = trace.ICMPTrace
|
||
}
|
||
|
||
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,
|
||
}
|
||
|
||
if m == trace.ICMPTrace && !*tablePrint {
|
||
conf.RealtimePrinter = printer.RealtimePrinter
|
||
}
|
||
|
||
res, err := trace.Traceroute(m, conf)
|
||
|
||
if err != nil {
|
||
log.Fatalln(err)
|
||
}
|
||
|
||
if *routePath {
|
||
// 如果为TCP SYN，也打印路由跟踪结果
|
||
if *tcpSYNFlag {
|
||
printer.TracerouteTablePrinter(res)
|
||
}
|
||
r := reporter.New(res, ip.String())
|
||
r.Print()
|
||
return
|
||
}
|
||
|
||
if m == trace.ICMPTrace && *tablePrint {
|
||
printer.TracerouteTablePrinter(res)
|
||
}
|
||
|
||
if m == trace.TCPTrace || m == trace.UDPTrace {
|
||
if *realtimePrint {
|
||
printer.TraceroutePrinter(res)
|
||
} else {
|
||
printer.TracerouteTablePrinter(res)
|
||
}
|
||
}
|
||
}
|