• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /***************************************************************************
2  *                                  _   _ ____  _
3  *  Project                     ___| | | |  _ \| |
4  *                             / __| | | | |_) | |
5  *                            | (__| |_| |  _ <| |___
6  *                             \___|\___/|_| \_\_____|
7  *
8  * Copyright (C) 1998 - 2016, 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 
select_wrapper(int nfds,fd_set * rd,fd_set * wr,fd_set * exc,struct timeval * tv)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 
wait_ms(int ms)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
memory_tracking_init(void)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_memdebug(fname);
100     /* this weird stuff here is to make curl_free() get called
101        before curl_memdebug() as otherwise memory tracking will
102        log a free() 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_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 */
hexdump(unsigned char * buffer,size_t len)119 char *hexdump(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     snprintf(p, 4, "%02x ", buffer[i]);
128   return dump;
129 }
130 
131 
main(int argc,char ** argv)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