2
0

memleaktest.c 1.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  1. /*
  2. * Copyright 2016-2017 The OpenSSL Project Authors. All Rights Reserved.
  3. *
  4. * Licensed under the Apache License 2.0 (the "License"). You may not use
  5. * this file except in compliance with the License. You can obtain a copy
  6. * in the file LICENSE in the source distribution or at
  7. * https://www.openssl.org/source/license.html
  8. */
  9. #include <string.h>
  10. #include <openssl/bio.h>
  11. #include <openssl/crypto.h>
  12. #include "testutil.h"
  13. /* __has_feature is a clang-ism, while __SANITIZE_ADDRESS__ is a gcc-ism */
  14. #if defined(__has_feature)
  15. # if __has_feature(address_sanitizer)
  16. # define __SANITIZE_ADDRESS__ 1
  17. # endif
  18. #endif
  19. /* If __SANITIZE_ADDRESS__ isn't defined, define it to be false */
  20. #ifndef __SANITIZE_ADDRESS__
  21. # define __SANITIZE_ADDRESS__ 0
  22. #endif
  23. /*
  24. * We use a proper main function here instead of the custom main from the
  25. * test framework to avoid CRYPTO_mem_leaks stuff.
  26. */
  27. int main(int argc, char *argv[])
  28. {
  29. #if __SANITIZE_ADDRESS__
  30. int exitcode = EXIT_SUCCESS;
  31. #else
  32. /*
  33. * When we don't sanitize, we set the exit code to what we would expect
  34. * to get when we are sanitizing. This makes it easy for wrapper scripts
  35. * to detect that we get the result we expect.
  36. */
  37. int exitcode = EXIT_FAILURE;
  38. #endif
  39. char *lost;
  40. lost = OPENSSL_malloc(3);
  41. if (!TEST_ptr(lost))
  42. return EXIT_FAILURE;
  43. strcpy(lost, "ab");
  44. if (argv[1] && strcmp(argv[1], "freeit") == 0) {
  45. OPENSSL_free(lost);
  46. exitcode = EXIT_SUCCESS;
  47. }
  48. lost = NULL;
  49. return exitcode;
  50. }