// 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.

#ifndef BASE_TRACE_EVENT_PROCESS_MEMORY_MAPS_H_
#define BASE_TRACE_EVENT_PROCESS_MEMORY_MAPS_H_

#include <stdint.h>

#include <string>
#include <vector>

#include "base/base_export.h"
#include "base/macros.h"

namespace base {
namespace trace_event {

class TracedValue;

// Data model for process-wide memory stats.
class BASE_EXPORT ProcessMemoryMaps {
 public:
  struct BASE_EXPORT VMRegion {
    static const uint32_t kProtectionFlagsRead;
    static const uint32_t kProtectionFlagsWrite;
    static const uint32_t kProtectionFlagsExec;
    static const uint32_t kProtectionFlagsMayshare;

    VMRegion();
    VMRegion(const VMRegion& other);

    uint64_t start_address;
    uint64_t size_in_bytes;
    uint32_t protection_flags;
    std::string mapped_file;

    // private_dirty_resident + private_clean_resident + shared_dirty_resident +
    // shared_clean_resident = resident set size.
    uint64_t byte_stats_private_dirty_resident;
    uint64_t byte_stats_private_clean_resident;
    uint64_t byte_stats_shared_dirty_resident;
    uint64_t byte_stats_shared_clean_resident;

    uint64_t byte_stats_swapped;

    // For multiprocess accounting.
    uint64_t byte_stats_proportional_resident;
  };

  ProcessMemoryMaps();
  ~ProcessMemoryMaps();

  void AddVMRegion(const VMRegion& region) { vm_regions_.push_back(region); }
  const std::vector<VMRegion>& vm_regions() const { return vm_regions_; }

  // Called at trace generation time to populate the TracedValue.
  void AsValueInto(TracedValue* value) const;

  // Clears up all the VMRegion(s) stored.
  void Clear();

 private:
  std::vector<VMRegion> vm_regions_;

  DISALLOW_COPY_AND_ASSIGN(ProcessMemoryMaps);
};

}  // namespace trace_event
}  // namespace base

#endif  // BASE_TRACE_EVENT_PROCESS_MEMORY_MAPS_H_