memfrob()还有什么限制？总是段错误的？

小锐同学

对于下面一个程序，就是把一个文件映射到内存，然后创建一个新文件也映射到内存，然后调用memfrob()函数把源文件的内存块加密，再把加密的内存块复制到新文件那里去。
但是却总是段错误，难道这个函数有什么限制么？
如果不要那个加密的函数的话就能够正常复制。

1. #include <fcntl.h>
   
2. #include <sys/mman.h>
   
3. #include <string.h>
   
4. #include <sys/types.h>
   
5. #include <sys/stat.h>
   
6. #include <unistd.h>
   
7. 
   
8. #define FILE_MODE (S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH)
   
9. 
   
10. int
    
11. main(int argc, char *argv[])
    
12. {
    
13.         int                fdin, fdout;
    
14.         void        *src, *dst;
    
15.         struct stat statbuf;
    
16. 
    
17.         if(argc != 3)
    
18.         {
    
19.                 printf("usage: %s <fromfile> <tofile>", argv[0]);
    
20.                 return 0;
    
21.         }
    
22.         fdin = open(argv[1], O_RDONLY);
    
23.         fdout = open(argv[2], O_RDWR | O_CREAT | O_TRUNC, FILE_MODE);
    
24.         fstat(fdin, &statbuf);
    
25.         lseek(fdout, statbuf.st_size - 1, SEEK_SET);
    
26.         write(fdout, "", 1);
    
27.         src = mmap(0, statbuf.st_size, PROT_READ, MAP_SHARED, fdin, 0);
    
28.         dst = mmap(0, statbuf.st_size, PROT_READ | PROT_WRITE, MAP_SHARED, fdout, 0);
    
29.         memfrob(src, statbuf.st_size);
    
30.         memcpy(dst, src, statbuf.st_size);
    
31.        
    
32.         return 0;
    
33. }
    

复制代码

尝试加密一个小文件。

1. lucifer@cross:~/tmp$ cat tmp3.c
   
2. /*printf 枚举可显示枚举的序号*/
   
3. #include <stdio.h>
   
4. int main(void)
   
5. {
   
6.         enum ABC{AAA,BBB,CCC};
   
7. 
   
8.         enum ABC e1,e2,e3;
   
9. 
   
10.         e1 = AAA;
    
11.         e2 = BBB;
    
12.         e3 = CCC;
    
13. 
    
14.         printf("%d,%d,%d\n",e1,e2,e3);
    
15.         getchar();
    
16.         return 0;
    
17. }
    
18. lucifer@cross:~/tmp$ ./a.out tmp3.c ddd
    
19. Segmentation fault
    
20. lucifer@cross:~/tmp$

复制代码

GDB的调试信息，说明了是 memfrob () 函数出问题了。

1. lucifer@cross:~/tmp$ gdb -q ./a.out
   
2. Reading symbols from /home/lucifer/tmp/a.out...done.
   
3. (gdb) r tmp3.c ddd
   
4. Starting program: /home/lucifer/tmp/a.out tmp3.c ddd
   
5. 
   
6. Program received signal SIGSEGV, Segmentation fault.
   
7. 0x00007ffff7ae184e in memfrob () from /lib/libc.so.6
   
8. (gdb) q
   
9. A debugging session is active.
   
10. 
    
11.         Inferior 1 [process 6216] will be killed.
    
12. 
    
13. Quit anyway? (y or n) y
    
14. lucifer@cross:~/tmp$
    

复制代码

x11

把memfrob和memcpy这两行改为

memcpy(dst, src, statbuf.st_size);
memfrob(dst, statbuf.st_size);

小锐同学

Post by x11;2097097
把memfrob和memcpy这两行改为

memcpy(dst, src, statbuf.st_size);
memfrob(dst, statbuf.st_size);

哦～

我知道了。