tests/libtest/libntlmconnect.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 2012 - 2016, 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 #include "test.h"
  23
  24 #ifdef HAVE_LIMITS_H
  25 #include <limits.h>
  26 #endif
  27 #include <assert.h>
  28
  29 #include "testutil.h"
  30 #include "warnless.h"
  31 #include "memdebug.h"
  32
  33 #define TEST_HANG_TIMEOUT 5 * 1000
  34 #define MAX_EASY_HANDLES 3
  35
  36 static CURL *easy[MAX_EASY_HANDLES];
  37 static curl_socket_t sockets[MAX_EASY_HANDLES];
  38 static int res = 0;
  39
  40 static size_t callback(char* ptr, size_t size, size_t nmemb, void* data)
  41 {
  42   ssize_t idx = ((CURL **) data) - easy;
  43   curl_socket_t sock;
  44   long longdata;
  45   CURLcode code;
  46
  47   const size_t failure = (size * nmemb) ? 0 : 1;
  48
  49   char *output = malloc(size * nmemb + 1);
  50   if(!output) {
  51     fprintf(stderr, "output, malloc() failed\n");
  52     res = TEST_ERR_MAJOR_BAD;
  53     return failure;
  54   }
  55
  56   memcpy(output, ptr, size * nmemb);
  57   output[size * nmemb] = '\0';
  58   fprintf(stdout, "%s", output);
  59   free(output);
  60
  61   /* Get socket being used for this easy handle, otherwise CURL_SOCKET_BAD */
  62   code = curl_easy_getinfo(easy[idx], CURLINFO_LASTSOCKET, &longdata);
  63   if(CURLE_OK != code) {
  64     fprintf(stderr, "%s:%d curl_easy_getinfo() failed, "
  65             "with code %d (%s)\n",
  66             __FILE__, __LINE__, (int)code, curl_easy_strerror(code));
  67     res = TEST_ERR_MAJOR_BAD;
  68     return failure;
  69   }
  70   if(longdata == -1L)
  71     sock = CURL_SOCKET_BAD;
  72   else
  73     sock = (curl_socket_t)longdata;
  74
  75   if(sock != CURL_SOCKET_BAD) {
  76     /* Track relationship between this easy handle and the socket. */
  77     if(sockets[idx] == CURL_SOCKET_BAD) {
  78       /* An easy handle without previous socket, record the socket. */
  79       sockets[idx] = sock;
  80     }
  81     else if(sock != sockets[idx]) {
  82       /* An easy handle with a socket different to previously
  83          tracked one, log and fail right away. Known bug #37. */
  84       fprintf(stderr, "Handle %d started on socket %d and moved to %d\n",
  85               curlx_sztosi(idx), (int)sockets[idx], (int)sock);
  86       res = TEST_ERR_MAJOR_BAD;
  87       return failure;
  88     }
  89   }
  90   return size * nmemb;
  91 }
  92
  93 enum HandleState {
  94   ReadyForNewHandle,
  95   NeedSocketForNewHandle,
  96   NoMoreHandles
  97 };
  98
  99 int test(char *url)
 100 {
 101   CURLM *multi = NULL;
 102   int running;
 103   int i, j;
 104   int num_handles = 0;
 105   enum HandleState state = ReadyForNewHandle;
 106   size_t urllen = strlen(url) + 4 + 1;
 107   char* full_url = malloc(urllen);
 108
 109   start_test_timing();
 110
 111   if(!full_url) {
 112     fprintf(stderr, "Not enough memory for full url\n");
 113     return TEST_ERR_MAJOR_BAD;
 114   }
 115
 116   for(i = 0; i < MAX_EASY_HANDLES; ++i) {
 117     easy[i] = NULL;
 118     sockets[i] = CURL_SOCKET_BAD;
 119   }
 120
 121   res_global_init(CURL_GLOBAL_ALL);
 122   if(res) {
 123     free(full_url);
 124     return res;
 125   }
 126
 127   multi_init(multi);
 128
 129 #ifdef USE_PIPELINING
 130   multi_setopt(multi, CURLMOPT_PIPELINING, 1L);
 131   multi_setopt(multi, CURLMOPT_MAX_HOST_CONNECTIONS, 5L);
 132   multi_setopt(multi, CURLMOPT_MAX_TOTAL_CONNECTIONS, 10L);
 133 #endif
 134
 135   for(;;) {
 136     struct timeval interval;
 137     fd_set fdread;
 138     fd_set fdwrite;
 139     fd_set fdexcep;
 140     long timeout = -99;
 141     int maxfd = -99;
 142     bool found_new_socket = FALSE;
 143
 144     /* Start a new handle if we aren't at the max */
 145     if(state == ReadyForNewHandle) {
 146       easy_init(easy[num_handles]);
 147
 148       if(num_handles % 3 == 2) {
 149         snprintf(full_url, urllen, "%s0200", url);
 150         easy_setopt(easy[num_handles], CURLOPT_HTTPAUTH, CURLAUTH_NTLM);
 151       }
 152       else {
 153         snprintf(full_url, urllen, "%s0100", url);
 154         easy_setopt(easy[num_handles], CURLOPT_HTTPAUTH, CURLAUTH_BASIC);
 155       }
 156       easy_setopt(easy[num_handles], CURLOPT_FRESH_CONNECT, 1L);
 157       easy_setopt(easy[num_handles], CURLOPT_URL, full_url);
 158       easy_setopt(easy[num_handles], CURLOPT_VERBOSE, 1L);
 159       easy_setopt(easy[num_handles], CURLOPT_HTTPGET, 1L);
 160       easy_setopt(easy[num_handles], CURLOPT_USERPWD, "testuser:testpass");
 161       easy_setopt(easy[num_handles], CURLOPT_WRITEFUNCTION, callback);
 162       easy_setopt(easy[num_handles], CURLOPT_WRITEDATA, easy + num_handles);
 163       easy_setopt(easy[num_handles], CURLOPT_HEADER, 1L);
 164
 165       multi_add_handle(multi, easy[num_handles]);
 166       num_handles += 1;
 167       state = NeedSocketForNewHandle;
 168     }
 169
 170     multi_perform(multi, &running);
 171
 172     abort_on_test_timeout();
 173
 174     if(!running && state == NoMoreHandles)
 175       break; /* done */
 176
 177     FD_ZERO(&fdread);
 178     FD_ZERO(&fdwrite);
 179     FD_ZERO(&fdexcep);
 180
 181     multi_fdset(multi, &fdread, &fdwrite, &fdexcep, &maxfd);
 182
 183     /* At this point, maxfd is guaranteed to be greater or equal than -1. */
 184
 185     /* Any socket which is new in fdread is associated with the new handle */
 186     for(i = 0; i <= maxfd; ++i) {
 187       bool socket_exists = FALSE;
 188       curl_socket_t curfd = (curl_socket_t)i;
 189
 190       if(!FD_ISSET(curfd, &fdread)) {
 191         continue;
 192       }
 193
 194       /* Check if this socket was already detected for an earlier handle (or
 195          for this handle, num_handles-1, in the callback */
 196       for(j = 0; j < num_handles; ++j) {
 197         if(sockets[j] == curfd) {
 198           socket_exists = TRUE;
 199           break;
 200         }
 201       }
 202       if(socket_exists) {
 203         continue;
 204       }
 205
 206       if(found_new_socket || state != NeedSocketForNewHandle) {
 207         fprintf(stderr, "Unexpected new socket\n");
 208         res = TEST_ERR_MAJOR_BAD;
 209         goto test_cleanup;
 210       }
 211
 212       /* Now we know the socket is for the most recent handle, num_handles-1 */
 213       if(sockets[num_handles-1] != CURL_SOCKET_BAD) {
 214         /* A socket for this handle was already detected in the callback; if it
 215            matched socket_exists should be true and we would never get here */
 216         assert(curfd != sockets[num_handles-1]);
 217         fprintf(stderr, "Handle %d wrote to socket %d then detected on %d\n",
 218                 num_handles-1, (int)sockets[num_handles-1], (int)curfd);
 219         res = TEST_ERR_MAJOR_BAD;
 220         goto test_cleanup;
 221       }
 222       else {
 223         sockets[num_handles-1] = curfd;
 224         found_new_socket = TRUE;
 225         /* continue to make sure there's only one new handle */
 226       }
 227     }
 228
 229     if(state == NeedSocketForNewHandle) {
 230       if(maxfd != -1 && !found_new_socket) {
 231         fprintf(stderr, "Warning: socket did not open immediately for new "
 232                 "handle (trying again)\n");
 233         continue;
 234       }
 235       state = num_handles < MAX_EASY_HANDLES ? ReadyForNewHandle
 236                                              : NoMoreHandles;
 237     }
 238
 239     multi_timeout(multi, &timeout);
 240
 241     /* At this point, timeout is guaranteed to be greater or equal than -1. */
 242
 243     fprintf(stderr, "%s:%d num_handles %d timeout %ld\n",
 244             __FILE__, __LINE__, num_handles, timeout);
 245
 246     if(timeout != -1L) {
 247       int itimeout = (timeout > (long)INT_MAX) ? INT_MAX : (int)timeout;
 248       interval.tv_sec = itimeout/1000;
 249       interval.tv_usec = (itimeout%1000)*1000;
 250     }
 251     else {
 252       interval.tv_sec = TEST_HANG_TIMEOUT/1000+1;
 253       interval.tv_usec = 0;
 254
 255       /* if there's no timeout and we get here on the last handle, we may
 256          already have read the last part of the stream so waiting makes no
 257          sense */
 258       if(!running && num_handles == MAX_EASY_HANDLES) {
 259         break;
 260       }
 261     }
 262
 263     select_test(maxfd+1, &fdread, &fdwrite, &fdexcep, &interval);
 264
 265     abort_on_test_timeout();
 266   }
 267
 268 test_cleanup:
 269
 270   /* proper cleanup sequence - type PB */
 271
 272   for(i = 0; i < MAX_EASY_HANDLES; i++) {
 273     curl_multi_remove_handle(multi, easy[i]);
 274     curl_easy_cleanup(easy[i]);
 275   }
 276
 277   curl_multi_cleanup(multi);
 278   curl_global_cleanup();
 279
 280   free(full_url);
 281
 282   return res;
 283 }