my_alloc.c
上传用户:jmzj888
上传日期:2007-01-02
资源大小:220k
文件大小:2k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright Abandoned 1996 TCX DataKonsult AB & Monty Program KB & Detron HB
  2.    This file is public domain and comes with NO WARRANTY of any kind */
  3. /* Routines to handle mallocing of results which will be freed the same time */
  4. #include "mysys_priv.h"
  5. #include <m_string.h>
  6. void init_alloc_root(MEM_ROOT *mem_root,uint block_size)
  7. {
  8.   mem_root->free=mem_root->used=0;
  9.   mem_root->min_malloc=16;
  10.   mem_root->block_size=block_size-MALLOC_OVERHEAD-sizeof(USED_MEM)-8;
  11.   mem_root->error_handler=0;
  12. }
  13. gptr alloc_root(MEM_ROOT *mem_root,unsigned int Size)
  14. {
  15.   uint get_size,max_left;
  16.   gptr point;
  17.   reg1 USED_MEM *next;
  18.   reg2 USED_MEM **prev;
  19.   Size= ALIGN_SIZE(Size);
  20.   prev= &mem_root->free;
  21.   max_left=0;
  22.   for (next= *prev ; next && next->left < Size ; next= next->next)
  23.   {
  24.     if (next->left > max_left)
  25.       max_left=next->left;
  26.     prev= &next->next;
  27.   }
  28.   if (! next)
  29.   { /* Time to alloc new block */
  30.     get_size= Size+ALIGN_SIZE(sizeof(USED_MEM));
  31.     if (max_left*4 < mem_root->block_size && get_size < mem_root->block_size)
  32.       get_size=mem_root->block_size; /* Normal alloc */
  33.     if (!(next = (USED_MEM*) my_malloc(get_size,MYF(MY_WME))))
  34.     {
  35.       if (mem_root->error_handler)
  36. (*mem_root->error_handler)();
  37.       return((gptr) 0); /* purecov: inspected */
  38.     }
  39.     next->next= *prev;
  40.     next->size= get_size;
  41.     next->left= get_size-ALIGN_SIZE(sizeof(USED_MEM));
  42.     *prev=next;
  43.   }
  44.   point= (gptr) ((char*) next+ (next->size-next->left));
  45.   if ((next->left-= Size) < mem_root->min_malloc)
  46.   { /* Full block */
  47.     *prev=next->next; /* Remove block from list */
  48.     next->next=mem_root->used;
  49.     mem_root->used=next;
  50.   }
  51.   return(point);
  52. }
  53. /* deallocate everything used by alloc_root */
  54. void free_root(MEM_ROOT *root)
  55. {
  56.   reg1 USED_MEM *next,*old;
  57.   DBUG_ENTER("free_root");
  58.   if (!root)
  59.     DBUG_VOID_RETURN; /* purecov: inspected */
  60.   for (next= root->used ; next ; )
  61.   {
  62.     old=next; next= next->next ;
  63.     my_free((gptr) old,MYF(0));
  64.   }
  65.   for (next= root->free ; next ; )
  66.   {
  67.     old=next; next= next->next ;
  68.     my_free((gptr) old,MYF(0));
  69.   }
  70.   root->used=root->free=0;
  71.   DBUG_VOID_RETURN;
  72. }
  73. char *strdup_root(MEM_ROOT *root,const char *str)
  74. {
  75.   uint len=strlen(str)+1;
  76.   char *pos;
  77.   if ((pos=alloc_root(root,len)))
  78.     memcpy(pos,str,len);
  79.   return pos;
  80. }
  81. char *memdup_root(MEM_ROOT *root,const char *str,uint len)
  82. {
  83.   char *pos;
  84.   if ((pos=alloc_root(root,len)))
  85.     memcpy(pos,str,len);
  86.   return pos;
  87. }