← Back to branch summary

~linuxjedi/drizzle/trunk-bug-667053

« back to all changes in this revision

Viewing changes to sql/thr_malloc.cc

  • Committer: brian
  • Date: 2008-06-25 05:29:13 UTC
  • Revision ID: brian@localhost.localdomain-20080625052913-6upwo0jsrl4lnapl
clean slate

Show diffs side-by-side

added added

removed removed

Lines of Context:
sql/thr_malloc.cc
 
1
/* Copyright (C) 2000-2001, 2003-2004 MySQL AB
 
2
 
 
3
   This program is free software; you can redistribute it and/or modify
 
4
   it under the terms of the GNU General Public License as published by
 
5
   the Free Software Foundation; version 2 of the License.
 
6
 
 
7
   This program is distributed in the hope that it will be useful,
 
8
   but WITHOUT ANY WARRANTY; without even the implied warranty of
 
9
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
10
   GNU General Public License for more details.
 
11
 
 
12
   You should have received a copy of the GNU General Public License
 
13
   along with this program; if not, write to the Free Software
 
14
   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
 
15
 
 
16
 
 
17
/* Mallocs for used in threads */
 
18
 
 
19
#include "mysql_priv.h"
 
20
 
 
21
extern "C" {
 
22
  void sql_alloc_error_handler(void)
 
23
  {
 
24
    sql_print_error(ER(ER_OUT_OF_RESOURCES));
 
25
  }
 
26
}
 
27
 
 
28
void init_sql_alloc(MEM_ROOT *mem_root, uint block_size, uint pre_alloc)
 
29
{
 
30
  init_alloc_root(mem_root, block_size, pre_alloc);
 
31
  mem_root->error_handler=sql_alloc_error_handler;
 
32
}
 
33
 
 
34
 
 
35
void *sql_alloc(size_t Size)
 
36
{
 
37
  MEM_ROOT *root= *my_pthread_getspecific_ptr(MEM_ROOT**,THR_MALLOC);
 
38
  return alloc_root(root,Size);
 
39
}
 
40
 
 
41
 
 
42
void *sql_calloc(size_t size)
 
43
{
 
44
  void *ptr;
 
45
  if ((ptr=sql_alloc(size)))
 
46
    bzero(ptr,size);
 
47
  return ptr;
 
48
}
 
49
 
 
50
 
 
51
char *sql_strdup(const char *str)
 
52
{
 
53
  size_t len= strlen(str)+1;
 
54
  char *pos;
 
55
  if ((pos= (char*) sql_alloc(len)))
 
56
    memcpy(pos,str,len);
 
57
  return pos;
 
58
}
 
59
 
 
60
 
 
61
char *sql_strmake(const char *str, size_t len)
 
62
{
 
63
  char *pos;
 
64
  if ((pos= (char*) sql_alloc(len+1)))
 
65
  {
 
66
    memcpy(pos,str,len);
 
67
    pos[len]=0;
 
68
  }
 
69
  return pos;
 
70
}
 
71
 
 
72
 
 
73
void* sql_memdup(const void *ptr, size_t len)
 
74
{
 
75
  void *pos;
 
76
  if ((pos= sql_alloc(len)))
 
77
    memcpy(pos,ptr,len);
 
78
  return pos;
 
79
}
 
80
 
 
81
void sql_element_free(void *ptr __attribute__((unused)))
 
82
{} /* purecov: deadcode */
 
83
 
 
84
 
 
85
 
 
86
char *sql_strmake_with_convert(const char *str, size_t arg_length,
 
87
                               CHARSET_INFO *from_cs,
 
88
                               size_t max_res_length,
 
89
                               CHARSET_INFO *to_cs, size_t *result_length)
 
90
{
 
91
  char *pos;
 
92
  size_t new_length= to_cs->mbmaxlen*arg_length;
 
93
  max_res_length--;                             // Reserve place for end null
 
94
 
 
95
  set_if_smaller(new_length, max_res_length);
 
96
  if (!(pos= (char*) sql_alloc(new_length+1)))
 
97
    return pos;                                 // Error
 
98
 
 
99
  if ((from_cs == &my_charset_bin) || (to_cs == &my_charset_bin))
 
100
  {
 
101
    // Safety if to_cs->mbmaxlen > 0
 
102
    new_length= min(arg_length, max_res_length);
 
103
    memcpy(pos, str, new_length);
 
104
  }
 
105
  else
 
106
  {
 
107
    uint dummy_errors;
 
108
    new_length= copy_and_convert((char*) pos, new_length, to_cs, str,
 
109
                                 arg_length, from_cs, &dummy_errors);
 
110
  }
 
111
  pos[new_length]= 0;
 
112
  *result_length= new_length;
 
113
  return pos;
 
114
}
 
115