Set transport protocol number during parsing

PiperOrigin-RevId: 334535896

1 files changed, 2 insertions, 0 deletions

diff --git a/pkg/tcpip/header/parse/parse.go b/pkg/tcpip/header/parse/parse.go
index 522135557..5ca75c834 100644
--- a/pkg/tcpip/header/parse/parse.go
+++ b/pkg/tcpip/header/parse/parse.go
@@ -139,6 +139,7 @@ traverseExtensions:
 // Returns true if the header was successfully parsed.
 func UDP(pkt *stack.PacketBuffer) bool {
 	_, ok := pkt.TransportHeader().Consume(header.UDPMinimumSize)
+	pkt.TransportProtocolNumber = header.UDPProtocolNumber
 	return ok
 }
 
@@ -162,5 +163,6 @@ func TCP(pkt *stack.PacketBuffer) bool {
 	}
 
 	_, ok = pkt.TransportHeader().Consume(hdrLen)
+	pkt.TransportProtocolNumber = header.TCPProtocolNumber
 	return ok
 }