1 /*************************************************************************** 2 * _ _ ____ _ 3 * Project ___| | | | _ \| | 4 * / __| | | | |_) | | 5 * | (__| |_| | _ <| |___ 6 * \___|\___/|_| \_\_____| 7 * 8 * Copyright (C) 1998 - 2017, Daniel Stenberg, <daniel (at) 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.haxx.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 ***************************************************************************/ 22 #include "test.h" 23 24 #include "testutil.h" 25 #include "warnless.h" 26 #include "memdebug.h" 27 28 #define TEST_HANG_TIMEOUT 60 * 1000 29 30 static char const testData[] = ".abc\0xyz"; 31 static off_t const testDataSize = sizeof(testData) - 1; 32 33 int test(char *URL) 34 { 35 CURL *easy; 36 CURLM *multi_handle; 37 int still_running; /* keep number of running handles */ 38 CURLMsg *msg; /* for picking up messages with the transfer status */ 39 int msgs_left; /* how many messages are left */ 40 int res = CURLE_OK; 41 42 start_test_timing(); 43 44 global_init(CURL_GLOBAL_ALL); 45 46 /* Allocate one CURL handle per transfer */ 47 easy = curl_easy_init(); 48 49 /* init a multi stack */ 50 multi_handle = curl_multi_init(); 51 52 /* add the individual transfer */ 53 curl_multi_add_handle(multi_handle, easy); 54 55 /* set the options (I left out a few, you'll get the point anyway) */ 56 curl_easy_setopt(easy, CURLOPT_URL, URL); 57 curl_easy_setopt(easy, CURLOPT_POSTFIELDSIZE_LARGE, 58 (curl_off_t)testDataSize); 59 curl_easy_setopt(easy, CURLOPT_POSTFIELDS, testData); 60 61 /* we start some action by calling perform right away */ 62 curl_multi_perform(multi_handle, &still_running); 63 64 abort_on_test_timeout(); 65 66 do { 67 struct timeval timeout; 68 int rc; /* select() return code */ 69 CURLMcode mc; /* curl_multi_fdset() return code */ 70 71 fd_set fdread; 72 fd_set fdwrite; 73 fd_set fdexcep; 74 int maxfd = -1; 75 76 long curl_timeo = -1; 77 78 FD_ZERO(&fdread); 79 FD_ZERO(&fdwrite); 80 FD_ZERO(&fdexcep); 81 82 /* set a suitable timeout to play around with */ 83 timeout.tv_sec = 1; 84 timeout.tv_usec = 0; 85 86 curl_multi_timeout(multi_handle, &curl_timeo); 87 if(curl_timeo >= 0) { 88 timeout.tv_sec = curl_timeo / 1000; 89 if(timeout.tv_sec > 1) 90 timeout.tv_sec = 1; 91 else 92 timeout.tv_usec = (curl_timeo % 1000) * 1000; 93 } 94 95 /* get file descriptors from the transfers */ 96 mc = curl_multi_fdset(multi_handle, &fdread, &fdwrite, &fdexcep, &maxfd); 97 98 if(mc != CURLM_OK) { 99 fprintf(stderr, "curl_multi_fdset() failed, code %d.\n", mc); 100 break; 101 } 102 103 /* On success the value of maxfd is guaranteed to be >= -1. We call 104 select(maxfd + 1, ...); specially in case of (maxfd == -1) there are 105 no fds ready yet so we call select(0, ...) --or Sleep() on Windows-- 106 to sleep 100ms, which is the minimum suggested value in the 107 curl_multi_fdset() doc. */ 108 109 if(maxfd == -1) { 110 #ifdef _WIN32 111 Sleep(100); 112 rc = 0; 113 #else 114 /* Portable sleep for platforms other than Windows. */ 115 struct timeval wait = { 0, 100 * 1000 }; /* 100ms */ 116 rc = select(0, NULL, NULL, NULL, &wait); 117 #endif 118 } 119 else { 120 /* Note that on some platforms 'timeout' may be modified by select(). 121 If you need access to the original value save a copy beforehand. */ 122 rc = select(maxfd + 1, &fdread, &fdwrite, &fdexcep, &timeout); 123 } 124 125 switch(rc) { 126 case -1: 127 /* select error */ 128 break; 129 case 0: /* timeout */ 130 default: /* action */ 131 curl_multi_perform(multi_handle, &still_running); 132 break; 133 } 134 135 abort_on_test_timeout(); 136 } while(still_running); 137 138 /* See how the transfers went */ 139 do { 140 msg = curl_multi_info_read(multi_handle, &msgs_left); 141 if(msg && msg->msg == CURLMSG_DONE) { 142 printf("HTTP transfer completed with status %d\n", msg->data.result); 143 break; 144 } 145 146 abort_on_test_timeout(); 147 } while(msg); 148 149 test_cleanup: 150 curl_multi_cleanup(multi_handle); 151 152 /* Free the CURL handles */ 153 curl_easy_cleanup(easy); 154 curl_global_cleanup(); 155 156 return res; 157 } 158