Annotation of embedaddon/curl/tests/libtest/lib540.c, revision 1.1.1.1

1.1       misho       1: /***************************************************************************
                      2:  *                                  _   _ ____  _
                      3:  *  Project                     ___| | | |  _ \| |
                      4:  *                             / __| | | | |_) | |
                      5:  *                            | (__| |_| |  _ <| |___
                      6:  *                             \___|\___/|_| \_\_____|
                      7:  *
                      8:  * Copyright (C) 1998 - 2018, 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:   msnprintf(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: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>