Home | History | Annotate | Download | only in unit
      1 /***************************************************************************
      2  *                                  _   _ ____  _
      3  *  Project                     ___| | | |  _ \| |
      4  *                             / __| | | | |_) | |
      5  *                            | (__| |_| |  _ <| |___
      6  *                             \___|\___/|_| \_\_____|
      7  *
      8  * Copyright (C) 1998 - 2016, 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 "curlcheck.h"
     23 
     24 #include "urldata.h"
     25 #include "connect.h"
     26 #include "memdebug.h" /* LAST include file */
     27 
     28 static struct Curl_easy *data;
     29 
     30 static CURLcode unit_setup(void)
     31 {
     32   data = curl_easy_init();
     33   if(!data)
     34     return CURLE_OUT_OF_MEMORY;
     35   return CURLE_OK;
     36 }
     37 
     38 static void unit_stop(void)
     39 {
     40   curl_easy_cleanup(data);
     41 }
     42 
     43 /* BASE is just a define to make us fool around with decently large number so
     44    that we aren't zero-based */
     45 #define BASE 1000000
     46 
     47 /* macro to set the pretended current time */
     48 #define NOW(x,y) now.tv_sec = x; now.tv_usec = y
     49 /* macro to set the millisecond based timeouts to use */
     50 #define TIMEOUTS(x,y) data->set.timeout = x; data->set.connecttimeout = y
     51 
     52 /*
     53  * To test:
     54  *
     55  * 00/10/01/11 timeouts set
     56  * 0/1         during connect
     57  * T           various values on the timeouts
     58  * N           various values of now
     59  */
     60 
     61 struct timetest {
     62   int now_s;
     63   int now_us;
     64   int timeout_ms;
     65   int connecttimeout_ms;
     66   bool connecting;
     67   long result;
     68   const char *comment;
     69 };
     70 
     71 UNITTEST_START
     72 {
     73   struct timeval now;
     74   long timeout;
     75   unsigned int i;
     76 
     77   const struct timetest run[] = {
     78   /* both timeouts set, not connecting */
     79   {BASE + 4, 0,      10000, 8000, FALSE, 6000, "6 seconds should be left"},
     80   {BASE + 4, 990000, 10000, 8000, FALSE, 5010, "5010 ms should be left"},
     81   {BASE + 10, 0,     10000, 8000, FALSE, -1,   "timeout is -1, expired"},
     82   {BASE + 12, 0,     10000, 8000, FALSE, -2000, "-2000, overdue 2 seconds"},
     83 
     84   /* both timeouts set, connecting */
     85   {BASE + 4, 0,      10000, 8000, TRUE, 4000, "4 seconds should be left"},
     86   {BASE + 4, 990000, 10000, 8000, TRUE, 3010, "3010 ms should be left"},
     87   {BASE + 8, 0,      10000, 8000, TRUE, -1,   "timeout is -1, expired"},
     88   {BASE + 10, 0,     10000, 8000, TRUE, -2000, "-2000, overdue 2 seconds"},
     89 
     90   /* no connect timeout set, not connecting */
     91   {BASE + 4, 0,      10000, 0, FALSE, 6000, "6 seconds should be left"},
     92   {BASE + 4, 990000, 10000, 0, FALSE, 5010, "5010 ms should be left"},
     93   {BASE + 10, 0,     10000, 0, FALSE, -1,   "timeout is -1, expired"},
     94   {BASE + 12, 0,     10000, 0, FALSE, -2000, "-2000, overdue 2 seconds"},
     95 
     96   /* no connect timeout set, connecting */
     97   {BASE + 4, 0,      10000, 0, FALSE, 6000, "6 seconds should be left"},
     98   {BASE + 4, 990000, 10000, 0, FALSE, 5010, "5010 ms should be left"},
     99   {BASE + 10, 0,     10000, 0, FALSE, -1,   "timeout is -1, expired"},
    100   {BASE + 12, 0,     10000, 0, FALSE, -2000, "-2000, overdue 2 seconds"},
    101 
    102   /* only connect timeout set, not connecting */
    103   {BASE + 4, 0,      0, 10000, FALSE, 0, "no timeout active"},
    104   {BASE + 4, 990000, 0, 10000, FALSE, 0, "no timeout active"},
    105   {BASE + 10, 0,     0, 10000, FALSE, 0, "no timeout active"},
    106   {BASE + 12, 0,     0, 10000, FALSE, 0, "no timeout active"},
    107 
    108   /* only connect timeout set, connecting */
    109   {BASE + 4, 0,      0, 10000, TRUE, 6000, "6 seconds should be left"},
    110   {BASE + 4, 990000, 0, 10000, TRUE, 5010, "5010 ms should be left"},
    111   {BASE + 10, 0,     0, 10000, TRUE, -1,   "timeout is -1, expired"},
    112   {BASE + 12, 0,     0, 10000, TRUE, -2000, "-2000, overdue 2 seconds"},
    113 
    114   /* no timeout set, not connecting */
    115   {BASE + 4, 0,      0, 0, FALSE, 0, "no timeout active"},
    116   {BASE + 4, 990000, 0, 0, FALSE, 0, "no timeout active"},
    117   {BASE + 10, 0,     0, 0, FALSE, 0, "no timeout active"},
    118   {BASE + 12, 0,     0, 0, FALSE, 0, "no timeout active"},
    119 
    120   /* no timeout set, connecting */
    121   {BASE + 4, 0,      0, 0, TRUE, 296000, "no timeout active"},
    122   {BASE + 4, 990000, 0, 0, TRUE, 295010, "no timeout active"},
    123   {BASE + 10, 0,     0, 0, TRUE, 290000, "no timeout active"},
    124   {BASE + 12, 0,     0, 0, TRUE, 288000, "no timeout active"},
    125 
    126   /* both timeouts set, connecting, connect timeout the longer one */
    127   {BASE + 4, 0,      10000, 12000, TRUE, 6000, "6 seconds should be left"},
    128 
    129   };
    130 
    131   /* this is the pretended start time of the transfer */
    132   data->progress.t_startsingle.tv_sec = BASE;
    133   data->progress.t_startsingle.tv_usec = 0;
    134   data->progress.t_startop.tv_sec = BASE;
    135   data->progress.t_startop.tv_usec = 0;
    136 
    137   for(i=0; i < sizeof(run)/sizeof(run[0]); i++) {
    138     NOW(run[i].now_s, run[i].now_us);
    139     TIMEOUTS(run[i].timeout_ms, run[i].connecttimeout_ms);
    140     timeout =  Curl_timeleft(data, &now, run[i].connecting);
    141     if(timeout != run[i].result)
    142       fail(run[i].comment);
    143   }
    144 }
    145 UNITTEST_STOP
    146