my_malloc.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:2k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #ifdef SAFEMALLOC /* We don't need SAFEMALLOC here */
  14. #undef SAFEMALLOC
  15. #endif
  16. #include "mysys_priv.h"
  17. #include "mysys_err.h"
  18. #include <m_string.h>
  19. /* My memory allocator */
  20. gptr my_malloc(unsigned int size, myf my_flags)
  21. {
  22.   gptr point;
  23.   DBUG_ENTER("my_malloc");
  24.   DBUG_PRINT("my",("size: %u  my_flags: %d",size, my_flags));
  25.   if (!size)
  26.     size=1; /* Safety */
  27.   if ((point = (char*)malloc(size)) == NULL)
  28.   {
  29.     my_errno=errno;
  30.     if (my_flags & MY_FAE)
  31.       error_handler_hook=fatal_error_handler_hook;
  32.     if (my_flags & (MY_FAE+MY_WME))
  33.       my_error(EE_OUTOFMEMORY, MYF(ME_BELL+ME_WAITTANG),size);
  34.     if (my_flags & MY_FAE)
  35.       exit(1);
  36.   }
  37.   else if (my_flags & MY_ZEROFILL)
  38.     bzero(point,size);
  39.   DBUG_PRINT("exit",("ptr: 0x%lx",point));
  40.   DBUG_RETURN(point);
  41. } /* my_malloc */
  42. /* Free memory allocated with my_malloc */
  43. /*ARGSUSED*/
  44. void my_no_flags_free(gptr ptr)
  45. {
  46.   DBUG_ENTER("my_free");
  47.   DBUG_PRINT("my",("ptr: 0x%lx",ptr));
  48.   if (ptr)
  49.     free(ptr);
  50.   DBUG_VOID_RETURN;
  51. } /* my_free */
  52. /* malloc and copy */
  53. gptr my_memdup(const byte *from, uint length, myf my_flags)
  54. {
  55.   gptr ptr;
  56.   if ((ptr=my_malloc(length,my_flags)) != 0)
  57.     memcpy((byte*) ptr, (byte*) from,(size_t) length);
  58.   return(ptr);
  59. }
  60. char *my_strdup(const char *from, myf my_flags)
  61. {
  62.   gptr ptr;
  63.   uint length=(uint) strlen(from)+1;
  64.   if ((ptr=my_malloc(length,my_flags)) != 0)
  65.     memcpy((byte*) ptr, (byte*) from,(size_t) length);
  66.   return((my_string) ptr);
  67. }
  68. char *my_strdup_with_length(const byte *from, uint length, myf my_flags)
  69. {
  70.   gptr ptr;
  71.   if ((ptr=my_malloc(length+1,my_flags)) != 0)
  72.   {
  73.     memcpy((byte*) ptr, (byte*) from,(size_t) length);
  74.     ((char*) ptr)[length]=0;
  75.   }
  76.   return((char*) ptr);
  77. }