无阻塞以太网捕获

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

来自the docs：

func (p *PacketSource) Packets() chan Packet

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

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

如果多次调用，返回同一个频道。

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

在最简单的情况下，可以这样做：

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
    }
}

根据具体的用例，第二个渠道可以来自其他来源，例如context。