Home | History | Annotate | Download | only in unit
      1 /***************************************************************************
      2  *                                  _   _ ____  _
      3  *  Project                     ___| | | |  _ \| |
      4  *                             / __| | | | |_) | |
      5  *                            | (__| |_| |  _ <| |___
      6  *                             \___|\___/|_| \_\_____|
      7  *
      8  * Copyright (C) 1998 - 2019, Daniel Stenberg, <daniel (at) 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 "curlcheck.h"
     23 
     24 #include "hostip.h"
     25 
     26 CURLcode Curl_shuffle_addr(struct Curl_easy *data,
     27                            Curl_addrinfo **addr);
     28 
     29 #define NUM_ADDRS 8
     30 static struct Curl_addrinfo addrs[NUM_ADDRS];
     31 
     32 static CURLcode unit_setup(void)
     33 {
     34   int i;
     35   for(i = 0; i < NUM_ADDRS - 1; i++) {
     36     addrs[i].ai_next = &addrs[i + 1];
     37   }
     38 
     39   return CURLE_OK;
     40 }
     41 
     42 static void unit_stop(void)
     43 {
     44 
     45 }
     46 
     47 UNITTEST_START
     48 {
     49   int i;
     50   CURLcode code;
     51   struct Curl_addrinfo* addrhead = addrs;
     52 
     53   struct Curl_easy *easy = curl_easy_init();
     54   abort_unless(easy, "out of memory");
     55 
     56   code = curl_easy_setopt(easy, CURLOPT_DNS_SHUFFLE_ADDRESSES, 1L);
     57   abort_unless(code == CURLE_OK, "curl_easy_setopt failed");
     58 
     59   /* Shuffle repeatedly and make sure that the list changes */
     60   for(i = 0; i < 10; i++) {
     61     if(CURLE_OK != Curl_shuffle_addr(easy, &addrhead))
     62       break;
     63     if(addrhead != addrs)
     64       break;
     65   }
     66 
     67   curl_easy_cleanup(easy);
     68 
     69   abort_unless(addrhead != addrs, "addresses are not being reordered");
     70 
     71   return 0;
     72 }
     73 UNITTEST_STOP
     74