summaryrefslogtreecommitdiffstats
path: root/src/lib/util/unittests/newhook.cc
blob: 09686d43ce103eb21ee06e57eeddd3a3049d70ec (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
// Copyright (C) 2011-2015 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#include <stdlib.h>

#include <new>
#include <stdexcept>

#include <util/unittests/newhook.h>

#ifdef ENABLE_CUSTOM_OPERATOR_NEW
void*
operator new(size_t size) throw(std::bad_alloc) {
    if (isc::util::unittests::force_throw_on_new &&
        size == isc::util::unittests::throw_size_on_new) {
        throw std::bad_alloc();
    }
    void* p = malloc(size);
    if (p == NULL) {
        throw std::bad_alloc();
    }
    return (p);
}

void
operator delete(void* p) throw() {
    if (p != NULL) {
        free(p);
    }
}
#endif

namespace isc {
namespace util {
namespace unittests {
bool force_throw_on_new = false;
size_t throw_size_on_new = 0;
}
}
}