linux与应用层通信方式 之nf_sock_opt

1. //内核部分代码。大家可以COPY ipt_sockopts

[cpp] view plain copy

1. //#include <linux/config.h>
2. #include <linux/module.h>
3. #include <linux/moduleparam.h>
4. #include <linux/init.h>
5. #include "hello_sock_opt.h"
7. #include <linux/kmod.h>
8. #include <linux/module.h>
9. #include <linux/vmalloc.h>
10. #include <linux/netfilter/x_tables.h>
11. #include <linux/netfilter_bridge/ebtables.h>
12. #include <linux/spinlock.h>
13. #include <linux/mutex.h>
14. #include <linux/slab.h>
15. #include <asm/uaccess.h>
16. #include <linux/smp.h>
17. #include <linux/cpumask.h>
18. #include <linux/audit.h>
19. #include <net/sock.h>
21. char hello[10][20]; 
24. static int hello_set_ctl(struct sock *sk, int cmd, void __user *user, unsigned int len) 
25. { 
26. int ret=0; 
28. if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN)) 
29. return -EPERM; 
30.         printk("%s: len is %d \n",__func__,len); 
31. switch (cmd) { 
32. case HELLO_LISA: 
33. if (copy_from_user(hello[0], user,20) != 0) 
34. return -EFAULT; 
36. break; 
38. case HELLO_MONA: 
39. if (copy_from_user(hello[1], user,20) != 0) 
40. return -EFAULT; 
42. break; 
44. default: 
45.                 printk("hello_set_ctl:  unknown request %i\n", cmd); 
46.                 ret = -EINVAL; 
47.         } 
49. return ret; 
50. } 
52. static int hello_get_ctl(struct sock *sk, int cmd, void __user *user, int *len) 
53. { 
54. int ret=0; 
56. if (!ns_capable(sock_net(sk)->user_ns, CAP_NET_ADMIN)) 
57. return -EPERM; 
59. switch (cmd) { 
60. case HELLO_LISA: 
61. if (copy_to_user(user,hello[0] , 20) != 0) 
62.                 { 
63. break; 
64.                 } 
65. break; 
66. case HELLO_MONA: 
67. if (copy_to_user(user,hello[1] , 20) != 0) 
68. break; 
69. else
70. break; 
72. default: 
73.                 printk("hello_get_ctl:  unknown request %i\n", cmd); 
74.                 ret = -EINVAL; 
75.         } 
77. return ret; 
78. } 
79. static struct nf_sockopt_ops   arpt_sockopts = { 
80.         .pf             = PF_INET, 
81.         .set_optmin     = HELLO_BASE_CTL, 
82.         .set_optmax     = HELLO_SO_SET_MAX+1, 
83.         .set            = hello_set_ctl, 
84.         .get_optmin     = HELLO_BASE_CTL, 
85.         .get_optmax     = HELLO_SO_SET_MAX+1, 
86.         .get            = hello_get_ctl, 
87.         .owner          = THIS_MODULE, 
88. }; 
89. static int simple_init(void) 
90. { 
91. int ret; 
93. /* Register setsockopt */
94.         ret = nf_register_sockopt(&arpt_sockopts); 
95. if (ret < 0) 
96.         { 
97.                 printk("nf_register_sockopt error! \n"); 
98. return -1; 
99.         } 
101.     printk("hello !!\n"); 
102. return 0; 
103. } 
106. static void simple_cleanup(void) 
107. { 
108.         nf_unregister_sockopt(&arpt_sockopts); 
109.         printk("Hello byebye\n"); 
110. } 
113. module_init(simple_init); 
114. module_exit(simple_cleanup); 

[cpp] view plain copy

1. /*
2. 上下通信的通用头文件。
4. */
7. #define  HELLO_BASE_CTL  102400 //在系统中这个号不能有重复的。建议最好是取较大值。
8. #define  HELLO_LISA      HELLO_BASE_CTL+1
9. #define  HELLO_MONA      HELLO_BASE_CTL+2
10. #define  HELLO_SO_SET_MAX  (HELLO_BASE_CTL+5)

[cpp] view plain copy

1. #include <sys/socket.h>
2.        #include <netinet/in.h>
3.        #include <netinet/ip.h> /* superset of previous */
4. #include <sys/types.h>          /* See NOTES */
5. #include <sys/socket.h>
6. #include <string.h>
7. #include <stdio.h>
8. #include <unistd.h>
9. #include <errno.h>
10. #include "hello_sock_opt.h"
13. int common_set_sock_opt(int id,void* msg,int len) 
14. { 
15. int sock; 
16. int ret; 
17.         sock = socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP); 
18. if(sock <=0){ 
19.                 printf("socket error\n"); 
20. return -1; 
21.         } 
22.         ret=setsockopt(sock,0,id,msg,len); 
24.         close(sock); 
25. return ret; 
26. } 
27. int common_get_sock_opt(int id,void* msg,int* len) 
28. { 
29. int sock; 
30. int ret; 
31.         sock = socket(AF_INET,SOCK_DGRAM,IPPROTO_UDP); 
32. if(sock <=0){ 
33.                 printf("socket error\n"); 
34. return -1; 
35.         } 
36.         ret=getsockopt(sock,0,id,msg,len); 
37.         close(sock); 
38. return ret; 
39. } 
40. int __set_lisa(char* s) 
41. { 
42. return common_set_sock_opt(HELLO_LISA,s,strlen(s)); 
44. } 
46. int __get_lisa(char* s,int *len) 
47. { 
48. return common_get_sock_opt(HELLO_LISA,s,len); 
51. } 
54. void main(int argc,char* argv[]) 
55. { 
56. char ret[100]={0}; 
57. int len=0; 
58. int t; 
59. if(argc ==2){ 
60.                 printf("argv : %s :%s \n",argv[0],argv[1]); 
61.                 t = __set_lisa(argv[1]);  //把参数配置到内核。
62. if(t<0) 
63.                         perror("__set_lisa:"); 
64.                 __get_lisa(ret,&len);  //从内核读出信息。
65. if(t<0) 
66.                         perror("__GET_lisa:"); 
67. else
68.                 { 
69.                         printf("ret %s; len %d\n",ret,len); 
70.                 } 
71.         } 
72. }