// Copyright 2015 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/bind.h"
#include "base/command_line.h"
#include "base/memory/discardable_memory.h"
#include "content/child/child_discardable_shared_memory_manager.h"
#include "content/child/child_thread_impl.h"
#include "content/common/host_discardable_shared_memory_manager.h"
#include "content/public/common/content_switches.h"
#include "content/public/test/content_browser_test.h"
#include "content/public/test/content_browser_test_utils.h"
#include "content/shell/browser/shell.h"
#include "url/gurl.h"

namespace content {

class ChildDiscardableSharedMemoryManagerBrowserTest
    : public ContentBrowserTest {
 public:
  void SetUpCommandLine(base::CommandLine* command_line) override {
    command_line->AppendSwitch(switches::kSingleProcess);
  }

  static void ReleaseFreeMemory() {
    ChildThreadImpl::current()
        ->discardable_shared_memory_manager()
        ->ReleaseFreeMemory();
  }

  static void AllocateLockedMemory(
      size_t size,
      scoped_ptr<base::DiscardableMemory>* memory) {
    *memory = ChildThreadImpl::current()
                  ->discardable_shared_memory_manager()
                  ->AllocateLockedDiscardableMemory(size);
  }

  static void LockMemory(base::DiscardableMemory* memory, bool* result) {
    *result = memory->Lock();
  }

  static void UnlockMemory(base::DiscardableMemory* memory) {
    memory->Unlock();
  }

  static void FreeMemory(scoped_ptr<base::DiscardableMemory> memory) {}
};

IN_PROC_BROWSER_TEST_F(ChildDiscardableSharedMemoryManagerBrowserTest,
                       DISABLED_LockMemory) {
  const size_t kSize = 1024 * 1024;  // 1MiB.

  NavigateToURL(shell(), GURL(url::kAboutBlankURL));

  scoped_ptr<base::DiscardableMemory> memory;
  PostTaskToInProcessRendererAndWait(base::Bind(
      &ChildDiscardableSharedMemoryManagerBrowserTest::AllocateLockedMemory,
      kSize, &memory));

  ASSERT_TRUE(memory);
  void* addr = memory->Memory();
  ASSERT_NE(nullptr, addr);

  PostTaskToInProcessRendererAndWait(
      base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::UnlockMemory,
                 memory.get()));

  // Purge all unlocked memory.
  HostDiscardableSharedMemoryManager::current()->SetMemoryLimit(0);

  bool result = true;
  PostTaskToInProcessRendererAndWait(
      base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::LockMemory,
                 memory.get(), &result));

  // Should fail as memory should have been purged.
  EXPECT_FALSE(result);

  PostTaskToInProcessRendererAndWait(
      base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::FreeMemory,
                 base::Passed(&memory)));
}

IN_PROC_BROWSER_TEST_F(ChildDiscardableSharedMemoryManagerBrowserTest,
                       DISABLED_AddressSpace) {
  const size_t kLargeSize = 4 * 1024 * 1024;   // 4MiB.
  const size_t kNumberOfInstances = 1024 + 1;  // >4GiB total.

  NavigateToURL(shell(), GURL(url::kAboutBlankURL));

  scoped_ptr<base::DiscardableMemory> instances[kNumberOfInstances];
  for (auto& memory : instances) {
    PostTaskToInProcessRendererAndWait(base::Bind(
        &ChildDiscardableSharedMemoryManagerBrowserTest::AllocateLockedMemory,
        kLargeSize, &memory));
    ASSERT_TRUE(memory);
    void* addr = memory->Memory();
    ASSERT_NE(nullptr, addr);
    PostTaskToInProcessRendererAndWait(base::Bind(
        &ChildDiscardableSharedMemoryManagerBrowserTest::UnlockMemory,
        memory.get()));
  }

  for (auto& memory : instances) {
    PostTaskToInProcessRendererAndWait(
        base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::FreeMemory,
                   base::Passed(&memory)));
  }
}

IN_PROC_BROWSER_TEST_F(ChildDiscardableSharedMemoryManagerBrowserTest,
                       DISABLED_ReleaseFreeMemory) {
  const size_t kSize = 1024 * 1024;  // 1MiB.

  NavigateToURL(shell(), GURL(url::kAboutBlankURL));

  scoped_ptr<base::DiscardableMemory> memory;
  PostTaskToInProcessRendererAndWait(base::Bind(
      &ChildDiscardableSharedMemoryManagerBrowserTest::AllocateLockedMemory,
      kSize, &memory));
  PostTaskToInProcessRendererAndWait(
      base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::UnlockMemory,
                 memory.get()));
  PostTaskToInProcessRendererAndWait(
      base::Bind(&ChildDiscardableSharedMemoryManagerBrowserTest::FreeMemory,
                 base::Passed(&memory)));

  EXPECT_GE(HostDiscardableSharedMemoryManager::current()->GetBytesAllocated(),
            kSize);

  PostTaskToInProcessRendererAndWait(base::Bind(
      &ChildDiscardableSharedMemoryManagerBrowserTest::ReleaseFreeMemory));

  EXPECT_EQ(HostDiscardableSharedMemoryManager::current()->GetBytesAllocated(),
            0u);
}

}  // content