Home | History | Annotate | Download | only in libtest 
      1 /***************************************************************************
      2  *                                  _   _ ____  _
      3  *  Project                     ___| | | |  _ \| |
      4  *                             / __| | | | |_) | |
      5  *                            | (__| |_| |  _ <| |___
      6  *                             \___|\___/|_| \_\_____|
      7  *
      8  * Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel (at) 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_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 */
    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     snprintf(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