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


相关文章
|
6天前
|
算法 Linux
深入探索Linux内核的内存管理机制
本文旨在为读者提供对Linux操作系统内核中内存管理机制的深入理解。通过探讨Linux内核如何高效地分配、回收和优化内存资源,我们揭示了这一复杂系统背后的原理及其对系统性能的影响。不同于常规的摘要,本文将直接进入主题,不包含背景信息或研究目的等标准部分,而是专注于技术细节和实际操作。
|
6天前
|
存储 缓存 网络协议
Linux操作系统的内核优化与性能调优####
本文深入探讨了Linux操作系统内核的优化策略与性能调优方法,旨在为系统管理员和高级用户提供一套实用的指南。通过分析内核参数调整、文件系统选择、内存管理及网络配置等关键方面,本文揭示了如何有效提升Linux系统的稳定性和运行效率。不同于常规摘要仅概述内容的做法,本摘要直接指出文章的核心价值——提供具体可行的优化措施,助力读者实现系统性能的飞跃。 ####
|
7天前
|
监控 算法 Linux
Linux内核锁机制深度剖析与实践优化####
本文作为一篇技术性文章,深入探讨了Linux操作系统内核中锁机制的工作原理、类型及其在并发控制中的应用,旨在为开发者提供关于如何有效利用这些工具来提升系统性能和稳定性的见解。不同于常规摘要的概述性质,本文将直接通过具体案例分析,展示在不同场景下选择合适的锁策略对于解决竞争条件、死锁问题的重要性,以及如何根据实际需求调整锁的粒度以达到最佳效果,为读者呈现一份实用性强的实践指南。 ####
|
7天前
|
缓存 监控 网络协议
Linux操作系统的内核优化与实践####
本文旨在探讨Linux操作系统内核的优化策略与实际应用案例,深入分析内核参数调优、编译选项配置及实时性能监控的方法。通过具体实例讲解如何根据不同应用场景调整内核设置,以提升系统性能和稳定性,为系统管理员和技术爱好者提供实用的优化指南。 ####
|
9天前
|
负载均衡 算法 Linux
深入探索Linux内核调度机制:公平与效率的平衡####
本文旨在剖析Linux操作系统内核中的进程调度机制,特别是其如何通过CFS(完全公平调度器)算法实现多任务环境下资源分配的公平性与系统响应速度之间的微妙平衡。不同于传统摘要的概览性质,本文摘要将直接聚焦于CFS的核心原理、设计目标及面临的挑战,为读者揭开Linux高效调度的秘密。 ####
26 3
|
24天前
|
算法 Linux 调度
深入理解Linux内核调度器:从基础到优化####
本文旨在通过剖析Linux操作系统的心脏——内核调度器,为读者揭开其高效管理CPU资源的神秘面纱。不同于传统的摘要概述,本文将直接以一段精简代码片段作为引子,展示一个简化版的任务调度逻辑,随后逐步深入,详细探讨Linux内核调度器的工作原理、关键数据结构、调度算法演变以及性能调优策略,旨在为开发者与系统管理员提供一份实用的技术指南。 ####
64 4
|
13天前
|
缓存 并行计算 Linux
深入解析Linux操作系统的内核优化策略
本文旨在探讨Linux操作系统内核的优化策略,包括内核参数调整、内存管理、CPU调度以及文件系统性能提升等方面。通过对这些关键领域的分析,我们可以理解如何有效地提高Linux系统的性能和稳定性,从而为用户提供更加流畅和高效的计算体验。
26 2
|
13天前
|
缓存 网络协议 Linux
深入探索Linux操作系统的内核优化策略####
本文旨在探讨Linux操作系统内核的优化方法,通过分析当前主流的几种内核优化技术,结合具体案例,阐述如何有效提升系统性能与稳定性。文章首先概述了Linux内核的基本结构,随后详细解析了内核优化的必要性及常用手段,包括编译优化、内核参数调整、内存管理优化等,最后通过实例展示了这些优化技巧在实际场景中的应用效果,为读者提供了一套实用的Linux内核优化指南。 ####
40 1
|
27天前
|
缓存 资源调度 安全
深入探索Linux操作系统的心脏——内核配置与优化####
本文作为一篇技术性深度解析文章,旨在引领读者踏上一场揭秘Linux内核配置与优化的奇妙之旅。不同于传统的摘要概述,本文将以实战为导向,直接跳入核心内容,探讨如何通过精细调整内核参数来提升系统性能、增强安全性及实现资源高效利用。从基础概念到高级技巧,逐步揭示那些隐藏在命令行背后的强大功能,为系统管理员和高级用户打开一扇通往极致性能与定制化体验的大门。 --- ###
59 9
|
27天前
|
算法 Unix Linux
深入理解Linux内核调度器:原理与优化
本文探讨了Linux操作系统的心脏——内核调度器(Scheduler)的工作原理,以及如何通过参数调整和代码优化来提高系统性能。不同于常规摘要仅概述内容,本摘要旨在激发读者对Linux内核调度机制深层次运作的兴趣,并简要介绍文章将覆盖的关键话题,如调度算法、实时性增强及节能策略等。