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