unit1608.c 2.1 KB

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