MemoryUtils.h

 
#pragma once
 
#include "core/GpgFrontendCoreExport.h"
#include "core/function/SecureMemoryAllocator.h"
 
/* To avoid that a compiler optimizes certain memset calls away, these
   macros may be used instead. */
#define wipememory2(_ptr, _set, _len)               \
  do {                                              \
    volatile char *_vptr = (volatile char *)(_ptr); \
    size_t _vlen = (_len);                          \
    while (_vlen) {                                 \
      *_vptr = (_set);                              \
      _vptr++;                                      \
      _vlen--;                                      \
    }                                               \
  } while (0)
#define wipememory(_ptr, _len) wipememory2(_ptr, 0, _len)
#define wipe(_ptr, _len) wipememory2(_ptr, 0, _len)
 
#define xtoi_1(p)                    \
  (*(p) <= '9'   ? (*(p) - '0')      \
   : *(p) <= 'F' ? (*(p) - 'A' + 10) \
                 : (*(p) - 'a' + 10))
#define xtoi_2(p) ((xtoi_1(p) * 16) + xtoi_1((p) + 1))
 
namespace GpgFrontend {
 
template <typename T>
class PointerConverter {
 public:
  explicit PointerConverter(void *ptr) : ptr_(ptr) {}
 
  auto AsType() const -> T * { return static_cast<T *>(ptr_); }
 
 private:
  void *ptr_;
};
 
auto GPGFRONTEND_CORE_EXPORT SecureMalloc(std::size_t) -> void *;
 
auto GPGFRONTEND_CORE_EXPORT SecureRealloc(void *, std::size_t) -> void *;
 
template <typename T>
auto SecureMallocAsType(std::size_t size) -> T * {
  return PointerConverter<T>(SecureMemoryAllocator::Allocate(size)).AsType();
}
 
template <typename T>
auto SecureReallocAsType(T *ptr, std::size_t size) -> T * {
  return PointerConverter<T>(SecureMemoryAllocator::Reallocate(ptr, size))
      .AsType();
}
 
void GPGFRONTEND_CORE_EXPORT SecureFree(void *);
 
template <typename T, typename... Args>
static auto SecureCreateObject(Args &&...args) -> T * {
  void *mem = SecureMemoryAllocator::Allocate(sizeof(T));
  if (!mem) return nullptr;
 
  try {
    return new (mem) T(std::forward<Args>(args)...);
  } catch (...) {
    SecureMemoryAllocator::Deallocate(mem);
    throw;
  }
}
 
template <typename T>
static void SecureDestroyObject(T *obj) {
  if (!obj) return;
  obj->~T();
  SecureMemoryAllocator::Deallocate(obj);
}
 
template <typename T, typename... Args>
static auto SecureCreateUniqueObject(Args &&...args)
    -> std::unique_ptr<T, SecureObjectDeleter<T>> {
  void *mem = SecureMemoryAllocator::Allocate(sizeof(T));
  if (!mem) throw std::bad_alloc();
 
  try {
    return std::unique_ptr<T, SecureObjectDeleter<T>>(
        new (mem) T(std::forward<Args>(args)...));
  } catch (...) {
    SecureMemoryAllocator::Deallocate(mem);
    throw;
  }
}
 
template <typename T, typename... Args>
auto SecureCreateSharedObject(Args &&...args) -> std::shared_ptr<T> {
  void *mem = SecureMemoryAllocator::Allocate(sizeof(T));
  if (!mem) throw std::bad_alloc();
 
  try {
    T *obj = new (mem) T(std::forward<Args>(args)...);
    return std::shared_ptr<T>(obj, [](T *ptr) {
      ptr->~T();
      SecureMemoryAllocator::Deallocate(ptr);
    });
  } catch (...) {
    SecureMemoryAllocator::Deallocate(mem);
    throw;
  }
}
 
template <typename T, typename... Args>
auto SecureCreateQSharedObject(Args &&...args) -> QSharedPointer<T> {
  void *mem = SecureMemoryAllocator::Allocate(sizeof(T));
  if (!mem) throw std::bad_alloc();
 
  try {
    T *obj = new (mem) T(std::forward<Args>(args)...);
    return QSharedPointer<T>(obj, [](T *ptr) {
      ptr->~T();
      SecureMemoryAllocator::Deallocate(ptr);
    });
  } catch (...) {
    SecureMemoryAllocator::Deallocate(mem);
    throw;
  }
}
 
};  // namespace GpgFrontend