Linux下非轮询方式监控网卡连接状态
在Linux下监控网卡的连接状态有多种方式，我想要的方式，不是以轮询方式定时查询或主动获取某个值，而是在网卡连接状态变化时我的程序能收到通知。
通过搜索找到stackoverflow上的一帖子（地址见后面的参考）得到了线索，做法是：
1. 使用AF_NETLINK socket
2. 绑定到RTMGRP_LINK组
3. 等待接收RTM_NEWLINK和RTM_DELLINK类型的message
4. 解析收到的消息中ifinfomsg结构体的ifi_flags成员是否被设置了IFF_RUNNING
太多文字段落看着容易吃力啊，就让文字尽可能地作为代码注释吧，理论与实践相结合啊，
代码和那些man命令的运行环境为debian wheezy 7.7。
参考：
http://stackoverflow.com/questions/10340145/get-event-for-nic-ethernet-card-link-status-on-linux
man 3 netlink
man 7 netlink
man 3 rtnetlink
man 7 rtnetlink
man 7 netdevice
http://centaur.sch.bme.hu/~leait/projects/openwrt/
http://fossies.org/linux/misc/open-fcoe-3.19.tar.gz/open-fcoe-3.19/fcoe-utils/lib/rtnetlink.c
通过搜索找到stackoverflow上的一帖子（地址见后面的参考）得到了线索，做法是：
1. 使用AF_NETLINK socket
2. 绑定到RTMGRP_LINK组
3. 等待接收RTM_NEWLINK和RTM_DELLINK类型的message
4. 解析收到的消息中ifinfomsg结构体的ifi_flags成员是否被设置了IFF_RUNNING
太多文字段落看着容易吃力啊，就让文字尽可能地作为代码注释吧，理论与实践相结合啊，
端上代码：
#include
#include
#include
#include
#include
#include
#include /* 根据`man 7 rtnetlink`，包含此头文件。（此头文件不包含也能编译通过，但为了兼容性，或者说平台移植性，包含上吧，万一去掉后放到非Linux系统下编不过了呢） */
#include
#include /* `man 7 rtnetlink`中有"ifi_flags contains the device flags, see netdevice(7)"这句话，`man 7 netdevice`中提到包含此头文件。 */
//#include /* use the following header instead of this one */
#include /* this one is suite for the Linux, more related to interact with Linux kernel's network interfaces */
#include /* 包含ARPHRD_LOOPBACK的定义 */
#include /* 根据`man 7 rtnetlink`，包含此头文件 */
#include /* 根据`man 7 rtnetlink`，包含此头文件 */
#define PRINTF(fmt, ...) printf("%s:%d: " fmt, __FUNCTION__, __LINE__, ## __VA_ARGS__)
#define BUF_SIZE 4096 /* 看到`man 7 netlink`中reading netlink message那个EXAMPLE的buf的大小是4096，我也就写这么大了。 */
static int init(struct sockaddr_nl *psa);
static int deinit();
static int msg_req();
static int msg_loop(struct msghdr *pmh);
static void sig_handler(int sig);
static int sfd = -1;
int main (int argc, char *argv[])
{
int ret = 0;
/* 下面4行变量声明参考`man 7 netlink`中reading netlink message那个EXAMPLE，和`man 2 recvmsg`中"The recvmsg() call uses a msghdr structure to minimize the number of directly supplied arguments"这行之后的结构体说明。 */
char buf[BUF_SIZE];
struct iovec iov = {buf, sizeof(buf)};
struct sockaddr_nl sa;
struct msghdr msg = {(void *)&sa, sizeof(sa), &iov, 1, NULL, 0, 0};
/* 程序的主要流程体现在下面4个函数调用：
* 1. 初始化socket
* 2. 请求获取当前link状态
* 3. 循环等待接收link状态的消息
* 4. 反初始化
*/
ret = init(&sa);
if (!ret) {
ret = msg_req();
}
if (!ret) {
ret = msg_loop(&msg);
}
ret = deinit();
return ret;
}
static int init(struct sockaddr_nl *psa)
{
int ret = 0;
struct sigaction sigact;
sigact.sa_handler = sig_handler;
if (!ret && -1 == sigemptyset(&sigact.sa_mask)) {
PRINTF("ERROR! sigemptyset\n");
ret = -1;
}
if (!ret && -1 == sigaction(SIGINT, &sigact, NULL)) { /* SIGINT用于保证按Ctrl+C时进程能够在退出前执行反初始化（资源释放） */
PRINTF("ERROR! sigaction SIGINT\n");
ret = -1;
}
if (!ret && -1 == sigaction(SIGTERM, &sigact, NULL)) { /* SIGINT用于保证按Ctrl+C时进程能够在退出前执行反初始化（资源释放） */
PRINTF("ERROR! sigaction SIGTERM\n");
ret = -1;
}
if (!ret) {
/* `man 7 netlink`中有"the netlink protocol does not distinguish between datagram and raw sockets"这样一句话，所以socket函数的第2个参数填SOCK_RAW和SOCK_DGRAM均可。 */
/* 根据`man 7 netlink`中netlink_family参数的解释，NETLINK_ROUTE作为netlink_family参数的值能够获取link状态信息。 */
sfd = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_ROUTE);
if (-1 == sfd) {
PRINTF("ERROR! socket: %s\n", strerror(errno));
ret = -1;
}
}
/* bind代码参考`man 7 netlink`中EXAMPLE部分的第1段代码 */
memset(psa, 0, sizeof(*psa));
psa->nl_family = AF_NETLINK;
psa->nl_groups = RTMGRP_LINK; /* 绑定到RTMGRP_LINK组 */
if (!ret && bind(sfd, (struct sockaddr *)psa, sizeof(*psa))) {
PRINTF("ERROR! bind: %s\n", strerror(errno));
ret = -1;
}
/* deinit if init failed */
if (0 != ret) {
deinit();
}
return ret;
}
static int deinit()
{
int ret = 0;
if (-1 != sfd) {
if (-1 == close(sfd)) {
PRINTF("ERROR! close: %s\n", strerror(errno));
ret = -1;
}
sfd = -1;
}
return ret;
}
static int msg_req()
{
int ret = 0;
struct {
struct nlmsghdr nh;
struct ifinfomsg ifimsg;
} req;
/* 发送消息来请求获取当前link状态，这是为了当程序启动后能够显示一下当前的网卡连接状态，否则程序运行时网卡连接状态没有变化的话，程序就一直没有任何输出 */
/* 代码参考`man 3 rtnetlink`中的EXAMPLE，主要参考了http://fossies.org/linux/misc/open-fcoe-3.19.tar.gz/open-fcoe-3.19/fcoe-utils/lib/rtnetlink.c中的send_getlink_dump函数 */
memset(&req, 0, sizeof(req));
req.nh.nlmsg_len = NLMSG_LENGTH(sizeof(struct ifinfomsg)); /* NLMSG_LENGTH会返回包含nlmsghdr这个header和ifinfomsg这个payload加起来的size，返回的是对齐后的值 */
req.nh.nlmsg_flags = NLM_F_REQUEST | NLM_F_DUMP; /* 根据`man 7 netlink`中NLM_F_REQUEST的解释，所有请求类型的消息都要设置NLM_F_REQUEST；设置了NLM_F_DUMP就能够获取link状态，根据`man 7 netlink`，NLM_F_DUMP相当于NLM_F_ROOT|NLM_F_MATCH，`man 7 netlink`中说NLM_F_MATCH是"Not implemented yet"，但去掉NLM_F_DUMP，只设置NLM_F_MATCH也是有效果的。 */
req.nh.nlmsg_type = RTM_GETLINK; /* `man 7 rtnetlink`中说RTM_GETLINK用于"get information about a specific network interface"，`man 7 rtnetlink`中的"These messages contain an ifinfomsg structure followed by a series of rtattr structures"这句话解释了RTM_GETLINK的消息结构 */
req.ifimsg.ifi_family = AF_UNSPEC; /* 根据`man 7 rtnetlink`中ifinfomsg结构体的定义，恒为AF_UNSPEC */
req.ifimsg.ifi_index = 0; /* http://man7.org/linux/man-pages/man7/rtnetlink.7.html中有"ifi_index is the unique interface index (since Linux 3.7, it is possible to feed a nonzero value with the RTM_NEWLINK message, thus creating a link with the given ifindex)"这句话，说明这个成员设置0没问题 */
req.ifimsg.ifi_change = 0xFFFFFFFF; /* `man 7 rtnetlink`中有"ifi_change is reserved for future use and should be always set to 0xFFFFFFFF"这句话，所以设置为0xFFFFFFFF */
if (-1 == send(sfd, &req, req.nh.nlmsg_len, 0)) {
PRINTF("ERROR! send: %s\n", strerror(errno));
ret = -1;
}
return ret;
}
static int msg_loop(struct msghdr *pmh)
{
int ret = 0;
ssize_t nread = -1;
char *buf = (char *)(pmh->msg_iov->iov_base);
struct nlmsghdr *nh;
struct ifinfomsg *ifimsg;
struct rtattr *rta;
int attrlen;
/* 循环体参考`man 7 netlink`中reading netlink message那个EXAMPLE，连注释都照搬过来了 :D */
while (!ret) {
nread = recvmsg(sfd, pmh, 0);
if (-1 == nread) {
PRINTF("ERROR! recvmsg: %s\n", strerror(errno));
ret = -1;
}
for (nh = (struct nlmsghdr *)buf; /* `man 7 netlink`中"Netlink messages consist of a byte stream with one or multiple nlmsghdr headers and associated payload"这句话的周围解释了netlink消息的结构，所以消息的开头是nlmsghdr类型的header */
!ret && NLMSG_OK(nh, nread); /* 根据`man 3 netlink`，NLMSG_OK用来检查收到的netlink消息是否OK */
nh = NLMSG_NEXT(nh, nread)) { /* 根据`man 3 netlink`，NLMSG_NEXT用于定位到下一个nlmsghdr header的开头，注意这个宏会改变nread的值 */
/* `man 3 netlink`中有"The caller must check if the current nlmsghdr didn't have the NLMSG_DONE set—this function doesn't return NULL on end"这句话，所以要检查NLMSG_DONE */
/* The end of multipart message. */
if (NLMSG_DONE == nh->nlmsg_type) {
break;
}
if (NLMSG_ERROR == nh->nlmsg_type) {
/* Do some error handling. */
PRINTF("ERROR! NLMSG_ERROR\n");
ret = -1;
}
/* Continue with parsing payload. */
if (!ret && (RTM_NEWLINK == nh->nlmsg_type || RTM_DELLINK == nh->nlmsg_type)) { /* `man 7 rtnetlink`中的"These messages contain an ifinfomsg structure followed by a series of rtattr structures"这句话解释了RTM_NEWLINK和RTM_DELLINK的消息结构，前一句说明了RTM_NEWLINK和RTM_DELLINK分别用于"Create"和"remove" "information about a specific network interface"，link的"Create"和"remove"动作产生的消息是程序要解析的。 */
ifimsg = (struct ifinfomsg *)NLMSG_DATA(nh); /* 根据`man 3 netlink`，NLMSG_DATA用于定位nlmsghdr header后面的payload的内存地址 */
if (ARPHRD_LOOPBACK != ifimsg->ifi_type) { /* 排除loopback网卡适配器 */
/* `man 7 rtnetlink`中的"These messages contain an ifinfomsg structure followed by a series of rtattr structures"这句话说明ifinfomsg之后的内容就是属性 */
attrlen = nh->nlmsg_len - NLMSG_LENGTH(sizeof(struct ifinfomsg)); /* NLMSG_LENGTH返回的值是nlmsghdr和ifinfomsg加起来的长度，减去后，剩下的就是属性的长度 */
for (rta = IFLA_RTA(ifimsg); /* IFLA_RTA用于定位ifinfomsg payload后面的属性的内存地址 */
RTA_OK(rta, attrlen) && rta->rta_type <= IFLA_MAX; /* 根据`man 3 rtnetlink`，RTA_OK用来检查rta指向的属性是否有效 */
rta = RTA_NEXT(rta, attrlen)) { /* 根据`man 3 rtnetlink`，RTA_NEXT用于定位到下一个属性的开头，注意这个宏会改变attrlen的值 */
if (IFLA_IFNAME == rta->rta_type) { /* 当rta_type为IFLA_IFNAME时，属性包含的数据是网卡名字符串 */
printf("%s: ", (char*)RTA_DATA(rta)); /* 打印网卡名，如eth0 */
}
}
/* 实际上，当link up时，IFF_RUNNING和IFF_LOWER_UP都会被设置，link down时都会被清除，以IFF_LOWER_UP作为判断依据也行，但根据`man 7 netdevice`，IFF_LOWER_UP是从内核2.6.17版本才有的，所以使用IFF_RUNNING兼容性可能好点 */
if (IFF_RUNNING & ifimsg->ifi_flags)
printf("link up\n");
else
printf("link down\n");
}
}
}
}
return ret;
}
static void sig_handler(int sig)
{
exit(deinit());
}
代码和那些man命令的运行环境为debian wheezy 7.7。
参考：
http://stackoverflow.com/questions/10340145/get-event-for-nic-ethernet-card-link-status-on-linux
man 3 netlink
man 7 netlink
man 3 rtnetlink
man 7 rtnetlink
man 7 netdevice
http://centaur.sch.bme.hu/~leait/projects/openwrt/
http://fossies.org/linux/misc/open-fcoe-3.19.tar.gz/open-fcoe-3.19/fcoe-utils/lib/rtnetlink.c