Linux下netlink通信的实例代码

Linux下netlink通信的实例代码

 http://blog.csdn.net/love_life2011/article/details/7596190#comments
源代码共分三个文件：

内核模块：netlink-exam-kern.c

应用接收：netlink-exam-user-recv.c

应用发送：netlink-exam-user-send.c

内核模块源码：

[cpp]  view plain copy

1. <span style="font-size:18px;">//kernel module: netlink-exam-kern.c  
2. #include <linux/config.h>  
3. #include <linux/module.h>  
4. #include <linux/netlink.h>  
5. #include <linux/sched.h>  
6. #include <net/sock.h>  
7. #include <linux/proc_fs.h>  
8.   
9. #define BUF_SIZE 16384  
10. static struct sock *netlink_exam_sock;  
11. static unsigned char buffer[BUF_SIZE];  
12. static unsigned int buffer_tail = 0;  
13. static int exit_flag = 0;  
14. static DECLARE_COMPLETION(exit_completion);  
15.   
16. static void recv_handler(struct sock * sk, int length)  
17. {  
18.         wake_up(sk->sk_sleep);  
19. }  
20.   
21. static int process_message_thread(void * data)  
22. {  
23.         struct sk_buff * skb = NULL;  
24.         struct nlmsghdr * nlhdr = NULL;  
25.         int len;  
26.         DEFINE_WAIT(wait);  
27.   
28.         daemonize("mynetlink");  
29.   
30.         while (exit_flag == 0) {  
31.                 prepare_to_wait(netlink_exam_sock->sk_sleep, &wait, TASK_INTERRUPTIBLE);  
32.                 schedule();  
33.                 finish_wait(netlink_exam_sock->sk_sleep, &wait);   
34.   
35.                 while ((skb = skb_dequeue(&netlink_exam_sock->sk_receive_queue))  
36.                          != NULL) {  
37.                         nlhdr = (struct nlmsghdr *)skb->data;  
38.                         if (nlhdr->nlmsg_len < sizeof(struct nlmsghdr)) {  
39.                                 printk("Corrupt netlink message.\n");  
40.                                 continue;  
41.                         }  
42.                         len = nlhdr->nlmsg_len - NLMSG_LENGTH(0);  
43.                         if (len + buffer_tail > BUF_SIZE) {  
44.                                 printk("netlink buffer is full.\n");  
45.                         }  
46.                         else {  
47.                                 memcpy(buffer + buffer_tail, NLMSG_DATA(nlhdr), len);  
48.                                 buffer_tail += len;  
49.                         }  
50.                         nlhdr->nlmsg_pid = 0;  
51.                         nlhdr->nlmsg_flags = 0;  
52.                         NETLINK_CB(skb).pid = 0;  
53.                         NETLINK_CB(skb).dst_pid = 0;  
54.                         NETLINK_CB(skb).dst_group = 1;  
55.                         netlink_broadcast(netlink_exam_sock, skb, 0, 1, GFP_KERNEL);  
56.                 }  
57.         }  
58.         complete(&exit_completion);  
59.         return 0;  
60. }  
61.   
62. static int netlink_exam_readproc(char *page, char **start, off_t off,  
63.                           int count, int *eof, void *data)  
64. {  
65.         int len;  
66.   
67.         if (off >= buffer_tail) {  
68.                 * eof = 1;  
69.                 return 0;  
70.         }  
71.         else {  
72.                 len = count;  
73.                 if (count > PAGE_SIZE) {  
74.                         len = PAGE_SIZE;  
75.                 }  
76.                 if (len > buffer_tail - off) {  
77.                         len = buffer_tail - off;  
78.                 }  
79.                 memcpy(page, buffer + off, len);  
80.                 *start = page;  
81.                 return len;  
82.         }  
83.   
84. }  
85.   
86. static int __init netlink_exam_init(void)  
87. {  
88.         netlink_exam_sock = netlink_kernel_create(NETLINK_GENERIC, 0, recv_handler, THIS_MODULE);  
89.         if (!netlink_exam_sock) {  
90.                 printk("Fail to create netlink socket.\n");  
91.                 return 1;  
92.         }  
93.         kernel_thread(process_message_thread, NULL, CLONE_KERNEL);  
94.         create_proc_read_entry("netlink_exam_buffer", 0444, NULL, netlink_exam_readproc, 0);  
95.         return 0;  
96. }  
97.   
98. static void __exit netlink_exam_exit(void)  
99. {  
100.         exit_flag = 1;  
101.         wake_up(netlink_exam_sock->sk_sleep);  
102.         wait_for_completion(&exit_completion);  
103.         sock_release(netlink_exam_sock->sk_socket);  
104. }  
105.   
106. module_init(netlink_exam_init);  
107. module_exit(netlink_exam_exit);  
108. MODULE_LICENSE("GPL");  
109.   
110. </span>  

用户接收源码：

[cpp]  view plain copy

1. <span style="font-size:18px;">//application receiver: netlink-exam-user-recv.c  
2. #include <stdio.h>  
3. #include <sys/types.h>  
4. #include <sys/socket.h>  
5. #include <linux/netlink.h>  
6.   
7. #define MAX_MSGSIZE 1024  
8.   
9.   
10. int main(void)  
11. {  
12.         struct sockaddr_nl saddr, daddr;  
13.         struct nlmsghdr *nlhdr = NULL;  
14.         struct msghdr msg;  
15.         struct iovec iov;  
16.         int sd;  
17.         int ret = 1;  
18.   
19.         sd = socket(AF_NETLINK, SOCK_RAW,NETLINK_GENERIC);  
20.         memset(&saddr, 0, sizeof(saddr));  
21.         memset(&daddr, 0, sizeof(daddr));  
22.   
23.         saddr.nl_family = AF_NETLINK;        
24.         saddr.nl_pid = getpid();  
25.         saddr.nl_groups = 1;  
26.         bind(sd, (struct sockaddr*)&saddr, sizeof(saddr));  
27.   
28.         nlhdr = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_MSGSIZE));  
29.   
30.         while (1) {  
31.                 memset(nlhdr, 0, NLMSG_SPACE(MAX_MSGSIZE));  
32.   
33.                 iov.iov_base = (void *)nlhdr;  
34.                 iov.iov_len = NLMSG_SPACE(MAX_MSGSIZE);  
35.                 msg.msg_name = (void *)&daddr;  
36.                 msg.msg_namelen = sizeof(daddr);  
37.                 msg.msg_iov = &iov;  
38.                 msg.msg_iovlen = 1;  
39.   
40.                 ret = recvmsg(sd, &msg, 0);  
41.                 if (ret == 0) {  
42.                         printf("Exit.\n");  
43.                         exit(0);  
44.                 }  
45.                 else if (ret == -1) {  
46.                         perror("recvmsg:");  
47.                         exit(1);  
48.                 }  
49.                 printf("%s", NLMSG_DATA(nlhdr));  
50.         }  
51.    
52.         close(sd);  
53. }  
54.   
55. </span>  

 

用户发送源码：

[cpp]  view plain copy

1. <span style="font-size:18px;">//application sender: netlink-exam-user-send.c  
2. #include <stdio.h>  
3. #include <sys/types.h>  
4. #include <sys/socket.h>  
5. #include <linux/netlink.h>  
6.   
7. #define MAX_MSGSIZE 1024  
8.   
9.   
10. int main(int argc, char * argv[])  
11. {  
12.         FILE * fp;  
13.         struct sockaddr_nl saddr, daddr;  
14.         struct nlmsghdr *nlhdr = NULL;  
15.         struct msghdr msg;  
16.         struct iovec iov;  
17.         int sd;  
18.         char text_line[MAX_MSGSIZE];  
19.         int ret = -1;  
20.   
21.         if (argc < 2) {  
22.                 printf("Usage: %s atextfilename\n", argv[0]);  
23.                 exit(1);  
24.         }  
25.   
26.         if ((fp = fopen(argv[1], "r")) == NULL) {  
27.                 printf("File %s dosen't exist.\n");  
28.                 exit(1);  
29.         }  
30.   
31.         sd = socket(AF_NETLINK, SOCK_RAW,NETLINK_GENERIC);  
32.         memset(&saddr, 0, sizeof(saddr));  
33.         memset(&daddr, 0, sizeof(daddr));  
34.   
35.         saddr.nl_family = AF_NETLINK;        
36.         saddr.nl_pid = getpid();  
37.         saddr.nl_groups = 0;  
38.         bind(sd, (struct sockaddr*)&saddr, sizeof(saddr));  
39.   
40.         daddr.nl_family = AF_NETLINK;  
41.         daddr.nl_pid = 0;  
42.         daddr.nl_groups = 0;  
43.   
44.         nlhdr = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_MSGSIZE));  
45.   
46.         while (fgets(text_line, MAX_MSGSIZE, fp)) {  
47.                 memcpy(NLMSG_DATA(nlhdr), text_line, strlen(text_line));  
48.                 memset(&msg, 0 ,sizeof(struct msghdr));  
49.   
50.                 nlhdr->nlmsg_len = NLMSG_LENGTH(strlen(text_line));  
51.                 nlhdr->nlmsg_pid = getpid();  /* self pid */  
52.                 nlhdr->nlmsg_flags = 0;  
53.   
54.                 iov.iov_base = (void *)nlhdr;  
55.                 iov.iov_len = nlhdr->nlmsg_len;  
56.                 msg.msg_name = (void *)&daddr;  
57.                 msg.msg_namelen = sizeof(daddr);  
58.                 msg.msg_iov = &iov;  
59.                 msg.msg_iovlen = 1;  
60.                 ret = sendmsg(sd, &msg, 0);  
61.                 if (ret == -1) {  
62.                         perror("sendmsg error:");  
63.                 }  
64.         }  
65.    
66.         close(sd);  
67. }  
68.   
69. </span>