tests/libtest/lib540.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2017, 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.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 /* This is the 'proxyauth.c' test app posted by Shmulik Regev on the libcurl
  23  * mailing list on 10 Jul 2007, converted to a test case.
  24  *
  25  * argv1 = URL
  26  * argv2 = proxy
  27  * argv3 = proxyuser:password
  28  * argv4 = host name to use for the custom Host: header
  29  */
  30
  31 #include "test.h"
  32
  33 #include <limits.h>
  34
  35 #include "testutil.h"
  36 #include "warnless.h"
  37 #include "memdebug.h"
  38
  39 #define TEST_HANG_TIMEOUT 60 * 1000
  40
  41 #define PROXY libtest_arg2
  42 #define PROXYUSERPWD libtest_arg3
  43 #define HOST test_argv[4]
  44
  45 #define NUM_HANDLES 2
  46
  47 static CURL *eh[NUM_HANDLES];
  48
  49 static int init(int num, CURLM *cm, const char *url, const char *userpwd,
  50                 struct curl_slist *headers)
  51 {
  52   int res = 0;
  53
  54   res_easy_init(eh[num]);
  55   if(res)
  56     goto init_failed;
  57
  58   res_easy_setopt(eh[num], CURLOPT_URL, url);
  59   if(res)
  60     goto init_failed;
  61
  62   res_easy_setopt(eh[num], CURLOPT_PROXY, PROXY);
  63   if(res)
  64     goto init_failed;
  65
  66   res_easy_setopt(eh[num], CURLOPT_PROXYUSERPWD, userpwd);
  67   if(res)
  68     goto init_failed;
  69
  70   res_easy_setopt(eh[num], CURLOPT_PROXYAUTH, (long)CURLAUTH_ANY);
  71   if(res)
  72     goto init_failed;
  73
  74   res_easy_setopt(eh[num], CURLOPT_VERBOSE, 1L);
  75   if(res)
  76     goto init_failed;
  77
  78   res_easy_setopt(eh[num], CURLOPT_HEADER, 1L);
  79   if(res)
  80     goto init_failed;
  81
  82   res_easy_setopt(eh[num], CURLOPT_HTTPHEADER, headers); /* custom Host: */
  83   if(res)
  84     goto init_failed;
  85
  86   res_multi_add_handle(cm, eh[num]);
  87   if(res)
  88     goto init_failed;
  89
  90   return 0; /* success */
  91
  92 init_failed:
  93
  94   curl_easy_cleanup(eh[num]);
  95   eh[num] = NULL;
  96
  97   return res; /* failure */
  98 }
  99
 100 static int loop(int num, CURLM *cm, const char *url, const char *userpwd,
 101                 struct curl_slist *headers)
 102 {
 103   CURLMsg *msg;
 104   long L;
 105   int Q, U = -1;
 106   fd_set R, W, E;
 107   struct timeval T;
 108   int res = 0;
 109
 110   res = init(num, cm, url, userpwd, headers);
 111   if(res)
 112     return res;
 113
 114   while(U) {
 115
 116     int M = -99;
 117
 118     res_multi_perform(cm, &U);
 119     if(res)
 120       return res;
 121
 122     res_test_timedout();
 123     if(res)
 124       return res;
 125
 126     if(U) {
 127       FD_ZERO(&R);
 128       FD_ZERO(&W);
 129       FD_ZERO(&E);
 130
 131       res_multi_fdset(cm, &R, &W, &E, &M);
 132       if(res)
 133         return res;
 134
 135       /* At this point, M is guaranteed to be greater or equal than -1. */
 136
 137       res_multi_timeout(cm, &L);
 138       if(res)
 139         return res;
 140
 141       /* At this point, L is guaranteed to be greater or equal than -1. */
 142
 143       if(L != -1) {
 144         int itimeout = (L > (long)INT_MAX) ? INT_MAX : (int)L;
 145         T.tv_sec = itimeout/1000;
 146         T.tv_usec = (itimeout%1000)*1000;
 147       }
 148       else {
 149         T.tv_sec = 5;
 150         T.tv_usec = 0;
 151       }
 152
 153       res_select_test(M + 1, &R, &W, &E, &T);
 154       if(res)
 155         return res;
 156     }
 157
 158     while((msg = curl_multi_info_read(cm, &Q)) != NULL) {
 159       if(msg->msg == CURLMSG_DONE) {
 160         int i;
 161         CURL *e = msg->easy_handle;
 162         fprintf(stderr, "R: %d - %s\n", (int)msg->data.result,
 163                 curl_easy_strerror(msg->data.result));
 164         curl_multi_remove_handle(cm, e);
 165         curl_easy_cleanup(e);
 166         for(i = 0; i < NUM_HANDLES; i++) {
 167           if(eh[i] == e) {
 168             eh[i] = NULL;
 169             break;
 170           }
 171         }
 172       }
 173       else
 174         fprintf(stderr, "E: CURLMsg (%d)\n", (int)msg->msg);
 175     }
 176
 177     res_test_timedout();
 178     if(res)
 179       return res;
 180   }
 181
 182   return 0; /* success */
 183 }
 184
 185 int test(char *URL)
 186 {
 187   CURLM *cm = NULL;
 188   struct curl_slist *headers = NULL;
 189   char buffer[246]; /* naively fixed-size */
 190   int res = 0;
 191   int i;
 192
 193   for(i = 0; i < NUM_HANDLES; i++)
 194     eh[i] = NULL;
 195
 196   start_test_timing();
 197
 198   if(test_argc < 4)
 199     return 99;
 200
 201   snprintf(buffer, sizeof(buffer), "Host: %s", HOST);
 202
 203   /* now add a custom Host: header */
 204   headers = curl_slist_append(headers, buffer);
 205   if(!headers) {
 206     fprintf(stderr, "curl_slist_append() failed\n");
 207     return TEST_ERR_MAJOR_BAD;
 208   }
 209
 210   res_global_init(CURL_GLOBAL_ALL);
 211   if(res) {
 212     curl_slist_free_all(headers);
 213     return res;
 214   }
 215
 216   res_multi_init(cm);
 217   if(res) {
 218     curl_global_cleanup();
 219     curl_slist_free_all(headers);
 220     return res;
 221   }
 222
 223   res = loop(0, cm, URL, PROXYUSERPWD, headers);
 224   if(res)
 225     goto test_cleanup;
 226
 227   fprintf(stderr, "lib540: now we do the request again\n");
 228
 229   res = loop(1, cm, URL, PROXYUSERPWD, headers);
 230
 231 test_cleanup:
 232
 233   /* proper cleanup sequence - type PB */
 234
 235   for(i = 0; i < NUM_HANDLES; i++) {
 236     curl_multi_remove_handle(cm, eh[i]);
 237     curl_easy_cleanup(eh[i]);
 238   }
 239
 240   curl_multi_cleanup(cm);
 241   curl_global_cleanup();
 242
 243   curl_slist_free_all(headers);
 244
 245   return res;
 246 }