113 lines
2.6 KiB
C++
113 lines
2.6 KiB
C++
/* Copyright (c) 2000, 2015, Oracle and/or its affiliates. All rights reserved.
|
|
|
|
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; version 2 of the License.
|
|
|
|
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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA */
|
|
|
|
|
|
/* Mallocs for used in threads */
|
|
|
|
#include "thr_malloc.h"
|
|
#include "sql_class.h"
|
|
|
|
#include <algorithm>
|
|
|
|
using std::min;
|
|
using std::max;
|
|
|
|
extern "C" void sql_alloc_error_handler(void);
|
|
|
|
void init_sql_alloc(PSI_memory_key key,
|
|
MEM_ROOT *mem_root, size_t block_size, size_t pre_alloc)
|
|
{
|
|
init_alloc_root(key, mem_root, block_size, pre_alloc);
|
|
mem_root->error_handler=sql_alloc_error_handler;
|
|
}
|
|
|
|
void *sql_alloc(size_t Size)
|
|
{
|
|
MEM_ROOT *root= *my_thread_get_THR_MALLOC();
|
|
return alloc_root(root,Size);
|
|
}
|
|
|
|
|
|
void *sql_calloc(size_t size)
|
|
{
|
|
void *ptr;
|
|
if ((ptr=sql_alloc(size)))
|
|
memset(ptr, 0, size);
|
|
return ptr;
|
|
}
|
|
|
|
|
|
char *sql_strdup(const char *str)
|
|
{
|
|
size_t len= strlen(str)+1;
|
|
char *pos;
|
|
if ((pos= (char*) sql_alloc(len)))
|
|
memcpy(pos,str,len);
|
|
return pos;
|
|
}
|
|
|
|
|
|
char *sql_strmake(const char *str, size_t len)
|
|
{
|
|
char *pos;
|
|
if ((pos= (char*) sql_alloc(len+1)))
|
|
{
|
|
memcpy(pos,str,len);
|
|
pos[len]=0;
|
|
}
|
|
return pos;
|
|
}
|
|
|
|
|
|
void* sql_memdup(const void *ptr, size_t len)
|
|
{
|
|
void *pos;
|
|
if ((pos= sql_alloc(len)))
|
|
memcpy(pos,ptr,len);
|
|
return pos;
|
|
}
|
|
|
|
|
|
char *sql_strmake_with_convert(const char *str, size_t arg_length,
|
|
const CHARSET_INFO *from_cs,
|
|
size_t max_res_length,
|
|
const CHARSET_INFO *to_cs, size_t *result_length)
|
|
{
|
|
char *pos;
|
|
size_t 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= (char*) 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(pos, new_length, to_cs, str,
|
|
arg_length, from_cs, &dummy_errors);
|
|
}
|
|
pos[new_length]= 0;
|
|
*result_length= new_length;
|
|
return pos;
|
|
}
|
|
|