security.hpp

// Copyright Take Vos 2022.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#include "utility/module.hpp"
 
#pragma once
 
namespace tt::inline v1 {
 
void secure_clear(void *ptr, size_t size) noexcept;
 
void secure_clear(trivially_copyable auto &object) noexcept
{
    secure_clear(&object, sizeof(object));
}
 
template<typename It>
void secure_clear(It first, It last) noexcept
{
    using value_type = decltype(*first);
 
    if constexpr (requires { std::distance(first, last); }) {
        secure_clear(std::addressof(*first), std::distance(first, last) * sizeof(value_type));
 
    } else {
        for (auto it = first; it != last; ++it) {
            secure_clear(*it);
        }
    }
}
 
void secure_destroy_at(auto *p)
{
    std::destroy_at(p);
    secure_clear(*p);
}
 
template<typename It>
void secure_destroy(It first, It last)
{
    std::destroy(first, last);
    secure_clear(first, last);
}
 
template<typename It, typename OutIt>
void secure_uninitialized_move(It first, It last, OutIt d_first)
{
    std::uninitialized_move(first, last, d_first);
    secure_erase(first, last);
}
 
}