1 | /*************************************************************************** |
2 | * _ _ ____ _ |
3 | * Project ___| | | | _ \| | |
4 | * / __| | | | |_) | | |
5 | * | (__| |_| | _ <| |___ |
6 | * \___|\___/|_| \_\_____| |
7 | * |
8 | * Copyright (C) 1998 - 2019, 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 | #ifdef HAVE_LOCALE_H |
25 | # include <locale.h> /* for setlocale() */ |
26 | #endif |
27 | |
28 | #ifdef HAVE_IO_H |
29 | # include <io.h> /* for setmode() */ |
30 | #endif |
31 | |
32 | #ifdef HAVE_FCNTL_H |
33 | # include <fcntl.h> /* for setmode() */ |
34 | #endif |
35 | |
36 | #ifdef USE_NSS |
37 | #include <nspr.h> |
38 | #endif |
39 | |
40 | #ifdef CURLDEBUG |
41 | # define MEMDEBUG_NODEFINES |
42 | # include "memdebug.h" |
43 | #endif |
44 | |
45 | int select_wrapper(int nfds, fd_set *rd, fd_set *wr, fd_set *exc, |
46 | struct timeval *tv) |
47 | { |
48 | if(nfds < 0) { |
49 | SET_SOCKERRNO(EINVAL); |
50 | return -1; |
51 | } |
52 | #ifdef USE_WINSOCK |
53 | /* |
54 | * Winsock select() requires that at least one of the three fd_set |
55 | * pointers is not NULL and points to a non-empty fdset. IOW Winsock |
56 | * select() can not be used to sleep without a single fd_set. |
57 | */ |
58 | if(!nfds) { |
59 | Sleep((1000*tv->tv_sec) + (DWORD)(((double)tv->tv_usec)/1000.0)); |
60 | return 0; |
61 | } |
62 | #endif |
63 | return select(nfds, rd, wr, exc, tv); |
64 | } |
65 | |
66 | void wait_ms(int ms) |
67 | { |
68 | struct timeval t; |
69 | t.tv_sec = ms/1000; |
70 | ms -= (int)t.tv_sec * 1000; |
71 | t.tv_usec = ms * 1000; |
72 | select_wrapper(0, NULL, NULL, NULL, &t); |
73 | } |
74 | |
75 | char *libtest_arg2 = NULL; |
76 | char *libtest_arg3 = NULL; |
77 | int test_argc; |
78 | char **test_argv; |
79 | |
80 | struct timeval tv_test_start; /* for test timing */ |
81 | |
82 | #ifdef UNITTESTS |
83 | int unitfail; /* for unittests */ |
84 | #endif |
85 | |
86 | #ifdef CURLDEBUG |
87 | static void memory_tracking_init(void) |
88 | { |
89 | char *env; |
90 | /* if CURL_MEMDEBUG is set, this starts memory tracking message logging */ |
91 | env = curl_getenv("CURL_MEMDEBUG" ); |
92 | if(env) { |
93 | /* use the value as file name */ |
94 | char fname[CURL_MT_LOGFNAME_BUFSIZE]; |
95 | if(strlen(env) >= CURL_MT_LOGFNAME_BUFSIZE) |
96 | env[CURL_MT_LOGFNAME_BUFSIZE-1] = '\0'; |
97 | strcpy(fname, env); |
98 | curl_free(env); |
99 | curl_dbg_memdebug(fname); |
100 | /* this weird stuff here is to make curl_free() get called before |
101 | curl_dbg_memdebug() as otherwise memory tracking will log a free() |
102 | without an alloc! */ |
103 | } |
104 | /* if CURL_MEMLIMIT is set, this enables fail-on-alloc-number-N feature */ |
105 | env = curl_getenv("CURL_MEMLIMIT" ); |
106 | if(env) { |
107 | char *endptr; |
108 | long num = strtol(env, &endptr, 10); |
109 | if((endptr != env) && (endptr == env + strlen(env)) && (num > 0)) |
110 | curl_dbg_memlimit(num); |
111 | curl_free(env); |
112 | } |
113 | } |
114 | #else |
115 | # define memory_tracking_init() Curl_nop_stmt |
116 | #endif |
117 | |
118 | /* returns a hexdump in a static memory area */ |
119 | char *hexdump(const unsigned char *buffer, size_t len) |
120 | { |
121 | static char dump[200 * 3 + 1]; |
122 | char *p = dump; |
123 | size_t i; |
124 | if(len > 200) |
125 | return NULL; |
126 | for(i = 0; i<len; i++, p += 3) |
127 | msnprintf(p, 4, "%02x " , buffer[i]); |
128 | return dump; |
129 | } |
130 | |
131 | |
132 | int main(int argc, char **argv) |
133 | { |
134 | char *URL; |
135 | int result; |
136 | |
137 | #ifdef O_BINARY |
138 | # ifdef __HIGHC__ |
139 | _setmode(stdout, O_BINARY); |
140 | # else |
141 | setmode(fileno(stdout), O_BINARY); |
142 | # endif |
143 | #endif |
144 | |
145 | memory_tracking_init(); |
146 | |
147 | /* |
148 | * Setup proper locale from environment. This is needed to enable locale- |
149 | * specific behaviour by the C library in order to test for undesired side |
150 | * effects that could cause in libcurl. |
151 | */ |
152 | #ifdef HAVE_SETLOCALE |
153 | setlocale(LC_ALL, "" ); |
154 | #endif |
155 | |
156 | if(argc< 2) { |
157 | fprintf(stderr, "Pass URL as argument please\n" ); |
158 | return 1; |
159 | } |
160 | |
161 | test_argc = argc; |
162 | test_argv = argv; |
163 | |
164 | if(argc>2) |
165 | libtest_arg2 = argv[2]; |
166 | |
167 | if(argc>3) |
168 | libtest_arg3 = argv[3]; |
169 | |
170 | URL = argv[1]; /* provide this to the rest */ |
171 | |
172 | fprintf(stderr, "URL: %s\n" , URL); |
173 | |
174 | result = test(URL); |
175 | |
176 | #ifdef USE_NSS |
177 | if(PR_Initialized()) |
178 | /* prevent valgrind from reporting possibly lost memory (fd cache, ...) */ |
179 | PR_Cleanup(); |
180 | #endif |
181 | |
182 | return result; |
183 | } |
184 | |