免费注册 查看新帖 |

Chinaunix

  平台 论坛 博客 文库
› 论坛 › 程序设计 › C/C++ › 写了一段很丢人的代码,内存free不了,谁有BSD帮我测试一 ...
12
最近访问板块 发新帖
楼主: redor
打印 上一主题 下一主题

写了一段很丢人的代码,内存free不了,谁有BSD帮我测试一下. [复制链接]

redor

论坛徽章:
0
11楼 [报告]
发表于 2010-10-09 14:53 |只看该作者
glibc库里面有自己维护内存的方式，不一定是你free它就交给os去了。
davelv 发表于 2010-10-09 11:41



    倒...... 重新用mmap搞了一下, 把新代码也贴出来. 可以通过命令测试 gcc -o vmm mmblock.c -D_DEBUG_MMBLOCK -DHAVE_MMAP -g && ./vmm
mmblock.h

  2. #ifndef _MMBLOCK_H_
  3. #define _MMBLOCK_H_
  4. typedef struct _MMBLOCK
  5. {
  6.            char *data;
  7.            int  ndata;
  8.           int  size;
  9.         int  left;
  10.         char *end;
  11. }MMBLOCK;
  12. #define  MMBLOCK_BASE             4096
  13. //#define  MMBLOCK_BASE         32768
  14. //#define  MMBLOCK_BASE         131072
  15. //#define  MMBLOCK_BASE         524288
  16. #define  MMBLOCK_MIN             1024
  17. #define  MMBLOCK_MAX             262144
  18. //#define  MMBLOCK_MAX             1048576
  19. /* initialize() */
  20. MMBLOCK *mmblock_init();
  21. /* recv() */
  22. int mmblock_recv(MMBLOCK *mmblock, int fd, int flag);
  23. /* read() */
  24. int mmblock_read(MMBLOCK *mmblock, int fd);
  25. /* SSL_read() */
  26. int mmblock_read_SSL(MMBLOCK *mmblock, void *ssl);
  27. /* push() */
  28. int mmblock_push(MMBLOCK *mmblock, char *data, int ndata);
  29. /* del() */
  30. int mmblock_del(MMBLOCK *mmblock, int ndata);
  31. /* reset() */
  32. void mmblock_reset(MMBLOCK *mmblock);
  33. /* clean() */
  34. void mmblock_clean(MMBLOCK *mmblock);
  35. #define MMB(x) ((MMBLOCK *)x)
  36. #define MMB_NDATA(x) ((MMBLOCK *)x)->ndata
  37. #define MMB_SIZE(x) ((MMBLOCK *)x)->size
  38. #define MMB_LEFT(x) ((MMBLOCK *)x)->left
  39. #define MMB_DATA(x) ((MMBLOCK *)x)->data
  40. #define MMB_END(x) ((MMBLOCK *)x)->end
  41. #define MMB_RECV(x, fd, flag) mmblock_recv(MMB(x), fd, flag)
  42. #define MMB_READ(x, fd) mmblock_read(MMB(x), fd)
  43. #define MMB_READ_SSL(x, ssl) mmblock_read_SSL(MMB(x), ssl)
  44. #define MMB_PUSH(x, data, ndata) mmblock_push(MMB(x), data, ndata)
  45. #define MMB_DELETE(x, ndata) mmblock_del(MMB(x), ndata)
  46. #define MMB_RESET(x) mmblock_reset(MMB(x))
  47. #define MMB_CLEAN(x) mmblock_clean(MMB(x))
  48. #endif
复制代码
mmblock.c

  2. #include <stdlib.h>
  3. #include <string.h>
  4. #include <stdio.h>
  5. #include <errno.h>
  6. #ifdef   HAVE_MMAP
  7. #include <sys/mman.h>
  8. #endif
  9. #ifdef  HAVE_SSL
  10. #include "xssl.h"
  11. #endif
  12. #include "mmblock.h"
  13. /* initialize() */
  14. MMBLOCK *mmblock_init()
  15. {
  16.         MMBLOCK *mmblock = NULL;
  17.         return mmblock = (MMBLOCK *)calloc(1, sizeof(MMBLOCK));
  18. }

  20. /* incre() */
  21. int mmblock_incre(MMBLOCK *mmblock, int incre_size)
  22. {
  23.         int size = 0, n = 0;
  24.     char *old = NULL;

  26.         if(mmblock && incre_size > 0)
  27.         {
  28.                 size = mmblock->ndata + incre_size;
  29.                 n = size / MMBLOCK_BASE;
  30.                 if(size % MMBLOCK_BASE) ++n;
  31.                 size = n * MMBLOCK_BASE;
  32.                 //fprintf(stdout, "%s::%d data:%p size:%d\n", __FILE__, __LINE__, mmblock->data, size);
  33. #ifdef         HAVE_MMAP
  34.                 if((old = mmblock->data))
  35.                 {
  36.                         if((mmblock->data = (char *)mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_ANON|MAP_PRIVATE,-1,0)) == (void *)-1)
  37.                 mmblock->data = NULL;
  38.             if(mmblock->data && mmblock->ndata > 0) memcpy(mmblock->data, old, mmblock->ndata);
  39.             munmap(old, mmblock->size);
  40.                 }
  41.                 else
  42.                 {
  43.                         if((mmblock->data = (char *)mmap(NULL, size, PROT_READ|PROT_WRITE, MAP_ANON|MAP_PRIVATE,-1,0)) == (void *)-1)
  44.             {
  45.                 mmblock->data = NULL;
  46.             }
  47.                 }
  48. #else

  50.                 mmblock->data = (char *)realloc(mmblock->data, size);
  51. #endif
  52.                 if(mmblock->data)
  53.                 {
  54.                         mmblock->end = mmblock->data + mmblock->ndata;
  55.                         mmblock->left = size - mmblock->ndata;
  56.                         mmblock->size = size;
  57.                         return 0;
  58.                 }
  59.                 else
  60.                 {
  61.                         mmblock->end = mmblock->data = NULL;
  62.                         mmblock->left = mmblock->ndata = mmblock->size = 0;
  63.                 }
  64.                        
  65.         }
  66.         return -1;
  67. }

  69. /* check() */
  70. int mmblock_check(MMBLOCK *mmblock)
  71. {
  72.         if(mmblock && mmblock->left < MMBLOCK_MIN)
  73.         {
  74.                 return mmblock_incre(mmblock, MMBLOCK_BASE);               
  75.         }
  76.         return -1;
  77. }

  79. /* recv() */
  80. int mmblock_recv(MMBLOCK *mmblock, int fd, int flag)
  81. {
  82.         int n = -1;

  84.         if(mmblock && fd > 0)
  85.         {
  86.                 mmblock_check(mmblock);
  87.                 if(mmblock->data && mmblock->end && mmblock->left > 0
  88.                 && (n = recv(fd, mmblock->end, mmblock->left, flag)) > 0)
  89.                 {
  90.                         mmblock->end += n;
  91.                         mmblock->ndata += n;
  92.                         mmblock->left -= n;               
  93.                 }       
  94.         }
  95.         return n;
  96. }

  98. /* read() */
  99. int mmblock_read(MMBLOCK *mmblock, int fd)
  100. {
  101.         int n = -1;

  103.         if(mmblock && fd > 0)
  104.         {
  105.                 mmblock_check(mmblock);
  106.                 if(mmblock->data && mmblock->end && mmblock->left > 0
  107.                 && (n = read(fd, mmblock->end, mmblock->left)) > 0)
  108.                 {
  109.                         mmblock->ndata += n;
  110.                         mmblock->end += n;
  111.                         mmblock->left -= n;
  112.                 }
  113.         }
  114.         return n;
  115. }

  117. /* SSL_read() */
  118. int mmblock_read_SSL(MMBLOCK *mmblock, void *ssl)
  119. {
  120.         int n = -1;
  121.         if(mmblock && ssl)
  122.         {
  123.                 mmblock_check(mmblock);
  124. #ifdef HAVE_SSL
  125.                 if(mmblock->data && mmblock->end && mmblock->left > 0
  126.                 && (n = SSL_read(XSSL(ssl), mmblock->end, mmblock->left)) > 0)
  127.                 {
  128.                         mmblock->ndata += n;
  129.                         mmblock->end += n;
  130.                         mmblock->left -= n;
  131.                 }
  132. #endif
  133.         }
  134.         return n;
  135. }

  137. /* push() */
  138. int mmblock_push(MMBLOCK *mmblock, char *data, int ndata)
  139. {
  140.         if(mmblock && data && ndata > 0)
  141.         {
  142.                 if(mmblock->left < ndata) mmblock_incre(mmblock, ndata);
  143.                 if(mmblock->left > ndata && mmblock->data && mmblock->end)
  144.                 {
  145.                         memcpy(mmblock->end, data, ndata);
  146.                         mmblock->ndata += ndata;
  147.                         mmblock->end += ndata;
  148.                         mmblock->left -= ndata;
  149.                         return ndata;
  150.                 }
  151.         }
  152.         return -1;
  153. }

  155. /* del() */
  156. int mmblock_del(MMBLOCK *mmblock, int ndata)
  157. {
  158.         char *s = NULL, *p = NULL;

  160.         if(mmblock && ndata > 0)
  161.         {
  162.                 if(mmblock->ndata <= ndata)
  163.                 {
  164.                         mmblock->end = mmblock->data;
  165.                         mmblock->left = mmblock->size;
  166.                         mmblock->ndata = 0;
  167.                 }
  168.                 else
  169.                 {
  170.                         p = mmblock->data;                       
  171.                         s = mmblock->data + ndata;
  172.                         while(s < mmblock->end) *p++ = *s++;
  173.                         mmblock->end = p;
  174.                         mmblock->left += ndata;
  175.                         mmblock->ndata -= ndata;
  176.                 }
  177.                 return 0;
  178.         }
  179.         return -1;
  180. }

  182. /* reset() */
  183. void mmblock_reset(MMBLOCK *mmblock)
  184. {
  185.         if(mmblock)
  186.         {
  187.                 if(mmblock->size > MMBLOCK_MAX)
  188.                 {
  189.                         if(mmblock->data)
  190.             {

  192. #ifdef HAVE_MMAP
  193.                 munmap(mmblock->data, mmblock->size);
  194. #else
  195.                 free(mmblock->data);
  196. #endif
  197.             }
  198.                         mmblock->size = mmblock->ndata = mmblock->left = 0;
  199.                         mmblock->end = mmblock->data  = NULL;
  200.                 }
  201.                 else
  202.                 {
  203.                         mmblock->end = mmblock->data;
  204.                         mmblock->left = mmblock->size;
  205.                         mmblock->ndata = 0;
  206.                 }
  207.         }
  208.         return ;
  209. }

  211. /* clean() */
  212. void mmblock_clean(MMBLOCK *mmblock)
  213. {
  214.         if(mmblock)
  215.         {
  216.                 if(mmblock->data)
  217.         {
  218. #ifdef HAVE_MMAP
  219.             munmap(mmblock->data, mmblock->size);
  220. #else
  221.             free(mmblock->data);
  222. #endif
  223.         }
  224.                 free(mmblock);
  225.         }
  226.         return ;
  227. }


  230. #ifdef _DEBUG_MMBLOCK
  231. #include <fcntl.h>
  232. int main()
  233. {
  234.         char *s = "DSKALFJLSDJFLKASJFKLDAJFKLAKLkljklajfldkfmnklasnfkladsnk";
  235.         MMBLOCK *mmblocks[40960];
  236.         int i = 0, fd = -1, n = strlen(s);
  237.     if((fd = open("/tmp/testfile", O_RDONLY)) > 0)
  238.     {
  239.         //initialize
  240.         for(i = 0; i < 40960; i++)
  241.         {
  242.             if((mmblocks[i] = mmblock_init()))
  243.             {
  244.                 mmblock_push(mmblocks[i], s, n);
  245.                 lseek(fd, 0, SEEK_SET);
  246.                 mmblock_read(mmblocks[i], fd);
  247.             }
  248.             //fprintf(stdout, "last_mmblock[%p]->ndata:%d\n", mmblocks[i], mmblocks[i]->ndata);
  249.         }
  250.         //while(1)sleep(1);
  251.         //clean
  252.         for(i = 0; i < 40960; i++)
  253.         {
  254.             if(mmblocks[i]) mmblock_clean(mmblocks[i]);
  255.         }
  256.         close(fd);
  257.     }
  258.         while(1)sleep(1);
  259.         return 0;
  260. }
  261. //gcc -o vmm mmblock.c -D_DEBUG_MMBLOCK -DHAVE_MMAP -g && ./vmm
  262. #endif
复制代码
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
redor

论坛徽章:
0
12楼 [报告]
发表于 2010-10-09 14:54 |只看该作者
回复 9# cugb_cat

hehe 我重新通过mmap实现了一下. 太搞了 之前没碰到过这个问题
实战分享：从技术角度谈机器学习入门| 【大话IT】RadonDB低门槛向MySQL集群下战书 | ChinaUnix打赏功能已上线！ | 新一代分布式关系型数据库RadonDB知多少？
12
返回列表 发新帖
高级模式
B Color Image Link Quote Code Smilies
您需要登录后才可以回帖 登录 | 注册

本版积分规则 发表回复

  

北京盛拓优讯信息技术有限公司. 版权所有 京ICP备16024965号-6 北京市公安局海淀分局网监中心备案编号：11010802020122 niuxiaotong@pcpop.com 17352615567
未成年举报专区
中国互联网协会会员  联系我们：huangweiwei@itpub.net
感谢所有关心和支持过ChinaUnix的朋友们 转载本站内容请注明原作者名及出处

清除 Cookies - ChinaUnix - Archiver - WAP - TOP