blob: 5f9ac705a0f8cf948f2ab1c84211936fee860286 (
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
|
/*
* Copyright (c) 2011-2015 Isode Limited.
* All rights reserved.
* See the COPYING file for more information.
*/
#pragma once
#include <vector>
#include <algorithm>
#include <Swiften/Base/API.h>
namespace Swift {
void secureZeroMemory(char* memory, size_t numberOfBytes);
template<typename T>
class SWIFTEN_API SafeAllocator : public std::allocator<T> {
public:
template <class U> struct rebind {
typedef SafeAllocator<U> other;
};
SafeAllocator() SWIFTEN_NOEXCEPT {}
SafeAllocator(const SafeAllocator&) SWIFTEN_NOEXCEPT : std::allocator<T>() {}
template <class U> SafeAllocator(const SafeAllocator<U>&) SWIFTEN_NOEXCEPT {}
~SafeAllocator() SWIFTEN_NOEXCEPT {}
void deallocate (T* p, size_t num) {
secureZeroMemory(reinterpret_cast<char*>(p), num);
std::allocator<T>::deallocate(p, num);
}
SWIFTEN_DEFAULT_COPY_ASSIGMNENT_OPERATOR(SafeAllocator)
};
}
|