789dfc478e
It isn't easy to use the test framework since it turns memory debugging on as well and the CRYPTO_mem_leaks_fp function cannot be called twice. Reviewed-by: Richard Levitte <levitte@openssl.org> Reviewed-by: Matt Caswell <matt@openssl.org> (Merged from https://github.com/openssl/openssl/pull/3169)
56 lines
1.6 KiB
C
56 lines
1.6 KiB
C
/*
|
|
* 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
|
|
}
|