xref: /php-src/ext/mysqlnd/mysqlnd_alloc.h (revision a893a490)
1 /*
2   +----------------------------------------------------------------------+
3   | Copyright (c) The PHP Group                                          |
4   +----------------------------------------------------------------------+
5   | This source file is subject to version 3.01 of the PHP license,      |
6   | that is bundled with this package in the file LICENSE, and is        |
7   | available through the world-wide-web at the following url:           |
8   | https://www.php.net/license/3_01.txt                                 |
9   | If you did not receive a copy of the PHP license and are unable to   |
10   | obtain it through the world-wide-web, please send a note to          |
11   | license@php.net so we can mail you a copy immediately.               |
12   +----------------------------------------------------------------------+
13   | Authors: Andrey Hristov <andrey@php.net>                             |
14   |          Ulf Wendel <uw@php.net>                                     |
15   +----------------------------------------------------------------------+
16 */
17 
18 #ifndef MYSQLND_ALLOC_H
19 #define MYSQLND_ALLOC_H
20 
21 PHPAPI extern const char * mysqlnd_debug_std_no_trace_funcs[];
22 
23 #define MYSQLND_MEM_D	ZEND_FILE_LINE_DC
24 #define MYSQLND_MEM_C	ZEND_FILE_LINE_CC
25 
26 struct st_mysqlnd_allocator_methods
27 {
28 	void *	(*m_emalloc)(size_t size MYSQLND_MEM_D);
29 	void *	(*m_pemalloc)(size_t size, bool persistent MYSQLND_MEM_D);
30 	void *	(*m_ecalloc)(unsigned int nmemb, size_t size MYSQLND_MEM_D);
31 	void *	(*m_pecalloc)(unsigned int nmemb, size_t size, bool persistent MYSQLND_MEM_D);
32 	void *	(*m_erealloc)(void *ptr, size_t new_size MYSQLND_MEM_D);
33 	void *	(*m_perealloc)(void *ptr, size_t new_size, bool persistent MYSQLND_MEM_D);
34 	void	(*m_efree)(void *ptr MYSQLND_MEM_D);
35 	void	(*m_pefree)(void *ptr, bool persistent MYSQLND_MEM_D);
36 	char *	(*m_pememdup)(const char * const ptr, size_t size, bool persistent MYSQLND_MEM_D);
37 	char *	(*m_pestrndup)(const char * const ptr, size_t size, bool persistent MYSQLND_MEM_D);
38 	char *	(*m_pestrdup)(const char * const ptr, bool persistent MYSQLND_MEM_D);
39 };
40 
41 PHPAPI extern struct st_mysqlnd_allocator_methods mysqlnd_allocator;
42 
43 #define mnd_emalloc(size)				mysqlnd_allocator.m_emalloc((size) MYSQLND_MEM_C)
44 #define mnd_pemalloc(size, pers)		mysqlnd_allocator.m_pemalloc((size), (pers) MYSQLND_MEM_C)
45 #define mnd_ecalloc(nmemb, size)		mysqlnd_allocator.m_ecalloc((nmemb), (size) MYSQLND_MEM_C)
46 #define mnd_pecalloc(nmemb, size, p)	mysqlnd_allocator.m_pecalloc((nmemb), (size), (p) MYSQLND_MEM_C)
47 #define mnd_erealloc(ptr, new_size)		mysqlnd_allocator.m_erealloc((ptr), (new_size) MYSQLND_MEM_C)
48 #define mnd_perealloc(ptr, new_size, p)	mysqlnd_allocator.m_perealloc((ptr), (new_size), (p) MYSQLND_MEM_C)
49 #define mnd_efree(ptr)					mysqlnd_allocator.m_efree((ptr) MYSQLND_MEM_C)
50 #define mnd_pefree(ptr, pers)			mysqlnd_allocator.m_pefree((ptr), (pers) MYSQLND_MEM_C)
51 #define mnd_pememdup(ptr, size, pers)	mysqlnd_allocator.m_pememdup((ptr), (size), (pers) MYSQLND_MEM_C)
52 #define mnd_pestrndup(ptr, size, pers)	mysqlnd_allocator.m_pestrndup((ptr), (size), (pers) MYSQLND_MEM_C)
53 #define mnd_pestrdup(ptr, pers)			mysqlnd_allocator.m_pestrdup((ptr), (pers) MYSQLND_MEM_C)
54 #define mnd_sprintf(p, mx_len, fmt,...) spprintf((p), (mx_len), (fmt), __VA_ARGS__)
55 #define mnd_vsprintf(p, mx_len, fmt,ap) vspprintf((p), (mx_len), (fmt), (ap))
56 #define mnd_sprintf_free(p)				efree((p))
57 
mnd_dup_cstring(const MYSQLND_CSTRING str,const bool persistent)58 static inline MYSQLND_STRING mnd_dup_cstring(const MYSQLND_CSTRING str, const bool persistent)
59 {
60 	const MYSQLND_STRING ret = {(char*) mnd_pemalloc(str.l + 1, persistent), str.l};
61 	if (ret.s) {
62 		memcpy(ret.s, str.s, str.l);
63 		ret.s[str.l] = '\0';
64 	}
65 	return ret;
66 }
67 
mnd_str2c(const MYSQLND_STRING str)68 static inline MYSQLND_CSTRING mnd_str2c(const MYSQLND_STRING str)
69 {
70 	const MYSQLND_CSTRING ret = {str.s, str.l};
71 	return ret;
72 }
73 
mysqlnd_set_string(MYSQLND_STRING * buf,const char * string,size_t len)74 static inline void mysqlnd_set_string(MYSQLND_STRING *buf, const char *string, size_t len) {
75 	if (buf->s) {
76 		mnd_efree(buf->s);
77 		buf->s = NULL;
78 		buf->l = 0;
79 	}
80 	if (string) {
81 		buf->s = mnd_pestrndup(string, len, 0);
82 		buf->l = len;
83 	}
84 }
85 
mysqlnd_set_persistent_string(MYSQLND_STRING * buf,const char * string,size_t len,bool persistent)86 static inline void mysqlnd_set_persistent_string(MYSQLND_STRING *buf, const char *string, size_t len, bool persistent) {
87 	if (buf->s) {
88 		mnd_pefree(buf->s, persistent);
89 		buf->s = NULL;
90 		buf->l = 0;
91 	}
92 	if (string) {
93 		buf->s = mnd_pestrndup(string, len, persistent);
94 		buf->l = len;
95 	}
96 }
97 
98 #endif /* MYSQLND_ALLOC_H */
99