thr_malloc.cpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:3k
- /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
- This program is free software; you can redistribute it and/or modify
- it under the terms of the GNU General Public License as published by
- the Free Software Foundation; either version 2 of the License, or
- (at your option) any later version.
- This program is distributed in the hope that it will be useful,
- but WITHOUT ANY WARRANTY; without even the implied warranty of
- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- GNU General Public License for more details.
- You should have received a copy of the GNU General Public License
- along with this program; if not, write to the Free Software
- Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */
- /* Mallocs for used in threads */
- #include "mysql_priv.h"
- extern "C" {
- void sql_alloc_error_handler(void)
- {
- THD *thd=current_thd;
- if (thd) // QQ; To be removed
- thd->fatal_error(); /* purecov: inspected */
- sql_print_error(ER(ER_OUT_OF_RESOURCES));
- }
- }
- void init_sql_alloc(MEM_ROOT *mem_root, uint block_size, uint pre_alloc)
- {
- init_alloc_root(mem_root, block_size, pre_alloc);
- mem_root->error_handler=sql_alloc_error_handler;
- }
- gptr sql_alloc(uint Size)
- {
- MEM_ROOT *root= *my_pthread_getspecific_ptr(MEM_ROOT**,THR_MALLOC);
- char *ptr= (char*) alloc_root(root,Size);
- return ptr;
- }
- gptr sql_calloc(uint size)
- {
- gptr ptr;
- if ((ptr=sql_alloc(size)))
- bzero((char*) ptr,size);
- return ptr;
- }
- char *sql_strdup(const char *str)
- {
- uint len=(uint) strlen(str)+1;
- char *pos;
- if ((pos= (char*) sql_alloc(len)))
- memcpy(pos,str,len);
- return pos;
- }
- char *sql_strmake(const char *str,uint len)
- {
- char *pos;
- if ((pos= (char*) sql_alloc(len+1)))
- {
- memcpy(pos,str,len);
- pos[len]=0;
- }
- return pos;
- }
- gptr sql_memdup(const void *ptr,uint len)
- {
- char *pos;
- if ((pos= (char*) sql_alloc(len)))
- memcpy(pos,ptr,len);
- return pos;
- }
- void sql_element_free(void *ptr __attribute__((unused)))
- {} /* purecov: deadcode */
- char *sql_strmake_with_convert(const char *str, uint32 arg_length,
- CHARSET_INFO *from_cs,
- uint32 max_res_length,
- CHARSET_INFO *to_cs, uint32 *result_length)
- {
- char *pos;
- uint32 new_length= to_cs->mbmaxlen*arg_length;
- max_res_length--; // Reserve place for end null
- set_if_smaller(new_length, max_res_length);
- if (!(pos= sql_alloc(new_length+1)))
- return pos; // Error
- if ((from_cs == &my_charset_bin) || (to_cs == &my_charset_bin))
- {
- // Safety if to_cs->mbmaxlen > 0
- new_length= min(arg_length, max_res_length);
- memcpy(pos, str, new_length);
- }
- else
- {
- uint dummy_errors;
- new_length= copy_and_convert((char*) pos, new_length, to_cs, str,
- arg_length, from_cs, &dummy_errors);
- }
- pos[new_length]= 0;
- *result_length= new_length;
- return pos;
- }