quickjs-tart

lib540.c (5779B)

---

 /***************************************************************************
  *                                  _   _ ____  _
  *  Project                     ___| | | |  _ \| |
  *                             / __| | | | |_) | |
  *                            | (__| |_| |  _ <| |___
  *                             \___|\___/|_| \_\_____|
  *
  * Copyright (C) Daniel Stenberg, <daniel@haxx.se>, et al.
  *
  * This software is licensed as described in the file COPYING, which
  * you should have received as part of this distribution. The terms
  * are also available at https://curl.se/docs/copyright.html.
  *
  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  * copies of the Software, and permit persons to whom the Software is
  * furnished to do so, under the terms of the COPYING file.
  *
  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  * KIND, either express or implied.
  *
  * SPDX-License-Identifier: curl
  *
  ***************************************************************************/
 /* This is the 'proxyauth.c' test app posted by Shmulik Regev on the libcurl
  * mailing list on 10 Jul 2007, converted to a test case.
  *
  * argv1 = URL
  * argv2 = proxy
  * argv3 = proxyuser:password
  * argv4 = host name to use for the custom Host: header
  */
 
 #include "first.h"
 
 #include "memdebug.h"
 
 #define PROXY libtest_arg2
 #define PROXYUSERPWD libtest_arg3
 #define HOST test_argv[4]
 
 static CURL *testeh[2];
 
 static CURLcode init(int num, CURLM *cm, const char *url, const char *userpwd,
                      struct curl_slist *headers)
 {
   CURLcode res = CURLE_OK;
 
   res_easy_init(testeh[num]);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_URL, url);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_PROXY, PROXY);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_PROXYUSERPWD, userpwd);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_PROXYAUTH, (long)CURLAUTH_ANY);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_VERBOSE, 1L);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_HEADER, 1L);
   if(res)
     goto init_failed;
 
   res_easy_setopt(testeh[num], CURLOPT_HTTPHEADER, headers); /* custom Host: */
   if(res)
     goto init_failed;
 
   res_multi_add_handle(cm, testeh[num]);
   if(res)
     goto init_failed;
 
   return CURLE_OK; /* success */
 
 init_failed:
 
   curl_easy_cleanup(testeh[num]);
   testeh[num] = NULL;
 
   return res; /* failure */
 }
 
 static CURLcode loop(int num, CURLM *cm, const char *url, const char *userpwd,
                      struct curl_slist *headers)
 {
   CURLMsg *msg;
   long L;
   int Q, U = -1;
   fd_set R, W, E;
   struct timeval T;
   CURLcode res = CURLE_OK;
 
   res = init(num, cm, url, userpwd, headers);
   if(res)
     return res;
 
   while(U) {
 
     int M = -99;
 
     res_multi_perform(cm, &U);
     if(res)
       return res;
 
     res_test_timedout();
     if(res)
       return res;
 
     if(U) {
       FD_ZERO(&R);
       FD_ZERO(&W);
       FD_ZERO(&E);
 
       res_multi_fdset(cm, &R, &W, &E, &M);
       if(res)
         return res;
 
       /* At this point, M is guaranteed to be greater or equal than -1. */
 
       res_multi_timeout(cm, &L);
       if(res)
         return res;
 
       /* At this point, L is guaranteed to be greater or equal than -1. */
 
       if(L != -1) {
         int itimeout;
 #if LONG_MAX > INT_MAX
         itimeout = (L > INT_MAX) ? INT_MAX : (int)L;
 #else
         itimeout = (int)L;
 #endif
         T.tv_sec = itimeout/1000;
         T.tv_usec = (itimeout%1000)*1000;
       }
       else {
         T.tv_sec = 5;
         T.tv_usec = 0;
       }
 
       res_select_test(M + 1, &R, &W, &E, &T);
       if(res)
         return res;
     }
 
     while(1) {
       msg = curl_multi_info_read(cm, &Q);
       if(!msg)
         break;
       if(msg->msg == CURLMSG_DONE) {
         size_t i;
         CURL *e = msg->easy_handle;
         curl_mfprintf(stderr, "R: %d - %s\n", (int)msg->data.result,
                       curl_easy_strerror(msg->data.result));
         curl_multi_remove_handle(cm, e);
         curl_easy_cleanup(e);
         for(i = 0; i < CURL_ARRAYSIZE(testeh); i++) {
           if(testeh[i] == e) {
             testeh[i] = NULL;
             break;
           }
         }
       }
       else
         curl_mfprintf(stderr, "E: CURLMsg (%d)\n", (int)msg->msg);
     }
 
     res_test_timedout();
     if(res)
       return res;
   }
 
   return CURLE_OK;
 }
 
 static CURLcode test_lib540(char *URL)
 {
   CURLM *cm = NULL;
   struct curl_slist *headers = NULL;
   char buffer[246]; /* naively fixed-size */
   CURLcode res = CURLE_OK;
   size_t i;
 
   for(i = 0; i < CURL_ARRAYSIZE(testeh); i++)
     testeh[i] = NULL;
 
   start_test_timing();
 
   if(test_argc < 4)
     return TEST_ERR_MAJOR_BAD;
 
   curl_msnprintf(buffer, sizeof(buffer), "Host: %s", HOST);
 
   /* now add a custom Host: header */
   headers = curl_slist_append(headers, buffer);
   if(!headers) {
     curl_mfprintf(stderr, "curl_slist_append() failed\n");
     return TEST_ERR_MAJOR_BAD;
   }
 
   res_global_init(CURL_GLOBAL_ALL);
   if(res) {
     curl_slist_free_all(headers);
     return res;
   }
 
   res_multi_init(cm);
   if(res) {
     curl_global_cleanup();
     curl_slist_free_all(headers);
     return res;
   }
 
   res = loop(0, cm, URL, PROXYUSERPWD, headers);
   if(res)
     goto test_cleanup;
 
   curl_mfprintf(stderr, "lib540: now we do the request again\n");
 
   res = loop(1, cm, URL, PROXYUSERPWD, headers);
 
 test_cleanup:
 
   /* proper cleanup sequence - type PB */
 
   for(i = 0; i < CURL_ARRAYSIZE(testeh); i++) {
     curl_multi_remove_handle(cm, testeh[i]);
     curl_easy_cleanup(testeh[i]);
   }
 
   curl_multi_cleanup(cm);
   curl_global_cleanup();
 
   curl_slist_free_all(headers);
 
   return res;
 }