141 строка
3.4 KiB
C
141 строка
3.4 KiB
C
/*****************************************************************************
|
|
* _ _ ____ _
|
|
* Project ___| | | | _ \| |
|
|
* / __| | | | |_) | |
|
|
* | (__| |_| | _ <| |___
|
|
* \___|\___/|_| \_\_____|
|
|
*
|
|
* $Id$
|
|
*
|
|
* This is the 'proxyauth.c' test app posted by Shmulik Regev on the libcurl
|
|
* mailing list on 10 Jul 2007, converted to a test case.
|
|
*
|
|
* argv1 = URL
|
|
* argv2 = proxy
|
|
* argv3 = proxyuser:password
|
|
* argv4 = host name to use for the custom Host: header
|
|
*/
|
|
|
|
#include "test.h"
|
|
|
|
#include "memdebug.h"
|
|
|
|
#define PROXY libtest_arg2
|
|
#define PROXYUSERPWD libtest_arg3
|
|
#define HOST test_argv[4]
|
|
|
|
static void init(CURLM *cm, const char* url, const char* userpwd,
|
|
struct curl_slist *headers)
|
|
{
|
|
CURL *eh = curl_easy_init();
|
|
|
|
curl_easy_setopt(eh, CURLOPT_URL, url);
|
|
curl_easy_setopt(eh, CURLOPT_PROXY, PROXY);
|
|
curl_easy_setopt(eh, CURLOPT_PROXYUSERPWD, userpwd);
|
|
curl_easy_setopt(eh, CURLOPT_PROXYAUTH, (long)CURLAUTH_ANY);
|
|
curl_easy_setopt(eh, CURLOPT_VERBOSE, 1L);
|
|
curl_easy_setopt(eh, CURLOPT_HEADER, 1L);
|
|
curl_easy_setopt(eh, CURLOPT_HTTPHEADER, headers); /* custom Host: */
|
|
|
|
curl_multi_add_handle(cm, eh);
|
|
}
|
|
|
|
static int loop(CURLM *cm, const char* url, const char* userpwd,
|
|
struct curl_slist *headers)
|
|
{
|
|
CURLMsg *msg;
|
|
CURLMcode code;
|
|
long L;
|
|
int M, Q, U = -1;
|
|
fd_set R, W, E;
|
|
struct timeval T;
|
|
|
|
init(cm, url, userpwd, headers);
|
|
|
|
while (U) {
|
|
|
|
do {
|
|
code = curl_multi_perform(cm, &U);
|
|
} while (code == CURLM_CALL_MULTI_PERFORM);
|
|
|
|
if (U) {
|
|
FD_ZERO(&R);
|
|
FD_ZERO(&W);
|
|
FD_ZERO(&E);
|
|
|
|
if (curl_multi_fdset(cm, &R, &W, &E, &M)) {
|
|
fprintf(stderr, "E: curl_multi_fdset\n");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
/* In a real-world program you OF COURSE check the return that maxfd is
|
|
bigger than -1 so that the call to select() below makes sense! */
|
|
|
|
if (curl_multi_timeout(cm, &L)) {
|
|
fprintf(stderr, "E: curl_multi_timeout\n");
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if(L != -1) {
|
|
T.tv_sec = L/1000;
|
|
T.tv_usec = (L%1000)*1000;
|
|
}
|
|
else {
|
|
T.tv_sec = 5;
|
|
T.tv_usec = 0;
|
|
}
|
|
|
|
if (0 > select(M+1, &R, &W, &E, &T)) {
|
|
fprintf(stderr, "E: select\n");
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
while ((msg = curl_multi_info_read(cm, &Q))) {
|
|
if (msg->msg == CURLMSG_DONE) {
|
|
CURL *e = msg->easy_handle;
|
|
fprintf(stderr, "R: %d - %s\n", (int)msg->data.result,
|
|
curl_easy_strerror(msg->data.result));
|
|
curl_multi_remove_handle(cm, e);
|
|
curl_easy_cleanup(e);
|
|
}
|
|
else {
|
|
fprintf(stderr, "E: CURLMsg (%d)\n", (int)msg->msg);
|
|
}
|
|
}
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
int test(char *URL)
|
|
{
|
|
CURLM *cm;
|
|
struct curl_slist *headers = NULL;
|
|
char buffer[246]; /* naively fixed-size */
|
|
|
|
if(test_argc < 4)
|
|
return 99;
|
|
|
|
sprintf(buffer, "Host: %s", HOST);
|
|
|
|
/* now add a custom Host: header */
|
|
headers = curl_slist_append(headers, buffer);
|
|
|
|
curl_global_init(CURL_GLOBAL_ALL);
|
|
|
|
cm = curl_multi_init();
|
|
loop(cm, URL, PROXYUSERPWD, headers);
|
|
|
|
fprintf(stderr, "lib540: now we do the request again\n");
|
|
loop(cm, URL, PROXYUSERPWD, headers);
|
|
|
|
curl_multi_cleanup(cm);
|
|
|
|
curl_global_cleanup();
|
|
|
|
curl_slist_free_all(headers);
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|