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

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult 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. /* Mallocs for used in threads */
  14. #include "mysql_priv.h"
  15. extern "C" {
  16.   void sql_alloc_error_handler(void)
  17.   {
  18.     THD *thd=current_thd;
  19.     if (thd) // QQ;  To be removed
  20.       thd->fatal_error(); /* purecov: inspected */
  21.     sql_print_error(ER(ER_OUT_OF_RESOURCES));
  22.   }
  23. }
  24. void init_sql_alloc(MEM_ROOT *mem_root, uint block_size, uint pre_alloc)
  25. {
  26.   init_alloc_root(mem_root, block_size, pre_alloc);
  27.   mem_root->error_handler=sql_alloc_error_handler;
  28. }
  29. gptr sql_alloc(uint Size)
  30. {
  31.   MEM_ROOT *root= *my_pthread_getspecific_ptr(MEM_ROOT**,THR_MALLOC);
  32.   char *ptr= (char*) alloc_root(root,Size);
  33.   return ptr;
  34. }
  35. gptr sql_calloc(uint size)
  36. {
  37.   gptr ptr;
  38.   if ((ptr=sql_alloc(size)))
  39.     bzero((char*) ptr,size);
  40.   return ptr;
  41. }
  42. char *sql_strdup(const char *str)
  43. {
  44.   uint len=(uint) strlen(str)+1;
  45.   char *pos;
  46.   if ((pos= (char*) sql_alloc(len)))
  47.     memcpy(pos,str,len);
  48.   return pos;
  49. }
  50. char *sql_strmake(const char *str,uint len)
  51. {
  52.   char *pos;
  53.   if ((pos= (char*) sql_alloc(len+1)))
  54.   {
  55.     memcpy(pos,str,len);
  56.     pos[len]=0;
  57.   }
  58.   return pos;
  59. }
  60. gptr sql_memdup(const void *ptr,uint len)
  61. {
  62.   char *pos;
  63.   if ((pos= (char*) sql_alloc(len)))
  64.     memcpy(pos,ptr,len);
  65.   return pos;
  66. }
  67. void sql_element_free(void *ptr __attribute__((unused)))
  68. {} /* purecov: deadcode */
  69. char *sql_strmake_with_convert(const char *str, uint32 arg_length,
  70.        CHARSET_INFO *from_cs,
  71.        uint32 max_res_length,
  72.        CHARSET_INFO *to_cs, uint32 *result_length)
  73. {
  74.   char *pos;
  75.   uint32 new_length= to_cs->mbmaxlen*arg_length;
  76.   max_res_length--; // Reserve place for end null
  77.   set_if_smaller(new_length, max_res_length);
  78.   if (!(pos= sql_alloc(new_length+1)))
  79.     return pos; // Error
  80.   if ((from_cs == &my_charset_bin) || (to_cs == &my_charset_bin))
  81.   {
  82.     // Safety if to_cs->mbmaxlen > 0
  83.     new_length= min(arg_length, max_res_length);
  84.     memcpy(pos, str, new_length);
  85.   }
  86.   else
  87.   {
  88.     uint dummy_errors;
  89.     new_length= copy_and_convert((char*) pos, new_length, to_cs, str,
  90.  arg_length, from_cs, &dummy_errors);
  91.   }
  92.   pos[new_length]= 0;
  93.   *result_length= new_length;
  94.   return pos;
  95. }