Expose MemoryCoordinator's global budget information.

content/browser/memory/memory_coordinator.cc

 // Copyright 2016 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 "content/browser/memory/memory_coordinator.h"
 
 #include "base/memory/memory_coordinator_client_registry.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/common/content_client.h"
@@ skipping 94 matching lines @@
   if (memory_state == mojom::MemoryState::SUSPENDED &&
       !CanSuspendRenderer(render_process_id))
     return false;
 
   // Update the internal state and send the message.
   iter->second.memory_state = memory_state;
   iter->second.handle->child()->OnStateChange(memory_state);
   return true;
 }
 
+bool MemoryCoordinator::SetRemainingGlobalBudget(
+    int render_process_id, int64_t remaining_global_budget_mb) {
+  // Can't send a message to a child that doesn't exist.
+  auto iter = children_.find(render_process_id);
+  if (iter == children_.end())
+    return false;
+
+  // Can't send a message to a child that isn't bound.
+  if (!iter->second.handle->child().is_bound())
+    return false;
+
+  // Send the new budget.
+  iter->second.handle->child()->SetRemainingGlobalBudget(
+      remaining_global_budget_mb);
+
+  return true;
+}
+
 mojom::MemoryState MemoryCoordinator::GetMemoryState(
     int render_process_id) const {
   auto iter = children_.find(render_process_id);
   if (iter == children_.end())
     return mojom::MemoryState::UNKNOWN;
   return iter->second.memory_state;
 }
 
 void MemoryCoordinator::EnableFeaturesForTesting() {
   base::FeatureList::ClearInstanceForTesting();
   std::unique_ptr<base::FeatureList> feature_list(new base::FeatureList);
   feature_list->InitializeFromCommandLine(features::kMemoryCoordinator.name,
                                           "");
   base::FeatureList::SetInstance(std::move(feature_list));
 }
 
 base::MemoryState MemoryCoordinator::GetCurrentMemoryState() const {
   return base::MemoryState::UNKNOWN;
 }
 
 void MemoryCoordinator::SetCurrentMemoryStateForTesting(
     base::MemoryState memory_state) {
 }
 
+int64_t MemoryCoordinator::GetRemainingGlobalBudget() const {
+  return 0;
+}
+
 void MemoryCoordinator::AddChildForTesting(
     int dummy_render_process_id, mojom::ChildMemoryCoordinatorPtr child) {
   mojom::MemoryCoordinatorHandlePtr mch;
   auto request = mojo::GetProxy(&mch);
   std::unique_ptr<MemoryCoordinatorHandleImpl> handle(
       new MemoryCoordinatorHandleImpl(std::move(request), this,
                                       dummy_render_process_id));
   handle->AddChild(std::move(child));
   CreateChildInfoMapEntry(dummy_render_process_id, std::move(handle));
 }
@@ skipping 35 matching lines @@
 
 MemoryCoordinator::ChildInfo::ChildInfo() {}
 
 MemoryCoordinator::ChildInfo::ChildInfo(const ChildInfo& rhs) {
   // This is a nop, but exists for compatibility with STL containers.
 }
 
 MemoryCoordinator::ChildInfo::~ChildInfo() {}
 
 }  // namespace content