[发明专利]抓取网卡数据包的方法、装置及电子设备

说明书

本发明提供了一种抓取网卡数据包的方法、装置及电子设备，涉及计算机技术领域，该方法包括：设置网卡的网卡队列按照指定顺序接收数据包；其中，网卡的网卡队列为至少一个；查找网卡队列中的目标网卡队列；获取目标网卡队列接收的数据包，并通过执行预设函数对获取到的数据包进行内存拷贝，将内存拷贝后的数据包作为抓取到的网卡数据包；其中，预设函数是用于修改数据包引用计数的函数。本发明有效提升了抓取网卡数据包的效率。

技术领域

本发明涉及计算机技术领域，尤其是涉及一种抓取网卡数据包的方法、装置及电子设备。

背景技术

随着互联网的发展，在网络中的数据包流量越来越大，提升抓取网卡数据包的抓取效率至关重要。目前，在抓取网卡数据包时，通常先将数据包进行一次内存拷贝，将拷贝后的结果写入pcap文件中，并将内存拷贝后的网卡数据包作为抓取到的网卡数据包。但是通常数据包的内存都较大，对数据包进行内存拷贝耗时较长，导致抓取网卡数据包效率较低。针对上述抓取网卡数据包在实际应用中的问题，目前尚未提出有效的解决方案。

发明内容

本发明的目的在于提供一种抓取网卡数据包的方法、装置及电子设备，以缓解了现有技术中在抓取网卡数据包时由于内存拷贝耗时较长导致抓取数据包效率较低的技术问题，有效提升了抓取网卡数据包的效率。

第一方面，本发明实施例提供了一种抓取网卡数据包的方法，包括：设置网卡的网卡队列按照指定顺序接收数据包；其中，网卡的网卡队列为至少一个；查找网卡队列中的目标网卡队列；获取目标网卡队列接收的数据包，并通过执行预设函数对获取到的数据包进行内存拷贝，将内存拷贝后的数据包作为抓取到的网卡数据包；其中，预设函数是用于修改数据包引用计数的函数。

进一步，上述通过执行预设函数对获取到的数据包进行内存拷贝的步骤，包括：获取数据包的第一引用计数，并通过执行预设函数对第一引用计数进行加值运算，得到第二引用计数；根据第二引用计数对数据包进行内存拷贝。

进一步，上述设置网卡的网卡队列按照指定顺序接收数据包的步骤，包括：获取网卡的网卡队列的队列配置信息；队列配置信息包括队列个数及队列长度；根据获取的队列配置信息，设置网卡的网卡队列按照指定顺序接收数据包。

进一步，上述查找网卡队列中的目标网卡队列的步骤，包括：查找网卡队列中接收到数据包的网卡队列；将接收到数据包的网卡队列确定为目标网卡队列。

进一步，上述方法还包括：将抓取到的网卡数据包写入指定文件。

第二方面，本发明实施例提供了一种抓取网卡数据包的装置，包括：网卡设置模块，用于设置网卡的网卡队列按照指定顺序接收数据包；其中，网卡的网卡队列为至少一个；队列查找模块，用于查找网卡队列中的目标网卡队列；抓包模块，用于获取目标网卡队列接收的数据包，并通过执行预设函数对获取到的数据包进行内存拷贝，将内存拷贝后的数据包作为抓取到的网卡数据包；其中，预设函数是用于修改数据包引用计数的函数。

进一步，上述抓包模块用于：获取数据包的第一引用计数，并通过执行预设函数对第一引用计数进行加值得到第二引用计数；根据第二引用计数对数据包进行内存拷贝。

第三方面，本发明实施例提供一种电子设备，包括处理器和存储器，存储器存储有能够被处理器执行的计算机可执行指令，处理器执行计算机可执行指令以实现前述实施方式任一项的抓取网卡数据包的方法的步骤。

第四方面，本发明实施例提供一种计算机可读存储介质，计算机可读存储介质上存储有计算机程序，计算机程序被处理器运行时执行前述实施方式任一项的抓取网卡数据包的方法的步骤。