Linux环境下纯Python无第三方库读写Netlink

最新推荐文章于 2024-06-07 09:49:13 发布

沙老师

最新推荐文章于 2024-06-07 09:49:13 发布

阅读量2.9k

点赞数 1

本文链接：https://blog.csdn.net/shajunxing/article/details/79755996

版权

原文地址：http://shajunxing.tpddns.cn:8888/web/blog/2018-03-28-Linux环境下纯Python无第三方库读写Netlink/index.html

基础知识

Netlink使用套接字的方式提供了应用程序和内核之间信息交互的机制，用来替代传统的ioctl等机制。当前，Netlink主要涵盖了Linux网络相关的功能，注意，不要把Netlink和Linux网络本身混为一谈，它只是一种进程间通信（IPC）而已。例如，早期的net-tools工具集（ifconfig、netstat...）使用的是ioctl，现在取而代之的iproute2工具集（ip、ss...）则使用Netlink。

Netlink套接字的创建、发送和接收

socket函数用于创建套接字，同样也用与创建Netlink套接字。第一个参数af填写AF_INET，和普通的套接字一样；因为需要自己构造数据包的首部，所以第二个参数type填写SOCK_RAW；第三个参数protocol填写Netlink专有的协议类型，比如NETLINK_ROUTE用于读写网络相关信息，当然，这些在Python3的socket模块中均有定义：

s = socket.socket(socket.AF_NETLINK, socket.SOCK_RAW, socket.NETLINK_ROUTE)

数据收发必须指定自己和对端的地址，Netlink的地址结构为pid, groups二元组，第一个参数相当于普通套接字的端口号，用来唯一标志一条“连接”，当然这里使用进程ID作为唯一标志是很方便的，第二个参数略，填0即可：

s.bind((os.getpid(), 0))

套接字的收发缓存要设置得尽量大一些，比如：

s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 65536)
s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 65536)

然后就可以使用sendto()和recvfrom()函数收发数据了，因为Linux内核进程的PID是0，所以对端地址为0, 0。注意，即便接收缓存设置得足够大，内核也很有可能会分批次发回数据，所以不能指望一次recvfrom()就接收到所有的数据，以下代码演示了获取所有网络接口信息，使用无限循环接收，当然了，在接收完毕后recvfrom()函数会阻塞，要手动结束程序，不过可以看到内核返回的结果了。

import os
import socket

s = socket.socket(socket.AF_NETLINK, socket.SOCK_RAW, socket.NETLINK_ROUTE)
s.bind((os.getpid(), 0))
s.setsockopt(socket.SOL_SOCKET, socket.SO_SNDBUF, 65536)
s.setsockopt(socket.SOL_SOCKET, socket.SO_RCVBUF, 65536)
s.sendto(b'\x14\x00\x00\x00\x12\x00\x01\x03\x00\x00\x00\x00\xd5\x1b\x00\x00\x11\x00\x00\x00', (0, 0))
while True:
    print(s.recvfrom(65536))

细心的读者应该会在接收到的二进制数据中查找到网络接口的字符串名，比如“lo”、“eth0”等。