go - 无阻塞以太网捕获

Question

我使用此代码来捕获以太网数据包：

var snapshotLen int32 = 102400
var promiscuous bool = true
var timeout time.Duration = 1 * time.Second
var handle *pcap.Handle
handle, err = pcap.OpenLive(device, snapshotLen, promiscuous, timeout)
err = handle.SetBPFFilter(filter)

packetSource := gopacket.NewPacketSource(handle, handle.LinkType())
for packet := range packetSource.Packets() {

}

问题是 packetSource.Packets() 被阻塞：如果没有收到数据包，则存在无限循环。如何设置超时？

Answer 1

从文档：

func (p *PacketSource) Packets() chan Packet

Packets 返回一个数据包通道，允许轻松迭代数据包。数据包将从底层的 PacketDataSource 异步读取并写入返回的通道。如果底层证券PacketDataSource返回io.EOF错误，通道将被关闭。如果遇到任何其他错误，则将其忽略。

for packet := range packetSource.Packets() {   handlePacket(packet) 
// Do something with each packet. }

如果多次调用，则返回相同的通道。

因此，由于gopacket.PacketSource.Packets()返回一个通道，您可以select在两个通道上使用通常的 ing 策略——其中一个通道上的操作由超时控制。

在最简单的情况下，这样的事情会做：

timer := time.NewTicker(2 * time.Second)
defer timer.Stop()

src := gopacket.NewPacketSource(handle, handle.LinkType())
for {
    select {
        case pkt := <- src.Packets():
            // process a packet in pkt
        case <- timer.C:
            // timed out waiting for 2 seconds
    }
}

根据具体的用例，第二个通道可以来自另一个来源——例如上下文。