un-nest curl
This commit is contained in:
248
curl-8.15.0/tests/libtest/lib670.c
Normal file
248
curl-8.15.0/tests/libtest/lib670.c
Normal file
@@ -0,0 +1,248 @@
|
||||
/***************************************************************************
|
||||
* _ _ ____ _
|
||||
* Project ___| | | | _ \| |
|
||||
* / __| | | | |_) | |
|
||||
* | (__| |_| | _ <| |___
|
||||
* \___|\___/|_| \_\_____|
|
||||
*
|
||||
* Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
|
||||
*
|
||||
* This software is licensed as described in the file COPYING, which
|
||||
* you should have received as part of this distribution. The terms
|
||||
* are also available at https://curl.se/docs/copyright.html.
|
||||
*
|
||||
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
|
||||
* copies of the Software, and permit persons to whom the Software is
|
||||
* furnished to do so, under the terms of the COPYING file.
|
||||
*
|
||||
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
|
||||
* KIND, either express or implied.
|
||||
*
|
||||
* SPDX-License-Identifier: curl
|
||||
*
|
||||
***************************************************************************/
|
||||
#include "first.h"
|
||||
|
||||
#include "memdebug.h"
|
||||
|
||||
#define PAUSE_TIME 5
|
||||
|
||||
struct t670_ReadThis {
|
||||
CURL *easy;
|
||||
time_t origin;
|
||||
int count;
|
||||
};
|
||||
|
||||
static size_t t670_read_cb(char *ptr, size_t size, size_t nmemb, void *userp)
|
||||
{
|
||||
struct t670_ReadThis *pooh = (struct t670_ReadThis *) userp;
|
||||
time_t delta;
|
||||
|
||||
if(size * nmemb < 1)
|
||||
return 0;
|
||||
|
||||
switch(pooh->count++) {
|
||||
case 0:
|
||||
*ptr = '\x41'; /* ASCII A. */
|
||||
return 1;
|
||||
case 1:
|
||||
pooh->origin = time(NULL);
|
||||
return CURL_READFUNC_PAUSE;
|
||||
case 2:
|
||||
delta = time(NULL) - pooh->origin;
|
||||
*ptr = delta >= PAUSE_TIME ? '\x42' : '\x41'; /* ASCII A or B. */
|
||||
return 1;
|
||||
case 3:
|
||||
return 0;
|
||||
}
|
||||
curl_mfprintf(stderr, "Read callback called after EOF\n");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static int t670_xferinfo(void *clientp,
|
||||
curl_off_t dltotal, curl_off_t dlnow,
|
||||
curl_off_t ultotal, curl_off_t ulnow)
|
||||
{
|
||||
struct t670_ReadThis *pooh = (struct t670_ReadThis *) clientp;
|
||||
|
||||
(void) dltotal;
|
||||
(void) dlnow;
|
||||
(void) ultotal;
|
||||
(void) ulnow;
|
||||
|
||||
if(pooh->origin) {
|
||||
time_t delta = time(NULL) - pooh->origin;
|
||||
|
||||
if(delta >= 4 * PAUSE_TIME) {
|
||||
curl_mfprintf(stderr, "unpausing failed: drain problem?\n");
|
||||
return CURLE_ABORTED_BY_CALLBACK;
|
||||
}
|
||||
|
||||
if(delta >= PAUSE_TIME)
|
||||
curl_easy_pause(pooh->easy, CURLPAUSE_CONT);
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static CURLcode test_lib670(char *URL)
|
||||
{
|
||||
static const char testname[] = "field";
|
||||
|
||||
curl_mime *mime = NULL;
|
||||
struct curl_httppost *formpost = NULL;
|
||||
|
||||
struct t670_ReadThis pooh;
|
||||
CURLcode res = TEST_ERR_FAILURE;
|
||||
|
||||
/*
|
||||
* Check proper pausing/unpausing from a mime or form read callback.
|
||||
*/
|
||||
|
||||
if(curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
|
||||
curl_mfprintf(stderr, "curl_global_init() failed\n");
|
||||
return TEST_ERR_MAJOR_BAD;
|
||||
}
|
||||
|
||||
pooh.origin = (time_t) 0;
|
||||
pooh.count = 0;
|
||||
pooh.easy = curl_easy_init();
|
||||
|
||||
/* First set the URL that is about to receive our POST. */
|
||||
test_setopt(pooh.easy, CURLOPT_URL, URL);
|
||||
|
||||
/* get verbose debug output please */
|
||||
test_setopt(pooh.easy, CURLOPT_VERBOSE, 1L);
|
||||
|
||||
/* include headers in the output */
|
||||
test_setopt(pooh.easy, CURLOPT_HEADER, 1L);
|
||||
|
||||
if(testnum == 670 || testnum == 671) {
|
||||
curl_mimepart *part;
|
||||
/* Build the mime tree. */
|
||||
mime = curl_mime_init(pooh.easy);
|
||||
part = curl_mime_addpart(mime);
|
||||
res = curl_mime_name(part, testname);
|
||||
if(res != CURLE_OK) {
|
||||
curl_mfprintf(stderr,
|
||||
"Something went wrong when building the mime structure: %d\n",
|
||||
res);
|
||||
goto test_cleanup;
|
||||
}
|
||||
|
||||
res = curl_mime_data_cb(part, (curl_off_t) 2, t670_read_cb,
|
||||
NULL, NULL, &pooh);
|
||||
|
||||
/* Bind mime data to its easy handle. */
|
||||
if(res == CURLE_OK)
|
||||
test_setopt(pooh.easy, CURLOPT_MIMEPOST, mime);
|
||||
}
|
||||
else {
|
||||
struct curl_httppost *lastptr = NULL;
|
||||
CURLFORMcode formrc;
|
||||
/* Build the form. */
|
||||
formrc = curl_formadd(&formpost, &lastptr,
|
||||
CURLFORM_COPYNAME, testname,
|
||||
CURLFORM_STREAM, &pooh,
|
||||
CURLFORM_CONTENTLEN, (curl_off_t) 2,
|
||||
CURLFORM_END);
|
||||
if(formrc) {
|
||||
curl_mfprintf(stderr, "curl_formadd() = %d\n", (int) formrc);
|
||||
goto test_cleanup;
|
||||
}
|
||||
|
||||
/* We want to use our own read function. */
|
||||
test_setopt(pooh.easy, CURLOPT_READFUNCTION, t670_read_cb);
|
||||
|
||||
/* Send a multi-part formpost. */
|
||||
test_setopt(pooh.easy, CURLOPT_HTTPPOST, formpost);
|
||||
}
|
||||
|
||||
if(testnum == 670 || testnum == 672) {
|
||||
CURLMcode mres;
|
||||
CURLM *multi;
|
||||
/* Use the multi interface. */
|
||||
multi = curl_multi_init();
|
||||
mres = curl_multi_add_handle(multi, pooh.easy);
|
||||
while(!mres) {
|
||||
struct timeval timeout;
|
||||
int rc = 0;
|
||||
fd_set fdread;
|
||||
fd_set fdwrite;
|
||||
fd_set fdexcept;
|
||||
int maxfd = -1;
|
||||
int still_running = 0;
|
||||
|
||||
mres = curl_multi_perform(multi, &still_running);
|
||||
if(!still_running || mres != CURLM_OK)
|
||||
break;
|
||||
|
||||
if(pooh.origin) {
|
||||
time_t delta = time(NULL) - pooh.origin;
|
||||
|
||||
if(delta >= 4 * PAUSE_TIME) {
|
||||
curl_mfprintf(stderr, "unpausing failed: drain problem?\n");
|
||||
res = CURLE_OPERATION_TIMEDOUT;
|
||||
break;
|
||||
}
|
||||
|
||||
if(delta >= PAUSE_TIME)
|
||||
curl_easy_pause(pooh.easy, CURLPAUSE_CONT);
|
||||
}
|
||||
|
||||
FD_ZERO(&fdread);
|
||||
FD_ZERO(&fdwrite);
|
||||
FD_ZERO(&fdexcept);
|
||||
timeout.tv_sec = 0;
|
||||
timeout.tv_usec = 1000000 * PAUSE_TIME / 10;
|
||||
mres = curl_multi_fdset(multi, &fdread, &fdwrite, &fdexcept, &maxfd);
|
||||
if(mres)
|
||||
break;
|
||||
#ifdef _WIN32
|
||||
if(maxfd == -1)
|
||||
curlx_wait_ms(100);
|
||||
else
|
||||
#endif
|
||||
rc = select(maxfd + 1, &fdread, &fdwrite, &fdexcept, &timeout);
|
||||
if(rc == -1) {
|
||||
curl_mfprintf(stderr, "Select error\n");
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if(mres != CURLM_OK)
|
||||
for(;;) {
|
||||
int msgs_left;
|
||||
CURLMsg *msg;
|
||||
msg = curl_multi_info_read(multi, &msgs_left);
|
||||
if(!msg)
|
||||
break;
|
||||
if(msg->msg == CURLMSG_DONE) {
|
||||
res = msg->data.result;
|
||||
}
|
||||
}
|
||||
|
||||
curl_multi_remove_handle(multi, pooh.easy);
|
||||
curl_multi_cleanup(multi);
|
||||
}
|
||||
else {
|
||||
/* Use the easy interface. */
|
||||
test_setopt(pooh.easy, CURLOPT_XFERINFODATA, &pooh);
|
||||
test_setopt(pooh.easy, CURLOPT_XFERINFOFUNCTION, t670_xferinfo);
|
||||
test_setopt(pooh.easy, CURLOPT_NOPROGRESS, 0L);
|
||||
res = curl_easy_perform(pooh.easy);
|
||||
}
|
||||
|
||||
test_cleanup:
|
||||
curl_easy_cleanup(pooh.easy);
|
||||
|
||||
if(testnum == 670 || testnum == 671) {
|
||||
curl_mime_free(mime);
|
||||
}
|
||||
else {
|
||||
curl_formfree(formpost);
|
||||
}
|
||||
|
||||
curl_global_cleanup();
|
||||
return res;
|
||||
}
|
||||
Reference in New Issue
Block a user