// Copyright 2020 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 "third_party/blink/renderer/controller/memory_usage_monitor_win.h"

#include <windows.h>  // Must be in front of other Windows header files.

#include <psapi.h>
#include <tchar.h>

#include "third_party/blink/public/platform/platform.h"

namespace blink {

namespace {

static MemoryUsageMonitor* g_instance_for_testing = nullptr;

}  // namespace

// static
MemoryUsageMonitor& MemoryUsageMonitor::Instance() {
  DEFINE_STATIC_LOCAL(MemoryUsageMonitorWin, monitor, ());
  return g_instance_for_testing ? *g_instance_for_testing : monitor;
}

// static
void MemoryUsageMonitor::SetInstanceForTesting(MemoryUsageMonitor* instance) {
  g_instance_for_testing = instance;
}

// CalculateProcessMemoryFootprint is generated from:
// - CalculatePrivateFootprintKb defined in
//   //services/resource_coordinator/memory_instrumentation/queued_request_dispatcher.cc
// - OSMetrics::FillOSMemoryDump defined in
//   //services/resource_coordinator/public/cpp/memory_instrumentation/os_metrics_win.cc
bool MemoryUsageMonitorWin::CalculateProcessMemoryFootprint(
    uint64_t* private_footprint) {
  PROCESS_MEMORY_COUNTERS_EX pmc;
  if (!::GetProcessMemoryInfo(::GetCurrentProcess(),
                              reinterpret_cast<PROCESS_MEMORY_COUNTERS*>(&pmc),
                              sizeof(pmc)))
    return false;
  *private_footprint = pmc.PrivateUsage;
  return true;
}

void MemoryUsageMonitorWin::GetProcessMemoryUsage(MemoryUsage& usage) {
  uint64_t private_footprint;
  if (CalculateProcessMemoryFootprint(&private_footprint))
    usage.private_footprint_bytes = static_cast<double>(private_footprint);
}

}  // namespace blink