tests/libtest/lib564.c

   1 /*****************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * $Id$
   9  */
  10
  11 #include "test.h"
  12
  13 #include <sys/types.h>
  14 #include <sys/stat.h>
  15 #include <fcntl.h>
  16
  17 #include "testutil.h"
  18 #include "memdebug.h"
  19
  20 #define MAIN_LOOP_HANG_TIMEOUT     90 * 1000
  21 #define MULTI_PERFORM_HANG_TIMEOUT 60 * 1000
  22
  23 int test(char *URL)
  24 {
  25   int res = 0;
  26   CURL *curl;
  27   int running;
  28   char done=FALSE;
  29   CURLM *m = NULL;
  30   struct timeval ml_start;
  31   struct timeval mp_start;
  32   char ml_timedout = FALSE;
  33   char mp_timedout = FALSE;
  34
  35   if (curl_global_init(CURL_GLOBAL_ALL) != CURLE_OK) {
  36     fprintf(stderr, "curl_global_init() failed\n");
  37     return TEST_ERR_MAJOR_BAD;
  38   }
  39
  40   if ((curl = curl_easy_init()) == NULL) {
  41     fprintf(stderr, "curl_easy_init() failed\n");
  42     curl_global_cleanup();
  43     return TEST_ERR_MAJOR_BAD;
  44   }
  45
  46   test_setopt(curl, CURLOPT_URL, URL);
  47   test_setopt(curl, CURLOPT_VERBOSE, 1);
  48   test_setopt(curl, CURLOPT_PROXY, libtest_arg2);
  49   test_setopt(curl, CURLOPT_PROXYTYPE, CURLPROXY_SOCKS4);
  50
  51   if ((m = curl_multi_init()) == NULL) {
  52     fprintf(stderr, "curl_multi_init() failed\n");
  53     curl_easy_cleanup(curl);
  54     curl_global_cleanup();
  55     return TEST_ERR_MAJOR_BAD;
  56   }
  57
  58   if ((res = (int)curl_multi_add_handle(m, curl)) != CURLM_OK) {
  59     fprintf(stderr, "curl_multi_add_handle() failed, "
  60             "with code %d\n", res);
  61     curl_multi_cleanup(m);
  62     curl_easy_cleanup(curl);
  63     curl_global_cleanup();
  64     return TEST_ERR_MAJOR_BAD;
  65   }
  66
  67   ml_timedout = FALSE;
  68   ml_start = tutil_tvnow();
  69
  70   fprintf(stderr, "Start at URL 0\n");
  71
  72   while (!done) {
  73     fd_set rd, wr, exc;
  74     int max_fd;
  75     struct timeval interval;
  76
  77     interval.tv_sec = 1;
  78     interval.tv_usec = 0;
  79
  80     if (tutil_tvdiff(tutil_tvnow(), ml_start) >
  81         MAIN_LOOP_HANG_TIMEOUT) {
  82       ml_timedout = TRUE;
  83       break;
  84     }
  85     mp_timedout = FALSE;
  86     mp_start = tutil_tvnow();
  87
  88     while (res == CURLM_CALL_MULTI_PERFORM) {
  89       res = (int)curl_multi_perform(m, &running);
  90       if (tutil_tvdiff(tutil_tvnow(), mp_start) >
  91           MULTI_PERFORM_HANG_TIMEOUT) {
  92         mp_timedout = TRUE;
  93         break;
  94       }
  95       if (running <= 0) {
  96         done = TRUE; /* bail out */
  97         break;
  98       }
  99     }
 100     if (mp_timedout || done)
 101       break;
 102
 103     if (res != CURLM_OK) {
 104       fprintf(stderr, "not okay???\n");
 105       break;
 106     }
 107
 108     FD_ZERO(&rd);
 109     FD_ZERO(&wr);
 110     FD_ZERO(&exc);
 111     max_fd = 0;
 112
 113     if (curl_multi_fdset(m, &rd, &wr, &exc, &max_fd) != CURLM_OK) {
 114       fprintf(stderr, "unexpected failured of fdset.\n");
 115       res = 189;
 116       break;
 117     }
 118
 119     if (select_test(max_fd+1, &rd, &wr, &exc, &interval) == -1) {
 120       fprintf(stderr, "bad select??\n");
 121       res = 195;
 122       break;
 123     }
 124
 125     res = CURLM_CALL_MULTI_PERFORM;
 126   }
 127
 128   if (ml_timedout || mp_timedout) {
 129     if (ml_timedout) fprintf(stderr, "ml_timedout\n");
 130     if (mp_timedout) fprintf(stderr, "mp_timedout\n");
 131     fprintf(stderr, "ABORTING TEST, since it seems "
 132             "that it would have run forever.\n");
 133     res = TEST_ERR_RUNS_FOREVER;
 134   }
 135
 136 test_cleanup:
 137
 138   curl_easy_cleanup(curl);
 139   if(m)
 140     curl_multi_cleanup(m);
 141   curl_global_cleanup();
 142
 143   return res;
 144 }