← Back to branch summary

~drizzle-trunk/drizzle/development

« back to all changes in this revision

Viewing changes to drizzled/memory/sql_alloc.cc

  • Committer: Monty Taylor
  • Date: 2010-10-15 17:18:02 UTC
  • mto: (1859.1.2 build)
  • mto: This revision was merged to the branch mainline in revision 1860.
  • Revision ID: mordred@inaugust.com-20101015171802-qhk6zyfhrkvprr1n
Added support for registering regular sys_var instances via module::Context.

Show diffs side-by-side

added added

removed removed

Lines of Context:
drizzled/memory/sql_alloc.cc
11
11
 
12
12
   You should have received a copy of the GNU General Public License
13
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 */
 
14
   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA */
15
15
 
16
16
 
17
17
/* Mallocs for used in threads */
18
18
 
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)
 
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
  mem_root->init_alloc_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 root->alloc_root(Size);
 
50
}
 
51
 
 
52
 
 
53
void *memory::sql_calloc(size_t size)
44
54
{
45
55
  void *ptr;
46
 
  if ((ptr=sql_alloc(size)))
 
56
 
 
57
  if ((ptr=memory::sql_alloc(size)))
47
58
    memset(ptr, 0, size);
 
59
 
48
60
  return ptr;
49
61
}
50
62
 
51
63
 
52
 
char *sql_strdup(const char *str)
 
64
char *memory::sql_strdup(const char *str)
53
65
{
54
66
  size_t len= strlen(str)+1;
55
67
  char *pos;
56
 
  if ((pos= (char*) sql_alloc(len)))
 
68
  if ((pos= (char*) memory::sql_alloc(len)))
57
69
    memcpy(pos,str,len);
58
70
  return pos;
59
71
}
60
72
 
61
73
 
62
 
char *sql_strmake(const char *str, size_t len)
 
74
char *memory::sql_strmake(const char *str, size_t len)
63
75
{
64
76
  char *pos;
65
 
  if ((pos= (char*) sql_alloc(len+1)))
 
77
  if ((pos= (char*) memory::sql_alloc(len+1)))
66
78
  {
67
79
    memcpy(pos,str,len);
68
80
    pos[len]=0;
71
83
}
72
84
 
73
85
 
74
 
void* sql_memdup(const void *ptr, size_t len)
 
86
void* memory::sql_memdup(const void *ptr, size_t len)
75
87
{
76
88
  void *pos;
77
 
  if ((pos= sql_alloc(len)))
 
89
  if ((pos= memory::sql_alloc(len)))
78
90
    memcpy(pos,ptr,len);
79
91
  return pos;
80
92
}
81
93
 
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
 
 
 
94
void *memory::SqlAlloc::operator new(size_t size)
 
95
{
 
96
  return memory::sql_alloc(size);
 
97
}
 
98
 
 
99
void *memory::SqlAlloc::operator new[](size_t size)
 
100
{
 
101
  return memory::sql_alloc(size);
 
102
}
 
103
 
 
104
void *memory::SqlAlloc::operator new[](size_t size, memory::Root *mem_root)
 
105
{
 
106
  return mem_root->alloc_root(size);
 
107
}
 
108
 
 
109
void *memory::SqlAlloc::operator new(size_t size, memory::Root *mem_root)
 
110
{
 
111
  return mem_root->alloc_root(size);
 
112
}
 
113
 
 
114
} /* namespace drizzled */