linux内核探索--编写一个netlink通信模块

简介: 如何编写一个netlink通信模块

 目录

1,编写内核模块

2,增加相应的生成文件

3,编写用户进程代码

4,测试结果


要求:实现一个网络链路通信模块,用户发送字符串数据后,内核将字符串翻转,然后回复给用户态程序


与系统调用,/ proc,sysfs等类似,netlink也是一种用于用户进程与内核通信的机制,它是基于BSD套接字协议,使用AF_NETLINK地址簇。与系统调用,proc,sysfs文件系统等方式相比,网络链路具有简单,支持双向通信的特点,并支持消息多播机制。

1,编写内核模块

vim netlink_mod.c:

#include <linux/module.h>
#include <net/sock.h>
#include <linux/netlink.h>
#include <linux/skbuff.h>
#define MY_NETLINK_TYPE 31 //max is 32, see netlink.h
struct sock *my_nl_sock = NULL;
static void hello_nl_recv_msg(struct sk_buff *skb)
{
    struct nlmsghdr *nlh;
    int pid;
    struct sk_buff *skb_out;
    int msg_size;
    char *msg=NULL;
    char *msg_zj=NULL;
    int res;
    int i;
    printk(KERN_INFO "Entering: %s\n", __FUNCTION__);
    //msg_size = strlen(msg);
    nlh = (struct nlmsghdr *)skb->data;//Kernel Space receive message from user
    printk(KERN_INFO "Netlink receive msg: %s\n", (char *)nlmsg_data(nlh));
    msg =nlmsg_data(nlh);
    printk(KERN_INFO "msg:%s\n",msg);    
    msg_size = strlen(msg);
    printk(KERN_INFO "size of msg:%d\n",msg_size);    
    //String inversion
    msg_zj = (char *)kmalloc(msg_size + 1,GFP_KERNEL);
    memset(msg_zj, 0, msg_size + 1);
    for(i = 0; i < msg_size;i++)
    {
  msg_zj[i]=msg[msg_size-i-1];
    }    
    printk(KERN_INFO "msg_zj:%s\n",msg_zj);    
    pid = nlh->nlmsg_pid;
    skb_out = nlmsg_new(msg_size, 0);
    if (!skb_out) {
        printk(KERN_INFO "alloc nlmsg failed\n");
        return;
    }
    nlh = nlmsg_put(skb_out, 0, 0, NLMSG_DONE, msg_size, 0);
    NETLINK_CB(skb_out).dst_group = 0;
    //strncpy(nlmsg_data(nlh), msg, msg_size);
    strncpy(nlmsg_data(nlh), msg_zj, msg_size);
    res = nlmsg_unicast(my_nl_sock, skb_out, pid);//Kernel Space send message toUser Space
    if (res < 0)
        printk(KERN_INFO "nlmsg unicast failed\n");
    kfree(msg_zj);
    return;
}
static int __init hello_init(void)
{
    struct netlink_kernel_cfg cfg = {
            .groups = 0,
            .input = hello_nl_recv_msg
    };
    printk("Entering: %s:\n", __FUNCTION__);
    my_nl_sock = netlink_kernel_create(&init_net, MY_NETLINK_TYPE, &cfg);
    if (!my_nl_sock) {
        printk(KERN_ALERT "netlink create sock failed\n");
        return -10;
    }
    return 0;
}
static void __exit hello_exit(void)
{
    printk(KERN_INFO "Exiting hello module\n");
    netlink_kernel_release(my_nl_sock);
    my_nl_sock = NULL;
}
module_init(hello_init);
module_exit(hello_exit);
MODULE_LICENSE("GPL");

image.gif

2,增加相应的生成文件

vim Makefile:

obj-m += netlink_mod.o
#KVERSION = $(shell uname -r)
#PWD = $(shell pwd)
all:
  make -C /lib/modules/$(shell uname -r)/build M=$(PWD) modules
clean:
  make -C /lib/modules/$(shell uname -r)/build M=$(PWD) clean

如果觉得文章对你有帮助,欢迎关注微信公众号:小牛呼噜噜

3,编写用户进程代码

用户进程主要使用netlink的socket与内核进行通信,首先向内核发送一条消息,然后再接收内核的消息,代码如下:

#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/socket.h>
#include <linux/netlink.h>
#define MY_NETLINK_TYPE 31
#define MAX_PAYLOAD 1024 /* maximum payload size*/
struct sockaddr_nl src_addr, dest_addr;
struct nlmsghdr *nlh = NULL;
struct iovec iov;
int sock_fd;
struct msghdr msg;
int 
main(int argc, char *argv[])
{
    sock_fd = socket(PF_NETLINK, SOCK_DGRAM, MY_NETLINK_TYPE);
    if (sock_fd < 0)
        return -1;
    memset(&src_addr, 0, sizeof(src_addr));
    src_addr.nl_family = AF_NETLINK;
    src_addr.nl_pid = getpid(); /* self pid */
    bind(sock_fd, (struct sockaddr *)&src_addr, sizeof(src_addr));
    memset(&dest_addr, 0, sizeof(dest_addr));
    memset(&dest_addr, 0, sizeof(dest_addr));
    dest_addr.nl_family = AF_NETLINK;
    dest_addr.nl_pid = 0; /* For Linux Kernel */
    dest_addr.nl_groups = 0; /* unicast */
    nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));
    memset(nlh, 0, NLMSG_SPACE(MAX_PAYLOAD));
    nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);
    nlh->nlmsg_pid = getpid();
    nlh->nlmsg_flags = 0;
    strcpy(NLMSG_DATA(nlh), "Hello");
    iov.iov_base = (void *)nlh;
    iov.iov_len = nlh->nlmsg_len;
    msg.msg_name = (void *)&dest_addr;
    msg.msg_namelen = sizeof(dest_addr);
    msg.msg_iov = &iov;
    msg.msg_iovlen = 1;
    printf("Sending message to kernel<<<%s\n",NLMSG_DATA(nlh));
    sendmsg(sock_fd, &msg, 0);//User Space send message to kernel
    printf("Waiting for message from kernel\n");
    /* Read message from kernel */
    recvmsg(sock_fd, &msg, 0);//User Space receive message from kernel
    printf("Received message payload>>>%s\n", NLMSG_DATA(nlh));
    close(sock_fd);
    return 0;
}

image.gif

4,测试结果

编译模块:

使

image.png

加载该模块:

sudo insmod netlink_mod.ko

image.gif

加载后可用的lsmod查看是否加载成功,然后编译并运行用户进程hello_user

gcc user.c -o user

image.gif

运行

./user

image.gif

可在终端中看到如下结果:

Sending message to kernel<<<Hello
Waiting for message from kernel
Received message payload>>>olleH
上面是用户进程的输出,可用dmesg命令查看内核模块的输出(或者直接查看/var/log/kern).

image.gif

image.png

卸载模块

sudo rmmod netlink_mod

参考:https//yangjin-unique.github.io/ru-he-zai-linuxnei-he-mo-kuai-zhong-jia-ru-netlinktong-xin-jie-kou.html


相关文章
|
2月前
|
安全 网络协议 Linux
深入理解Linux内核模块:加载机制、参数传递与实战开发
本文深入解析了Linux内核模块的加载机制、参数传递方式及实战开发技巧。内容涵盖模块基础概念、加载与卸载流程、生命周期管理、参数配置方法,并通过“Hello World”模块和字符设备驱动实例,带领读者逐步掌握模块开发技能。同时,介绍了调试手段、常见问题排查、开发规范及高级特性,如内核线程、模块间通信与性能优化策略。适合希望深入理解Linux内核机制、提升系统编程能力的技术人员阅读与实践。
232 1
|
2月前
|
监控 Ubuntu Linux
什么Linux,Linux内核及Linux操作系统
上面只是简单的介绍了一下Linux操作系统的几个核心组件,其实Linux的整体架构要复杂的多。单纯从Linux内核的角度,它要管理CPU、内存、网卡、硬盘和输入输出等设备,因此内核本身分为进程调度,内存管理,虚拟文件系统,网络接口等4个核心子系统。
209 0
|
2月前
|
Web App开发 缓存 Rust
|
2月前
|
Ubuntu 安全 Linux
Ubuntu 发行版更新 Linux 内核,修复 17 个安全漏洞
本地攻击者可以利用上述漏洞,攻击 Ubuntu 22.10、Ubuntu 22.04、Ubuntu 20.04 LTS 发行版,导致拒绝服务(系统崩溃)或执行任意代码。
|
11月前
|
算法 Linux 调度
深入理解Linux内核调度器:从基础到优化####
本文旨在通过剖析Linux操作系统的心脏——内核调度器,为读者揭开其高效管理CPU资源的神秘面纱。不同于传统的摘要概述,本文将直接以一段精简代码片段作为引子,展示一个简化版的任务调度逻辑,随后逐步深入,详细探讨Linux内核调度器的工作原理、关键数据结构、调度算法演变以及性能调优策略,旨在为开发者与系统管理员提供一份实用的技术指南。 ####
394 4
|
机器学习/深度学习 人工智能 负载均衡
深度解析:Linux内核调度策略的演变与优化
【5月更文挑战第30天】 随着计算技术的不断进步,操作系统的性能调优成为了提升计算机系统效率的关键。在众多操作系统中,Linux因其开源和高度可定制性而备受青睐。本文将深入剖析Linux操作系统的内核调度策略,追溯其历史演变过程,并重点探讨近年来为适应多核处理器和实时性要求而产生的调度策略优化。通过分析比较不同的调度算法,如CFS(完全公平调度器)、实时调度类和批处理作业的调度需求,本文旨在为系统管理员和开发者提供对Linux调度机制深层次理解,同时指出未来可能的发展趋势。
|
11月前
|
缓存 并行计算 Linux
深入解析Linux操作系统的内核优化策略
本文旨在探讨Linux操作系统内核的优化策略,包括内核参数调整、内存管理、CPU调度以及文件系统性能提升等方面。通过对这些关键领域的分析,我们可以理解如何有效地提高Linux系统的性能和稳定性,从而为用户提供更加流畅和高效的计算体验。
435 24
|
10月前
|
缓存 监控 网络协议
Linux操作系统的内核优化与实践####
本文旨在探讨Linux操作系统内核的优化策略与实际应用案例,深入分析内核参数调优、编译选项配置及实时性能监控的方法。通过具体实例讲解如何根据不同应用场景调整内核设置,以提升系统性能和稳定性,为系统管理员和技术爱好者提供实用的优化指南。 ####
|
10月前
|
监控 算法 Linux
Linux内核锁机制深度剖析与实践优化####
本文作为一篇技术性文章,深入探讨了Linux操作系统内核中锁机制的工作原理、类型及其在并发控制中的应用,旨在为开发者提供关于如何有效利用这些工具来提升系统性能和稳定性的见解。不同于常规摘要的概述性质,本文将直接通过具体案例分析,展示在不同场景下选择合适的锁策略对于解决竞争条件、死锁问题的重要性,以及如何根据实际需求调整锁的粒度以达到最佳效果,为读者呈现一份实用性强的实践指南。 ####