Ubuntu Netlink 套接字使用介绍

news/2024/12/23 20:41:47/

Netlink 套接字 是 Linux 特有的一种 IPC(进程间通信)机制,用于用户态进程和内核模块之间的通信。它可以用来完成路由管理、设备通知、网络状态更新等任务。


1. Netlink 的基本工作原理

  • Netlink 是一种双向通信机制。
  • Netlink 消息分为请求和响应:
    • 用户态进程发送请求消息到内核。
    • 内核处理请求并返回响应消息到用户态进程。
    • 也可以由内核主动向用户态进程发送事件通知。

Netlink 的消息通常通过结构体 struct nlmsghdr 包装,消息正文是可选的数据内容。


2. Netlink 的常用 API

Netlink 通信使用的是普通的 BSD 套接字接口:

  • 创建套接字

    int socket(int domain, int type, int protocol);
    
    • domain 使用 AF_NETLINK
    • type 通常为 SOCK_RAWSOCK_DGRAM
    • protocol 指定 Netlink 子系统编号(例如 NETLINK_ROUTE)。
  • 绑定套接字

    int bind(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
    
    • 使用 struct sockaddr_nl 作为地址。
  • 发送消息

    ssize_t sendto(int sockfd, const void *buf, size_t len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen);
    
  • 接收消息

    ssize_t recv(int sockfd, void *buf, size_t len, int flags);
    

3. Netlink 示例代码

以下示例演示了如何使用 Netlink 套接字与内核进行简单的通信。

完整代码:用户态程序
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <sys/socket.h>
#include <linux/netlink.h>
#include <unistd.h>#define NETLINK_USER 31  // 自定义 Netlink 协议号(大于 31 时需内核支持)
#define MSG_LEN 1024     // 消息缓冲区大小int main() {// 创建 Netlink 套接字int sock_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_USER);if (sock_fd < 0) {std::cerr << "Error creating Netlink socket\n";return -1;}// 本地地址配置struct sockaddr_nl src_addr;memset(&src_addr, 0, sizeof(src_addr));src_addr.nl_family = AF_NETLINK;src_addr.nl_pid = getpid(); // 绑定到当前进程src_addr.nl_groups = 0;     // 不订阅多播if (bind(sock_fd, (struct sockaddr *)&src_addr, sizeof(src_addr)) < 0) {std::cerr << "Error binding Netlink socket\n";close(sock_fd);return -1;}// 目标地址配置(内核)struct sockaddr_nl dest_addr;memset(&dest_addr, 0, sizeof(dest_addr));dest_addr.nl_family = AF_NETLINK;dest_addr.nl_pid = 0;       // 发送到内核dest_addr.nl_groups = 0;    // 不订阅多播// 构造发送消息struct nlmsghdr *nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MSG_LEN));memset(nlh, 0, NLMSG_SPACE(MSG_LEN));nlh->nlmsg_len = NLMSG_SPACE(MSG_LEN); // 消息长度nlh->nlmsg_pid = getpid();             // 发送者 PIDnlh->nlmsg_flags = 0;                  // 无特殊标志位strcpy((char *)NLMSG_DATA(nlh), "Hello from user space!"); // 消息内容// 发送消息到内核if (sendto(sock_fd, nlh, nlh->nlmsg_len, 0, (struct sockaddr *)&dest_addr, sizeof(dest_addr)) < 0) {std::cerr << "Error sending message\n";free(nlh);close(sock_fd);return -1;}std::cout << "Message sent to kernel: " << (char *)NLMSG_DATA(nlh) << "\n";// 接收内核响应memset(nlh, 0, NLMSG_SPACE(MSG_LEN));if (recv(sock_fd, nlh, NLMSG_SPACE(MSG_LEN), 0) < 0) {std::cerr << "Error receiving message\n";free(nlh);close(sock_fd);return -1;}std::cout << "Message received from kernel: " << (char *)NLMSG_DATA(nlh) << "\n";// 清理资源free(nlh);close(sock_fd);return 0;
}

4. 编写内核模块以响应 Netlink 消息

内核模块代码
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/netlink.h>
#include <linux/skbuff.h>
#include <net/sock.h>#define NETLINK_USER 31struct sock *nl_sk = NULL;static void netlink_recv_msg(struct sk_buff *skb) {struct nlmsghdr *nlh;int pid;struct sk_buff *skb_out;char *msg = "Hello from kernel!";int msg_size = strlen(msg);int res;// 获取 Netlink 消息头部nlh = (struct nlmsghdr *)skb->data;printk(KERN_INFO "Kernel received message: %s\n", (char *)NLMSG_DATA(nlh));pid = nlh->nlmsg_pid; // 获取用户进程 PID// 构造响应消息skb_out = nlmsg_new(msg_size, 0);if (!skb_out) {printk(KERN_ERR "Failed to allocate new skb\n");return;}nlh = nlmsg_put(skb_out, 0, 0, NLMSG_DONE, msg_size, 0);strncpy(NLMSG_DATA(nlh), msg, msg_size);res = nlmsg_unicast(nl_sk, skb_out, pid); // 发送响应if (res < 0) {printk(KERN_INFO "Error sending message to user\n");}
}static int __init netlink_init(void) {struct netlink_kernel_cfg cfg = {.input = netlink_recv_msg, // 注册消息接收回调};nl_sk = netlink_kernel_create(&init_net, NETLINK_USER, &cfg);if (!nl_sk) {printk(KERN_ALERT "Error creating Netlink socket\n");return -10;}printk(KERN_INFO "Netlink module loaded\n");return 0;
}static void __exit netlink_exit(void) {netlink_kernel_release(nl_sk);printk(KERN_INFO "Netlink module unloaded\n");
}module_init(netlink_init);
module_exit(netlink_exit);MODULE_LICENSE("GPL");

5. 编译和测试

编译内核模块
make -C /lib/modules/$(uname -r)/build M=$(pwd) modules
加载模块
sudo insmod netlink_test.ko
运行用户态程序
./user_netlink

6. 输出示例

  • 用户态程序:

    Message sent to kernel: Hello from user space!
    Message received from kernel: Hello from kernel!
    
  • 内核日志(dmesg):

    [INFO] Kernel received message: Hello from user space!
    

总结

通过以上代码,用户态程序和内核模块实现了简单的双向 Netlink 通信。根据实际需求,可以扩展消息内容和通信逻辑。


http://www.ppmy.cn/news/1557556.html

相关文章

一款轻量级的开源笔记服务软件

大家好&#xff0c;我是兔兔&#xff0c;一位写作爱好者&#xff0c;今天分享的内容是&#xff0c;如何搭建一个开源的、隐私优先的轻量级笔记服务应用。 不知道大家是否有这样的需求&#xff1a; 1、自己想搭建一个个人的学习笔记文档&#xff0c;既要自己看也可以单独分享给…

Scrcpy:安卓投屏神器介绍及详细使用步骤

目录 一、Scrcpy简介 1.1. 基本概述 1.2. 主要功能 1.2.1. 屏幕镜像 1.2.2. 操作控制 1.2.3. 视频录制和截图 1.2.4. 无线连接 1.2.5. 多设备连接 二、使用场景 2.1. 开发调试 2.2. 游戏录制 2.3. 教学演示 2.4. 远程协助 三、安装与配置 3.1. 安装Scrcpy 3.1.…

GoTime#34期 Pachyderm, Provenance, Data Lakes

本篇内容是根据2017年2月份#34 Pachyderm, Provenance, Data Lakes音频录制内容的整理与翻译 Joe Doliner 加入了节目&#xff0c;谈论使用 Pachyderm 管理数据湖、数据容器、溯源(provenance) 以及其他有趣的 Go 项目和新闻。 Erik St. Martin: 大家好&#xff0c;欢迎收听新…

Qt有哪些读取文件的方式

1. 使用 QFile 和 QTextStream&#xff08;文本文件读取&#xff09; 适用于纯文本文件&#xff0c;按行或整体读取。 示例代码&#xff1a;逐行读取 QFile file("example.txt"); if (file.open(QIODevice::ReadOnly | QIODevice::Text)) {QTextStream in(&fi…

C# 面试中常见递归算法

前言 今天我们主要总结一下C#面试中常见递归算法。 C#经典十大排序算法&#xff08;完结&#xff09; C#递归算法计算阶乘的方法 一个正整数的阶乘&#xff08;factorial&#xff09;是所有小于及等于该数的正整数的积&#xff0c;并且0的阶乘为1。自然数n的阶乘写作n!。180…

[Unity]【图形渲染】【游戏开发】Shader数学基础4-更多矢量运算

在计算机图形学和着色器编程中,矢量运算是核心的数学工具之一。矢量用于描述空间中的位置、方向、速度等各种物理量,并在图形变换、光照计算、纹理映射等方面起着至关重要的作用。本篇文章将详细讲解矢量和标量之间的乘法与除法、矢量的加法与减法、矢量的模与单位矢量、点积…

自动呼入机器人如何实现自动化学习?

自动呼入机器人如何实现自动化学习&#xff1f; 原作者&#xff1a;开源呼叫中心FreeIPCC&#xff0c;其Github&#xff1a;https://github.com/lihaiya/freeipcc 自动呼入机器人的自动化学习是其智能化和持续改进的核心驱动力。通过不断从用户交互中获取新知识&#xff0c;并…

【集合】Java 8 - Stream API 17种常用操作与案例详解

文章目录 Java8 Stream API 17种常用操作与案例详解1. collect()&#xff1a;将流中的元素收集到集合中2. filter()&#xff1a;根据条件过滤流中的元素3. map()&#xff1a;元素映射为另一个值4. forEach()&#xff1a;对流中的元素执行操作5. flatMap()&#xff1a;将流中的元…