feature: capture MetadataRequest and ProduceRequest

core/dispatch.go

@@ -2,25 +2,26 @@ package core
 
 import (
 	"fmt"
-	"github.com/google/gopacket/pcap"
 	"log"
+	"time"
+
 	"github.com/google/gopacket"
 	"github.com/google/gopacket/layers"
+	"github.com/google/gopacket/pcap"
 	"github.com/google/gopacket/tcpassembly"
 	"github.com/google/gopacket/tcpassembly/tcpreader"
-	"time"
 )
 
 type Dispatch struct {
-	device string
+	device  string
 	payload []byte
-	Plug *Plug
+	Plug    *Plug
 }
 
 func NewDispatch(plug *Plug, cmd *Cmd) *Dispatch {
-	return &Dispatch {
-		Plug: plug,
-		device:cmd.Device,
+	return &Dispatch{
+		Plug:   plug,
+		device: cmd.Device,
 	}
 }
 
@@ -40,16 +41,16 @@ func (d *Dispatch) Capture() {
 	}
 
 	// Capture
-	src     := gopacket.NewPacketSource(handle, handle.LinkType())
-	packets := src.Packets()
+	src := gopacket.NewPacketSource(handle, handle.LinkType())
+	packets := src.Packets() // get packet chan
 
 	// Set up assembly
 	streamFactory := &ProtocolStreamFactory{
-		dispatch:d,
+		dispatch: d,
 	}
 	streamPool := NewStreamPool(streamFactory)
-	assembler  := NewAssembler(streamPool)
-	ticker     := time.Tick(time.Minute)
+	assembler := NewAssembler(streamPool)
+	ticker := time.Tick(time.Minute)
 
 	// Loop until ctrl+z
 	for {
@@ -84,7 +85,7 @@ type ProtocolStream struct {
 func (m *ProtocolStreamFactory) New(net, transport gopacket.Flow) tcpassembly.Stream {
 
 	//init stream struct
-	stm := &ProtocolStream {
+	stm := &ProtocolStream{
 		net:       net,
 		transport: transport,
 		r:         tcpreader.NewReaderStream(),
@@ -97,4 +98,4 @@ func (m *ProtocolStreamFactory) New(net, transport gopacket.Flow) tcpassembly.St
 	go m.dispatch.Plug.ResolveStream(net, transport, &(stm.r))
 
 	return &(stm.r)
-}
+}