mirror of https://github.com/openssl/openssl.git
				
				
				
			
		
			
				
	
	
		
			57 lines
		
	
	
		
			1.6 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			57 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
 | |
| }
 |