password.h

Go to the documentation of this file.

 
#ifndef CPPSECURITY_STRING_PASSWORD_H
#define CPPSECURITY_STRING_PASSWORD_H
 
#include "memory/memory.h"
 
#include <string>
 
namespace CppSecurity {
 
template <class T>
class PasswordAllocator : public std::allocator<T>
{
public:
    typedef typename std::allocator<T>::value_type value_type;
    typedef typename std::allocator<T>::size_type size_type;
    typedef typename std::allocator<T>::difference_type difference_type;
 
    template<class U> struct rebind { typedef PasswordAllocator<U> other; };
 
    PasswordAllocator() throw() {}
    PasswordAllocator(const PasswordAllocator&) throw() {}
    template <class U> PasswordAllocator(const PasswordAllocator<U>&) throw() {}
 
    void deallocate(T* p, size_type n)
    {
        CppCommon::Memory::ZeroFill(p, n * sizeof(T));
        std::allocator<T>::deallocate(p, n);
    }
};
 
} // namespace CppSecurity
 
namespace std {
 
 
class password : public std::basic_string<char, std::char_traits<char>, CppSecurity::PasswordAllocator<char>>
{
public:
    using basic_string::basic_string;
    ~password() { clear(); }
    void clear() noexcept { CppCommon::Memory::ZeroFill(data(), size()); }
};
 
} // namespace std
 
#endif // CPPSECURITY_STRING_PASSWORD_H