1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2016, 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.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
23 /*
24 * This test sends data with CURLOPT_KEEP_SENDING_ON_ERROR.
25 * The server responds with an early error response.
26 * The test is successful if the connection can be reused for the next request,
27 * because this implies that the data has been sent completely to the server.
28 */
29
30 #include "test.h"
31
32 #include "memdebug.h"
33
34 struct cb_data {
35 CURL *easy_handle;
36 int response_received;
37 int paused;
38 size_t remaining_bytes;
39 };
40
41
reset_data(struct cb_data * data,CURL * curl)42 static void reset_data(struct cb_data *data, CURL *curl)
43 {
44 data->easy_handle = curl;
45 data->response_received = 0;
46 data->paused = 0;
47 data->remaining_bytes = 3;
48 }
49
50
read_callback(void * ptr,size_t size,size_t nitems,void * userdata)51 static size_t read_callback(void *ptr, size_t size, size_t nitems,
52 void *userdata)
53 {
54 struct cb_data *data = (struct cb_data *)userdata;
55
56 /* wait until the server has sent all response headers */
57 if(data->response_received) {
58 size_t totalsize = nitems * size;
59
60 size_t bytes_to_send = data->remaining_bytes;
61 if(bytes_to_send > totalsize) {
62 bytes_to_send = totalsize;
63 }
64
65 memset(ptr, 'a', bytes_to_send);
66 data->remaining_bytes -= bytes_to_send;
67
68 return bytes_to_send;
69 }
70 else {
71 data->paused = 1;
72 return CURL_READFUNC_PAUSE;
73 }
74 }
75
76
write_callback(char * ptr,size_t size,size_t nmemb,void * userdata)77 static size_t write_callback(char *ptr, size_t size, size_t nmemb,
78 void *userdata)
79 {
80 struct cb_data *data = (struct cb_data *)userdata;
81 size_t totalsize = nmemb * size;
82
83 /* unused parameter */
84 (void)ptr;
85
86 /* all response headers have been received */
87 data->response_received = 1;
88
89 if(data->paused) {
90 /* continue to send request body data */
91 data->paused = 0;
92 curl_easy_pause(data->easy_handle, CURLPAUSE_CONT);
93 }
94
95 return totalsize;
96 }
97
98
perform_and_check_connections(CURL * curl,const char * description,long expected_connections)99 static int perform_and_check_connections(CURL *curl, const char *description,
100 long expected_connections)
101 {
102 CURLcode res;
103 long connections = 0;
104
105 res = curl_easy_perform(curl);
106 if(res != CURLE_OK) {
107 fprintf(stderr, "curl_easy_perform() failed\n");
108 return TEST_ERR_MAJOR_BAD;
109 }
110
111 res = curl_easy_getinfo(curl, CURLINFO_NUM_CONNECTS, &connections);
112 if(res != CURLE_OK) {
113 fprintf(stderr, "curl_easy_getinfo() failed\n");
114 return TEST_ERR_MAJOR_BAD;
115 }
116
117 fprintf(stderr, "%s: expected: %ld connections; actual: %ld connections\n",
118 description, expected_connections, connections);
119
120 if(connections != expected_connections) {
121 return TEST_ERR_FAILURE;
122 }
123
124 return TEST_ERR_SUCCESS;
125 }
126
127
test(char * URL)128 int test(char *URL)
129 {
130 struct cb_data data;
131 CURL *curl = NULL;
132 CURLcode res = CURLE_FAILED_INIT;
133
134 if(curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
135 fprintf(stderr, "curl_global_init() failed\n");
136 return TEST_ERR_MAJOR_BAD;
137 }
138
139 curl = curl_easy_init();
140 if(curl == NULL) {
141 fprintf(stderr, "curl_easy_init() failed\n");
142 curl_global_cleanup();
143 return TEST_ERR_MAJOR_BAD;
144 }
145
146 reset_data(&data, curl);
147
148 test_setopt(curl, CURLOPT_URL, URL);
149 test_setopt(curl, CURLOPT_POST, 1L);
150 test_setopt(curl, CURLOPT_POSTFIELDSIZE_LARGE,
151 (curl_off_t)data.remaining_bytes);
152 test_setopt(curl, CURLOPT_VERBOSE, 1L);
153 test_setopt(curl, CURLOPT_READFUNCTION, read_callback);
154 test_setopt(curl, CURLOPT_READDATA, &data);
155 test_setopt(curl, CURLOPT_WRITEFUNCTION, write_callback);
156 test_setopt(curl, CURLOPT_WRITEDATA, &data);
157
158 res = perform_and_check_connections(curl,
159 "First request without CURLOPT_KEEP_SENDING_ON_ERROR", 1);
160 if(res != TEST_ERR_SUCCESS) {
161 goto test_cleanup;
162 }
163
164 reset_data(&data, curl);
165
166 res = perform_and_check_connections(curl,
167 "Second request without CURLOPT_KEEP_SENDING_ON_ERROR", 1);
168 if(res != TEST_ERR_SUCCESS) {
169 goto test_cleanup;
170 }
171
172 test_setopt(curl, CURLOPT_KEEP_SENDING_ON_ERROR, 1L);
173
174 reset_data(&data, curl);
175
176 res = perform_and_check_connections(curl,
177 "First request with CURLOPT_KEEP_SENDING_ON_ERROR", 1);
178 if(res != TEST_ERR_SUCCESS) {
179 goto test_cleanup;
180 }
181
182 reset_data(&data, curl);
183
184 res = perform_and_check_connections(curl,
185 "Second request with CURLOPT_KEEP_SENDING_ON_ERROR", 0);
186 if(res != TEST_ERR_SUCCESS) {
187 goto test_cleanup;
188 }
189
190 res = TEST_ERR_SUCCESS;
191
192 test_cleanup:
193
194 curl_easy_cleanup(curl);
195
196 curl_global_cleanup();
197
198 return (int)res;
199 }
200