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

cc/surfaces/frame_sink_manager.h

 // Copyright 2014 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 CC_SURFACES_FRAME_SINK_MANAGER_H_
 #define CC_SURFACES_FRAME_SINK_MANAGER_H_
 
 #include <stdint.h>
 
 #include <unordered_map>
 #include <unordered_set>
 #include <vector>
 
 #include "base/logging.h"
 #include "base/macros.h"
+#include "base/observer_list.h"
 #include "cc/surfaces/frame_sink_id.h"
+#include "cc/surfaces/frame_sink_observer.h"
 #include "cc/surfaces/surfaces_export.h"
 
 namespace cc {
 class BeginFrameSource;
 class FrameSinkManagerClient;
 
 namespace test {
 class SurfaceSynchronizationTest;
 }
 
@@ skipping 16 matching lines @@
   // However, DelegatedFrameHost can register itself as a client before its
   // relationship with the ui::Compositor is known.
 
   // Associates a FrameSinkManagerClient with the frame_sink_id it uses.
   // FrameSinkManagerClient and framesink allocators have a 1:1 mapping.
   // Caller guarantees the client is alive between register/unregister.
   void RegisterFrameSinkManagerClient(const FrameSinkId& frame_sink_id,
                                       FrameSinkManagerClient* client);
   void UnregisterFrameSinkManagerClient(const FrameSinkId& frame_sink_id);
 
+  // Creates a BeginFrameSource that tracks the first registered
+  // BeginFrameSource.
+  std::unique_ptr<BeginFrameSource> CreatePrimaryBeginFrameSource();
+
+  void AddObserver(FrameSinkObserver* obs) { observer_list_.AddObserver(obs); }
+
+  void RemoveObserver(FrameSinkObserver* obs) {
+    observer_list_.RemoveObserver(obs);
+  }
+
   // Associates a |source| with a particular framesink.  That framesink and
   // any children of that framesink with valid clients can potentially use
   // that |source|.
   void RegisterBeginFrameSource(BeginFrameSource* source,
                                 const FrameSinkId& frame_sink_id);
   void UnregisterBeginFrameSource(BeginFrameSource* source);
+  BeginFrameSource* GetPrimaryBeginFrameSource();
 
   // Register a relationship between two framesinks.  This relationship means
   // that surfaces from the child framesik will be displayed in the parent.
   // Children are allowed to use any begin frame source that their parent can
   // use.
   void RegisterFrameSinkHierarchy(const FrameSinkId& parent_frame_sink_id,
                                   const FrameSinkId& child_frame_sink_id);
   void UnregisterFrameSinkHierarchy(const FrameSinkId& parent_frame_sink_id,
                                     const FrameSinkId& child_frame_sink_id);
 
@@ skipping 23 matching lines @@
 
   // Begin frame source routing. Both BeginFrameSource and
   // CompositorFrameSinkSupport pointers guaranteed alive by callers until
   // unregistered.
   struct FrameSinkSourceMapping {
     FrameSinkSourceMapping();
     FrameSinkSourceMapping(const FrameSinkSourceMapping& other);
     ~FrameSinkSourceMapping();
     bool has_children() const { return !children.empty(); }
     // The currently assigned begin frame source for this client.
-    BeginFrameSource* source;
+    BeginFrameSource* source = nullptr;
     // This represents a dag of parent -> children mapping.
     std::vector<FrameSinkId> children;
   };
 
+  base::ObserverList<FrameSinkObserver> observer_list_;
+
   std::unordered_map<FrameSinkId, FrameSinkManagerClient*, FrameSinkIdHash>
       clients_;
 
   std::unordered_map<FrameSinkId, FrameSinkSourceMapping, FrameSinkIdHash>
       frame_sink_source_map_;
 
-  // Set of which sources are registered to which frmesinks.  Any child
+  // Set of BeginFrameSource along with associated FrameSinkIds. Any child
   // that is implicitly using this framesink must be reachable by the
   // parent in the dag.
   std::unordered_map<BeginFrameSource*, FrameSinkId> registered_sources_;
 
   DISALLOW_COPY_AND_ASSIGN(FrameSinkManager);
 };
 
 }  // namespace cc
 
 #endif  // CC_SURFACES_FRAME_SINK_MANAGER_H_