add windows support

capture/pacp_macos.go

@@ -1,15 +1,20 @@
+//go:build darwin
 // +build darwin
 
 package capture
 
 import (
 	"fmt"
+	"time"
 
+	log "github.com/golang/glog"
+	"github.com/google/gopacket"
+	"github.com/google/gopacket/layers"
 	"github.com/google/gopacket/pcap"
 )
 
 // in online use, we found a strange bug: pcap cost 100% core CPU and memory increase along
-func initEthernetHandlerFromPacp() (handler PcapHandler) {
+func initEthernetHandlerFromPacp() (pcapHandler *pcap.Handle) {
 	pcapHandler, err := pcap.OpenLive(DeviceName, 65536, false, pcap.BlockForever)
 	if err != nil {
 		panic(fmt.Sprintf("cannot open network interface %s <-- %s", DeviceName, err.Error()))
@@ -20,6 +25,58 @@ func initEthernetHandlerFromPacp() (handler PcapHandler) {
 		panic(err.Error())
 	}
 
-	handler = pcapHandler
 	return
 }
+
+func dealEachTCPIPPacket(dealTCPIPPacket func(tcpIPPkt *TCPIPPair)) {
+	handler := initEthernetHandlerFromPacp()
+	defer func() {
+		handler.Close()
+	}()
+
+	for {
+		var ci gopacket.CaptureInfo
+		data, ci, err := handler.ZeroCopyReadPacketData()
+		if err != nil {
+			log.Error(err.Error())
+			time.Sleep(time.Second * 3)
+			continue
+		}
+
+		packet := gopacket.NewPacket(data, layers.LayerTypeEthernet, gopacket.NoCopy)
+		m := packet.Metadata()
+		m.CaptureInfo = ci
+
+		tcpPkt, ok := packet.TransportLayer().(*layers.TCP)
+		if !ok {
+			continue
+		}
+
+		ipLayer := packet.NetworkLayer()
+		if ipLayer == nil {
+			log.Error("no ip layer found in package")
+			continue
+		}
+
+		var srcIP, dstIP string
+		switch realIPLayer := ipLayer.(type) {
+		case *layers.IPv6:
+			{
+				srcIP = realIPLayer.SrcIP.String()
+				dstIP = realIPLayer.DstIP.String()
+			}
+		case *layers.IPv4:
+			{
+				srcIP = realIPLayer.SrcIP.String()
+				dstIP = realIPLayer.DstIP.String()
+			}
+		}
+
+		tcpipPair := &TCPIPPair{
+			srcIP:  srcIP,
+			dstIP:  dstIP,
+			tcpPkt: tcpPkt,
+		}
+		dealTCPIPPacket(tcpipPair)
+	}
+}