nits

netsampler · Aug 12, 2024 · 08c553c · 08c553c
1 parent 4d6e0fd
commit 08c553c
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 7 deletions.
diff --git a/pb/flow.pb.go b/pb/flow.pb.go
diff --git a/pb/flow.proto b/pb/flow.proto
@@ -109,6 +109,8 @@ message FlowMessage {
     GRE = 9;
     IPv6HeaderRouting = 10;
     IPv6HeaderFragment = 11;
+    Geneve = 12;
+    Teredo = 13;
     Custom = 99;
     // todo: add nsh
   }

diff --git a/producer/proto/producer_packet.go b/producer/proto/producer_packet.go
@@ -743,6 +743,8 @@ func ParseGRE(flowMessage *ProtoProducerMessage, data []byte, pc ParseConfig) (r
 }
 
 func ParseTeredoDst(flowMessage *ProtoProducerMessage, data []byte, pc ParseConfig) (res ParseResult, err error) {
+	flowMessage.AddLayer("Teredo")
+
 	// get next parser
 	res.NextParser = parserIPv6
 
@@ -756,6 +758,8 @@ func ParseGeneve(flowMessage *ProtoProducerMessage, data []byte, pc ParseConfig)
 
 	res.Size = int(data[0]&0x3f)*4 + 8
 
+	flowMessage.AddLayer("Geneve")
+
 	eType := data[2:4]
 
 	// get next parser