Go语言 无阻塞以太网捕获

kxxlusnw  于 5个月前  发布在  Go
关注(0)|答案(1)|浏览(85)

我用这个代码来捕获Ehternet数据包:

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()阻塞:如果没有收到数据包,就会出现无限循环。如何设置超时?

smtd7mpg

smtd7mpg1#

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

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

字符串
如果多次调用,则返回相同的通道。
因此,由于gopacket.PacketSource.Packets()返回一个通道,因此可以在两个通道上使用select的常用策略其中一个通道上的操作由超时控制。
在最简单的情况下,类似这样的东西可以做到:

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

相关问题