curl/tests/libtest/libprereq.c
Max Dymond a517378de5
CURLOPT_PREREQFUNCTION: add new callback
Triggered before a request is made but after a connection is set up

Changes:

- callback: Update docs and callback for pre-request callback
- Add documentation for CURLOPT_PREREQDATA and CURLOPT_PREREQFUNCTION,
- Add redirect test and callback failure test
- Note that the function may be called multiple times on a redirection
- Disable new 2086 test due to Windows weirdness

Closes #7477
2021-09-27 17:16:43 +02:00

99 lines
2.9 KiB
C

/***************************************************************************
* _ _ ____ _
* Project ___| | | | _ \| |
* / __| | | | |_) | |
* | (__| |_| | _ <| |___
* \___|\___/|_| \_\_____|
*
* Copyright (C) 2021, Max Dymond, <max.dymond@microsoft.com>
*
* This software is licensed as described in the file COPYING, which
* you should have received as part of this distribution. The terms
* are also available at https://curl.se/docs/copyright.html.
*
* You may opt to use, copy, modify, merge, publish, distribute and/or sell
* copies of the Software, and permit persons to whom the Software is
* furnished to do so, under the terms of the COPYING file.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
***************************************************************************/
#include "test.h"
typedef struct prcs {
int prereq_retcode;
int ipv6;
} PRCS;
static int prereq_callback(void *clientp,
char *conn_primary_ip,
char *conn_local_ip,
int conn_primary_port,
int conn_local_port)
{
PRCS *prereq_cb = (PRCS *)clientp;
if(prereq_cb->ipv6) {
printf("Connected to [%s]\n", conn_primary_ip);
printf("Connected from [%s]\n", conn_local_ip);
}
else {
printf("Connected to %s\n", conn_primary_ip);
printf("Connected from %s\n", conn_local_ip);
}
printf("Remote port = %d\n", conn_primary_port);
printf("Local port = %d\n", conn_local_port);
printf("Returning = %d\n", prereq_cb->prereq_retcode);
return prereq_cb->prereq_retcode;
}
int test(char *URL)
{
PRCS prereq_cb;
CURLcode ret = CURLE_OK;
CURL *curl = NULL;
prereq_cb.prereq_retcode = CURL_PREREQFUNC_OK;
prereq_cb.ipv6 = 0;
curl_global_init(CURL_GLOBAL_ALL);
curl = curl_easy_init();
if(curl) {
if(strstr(URL, "#ipv6")) {
/* The IP addresses should be surrounded by brackets! */
prereq_cb.ipv6 = 1;
}
if(strstr(URL, "#err")) {
/* Set the callback to exit with failure */
prereq_cb.prereq_retcode = CURL_PREREQFUNC_ABORT;
}
curl_easy_setopt(curl, CURLOPT_URL, URL);
curl_easy_setopt(curl, CURLOPT_PREREQFUNCTION, prereq_callback);
curl_easy_setopt(curl, CURLOPT_PREREQDATA, &prereq_cb);
curl_easy_setopt(curl, CURLOPT_WRITEDATA, stderr);
if(strstr(URL, "#redir")) {
/* Enable follow-location */
curl_easy_setopt(curl, CURLOPT_FOLLOWLOCATION, 1);
}
ret = curl_easy_perform(curl);
if(ret) {
fprintf(stderr, "%s:%d curl_easy_perform() failed with code %d (%s)\n",
__FILE__, __LINE__, ret, curl_easy_strerror(ret));
goto test_cleanup;
}
}
test_cleanup:
curl_easy_cleanup(curl);
curl_global_cleanup();
return ret;
}