update

Author: wardlin

.gitignore

@@ -0,0 +1,15 @@
+# Binaries for programs and plugins
+*.exe
+*.exe~
+*.dll
+*.so
+*.dylib
+
+# Test binary, built with `go test -c`
+*.test
+
+# Output of the go coverage tool, specifically when used with LiteIDE
+*.out
+
+# Dependency directories (remove the comment below to include it)
+# vendor/

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 Mie~~~
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -0,0 +1,8 @@
+module github.com/eaglesunshine/trace
+
+go 1.16
+
+require (
+	github.com/sirupsen/logrus v1.8.1
+	golang.org/x/net v0.0.0-20210813160813-60bc85c4be6d
+)

go.mod

@@ -0,0 +1,17 @@
+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/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
+github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
+github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+golang.org/x/net v0.0.0-20210813160813-60bc85c4be6d h1:LO7XpTYMwTqxjLcGWPijK3vRXg1aWdlNOVOHRq45d7c=
+golang.org/x/net v0.0.0-20210813160813-60bc85c4be6d/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y=
+golang.org/x/sys v0.0.0-20191026070338-33540a1f6037/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210423082822-04245dca01da h1:b3NXsE2LusjYGGjL5bxEVZZORm/YEFFrWFjR8eFrw/c=
+golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=

go.sum

@@ -0,0 +1,109 @@
+package ztrace
+
+import (
+	"encoding/binary"
+	"net"
+	"sync/atomic"
+	"time"
+
+	"github.com/sirupsen/logrus"
+	"golang.org/x/net/ipv4"
+)
+
+func (t *TraceRoute) SendIPv4ICMP() {
+	key := GetHash(t.netSrcAddr.To4(), t.netDstAddr.To4(), 65535, 65535, 1)
+	db := NewStatsDB(key)
+
+	t.DB.Store(key, db)
+	go db.Cache.Run()
+	conn, err := net.ListenPacket("ip4:icmp", t.netSrcAddr.String())
+	if err != nil {
+		logrus.Error(err)
+		return
+	}
+	defer conn.Close()
+
+	rSocket, err := ipv4.NewRawConn(conn)
+	if err != nil {
+		logrus.Error("can not create raw socket:", err)
+		return
+	}
+	defer rSocket.Close()
+
+	id := uint16(1)
+	mod := uint16(1 << 15)
+
+	for ttl := 1; ttl <= int(t.MaxTTL); ttl++ {
+		hdr, payload := t.BuildIPv4ICMP(uint8(ttl), id, id, 0)
+		rSocket.WriteTo(hdr, payload, nil)
+		report := &SendMetric{
+			FlowKey:   key,
+			ID:        uint32(hdr.ID),
+			TTL:       uint8(ttl),
+			TimeStamp: time.Now(),
+		}
+		t.SendChan <- report
+		id = (id + 1) % mod
+
+		logrus.Info("send icmp ttl:", ttl)
+
+		//atomic.AddUint64(db.SendCnt, 1)
+	}
+}
+
+func (t *TraceRoute) ListenIPv4ICMP() {
+	defer t.Stop()
+
+	laddr := &net.IPAddr{IP: t.netSrcAddr}
+
+	var err error
+	t.recvICMPConn, err = net.ListenIP("ip4:icmp", laddr)
+
+	if err != nil {
+		logrus.Error("bind failure:", err)
+		return
+	}
+	defer t.recvICMPConn.Close()
+
+	//设置超时时间
+	t.recvICMPConn.SetReadDeadline(time.Now().Add(t.Timeout))
+
+	for {
+		if atomic.LoadInt32(t.stopSignal) == 1 {
+			return
+		}
+
+		buf := make([]byte, 1500)
+		n, raddr, err := t.recvICMPConn.ReadFrom(buf)
+		if err != nil {
+			logrus.Error("recvICMPConn.ReadFrom failed:", err)
+			return
+		}
+
+		icmpType := buf[0]
+		//logrus.Info(raddr, "|", icmpType, "|", n)
+
+		//TTL Exceeded or Port Unreachable
+		if (icmpType == 11 || (icmpType == 3 && buf[1] == 3)) && (n >= 36) {
+			id := binary.BigEndian.Uint16(buf[32:34])
+
+			dstip := net.IP(buf[24:28])
+			srcip := net.IP(buf[20:24])
+
+			if dstip.Equal(t.netDstAddr) {
+				key := GetHash(srcip, dstip, 65535, 65535, 1)
+
+				m := &RecvMetric{
+					FlowKey:   key,
+					ID:        uint32(id),
+					RespAddr:  raddr.String(),
+					TimeStamp: time.Now(),
+				}
+				logrus.Info("recv icmp ttl:", id)
+
+				t.RecvChan <- m
+			}
+		}
+	}
+
+}

icmp.go

@@ -0,0 +1,130 @@
+package ztrace
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"golang.org/x/net/icmp"
+	"golang.org/x/net/ipv6"
+	"math/rand"
+	"net"
+	"time"
+)
+
+func (t *TraceRoute) TraceIpv6ICMP() {
+	var dst net.IPAddr
+
+	if _, err := t.dnsResolve(t.Dest, true, &dst); err != nil {
+		logrus.Error("TraceIpv6ICMP failed:", err)
+		return
+	}
+
+	//TODO：获取本地ipv6地址
+	localAddr := "::"
+
+	icmp6Sock, err := net.ListenPacket("ip6:ipv6-icmp", localAddr)
+	if err != nil {
+		logrus.Error("Could not set a listening ICMP6 socket: %s\n", err)
+		return
+	}
+	defer icmp6Sock.Close()
+
+	ipv6Sock := ipv6.NewPacketConn(icmp6Sock)
+	defer ipv6Sock.Close()
+
+	if err := ipv6Sock.SetControlMessage(ipv6.FlagHopLimit|ipv6.FlagDst|ipv6.FlagInterface|ipv6.FlagSrc, true); err != nil {
+		logrus.Error("Could not set options on the ipv6 socket: %s\n", err)
+		return
+	}
+
+	icmp6Echo := icmp.Message{
+		Type: ipv6.ICMPTypeEchoRequest, Code: 0, Body: &icmp.Echo{ID: rand.Int(), Data: []byte("")},
+	}
+
+	buf := make([]byte, 1500)
+
+	isDest := false
+	for i := 1; i <= int(t.MaxTTL); i++ {
+
+		hopData := HopData{
+			Hop: i,
+		}
+
+		for j := 0; j < t.MaxPath; j++ {
+
+			icmp6Echo.Body.(*icmp.Echo).Seq = i
+
+			buffer, err := icmp6Echo.Marshal(nil)
+
+			if err != nil {
+				logrus.Error("Could not serialize the ICMP6 echo request: %s\n", err)
+				return
+			}
+
+			if err := ipv6Sock.SetHopLimit(i); err != nil {
+				logrus.Error("Could not set the HopLimit field: %s\n", err)
+				return
+			}
+
+			timeNow := time.Now()
+
+			if _, err := ipv6Sock.WriteTo(buffer, nil, &dst); err != nil {
+				logrus.Error("Could not send the ICMP6 echo packet: %s\n", err)
+				return
+			}
+
+			if err := ipv6Sock.SetReadDeadline(time.Now().Add(t.Timeout)); err != nil {
+				logrus.Error("Could not set the read timeout on the ipv6 socket: %s\n", err)
+				return
+			}
+
+			n, _, node, err := ipv6Sock.ReadFrom(buf)
+
+			//收集一个响应包的数据
+			hop := map[string]interface{}{}
+
+			if err != nil {
+				fmt.Printf("%d %40s\n", i, "*")
+			} else {
+				answer, err := icmp.ParseMessage(58, buf[:n])
+
+				if err != nil {
+					logrus.Error("Could not parse the ICMP6 packet from: %s\n", node.String())
+					return
+				}
+
+				timeCost := time.Since(timeNow)
+
+				if answer.Type == ipv6.ICMPTypeTimeExceeded {
+					fmt.Printf("%d   %40s   %40s\n", i, node.String(), timeCost)
+
+					hop = map[string]interface{}{
+						"rtt":   fmt.Sprintf("%s", timeCost),
+						"saddr": node.String(),
+					}
+
+					hopData.Details = append(hopData.Details, hop)
+				} else if answer.Type == ipv6.ICMPTypeEchoReply {
+					fmt.Printf("%d   %40s   %40s\n", i, node.String(), timeCost)
+					hop = map[string]interface{}{
+						"rtt":   fmt.Sprintf("%s", timeCost),
+						"saddr": node.String(),
+					}
+					hopData.Details = append(hopData.Details, hop)
+
+					isDest = true
+					break
+				} else {
+					fmt.Printf("%d %40s\n", i, "*")
+				}
+
+			}
+
+		}
+
+		t.hops = append(t.hops, hopData)
+
+		if isDest {
+			return
+		}
+	}
+}