← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/thr_malloc.cc

Cleanup around SAFEMALLOC

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/thr_malloc.cc
16
16
 
17
17
/* Mallocs for used in threads */
18
18
 
19
 
#include "config.h"
20
 
 
21
 
#include <string.h>
22
 
 
23
 
#include "drizzled/errmsg_print.h"
24
 
#include "drizzled/memory/sql_alloc.h"
25
 
#include "drizzled/current_session.h"
26
 
#include "drizzled/error.h"
27
 
#include "drizzled/definitions.h"
28
 
 
29
 
#include "drizzled/internal/my_sys.h"
30
 
 
31
 
namespace drizzled
32
 
{
33
 
 
34
 
static void sql_alloc_error_handler(void)
35
 
{
36
 
  errmsg_printf(ERRMSG_LVL_ERROR, "%s",ER(ER_OUT_OF_RESOURCES));
37
 
}
38
 
 
39
 
void memory::init_sql_alloc(memory::Root *mem_root, size_t block_size, size_t)
40
 
{
41
 
  memory::init_alloc_root(mem_root, block_size);
42
 
  mem_root->error_handler= sql_alloc_error_handler;
43
 
}
44
 
 
45
 
 
46
 
void *memory::sql_alloc(size_t Size)
47
 
{
48
 
  memory::Root *root= current_mem_root();
49
 
  return memory::alloc_root(root,Size);
50
 
}
51
 
 
52
 
 
53
 
void *memory::sql_calloc(size_t size)
 
19
#include <drizzled/server_includes.h>
 
20
#include <drizzled/drizzled_error_messages.h>
 
21
 
 
22
extern "C" {
 
23
  void sql_alloc_error_handler(void)
 
24
  {
 
25
    sql_print_error(ER(ER_OUT_OF_RESOURCES));
 
26
  }
 
27
}
 
28
 
 
29
void init_sql_alloc(MEM_ROOT *mem_root, uint block_size, uint pre_alloc)
 
30
{
 
31
  init_alloc_root(mem_root, block_size, pre_alloc);
 
32
  mem_root->error_handler=sql_alloc_error_handler;
 
33
}
 
34
 
 
35
 
 
36
void *sql_alloc(size_t Size)
 
37
{
 
38
  MEM_ROOT *root= *(MEM_ROOT **)pthread_getspecific(THR_MALLOC);
 
39
  return alloc_root(root,Size);
 
40
}
 
41
 
 
42
 
 
43
void *sql_calloc(size_t size)
54
44
{
55
45
  void *ptr;
56
 
  if ((ptr=memory::sql_alloc(size)))
 
46
  if ((ptr=sql_alloc(size)))
57
47
    memset(ptr, 0, size);
58
48
  return ptr;
59
49
}
60
50
 
61
51
 
62
 
char *memory::sql_strdup(const char *str)
 
52
char *sql_strdup(const char *str)
63
53
{
64
54
  size_t len= strlen(str)+1;
65
55
  char *pos;
66
 
  if ((pos= (char*) memory::sql_alloc(len)))
 
56
  if ((pos= (char*) sql_alloc(len)))
67
57
    memcpy(pos,str,len);
68
58
  return pos;
69
59
}
70
60
 
71
61
 
72
 
char *memory::sql_strmake(const char *str, size_t len)
 
62
char *sql_strmake(const char *str, size_t len)
73
63
{
74
64
  char *pos;
75
 
  if ((pos= (char*) memory::sql_alloc(len+1)))
 
65
  if ((pos= (char*) sql_alloc(len+1)))
76
66
  {
77
67
    memcpy(pos,str,len);
78
68
    pos[len]=0;
81
71
}
82
72
 
83
73
 
84
 
void* memory::sql_memdup(const void *ptr, size_t len)
 
74
void* sql_memdup(const void *ptr, size_t len)
85
75
{
86
76
  void *pos;
87
 
  if ((pos= memory::sql_alloc(len)))
 
77
  if ((pos= sql_alloc(len)))
88
78
    memcpy(pos,ptr,len);
89
79
  return pos;
90
80
}
91
81
 
92
 
void *memory::SqlAlloc::operator new(size_t size)
93
 
{
94
 
  return memory::sql_alloc(size);
95
 
}
96
 
 
97
 
void *memory::SqlAlloc::operator new[](size_t size)
98
 
{
99
 
  return memory::sql_alloc(size);
100
 
}
101
 
 
102
 
void *memory::SqlAlloc::operator new[](size_t size, memory::Root *mem_root)
103
 
{
104
 
  return memory::alloc_root(mem_root, size);
105
 
}
106
 
 
107
 
void *memory::SqlAlloc::operator new(size_t size, memory::Root *mem_root)
108
 
{
109
 
  return memory::alloc_root(mem_root, size);
110
 
}
111
 
 
112
 
} /* namespace drizzled */
 
82
void sql_element_free(void *ptr __attribute__((unused)))
 
83
{} /* purecov: deadcode */
 
84
 
 
85
 
 
86
 
 
87
char *sql_strmake_with_convert(const char *str, size_t arg_length,
 
88
                               const CHARSET_INFO * const from_cs,
 
89
                               size_t max_res_length,
 
90
                               const CHARSET_INFO * const to_cs, size_t *result_length)
 
91
{
 
92
  char *pos;
 
93
  size_t new_length= to_cs->mbmaxlen*arg_length;
 
94
  max_res_length--;                             // Reserve place for end null
 
95
 
 
96
  set_if_smaller(new_length, max_res_length);
 
97
  if (!(pos= (char*) sql_alloc(new_length+1)))
 
98
    return pos;                                 // Error
 
99
 
 
100
  if ((from_cs == &my_charset_bin) || (to_cs == &my_charset_bin))
 
101
  {
 
102
    // Safety if to_cs->mbmaxlen > 0
 
103
    new_length= min(arg_length, max_res_length);
 
104
    memcpy(pos, str, new_length);
 
105
  }
 
106
  else
 
107
  {
 
108
    uint dummy_errors;
 
109
    new_length= copy_and_convert((char*) pos, new_length, to_cs, str,
 
110
                                 arg_length, from_cs, &dummy_errors);
 
111
  }
 
112
  pos[new_length]= 0;
 
113
  *result_length= new_length;
 
114
  return pos;
 
115
}
 
116