fix: use ip.To4()

Charlie17Li · Charlie17Li · commit 00ae87d03da0 · 2024-09-09T22:04:41.000+08:00
diff --git a/nl/tc_linux.go b/nl/tc_linux.go
@@ -1460,15 +1460,15 @@ func (p *TcPedit) SetIPv6Src(ip6 net.IP) {
 
 func (p *TcPedit) SetDstIP(ip net.IP) {
 	if ip.To4() != nil {
-		p.SetIPv4Dst(ip)
+		p.SetIPv4Dst(ip.To4())
 	} else {
 		p.SetIPv6Dst(ip)
 	}
 }
 
 func (p *TcPedit) SetSrcIP(ip net.IP) {
 	if ip.To4() != nil {
-		p.SetIPv4Src(ip)
+		p.SetIPv4Src(ip.To4())
 	} else {
 		p.SetIPv6Src(ip)
 	}
@@ -1533,7 +1533,7 @@ func (p *TcPedit) SetIPv6Dst(ip6 net.IP) {
 }
 
 func (p *TcPedit) SetIPv4Src(ip net.IP) {
-	u32 := NativeEndian().Uint32(ip[12:16])
+	u32 := NativeEndian().Uint32(ip[:4])
 
 	tKey := TcPeditKey{}
 	tKeyEx := TcPeditKeyEx{}
@@ -1549,7 +1549,7 @@ func (p *TcPedit) SetIPv4Src(ip net.IP) {
 }
 
 func (p *TcPedit) SetIPv4Dst(ip net.IP) {
-	u32 := NativeEndian().Uint32(ip[12:16])
+	u32 := NativeEndian().Uint32(ip[:4])
 
 	tKey := TcPeditKey{}
 	tKeyEx := TcPeditKeyEx{}

Original file line number	Diff line number	Diff line change
`@@ -1460,15 +1460,15 @@ func (p *TcPedit) SetIPv6Src(ip6 net.IP) {`
`1460`	`1460`
`1461`	`1461`	`func (p *TcPedit) SetDstIP(ip net.IP) {`
`1462`	`1462`	`if ip.To4() != nil {`
`1463`		`- p.SetIPv4Dst(ip)`
	`1463`	`+ p.SetIPv4Dst(ip.To4())`
`1464`	`1464`	`} else {`
`1465`	`1465`	`p.SetIPv6Dst(ip)`
`1466`	`1466`	`}`
`1467`	`1467`	`}`
`1468`	`1468`
`1469`	`1469`	`func (p *TcPedit) SetSrcIP(ip net.IP) {`
`1470`	`1470`	`if ip.To4() != nil {`
`1471`		`- p.SetIPv4Src(ip)`
	`1471`	`+ p.SetIPv4Src(ip.To4())`
`1472`	`1472`	`} else {`
`1473`	`1473`	`p.SetIPv6Src(ip)`
`1474`	`1474`	`}`
`@@ -1533,7 +1533,7 @@ func (p *TcPedit) SetIPv6Dst(ip6 net.IP) {`
`1533`	`1533`	`}`
`1534`	`1534`
`1535`	`1535`	`func (p *TcPedit) SetIPv4Src(ip net.IP) {`
`1536`		`- u32 := NativeEndian().Uint32(ip[12:16])`
	`1536`	`+ u32 := NativeEndian().Uint32(ip[:4])`
`1537`	`1537`
`1538`	`1538`	`tKey := TcPeditKey{}`
`1539`	`1539`	`tKeyEx := TcPeditKeyEx{}`
`@@ -1549,7 +1549,7 @@ func (p *TcPedit) SetIPv4Src(ip net.IP) {`
`1549`	`1549`	`}`
`1550`	`1550`
`1551`	`1551`	`func (p *TcPedit) SetIPv4Dst(ip net.IP) {`
`1552`		`- u32 := NativeEndian().Uint32(ip[12:16])`
	`1552`	`+ u32 := NativeEndian().Uint32(ip[:4])`
`1553`	`1553`
`1554`	`1554`	`tKey := TcPeditKey{}`
`1555`	`1555`	`tKeyEx := TcPeditKeyEx{}`