// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/memory/discardable_memory.h"

#include <mach/mach.h>

#include "base/logging.h"

namespace base {

namespace {

// The VM subsystem allows tagging of memory and 240-255 is reserved for
// application use (see mach/vm_statistics.h). Pick 252 (after chromium's atomic
// weight of ~52).
const int kDiscardableMemoryTag = VM_MAKE_TAG(252);

}  // namespace

// static
bool DiscardableMemory::Supported() {
  return true;
}

DiscardableMemory::~DiscardableMemory() {
  if (memory_) {
    vm_deallocate(mach_task_self(),
                  reinterpret_cast<vm_address_t>(memory_),
                  size_);
  }
}

bool DiscardableMemory::InitializeAndLock(size_t size) {
  DCHECK(!memory_);
  size_ = size;

  vm_address_t buffer = 0;
  kern_return_t ret = vm_allocate(mach_task_self(),
                                  &buffer,
                                  size,
                                  VM_FLAGS_PURGABLE |
                                  VM_FLAGS_ANYWHERE |
                                  kDiscardableMemoryTag);

  if (ret != KERN_SUCCESS) {
    DLOG(ERROR) << "vm_allocate() failed";
    return false;
  }

  is_locked_ = true;
  memory_ = reinterpret_cast<void*>(buffer);
  return true;
}

LockDiscardableMemoryStatus DiscardableMemory::Lock() {
  DCHECK(!is_locked_);

  int state = VM_PURGABLE_NONVOLATILE;
  kern_return_t ret = vm_purgable_control(
      mach_task_self(),
      reinterpret_cast<vm_address_t>(memory_),
      VM_PURGABLE_SET_STATE,
      &state);

  if (ret != KERN_SUCCESS)
    return DISCARDABLE_MEMORY_FAILED;

  is_locked_ = true;
  return state & VM_PURGABLE_EMPTY ? DISCARDABLE_MEMORY_PURGED
                                   : DISCARDABLE_MEMORY_SUCCESS;
}

void DiscardableMemory::Unlock() {
  DCHECK(is_locked_);

  int state = VM_PURGABLE_VOLATILE | VM_VOLATILE_GROUP_DEFAULT;
  kern_return_t ret = vm_purgable_control(
      mach_task_self(),
      reinterpret_cast<vm_address_t>(memory_),
      VM_PURGABLE_SET_STATE,
      &state);

  if (ret != KERN_SUCCESS)
    DLOG(ERROR) << "Failed to unlock memory.";

  is_locked_ = false;
}

// static
bool DiscardableMemory::PurgeForTestingSupported() {
  return true;
}

// static
void DiscardableMemory::PurgeForTesting() {
  int state = 0;
  vm_purgable_control(mach_task_self(), 0, VM_PURGABLE_PURGE_ALL, &state);
}

}  // namespace base