luo980/curl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test556: improve robustness

Browse Source 
Fix handling of partial/blocked sends.

Fixes test with CURL_DBG_SOCK_WBLOCK=90 set.

Closes #14453
This commit is contained in:
Stefan Eissing 2024-08-08 13:23:32 +02:00 committed by Daniel Stenberg
parent 32f9130ae1
commit ac6349b451
No known key found for this signature in database
GPG Key ID: 5CC908FDB71E12C2
1 changed files with 24 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
tests/libtest/lib556.c
View File

@ -65,28 +65,35 @@ CURLcode test(char *URL)
const char *request = const char *request =
"GET /556 HTTP/1.1\r\n" "GET /556 HTTP/1.1\r\n"
"Host: ninja\r\n\r\n"; "Host: ninja\r\n\r\n";
size_t iolen = 0; const char *sbuf = request;
size_t sblen = strlen(request);
size_t nwritten = 0, nread = 0;
res = curl_easy_send(curl, request, strlen(request), &iolen); do {
char buf[1024];
if(!res) { if(sblen) {
/* we assume that sending always work */ res = curl_easy_send(curl, sbuf, sblen, &nwritten);
if(res && res != CURLE_AGAIN)
do { break;
char buf[1024]; if(nwritten > 0) {
/* busy-read like crazy */ sbuf += nwritten;
res = curl_easy_recv(curl, buf, sizeof(buf), &iolen); sblen -= nwritten;
if(iolen) {
/* send received stuff to stdout */
if(!write(STDOUT_FILENO, buf, iolen))
break;
} }
}
} while((res == CURLE_OK && iolen) || (res == CURLE_AGAIN)); /* busy-read like crazy */
} res = curl_easy_recv(curl, buf, sizeof(buf), &nread);
if(iolen) if(nread) {
/* send received stuff to stdout */
if(!write(STDOUT_FILENO, buf, nread))
break;
}
} while((res == CURLE_OK && nread) || (res == CURLE_AGAIN));
if(res && res != CURLE_AGAIN)
res = TEST_ERR_FAILURE; res = TEST_ERR_FAILURE;
} }