1/*
2 * Copyright 2016-2021 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
35int 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 *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