| /***************************************************************************** |
| * _ _ ____ _ |
| * 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" |
| #include "memdebug.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; |
| } |
| |