mirror of
https://github.com/bitcoin/bitcoin.git
synced 2025-06-02 19:22:22 +02:00
ea4b61a157
Includes allocator::pointer, allocator::const_pointer, allocator::reference and allocator::const_reference which are deprecated in c++17 and removed in c++20. See https://en.cppreference.com/w/cpp/memory/allocator Also prefer `using` over `typedef` see: https://isocpp.github.io/CppCoreGuidelines/CppCoreGuidelines#Rt-using
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|
// Copyright (c) 2009-2019 The Bitcoin Core developers
|
|
// Distributed under the MIT software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#ifndef BITCOIN_SUPPORT_ALLOCATORS_SECURE_H
|
|
#define BITCOIN_SUPPORT_ALLOCATORS_SECURE_H
|
|
|
|
#include <support/lockedpool.h>
|
|
#include <support/cleanse.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
//
|
|
// Allocator that locks its contents from being paged
|
|
// out of memory and clears its contents before deletion.
|
|
//
|
|
template <typename T>
|
|
struct secure_allocator : public std::allocator<T> {
|
|
using base = std::allocator<T>;
|
|
using traits = std::allocator_traits<base>;
|
|
using size_type = typename traits::size_type;
|
|
using difference_type = typename traits::difference_type;
|
|
using pointer = typename traits::pointer;
|
|
using const_pointer = typename traits::const_pointer;
|
|
using value_type = typename traits::value_type;
|
|
secure_allocator() noexcept {}
|
|
secure_allocator(const secure_allocator& a) noexcept : base(a) {}
|
|
template <typename U>
|
|
secure_allocator(const secure_allocator<U>& a) noexcept : base(a)
|
|
{
|
|
}
|
|
~secure_allocator() noexcept {}
|
|
template <typename _Other>
|
|
struct rebind {
|
|
typedef secure_allocator<_Other> other;
|
|
};
|
|
|
|
T* allocate(std::size_t n, const void* hint = 0)
|
|
{
|
|
T* allocation = static_cast<T*>(LockedPoolManager::Instance().alloc(sizeof(T) * n));
|
|
if (!allocation) {
|
|
throw std::bad_alloc();
|
|
}
|
|
return allocation;
|
|
}
|
|
|
|
void deallocate(T* p, std::size_t n)
|
|
{
|
|
if (p != nullptr) {
|
|
memory_cleanse(p, sizeof(T) * n);
|
|
}
|
|
LockedPoolManager::Instance().free(p);
|
|
}
|
|
};
|
|
|
|
// This is exactly like std::string, but with a custom allocator.
|
|
typedef std::basic_string<char, std::char_traits<char>, secure_allocator<char> > SecureString;
|
|
|
|
#endif // BITCOIN_SUPPORT_ALLOCATORS_SECURE_H
|