1/*************************************************************************** 2 * _ _ ____ _ 3 * Project ___| | | | _ \| | 4 * / __| | | | |_) | | 5 * | (__| |_| | _ <| |___ 6 * \___|\___/|_| \_\_____| 7 * 8 * Copyright (C) 1998 - 2011, 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 http://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#ifdef HAVE_SYS_TYPES_H 25#include <sys/types.h> 26#endif 27#ifdef HAVE_SYS_STAT_H 28#include <sys/stat.h> 29#endif 30#ifdef HAVE_STRING_H 31#include <string.h> 32#endif 33 34#include "testutil.h" 35#include "warnless.h" 36#include "memdebug.h" 37 38#define MAIN_LOOP_HANG_TIMEOUT 90 * 1000 39#define MULTI_PERFORM_HANG_TIMEOUT 60 * 1000 40 41#define NUM_HANDLES 4 42 43int test(char *URL) 44{ 45 int res = 0; 46 CURL *curl[NUM_HANDLES]; 47 int running; 48 char done=FALSE; 49 CURLM *m; 50 int i, j; 51 struct timeval ml_start; 52 struct timeval mp_start; 53 char ml_timedout = FALSE; 54 char mp_timedout = FALSE; 55 char target_url[256]; 56 57 if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) { 58 fprintf(stderr, "curl_global_init() failed\n"); 59 return TEST_ERR_MAJOR_BAD; 60 } 61 62 if ((m = curl_multi_init()) == NULL) { 63 fprintf(stderr, "curl_multi_init() failed\n"); 64 curl_global_cleanup(); 65 return TEST_ERR_MAJOR_BAD; 66 } 67 68 /* get NUM_HANDLES easy handles */ 69 for(i=0; i < NUM_HANDLES; i++) { 70 curl[i] = curl_easy_init(); 71 if(!curl[i]) { 72 fprintf(stderr, "curl_easy_init() failed " 73 "on handle #%d\n", i); 74 for (j=i-1; j >= 0; j--) { 75 curl_multi_remove_handle(m, curl[j]); 76 curl_easy_cleanup(curl[j]); 77 } 78 curl_multi_cleanup(m); 79 curl_global_cleanup(); 80 return TEST_ERR_MAJOR_BAD + i; 81 } 82 sprintf(target_url, "%s%04i", URL, i + 1); 83 target_url[sizeof(target_url) - 1] = '\0'; 84 85 res = curl_easy_setopt(curl[i], CURLOPT_URL, target_url); 86 if(res) { 87 fprintf(stderr, "curl_easy_setopt() failed " 88 "on handle #%d\n", i); 89 for (j=i; j >= 0; j--) { 90 curl_multi_remove_handle(m, curl[j]); 91 curl_easy_cleanup(curl[j]); 92 } 93 curl_multi_cleanup(m); 94 curl_global_cleanup(); 95 return TEST_ERR_MAJOR_BAD + i; 96 } 97 98 /* go verbose */ 99 res = curl_easy_setopt(curl[i], CURLOPT_VERBOSE, 1L); 100 if(res) { 101 fprintf(stderr, "curl_easy_setopt() failed " 102 "on handle #%d\n", i); 103 for (j=i; j >= 0; j--) { 104 curl_multi_remove_handle(m, curl[j]); 105 curl_easy_cleanup(curl[j]); 106 } 107 curl_multi_cleanup(m); 108 curl_global_cleanup(); 109 return TEST_ERR_MAJOR_BAD + i; 110 } 111 112 /* include headers */ 113 res = curl_easy_setopt(curl[i], CURLOPT_HEADER, 1L); 114 if(res) { 115 fprintf(stderr, "curl_easy_setopt() failed " 116 "on handle #%d\n", i); 117 for (j=i; j >= 0; j--) { 118 curl_multi_remove_handle(m, curl[j]); 119 curl_easy_cleanup(curl[j]); 120 } 121 curl_multi_cleanup(m); 122 curl_global_cleanup(); 123 return TEST_ERR_MAJOR_BAD + i; 124 } 125 126 /* add handle to multi */ 127 if ((res = (int)curl_multi_add_handle(m, curl[i])) != CURLM_OK) { 128 fprintf(stderr, "curl_multi_add_handle() failed, " 129 "on handle #%d with code %d\n", i, res); 130 curl_easy_cleanup(curl[i]); 131 for (j=i-1; j >= 0; j--) { 132 curl_multi_remove_handle(m, curl[j]); 133 curl_easy_cleanup(curl[j]); 134 } 135 curl_multi_cleanup(m); 136 curl_global_cleanup(); 137 return TEST_ERR_MAJOR_BAD + i; 138 } 139 } 140 141 curl_multi_setopt(m, CURLMOPT_PIPELINING, 1L); 142 143 ml_timedout = FALSE; 144 ml_start = tutil_tvnow(); 145 146 fprintf(stderr, "Start at URL 0\n"); 147 148 while (!done) { 149 fd_set rd, wr, exc; 150 int max_fd; 151 struct timeval interval; 152 153 interval.tv_sec = 1; 154 interval.tv_usec = 0; 155 156 if (tutil_tvdiff(tutil_tvnow(), ml_start) > 157 MAIN_LOOP_HANG_TIMEOUT) { 158 ml_timedout = TRUE; 159 break; 160 } 161 mp_timedout = FALSE; 162 mp_start = tutil_tvnow(); 163 164 while (res == CURLM_CALL_MULTI_PERFORM) { 165 res = (int)curl_multi_perform(m, &running); 166 if (tutil_tvdiff(tutil_tvnow(), mp_start) > 167 MULTI_PERFORM_HANG_TIMEOUT) { 168 mp_timedout = TRUE; 169 break; 170 } 171 if (running <= 0) { 172 done = TRUE; /* bail out */ 173 break; 174 } 175 } 176 if (mp_timedout || done) 177 break; 178 179 if (res != CURLM_OK) { 180 fprintf(stderr, "not okay???\n"); 181 break; 182 } 183 184 FD_ZERO(&rd); 185 FD_ZERO(&wr); 186 FD_ZERO(&exc); 187 max_fd = 0; 188 189 if (curl_multi_fdset(m, &rd, &wr, &exc, &max_fd) != CURLM_OK) { 190 fprintf(stderr, "unexpected failured of fdset.\n"); 191 res = 189; 192 break; 193 } 194 195 if (select_test(max_fd+1, &rd, &wr, &exc, &interval) == -1) { 196 fprintf(stderr, "bad select??\n"); 197 res = 195; 198 break; 199 } 200 201 res = CURLM_CALL_MULTI_PERFORM; 202 } 203 204 if (ml_timedout || mp_timedout) { 205 if (ml_timedout) fprintf(stderr, "ml_timedout\n"); 206 if (mp_timedout) fprintf(stderr, "mp_timedout\n"); 207 fprintf(stderr, "ABORTING TEST, since it seems " 208 "that it would have run forever.\n"); 209 res = TEST_ERR_RUNS_FOREVER; 210 } 211 212/* test_cleanup: */ 213 214 /* cleanup NUM_HANDLES easy handles */ 215 for(i=0; i < NUM_HANDLES; i++) { 216 curl_multi_remove_handle(m, curl[i]); 217 curl_easy_cleanup(curl[i]); 218 } 219 220 curl_multi_cleanup(m); 221 curl_global_cleanup(); 222 223 return res; 224} 225