当前位置：首页 > backend >正文

Linux基础 -- 用户态Generic Netlink库高性能接收与回调框架

backend 2025/5/14 7:17:10

用户态Generic Netlink库高性能接收与回调框架

一、概述

在 Linux 系统中，Netlink 是用户态与内核态通信的强大机制。libnl 是一个专为简化 Netlink 编程而设计的库，提供了接收和处理 Netlink 消息的高级接口。libnl-genl 是其通用 Netlink (Generic Netlink) 扩展库，专门用于处理自定义协议。本文将详细介绍如何通过 nl_socket_modify_cb 和 nl_cb_set 两种方法实现高效的 Netlink 消息接收和处理框架。

二、Netlink 回调机制

Netlink 使用回调机制来处理接收到的消息。常用的回调设置方法有两种：

1. `nl_socket_modify_cb`

仅在指定的 Netlink 套接字上设置回调。
适合在每个套接字上使用不同回调。

2. `nl_cb_set`

在回调上下文 (struct nl_cb *) 上设置回调。
可以在多个套接字之间共享。

三、高效 Netlink 接收框架（libnl + libnl-genl + epoll）

1. epoll + 非阻塞 + 批量接收

使用 epoll 进行非阻塞监听。
通过大缓冲区批量接收消息。
结合 libnl 和 libnl-genl 实现高效的 Netlink 消息接收。

2. 完整示例：Generic Netlink 消息接收 + epoll

#include <netlink/netlink.h>
#include <netlink/msg.h>
#include <netlink/genl/genl.h>
#include <netlink/genl/ctrl.h>
#include <sys/epoll.h>
#include <stdio.h>
#include <fcntl.h>
#include <unistd.h>#define MY_GENL_FAMILY "my_custom_family"// 自定义消息处理回调
int my_genl_handler(struct nl_msg *msg, void *arg) {struct nlmsghdr *nlh = nlmsg_hdr(msg);struct genlmsghdr *gnlh = nlmsg_data(nlh);printf("[Genl Handler] Received a Netlink message.\n");printf("Command: %d, Version: %d\n", gnlh->cmd, gnlh->version);struct nlattr *attrs[GENL_MAX_ATTRS + 1] = {};nla_parse(attrs, GENL_MAX_ATTRS, genlmsg_attrdata(gnlh, 0), genlmsg_attrlen(gnlh, 0), NULL);if (attrs[GENL_ATTR_MSG]) {printf("Message: %s\n", nla_get_string(attrs[GENL_ATTR_MSG]));}return NL_OK;
}// 设置非阻塞模式
void set_nonblocking(int fd) {int flags = fcntl(fd, F_GETFL, 0);fcntl(fd, F_SETFL, flags | O_NONBLOCK);
}int main() {struct nl_sock *sock = nl_socket_alloc();nl_connect(sock, NETLINK_GENERIC);// 查找 Generic Netlink familyint family_id = genl_ctrl_resolve(sock, MY_GENL_FAMILY);if (family_id < 0) {printf("[Error] Could not resolve Generic Netlink family\n");return -1;}struct nl_cb *cb = nl_cb_alloc(NL_CB_DEFAULT);nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, my_genl_handler, NULL);nl_socket_set_cb(sock, cb);int fd = nl_socket_get_fd(sock);set_nonblocking(fd);int epfd = epoll_create1(0);struct epoll_event ev, events[10];ev.events = EPOLLIN;ev.data.fd = fd;epoll_ctl(epfd, EPOLL_CTL_ADD, fd, &ev);while (1) {int nfds = epoll_wait(epfd, events, 10, -1);for (int i = 0; i < nfds; i++) {if (events[i].data.fd == fd) {nl_recvmsgs(sock, cb); // 使用 libnl + epoll 处理消息}}}nl_cb_put(cb);nl_close(sock);nl_socket_free(sock);close(epfd);return 0;
}