Home | History | Annotate | Line # | Download | only in test 
      1 /*
      2  * Copyright 2016-2025 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 
     10 #include <string.h>
     11 #include <openssl/bio.h>
     12 #include <openssl/crypto.h>
     13 
     14 #include "testutil.h"
     15 
     16 /* __has_feature is a clang-ism, while __SANITIZE_ADDRESS__ is a gcc-ism */
     17 #if defined(__has_feature)
     18 # if __has_feature(address_sanitizer)
     19 #  define __SANITIZE_ADDRESS__ 1
     20 # endif
     21 #endif
     22 /* If __SANITIZE_ADDRESS__ isn't defined, define it to be false */
     23 /* Leak detection is not yet supported with MSVC on Windows, so */
     24 /* set __SANITIZE_ADDRESS__ to false in this case as well.      */
     25 #if !defined(__SANITIZE_ADDRESS__) || defined(_MSC_VER)
     26 # undef __SANITIZE_ADDRESS__
     27 # define __SANITIZE_ADDRESS__ 0
     28 #endif
     29 
     30 /*
     31  * We use a proper main function here instead of the custom main from the
     32  * test framework to avoid CRYPTO_mem_leaks stuff.
     33  */
     34 
     35 int main(int argc, char *argv[])
     36 {
     37 #if __SANITIZE_ADDRESS__
     38     int exitcode = EXIT_SUCCESS;
     39 #else
     40     /*
     41      * When we don't sanitize, we set the exit code to what we would expect
     42      * to get when we are sanitizing.  This makes it easy for wrapper scripts
     43      * to detect that we get the result we expect.
     44      */
     45     int exitcode = EXIT_FAILURE;
     46 #endif
     47     char *volatile lost;
     48 
     49     lost = OPENSSL_malloc(3);
     50     if (!TEST_ptr(lost))
     51         return EXIT_FAILURE;
     52 
     53     strcpy(lost, "ab");
     54 
     55     if (argv[1] && strcmp(argv[1], "freeit") == 0) {
     56         OPENSSL_free(lost);
     57         exitcode = EXIT_SUCCESS;
     58     }
     59 
     60     lost = NULL;
     61     return exitcode;
     62 }
     63