datacontainer.cc

Go to the documentation of this file.

#include "ion/base/datacontainer.h"

#include "ion/base/allocationmanager.h"
#include "ion/base/lockguards.h"
#include "ion/base/staticsafedeclare.h"
#include "ion/port/mutex.h"

namespace ion {
namespace base {

DataContainer::DataContainer(const Deleter& deleter, bool is_wipeable)
    : data_(NULL),
      is_wipeable_(is_wipeable),
      deleter_(deleter) {
}

DataContainer::~DataContainer() {
  AddOrRemoveDataFromCheck(data_, false);
  InternalWipeData();
}

void DataContainer::WipeData() {
  AddOrRemoveDataFromCheck(data_, false);
  if (is_wipeable_)
    InternalWipeData();
}

DataContainer* DataContainer::Allocate(
    size_t extra_bytes, const Deleter& deleter, bool is_wipeable,
    const AllocatorPtr& allocator) {
  const AllocatorPtr& alloc = allocator.Get()
      ? allocator
      : AllocationManager::GetDefaultAllocatorForLifetime(kMediumTerm);

  void* ptr =
      DataContainer::operator new(sizeof(DataContainer) + extra_bytes, alloc);
  DataContainer* container = new(ptr) DataContainer(deleter, is_wipeable);
  return container;
}

bool DataContainer::AddOrRemoveDataFromCheck(void* data, bool is_add) {
#if ION_DEBUG
  ION_DECLARE_SAFE_STATIC_POINTER(std::set<void*>, client_pointers_used);

  static port::Mutex mutex;
  LockGuard guard(&mutex);

  if (is_add) {
    if (client_pointers_used->count(data)) {
      LOG(ERROR) << "Duplicate client-space pointer passed to "
                 << "DataContainer::Create(). This is very dangerous and may "
                 << "result in double-deletion! It is much safer to simply "
                 << "use the same DataContainerPtr.";
      return false;
    }
    client_pointers_used->insert(data);
  } else {
    client_pointers_used->erase(data);
  }
#endif
  return true;
}

void DataContainer::InternalWipeData() {
  if (data_ != NULL && deleter_) {
    deleter_(data_);
    data_ = NULL;
  }
}

}  // namespace base
}  // namespace ion