renun/NTrace-core
1
0
Fork 0
mirror of https://github.com/nxtrace/NTrace-core.git synced 2025-08-12 06:26:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

重构mpls icmpv4 代码

Browse Source
This commit is contained in:
tsosunchia
2023-10-05 06:47:28 +08:00
parent db1e9e2b0c
commit ba7d6ea87b
1 changed files with 40 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
trace/icmp_ipv4.go
View File

@@ -175,43 +175,8 @@ func (t *ICMPTracer) listenICMP() {
func (t *ICMPTracer) handleICMPMessage(msg ReceivedMessage, icmpType int8, data []byte, ttl int) {
t.inflightRequestRWLock.RLock()
defer t.inflightRequestRWLock.RUnlock()
label, err := int64(0), error(nil)
tc, s, ttlMpls := int64(0), "", int64(0)
mpls := ""
extensionOffset := 20 + 8 + 52
if len(data) > extensionOffset {
extensionBody := data[extensionOffset:]
if len(extensionBody) >= 8 && len(extensionBody)%8 == 0 {
//log.Println("ICMP Multi-Part Extensions detected for TTL:", ttl)
//这里感觉会有问题，能力所限先这样吧，等有缘人来改
tmp := fmt.Sprintf("%x", msg.Msg[:*msg.N])
substring := tmp[len(tmp)-8 : len(tmp)-3]
label, err = strconv.ParseInt(substring, 16, 32)
if err != nil {
return
}
strSlice := []byte(tmp[len(tmp)-3 : len(tmp)-2])
charValue := int(strSlice[0] - '0')
binaryStr := fmt.Sprintf("%04b", charValue)
tc, err = strconv.ParseInt(binaryStr[:3], 2, 32)
if err != nil {
return
}
s = binaryStr[3:]
substring = tmp[len(tmp)-2:]
ttlMpls, err = strconv.ParseInt(substring, 16, 32)
if err != nil {
return
}
//fmt.Printf("MPLS for TTL %d: Lbl %d, TC %d, S %s, TTL %d\n", ttl, label, tc, s, ttlMpls)
mpls = fmt.Sprintf("Lbl %d, TC %d, S %s, TTL %d", label, tc, s, ttlMpls)
}
}
mpls := t.extractMPLS(msg, data)
if _, ok := t.inflightRequest[ttl]; ok {
t.inflightRequest[ttl] <- Hop{
Success: true,
@@ -221,6 +186,45 @@ func (t *ICMPTracer) handleICMPMessage(msg ReceivedMessage, icmpType int8, data
}
}
func (t *ICMPTracer) extractMPLS(msg ReceivedMessage, data []byte) string {
extensionOffset := 20 + 8 + 52
if len(data) <= extensionOffset {
return ""
}
extensionBody := data[extensionOffset:]
if len(extensionBody) < 8 || len(extensionBody)%8 != 0 {
return ""
}
tmp := fmt.Sprintf("%x", msg.Msg[:*msg.N])
if len(tmp) < 8 {
return ""
}
label, err := strconv.ParseInt(tmp[len(tmp)-8:len(tmp)-3], 16, 32)
if err != nil {
return ""
}
strSlice := []byte(tmp[len(tmp)-3 : len(tmp)-2])
charValue := int(strSlice[0] - '0')
binaryStr := fmt.Sprintf("%04b", charValue)
tc, err := strconv.ParseInt(binaryStr[:3], 2, 32)
if err != nil {
return ""
}
s := binaryStr[3:]
ttlMpls, err := strconv.ParseInt(tmp[len(tmp)-2:], 16, 32)
if err != nil {
return ""
}
return fmt.Sprintf("Lbl %d, TC %d, S %s, TTL %d", label, tc, s, ttlMpls)
}
func gernerateID(ttl_int int) int {
const ID_FIXED_HEADER = "10"
var processID = fmt.Sprintf("%07b", os.Getpid()&0x7f) //取进程ID的前7位