mirror of
https://github.com/moparisthebest/curl
synced 2024-08-13 17:03:50 -04:00
2f4fe0175b
part of the official libcurl API http://curl.haxx.se/lxr/source/lib/README.curlx The documented way of using them would be to use timeval.c as a source code file. The above described method works very well when statically linking libcurl and apps, test programs, but has several drawbacks when you build a true shared libcurl (i.e. Name space clash at linkage stage as functions are defined more than once. Windows makefiles are not capable of handling this system of source-level sharing) So... Now testutil.h and testutil.c define and implement tutil_tvnow and tutil_tvdiff which replace curlx_tvnow and curlx_tvdiff for the libtest programs. Doing this we avoid the above described problems, and the code in the testsuite does not impose the need to keep those functions public in libcurl even when not part of the API.
123 lines
3.0 KiB
C
123 lines
3.0 KiB
C
/*****************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* $Id$
|
|
*/
|
|
|
|
#include "test.h"
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <fcntl.h>
|
|
|
|
#include "testutil.h"
|
|
|
|
#define MAIN_LOOP_HANG_TIMEOUT 30 * 1000
|
|
#define MULTI_PERFORM_HANG_TIMEOUT 20 * 1000
|
|
|
|
static CURLMcode perform(CURLM * multi);
|
|
|
|
static CURLMcode perform(CURLM * multi)
|
|
{
|
|
int handles, maxfd;
|
|
CURLMcode code;
|
|
fd_set fdread, fdwrite, fdexcep;
|
|
struct timeval mp_start;
|
|
char mp_timedout = FALSE;
|
|
|
|
mp_timedout = FALSE;
|
|
mp_start = tutil_tvnow();
|
|
|
|
for (;;) {
|
|
code = curl_multi_perform(multi, &handles);
|
|
if (tutil_tvdiff(tutil_tvnow(), mp_start) >
|
|
MULTI_PERFORM_HANG_TIMEOUT) {
|
|
mp_timedout = TRUE;
|
|
break;
|
|
}
|
|
if (handles <= 0)
|
|
return CURLM_OK;
|
|
|
|
switch (code) {
|
|
case CURLM_OK:
|
|
break;
|
|
case CURLM_CALL_MULTI_PERFORM:
|
|
continue;
|
|
default:
|
|
return code;
|
|
}
|
|
|
|
FD_ZERO(&fdread);
|
|
FD_ZERO(&fdwrite);
|
|
FD_ZERO(&fdexcep);
|
|
curl_multi_fdset(multi, &fdread, &fdwrite, &fdexcep, &maxfd);
|
|
if (maxfd < 0)
|
|
return (CURLMcode) ~CURLM_OK;
|
|
if (select(maxfd + 1, &fdread, &fdwrite, &fdexcep, 0) == -1)
|
|
return (CURLMcode) ~CURLM_OK;
|
|
}
|
|
|
|
/* We only reach this point 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");
|
|
return (CURLMcode) ~CURLM_OK;
|
|
}
|
|
|
|
int test(char *URL)
|
|
{
|
|
CURLM *multi;
|
|
CURL *easy;
|
|
|
|
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 ((easy = curl_easy_init()) == NULL) {
|
|
fprintf(stderr, "curl_easy_init() failed\n");
|
|
curl_multi_cleanup(multi);
|
|
curl_global_cleanup();
|
|
return TEST_ERR_MAJOR_BAD;
|
|
}
|
|
|
|
curl_multi_setopt(multi, CURLMOPT_PIPELINING, 1);
|
|
|
|
curl_easy_setopt(easy, CURLOPT_WRITEFUNCTION, fwrite);
|
|
curl_easy_setopt(easy, CURLOPT_FAILONERROR, 1);
|
|
curl_easy_setopt(easy, CURLOPT_URL, URL);
|
|
|
|
curl_multi_add_handle(multi, easy);
|
|
if (perform(multi) != CURLM_OK)
|
|
printf("retrieve 1 failed\n");
|
|
|
|
curl_multi_remove_handle(multi, easy);
|
|
curl_easy_reset(easy);
|
|
|
|
curl_easy_setopt(easy, CURLOPT_FAILONERROR, 1);
|
|
curl_easy_setopt(easy, CURLOPT_URL, arg2);
|
|
|
|
curl_multi_add_handle(multi, easy);
|
|
if (perform(multi) != CURLM_OK)
|
|
printf("retrieve 2 failed\n");
|
|
|
|
curl_multi_remove_handle(multi, easy);
|
|
curl_easy_cleanup(easy);
|
|
curl_multi_cleanup(multi);
|
|
curl_global_cleanup();
|
|
|
|
printf("Finished!\n");
|
|
|
|
return 0;
|
|
}
|