1/***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 2012 - 2020, 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 "test.h"
23
24#include <limits.h>
25#include <assert.h>
26
27#include "testutil.h"
28#include "warnless.h"
29#include "memdebug.h"
30
31#define TEST_HANG_TIMEOUT 5 * 1000
32#define MAX_EASY_HANDLES 3
33
34static int counter[MAX_EASY_HANDLES];
35static CURL *easy[MAX_EASY_HANDLES];
36static curl_socket_t sockets[MAX_EASY_HANDLES];
37static int res = 0;
38
39static size_t callback(char *ptr, size_t size, size_t nmemb, void *data)
40{
41 ssize_t idx = ((CURL **) data) - easy;
42 curl_socket_t sock;
43 long longdata;
44 CURLcode code;
45 const size_t failure = (size && nmemb) ? 0 : 1;
46 (void)ptr;
47
48 counter[idx] += (int)(size * nmemb);
49
50 /* Get socket being used for this easy handle, otherwise CURL_SOCKET_BAD */
51 code = curl_easy_getinfo(easy[idx], CURLINFO_LASTSOCKET, &longdata);
52 if(CURLE_OK != code) {
53 fprintf(stderr, "%s:%d curl_easy_getinfo() failed, "
54 "with code %d (%s)\n",
55 __FILE__, __LINE__, (int)code, curl_easy_strerror(code));
56 res = TEST_ERR_MAJOR_BAD;
57 return failure;
58 }
59 if(longdata == -1L)
60 sock = CURL_SOCKET_BAD;
61 else
62 sock = (curl_socket_t)longdata;
63
64 if(sock != CURL_SOCKET_BAD) {
65 /* Track relationship between this easy handle and the socket. */
66 if(sockets[idx] == CURL_SOCKET_BAD) {
67 /* An easy handle without previous socket, record the socket. */
68 sockets[idx] = sock;
69 }
70 else if(sock != sockets[idx]) {
71 /* An easy handle with a socket different to previously
72 tracked one, log and fail right away. Known bug #37. */
73 fprintf(stderr, "Handle %d started on socket %d and moved to %d\n",
74 curlx_sztosi(idx), (int)sockets[idx], (int)sock);
75 res = TEST_ERR_MAJOR_BAD;
76 return failure;
77 }
78 }
79 return size * nmemb;
80}
81
82enum HandleState {
83 ReadyForNewHandle,
84 NeedSocketForNewHandle,
85 NoMoreHandles
86};
87
88int test(char *url)
89{
90 CURLM *multi = NULL;
91 int running;
92 int i;
93 int num_handles = 0;
94 enum HandleState state = ReadyForNewHandle;
95 size_t urllen = strlen(url) + 4 + 1;
96 char *full_url = malloc(urllen);
97
98 start_test_timing();
99
100 if(!full_url) {
101 fprintf(stderr, "Not enough memory for full url\n");
102 return TEST_ERR_MAJOR_BAD;
103 }
104
105 for(i = 0; i < MAX_EASY_HANDLES; ++i) {
106 easy[i] = NULL;
107 sockets[i] = CURL_SOCKET_BAD;
108 }
109
110 res_global_init(CURL_GLOBAL_ALL);
111 if(res) {
112 free(full_url);
113 return res;
114 }
115
116 multi_init(multi);
117
118#ifdef USE_PIPELINING
119 multi_setopt(multi, CURLMOPT_PIPELINING, 1L);
120 multi_setopt(multi, CURLMOPT_MAX_HOST_CONNECTIONS, 5L);
121 multi_setopt(multi, CURLMOPT_MAX_TOTAL_CONNECTIONS, 10L);
122#endif
123
124 for(;;) {
125 struct timeval interval;
126 fd_set fdread;
127 fd_set fdwrite;
128 fd_set fdexcep;
129 long timeout = -99;
130 int maxfd = -99;
131 bool found_new_socket = FALSE;
132
133 /* Start a new handle if we aren't at the max */
134 if(state == ReadyForNewHandle) {
135 easy_init(easy[num_handles]);
136
137 if(num_handles % 3 == 2) {
138 msnprintf(full_url, urllen, "%s0200", url);
139 easy_setopt(easy[num_handles], CURLOPT_HTTPAUTH, CURLAUTH_NTLM);
140 }
141 else {
142 msnprintf(full_url, urllen, "%s0100", url);
143 easy_setopt(easy[num_handles], CURLOPT_HTTPAUTH, CURLAUTH_BASIC);
144 }
145 easy_setopt(easy[num_handles], CURLOPT_FRESH_CONNECT, 1L);
146 easy_setopt(easy[num_handles], CURLOPT_URL, full_url);
147 easy_setopt(easy[num_handles], CURLOPT_VERBOSE, 1L);
148 easy_setopt(easy[num_handles], CURLOPT_HTTPGET, 1L);
149 easy_setopt(easy[num_handles], CURLOPT_USERPWD, "testuser:testpass");
150 easy_setopt(easy[num_handles], CURLOPT_WRITEFUNCTION, callback);
151 easy_setopt(easy[num_handles], CURLOPT_WRITEDATA, easy + num_handles);
152 easy_setopt(easy[num_handles], CURLOPT_HEADER, 1L);
153
154 multi_add_handle(multi, easy[num_handles]);
155 num_handles += 1;
156 state = NeedSocketForNewHandle;
157 }
158
159 multi_perform(multi, &running);
160
161 fprintf(stderr, "%s:%d running %d state %d\n",
162 __FILE__, __LINE__, running, state);
163
164 abort_on_test_timeout();
165
166 if(!running && state == NoMoreHandles)
167 break; /* done */
168
169 FD_ZERO(&fdread);
170 FD_ZERO(&fdwrite);
171 FD_ZERO(&fdexcep);
172
173 multi_fdset(multi, &fdread, &fdwrite, &fdexcep, &maxfd);
174
175 /* At this point, maxfd is guaranteed to be greater or equal than -1. */
176
177 if(state == NeedSocketForNewHandle) {
178 if(maxfd != -1 && !found_new_socket) {
179 fprintf(stderr, "Warning: socket did not open immediately for new "
180 "handle (trying again)\n");
181 continue;
182 }
183 state = num_handles < MAX_EASY_HANDLES ? ReadyForNewHandle
184 : NoMoreHandles;
185 fprintf(stderr, "%s:%d new state %d\n",
186 __FILE__, __LINE__, state);
187 }
188
189 multi_timeout(multi, &timeout);
190
191 /* At this point, timeout is guaranteed to be greater or equal than -1. */
192
193 fprintf(stderr, "%s:%d num_handles %d timeout %ld running %d\n",
194 __FILE__, __LINE__, num_handles, timeout, running);
195
196 if(timeout != -1L) {
197 int itimeout = (timeout > (long)INT_MAX) ? INT_MAX : (int)timeout;
198 interval.tv_sec = itimeout/1000;
199 interval.tv_usec = (itimeout%1000)*1000;
200 }
201 else {
202 interval.tv_sec = 0;
203 interval.tv_usec = 5000;
204
205 /* if there's no timeout and we get here on the last handle, we may
206 already have read the last part of the stream so waiting makes no
207 sense */
208 if(!running && num_handles == MAX_EASY_HANDLES) {
209 break;
210 }
211 }
212
213 select_test(maxfd + 1, &fdread, &fdwrite, &fdexcep, &interval);
214
215 abort_on_test_timeout();
216 }
217
218test_cleanup:
219
220 /* proper cleanup sequence - type PB */
221
222 for(i = 0; i < MAX_EASY_HANDLES; i++) {
223 printf("Data connection %d: %d\n", i, counter[i]);
224 curl_multi_remove_handle(multi, easy[i]);
225 curl_easy_cleanup(easy[i]);
226 }
227
228 curl_multi_cleanup(multi);
229 curl_global_cleanup();
230
231 free(full_url);
232
233 return res;
234}
235