1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel@haxx.se>, et al.
9 *
10 * This software is licensed as described in the file COPYING, which
11 * you should have received as part of this distribution. The terms
12 * are also available at https://curl.haxx.se/docs/copyright.html.
13 *
14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell
15 * copies of the Software, and permit persons to whom the Software is
16 * furnished to do so, under the terms of the COPYING file.
17 *
18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
19 * KIND, either express or implied.
20 *
21 ***************************************************************************/
22 #include "test.h"
23
24 #include <string.h>
25
26 /*
27 * This test uses these funny custom memory callbacks for the only purpose
28 * of verifying that curl_global_init_mem() functionality is present in
29 * libcurl and that it works unconditionally no matter how libcurl is built,
30 * nothing more.
31 *
32 * Do not include memdebug.h in this source file, and do not use directly
33 * memory related functions in this file except those used inside custom
34 * memory callbacks which should be calling 'the real thing'.
35 */
36
37 /*
38 #include "memdebug.h"
39 */
40
41 static int seen_malloc = 0;
42 static int seen_free = 0;
43 static int seen_realloc = 0;
44 static int seen_strdup = 0;
45 static int seen_calloc = 0;
46
47 void *custom_malloc(size_t size);
48 void custom_free(void *ptr);
49 void *custom_realloc(void *ptr, size_t size);
50 char *custom_strdup(const char *ptr);
51 void *custom_calloc(size_t nmemb, size_t size);
52
53
custom_calloc(size_t nmemb,size_t size)54 void *custom_calloc(size_t nmemb, size_t size)
55 {
56 if(!seen_calloc) {
57 printf("seen custom_calloc()\n");
58 seen_calloc = 1;
59 }
60 return (calloc)(nmemb, size);
61 }
62
custom_malloc(size_t size)63 void *custom_malloc(size_t size)
64 {
65 if(!seen_malloc && seen_calloc) {
66 printf("seen custom_malloc()\n");
67 seen_malloc = 1;
68 }
69 return (malloc)(size);
70 }
71
custom_strdup(const char * ptr)72 char *custom_strdup(const char *ptr)
73 {
74 if(!seen_strdup && seen_malloc) {
75 /* currently (2013.03.13), memory tracking enabled builds do not call
76 the strdup callback, in this case malloc callback and memcpy are used
77 instead. If some day this is changed the following printf() should be
78 uncommented, and a line added to test definition.
79 printf("seen custom_strdup()\n");
80 */
81 seen_strdup = 1;
82 }
83 return (strdup)(ptr);
84 }
85
custom_realloc(void * ptr,size_t size)86 void *custom_realloc(void *ptr, size_t size)
87 {
88 if(!seen_realloc && seen_malloc) {
89 printf("seen custom_realloc()\n");
90 seen_realloc = 1;
91 }
92 return (realloc)(ptr, size);
93 }
94
custom_free(void * ptr)95 void custom_free(void *ptr)
96 {
97 if(!seen_free && seen_realloc) {
98 printf("seen custom_free()\n");
99 seen_free = 1;
100 }
101 (free)(ptr);
102 }
103
104
test(char * URL)105 int test(char *URL)
106 {
107 unsigned char a[] = {0x2f, 0x3a, 0x3b, 0x3c, 0x3d, 0x3e, 0x3f,
108 0x91, 0xa2, 0xb3, 0xc4, 0xd5, 0xe6, 0xf7};
109 CURLcode res;
110 CURL *curl;
111 int asize;
112 char *str = NULL;
113
114 (void)URL;
115
116 res = curl_global_init_mem(CURL_GLOBAL_ALL,
117 custom_malloc,
118 custom_free,
119 custom_realloc,
120 custom_strdup,
121 custom_calloc);
122 if(res != CURLE_OK) {
123 fprintf(stderr, "curl_global_init_mem() failed\n");
124 return TEST_ERR_MAJOR_BAD;
125 }
126
127 curl = curl_easy_init();
128 if(!curl) {
129 fprintf(stderr, "curl_easy_init() failed\n");
130 curl_global_cleanup();
131 return TEST_ERR_MAJOR_BAD;
132 }
133
134 test_setopt(curl, CURLOPT_USERAGENT, "test509"); /* uses strdup() */
135
136 asize = (int)sizeof(a);
137 str = curl_easy_escape(curl, (char *)a, asize); /* uses realloc() */
138
139 test_cleanup:
140
141 if(str)
142 curl_free(str);
143
144 curl_easy_cleanup(curl);
145 curl_global_cleanup();
146
147 return (int)res;
148 }
149