mirror of
https://github.com/curl/curl.git
synced 2024-12-27 06:59:43 +08:00
25cbc2f79a
Before this patch, the result code was a mixture of `int` and `CURLcode`. Also adjust casts and fix a couple of minor issues found along the way. Cherry-picked from #13489 Closes #13600
131 lines
3.7 KiB
C
131 lines
3.7 KiB
C
/***************************************************************************
|
|
* _ _ ____ _
|
|
* 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 "test.h"
|
|
|
|
#include "testutil.h"
|
|
#include "warnless.h"
|
|
#include "memdebug.h"
|
|
|
|
static const char cmd[] = "A1 IDLE\r\n";
|
|
static char buf[1024];
|
|
|
|
CURLcode test(char *URL)
|
|
{
|
|
CURLM *mcurl;
|
|
CURL *curl = NULL;
|
|
int mrun;
|
|
curl_socket_t sock = CURL_SOCKET_BAD;
|
|
time_t start = time(NULL);
|
|
int state = 0;
|
|
ssize_t pos = 0;
|
|
CURLcode res = CURLE_OK;
|
|
|
|
global_init(CURL_GLOBAL_DEFAULT);
|
|
multi_init(mcurl);
|
|
easy_init(curl);
|
|
|
|
easy_setopt(curl, CURLOPT_VERBOSE, 1L);
|
|
easy_setopt(curl, CURLOPT_URL, URL);
|
|
easy_setopt(curl, CURLOPT_CONNECT_ONLY, 1L);
|
|
if(curl_multi_add_handle(mcurl, curl))
|
|
goto test_cleanup;
|
|
|
|
while(time(NULL) - start < 5) {
|
|
struct curl_waitfd waitfd;
|
|
|
|
multi_perform(mcurl, &mrun);
|
|
for(;;) {
|
|
int i;
|
|
struct CURLMsg *m = curl_multi_info_read(mcurl, &i);
|
|
|
|
if(!m)
|
|
break;
|
|
if(m->msg == CURLMSG_DONE && m->easy_handle == curl) {
|
|
curl_easy_getinfo(curl, CURLINFO_ACTIVESOCKET, &sock);
|
|
if(sock == CURL_SOCKET_BAD)
|
|
goto test_cleanup;
|
|
printf("Connected fine, extracted socket. Moving on\n");
|
|
}
|
|
}
|
|
|
|
if(sock != CURL_SOCKET_BAD) {
|
|
waitfd.events = state ? CURL_WAIT_POLLIN : CURL_WAIT_POLLOUT;
|
|
waitfd.revents = 0;
|
|
curl_easy_getinfo(curl, CURLINFO_ACTIVESOCKET, &sock);
|
|
waitfd.fd = sock;
|
|
}
|
|
curl_multi_wait(mcurl, &waitfd, sock == CURL_SOCKET_BAD ? 0 : 1, 500,
|
|
&mrun);
|
|
if((sock != CURL_SOCKET_BAD) && (waitfd.revents & waitfd.events)) {
|
|
size_t len = 0;
|
|
|
|
if(!state) {
|
|
CURLcode ec;
|
|
ec = curl_easy_send(curl, cmd + pos, sizeof(cmd) - 1 - pos, &len);
|
|
if(ec != CURLE_OK) {
|
|
fprintf(stderr, "curl_easy_send() failed, with code %d (%s)\n",
|
|
(int)ec, curl_easy_strerror(ec));
|
|
res = ec;
|
|
goto test_cleanup;
|
|
}
|
|
if(len > 0)
|
|
pos += len;
|
|
else
|
|
pos = 0;
|
|
if(pos == sizeof(cmd) - 1) {
|
|
state++;
|
|
pos = 0;
|
|
}
|
|
}
|
|
else if(pos < (ssize_t)sizeof(buf)) {
|
|
CURLcode ec;
|
|
ec = curl_easy_recv(curl, buf + pos, sizeof(buf) - pos, &len);
|
|
if(ec != CURLE_OK) {
|
|
fprintf(stderr, "curl_easy_recv() failed, with code %d (%s)\n",
|
|
(int)ec, curl_easy_strerror(ec));
|
|
res = ec;
|
|
goto test_cleanup;
|
|
}
|
|
if(len > 0)
|
|
pos += len;
|
|
}
|
|
if(len <= 0)
|
|
sock = CURL_SOCKET_BAD;
|
|
}
|
|
}
|
|
|
|
if(state) {
|
|
fwrite(buf, pos, 1, stdout);
|
|
putchar('\n');
|
|
}
|
|
|
|
curl_multi_remove_handle(mcurl, curl);
|
|
test_cleanup:
|
|
curl_easy_cleanup(curl);
|
|
curl_multi_cleanup(mcurl);
|
|
|
|
curl_global_cleanup();
|
|
return res;
|
|
}
|