cc::SurfaceDependencyTracker should not crash when a Display goes away

cc/surfaces/frame_sink_manager.cc

Index: cc/surfaces/frame_sink_manager.cc
diff --git a/cc/surfaces/frame_sink_manager.cc b/cc/surfaces/frame_sink_manager.cc
index c16a921fc47a38413dc4e0724903b8e7171c4815..22f1d172dfcb8b4dcfb5799d211c5a4c8808e3b2 100644
--- a/cc/surfaces/frame_sink_manager.cc
+++ b/cc/surfaces/frame_sink_manager.cc
@@ -9,6 +9,7 @@
 
 #include "base/logging.h"
 #include "cc/surfaces/frame_sink_manager_client.h"
+#include "cc/surfaces/primary_begin_frame_source.h"
 
 #if DCHECK_IS_ON()
 #include <sstream>
@@ -16,15 +17,14 @@
 
 namespace cc {
 
-FrameSinkManager::FrameSinkSourceMapping::FrameSinkSourceMapping()
-    : source(nullptr) {}
+FrameSinkManager::FrameSinkSourceMapping::FrameSinkSourceMapping() = default;
 
 FrameSinkManager::FrameSinkSourceMapping::FrameSinkSourceMapping(
     const FrameSinkSourceMapping& other) = default;
 
-FrameSinkManager::FrameSinkSourceMapping::~FrameSinkSourceMapping() {}
+FrameSinkManager::FrameSinkSourceMapping::~FrameSinkSourceMapping() = default;
 
-FrameSinkManager::FrameSinkManager() {}
+FrameSinkManager::FrameSinkManager() = default;
 
 FrameSinkManager::~FrameSinkManager() {
   // All CompositorFrameSinks should be unregistered prior to
@@ -82,6 +82,8 @@ void FrameSinkManager::RegisterBeginFrameSource(
 
   registered_sources_[source] = frame_sink_id;
   RecursivelyAttachBeginFrameSource(frame_sink_id, source);
+
+  primary_source_.OnBeginFrameSourceAdded(source);
 }
 
 void FrameSinkManager::UnregisterBeginFrameSource(BeginFrameSource* source) {
@@ -91,6 +93,8 @@ void FrameSinkManager::UnregisterBeginFrameSource(BeginFrameSource* source) {
   FrameSinkId frame_sink_id = registered_sources_[source];
   registered_sources_.erase(source);
 
+  primary_source_.OnBeginFrameSourceRemoved(source);
+
   if (frame_sink_source_map_.count(frame_sink_id) == 0u)
     return;
 
@@ -103,6 +107,10 @@ void FrameSinkManager::UnregisterBeginFrameSource(BeginFrameSource* source) {
     RecursivelyAttachBeginFrameSource(source_iter.second, source_iter.first);
 }
 
+BeginFrameSource* FrameSinkManager::GetPrimaryBeginFrameSource() {
+  return &primary_source_;
+}
+
 void FrameSinkManager::RecursivelyAttachBeginFrameSource(
     const FrameSinkId& frame_sink_id,
     BeginFrameSource* source) {