1.NetLink机制
NetLink是一种基于应用层跟内核态的通信机制,其特点是一种异步全双工的通信方式,支持内核态主动发起通信的机制。该机制提供了一组特殊的API接口,用户态则通过socket API调用。内核发送的数据再应用层接收后会保存在接收进程socket的缓存中,再由接受进程处理。
2.NetLink通信示意图:
3.驱动文件ko文件的初始化与停止使用方式:
/* 模块入口函数 */
static int __init my_ko_init(void) {/**doing*/printk("I am init self ko !!!");return 0;
}/* 模块退出函数 */
static void __exit my_ko_exit(void) {// 注销netlink协议printk("I am exit self ko !!!");return;
}module_init(my_ko_init)
module_exit(my_ko__exit)
MODULE_AUTHOR("my_self_ko");
MODULE_DESCRIPTION("self ko study");
MODULE_LICENSE("GPL");
注:自定义的ko文件使用,按照如上图的模版标准来进行实现对应的功能,该模版是初始化跟退出还原方法,通过module_init,module_exit,两个函数进行插入加载跟卸载还原内核态信息(避免出现内核崩溃。
makefile:
obj-m += my_self_ko.oall:make -C /lib/modules/$(shell uname -r)/build/ M=$(PWD) modules
clean:make -C /lib/modules/$(shell uname -r)/build/ M=$(PWD) clean
4.用户态创建NetLink套接字通信
4.1 netlink_kernel_create函数
用于创建一个内核态的netlink的socket服务。该函数对应的参数在不同的版本之间是不同的设定:
头文件:linux/netlink.h
netlink_kernel_create(struct net *net, int unit, struct netlink_kernel_cfg *cfg)a. LINUX_VERSION_CODE < KERNEL_VERSION(2,6,24)的函数参数定义:netlink_kernel_create(NETLINK_SELF_MODULE,0,netlink_kernel_rcv,THIS_MODULE);b. KERNEL_VERSION(2,6,24)< LINUX_VERSION_CODE < KERNEL_VERSION(3,6,0)netlink_kernel_create(&init_net, NETLINK_SELF_MODULE,0, netlink_kernel_rcv, NULL, THIS_MODULE);c. LINUX_VERSION_CODE >= KERNEL_VERSION(3,6,0)struct netlink_kernel_cfg cfg = {.input = netlink_kernel_rcv,};netlink_kernel_create(&init_net, NETLINK_SELF_MODULE,0, &cfg);
以上为内核驱动ko文件内创建netlink的方法,netlink_kernel_rcv自定义的回调处理函数。
netlink_kernel_create函数:
返回值: sock 对象指针.
参数1: init_net,系统内部定义的类型,默认使用即可.
参数2:自定义的通信消息类型,根据实际定义,但是避免跟系统冲以及不要超过系统范围。
参数3:默认设置0
参数4:设置系统消息回调函数,不同版本该接口有差异性,如上代码。
具体的方式通过客户端主动链接,双向通信。例如:
内核ko驱动代码:
#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/init.h>
#include <linux/netlink.h>
#include <net/netlink.h>#define NETLINK_TEST 17struct sock *nl_sk = 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 = "Hello from kernel";int res;printk(KERN_INFO "Entering: %s\n", __FUNCTION__);msg_size = strlen(msg);nlh = (struct nlmsghdr *)skb->data;printk(KERN_INFO "Netlink received msg payload:%s\n", (char *)nlmsg_data(nlh));pid = nlh->nlmsg_pid; /*pid of 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 mcast 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 bak to user\n");
}static int __init hello_init(void)
{printk("Entering: %s\n", __FUNCTION__);nl_sk = netlink_kernel_create(&init_net, NETLINK_TEST, 0, hello_nl_recv_msg, NULL, THIS_MODULE);if (!nl_sk) {printk(KERN_ALERT "Error creating socket.\n");return -10;}return 0;
}static void __exit hello_exit(void)
{printk(KERN_INFO "exiting hello module\n");netlink_kernel_release(nl_sk);
}module_init(hello_init);
module_exit(hello_exit);
客户端:
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/socket.h>
#include <linux/netlink.h>#define NETLINK_TEST 17struct sockaddr_nl src_addr, dest_addr;
struct nlmsghdr *nlh = NULL;
struct iovec iov;
struct msghdr msg;int main()
{int sock_fd;sock_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_TEST);if(sock_fd < 0)return -1;memset(&src_addr, 0, sizeof(src_addr));src_addr.nl_family = AF_NETLINK;src_addr.nl_pid = getpid();bind(sock_fd, (struct sockaddr*)&src_addr, sizeof(src_addr));memset(&dest_addr, 0, sizeof(dest_addr));dest_addr.nl_family = AF_NETLINK;dest_addr.nl_pid = 0; // For Linux Kerneldest_addr.nl_groups = 0; // unicastnlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(1024));memset(nlh, 0, NLMSG_SPACE(1024));nlh->nlmsg_len = NLMSG_SPACE(1024);nlh->nlmsg_pid = getpid();nlh->nlmsg_flags = 0;strcpy(NLMSG_DATA(nlh), "Hello from user space!");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\n");sendmsg(sock_fd, &msg, 0);printf("Waiting for message from kernel\n");memset(nlh, 0, NLMSG_SPACE(1024));recvmsg(sock_fd, &msg, 0);printf("Received message: %s\n", NLMSG_DATA(nlh));close(sock_fd);return 0;
}
上述代码中自定义通信类型,不可以跟系统重复,否则会导致写入ko文件失败.如下:
注:驱动文件ko插入内核NetLink启动失败导致内核ko驱动不可用:
如调用失败,最可能原因,第二个参数NETLINK_TEST_MODULE错误,可能已经被系统占用。需 要重新更换该类型参数。可以通过: cat /proc/net/netlink 查看当前使用的内核类型,自我使用是 否跟内核冲突,修改即可。另外应用层也需同步修改NETLINK_TEST_MODULE ,如下所示:
nlsock = netlink_kernel_create(&init_net, NETLINK_TEST_MODULE, &cfg);
注:自定义的文件,是不会主动从内核卸载的,当前系统使用的可以通过lsmod |grep 自定义ko文件名,查看当前的引用计数,只有当引用计数为0时,才可以卸载该驱动文件。可以使用rmmod 驱动文件名。