flows: implement ipv

2026-05-14 09:41:07 +05:30
parent 7a3f6e945d
commit 1e5c1dea75
5 changed files with 435 additions and 69 deletions
@@ -22,8 +22,6 @@ const (

 type tcpFlow struct {
 	streamID uint32
-	srcIP    [4]byte
-	dstIP    [4]byte
 	srcPort  uint16
 	dstPort  uint16

@@ -199,8 +197,8 @@ func (m *tcpFlowManager) handle(streamID uint32, l3 L3Info, tcp TCPInfo, payload

 func (m *tcpFlowManager) createFlow(streamID uint32, l3 L3Info, tcp TCPInfo, payload []byte, srcMAC, dstMAC net.HardwareAddr) *tcpFlow {
 	id := m.sfNode.Generate()
-	ipSrc := net.IP(l3.SrcIP[:])
-	ipDst := net.IP(l3.DstIP[:])
+	ipSrc := l3.SrcIPAddr()
+	ipDst := l3.DstIPAddr()
 	if len(srcMAC) == 0 && m.macResolver != nil {
 		srcMAC = m.macResolver.Resolve(ipSrc)
 	}
@@ -244,8 +242,6 @@ func (m *tcpFlowManager) createFlow(streamID uint32, l3 L3Info, tcp TCPInfo, pay

 	flow := &tcpFlow{
 		streamID:       streamID,
-		srcIP:          l3.SrcIP,
-		dstIP:          l3.DstIP,
 		srcPort:        tcp.SrcPort,
 		dstPort:        tcp.DstPort,
 		info:           info,