| /***************************************************************************** |
| * _ _ ____ _ |
| * Project ___| | | | _ \| | |
| * / __| | | | |_) | | |
| * | (__| |_| | _ <| |___ |
| * \___|\___/|_| \_\_____| |
| * |
| * $Id$ |
| */ |
| |
| #include "test.h" |
| |
| #include "testutil.h" |
| #include "memdebug.h" |
| |
| #define MAIN_LOOP_HANG_TIMEOUT 90 * 1000 |
| #define MULTI_PERFORM_HANG_TIMEOUT 60 * 1000 |
| |
| int test(char *URL) |
| { |
| CURL* curls; |
| CURLM* multi; |
| int still_running; |
| int i = -1; |
| CURLMsg *msg; |
| CURLMcode res; |
| struct timeval ml_start; |
| struct timeval mp_start; |
| char ml_timedout = FALSE; |
| 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 ((multi = curl_multi_init()) == NULL) { |
| fprintf(stderr, "curl_multi_init() failed\n"); |
| curl_global_cleanup(); |
| return TEST_ERR_MAJOR_BAD; |
| } |
| |
| if ((curls = curl_easy_init()) == NULL) { |
| fprintf(stderr, "curl_easy_init() failed\n"); |
| curl_multi_cleanup(multi); |
| curl_global_cleanup(); |
| return TEST_ERR_MAJOR_BAD; |
| } |
| |
| curl_easy_setopt(curls, CURLOPT_URL, URL); |
| |
| if ((res = curl_multi_add_handle(multi, curls)) != CURLM_OK) { |
| fprintf(stderr, "curl_multi_add_handle() failed, " |
| "with code %d\n", res); |
| curl_easy_cleanup(curls); |
| curl_multi_cleanup(multi); |
| curl_global_cleanup(); |
| return TEST_ERR_MAJOR_BAD; |
| } |
| |
| mp_timedout = FALSE; |
| mp_start = tutil_tvnow(); |
| |
| do { |
| res = curl_multi_perform(multi, &still_running); |
| if (tutil_tvdiff(tutil_tvnow(), mp_start) > |
| MULTI_PERFORM_HANG_TIMEOUT) { |
| mp_timedout = TRUE; |
| break; |
| } |
| } while (res == CURLM_CALL_MULTI_PERFORM); |
| |
| ml_timedout = FALSE; |
| ml_start = tutil_tvnow(); |
| |
| while ((!ml_timedout) && (!mp_timedout) && (still_running)) { |
| struct timeval timeout; |
| int rc; |
| fd_set fdread; |
| fd_set fdwrite; |
| fd_set fdexcep; |
| int maxfd; |
| |
| FD_ZERO(&fdread); |
| FD_ZERO(&fdwrite); |
| FD_ZERO(&fdexcep); |
| timeout.tv_sec = 1; |
| timeout.tv_usec = 0; |
| |
| if (tutil_tvdiff(tutil_tvnow(), ml_start) > |
| MAIN_LOOP_HANG_TIMEOUT) { |
| ml_timedout = TRUE; |
| break; |
| } |
| |
| curl_multi_fdset(multi, &fdread, &fdwrite, &fdexcep, &maxfd); |
| rc = select_test(maxfd+1, &fdread, &fdwrite, &fdexcep, &timeout); |
| switch(rc) { |
| case -1: |
| break; |
| case 0: |
| default: |
| mp_timedout = FALSE; |
| mp_start = tutil_tvnow(); |
| do { |
| res = curl_multi_perform(multi, &still_running); |
| if (tutil_tvdiff(tutil_tvnow(), mp_start) > |
| MULTI_PERFORM_HANG_TIMEOUT) { |
| mp_timedout = TRUE; |
| break; |
| } |
| } while (res == CURLM_CALL_MULTI_PERFORM); |
| break; |
| } |
| } |
| if (ml_timedout || mp_timedout) { |
| if (ml_timedout) fprintf(stderr, "ml_timedout\n"); |
| if (mp_timedout) fprintf(stderr, "mp_timedout\n"); |
| fprintf(stderr, "ABORTING TEST, since it seems " |
| "that it would have run forever.\n"); |
| i = TEST_ERR_RUNS_FOREVER; |
| } |
| else { |
| msg = curl_multi_info_read(multi, &still_running); |
| if(msg) |
| /* this should now contain a result code from the easy handle, |
| get it */ |
| i = msg->data.result; |
| } |
| |
| curl_multi_cleanup(multi); |
| curl_easy_cleanup(curls); |
| curl_global_cleanup(); |
| |
| return i; /* return the final return code */ |
| } |