forked from OSchip/llvm-project
110 lines
3.1 KiB
C++
110 lines
3.1 KiB
C++
//===-- harness.h -----------------------------------------------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef GWP_ASAN_TESTS_HARNESS_H_
|
|
#define GWP_ASAN_TESTS_HARNESS_H_
|
|
|
|
#include <stdarg.h>
|
|
|
|
#if defined(__Fuchsia__)
|
|
#include <zxtest/zxtest.h>
|
|
using Test = ::zxtest::Test;
|
|
#else
|
|
#include "gtest/gtest.h"
|
|
using Test = ::testing::Test;
|
|
#endif
|
|
|
|
#include "gwp_asan/guarded_pool_allocator.h"
|
|
#include "gwp_asan/optional/backtrace.h"
|
|
#include "gwp_asan/optional/printf.h"
|
|
#include "gwp_asan/optional/segv_handler.h"
|
|
#include "gwp_asan/options.h"
|
|
|
|
namespace gwp_asan {
|
|
namespace test {
|
|
// This printf-function getter allows other platforms (e.g. Android) to define
|
|
// their own signal-safe Printf function. In LLVM, we use
|
|
// `optional/printf_sanitizer_common.cpp` which supplies the __sanitizer::Printf
|
|
// for this purpose.
|
|
Printf_t getPrintfFunction();
|
|
|
|
// First call returns true, all the following calls return false.
|
|
bool OnlyOnce();
|
|
|
|
}; // namespace test
|
|
}; // namespace gwp_asan
|
|
|
|
class DefaultGuardedPoolAllocator : public Test {
|
|
public:
|
|
void SetUp() override {
|
|
gwp_asan::options::Options Opts;
|
|
Opts.setDefaults();
|
|
MaxSimultaneousAllocations = Opts.MaxSimultaneousAllocations;
|
|
|
|
Opts.InstallForkHandlers = gwp_asan::test::OnlyOnce();
|
|
GPA.init(Opts);
|
|
}
|
|
|
|
void TearDown() override { GPA.uninitTestOnly(); }
|
|
|
|
protected:
|
|
gwp_asan::GuardedPoolAllocator GPA;
|
|
decltype(gwp_asan::options::Options::MaxSimultaneousAllocations)
|
|
MaxSimultaneousAllocations;
|
|
};
|
|
|
|
class CustomGuardedPoolAllocator : public Test {
|
|
public:
|
|
void
|
|
InitNumSlots(decltype(gwp_asan::options::Options::MaxSimultaneousAllocations)
|
|
MaxSimultaneousAllocationsArg) {
|
|
gwp_asan::options::Options Opts;
|
|
Opts.setDefaults();
|
|
|
|
Opts.MaxSimultaneousAllocations = MaxSimultaneousAllocationsArg;
|
|
MaxSimultaneousAllocations = MaxSimultaneousAllocationsArg;
|
|
|
|
Opts.InstallForkHandlers = gwp_asan::test::OnlyOnce();
|
|
GPA.init(Opts);
|
|
}
|
|
|
|
void TearDown() override { GPA.uninitTestOnly(); }
|
|
|
|
protected:
|
|
gwp_asan::GuardedPoolAllocator GPA;
|
|
decltype(gwp_asan::options::Options::MaxSimultaneousAllocations)
|
|
MaxSimultaneousAllocations;
|
|
};
|
|
|
|
class BacktraceGuardedPoolAllocator : public Test {
|
|
public:
|
|
void SetUp() override {
|
|
gwp_asan::options::Options Opts;
|
|
Opts.setDefaults();
|
|
|
|
Opts.Backtrace = gwp_asan::backtrace::getBacktraceFunction();
|
|
Opts.InstallForkHandlers = gwp_asan::test::OnlyOnce();
|
|
GPA.init(Opts);
|
|
|
|
gwp_asan::segv_handler::installSignalHandlers(
|
|
&GPA, gwp_asan::test::getPrintfFunction(),
|
|
gwp_asan::backtrace::getPrintBacktraceFunction(),
|
|
gwp_asan::backtrace::getSegvBacktraceFunction());
|
|
}
|
|
|
|
void TearDown() override {
|
|
GPA.uninitTestOnly();
|
|
gwp_asan::segv_handler::uninstallSignalHandlers();
|
|
}
|
|
|
|
protected:
|
|
gwp_asan::GuardedPoolAllocator GPA;
|
|
};
|
|
|
|
#endif // GWP_ASAN_TESTS_HARNESS_H_
|