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