unit1608.c 2.0 KB

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