2008年10月29日 星期三

[轉載]How to use netlink

轉載自:http://linux.chinaunix.net/bbs/thread-1036826-1-1.html

參考:http://www.linuxjournal.com/article/7356


首先声明自己是kernel编程的纯新手,由于要用到netlink实现内核与user space通信,
花了很多时间在这上面。

今天刚刚调通了一个最简单的netlink程序,放在这里,希望其他的新手少走弯路。
网上有很多教程和详细的解释,我就是看了duanjigang的这个帖子,弄懂了很多东西
http://linux.chinaunix.net/bbs/thread-1031932-1-2.html

现在的问题是,netlink的kernel api变了很多(2.6.26 vs 原帖中的程序)。
对于kernel新手来说,没有文档,没有教程
太痛苦。我去翻了从kernel 2.6.20 到 2.6.26所有的change log,看到好像
是在2.6.25里面,netlink socket create api变成了现在这个样子。而且这个过程中,
不止变了一次,因为我在其他网站看到另外一种(第三种)netlink_kernel_create 调用的例程,可能用于更老的kernel。

对于新的kernel没有例程,只能到kerenel源代码里面找例子。感觉这种方法很土,不知道有没有大侠给指点指点,除了跟踪kerenel mail list以外,有没有好的办法应对kernel api的改变。

下面这个程序基本实现了duanjigang那个帖子里面的功能,只不过是应用于kernel 2.6.26下面。
详细的解释还是看原帖吧。其他的系统我没有试过,我的系统是debian lenny 2.6.26-1-686

//net_link.c

#include <&linux/kernel.h&>
#include <&linux/module.h>
#include <&linux/types.h&>
#include <&linux/sched.h&>
#include <&net/sock.h&>
#include <&net/netlink.h&>

#define NETLINK_TEST 21

struct sock *nl_sk = NULL;
EXPORT_SYMBOL_GPL(nl_sk);

void nl_data_ready (struct sk_buff *__skb)
{
struct sk_buff *skb;
struct nlmsghdr *nlh;
u32 pid;
int rc;
int len = NLMSG_SPACE(1200);
char str[100];

printk("net_link: data is ready to read.\n");
skb = skb_get(__skb);

if (skb->len >= NLMSG_SPACE(0)) {
nlh = nlmsg_hdr(skb);
printk("net_link: recv %s.\n", (char *)NLMSG_DATA(nlh));
memcpy(str,NLMSG_DATA(nlh), sizeof(str));
pid = nlh->nlmsg_pid; /*pid of sending process */
printk("net_link: pid is %d\n", pid);
kfree_skb(skb);

skb = alloc_skb(len, GFP_ATOMIC);
if (!skb){
printk(KERN_ERR "net_link: allocate failed.\n");
return;
}
nlh = nlmsg_put(skb,0,0,0,1200,0);
NETLINK_CB(skb).pid = 0; /* from kernel */

memcpy(NLMSG_DATA(nlh), str, sizeof(str));
printk("net_link: going to send.\n");
rc = netlink_unicast(nl_sk, skb, pid, MSG_DONTWAIT);
if (rc < nl_sk =" netlink_kernel_create(&init_net,">sk_socket);
}
printk("net_link: remove ok.\n");
}
MODULE_LICENSE("GPL");
MODULE_AUTHOR("kidoln");



//sender.c
#include <&sys/stat.h&>
#include <&unistd.h&>
#include <&stdio.h&>
#include <&stdlib.h&>
#include <&sys/socket.h&>
#include <&sys/types.h&>
#include <&string.h&>
#include <&asm/types.h&>
#include <&linux/netlink.h&>
#include <&linux/socket.h&>

#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_RAW, 21);
memset(&msg, 0, sizeof(msg));
memset(&src_addr, 0, sizeof(src_addr));
src_addr.nl_family = AF_NETLINK;
src_addr.nl_pid = getpid(); /* self pid */
src_addr.nl_groups = 0; /* not in mcast groups */
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 Kernel */
dest_addr.nl_groups = 0; /* unicast */

nlh=(struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));
/* Fill the netlink message header */
nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);
nlh->nlmsg_pid = getpid(); /* self pid */
nlh->nlmsg_flags = 0;
/* Fill in the netlink message payload */
strcpy(NLMSG_DATA(nlh), "Hello you!");

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. ...\n");
sendmsg(sock_fd, &msg, 0);

/* Read message from kernel */
memset(nlh, 0, NLMSG_SPACE(MAX_PAYLOAD));
printf(" Waiting message. ...\n");
recvmsg(sock_fd, &msg, 0);
printf(" Received message payload: %s\n",
NLMSG_DATA(nlh));

/* Close Netlink Socket */
close(sock_fd);
}


#Makefile

MODULE_NAME :=net_link
obj-m :=$(MODULE_NAME).o
KERNELDIR ?= /lib/modules/$(shell uname -r)/build
PWD := $(shell pwd)
all:
$(MAKE) -C $(KERNELDIR) M=$(PWD)
gcc -o sender sender.c
clean:
rm -fr *.ko *.o *.cmd sender $(MODULE_NAME).mod.c


Example download URL:

http://linux.chinaunix.net/bbs/attachment.php?aid=217363

沒有留言: