linux 内核网络Publisher-Subscriber Pattern学习

Linux内核中的发布者-订阅者模式（Publisher-Subscriber Pattern）是一种用于实现事件通知和处理的机制。它通常用于内核模块之间的通信和事件处理，以及用户空间应用程序与内核之间的交互。该模式使得不同的组件能够松散地耦合，以便它们能够以一种高效且灵活的方式共享信息。在这个模式中，发布者负责生成事件或消息，而订阅者则注册以接收感兴趣的事件，并在事件发生时执行相应的操作。发布者-订阅者模式在内核中的应用非常广泛，可以用于各种用例，包括事件通知、数据传输、进程通信、设备驱动程序、系统监控等。

在Linux内核中，发布者-订阅者模式的核心概念是通过观察者模式来实现的，观察者模式中有以下关键角色：

发布者（Publisher）：发布者负责生成事件或信息，并将其发送给所有已注册的订阅者。在内核中，发布者通常是一个模块、子系统或内核组件。发布者负责通知订阅者有关事件的发生。

订阅者（Subscriber）：订阅者是对特定事件或信息感兴趣的组件。它们向发布者注册以接收相关事件的通知，并在事件发生时采取相应的行动。在内核中，订阅者通常是另一个模块、设备驱动程序或用户空间应用程序。

事件或消息（Event/Message）：发布者生成的信息，通常包含有关事件的相关数据。这些消息可以是简单的通知，也可以包含详细的信息。

Linux内核中的API

在Linux内核中，实现发布者-订阅者模式的API（Application Programming Interface，应用程序编程接口）以及实现机制包括：

（1）Netlink套接字 API：Netlink是Linux内核中用于进程之间和用户空间与内核之间通信的机制。通过Netlink套接字，内核模块或用户空间程序可以注册成为发布者或订阅者，发布者向用户空间发布事件，用户空间应用程序可以通过监听Netlink套接字接收事件通知。netlink_kernel_create() 用于创建内核中的Netlink套接字，nlmsg_unicast() 用于将消息发送给一个特定的订阅者。

（2）回调函数 API：发布者可以使用回调函数机制，将事件传递给已注册的订阅者。订阅者在注册时提供回调函数，通过向发布者注册回调函数来接收事件通知，以便在事件发生时被调用。内核中的事件通知通常使用回调函数来实现。

（3）事件通知 API：即内核模块通信，内核中的许多子系统（如文件系统、网络协议栈等不同的内核模块之间）具有内置的发布者-订阅者机制，可以使用此模式进行通信，允许内核模块注册以接收有关特定事件的通知。换句话说，内核的不同子系统通常使用发布者-订阅者模式来通知其他子系统或用户空间应用程序有关特定事件的发生。例如，内核可以通知文件系统模块有新的文件被创建，以便执行相关操作。字符设备和块设备驱动程序通常也会实现注册和通知机制，以便用户空间应用程序或其他内核模块能够订阅设备事件，如数据可用性或设备状态更改。不同子系统内核通常定义了自己的事件通知机制。例如，字符设备驱动程序可以使用 file_operations 结构中的回调函数通知订阅者。

演示在Linux内核模块中使用发布者-订阅者模式的示例如下：
1、例子一：

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/netlink.h>
#include <linux/skbuff.h>

#define NETLINK_USER 31

struct sock *nl_sk = NULL;

static void nl_recv_msg(struct sk_buff *skb) {
    struct nlmsghdr *nlh;
    int pid;
    struct sk_buff *skb_out;
    int msg_size;
    char *msg = "Hello from kernel";
    int res;

    msg_size = strlen(msg);

    nlh = nlmsg_hdr(skb);
    printk(KERN_INFO "Netlink received message: %s\n", (char *)nlmsg_data(nlh));
    pid = nlh->nlmsg_pid; // Get the process ID of the sending process

    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);
    NETLINK_CB(skb_out).dst_group = 0; // Not in any multicast group
    strncpy(nlmsg_data(nlh), msg, msg_size);

    res = nlmsg_unicast(nl_sk, skb_out, pid);
    if (res < 0)
        printk(KERN_INFO "Error while sending back to user\n");
}

static int __init init_module_func(void) {
    struct netlink_kernel_cfg cfg = {
        .input = nl_recv_msg,
    };

    printk(KERN_INFO "Netlink Kernel Module Loaded\n");

    nl_sk = netlink_kernel_create(&init_net, NETLINK_USER, &cfg);
    if (!nl_sk) {
        printk(KERN_ALERT "Error creating socket.\n");
        return -1;
    }
    return 0;
}

static void __exit exit_module_func(void) {
    printk(KERN_INFO "Netlink Kernel Module Unloaded\n");
    netlink_kernel_release(nl_sk);
}

module_init(init_module_func);
module_exit(exit_module_func);

MODULE_LICENSE("GPL");

以上代码创建了一个Netlink套接字，允许用户空间程序通过该套接字向内核模块发送消息。内核模块注册了一个消息接收回调函数 nl_recv_msg，它将在接收到消息时被调用。在消息接收回调函数中，内核可以执行特定的操作，然后向用户空间发送回复。

2、例子二：
创建一个内核模块，其中包括发布者和两个订阅者。发布者可以注册回调函数，而两个订阅者注册自己的回调函数以接收事件通知。当事件发生时，发布者会通知所有已注册的订阅者。

#include <linux/module.h>
#include <linux/kernel.h>

// 定义回调函数类型
typedef void (*EventCallback)(void);

// 发布者
static EventCallback publisher = NULL;

// 注册回调函数
void register_callback(EventCallback callback) {
    publisher = callback;
}

// 模拟事件发生
static void simulate_event(void) {
    if (publisher != NULL) {
        printk(KERN_INFO "Event occurred!\n");
        // 调用注册的回调函数
        publisher();
    }
}

// 订阅者1
static void subscriber1_callback(void) {
    printk(KERN_INFO "Subscriber 1 received the event!\n");
}

// 订阅者2
static void subscriber2_callback(void) {
    printk(KERN_INFO "Subscriber 2 received the event!\n");
}

static int __init init_module_func(void) {
    printk(KERN_INFO "Publisher-Subscriber Module Loaded\n");

    // 注册订阅者的回调函数
    register_callback(subscriber1_callback);
    register_callback(subscriber2_callback);

    // 模拟事件发生
    simulate_event();

    return 0;
}

static void __exit exit_module_func(void) {
    printk(KERN_INFO "Publisher-Subscriber Module Unloaded\n");
}

module_init(init_module_func);
module_exit(exit_module_func);

MODULE_LICENSE("GPL");

实际中，内核模块可能会处理更复杂的消息和事件，以及更多的错误检查和容错处理，可能涉及更复杂的数据结构。此外，内核常常通过Netlink实现更强大和更复杂的发布者-订阅者通信。