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