summaryrefslogtreecommitdiffstats
path: root/src/tests/rnp_tests.cpp
blob: 910d2d8bb700aea6586fc61bc5e0759f17bc0ba7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
/*
 * Copyright (c) 2017-2019 [Ribose Inc](https://www.ribose.com).
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *
 * 1.  Redistributions of source code must retain the above copyright notice,
 *     this list of conditions and the following disclaimer.
 *
 * 2.  Redistributions in binary form must reproduce the above copyright notice,
 *     this list of conditions and the following disclaimer in the documentation
 *     and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "gtest/gtest.h"
#include <crypto/rng.h>
#include "rnp_tests.h"
#include "support.h"
#ifdef _WIN32
#include <cstdlib>
#include <crtdbg.h>
#endif

static char original_dir[PATH_MAX];

/*
 * Handler used to access DRBG.
 */
rnp::SecurityContext global_ctx;

#ifdef _WIN32
void
rnpInvalidParameterHandler(const wchar_t *expression,
                           const wchar_t *function,
                           const wchar_t *file,
                           unsigned int   line,
                           uintptr_t      pReserved)
{
    wprintf(L"%s:%d %s: invalid param: %s.\n", file, line, function, expression);
}
#endif

rnp_tests::rnp_tests() : m_dir(make_temp_dir())
{
#ifdef _WIN32
    _invalid_parameter_handler handler = rnpInvalidParameterHandler;
    _set_invalid_parameter_handler(handler);
    _CrtSetReportMode(_CRT_ASSERT, 0);
#endif
    /* We use LOGNAME in a few places within the tests
     * and it isn't always set in every environment.
     */
    if (!getenv("LOGNAME")) {
        setenv("LOGNAME", "test-user", 1);
    }
    EXPECT_EQ(0, setenv("HOME", m_dir, 1));
    EXPECT_EQ(0, chdir(m_dir));
    /* fully specified path works correctly here with cp and xcopy */
    std::string data_str = std::string(m_dir) + "/data";
    copy_recursively(getenv("RNP_TEST_DATA"), data_str.c_str());
}

rnp_tests::~rnp_tests()
{
    clean_temp_dir(m_dir);
    free(m_dir);
}

const char *
rnp_tests::original_dir() const
{
    return ::original_dir;
}

int
main(int argc, char *argv[])
{
    EXPECT_NE(nullptr, getcwd(original_dir, sizeof(original_dir)));
    ::testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}