1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
|
/***************************************************************************
* _ _ ____ _
* Project ___| | | | _ \| |
* / __| | | | |_) | |
* | (__| |_| | _ <| |___
* \___|\___/|_| \_\_____|
*
* Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
*
* 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.
*
* SPDX-License-Identifier: curl
*
***************************************************************************/
#include "first.h"
#include "testtrace.h"
struct t753_transfer_status {
CURL *curl;
const char *name;
bool pause;
bool is_paused;
bool seen_welcome;
};
static size_t t753_write_cb(char *ptr, size_t size, size_t nmemb, void *userp)
{
struct t753_transfer_status *st = userp;
size_t len = size * nmemb;
(void)ptr;
if(st->pause) {
curl_mfprintf(stderr, "[%s] write_cb(len=%zu), PAUSE\n", st->name, len);
st->is_paused = TRUE;
return CURL_READFUNC_PAUSE;
}
curl_mfprintf(stderr, "[%s] write_cb(len=%zu), CONSUME\n", st->name, len);
st->is_paused = FALSE;
return len;
}
static size_t t753_hd_cb(char *ptr, size_t size, size_t nmemb, void *userp)
{
struct t753_transfer_status *st = userp;
size_t len = size * nmemb;
curl_mfprintf(stderr, "[%s] hd_cb '%.*s'\n", st->name, (int)len, ptr);
if(!strcmp("230 Welcome you silly person\r\n", ptr)) {
st->seen_welcome = TRUE;
st->curl = NULL;
}
return len;
}
static bool t753_setup(const char *URL, const char *name,
CURL **pcurl,
struct t753_transfer_status *st)
{
CURL *curl = NULL;
CURLcode result = CURLE_OK;
*pcurl = NULL;
memset(st, 0, sizeof(*st));
st->name = name;
st->curl = curl;
st->pause = TRUE;
easy_init(curl);
easy_setopt(curl, CURLOPT_URL, URL);
easy_setopt(curl, CURLOPT_WRITEFUNCTION, t753_write_cb);
easy_setopt(curl, CURLOPT_WRITEDATA, st);
easy_setopt(curl, CURLOPT_HEADERFUNCTION, t753_hd_cb);
easy_setopt(curl, CURLOPT_HEADERDATA, st);
easy_setopt(curl, CURLOPT_NOPROGRESS, 1L);
easy_setopt(curl, CURLOPT_DEBUGDATA, &debug_config);
easy_setopt(curl, CURLOPT_DEBUGFUNCTION, libtest_debug_cb);
easy_setopt(curl, CURLOPT_VERBOSE, 1L);
*pcurl = curl;
return TRUE;
test_cleanup:
if(curl)
curl_easy_cleanup(curl);
return FALSE;
}
static CURLcode test_lib753(const char *URL)
{
CURL *curl1 = NULL, *curl2 = NULL;
CURLM *multi = NULL;
struct t753_transfer_status st1, st2;
CURLcode result = CURLE_OK;
CURLMcode mresult;
int still_running;
start_test_timing();
debug_config.nohex = TRUE;
debug_config.tracetime = TRUE;
curl_global_init(CURL_GLOBAL_DEFAULT);
curl_mfprintf(stderr, "init multi\n");
multi = curl_multi_init();
if(!multi) {
result = CURLE_OUT_OF_MEMORY;
goto test_cleanup;
}
if(!t753_setup(URL, "EASY1", &curl1, &st1))
goto test_cleanup;
multi_add_handle(multi, curl1);
multi_perform(multi, &still_running);
abort_on_test_timeout();
curl_mfprintf(stderr, "multi_perform() -> %d running\n", still_running);
while(still_running) {
int num;
/* The purpose of this Test:
* 1. Violently cleanup EASY1 *without* removing it from the multi
* handle first. This MUST discard the connection that EASY1 holds,
* as EASY1 is not DONE at this point.
* With the env var CURL_FTP_PWD_STOP set, the connection will
* have no outstanding data at this point. This would allow
* reuse if the connection is not terminated by the cleanup.
* 2. Add EASY2 for the same URL and observe in the expected result
* that the connection is NOT reused, e.g. all FTP commands
* are sent again on the new connection.
*/
if(curl1 && st1.seen_welcome) {
curl_easy_cleanup(curl1);
curl1 = NULL;
if(!curl2) {
if(!t753_setup(URL, "EASY2", &curl2, &st2))
goto test_cleanup;
st2.pause = FALSE;
multi_add_handle(multi, curl2);
}
}
mresult = curl_multi_wait(multi, NULL, 0, 1, &num);
if(mresult != CURLM_OK) {
curl_mfprintf(stderr, "curl_multi_wait() returned %d\n", mresult);
result = TEST_ERR_MAJOR_BAD;
goto test_cleanup;
}
abort_on_test_timeout();
multi_perform(multi, &still_running);
curl_mfprintf(stderr, "multi_perform() -> %d running\n", still_running);
abort_on_test_timeout();
}
test_cleanup:
if(result)
curl_mfprintf(stderr, "ERROR: %s\n", curl_easy_strerror(result));
if(curl1)
curl_easy_cleanup(curl1);
if(curl2)
curl_easy_cleanup(curl2);
curl_multi_cleanup(multi);
curl_global_cleanup();
return result;
}
|