| /* |
| * Copyright 2016-2017 The OpenSSL Project Authors. All Rights Reserved. |
| * |
| * Licensed under the OpenSSL license (the "License"). You may not use |
| * this file except in compliance with the License. You can obtain a copy |
| * in the file LICENSE in the source distribution or at |
| * https://www.openssl.org/source/license.html |
| */ |
| |
| #include <string.h> |
| #include <openssl/bio.h> |
| #include <openssl/crypto.h> |
| |
| #include "testutil.h" |
| |
| /* |
| * We use a proper main function here instead of the custom main from the |
| * test framework because the CRYPTO_mem_leaks_fp function cannot be called |
| * a second time without trying to use a null pointer. The test framework |
| * calls this function as part of its close down. |
| * |
| * A work around is to call putenv("OPENSSL_DEBUG_MEMORY=0"); before exiting |
| * but that is worse than avoiding the test framework's main. |
| */ |
| |
| int main(int argc, char *argv[]) |
| { |
| #ifndef OPENSSL_NO_CRYPTO_MDEBUG |
| char *p; |
| char *lost; |
| int noleak; |
| |
| p = getenv("OPENSSL_DEBUG_MEMORY"); |
| if (p != NULL && strcmp(p, "on") == 0) |
| CRYPTO_set_mem_debug(1); |
| CRYPTO_mem_ctrl(CRYPTO_MEM_CHECK_ON); |
| |
| lost = OPENSSL_malloc(3); |
| if (!TEST_ptr(lost)) |
| return EXIT_FAILURE; |
| |
| if (argv[1] && strcmp(argv[1], "freeit") == 0) { |
| OPENSSL_free(lost); |
| lost = NULL; |
| } |
| |
| noleak = CRYPTO_mem_leaks_fp(stderr); |
| /* If -1 return value something bad happened */ |
| if (!TEST_int_ne(noleak, -1)) |
| return EXIT_FAILURE; |
| |
| return TEST_int_eq(lost != NULL, noleak == 0) ? EXIT_SUCCESS : EXIT_FAILURE; |
| #else |
| return EXIT_SUCCESS; |
| #endif |
| } |