add: 新增配置文件模块

fix some bugs

.cross_compile.sh

@@ -5,7 +5,11 @@ set -e
 DIST_PREFIX="nexttrace"
 DEBUG_MODE=${2}
 TARGET_DIR="dist"
-PLATFORMS="darwin/amd64 darwin/arm64 linux/amd64 linux/arm64"
+PLATFORMS="darwin/amd64 darwin/arm64 linux/386 linux/amd64 linux/arm64 linux/mips openbsd/amd64 openbsd/arm64 freebsd/amd64 freebsd/arm64"
+
+BUILD_VERSION="$(git describe --tags --always)"
+BUILD_DATE="$(date -u +'%Y-%m-%dT%H:%M:%SZ')"
+COMMIT_SHA1="$(git rev-parse --short HEAD)"
 
 rm -rf ${TARGET_DIR}
 mkdir ${TARGET_DIR}
@@ -21,16 +25,36 @@ for pl in ${PLATFORMS}; do
     echo "build => ${TARGET}"
     if [ "${DEBUG_MODE}" == "debug" ]; then
         go build -trimpath -gcflags "all=-N -l" -o ${TARGET} \
-            -ldflags    "-X 'main.version=${BUILD_VERSION}' \
-                        -X 'main.buildDate=${BUILD_DATE}' \
-                        -X 'main.commitID=${COMMIT_SHA1}'\
+            -ldflags    "-X 'github.com/xgadget-lab/nexttrace/printer.version=${BUILD_VERSION}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.buildDate=${BUILD_DATE}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.commitID=${COMMIT_SHA1}'\
                         -w -s"
     else
         go build -trimpath -o ${TARGET} \
-            -ldflags    "-X 'main.version=${BUILD_VERSION}' \
-                        -X 'main.buildDate=${BUILD_DATE}' \
-                        -X 'main.commitID=${COMMIT_SHA1}'\
+            -ldflags    "-X 'github.com/xgadget-lab/nexttrace/printer.version=${BUILD_VERSION}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.buildDate=${BUILD_DATE}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.commitID=${COMMIT_SHA1}'\
                         -w -s"
     fi
 done
 
+    export GOOS='linux'
+    export GOARCH='arm'
+    export GOARM='7'
+    export TARGET=${TARGET_DIR}/${DIST_PREFIX}_${GOOS}_${GOARCH}v7
+    echo "build => ${TARGET}"
+    if [ "${DEBUG_MODE}" == "debug" ]; then
+        go build -trimpath -gcflags "all=-N -l" -o ${TARGET} \
+            -ldflags    "-X 'github.com/xgadget-lab/nexttrace/printer.version=${BUILD_VERSION}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.buildDate=${BUILD_DATE}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.commitID=${COMMIT_SHA1}'\
+                        -w -s"
+    else
+        go build -trimpath -o ${TARGET} \
+            -ldflags    "-X 'github.com/xgadget-lab/nexttrace/printer.version=${BUILD_VERSION}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.buildDate=${BUILD_DATE}' \
+                        -X 'github.com/xgadget-lab/nexttrace/printer.commitID=${COMMIT_SHA1}'\
+                        -w -s"
+    fi
+
+

.github/ISSUE_TEMPLATE/bug_report.md

@@ -0,0 +1,40 @@
+---
+name: Bug report
+about: Create a report to help us improve
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+---
+name: nexttrace 程序问题
+about: "提交一个 nexttrace 的程序问题报告。"
+copyright: [v2fly](https://github.com/v2fly)
+---
+
+<!--
+除非特殊情况，请完整填写所有问题。不按模板发的 issue 将直接被关闭。
+如果你遇到的问题不是 nexttrace 的 bug，比如你不清楚如何配置，请在 https://github.com/xgadget-lab/nexttrace/discussions 进行讨论。
+-->
+
+## 你正在使用哪个版本的 nexttrace？
+
+<!-- 比如linux_amd64 macOS_arm64 -->
+
+
+## 你看到的异常现象是什么？
+
+<!-- 请描述具体现象 -->
+
+
+## 你期待看到的正常表现是怎样的？
+
+
+
+## 请附上你的命令
+
+<!-- 提交 issue 前，请隐去您的隐私信息 -->
+
+
+## 请附上出错时软件输出的错误信息

.github/ISSUE_TEMPLATE/feature_request.md

@@ -0,0 +1,20 @@
+---
+name: Feature request
+about: Suggest an idea for this project
+title: ''
+labels: ''
+assignees: ''
+
+---
+
+**Is your feature request related to a problem? Please describe.**
+A clear and concise description of what the problem is. Ex. I'm always frustrated when [...]
+
+**Describe the solution you'd like**
+A clear and concise description of what you want to happen.
+
+**Describe alternatives you've considered**
+A clear and concise description of any alternative solutions or features you've considered.
+
+**Additional context**
+Add any other context or screenshots about the feature request here.

.github/workflows/build.yml

@@ -1,16 +1,31 @@
 on:
-  push: # 每次 push 的时候触发
+  push:
+  pull_request:
 
-name: Build Release
+name: Test & Build Release
 jobs:
-  release:
-    if: startsWith(github.ref, 'refs/tags/') # 只有这次 Commit 是 创建 Tag 时，才进行后续发布操作
+  Test:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@master # checkout 代码
-      - uses: actions/setup-go@v2 # 配置 Go 环境
+      - uses: actions/checkout@v3
+
+      - uses: actions/setup-go@v2
         with:
-          go-version: "1.18" # 改成自己的版本
+          go-version: "1.18"
+
+      - name: Test
+        run: go test -v -coverprofile='coverage.out' -covermode=count ./...
+
+  Build:
+    needs: test
+    if: startsWith(github.ref, 'refs/tags/v')
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+
+      - uses: actions/setup-go@v2
+        with:
+          go-version: "1.18"
 
       - run: bash .cross_compile.sh
 
@@ -21,7 +36,14 @@ jobs:
           files: |
             dist/nexttrace_darwin_amd64
             dist/nexttrace_darwin_arm64
+            dist/nexttrace_linux_386
             dist/nexttrace_linux_amd64
             dist/nexttrace_linux_arm64
+            dist/nexttrace_linux_armv7
+            dist/nexttrace_linux_mips
+            dist/nexttrace_openbsd_amd64
+            dist/nexttrace_openbsd_arm64
+            dist/nexttrace_freebsd_amd64
+            dist/nexttrace_freebsd_arm64
         env:
-          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          GITHUB_TOKEN: ${{ secrets.GT_Token }}

README.md

@@ -1,2 +1,201 @@
-# traceroute
-可视化路由跟踪工具
+<div align="center">
+
+<img src="asset/logo.png" height="200px" alt="NextTrace Logo"/>
+
+</div>
+
+# NextTrace
+
+一款开源的可视化路由跟踪工具，使用 Golang 开发。
+
+## How To Use
+
+### Automated Install
+
+```bash
+# Note: This Script Only Supports Linux/macOS, Other Unix-Like Systems are UNAVAILABLE
+curl -Ls https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh -O && sudo bash nt_install.sh
+```
+
+* `Release`里面为很多系统以及不同架构提供了编译好的二进制可执行文件，如果没有可以自行编译。
+* 一些本项目的必要依赖在`Windows`上`Golang`底层实现不完全，所以目前`NextTrace`在`Windows`平台不可用。
+
+### Fast Test
+
+此脚本旨在快速测试服务器的到中国内地的线路，建议新手或者没有自定义`NextTrace`功能需求的用户使用。
+
+```bash
+curl -Ls https://github.com/xgadget-lab/nexttrace/raw/main/quicklytest.sh -O && sudo bash quicklytest.sh
+```
+
+### Get Started
+
+`NextTrace`默认使用`ICMP`协议发起`TraceRoute`请求，该协议同时支持`IPv4`和`IPv6`
+
+```bash
+# IPv4 ICMP Trace
+nexttrace 1.0.0.1
+
+# 表格打印（一次性输出全部跳数，需等待20-40秒）
+nexttrace -table 1.0.0.1
+
+# IPv6 ICMP Trace
+nexttrace 2606:4700:4700::1111
+```
+
+`NextTrace`也可以使用`TCP`和`UDP`协议发起`Traceroute`请求，不过目前只支持`IPv4`
+
+```bash
+# TCP SYN Trace
+nexttrace -T www.bing.com
+
+# 可以自行指定端口[此处为443]，默认80端口
+nexttrace -T -p 443 1.0.0.1
+
+# UDP Trace
+nexttrace -U 1.0.0.1
+
+nexttrace -U -p 53 1.0.0.1
+```
+
+`NextTrace`也同样支持一些进阶功能，如 IP 反向解析、并发数控制、模式切换等
+
+```bash
+# 每一跳发送2个探测包
+nexttrace -q 2 www.hkix.net
+
+# 无并发，每次只发送一个探测包
+nexttrace -r 1 www.hkix.net
+
+# 打开IP反向解析功能，在IPv6的骨干网定位辅助有较大帮助
+nexttrace -rdns www.bbix.net
+
+# 特色功能：打印Route-Path图
+# Route-Path图示例：
+# AS6453 塔塔通信「Singapore『Singapore』」
+#  ╭╯
+#  ╰AS9299 Philippine Long Distance Telephone Co.「Philippines『Metro Manila』」
+#  ╭╯
+#  ╰AS36776 Five9 Inc.「Philippines『Metro Manila』」
+#  ╭╯
+#  ╰AS37963 阿里云「ALIDNS.COM『ALIDNS.COM』」
+nexttrace -report www.time.com.my
+```
+
+`NextTrace`支持用户自主选择 IP 数据库（目前支持：`LeoMoeAPI`, `IP.SB`, `IPInfo`, `IPInsight`, `IPAPI.com`）
+
+```bash
+# 可以自行指定IP数据库[此处为IP.SB]，不指定则默认为LeoMoeAPI
+nexttrace -d IP.SB
+## 特别的：其中 ipinfo API 需要从ipinfo自行购买服务，如有需要可以clone本项目添加其提供的token自行编译
+##        TOKEN填写路径：ipgeo/tokens.go
+## 另外：由于IP.SB被滥用比较严重，会经常出现无法查询的问题，请知悉。
+##      IPAPI.com限制调用较为严格，如有查询不到的情况，请几分钟后再试。
+```
+
+`NextTrace`支持参数混合使用
+
+```bash
+Example:
+nexttrace -d IPInsight -m 20 -p 443 -q 5 -r 20 -rdns 1.1.1.1
+nexttrace -T -q 2 -r 1 -rdns -table -report 2001:4860:4860::8888
+```
+
+### IP 数据库
+
+目前使用的 IP 数据库默认为我们自己搭建的 API 服务，如果后期遇到滥用，我们可能会选择关闭。
+
+我们也会在后期开放服务端源代码，您也可以根据该项目的源码自行搭建属于您的 API 服务器。
+
+NextTrace 所有的的 IP 地理位置`API DEMO`可以参考[这里](https://github.com/xgadget-lab/nexttrace/blob/main/ipgeo/)
+
+### 全部用法详见 Usage 菜单
+
+```shell
+Usage of nexttrace:
+      'nexttrace [options] <hostname>' or 'nexttrace <hostname> [option...]'
+Options:
+  -T    Use TCP SYN for tracerouting (default port is 80)
+  -U    Use UDP Package for tracerouting (default port is 53 in UDP)
+  -V    Check Version
+  -d string
+        Choose IP Geograph Data Provider [LeoMoeAPI, IP.SB, IPInfo, IPInsight, IPAPI.com] (default "LeoMoeAPI")
+  -m int
+        Set the max number of hops (max TTL to be reached). (default 30)
+  -p int
+        Set SYN Traceroute Port (default 80)
+  -q int
+        Set the number of probes per each hop. (default 3)
+  -r int
+        Set ParallelRequests number. It should be 1 when there is a multi-routing. (default 18)
+  -rdns
+        Set whether rDNS will be display
+  -table
+        Output trace results as table
+  -report
+        Route Path
+
+```
+
+## 项目截图
+
+<img src=asset/screenshot.png alt="NextTrace Screenshot" width="683" height="688" />
+
+## FAQ 常见问题
+
+如果你在安装或者使用的时候遇到了问题，我们建议你不要把新建一个 `issue` 作为首选项
+
+以下是我们推荐的排错流程：
+
+1. 查看是否为常见问题 -> [前往 Github Wiki](https://github.com/xgadget-lab/nexttrace/wiki/FAQ---%E5%B8%B8%E8%A7%81%E9%97%AE%E9%A2%98%E8%A7%A3%E7%AD%94)
+2. 不会软件的参数使用 -> [前往 Github Discussions](https://github.com/xgadget-lab/nexttrace/discussions)
+3. 疑似BUG、或者功能建议 -> [前往 Github Issues](https://github.com/xgadget-lab/nexttrace/issues)
+
+## Thanks
+
+[Vincent Young](https://github.com/missuo) (i@yyt.moe)
+
+[Sam Sam](https://github.com/samleong123) (samsam123@samsam123.name.my)
+
+[tsosunchia](https://github.com/tsosunchia)
+
+[waiting4new](https://github.com/waiting4new)
+
+[FFEE_CO](https://github.com/fkx4-p)
+
+## IP Database Copyright
+
+### IPv4 Database
+
+#### China
+
+|      ISP       |  类型  |      数据源      | 占比 |
+| :------------: | :----: | :--------------: | :--: |
+| 电信/联通/移动 | 骨干网 | NextTrace |  10%  |
+| 电信/联通/移动 | 城域网 |     埃文科技     | 90%  |
+
+#### WorldWide
+
+|      ISP       |  类型  |      数据源      | 占比 |
+| :------------: | :----: | :--------------: | :--: |
+| Tier-01 | 骨干网 | IPInfo |  2%  |
+| Tier-01 | 骨干网 | 埃文科技 |  3%  |
+| Tier-01 | 骨干网 | IPInSight |  5%  |
+| Tier-01 | 城域网 | IPInSight |  90%  |
+
+|      ISP       |  类型  |      数据源      | 占比 |
+| :------------: | :----: | :--------------: | :--: |
+| Others | 骨干网 | IPInSight | 5%  |
+| Others | 城域网 |  IPInSight  | 95%  |
+
+### IPv6 Database
+
+|      ISP       |  类型  |      数据源      | 占比 |
+| :------------: | :----: | :--------------: | :--: |
+| All | 全部 | IP2Location Lite | 100%  |
+
+This product includes IP2Location LITE data available from <a href="https://lite.ip2location.com">https://lite.ip2location.com</a>.
+
+### Others
+
+其他第三方 API 尽管集成在本项目内，但是具体的 TOS 以及 AUP，请详见第三方 API 官网。如遇到 IP 数据错误，也请直接联系他们纠错。

config/config.go

@@ -0,0 +1,31 @@
+package config
+
+import "os"
+
+type tracerConfig struct {
+    Token `yaml:"Token"`
+    Preference `yaml:"Preference"`
+}
+
+type Token struct {
+	LeoMoeAPI string `yaml:"LeoMoeAPI"`
+	IPInfo    string `yaml:"IPInfo"`
+}
+
+type Preference struct {
+    AlwaysRoutePath bool `yaml:"AlwaysRoutePath"`
+}
+
+type configPath func() (string, error)
+
+func configFromRunDir() (string, error) {
+    return "./", nil
+}
+
+func configFromUserHomeDir() (string, error) {
+    dir, err := os.UserHomeDir()
+    if err != nil {
+        return "", err
+    }
+    return dir + "/.nexttrace/", nil
+}

config/config_test.go

@@ -0,0 +1,18 @@
+package config
+
+import (
+	"testing"
+	"log"
+)
+
+func TestReadConfig(t *testing.T) {
+	if res, err := Read(); err != nil {
+		log.Println(err)
+	} else {
+		log.Println(res)
+	}
+}
+
+func TestGenerateConfig(t *testing.T) {
+	Generate()
+}

config/generate_config.go

@@ -0,0 +1,94 @@
+package config
+
+import (
+	"os"
+	"io/ioutil"
+	"fmt"
+	"gopkg.in/yaml.v2"
+)
+
+func pathExists(path string) (bool, error) {
+	_, err := os.Stat(path)
+	if err == nil {
+		return true, nil
+	}
+	if os.IsNotExist(err) {
+		return false, nil
+	}
+	return false, err
+}
+
+func writeFile(content []byte) error {
+	var err error
+	var path string
+	path, err = configFromUserHomeDir()
+	if err != nil {
+		path, err = configFromRunDir()
+		if err != nil {
+			return err
+		}
+	}
+
+	if exist, _ := pathExists(path); !exist {
+		err := os.Mkdir(path, os.ModePerm)
+		if err != nil {
+			return err
+		}
+	}
+
+	if err = ioutil.WriteFile(path + "ntraceConfig.yml", []byte(content), 0644); err != nil{
+		return err
+	}
+
+	return nil
+}
+
+func Generate() (*tracerConfig, error) {
+	var leotoken string
+	var iPInfoToken string
+	var routePathEnable string
+	
+	fmt.Println("这是一个配置向导，我们会帮助您生成配置文件，它是一次性的，除非您主动要求重新生成，否则它将不会再出现")
+
+	fmt.Println("请输入您的LeoMoeAPI Token，如果您没有，请到 Telegram Bot @NextTraceBot 获取一个")
+	fmt.Scanln(&leotoken)
+	if leotoken == "" {
+		fmt.Println("检测到您的输入为空，您将使用公共Token。这意味着您将和所有使用此Token的客户端共用每分钟150次IP查询的额度")
+		leotoken = "NextTraceDemo"
+	}
+
+	fmt.Println("请输入您的IPInfo Token，如果您不需要使用IPInfo，可以直接回车")
+	fmt.Scanln(&iPInfoToken)
+
+	token := Token{
+		LeoMoeAPI: leotoken,
+		IPInfo:    iPInfoToken,
+	}
+
+	var preference Preference
+	fmt.Print("您是否希望在每次Traceroute结束后显示Route-Path图? (y/n)")
+	fmt.Scanln(&routePathEnable)
+	if routePathEnable == "n" || routePathEnable == "N" || routePathEnable == "no" || routePathEnable == "No" || routePathEnable == "NO" {
+		preference = Preference{AlwaysRoutePath: false}
+	} else {
+		preference = Preference{AlwaysRoutePath: true}
+	}
+
+	finalConfig := tracerConfig{
+		Token:      token,
+		Preference: preference,
+	}
+
+	yamlData, err := yaml.Marshal(&finalConfig)
+
+	if err != nil {
+        return nil, err
+    }
+
+	if err = writeFile(yamlData); err != nil {
+		return nil, err
+	} else {
+		fmt.Println("配置文件创建成功")
+		return &finalConfig, nil
+	}
+}

config/read_config.go

@@ -0,0 +1,47 @@
+package config
+
+import (
+    "io/ioutil"
+    "log"
+    "gopkg.in/yaml.v2"
+)
+
+func (c *tracerConfig) Parse(data []byte) error {
+    return yaml.Unmarshal(data, c)
+}
+
+func readFile(cp configPath) ([]byte, error) {
+    var content []byte
+    path, err := cp()
+    if err != nil {
+        log.Println(err)
+        return nil, err
+    }
+    content, err = ioutil.ReadFile(path + "ntraceConfig.yml")
+    if err != nil {
+        return nil, err
+    }
+    return content, nil
+}
+
+func Read() (*tracerConfig, error) {
+    var data []byte
+    var err  error
+    
+    data, err = readFile(configFromRunDir)
+
+    if err != nil {
+        data, err = readFile(configFromUserHomeDir)
+
+        if err != nil {
+            return nil, err
+        }
+    }
+
+    var config tracerConfig
+    if err := config.Parse(data); err != nil {
+        return nil, err
+    }
+    
+    return &config, err
+}

go.mod

@@ -5,22 +5,24 @@ go 1.18
 require (
 	github.com/google/gopacket v1.1.19
 	golang.org/x/net v0.0.0-20220425223048-2871e0cb64e4
+	golang.org/x/sync v0.0.0-20220513210516-0976fa681c29
 )
 
 require (
 	github.com/mattn/go-colorable v0.1.9 // indirect
 	github.com/mattn/go-isatty v0.0.14 // indirect
+	gopkg.in/yaml.v2 v2.4.0 // 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 // indirect
-	github.com/stretchr/testify v1.7.1 // indirect
-	github.com/tidwall/gjson v1.14.1 // indirect
+	github.com/rodaine/table v1.0.1
+	github.com/stretchr/testify v1.7.1
+	github.com/tidwall/gjson v1.14.1
 	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
 )

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 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+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=
@@ -9,6 +10,7 @@ github.com/mattn/go-colorable v0.1.9/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope
 github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
 github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y=
 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/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
@@ -33,18 +35,22 @@ 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=
 golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
 golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
-golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e h1:fLOSk5Q00efkSvAm+4xcoXD+RRmLmmulPn5I3Y9F2EM=
-golang.org/x/sys v0.0.0-20211216021012-1d35b9e2eb4e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6 h1:nonptSpoQ4vQjyraW20DXPAglgQfVnM9ZC6MmNLMR60=
 golang.org/x/sys v0.0.0-20220503163025-988cb79eb6c6/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/tools v0.0.0-20200130002326-2f3ba24bd6e7/go.mod h1:TB2adYChydJhpapKDTa4BR/hXlZSLoq2Wpct/0txZ28=
 golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 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.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
+gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b h1:h8qDotaEPuJATrMmW04NCwg7v22aHH28wwpauUhK9Oo=
+gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

ipgeo/ipapicom.go

@@ -0,0 +1,43 @@
+package ipgeo
+
+import (
+	"errors"
+	"io/ioutil"
+	"log"
+	"net/http"
+	"regexp"
+	"time"
+
+	"github.com/tidwall/gjson"
+)
+
+func IPApiCom(ip string) (*IPGeoData, error) {
+	url := "http://ip-api.com/json/" + ip + "?fields=status,message,country,regionName,city,isp,as"
+	client := &http.Client{
+		// 2 秒超时
+		Timeout: 2 * time.Second,
+	}
+	req, _ := http.NewRequest("GET", url, nil)
+	req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:100.0) Gecko/20100101 Firefox/100.0")
+	content, err := client.Do(req)
+	if err != nil {
+		log.Println("ip-api.com 请求超时(2s)，请切换其他API使用")
+		return nil, err
+	}
+	body, _ := ioutil.ReadAll(content.Body)
+	res := gjson.ParseBytes(body)
+
+	if res.Get("status").String() != "success" {
+		return &IPGeoData{}, errors.New("超过API阈值")
+	}
+
+	re := regexp.MustCompile("[0-9]+")
+
+	return &IPGeoData{
+		Asnumber: re.FindString(res.Get("as").String()),
+		Country:  res.Get("country").String(),
+		City:     res.Get("city").String(),
+		Prov:     res.Get("regionName").String(),
+		Isp:      res.Get("isp").String(),
+	}, nil
+}

ipgeo/ipgeo.go

@@ -1,5 +1,9 @@
 package ipgeo
 
+import (
+	"strings"
+)
+
 type IPGeoData struct {
 	Asnumber string
 	Country  string
@@ -11,3 +15,20 @@ type IPGeoData struct {
 }
 
 type Source = func(ip string) (*IPGeoData, error)
+
+func GetSource(s string) Source {
+	switch strings.ToUpper(s) {
+	case "LEOMOEAPI":
+		return LeoIP
+	case "IP.SB":
+		return IPSB
+	case "IPINSIGHT":
+		return IPInSight
+	case "IPAPI.COM":
+		return IPApiCom
+	case "IPINFO":
+		return IPInfo
+	default:
+		return LeoIP
+	}
+}

ipgeo/ipgeo_test.go

@@ -15,11 +15,12 @@ func TestLeoIP(t *testing.T) {
 }
 
 func TestIPSB(t *testing.T) {
-	res, err := IPSB("1.1.1.1")
-	assert.Nil(t, err)
-	assert.NotNil(t, res)
-	assert.NotEmpty(t, res.Asnumber)
-	assert.NotEmpty(t, res.Isp)
+	// Not available
+	//res, err := IPSB("1.1.1.1")
+	//assert.Nil(t, err)
+	//assert.NotNil(t, res)
+	//assert.NotEmpty(t, res.Asnumber)
+	//assert.NotEmpty(t, res.Isp)
 }
 
 func TestIPInfo(t *testing.T) {
@@ -40,3 +41,12 @@ func TestIPInSight(t *testing.T) {
 	// 这个库有时候不提供城市信息，返回值为""
 	//assert.NotEmpty(t, res.City)
 }
+
+func TestIPApiCom(t *testing.T) {
+	res, err := IPApiCom("1.1.1.1")
+	assert.Nil(t, err)
+	assert.NotNil(t, res)
+	assert.NotEmpty(t, res.Country)
+	assert.NotEmpty(t, res.City)
+	assert.NotEmpty(t, res.Prov)
+}

ipgeo/ipinfo.go

@@ -8,7 +8,6 @@ import (
 )
 
 func IPInfo(ip string) (*IPGeoData, error) {
-
 	resp, err := http.Get("https://ipinfo.io/" + ip + "?token=" + token.ipinfo)
 	if err != nil {
 		return nil, err

ipgeo/ipsb.go

@@ -2,24 +2,36 @@ package ipgeo
 
 import (
 	"io/ioutil"
+	"log"
 	"net/http"
+	"os"
+	"time"
 
 	"github.com/tidwall/gjson"
 )
 
 func IPSB(ip string) (*IPGeoData, error) {
-	resp, err := http.Get("https://api.ip.sb/geoip/" + ip)
+	url := "https://api.ip.sb/geoip/" + ip
+	client := &http.Client{
+		// 2 秒超时
+		Timeout: 2 * time.Second,
+	}
+	req, _ := http.NewRequest("GET", url, nil)
+	// 设置 UA，ip.sb 默认禁止 go-client User-Agent 的 api 请求
+	req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:100.0) Gecko/20100101 Firefox/100.0")
+	content, err := client.Do(req)
 	if err != nil {
+		log.Println("api.ip.sb 请求超时(2s)，请切换其他API使用")
 		return nil, err
 	}
-	defer resp.Body.Close()
-	body, err := ioutil.ReadAll(resp.Body)
-	if err != nil {
-		return nil, err
-	}
-
+	body, _ := ioutil.ReadAll(content.Body)
 	res := gjson.ParseBytes(body)
 
+	if res.Get("country").String() == "" {
+		// 什么都拿不到，证明被Cloudflare风控了
+		os.Exit(1)
+	}
+
 	return &IPGeoData{
 		Asnumber: res.Get("asn").String(),
 		Country:  res.Get("country").String(),

ipgeo/tokens.go

@@ -1,5 +1,7 @@
 package ipgeo
 
+import "github.com/xgadget-lab/nexttrace/config"
+
 type tokenData struct {
 	ipinsight string
 	ipinfo    string
@@ -8,6 +10,12 @@ type tokenData struct {
 
 var token = tokenData{
 	ipinsight: "",
-	ipinfo:    "42764a944dabd0",
+	ipinfo:    "",
 	ipleo:     "NextTraceDemo",
 }
+
+
+func SetToken(c config.Token) {
+	token.ipleo = c.LeoMoeAPI
+	token.ipinfo = c.IPInfo
+}

main.go

@@ -3,85 +3,128 @@ package main
 import (
 	"flag"
 	"fmt"
+	"log"
 	"os"
+	"strings"
 	"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/config"
+	"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"
-	"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)")
-var port = flag.Int("p", 80, "Set SYN Traceroute Port")
-var numMeasurements = flag.Int("q", 3, "Set the number of probes per each hop.")
-var parallelRequests = flag.Int("r", 18, "Set ParallelRequests number. It should be 1 when there is a multi-routing.")
-var maxHops = flag.Int("m", 30, "Set the max number of hops (max TTL to be reached).")
-var dataOrigin = flag.String("d", "LeoMoeAPI", "Choose IP Geograph Data Provider [LeoMoeAPI, IP.SB, IPInfo, IPInsight]")
-var displayMode = flag.String("displayMode", "table", "Choose The Display Mode [table, Besttrace]")
+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 = fSet.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 tablePrint = fSet.Bool("table", false, "Output trace results as table")
+var ver = fSet.Bool("V", false, "Check Version")
 
-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 {
-			if *displayMode == "Besttrace" {
-				printer.TraceroutePrinter(ip, *res, *dataOrigin)
-			} else if *displayMode == "table" {
-				printer.TracerouteTablePrinter(ip, *res, *dataOrigin)
-			} else {
-				printer.TracerouteTablePrinter(ip, *res, *dataOrigin)
-			}
-		}
-
-	} 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 {
-			if *displayMode == "Besttrace" {
-				printer.TraceroutePrinter(ip, *res, *dataOrigin)
-			} else if *displayMode == "table" {
-				printer.TracerouteTablePrinter(ip, *res, *dataOrigin)
-			} else {
-				printer.TracerouteTablePrinter(ip, *res, *dataOrigin)
-			}
-		}
-	}
+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] [ -table ] -report")
+	fSet.PrintDefaults()
+	os.Exit(2)
 }
 
 func flagApply() string {
-	flag.Parse()
-	ipArg := flag.Args()
-	if flag.NArg() != 1 {
-		fmt.Println("Args Error\nUsage : ./bettertrace [-T] [-d <dataOrigin> ] [ -m <hops> ] [ -p <port> ] [ -q <probes> ] [ -r <parallelrequests> ] <hostname>")
-		os.Exit(2)
+	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.")
+	}
+
+	configData, err := config.Read();
+
+	if err != nil {
+		if configData, err = config.Generate(); err != nil {
+			log.Fatal(err)
+		}
+	}
+
+	ipgeo.SetToken(configData.Token)
+
+	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 (*tcpSYNFlag && *udpPackageFlag) || *tablePrint {
+		printer.TracerouteTablePrinter(res)
+	} else if *tcpSYNFlag || *udpPackageFlag {
+		printer.TraceroutePrinter(res)
+	}
+
+	p := configData.Preference
+
+	if *routePath || p.AlwaysRoutePath {
+		r := reporter.New(res, ip.String())
+		r.Print()
 	}
-	return ipArg[0]
 }

methods/methods.go

@@ -1,79 +0,0 @@
-package methods
-
-import (
-	"encoding/binary"
-	"errors"
-	"net"
-	"time"
-)
-
-// TracerouteHop type
-type TracerouteHop struct {
-	Success bool
-	Address net.Addr
-	TTL     uint16
-	RTT     *time.Duration
-}
-
-type TracerouteConfig struct {
-	MaxHops          uint16
-	NumMeasurements  uint16
-	ParallelRequests uint16
-
-	Port    int
-	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{}
-	for i := uint16(1); i < maxHops; i++ {
-		foundFinal := false
-		probes := preliminary[i]
-		if probes == nil {
-			break
-		}
-		finalResults[i] = []TracerouteHop{}
-		for _, probe := range probes {
-			if probe.Success && probe.Address.String() == destIP.String() {
-				foundFinal = true
-			}
-			finalResults[i] = append(finalResults[i], probe)
-		}
-		if foundFinal {
-			break
-		}
-	}
-	return finalResults
-}

methods/tcp/tcp.go

@@ -1,329 +0,0 @@
-package tcp
-
-import (
-	"log"
-	"math"
-	"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/parallel_limiter"
-	"github.com/xgadget-lab/nexttrace/signal"
-	"github.com/xgadget-lab/nexttrace/util"
-	"golang.org/x/net/context"
-	"golang.org/x/net/icmp"
-	"golang.org/x/net/ipv4"
-)
-
-type inflightData struct {
-	start time.Time
-	ttl   uint16
-}
-
-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 {
-	opConfig    opConfig
-	trcrtConfig methods.TracerouteConfig
-	results     results
-}
-
-type opConfig struct {
-	icmpConn net.PacketConn
-	tcpConn  net.PacketConn
-	tcpMu    sync.Mutex
-
-	destIP net.IP
-	srcIP  net.IP
-
-	wg *sync.WaitGroup
-
-	ctx    context.Context
-	cancel context.CancelFunc
-}
-
-func New(destIP net.IP, config methods.TracerouteConfig) *Traceroute {
-	return &Traceroute{
-		opConfig: opConfig{
-			destIP: destIP,
-		},
-		trcrtConfig: config,
-	}
-}
-
-func (tr *Traceroute) Start() (*map[uint16][]methods.TracerouteHop, error) {
-	tr.opConfig.ctx, tr.opConfig.cancel = context.WithCancel(context.Background())
-
-	tr.opConfig.srcIP, _ = util.LocalIPPort(tr.opConfig.destIP)
-
-	var err error
-	tr.opConfig.tcpConn, err = net.ListenPacket("ip4:tcp", tr.opConfig.srcIP.String())
-	if err != nil {
-		return nil, err
-	}
-
-	tr.opConfig.icmpConn, err = icmp.ListenPacket("ip4:icmp", "0.0.0.0")
-	if err != nil {
-		return nil, err
-	}
-
-	var wg sync.WaitGroup
-	tr.opConfig.wg = &wg
-
-	tr.results = results{
-		inflightRequests:   sync.Map{},
-		concurrentRequests: parallel_limiter.New(int(tr.trcrtConfig.ParallelRequests)),
-		reachedFinalHop:    signal.New(),
-
-		results: map[uint16][]methods.TracerouteHop{},
-	}
-
-	return tr.start()
-}
-
-func (tr *Traceroute) timeoutLoop() {
-	ticker := time.NewTicker(tr.trcrtConfig.Timeout / 4)
-	go func() {
-		for range ticker.C {
-			tr.results.inflightRequests.Range(func(key, value interface{}) bool {
-				request := value.(inflightData)
-				expired := time.Since(request.start) > tr.trcrtConfig.Timeout
-				if !expired {
-					return true
-				}
-				tr.results.inflightRequests.Delete(key)
-				tr.addToResult(request.ttl, methods.TracerouteHop{
-					Success: false,
-					TTL:     request.ttl,
-				})
-				tr.results.concurrentRequests.Finished()
-				tr.opConfig.wg.Done()
-				return true
-			})
-		}
-	}()
-	select {
-	case <-tr.opConfig.ctx.Done():
-		ticker.Stop()
-	}
-}
-
-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) handleICMPMessage(msg listener_channel.ReceivedMessage, data []byte) {
-	header, err := methods.GetICMPResponsePayload(data)
-	if err != nil {
-		return
-	}
-	sequenceNumber := methods.GetTCPSeq(header)
-	val, ok := tr.results.inflightRequests.LoadAndDelete(sequenceNumber)
-	if !ok {
-		return
-	}
-	request := val.(inflightData)
-	elapsed := time.Since(request.start)
-	if msg.Peer.String() == tr.opConfig.destIP.String() {
-		tr.results.reachedFinalHop.Signal()
-	}
-	tr.addToResult(request.ttl, methods.TracerouteHop{
-		Success: true,
-		Address: msg.Peer,
-		TTL:     request.ttl,
-		RTT:     &elapsed,
-	})
-	tr.results.concurrentRequests.Finished()
-	tr.opConfig.wg.Done()
-}
-
-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")
-			}
-		}
-	}
-}
-
-func (tr *Traceroute) tcpListener() {
-	lc := listener_channel.New(tr.opConfig.tcpConn)
-
-	defer lc.Stop()
-
-	go lc.Start()
-
-	for {
-		select {
-		case <-tr.opConfig.ctx.Done():
-			return
-		case msg := <-lc.Messages:
-			if msg.N == nil {
-				continue
-			}
-			if msg.Peer.String() != tr.opConfig.destIP.String() {
-				continue
-			}
-			// Decode a packet
-			packet := gopacket.NewPacket(msg.Msg[:*msg.N], layers.LayerTypeTCP, gopacket.Default)
-			// Get the TCP layer from this packet
-			if tcpLayer := packet.Layer(layers.LayerTypeTCP); tcpLayer != nil {
-				tcp, _ := tcpLayer.(*layers.TCP)
-
-				val, ok := tr.results.inflightRequests.LoadAndDelete(tcp.Ack - 1)
-				if !ok {
-					continue
-				}
-				request := val.(inflightData)
-				tr.results.concurrentRequests.Finished()
-				elapsed := time.Since(request.start)
-				if msg.Peer.String() == tr.opConfig.destIP.String() {
-					tr.results.reachedFinalHop.Signal()
-				}
-				tr.addToResult(request.ttl, methods.TracerouteHop{
-					Success: true,
-					Address: msg.Peer,
-					TTL:     request.ttl,
-					RTT:     &elapsed,
-				})
-				tr.opConfig.wg.Done()
-			}
-		}
-	}
-}
-
-func (tr *Traceroute) sendMessage(ttl uint16) {
-	_, srcPort := util.LocalIPPort(tr.opConfig.destIP)
-	ipHeader := &layers.IPv4{
-		SrcIP:    tr.opConfig.srcIP,
-		DstIP:    tr.opConfig.destIP,
-		Protocol: layers.IPProtocolTCP,
-		TTL:      uint8(ttl),
-	}
-
-	sequenceNumber := uint32(rand.Intn(math.MaxUint32))
-
-	tcpHeader := &layers.TCP{
-		SrcPort: layers.TCPPort(srcPort),
-		DstPort: layers.TCPPort(tr.trcrtConfig.Port),
-		Seq:     sequenceNumber,
-		SYN:     true,
-		Window:  14600,
-	}
-	_ = tcpHeader.SetNetworkLayerForChecksum(ipHeader)
-
-	buf := gopacket.NewSerializeBuffer()
-	opts := gopacket.SerializeOptions{
-		ComputeChecksums: true,
-		FixLengths:       true,
-	}
-	if err := gopacket.SerializeLayers(buf, opts, tcpHeader); err != nil {
-		tr.results.err = err
-		tr.opConfig.cancel()
-		return
-	}
-
-	tr.opConfig.tcpMu.Lock()
-	defer tr.opConfig.tcpMu.Unlock()
-	err := ipv4.NewPacketConn(tr.opConfig.tcpConn).SetTTL(int(ttl))
-	if err != nil {
-		tr.results.err = err
-		tr.opConfig.cancel()
-		return
-	}
-
-	start := time.Now()
-	if _, err := tr.opConfig.tcpConn.WriteTo(buf.Bytes(), &net.IPAddr{IP: tr.opConfig.destIP}); err != nil {
-		tr.results.err = err
-		tr.opConfig.cancel()
-		return
-	}
-	tr.results.inflightRequests.Store(sequenceNumber, inflightData{start: start, ttl: ttl})
-}
-
-func (tr *Traceroute) sendLoop() {
-	rand.Seed(time.Now().UTC().UnixNano())
-	defer tr.opConfig.wg.Done()
-
-	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(1)
-				go tr.sendMessage(ttl)
-			}
-		}
-	}
-}
-
-func (tr *Traceroute) start() (*map[uint16][]methods.TracerouteHop, error) {
-	go tr.timeoutLoop()
-	go tr.icmpListener()
-	go tr.tcpListener()
-
-	tr.opConfig.wg.Add(1)
-	go tr.sendLoop()
-
-	tr.opConfig.wg.Wait()
-	tr.opConfig.cancel()
-
-	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
-}

methods/udp/udp.go

@@ -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
-}

nt_install.sh

@@ -0,0 +1,275 @@
+#!/bin/bash
+
+auto=False
+#是否忽略一切警告，按默认执行
+if [[ $1 == "--auto" ]]; then
+  auto=True
+  echo "自动运行中"
+fi
+
+usrPath="/usr/local/bin"
+
+function red() {
+  echo -e "\033[31m""${1}""\033[0m"
+}
+
+checkRootPermit() {
+  [[ $EUID -ne 0 ]] && red "请使用sudo/root权限运行本脚本" && exit 1
+}
+
+checkSystemArch() {
+  arch=$(uname -m)
+  case $arch in
+  'x86_64')
+    archParam='amd64'
+    ;;
+  'mips')
+    archParam='mips'
+    ;;
+  'arm64' | 'aarch64')
+    archParam="arm64"
+    ;;
+  'armv7l')
+    archParam='armv7'
+    ;;
+  'i386')
+    archParam="386"
+    ;;
+  *)
+    red "未知的系统架构，请联系开发者."
+    exit 1
+    ;;
+  esac
+}
+
+checkSystemDistribution() {
+  case "$OSTYPE" in
+  darwin*)
+    osDistribution="darwin"
+    downPath="/var/tmp/nexttrace"
+    ;;
+  linux*)
+    osDistribution="linux"
+    downPath="/var/tmp/nexttrace"
+    ;;
+  *)
+    red "安装脚本暂不支持的操作系统: $OSTYPE"
+    exit 1
+    ;;
+  esac
+}
+
+ask_if() {
+  local choice=""
+  red "${1}"
+  read -r choice
+  [[ $choice == y ]] && return 0 || return 1
+}
+
+#检查脚本更新
+check_script_update() {
+  if [[ ${osDistribution} == "darwin" ]]; then
+    [ "$(md5 <"${BASH_SOURCE[0]}")" == "$(curl -sL "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh" | md5)" ] && return 1 || return 0
+  else
+    [ "$(md5sum "${BASH_SOURCE[0]}" | awk '{print $1}')" == "$(md5sum <(curl -sL "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh") | awk '{print $1}')" ] && return 1 || return 0
+  fi
+}
+
+#更新脚本
+update_script() {
+  if curl -sL -o "${BASH_SOURCE[0]}" "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh" || curl -sL -o "${BASH_SOURCE[0]}" "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh"; then
+    red "nt_install.sh更新完成，正在重启脚本..."
+    exec bash "${BASH_SOURCE[0]}" --auto
+  else
+    red "更新nt_install.sh失败！"
+    exit 1
+  fi
+}
+
+ask_update_script() {
+  if check_script_update; then
+    red "nt_install.sh可升级"
+    [[ $auto == True ]] && update_script
+    ask_if "是否升级脚本？(n/y)：[n]" && update_script
+  else
+    red "nt_install.sh已经是最新版本"
+  fi
+}
+
+getLocation() {
+  red "正在获取地理位置信息..."
+  countryCode=$(curl -s "http://ip-api.com/line/?fields=countryCode")
+}
+
+checkPackageManger() {
+  if [[ ${osDistribution} == "darwin" ]]; then
+    # brew update
+    PACKAGE_MANAGEMENT_INSTALL='brew install'
+    PACKAGE_MANAGEMENT_REMOVE='brew uninstall'
+    return 0
+  fi
+  if [[ "$(which apt)" ]]; then
+    apt-get update
+    PACKAGE_MANAGEMENT_INSTALL='apt-get -y --no-install-recommends install'
+    PACKAGE_MANAGEMENT_REMOVE='apt-get purge'
+  elif [[ "$(which dnf)" ]]; then
+    dnf check-update
+    PACKAGE_MANAGEMENT_INSTALL='dnf -y install'
+    PACKAGE_MANAGEMENT_REMOVE='dnf remove'
+  elif [[ "$(which yum)" ]]; then
+    PACKAGE_MANAGEMENT_INSTALL='yum -y install'
+    PACKAGE_MANAGEMENT_REMOVE='yum remove'
+  elif [[ "$(which zypper)" ]]; then
+    zypper refresh
+    PACKAGE_MANAGEMENT_INSTALL='zypper install -y --no-recommends'
+    PACKAGE_MANAGEMENT_REMOVE='zypper remove'
+  elif [[ "$(which pacman)" ]]; then
+    PACKAGE_MANAGEMENT_INSTALL='pacman -Syu --noconfirm'
+    PACKAGE_MANAGEMENT_REMOVE='pacman -Rsn'
+  else
+    red "error: The script does not support the package manager in this operating system."
+    exit 1
+  fi
+}
+
+install_software() {
+  package_name="$1"
+  which "$package_name" >/dev/null 2>&1 && return
+  [[ ${osDistribution} == "darwin" ]] && echo -e "由于macOS brew的权限限制，请以非root权限执行下面一行提示的命令后再次运行本脚本(注意不要在该命令加sudo!):\nbrew update && ${PACKAGE_MANAGEMENT_INSTALL} $package_name " && exit 0
+  red "${package_name} 正在安装中...(此步骤时间可能较长，请耐心等待)"
+  if ${PACKAGE_MANAGEMENT_INSTALL} "$package_name"; then
+    red "info: $package_name is installed."
+  else
+    red "error: Installation of $package_name failed, please check your network."
+    exit 1
+  fi
+}
+
+checkVersion() {
+  which nexttrace >/dev/null 2>&1 || return
+  red "正在检查版本..."
+  version=$(curl -sL https://api.github.com/repos/xgadget-lab/nexttrace/releases/latest | jq -r '.tag_name')
+  if [[ $version == "" ]]; then
+    red "获取版本失败，请检查网络连接"
+    exit 1
+  fi
+  currentVersion=$(nexttrace -V | head -n 1 | awk '{print $2}')
+  if [[ $currentVersion == "$version" ]]; then
+    red "当前版本已是最新版本"
+    exit 0
+  fi
+  red "当前最新release版本：${version}"
+  red "您当前的版本：${currentVersion}"
+  if [[ $auto == True ]]; then
+    return 0
+  fi
+  read -r -p "是否更新软件? (n/y)：[n]" input
+  case $input in
+  [yY][eE][sS] | [yY])
+    return 0
+    ;;
+  [nN][oO] | [nN])
+    red "您选择了取消更新，脚本即将退出"
+    exit 1
+    ;;
+  *)
+    red "您选择了取消更新，脚本即将退出"
+    exit 1
+    ;;
+  esac
+}
+
+downloadBinrayFile() {
+  red "正在获取最新版的 NextTrace 发行版文件信息..."
+  # 简单说明一下，Github提供了一个API，可以获取最新发行版本的二进制文件下载地址（对应的是browser_download_url），根据刚刚测得的osDistribution、archParam，获取对应的下载地址
+  # red nexttrace_${osDistribution}_${archParam}
+  latestURL=$(curl -s https://api.github.com/repos/xgadget-lab/nexttrace/releases/latest | jq ".assets[] | select(.name == \"nexttrace_${osDistribution}_${archParam}\") | .browser_download_url")
+  latestURL=${latestURL:1:$((${#latestURL} - 1 - 1))}
+  if [ "$countryCode" == "CN" ]; then
+    if [[ $auto == True ]]; then
+      latestURL="https://ghproxy.com/"$latestURL
+    else
+      read -r -p "检测到国内网络环境，是否使用镜像下载以加速(n/y)[y]" input
+      case $input in
+      [yY][eE][sS] | [yY])
+        latestURL="https://ghproxy.com/"$latestURL
+        ;;
+
+      [nN][oO] | [nN])
+        red "您选择了不使用镜像，下载可能会变得异常缓慢，或者失败"
+        ;;
+
+      *)
+        latestURL="https://ghproxy.com/"$latestURL
+        ;;
+      esac
+    fi
+  fi
+
+  red "正在下载 NextTrace 二进制文件..."
+  if wget -O ${downPath} "${latestURL}"; then
+    red "NextTrace 现在已经在您的系统中可用"
+    changeMode
+    mv ${downPath} ${usrPath}
+  else
+    red "NextTrace 下载失败，请检查您的网络是否正常"
+    exit 1
+  fi
+}
+
+changeMode() {
+  chmod +x ${downPath}
+  [[ ${osDistribution} == "darwin" ]] && xattr -r -d com.apple.quarantine ${downPath}
+}
+
+runBinrayFileHelp() {
+  if [ -e ${usrPath} ]; then
+    ${usrPath}/nexttrace -h
+  fi
+  red "You may need to execute a command to remove dependent software: $PACKAGE_MANAGEMENT_REMOVE wget jq"
+}
+
+addCronTask() {
+  read -r -p "是否添加自动更新任务？(n/y)：[n]" input
+  case $input in
+  [yY][eE][sS] | [yY])
+    if [[ ${osDistribution} == "darwin" ]]; then
+      crontab -l >crontab.bak 2>/dev/null
+      sed -i '' '/nt_install.sh/d' crontab.bak
+    elif [[ ${osDistribution} == "linux" ]]; then
+      crontab -l >crontab.bak 2>/dev/null
+      sed -i '/nt_install.sh/d' crontab.bak
+    else
+      red "暂不支持您的系统,无法自动添加crontab任务"
+      return
+    fi
+    echo "1 1 * * * $(dirname "$(readlink -f "$0")")/nt_install.sh --auto >> /var/log/nt_install.log" >>crontab.bak
+    crontab crontab.bak
+    rm -f crontab.bak
+    ;;
+  [nN][oO] | [nN])
+    red "您选择了不添加自动更新任务，您也可以通过命令 再次执行此脚本 手动更新"
+    ;;
+  *)
+    red "您选择了不添加自动更新任务，您可以通过命令 再次执行此脚本 手动更新"
+    ;;
+  esac
+}
+
+# Check Procedure
+checkRootPermit
+checkSystemDistribution
+checkSystemArch
+ask_update_script
+checkPackageManger
+install_software wget
+install_software jq
+checkVersion
+
+# Download Procedure
+getLocation
+downloadBinrayFile
+
+# Run Procedure
+runBinrayFileHelp
+[[ $auto != True ]] && addCronTask

parallel_limiter/parallel_limiter.go

@@ -1,52 +0,0 @@
-package parallel_limiter
-
-import (
-	"sync"
-)
-
-type ParallelLimiter struct {
-	maxCount int
-
-	mu             sync.Mutex
-	currentRunning int
-
-	waiting []chan struct{}
-}
-
-func New(count int) *ParallelLimiter {
-	return &ParallelLimiter{
-		maxCount: count,
-
-		currentRunning: 0,
-		waiting:        []chan struct{}{},
-	}
-}
-
-func (p *ParallelLimiter) Start() chan struct{} {
-	p.mu.Lock()
-	if p.currentRunning+1 > p.maxCount {
-		waitChan := make(chan struct{})
-		p.waiting = append(p.waiting, waitChan)
-		p.mu.Unlock()
-		return waitChan
-	}
-	p.currentRunning++
-	p.mu.Unlock()
-	instantResolveChan := make(chan struct{})
-	go func() {
-		instantResolveChan <- struct{}{}
-	}()
-	return instantResolveChan
-}
-
-func (p *ParallelLimiter) Finished() {
-	p.mu.Lock()
-	if len(p.waiting) > 0 {
-		first := p.waiting[0]
-		p.waiting = p.waiting[1:]
-		first <- struct{}{}
-		p.currentRunning++
-	}
-	p.currentRunning--
-	p.mu.Unlock()
-}

printer/basic.go

@@ -5,8 +5,13 @@ import (
 	"net"
 )
 
-func PrintCopyRight() {
-	fmt.Println("NextTrace v0.1.0 Alpha \nxgadget-lab 实验室 zhsh (xzhsh.ch) & leo (leo.moe)")
+var version = "v0.0.0.alpha"
+var buildDate = ""
+var commitID = ""
+
+func Version() {
+	fmt.Println("NextTrace", version, buildDate, commitID)
+	fmt.Println("XGadget-lab Leo (leo.moe) & Vincent (vincent.moe) & zhshch (xzhsh.ch)")
 }
 
 func PrintTraceRouteNav(ip net.IP, domain string, dataOrigin string) {

printer/printer.go

@@ -0,0 +1,89 @@
+package printer
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/xgadget-lab/nexttrace/trace"
+
+	"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 {
+		// 有些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, ", ")
+}

printer/printer_test.go

@@ -0,0 +1,104 @@
+package printer
+
+import (
+	"errors"
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+	"github.com/xgadget-lab/nexttrace/trace"
+	"github.com/xgadget-lab/nexttrace/util"
+	"net"
+	"testing"
+	"time"
+)
+
+func TestPrintTraceRouteNav(t *testing.T) {
+	PrintTraceRouteNav(util.DomainLookUp("1.1.1.1"), "1.1.1.1", "dataOrigin")
+}
+
+var testGeo = &ipgeo.IPGeoData{
+	Asnumber: "TestAsnumber",
+	Country:  "TestCountry",
+	Prov:     "TestProv",
+	City:     "TestCity",
+	District: "TestDistrict",
+	Owner:    "TestOwner",
+	Isp:      "TestIsp",
+}
+
+var testResult = &trace.Result{
+	Hops: [][]trace.Hop{
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo:      testGeo,
+			},
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo:      testGeo,
+			},
+		},
+		{
+			{
+				Success:  false,
+				Address:  nil,
+				Hostname: "",
+				TTL:      0,
+				RTT:      0,
+				Error:    errors.New("test error"),
+				Geo:      nil,
+			},
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo:      nil,
+			},
+		},
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      0,
+				Error:    nil,
+				Geo:      &ipgeo.IPGeoData{},
+			},
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo:      testGeo,
+			},
+		},
+	},
+}
+
+func TestTraceroutePrinter(t *testing.T) {
+	TraceroutePrinter(testResult)
+}
+
+func TestTracerouteTablePrinter(t *testing.T) {
+	TracerouteTablePrinter(testResult)
+}
+
+func TestRealtimePrinter(t *testing.T) {
+	RealtimePrinter(testResult, 0)
+	RealtimePrinter(testResult, 1)
+	RealtimePrinter(testResult, 2)
+}

printer/realtime_printer.go

@@ -0,0 +1,15 @@
+package printer
+
+import (
+	"fmt"
+
+	"github.com/xgadget-lab/nexttrace/trace"
+)
+
+func RealtimePrinter(res *trace.Result, ttl int) {
+	fmt.Print(ttl + 1)
+	for i := range res.Hops[ttl] {
+		HopPrinter(res.Hops[ttl][i])
+	}
+
+}

printer/tableprinter.go

@@ -0,0 +1,124 @@
+package printer
+
+import (
+	"fmt"
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+	"strings"
+
+	"github.com/xgadget-lab/nexttrace/trace"
+
+	"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 = ""
+			}
+			if data.Country == "" && data.Prov == "" && data.City == "" {
+				tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, "", data.Owner)
+			} else {
+				if data.City != "" {
+					tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, data.Country+", "+data.Prov+", "+data.City, data.Owner)
+				} else {
+					tbl.AddRow(data.Hop, data.IP, data.Latency, data.Asnumber, data.Country, 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", "Location", "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, ") ")
+		}
+
+		if h.Geo == nil {
+			h.Geo = &ipgeo.IPGeoData{}
+		}
+
+		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
+	}
+}

quicklytest.sh

@@ -0,0 +1,271 @@
+#!/bin/bash
+
+Green_font="\033[32m" && Red_font="\033[31m" && Font_suffix="\033[0m"
+Info="${Green_font}[Info]${Font_suffix}"
+Error="${Red_font}[Error]${Font_suffix}"
+echo -e "${Green_font}
+#======================================
+# Project: NextTrace https://github.com/xgadget-lab/nexttrace
+# Copyright Notice:
+# This script is ported from @KANIKIG https://github.com/KANIKIG/
+# The developer team made some modifications to adapt to NextTrace under the GPL-3.0 LICENSE
+# NextTrace:
+#   XGadget-lab Leo (leo.moe) & Vincent (vincent.moe) & zhshch (xzhsh.ch)
+#   IP Geo Data Provider: LeoMoeAPI
+#======================================
+${Font_suffix}"
+
+check_root() {
+  [[ "$(id -u)" != "0" ]] && echo -e "${Error} must be root user !" && exit 1
+}
+
+checkNexttrace() {
+  echo -e "${Info} 正在检查Nexttrace...(若未安装NextTrace则开始安装)"
+  if curl -sL -O "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh" || curl -sL -O "https://github.com/xgadget-lab/nexttrace/raw/main/nt_install.sh"; then
+    bash nt_install.sh #--auto #>/dev/null
+  fi
+}
+
+ask_if() {
+  local choice=""
+  echo -e "${Info} $1"
+  read -r choice
+  [[ $choice == y ]] && return 0
+  return 1
+}
+
+checkSystemDistribution() {
+  case "$OSTYPE" in
+  darwin*)
+    osDistribution="darwin"
+    ;;
+  linux*)
+    osDistribution="linux"
+    ;;
+  *)
+    red "unknown: $OSTYPE"
+    exit 1
+    ;;
+  esac
+}
+
+#检查脚本更新
+check_script_update() {
+  if [[ ${osDistribution} == "darwin" ]]; then
+    [ "$(md5 <"${BASH_SOURCE[0]}")" == "$(curl -sL "https://github.com/xgadget-lab/nexttrace/raw/main/quicklytest.sh" | md5)" ] && return 1 || return 0
+  else
+    [ "$(md5sum "${BASH_SOURCE[0]}" | awk '{print $1}')" == "$(md5sum <(curl -sL "https://github.com/xgadget-lab/nexttrace/raw/main/quicklytest.sh") | awk '{print $1}')" ] && return 1 || return 0
+  fi
+}
+
+#更新脚本
+update_script() {
+  if curl -sL -o "${BASH_SOURCE[0]}" "https://github.com/xgadget-lab/nexttrace/raw/main/quicklytest.sh" || curl -sL -o "${BASH_SOURCE[0]}" "https://github.com/xgadget-lab/nexttrace/raw/main/quicklytest.sh"; then
+    echo -e "${Info} quickylytest.sh更新完成，正在重启脚本..."
+    exec bash "${BASH_SOURCE[0]}"
+  else
+    echo -e "${Info} 更新quickylytest.sh失败！"
+    exit 1
+  fi
+}
+
+ask_update_script() {
+  if check_script_update; then
+    echo -e "${Info} quickylytest.sh可升级"
+    ask_if "是否升级脚本？(n/y)[n]" && update_script
+  else
+    echo -e "${Info} quickylytest.sh已经是最新版本"
+  fi
+}
+
+check_mode() {
+  echo -e "${Info} Nexttrace目前支持以下三种协议发起Traceroute请求:\n1.ICMP\n2.TCP(速度最快,但部分节点不支持)\n3.UDP\n(IPv6暂只支持ICMP模式)" && read -r -p "输入数字以选择:" node
+
+  while [[ ! "${node}" =~ ^[1-3]$ ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" node
+  done
+
+  [[ "${node}" == "1" ]] && TRACECMD="nexttrace"
+  [[ "${node}" == "2" ]] && TRACECMD="nexttrace -T"
+  [[ "${node}" == "3" ]] && TRACECMD="nexttrace -U"
+
+  echo -e "${Info} 结果是否制表?(制表模式为非实时显示)"
+  if ask_if "输入n/y以选择模式:[n]"; then
+    TRACECMD=${TRACECMD}" -rdns -table"
+    #        ##Route-Path功能还未完善,临时替代:
+    #        [[ "${node}" == "2" ]] && TRACECMD=${TRACECMD}" -report"
+    #        ##
+  else
+    TRACECMD=${TRACECMD}" -rdns -realtime"
+    #        ##Route-Path功能还未完善,临时替代:
+    #        [[ "${node}" == "1" ]] && TRACECMD=${TRACECMD}" -report"
+    #        [[ "${node}" == "2" ]] && TRACECMD=${TRACECMD}" -report"
+    #        ##
+  fi
+
+  echo -e "${Info} 是否输出Route-Path?"
+  ask_if "输入n/y以选择模式:[n]" && TRACECMD=${TRACECMD}" -report"
+
+}
+
+test_single() {
+  echo -e "${Info} 请输入你要测试的目标 ip :"
+  read -r -p "输入 ip 地址:" ip
+
+  while [[ -z "${ip}" ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新输入" && read -r -p "输入 ip 地址:" ip
+  done
+
+  ${TRACECMD} "${ip}" | grep -v -E 'NextTrace|XGadget-lab|Data\ Provider'
+
+  repeat_test_single
+}
+
+repeat_test_single() {
+  echo -e "${Info} 是否继续测试其他目标 ip ?"
+  if ask_if "输入n/y以选择:[n]"; then
+    test_single
+  else
+    echo -e "${Info} 退出脚本 ..." && exit 0
+  fi
+}
+
+test_alternative() {
+  select_alternative
+  set_alternative
+  result_alternative
+}
+
+select_alternative() {
+  echo -e "${Info} 选择需要测速的目标网络: \n1.中国电信\n2.中国联通\n3.中国移动\n4.教育网"
+  read -r -p "输入数字以选择:" ISP
+
+  while [[ ! "${ISP}" =~ ^[1-4]$ ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" ISP
+  done
+}
+
+set_alternative() {
+  [[ "${ISP}" == "1" ]] && node_1
+  [[ "${ISP}" == "2" ]] && node_2
+  [[ "${ISP}" == "3" ]] && node_3
+  [[ "${ISP}" == "4" ]] && node_4
+}
+
+node_1() {
+  echo -e "1.上海电信(天翼云)\n2.厦门电信CN2\n3.北京电信\n4.江苏电信\n5.广东深圳电信\n6.广州电信(天翼云)\n7.浙江电信" && read -r -p "输入数字以选择:" node
+
+  while [[ ! "${node}" =~ ^[1-7]$ ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" node
+  done
+
+  [[ "${node}" == "1" ]] && ISP_name="上海电信" && ip=101.89.132.9
+  [[ "${node}" == "2" ]] && ISP_name="厦门电信CN2" && ip=117.28.254.129
+  [[ "${node}" == "3" ]] && ISP_name="北京电信" && ip=120.92.180.135
+  [[ "${node}" == "4" ]] && ISP_name="江苏电信" && ip=221.229.173.233
+  [[ "${node}" == "5" ]] && ISP_name="广东深圳电信" && ip=116.6.211.41
+  [[ "${node}" == "6" ]] && ISP_name="广州电信(天翼云)" && ip=14.215.116.1
+  [[ "${node}" == "7" ]] && ISP_name="浙江电信" && ip=115.236.169.86
+}
+
+node_2() {
+  echo -e "1.上海联通\n2.重庆联通\n3.北京联通\n4.安徽合肥联通\n5.江苏南京联通\n6.浙江杭州联通\n7.广东联通" && read -r -p "输入数字以选择:" node
+
+  while [[ ! "${node}" =~ ^[1-7]$ ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" node
+  done
+
+  [[ "${node}" == "1" ]] && ISP_name="上海联通" && ip=220.196.252.174
+  [[ "${node}" == "2" ]] && ISP_name="重庆联通" && ip=113.207.32.65
+  [[ "${node}" == "3" ]] && ISP_name="北京联通" && ip=202.106.54.150
+  [[ "${node}" == "4" ]] && ISP_name="安徽合肥联通" && ip=112.122.10.26
+  [[ "${node}" == "5" ]] && ISP_name="江苏联通" && ip=112.85.231.129
+  [[ "${node}" == "6" ]] && ISP_name="浙江联通" && ip=60.12.214.156
+  [[ "${node}" == "7" ]] && ISP_name="广东联通" && ip=58.252.2.194
+}
+
+node_3() {
+  echo -e "1.上海移动\n2.四川成都移动\n3.北京移动\n4.浙江杭州移动\n5.广东移动\n6.江苏移动\n7.浙江移动" && read -r -p "输入数字以选择:" node
+
+  while [[ ! "${node}" =~ ^[1-7]$ ]]; do
+    echo -e "${Error} 无效输入"
+    echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" node
+  done
+
+  [[ "${node}" == "1" ]] && ISP_name="上海移动" && ip=117.184.42.114
+  [[ "${node}" == "2" ]] && ISP_name="四川成都移动" && ip=183.221.247.9
+  [[ "${node}" == "3" ]] && ISP_name="北京移动" && ip=111.13.217.125
+  [[ "${node}" == "4" ]] && ISP_name="浙江移动" && ip=183.246.69.139
+  [[ "${node}" == "5" ]] && ISP_name="广东移动" && ip=221.179.44.57
+  [[ "${node}" == "6" ]] && ISP_name="江苏移动" && ip=120.195.6.129
+  [[ "${node}" == "7" ]] && ISP_name="浙江移动" && ip=183.246.69.139
+}
+
+node_4() {
+  ISP_name="北京教育网" && ip=211.68.69.240
+}
+
+result_alternative() {
+  echo -e "${Info} 测试路由 到 ${ISP_name} 中 ..."
+  ${TRACECMD} ${ip} | grep -v -E 'NextTrace|XGadget-lab|Data\ Provider'
+  echo -e "${Info} 测试路由 到 ${ISP_name} 完成 ！"
+
+  repeat_test_alternative
+}
+
+repeat_test_alternative() {
+  echo -e "${Info} 是否继续测试其他节点?"
+  if ask_if "输入n/y以选择:[n]"; then
+    test_alternative
+  else
+    echo -e "${Info} 退出脚本 ..." && exit 0
+  fi
+}
+
+test_all() {
+  result_all '116.6.211.41' '广东东莞CN2'
+
+  result_all '101.95.110.149' '上海电信'
+
+  result_all '112.85.231.129' '江苏徐州联通'
+
+  result_all '120.199.239.1' '浙江杭州移动'
+
+  result_all '211.68.69.240' '北京教育网'
+
+  echo -e "${Info} 四网路由快速测试 已完成 ！"
+}
+
+result_all() {
+  ISP_name=$2
+  echo -e "${Info} 测试路由 到 ${ISP_name} 中 ..."
+  ${TRACECMD} "${1}" | grep -v -E 'NextTrace|XGadget-lab|Data\ Provider'
+  echo -e "${Info} 测试路由 到 ${ISP_name} 完成 ！"
+}
+
+check_root
+checkSystemDistribution
+ask_update_script
+checkNexttrace
+check_mode
+echo -e "${Info} 选择你要使用的功能: "
+echo -e "1.选择一个节点进行测试\n2.四网路由快速测试\n3.手动输入 ip 进行测试"
+read -r -p "输入数字以选择:" function
+
+while [[ ! "${function}" =~ ^[1-3]$ ]]; do
+  echo -e "${Error} 缺少或无效输入"
+  echo -e "${Info} 请重新选择" && read -r -p "输入数字以选择:" function
+done
+
+if [[ "${function}" == "1" ]]; then
+  test_alternative
+elif [[ "${function}" == "2" ]]; then
+  test_all
+else
+  test_single
+fi

reporter/reporter.go

@@ -0,0 +1,183 @@
+package reporter
+
+import (
+	"fmt"
+	"net"
+	"strings"
+	"sync"
+
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+	"github.com/xgadget-lab/nexttrace/trace"
+)
+
+type Reporter interface {
+	Print()
+}
+
+func New(rs *trace.Result, ip string) Reporter {
+	experimentTag()
+	r := reporter{
+		routeResult: rs,
+		targetIP:    ip,
+	}
+	return &r
+}
+
+type reporter struct {
+	targetTTL       uint16
+	targetIP        string
+	routeReport     map[uint16][]routeReportNode
+	routeReportLock sync.Mutex
+	routeResult     *trace.Result
+	wg              sync.WaitGroup
+}
+
+type routeReportNode struct {
+	asn string
+	isp string
+	geo []string
+	ix  bool
+}
+
+func experimentTag() {
+	fmt.Println("Route-Path 功能实验室")
+}
+
+func (r *reporter) generateRouteReportNode(ip string, ipGeoData ipgeo.IPGeoData, ttl uint16) {
+
+	var success bool = true
+
+	defer r.wg.Done()
+
+	rpn := routeReportNode{}
+	ptr, err := net.LookupAddr(ip)
+
+	if err == nil {
+		if strings.Contains(strings.ToLower(ptr[0]), "ix") {
+			rpn.ix = true
+		} else {
+			rpn.ix = false
+		}
+	}
+	// TODO: 这种写法不好，后面再重构一下
+	// 判断反向解析的域名中又或者是IP地理位置数据库中，是否出现了 IX
+	if strings.Contains(strings.ToLower(ipGeoData.Isp), "exchange") || strings.Contains(strings.ToLower(ipGeoData.Isp), "ix") || strings.Contains(strings.ToLower(ipGeoData.Owner), "exchange") || strings.Contains(strings.ToLower(ipGeoData.Owner), "ix") {
+		rpn.ix = true
+	}
+
+	// TODO: 正则判断POP并且提取带宽大小等信息
+
+	// CN2 需要特殊处理，因为他们很多没有ASN
+	// 但是目前这种写法是不规范的，属于凭空标记4809的IP
+	// TODO: 用更好的方式显示 CN2 骨干网的路由 Path
+	if strings.HasPrefix(ip, "59.43") {
+		rpn.asn = "4809"
+	} else {
+		rpn.asn = ipGeoData.Asnumber
+	}
+
+	// 无论最后一跳是否为存在地理位置信息（AnyCast），都应该给予显示
+	if (ipGeoData.Country == "" || ipGeoData.Country == "LAN Address" || ipGeoData.Country == "-") && ip != r.targetIP {
+		success = false
+	} else {
+		if ipGeoData.City == "" {
+			rpn.geo = []string{ipGeoData.Country, ipGeoData.Prov}
+		} else {
+			rpn.geo = []string{ipGeoData.Country, ipGeoData.City}
+		}
+	}
+	if ipGeoData.Asnumber == "" {
+		rpn.asn = "*"
+	}
+
+	if ipGeoData.Isp == "" {
+		rpn.isp = ipGeoData.Owner
+	} else {
+		rpn.isp = ipGeoData.Isp
+	}
+
+	// 有效记录
+	if success {
+		// 锁住资源，防止同时写panic
+		r.routeReportLock.Lock()
+		// 添加到MAP中
+		r.routeReport[ttl] = append(r.routeReport[ttl], rpn)
+		// 写入完成，解锁释放资源给其他协程
+		r.routeReportLock.Unlock()
+	}
+}
+
+func (r *reporter) InitialBaseData() Reporter {
+	reportNodes := map[uint16][]routeReportNode{}
+
+	r.routeReport = reportNodes
+	r.targetTTL = uint16(len(r.routeResult.Hops))
+
+	for i := uint16(0); i < r.targetTTL; i++ {
+		traceHop := r.routeResult.Hops[i][0]
+		if traceHop.Success {
+			currentIP := traceHop.Address.String()
+			r.wg.Add(1)
+			go r.generateRouteReportNode(currentIP, *traceHop.Geo, i)
+		}
+	}
+
+	// 等待所有的子协程运行完毕
+	r.wg.Wait()
+	return r
+}
+
+func (r *reporter) Print() {
+	var beforeActiveTTL uint16 = 0
+	r.InitialBaseData()
+	// 尝试首个有效 TTL
+	for i := uint16(0); i < r.targetTTL; i++ {
+		if len(r.routeReport[i]) != 0 {
+			beforeActiveTTL = i
+			// 找到以后便不再循环
+			break
+		}
+	}
+
+	for i := beforeActiveTTL; i < r.targetTTL; i++ {
+		// 计算该TTL内的数据长度，如果为0，则代表没有有效数据
+		if len(r.routeReport[i]) == 0 {
+			// 跳过改跃点的数据整理
+			continue
+		}
+		nodeReport := r.routeReport[i][0]
+
+		if i == beforeActiveTTL {
+			fmt.Printf("AS%s %s「%s『%s", nodeReport.asn, nodeReport.isp, nodeReport.geo[0], nodeReport.geo[1])
+		} else {
+			nodeReportBefore := r.routeReport[beforeActiveTTL][0]
+			// ASN 相同，同个 ISP 内部的数据传递
+			if nodeReportBefore.asn == nodeReport.asn {
+				// Same ASN but Coutry or City Changed
+				if nodeReportBefore.geo[0] != nodeReport.geo[0] {
+					fmt.Printf("』→ %s『%s", nodeReport.geo[0], nodeReport.geo[1])
+				} else {
+					if nodeReportBefore.geo[1] != nodeReport.geo[1] {
+						fmt.Printf(" → %s", nodeReport.geo[1])
+					}
+				}
+			} else {
+				// ASN 不同，跨 ISP 的数据传递，这里可能会出现 POP、IP Transit、Peer、Exchange
+				fmt.Printf("』」")
+				if int(i) != len(r.routeReport)+1 {
+					// 部分 Shell 客户端可能无法很好的展示这个特殊字符
+					// TODO: 寻找其他替代字符
+					fmt.Printf("\n ╭╯\n ╰")
+				}
+				if nodeReport.ix {
+					fmt.Printf("AS%s \033[42;37mIXP\033[0m %s「%s『%s", nodeReport.asn, nodeReport.isp, nodeReport.geo[0], nodeReport.geo[1])
+				} else {
+					fmt.Printf("AS%s %s「%s『%s", nodeReport.asn, nodeReport.isp, nodeReport.geo[0], nodeReport.geo[1])
+				}
+			}
+		}
+		// 标记为最新的一个有效跃点
+		beforeActiveTTL = i
+	}
+	fmt.Println("』」")
+}

reporter/reporter_test.go

@@ -0,0 +1,115 @@
+package reporter
+
+import (
+	"net"
+	"testing"
+	"time"
+
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+	"github.com/xgadget-lab/nexttrace/trace"
+)
+
+var testResult = &trace.Result{
+	Hops: [][]trace.Hop{
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("192.168.3.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo: &ipgeo.IPGeoData{
+					Asnumber: "4808",
+					Country:  "中国",
+					Prov:     "北京市",
+					City:     "北京市",
+					District: "北京市",
+					Owner:    "",
+					Isp:      "中国联通",
+				},
+			},
+		},
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("114.249.16.1")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo: &ipgeo.IPGeoData{
+					Asnumber: "4808",
+					Country:  "中国",
+					Prov:     "北京市",
+					City:     "北京市",
+					District: "北京市",
+					Owner:    "",
+					Isp:      "中国联通",
+				},
+			},
+		},
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("219.158.5.150")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo: &ipgeo.IPGeoData{
+					Asnumber: "4837",
+					Country:  "中国",
+					Prov:     "",
+					City:     "",
+					District: "",
+					Owner:    "",
+					Isp:      "中国联通",
+				},
+			},
+		},
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("62.115.125.160")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo: &ipgeo.IPGeoData{
+					Asnumber: "1299",
+					Country:  "Sweden",
+					Prov:     "Stockholm County",
+					City:     "Stockholm",
+					District: "",
+					Owner:    "",
+					Isp:      "Telia Company AB",
+				},
+			},
+		},
+		{
+			{
+				Success:  true,
+				Address:  &net.IPAddr{IP: net.ParseIP("213.226.68.73")},
+				Hostname: "test",
+				TTL:      0,
+				RTT:      10 * time.Millisecond,
+				Error:    nil,
+				Geo: &ipgeo.IPGeoData{
+					Asnumber: "56630",
+					Country:  "Germany",
+					Prov:     "Hesse, Frankfurt",
+					City:     "",
+					District: "",
+					Owner:    "",
+					Isp:      "Melbikomas UAB",
+				},
+			},
+		},
+	},
+}
+
+func TestPrint(t *testing.T) {
+	r := New(testResult, "213.226.68.73")
+	r.Print()
+}

signal/signal.go

@@ -1,19 +0,0 @@
-package signal
-
-type Signal struct {
-	sigChan chan struct{}
-}
-
-func New() *Signal {
-	return &Signal{sigChan: make(chan struct{}, 1)}
-}
-
-func (s *Signal) Signal() {
-	if len(s.sigChan) == 0 {
-		s.sigChan <- struct{}{}
-	}
-}
-
-func (s *Signal) Chan() chan struct{} {
-	return s.sigChan
-}

taskgroup/taskgroup.go

@@ -1,45 +0,0 @@
-package taskgroup
-
-import (
-	"sync"
-)
-
-type TaskGroup struct {
-	count int
-	mu    sync.Mutex
-	done  []chan struct{}
-}
-
-func New() *TaskGroup {
-	return &TaskGroup{
-		count: 0,
-		mu:    sync.Mutex{},
-		done:  []chan struct{}{},
-	}
-}
-
-func (t *TaskGroup) Add() {
-	t.mu.Lock()
-	defer t.mu.Unlock()
-	t.count++
-}
-
-func (t *TaskGroup) Done() {
-	t.mu.Lock()
-	defer t.mu.Unlock()
-	if t.count-1 == 0 {
-		for _, doneChannel := range t.done {
-			doneChannel <- struct{}{}
-		}
-		t.done = []chan struct{}{}
-	}
-	t.count--
-}
-
-func (t *TaskGroup) Wait() {
-	doneChannel := make(chan struct{})
-	t.mu.Lock()
-	t.done = append(t.done, doneChannel)
-	t.mu.Unlock()
-	<-doneChannel
-}

trace/icmp_ipv4.go

@@ -0,0 +1,175 @@
+package trace
+
+import (
+	"log"
+	"net"
+	"os"
+	"sync"
+	"time"
+
+	"golang.org/x/net/context"
+	"golang.org/x/net/icmp"
+	"golang.org/x/net/ipv4"
+)
+
+type ICMPTracer struct {
+	Config
+	wg         sync.WaitGroup
+	res        Result
+	ctx        context.Context
+	resCh      chan Hop
+	icmpListen net.PacketConn
+	final      int
+	finalLock  sync.Mutex
+}
+
+func (t *ICMPTracer) Execute() (*Result, error) {
+	if len(t.res.Hops) > 0 {
+		return &t.res, ErrTracerouteExecuted
+	}
+
+	var err error
+
+	t.icmpListen, err = net.ListenPacket("ip4:1", "0.0.0.0")
+	if err != nil {
+		return &t.res, err
+	}
+	defer t.icmpListen.Close()
+
+	var cancel context.CancelFunc
+	t.ctx, cancel = context.WithCancel(context.Background())
+	defer cancel()
+	t.resCh = make(chan Hop)
+	t.final = -1
+
+	go t.listenICMP()
+
+	for ttl := 1; ttl <= t.MaxHops; ttl++ {
+		if t.final != -1 && ttl > t.final {
+			break
+		}
+		for i := 0; i < t.NumMeasurements; i++ {
+			t.wg.Add(1)
+			go t.send(ttl)
+		}
+		// 一组TTL全部退出（收到应答或者超时终止）以后，再进行下一个TTL的包发送
+		t.wg.Wait()
+		if t.RealtimePrinter != nil {
+			t.RealtimePrinter(&t.res, ttl-1)
+		}
+	}
+	t.res.reduce(t.final)
+
+	return &t.res, nil
+}
+
+func (t *ICMPTracer) listenICMP() {
+	lc := NewPacketListener(t.icmpListen, 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, 0, rm.Body.(*icmp.TimeExceeded).Data)
+			case ipv4.ICMPTypeEchoReply:
+				t.handleICMPMessage(msg, 1, rm.Body.(*icmp.Echo).Data)
+			default:
+				// log.Println("received icmp message of unknown type", rm.Type)
+			}
+		}
+	}
+
+}
+
+func (t *ICMPTracer) handleICMPMessage(msg ReceivedMessage, icmpType int8, data []byte) {
+	t.resCh <- Hop{
+		Success: true,
+		Address: msg.Peer,
+	}
+}
+
+func (t *ICMPTracer) send(ttl int) error {
+	defer t.wg.Done()
+	if t.final != -1 && ttl > t.final {
+		return nil
+	}
+
+	icmpHeader := icmp.Message{
+		Type: ipv4.ICMPTypeEcho, Code: 0,
+		Body: &icmp.Echo{
+			ID:   os.Getpid() & 0xffff,
+			Data: []byte("HELLO-R-U-THERE"),
+		},
+	}
+
+	ipv4.NewPacketConn(t.icmpListen).SetTTL(ttl)
+
+	wb, err := icmpHeader.Marshal(nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	start := time.Now()
+	if _, err := t.icmpListen.WriteTo(wb, &net.IPAddr{IP: t.DestIP}); err != nil {
+		log.Fatal(err)
+	}
+	if err := t.icmpListen.SetReadDeadline(time.Now().Add(3 * time.Second)); err != nil {
+		log.Fatal(err)
+	}
+
+	select {
+	case <-t.ctx.Done():
+		return nil
+	case h := <-t.resCh:
+		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.TCPAddr); 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
+}

trace/icmp_ipv6.go

@@ -0,0 +1,179 @@
+package trace
+
+import (
+	"log"
+	"net"
+	"os"
+	"sync"
+	"time"
+
+	"golang.org/x/net/context"
+	"golang.org/x/net/icmp"
+	"golang.org/x/net/ipv6"
+)
+
+type ICMPTracerv6 struct {
+	Config
+	wg         sync.WaitGroup
+	res        Result
+	ctx        context.Context
+	resCh      chan Hop
+	icmpListen net.PacketConn
+	final      int
+	finalLock  sync.Mutex
+}
+
+func (t *ICMPTracerv6) Execute() (*Result, error) {
+	if len(t.res.Hops) > 0 {
+		return &t.res, ErrTracerouteExecuted
+	}
+
+	var err error
+
+	t.icmpListen, err = net.ListenPacket("ip6:58", "::")
+	if err != nil {
+		return &t.res, err
+	}
+	defer t.icmpListen.Close()
+
+	var cancel context.CancelFunc
+	t.ctx, cancel = context.WithCancel(context.Background())
+	defer cancel()
+	t.resCh = make(chan Hop)
+	t.final = -1
+
+	go t.listenICMP()
+
+	for ttl := 1; ttl <= t.MaxHops; ttl++ {
+		if t.final != -1 && ttl > t.final {
+			break
+		}
+		for i := 0; i < t.NumMeasurements; i++ {
+			t.wg.Add(1)
+			go t.send(ttl)
+		}
+		// 一组TTL全部退出（收到应答或者超时终止）以后，再进行下一个TTL的包发送
+		t.wg.Wait()
+		if t.RealtimePrinter != nil {
+			t.RealtimePrinter(&t.res, ttl-1)
+		}
+	}
+	t.res.reduce(t.final)
+
+	return &t.res, nil
+}
+
+func (t *ICMPTracerv6) listenICMP() {
+	lc := NewPacketListener(t.icmpListen, 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(58, msg.Msg[:*msg.N])
+			if err != nil {
+				log.Println(err)
+				continue
+			}
+			// log.Println(msg.Peer)
+			switch rm.Type {
+			case ipv6.ICMPTypeTimeExceeded:
+				t.handleICMPMessage(msg, 0, rm.Body.(*icmp.TimeExceeded).Data)
+			case ipv6.ICMPTypeEchoReply:
+				t.handleICMPMessage(msg, 1, rm.Body.(*icmp.Echo).Data)
+			default:
+				// log.Println("received icmp message of unknown type", rm.Type)
+			}
+		}
+	}
+
+}
+
+func (t *ICMPTracerv6) handleICMPMessage(msg ReceivedMessage, icmpType int8, data []byte) {
+	t.resCh <- Hop{
+		Success: true,
+		Address: msg.Peer,
+	}
+}
+
+func (t *ICMPTracerv6) send(ttl int) error {
+	defer t.wg.Done()
+	if t.final != -1 && ttl > t.final {
+		return nil
+	}
+
+	icmpHeader := icmp.Message{
+		Type: ipv6.ICMPTypeEchoRequest, Code: 0,
+		Body: &icmp.Echo{
+			ID:   os.Getpid() & 0xffff,
+			Data: []byte("HELLO-R-U-THERE"),
+		},
+	}
+
+	p := ipv6.NewPacketConn(t.icmpListen)
+
+	icmpHeader.Body.(*icmp.Echo).Seq = ttl
+	p.SetHopLimit(ttl)
+
+	wb, err := icmpHeader.Marshal(nil)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	start := time.Now()
+	if _, err := t.icmpListen.WriteTo(wb, &net.IPAddr{IP: t.DestIP}); err != nil {
+		log.Fatal(err)
+	}
+	if err := t.icmpListen.SetReadDeadline(time.Now().Add(3 * time.Second)); err != nil {
+		log.Fatal(err)
+	}
+
+	select {
+	case <-t.ctx.Done():
+		return nil
+	case h := <-t.resCh:
+		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.TCPAddr); 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
+}

trace/packet_listener.go

@@ -1,4 +1,4 @@
-package listener_channel
+package trace
 
 import (
 	"golang.org/x/net/context"
@@ -13,21 +13,19 @@ type ReceivedMessage struct {
 	Err  error
 }
 
-type ListenerChannel struct {
+type PacketListener struct {
 	ctx      context.Context
-	cancel   context.CancelFunc
 	Conn     net.PacketConn
 	Messages chan ReceivedMessage
 }
 
-func New(conn net.PacketConn) *ListenerChannel {
-	ctx, cancel := context.WithCancel(context.Background())
+func NewPacketListener(conn net.PacketConn, ctx context.Context) *PacketListener {
 	results := make(chan ReceivedMessage, 50)
 
-	return &ListenerChannel{Conn: conn, ctx: ctx, cancel: cancel, Messages: results}
+	return &PacketListener{Conn: conn, ctx: ctx, Messages: results}
 }
 
-func (l *ListenerChannel) Start() {
+func (l *PacketListener) Start() {
 	for {
 		select {
 		case <-l.ctx.Done():
@@ -55,7 +53,3 @@ func (l *ListenerChannel) Start() {
 		}
 	}
 }
-
-func (l *ListenerChannel) Stop() {
-	l.cancel()
-}

trace/tcp_ipv4.go

@@ -0,0 +1,272 @@
+package trace
+
+import (
+	"log"
+	"math"
+	"math/rand"
+	"net"
+	"sync"
+	"time"
+
+	"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"
+)
+
+type TCPTracer struct {
+	Config
+	wg                  sync.WaitGroup
+	res                 Result
+	ctx                 context.Context
+	inflightRequest     map[int]chan Hop
+	inflightRequestLock sync.Mutex
+	SrcIP               net.IP
+	icmp                net.PacketConn
+	tcp                 net.PacketConn
+
+	final     int
+	finalLock sync.Mutex
+
+	sem *semaphore.Weighted
+}
+
+func (t *TCPTracer) Execute() (*Result, error) {
+	if len(t.res.Hops) > 0 {
+		return &t.res, ErrTracerouteExecuted
+	}
+
+	t.SrcIP, _ = util.LocalIPPort(t.DestIP)
+
+	var err error
+	t.tcp, err = net.ListenPacket("ip4:tcp", t.SrcIP.String())
+	if err != nil {
+		return nil, err
+	}
+	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()
+	go t.listenTCP()
+
+	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)
+
+		}
+		time.Sleep(1 * time.Millisecond)
+	}
+	t.wg.Wait()
+	t.res.reduce(t.final)
+
+	return &t.res, nil
+}
+
+func (t *TCPTracer) 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)
+			}
+		}
+	}
+
+}
+
+// @title    listenTCP
+// @description   监听TCP的响应数据包
+func (t *TCPTracer) listenTCP() {
+	lc := NewPacketListener(t.tcp, t.ctx)
+	go lc.Start()
+
+	for {
+		select {
+		case <-t.ctx.Done():
+			return
+		case msg := <-lc.Messages:
+			if msg.N == nil {
+				continue
+			}
+			if msg.Peer.String() != t.DestIP.String() {
+				continue
+			}
+
+			// 解包
+			packet := gopacket.NewPacket(msg.Msg[:*msg.N], layers.LayerTypeTCP, gopacket.Default)
+			// 从包中获取TCP layer信息
+			if tcpLayer := packet.Layer(layers.LayerTypeTCP); tcpLayer != nil {
+				tcp, _ := tcpLayer.(*layers.TCP)
+				// 取得目标主机的Sequence Number
+
+				if ch, ok := t.inflightRequest[int(tcp.Ack-1)]; ok {
+					// 最后一跳
+					ch <- Hop{
+						Success: true,
+						Address: msg.Peer,
+					}
+				}
+			}
+		}
+	}
+}
+
+func (t *TCPTracer) handleICMPMessage(msg ReceivedMessage, data []byte) {
+	header, err := util.GetICMPResponsePayload(data)
+	if err != nil {
+		return
+	}
+	sequenceNumber := util.GetTCPSeq(header)
+	t.inflightRequestLock.Lock()
+	defer t.inflightRequestLock.Unlock()
+	ch, ok := t.inflightRequest[int(sequenceNumber)]
+	if !ok {
+		return
+	}
+	ch <- Hop{
+		Success: true,
+		Address: msg.Peer,
+	}
+
+}
+
+func (t *TCPTracer) 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
+	}
+	// 随机种子
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	_, srcPort := util.LocalIPPort(t.DestIP)
+	ipHeader := &layers.IPv4{
+		SrcIP:    t.SrcIP,
+		DstIP:    t.DestIP,
+		Protocol: layers.IPProtocolTCP,
+		TTL:      uint8(ttl),
+	}
+	// 使用Uint16兼容32位系统，防止在rand的时候因使用int32而溢出
+	sequenceNumber := uint32(r.Intn(math.MaxUint16))
+	tcpHeader := &layers.TCP{
+		SrcPort: layers.TCPPort(srcPort),
+		DstPort: layers.TCPPort(t.DestPort),
+		Seq:     sequenceNumber,
+		SYN:     true,
+		Window:  14600,
+	}
+	_ = tcpHeader.SetNetworkLayerForChecksum(ipHeader)
+
+	buf := gopacket.NewSerializeBuffer()
+	opts := gopacket.SerializeOptions{
+		ComputeChecksums: true,
+		FixLengths:       true,
+	}
+	if err := gopacket.SerializeLayers(buf, opts, tcpHeader); err != nil {
+		return err
+	}
+
+	err = ipv4.NewPacketConn(t.tcp).SetTTL(ttl)
+	if err != nil {
+		return err
+	}
+
+	start := time.Now()
+	if _, err := t.tcp.WriteTo(buf.Bytes(), &net.IPAddr{IP: t.DestIP}); err != nil {
+		return err
+	}
+	t.inflightRequestLock.Lock()
+	hopCh := make(chan Hop)
+	t.inflightRequest[int(sequenceNumber)] = hopCh
+	t.inflightRequestLock.Unlock()
+	/*
+		// 这里属于 2个Sender，N个Reciever的情况，在哪里关闭Channel都容易导致Panic
+		defer func() {
+			t.inflightRequestLock.Lock()
+			close(hopCh)
+			delete(t.inflightRequest, srcPort)
+			t.inflightRequestLock.Unlock()
+		}()
+	*/
+	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.TCPAddr); 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
+}

trace/tcp_ipv6.go

@@ -0,0 +1,265 @@
+package trace
+
+import (
+	"log"
+	"math"
+	"math/rand"
+	"net"
+	"sync"
+	"time"
+
+	"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/ipv6"
+	"golang.org/x/sync/semaphore"
+)
+
+type TCPTracerv6 struct {
+	Config
+	wg                  sync.WaitGroup
+	res                 Result
+	ctx                 context.Context
+	inflightRequest     map[int]chan Hop
+	inflightRequestLock sync.Mutex
+	SrcIP               net.IP
+	icmp                net.PacketConn
+	tcp                 net.PacketConn
+
+	final     int
+	finalLock sync.Mutex
+
+	sem *semaphore.Weighted
+}
+
+func (t *TCPTracerv6) Execute() (*Result, error) {
+	if len(t.res.Hops) > 0 {
+		return &t.res, ErrTracerouteExecuted
+	}
+
+	t.SrcIP, _ = util.LocalIPPort(t.DestIP)
+	log.Println(util.LocalIPPort(t.DestIP))
+	var err error
+	t.tcp, err = net.ListenPacket("ip6:tcp", t.SrcIP.String())
+	if err != nil {
+		return nil, err
+	}
+	t.icmp, err = icmp.ListenPacket("ip6:53", "::")
+	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()
+	go t.listenTCP()
+
+	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)
+
+		}
+		time.Sleep(1 * time.Millisecond)
+	}
+	t.wg.Wait()
+	t.res.reduce(t.final)
+
+	return &t.res, nil
+}
+
+func (t *TCPTracerv6) 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(53, msg.Msg[:*msg.N])
+			if err != nil {
+				log.Println(err)
+				continue
+			}
+			log.Println(msg.Peer)
+			switch rm.Type {
+			case ipv6.ICMPTypeTimeExceeded:
+				t.handleICMPMessage(msg, rm.Body.(*icmp.TimeExceeded).Data)
+			case ipv6.ICMPTypeDestinationUnreachable:
+				t.handleICMPMessage(msg, rm.Body.(*icmp.DstUnreach).Data)
+			default:
+				//log.Println("received icmp message of unknown type", rm.Type)
+			}
+		}
+	}
+
+}
+
+// @title    listenTCP
+// @description   监听TCP的响应数据包
+func (t *TCPTracerv6) listenTCP() {
+	lc := NewPacketListener(t.tcp, t.ctx)
+	go lc.Start()
+
+	for {
+		select {
+		case <-t.ctx.Done():
+			return
+		case msg := <-lc.Messages:
+			if msg.N == nil {
+				continue
+			}
+			if msg.Peer.String() != t.DestIP.String() {
+				continue
+			}
+
+			// 解包
+			packet := gopacket.NewPacket(msg.Msg[:*msg.N], layers.LayerTypeTCP, gopacket.Default)
+			// 从包中获取TCP layer信息
+			if tcpLayer := packet.Layer(layers.LayerTypeTCP); tcpLayer != nil {
+				tcp, _ := tcpLayer.(*layers.TCP)
+				// 取得目标主机的Sequence Number
+
+				if ch, ok := t.inflightRequest[int(tcp.Ack-1)]; ok {
+					// 最后一跳
+					ch <- Hop{
+						Success: true,
+						Address: msg.Peer,
+					}
+				}
+			}
+		}
+	}
+}
+
+func (t *TCPTracerv6) handleICMPMessage(msg ReceivedMessage, data []byte) {
+	header, err := util.GetICMPResponsePayload(data)
+	if err != nil {
+		return
+	}
+	sequenceNumber := util.GetTCPSeq(header)
+	t.inflightRequestLock.Lock()
+	defer t.inflightRequestLock.Unlock()
+	ch, ok := t.inflightRequest[int(sequenceNumber)]
+	if !ok {
+		return
+	}
+	ch <- Hop{
+		Success: true,
+		Address: msg.Peer,
+	}
+
+}
+
+func (t *TCPTracerv6) 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
+	}
+	// 随机种子
+	r := rand.New(rand.NewSource(time.Now().UnixNano()))
+	_, srcPort := util.LocalIPPort(t.DestIP)
+	ipHeader := &layers.IPv6{
+		SrcIP:      t.SrcIP,
+		DstIP:      t.DestIP,
+		NextHeader: layers.IPProtocolTCP,
+		HopLimit:   uint8(ttl),
+	}
+	// 使用Uint16兼容32位系统，防止在rand的时候因使用int32而溢出
+	sequenceNumber := uint32(r.Intn(math.MaxUint16))
+	tcpHeader := &layers.TCP{
+		SrcPort: layers.TCPPort(srcPort),
+		DstPort: layers.TCPPort(t.DestPort),
+		Seq:     sequenceNumber,
+		SYN:     true,
+		Window:  14600,
+	}
+	_ = tcpHeader.SetNetworkLayerForChecksum(ipHeader)
+
+	buf := gopacket.NewSerializeBuffer()
+	opts := gopacket.SerializeOptions{
+		ComputeChecksums: true,
+		FixLengths:       true,
+	}
+	if err := gopacket.SerializeLayers(buf, opts, tcpHeader); err != nil {
+		return err
+	}
+
+	ipv6.NewPacketConn(t.tcp).SetHopLimit(ttl)
+	if err != nil {
+		return err
+	}
+
+	start := time.Now()
+	if _, err := t.tcp.WriteTo(buf.Bytes(), &net.IPAddr{IP: t.DestIP}); err != nil {
+		return err
+	}
+	t.inflightRequestLock.Lock()
+	hopCh := make(chan Hop)
+	t.inflightRequest[int(sequenceNumber)] = hopCh
+	t.inflightRequestLock.Unlock()
+
+	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.TCPAddr); 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
+}

trace/temp_printer.go

@@ -0,0 +1,76 @@
+package trace
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+)
+
+func HopPrinter(h 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 {
+		// 有些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, ", ")
+}

trace/trace.go

@@ -0,0 +1,126 @@
+package trace
+
+import (
+	"errors"
+	"net"
+	"sync"
+	"time"
+
+	"github.com/xgadget-lab/nexttrace/ipgeo"
+)
+
+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
+	RealtimePrinter  func(res *Result, ttl int)
+}
+
+type Method string
+
+const (
+	ICMPTrace Method = "icmp"
+	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 ICMPTrace:
+		if config.DestIP.To4() != nil {
+			tracer = &ICMPTracer{Config: config}
+		} else {
+			tracer = &ICMPTracerv6{Config: config}
+		}
+
+	case UDPTrace:
+		if config.DestIP.To4() != nil {
+			tracer = &UDPTracer{Config: config}
+		} else {
+			return nil, errors.New("IPv6 UDP Traceroute is not supported")
+		}
+	case TCPTrace:
+		if config.DestIP.To4() != nil {
+			tracer = &TCPTracer{Config: config}
+		} else {
+			// tracer = &TCPTracerv6{Config: config}
+			return nil, errors.New("IPv6 TCP Traceroute is not supported")
+		}
+	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
+}

trace/udp.go

@@ -0,0 +1,257 @@
+package trace
+
+import (
+	"log"
+	"net"
+	"sync"
+	"time"
+
+	"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"
+)
+
+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
+	}
+
+	// 在对inflightRequest进行写操作的时候应该加锁保护，以免多个goroutine协程试图同时写入造成panic
+	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
+}

util/methods.go

@@ -1,11 +0,0 @@
-package util
-
-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"`
-}

util/printer/printer.go

@@ -1,103 +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) {
-	for hi := uint16(1); hi < 30; hi++ {
-		fmt.Print(hi)
-		for _, v := range res[hi] {
-			hopPrinter(v)
-			if v.Address != nil && ip.String() == v.Address.String() {
-				hi = 31
-			}
-		}
-	}
-}
-
-func hopPrinter(v2 methods.TracerouteHop) {
-	if v2.Address == nil {
-		fmt.Println("\t*")
-	} else {
-		var iPGeoData *ipgeo.IPGeoData
-		var err error
-
-		ipStr := v2.Address.String()
-
-		// TODO: 判断 err 返回，并且在CLI终端提示错误
-		if dataOrigin == "LeoMoeAPI" {
-			iPGeoData, err = ipgeo.LeoIP(ipStr)
-		} else if dataOrigin == "IP.SB" {
-			iPGeoData, err = ipgeo.IPSB(ipStr)
-		} else if dataOrigin == "IPInfo" {
-			iPGeoData, err = ipgeo.IPInfo(ipStr)
-		} else if dataOrigin == "IPInsight" {
-			iPGeoData, err = ipgeo.IPInSight(ipStr)
-		} else {
-			iPGeoData, err = ipgeo.LeoIP(ipStr)
-		}
-
-		geo := ""
-		if err != nil {
-			geo = fmt.Sprint("Error: ", err)
-		} else {
-			geo = formatIpGeoData(ipStr, iPGeoData)
-		}
-
-		ptr, err := net.LookupAddr(ipStr)
-
-		txt := "\t"
-		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)
-		}
-		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.31.") || strings.HasPrefix(ip, "11.72.") {
-		res = append(res, "局域网", "腾讯云")
-	} else if strings.HasPrefix(ip, "11.13.") {
-		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, ", ")
-}

util/printer/tableprinter.go

@@ -1,114 +0,0 @@
-package printer
-
-import (
-	"fmt"
-	"net"
-
-	"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) {
-	// 初始化表格
-	tbl := New()
-	for hi := uint16(1); hi < 30; hi++ {
-		for _, v := range res[hi] {
-			data := tableDataGenerator(v)
-			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) *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()
-
-		// TODO: 判断 err 返回，并且在CLI终端提示错误
-		if dataOrigin == "LeoMoeAPI" {
-			iPGeoData, err = ipgeo.LeoIP(ipStr)
-		} else if dataOrigin == "IP.SB" {
-			iPGeoData, err = ipgeo.IPSB(ipStr)
-		} else if dataOrigin == "IPInfo" {
-			iPGeoData, err = ipgeo.IPInfo(ipStr)
-		} else if dataOrigin == "IPInsight" {
-			iPGeoData, err = ipgeo.IPInSight(ipStr)
-		} else {
-			iPGeoData, err = ipgeo.LeoIP(ipStr)
-		}
-
-		ptr, err_LookupAddr := net.LookupAddr(ipStr)
-
-		lantency = fmt.Sprintf("%.2fms", v2.RTT.Seconds()*1000)
-
-		if err_LookupAddr != nil {
-			IP = fmt.Sprint(ipStr)
-		} else {
-			IP = fmt.Sprint(ptr[0], " (", ipStr, ") ")
-		}
-
-		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,
-			}
-		}
-	}
-}

util/trace.go

@@ -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)
+}

util/util.go

@@ -36,12 +36,13 @@ func DomainLookUp(host string) net.IP {
 	var ipv6Flag = false
 
 	for _, ip := range ips {
+		ipSlice = append(ipSlice, ip)
 		// 仅返回ipv4的ip
-		if ip.To4() != nil {
-			ipSlice = append(ipSlice, ip)
-		} else {
-			ipv6Flag = true
-		}
+		// if ip.To4() != nil {
+		// 	ipSlice = append(ipSlice, ip)
+		// } else {
+		// 	ipv6Flag = true
+		// }
 	}
 
 	if ipv6Flag {