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

1.1       misho       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: /*
                     23:  * This code sets up multiple easy handles that transfer a single file from
                     24:  * the same URL, in a serial manner after each other. Due to the connection
                     25:  * sharing within the multi handle all transfers are performed on the same
                     26:  * persistent connection.
                     27:  *
                     28:  * This source code is used for lib526, lib527 and lib532 with only #ifdefs
                     29:  * controlling the small differences.
                     30:  *
                     31:  * - lib526 closes all easy handles after
                     32:  *   they all have transferred the file over the single connection
                     33:  * - lib527 closes each easy handle after each single transfer.
                     34:  * - lib532 uses only a single easy handle that is removed, reset and then
                     35:  *   re-added for each transfer
                     36:  *
                     37:  * Test case 526, 527 and 532 use FTP, while test 528 uses the lib526 tool but
                     38:  * with HTTP.
                     39:  */
                     40: 
                     41: #include "test.h"
                     42: 
                     43: #include <fcntl.h>
                     44: 
                     45: #include "testutil.h"
                     46: #include "warnless.h"
                     47: #include "memdebug.h"
                     48: 
                     49: #define TEST_HANG_TIMEOUT 60 * 1000
                     50: 
                     51: #define NUM_HANDLES 4
                     52: 
                     53: int test(char *URL)
                     54: {
                     55:   int res = 0;
                     56:   CURL *curl[NUM_HANDLES];
                     57:   int running;
                     58:   CURLM *m = NULL;
                     59:   int current = 0;
                     60:   int i;
                     61: 
                     62:   for(i = 0; i < NUM_HANDLES; i++)
                     63:     curl[i] = NULL;
                     64: 
                     65:   start_test_timing();
                     66: 
                     67:   global_init(CURL_GLOBAL_ALL);
                     68: 
                     69:   /* get NUM_HANDLES easy handles */
                     70:   for(i = 0; i < NUM_HANDLES; i++) {
                     71:     easy_init(curl[i]);
                     72:     /* specify target */
                     73:     easy_setopt(curl[i], CURLOPT_URL, URL);
                     74:     /* go verbose */
                     75:     easy_setopt(curl[i], CURLOPT_VERBOSE, 1L);
                     76:   }
                     77: 
                     78:   multi_init(m);
                     79: 
                     80:   multi_add_handle(m, curl[current]);
                     81: 
                     82:   fprintf(stderr, "Start at URL 0\n");
                     83: 
                     84:   for(;;) {
                     85:     struct timeval interval;
                     86:     fd_set rd, wr, exc;
                     87:     int maxfd = -99;
                     88: 
                     89:     interval.tv_sec = 1;
                     90:     interval.tv_usec = 0;
                     91: 
                     92:     multi_perform(m, &running);
                     93: 
                     94:     abort_on_test_timeout();
                     95: 
                     96:     if(!running) {
                     97: #ifdef LIB527
                     98:       /* NOTE: this code does not remove the handle from the multi handle
                     99:          here, which would be the nice, sane and documented way of working.
                    100:          This however tests that the API survives this abuse gracefully. */
                    101:       curl_easy_cleanup(curl[current]);
                    102:       curl[current] = NULL;
                    103: #endif
                    104:       if(++current < NUM_HANDLES) {
                    105:         fprintf(stderr, "Advancing to URL %d\n", current);
                    106: #ifdef LIB532
                    107:         /* first remove the only handle we use */
                    108:         curl_multi_remove_handle(m, curl[0]);
                    109: 
                    110:         /* make us re-use the same handle all the time, and try resetting
                    111:            the handle first too */
                    112:         curl_easy_reset(curl[0]);
                    113:         easy_setopt(curl[0], CURLOPT_URL, URL);
                    114:         /* go verbose */
                    115:         easy_setopt(curl[0], CURLOPT_VERBOSE, 1L);
                    116: 
                    117:         /* re-add it */
                    118:         multi_add_handle(m, curl[0]);
                    119: #else
                    120:         multi_add_handle(m, curl[current]);
                    121: #endif
                    122:       }
                    123:       else {
                    124:         break; /* done */
                    125:       }
                    126:     }
                    127: 
                    128:     FD_ZERO(&rd);
                    129:     FD_ZERO(&wr);
                    130:     FD_ZERO(&exc);
                    131: 
                    132:     multi_fdset(m, &rd, &wr, &exc, &maxfd);
                    133: 
                    134:     /* At this point, maxfd is guaranteed to be greater or equal than -1. */
                    135: 
                    136:     select_test(maxfd + 1, &rd, &wr, &exc, &interval);
                    137: 
                    138:     abort_on_test_timeout();
                    139:   }
                    140: 
                    141: test_cleanup:
                    142: 
                    143: #if defined(LIB526)
                    144: 
                    145:   /* test 526 and 528 */
                    146:   /* proper cleanup sequence - type PB */
                    147: 
                    148:   for(i = 0; i < NUM_HANDLES; i++) {
                    149:     curl_multi_remove_handle(m, curl[i]);
                    150:     curl_easy_cleanup(curl[i]);
                    151:   }
                    152:   curl_multi_cleanup(m);
                    153:   curl_global_cleanup();
                    154: 
                    155: #elif defined(LIB527)
                    156: 
                    157:   /* test 527 */
                    158: 
                    159:   /* Upon non-failure test flow the easy's have already been cleanup'ed. In
                    160:      case there is a failure we arrive here with easy's that have not been
                    161:      cleanup'ed yet, in this case we have to cleanup them or otherwise these
                    162:      will be leaked, let's use undocumented cleanup sequence - type UB */
                    163: 
                    164:   if(res)
                    165:     for(i = 0; i < NUM_HANDLES; i++)
                    166:       curl_easy_cleanup(curl[i]);
                    167: 
                    168:   curl_multi_cleanup(m);
                    169:   curl_global_cleanup();
                    170: 
                    171: #elif defined(LIB532)
                    172: 
                    173:   /* test 532 */
                    174:   /* undocumented cleanup sequence - type UB */
                    175: 
                    176:   for(i = 0; i < NUM_HANDLES; i++)
                    177:     curl_easy_cleanup(curl[i]);
                    178:   curl_multi_cleanup(m);
                    179:   curl_global_cleanup();
                    180: 
                    181: #endif
                    182: 
                    183:   return res;
                    184: }

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