148 lines
4.0 KiB
C
148 lines
4.0 KiB
C
/*****************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* $Id$
|
|
*
|
|
* This test case is supposed to be identical to 547 except that this uses the
|
|
* multi interface and 547 is easy interface.
|
|
*
|
|
* argv1 = URL
|
|
* argv2 = proxy
|
|
* argv3 = proxyuser:password
|
|
*/
|
|
|
|
#include "test.h"
|
|
#include "testutil.h"
|
|
|
|
#define MULTI_PERFORM_HANG_TIMEOUT 60 * 1000
|
|
|
|
#define UPLOADTHIS "this is the blurb we want to upload\n"
|
|
|
|
static size_t readcallback(void *ptr,
|
|
size_t size,
|
|
size_t nmemb,
|
|
void *clientp)
|
|
{
|
|
int *counter = (int *)clientp;
|
|
|
|
if(*counter) {
|
|
/* only do this once and then require a clearing of this */
|
|
fprintf(stderr, "READ ALREADY DONE!\n");
|
|
return 0;
|
|
}
|
|
(*counter)++; /* bump */
|
|
|
|
if(size * nmemb > strlen(UPLOADTHIS)) {
|
|
fprintf(stderr, "READ!\n");
|
|
strcpy(ptr, UPLOADTHIS);
|
|
return strlen(UPLOADTHIS);
|
|
}
|
|
fprintf(stderr, "READ NOT FINE!\n");
|
|
return 0;
|
|
}
|
|
static curlioerr ioctlcallback(CURL *handle,
|
|
int cmd,
|
|
void *clientp)
|
|
{
|
|
int *counter = (int *)clientp;
|
|
(void)handle; /* unused */
|
|
if(cmd == CURLIOCMD_RESTARTREAD) {
|
|
fprintf(stderr, "REWIND!\n");
|
|
*counter = 0; /* clear counter to make the read callback restart */
|
|
}
|
|
return CURLIOE_OK;
|
|
}
|
|
|
|
|
|
int test(char *URL)
|
|
{
|
|
int res;
|
|
CURL *curl;
|
|
int counter=0;
|
|
CURLM *m;
|
|
int running=1;
|
|
struct timeval mp_start;
|
|
char mp_timedout = FALSE;
|
|
|
|
if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
|
|
fprintf(stderr, "curl_global_init() failed\n");
|
|
return TEST_ERR_MAJOR_BAD;
|
|
}
|
|
|
|
if ((curl = curl_easy_init()) == NULL) {
|
|
fprintf(stderr, "curl_easy_init() failed\n");
|
|
curl_global_cleanup();
|
|
return TEST_ERR_MAJOR_BAD;
|
|
}
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL, URL);
|
|
curl_easy_setopt(curl, CURLOPT_VERBOSE, 1L);
|
|
curl_easy_setopt(curl, CURLOPT_HEADER, 1L);
|
|
|
|
/* read the POST data from a callback */
|
|
curl_easy_setopt(curl, CURLOPT_IOCTLFUNCTION, ioctlcallback);
|
|
curl_easy_setopt(curl, CURLOPT_IOCTLDATA, &counter);
|
|
curl_easy_setopt(curl, CURLOPT_READFUNCTION, readcallback);
|
|
curl_easy_setopt(curl, CURLOPT_READDATA, &counter);
|
|
/* We CANNOT do the POST fine without setting the size (or choose chunked)! */
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDSIZE, strlen(UPLOADTHIS));
|
|
|
|
curl_easy_setopt(curl, CURLOPT_POST, 1L);
|
|
curl_easy_setopt(curl, CURLOPT_PROXY, libtest_arg2);
|
|
curl_easy_setopt(curl, CURLOPT_PROXYUSERPWD, libtest_arg3);
|
|
curl_easy_setopt(curl, CURLOPT_PROXYAUTH,
|
|
(long) (CURLAUTH_NTLM | CURLAUTH_DIGEST | CURLAUTH_BASIC) );
|
|
|
|
if ((m = curl_multi_init()) == NULL) {
|
|
fprintf(stderr, "curl_multi_init() failed\n");
|
|
curl_easy_cleanup(curl);
|
|
curl_global_cleanup();
|
|
return TEST_ERR_MAJOR_BAD;
|
|
}
|
|
|
|
if ((res = (int)curl_multi_add_handle(m, curl)) != CURLM_OK) {
|
|
fprintf(stderr, "curl_multi_add_handle() failed, "
|
|
"with code %d\n", res);
|
|
curl_multi_cleanup(m);
|
|
curl_easy_cleanup(curl);
|
|
curl_global_cleanup();
|
|
return TEST_ERR_MAJOR_BAD;
|
|
}
|
|
|
|
mp_timedout = FALSE;
|
|
mp_start = tutil_tvnow();
|
|
|
|
while (running) {
|
|
res = (int)curl_multi_perform(m, &running);
|
|
if (tutil_tvdiff(tutil_tvnow(), mp_start) >
|
|
MULTI_PERFORM_HANG_TIMEOUT) {
|
|
mp_timedout = TRUE;
|
|
break;
|
|
}
|
|
if (running <= 0) {
|
|
fprintf(stderr, "nothing left running.\n");
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (mp_timedout) {
|
|
if (mp_timedout) fprintf(stderr, "mp_timedout\n");
|
|
fprintf(stderr, "ABORTING TEST, since it seems "
|
|
"that it would have run forever.\n");
|
|
res = TEST_ERR_RUNS_FOREVER;
|
|
}
|
|
|
|
curl_multi_remove_handle(m, curl);
|
|
curl_easy_cleanup(curl);
|
|
curl_multi_cleanup(m);
|
|
|
|
curl_global_cleanup();
|
|
|
|
return res;
|
|
}
|
|
|