Linux内核和用户空间通信的方式（一）— proc文件和mmap共享内存

jlwcb

动态的将内核空间的物理地址和大小传给用户空间。本文也演示了内核空间和用户空间进行通信可以使用的两种常用方法：proc文件系统和mmap共享内存。
整个内核模块，在模块插入时建立proc文件，分配内存。卸载模块的时候将用户空间写入的内容打印出来。
以下是内核模块的代码和用户空间的测试代码。

1. /*This program is used to allocate memory in kernel
   
2. and pass the physical address to userspace through proc file.*/
   
3. 
   
4. #include <linux/version.h>
   
5. #include <linux/module.h>
   
6. #include <linux/proc_fs.h>
   
7. #include <linux/mm.h>
   
8. 
   
9. #define PROC_MEMSHARE_DIR "memshare"
   
10. #define PROC_MEMSHARE_PHYADDR "phymem_addr"
    
11. #define PROC_MEMSHARE_SIZE "phymem_size"
    
12. 
    
13. /*alloc one page. 4096 bytes*/
    
14. #define PAGE_ORDER 0
    
15. /*this value can get from PAGE_ORDER*/
    
16. #define PAGES_NUMBER 1
    
17. 
    
18. struct proc_dir_entry *proc_memshare_dir ;
    
19. unsigned long kernel_memaddr = 0;
    
20. unsigned long kernel_memsize= 0;
    
21. 
    
22. static int proc_read_phymem_addr(char *page, char **start, off_t off, int count)
    
23. {
    
24.         return sprintf(page, "%08lx\n", __pa(kernel_memaddr));
    
25. }
    
26. static int proc_read_phymem_size(char *page, char **start, off_t off, int count)
    
27. {
    
28.         return sprintf(page, "%lu\n", kernel_memsize);
    
29. }
    
30. 
    
31. static int __init init(void)
    
32. {
    
33.         /*build proc dir "memshare"and two proc files: phymem_addr, phymem_size in the dir*/
    
34.          proc_memshare_dir = proc_mkdir(PROC_MEMSHARE_DIR, NULL);
    
35.          create_proc_info_entry(PROC_MEMSHARE_PHYADDR, 0, proc_memshare_dir, proc_read_phymem_addr);
    
36.          create_proc_info_entry(PROC_MEMSHARE_SIZE, 0, proc_memshare_dir, proc_read_phymem_size);
    
37. 
    
38.         /*alloc one page*/
    
39.          kernel_memaddr =__get_free_pages(GFP_KERNEL, PAGE_ORDER);
    
40.         if(!kernel_memaddr)
    
41.         {
    
42.                  printk("Allocate memory failure!\n");
    
43.         }
    
44.         else
    
45.         {
    
46.                  SetPageReserved(virt_to_page(kernel_memaddr));
    
47. 
    
48. 内核中申请到页面之后，要调用一下SetPageReserved，相当于告诉系统，这个页面我已经占了。对于每一个申请到的页面，应该都要这样做
    
49. 
    
50.                  kernel_memsize = PAGES_NUMBER * PAGE_SIZE;
    
51.                  printk("Allocate memory success!. The phy mem addr=%08lx, size=%lu\n", __pa(kernel_memaddr), kernel_memsize);
    
52.         }
    
53.         return 0;
    
54. }
    
55. 
    
56. static void __exit fini(void)
    
57. {
    
58.          printk("The content written by user is: %s\n", (unsigned char *) kernel_memaddr);
    
59.          ClearPageReserved(virt_to_page(kernel_memaddr));
    
60.          free_pages(kernel_memaddr, PAGE_ORDER);
    
61.          remove_proc_entry(PROC_MEMSHARE_PHYADDR, proc_memshare_dir);
    
62.          remove_proc_entry(PROC_MEMSHARE_SIZE, proc_memshare_dir);
    
63.          remove_proc_entry(PROC_MEMSHARE_DIR, NULL);
    
64. 
    
65.         return;
    
66. }
    
67. module_init(init);
    
68. module_exit(fini);
    
69. MODULE_LICENSE("GPL");
    
70. MODULE_AUTHOR("Godbach ([email]nylzhaowei@163.com[/email])");
    
71. MODULE_DESCRIPTION("Kernel memory share module.");
    

复制代码

用户空间的测试代码：

1. #include <stdio.h>
   
2. #include <stdlib.h>
   
3. #include <unistd.h>
   
4. #include <string.h>
   
5. #include <fcntl.h>
   
6. #include <sys/stat.h>
   
7. #include <sys/types.h>
   
8. #include <sys/mman.h>
   
9. 
   
10. int main(int argc, char* argv[])
    
11. {
    
12.         if(argc != 2)
    
13.         {
    
14.                 printf("Usage: %s string\n", argv[0]);
    
15.                 return 0;
    
16.         }
    
17.         
    
18.         unsigned long phymem_addr, phymem_size;
    
19.         char *map_addr;
    
20.         char s[256];
    
21.         int fd;
    
22.         
    
23.         /*get the physical address of allocated memory in kernel*/
    
24.          fd = open("/proc/memshare/phymem_addr", O_RDONLY);
    
25.         if(fd < 0)
    
26.         {
    
27.                 printf("cannot open file /proc/memshare/phymem_addr\n");
    
28.                 return 0;
    
29.         }
    
30.         read(fd, s, sizeof(s));
    
31.         sscanf(s, "%lx", &phymem_addr);
    
32.         close(fd);
    
33. 
    
34.         /*get the size of allocated memory in kernel*/
    
35.          fd = open("/proc/memshare/phymem_size", O_RDONLY);
    
36.         if(fd < 0)
    
37.         {
    
38.                 printf("cannot open file /proc/memshare/phymem_size\n");
    
39.                 return 0;
    
40.         }
    
41.         read(fd, s, sizeof(s));
    
42.         sscanf(s, "%lu", &phymem_size);
    
43.         close(fd);
    
44.         
    
45.         printf("phymem_addr=%lx, phymem_size=%lu\n", phymem_addr, phymem_size);
    
46.         /*memory map*/
    
47.         int map_fd = open("/dev/mem", O_RDWR);
    
48.         if(map_fd < 0)
    
49.         {
    
50.                 printf("cannot open file /dev/mem\n");
    
51.                 return 0;
    
52.         }
    
53.         
    
54.          map_addr = mmap(0, phymem_size, PROT_READ|PROT_WRITE, MAP_SHARED, map_fd, phymem_addr);
    
55.         strcpy(map_addr, argv[1]);
    
56.          munmap(map_addr, phymem_size);
    
57.         close(map_fd);
    
58.         return 0;
    
59.         
    
60. }
    

复制代码

测试的内核是2.6.25.以下是执行结果。

1. debian:/home/km/memshare# insmod memshare_kernel.ko
   
2. debian:/home/km/memshare# ./memshare_user 'hello,world!'
   
3. phymem_addr=e64e000, phymem_size=4096
   
4. debian:/home/km/memshare# cat /proc/memshare/phymem_addr
   
5. 0e64e000
   
6. debian:/home/km/memshare# cat /proc/memshare/phymem_size
   
7. 4096
   
8. debian:/home/km/memshare# rmmod memshare_kernel
   
9. debian:/home/km/memshare# tail /var/log/messages
   
10. Sep 27 18:14:24 debian kernel: [50527.567931] Allocate memory success!. The phy mem addr=0e64e000, size=4096
    
11. Sep 27 18:15:31 debian kernel: [50592.570986] The content written by user is: hello,world!
    

复制代码

用你这种方式，想申请4M的内存，

static int __init init(void)
{
     char * paddr;
     int order = 0;
     int offset = 0;
     int tmp = 0;
     
         /*build proc dir "memshare"and two proc files: phymem_addr, phymem_size in the dir*/
         proc_memshare_dir = proc_mkdir(PROC_MEMSHARE_DIR, NULL);
         create_proc_info_entry(PROC_MEMSHARE_PHYADDR, 0, proc_memshare_dir, proc_read_phymem_addr);
         create_proc_info_entry(PROC_MEMSHARE_SIZE,     0, proc_memshare_dir, proc_read_phymem_size);

         /*alloc one page*/
         //kernel_memaddr =__get_free_pages(GFP_KERNEL, PAGE_ORDER);
           order = get_order(4000000);
         kernel_memaddr =__get_free_pages(GFP_KERNEL, order);

         if(!kernel_memaddr)
         {
                 printk("Allocate memory failure!\n");
         }
         else
         {
                 SetPageReserved(virt_to_page(kernel_memaddr));

                 //kernel_memsize = PAGES_NUMBER * PAGE_SIZE;
                kernel_memsize = ((4000000 - 1) >> PAGE_SHIFT)* PAGE_SIZE;

...
}


我在内核中往第一页（4k范围内）写数据，在用户态是能够读出来的，但是如果向以后的其它页写数据时，用户态就读不出了，
用户态写也这样。

需要对每个4k的页面执行：SetPageReserved(virt_to_page(kernel_memaddr));

内核中申请到页面之后，要调用一下SetPageReserved，相当于告诉系统，这个页面我已经占了。对于每一个申请到的页面，应该都要这样做。

转载自：http://hi.baidu.com/liuming910/b ... 85df72dab4bdbf.html

Godbach

唉，这个帖子经过几手的转载，原来就是我自己发的。

内核版或者驱动板就有这个帖子。

建议 LZ 不要一味的转贴，而是对于一些帖子和知识，发一些自己的见解和看法。

瀚海书香

呵呵。我说这哥们怎么突然发这么多高质量的帖子呢，原来都是抄袭的啊。