1/*************************************************************************** 2 * _ _ ____ _ 3 * Project ___| | | | _ \| | 4 * / __| | | | |_) | | 5 * | (__| |_| | _ <| |___ 6 * \___|\___/|_| \_\_____| 7 * 8 * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al. 9 * 10 * This software is licensed as described in the file COPYING, which 11 * you should have received as part of this distribution. The terms 12 * are also available at https://curl.se/docs/copyright.html. 13 * 14 * You may opt to use, copy, modify, merge, publish, distribute and/or sell 15 * copies of the Software, and permit persons to whom the Software is 16 * furnished to do so, under the terms of the COPYING file. 17 * 18 * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY 19 * KIND, either express or implied. 20 * 21 * SPDX-License-Identifier: curl 22 * 23 ***************************************************************************/ 24#include "test.h" 25 26#include "testutil.h" 27#include "warnless.h" 28#include "memdebug.h" 29 30static const char cmd[] = "A1 IDLE\r\n"; 31static char buf[1024]; 32 33int test(char *URL) 34{ 35 CURLM *mcurl; 36 CURL *curl = NULL; 37 int mrun; 38 curl_socket_t sock = CURL_SOCKET_BAD; 39 time_t start = time(NULL); 40 int state = 0; 41 ssize_t pos = 0; 42 int res = 0; 43 44 global_init(CURL_GLOBAL_DEFAULT); 45 multi_init(mcurl); 46 easy_init(curl); 47 48 easy_setopt(curl, CURLOPT_VERBOSE, 1L); 49 easy_setopt(curl, CURLOPT_URL, URL); 50 easy_setopt(curl, CURLOPT_CONNECT_ONLY, 1L); 51 if(curl_multi_add_handle(mcurl, curl)) 52 goto test_cleanup; 53 54 while(time(NULL) - start < 5) { 55 struct curl_waitfd waitfd; 56 57 multi_perform(mcurl, &mrun); 58 for(;;) { 59 int i; 60 struct CURLMsg *m = curl_multi_info_read(mcurl, &i); 61 62 if(!m) 63 break; 64 if(m->msg == CURLMSG_DONE && m->easy_handle == curl) { 65 curl_easy_getinfo(curl, CURLINFO_ACTIVESOCKET, &sock); 66 if(sock == CURL_SOCKET_BAD) 67 goto test_cleanup; 68 printf("Connected fine, extracted socket. Moving on\n"); 69 } 70 } 71 72 if(sock != CURL_SOCKET_BAD) { 73 waitfd.events = state ? CURL_WAIT_POLLIN : CURL_WAIT_POLLOUT; 74 waitfd.revents = 0; 75 curl_easy_getinfo(curl, CURLINFO_ACTIVESOCKET, &sock); 76 waitfd.fd = sock; 77 } 78 curl_multi_wait(mcurl, &waitfd, sock == CURL_SOCKET_BAD ? 0 : 1, 500, 79 &mrun); 80 if((sock != CURL_SOCKET_BAD) && (waitfd.revents & waitfd.events)) { 81 size_t len = 0; 82 83 if(!state) { 84 CURLcode ec; 85 ec = curl_easy_send(curl, cmd + pos, sizeof(cmd) - 1 - pos, &len); 86 if(ec != CURLE_OK) { 87 fprintf(stderr, "curl_easy_send() failed, with code %d (%s)\n", 88 (int)ec, curl_easy_strerror(ec)); 89 res = ec; 90 goto test_cleanup; 91 } 92 if(len > 0) 93 pos += len; 94 else 95 pos = 0; 96 if(pos == sizeof(cmd) - 1) { 97 state++; 98 pos = 0; 99 } 100 } 101 else if(pos < (ssize_t)sizeof(buf)) { 102 CURLcode ec; 103 ec = curl_easy_recv(curl, buf + pos, sizeof(buf) - pos, &len); 104 if(ec != CURLE_OK) { 105 fprintf(stderr, "curl_easy_recv() failed, with code %d (%s)\n", 106 (int)ec, curl_easy_strerror(ec)); 107 res = ec; 108 goto test_cleanup; 109 } 110 if(len > 0) 111 pos += len; 112 } 113 if(len <= 0) 114 sock = CURL_SOCKET_BAD; 115 } 116 } 117 118 if(state) { 119 fwrite(buf, pos, 1, stdout); 120 putchar('\n'); 121 } 122 123 curl_multi_remove_handle(mcurl, curl); 124test_cleanup: 125 curl_easy_cleanup(curl); 126 curl_multi_cleanup(mcurl); 127 128 curl_global_cleanup(); 129 return res; 130} 131