[Media Router] Add a presentation id to MediaRoute mapping in the MR

chrome/browser/media/router/presentation_service_delegate_impl.cc

 // 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.
 
 #include "chrome/browser/media/router/presentation_service_delegate_impl.h"
 
 #include <string>
 #include <unordered_map>
 #include <utility>
 #include <vector>
@@ skipping 195 matching lines @@
 }
 
 PresentationFrame::~PresentationFrame() {
 }
 
 void PresentationFrame::OnPresentationSessionStarted(
     const content::PresentationSessionInfo& session,
     const MediaRoute& route) {
   presentation_id_to_route_.insert(
       std::make_pair(session.presentation_id, route));
+  router_->RegisterPresentationRoute(session.presentation_id, route);
 }
 
 const MediaRoute::Id PresentationFrame::GetRouteId(
     const std::string& presentation_id) const {
   auto it = presentation_id_to_route_.find(presentation_id);
   return it != presentation_id_to_route_.end() ? it->second.media_route_id()
                                                : "";
 }
 
 bool PresentationFrame::SetScreenAvailabilityListener(
@@ skipping 37 matching lines @@
   auto* offscreen_presentation_manager =
       OffscreenPresentationManagerFactory::GetOrCreateForWebContents(
           web_contents_);
 
   for (const auto& pid_route : presentation_id_to_route_) {
     if (IsOffscreenPresentation(pid_route.first)) {
       offscreen_presentation_manager->UnregisterOffscreenPresentationController(
           pid_route.first, render_frame_host_id_);
     }
     router_->DetachRoute(pid_route.second.media_route_id());
+    router_->UnregisterPresentationRoute(pid_route.first);
   }
 
   presentation_id_to_route_.clear();
   url_to_sinks_observer_.clear();
   connection_state_subscriptions_.clear();
   session_messages_observers_.clear();
   browser_connection_proxies_.clear();
 }
 
 void PresentationFrame::RemoveConnection(const std::string& presentation_id,
                                          const MediaRoute::Id& route_id) {
   // Remove the presentation id mapping so a later call to Reset is a no-op.
   presentation_id_to_route_.erase(presentation_id);
 
+  router_->UnregisterPresentationRoute(presentation_id);
+
   // We no longer need to observe route messages.
   session_messages_observers_.erase(route_id);
 
   browser_connection_proxies_.erase(route_id);
   // We keep the PresentationConnectionStateChangedCallback registered with MR
   // so the MRP can tell us when terminate() completed.
 }
 
 void PresentationFrame::ListenForConnectionStateChange(
     const content::PresentationSessionInfo& connection,
@@ skipping 757 matching lines @@
   const base::ListValue* origins =
       Profile::FromBrowserContext(web_contents_->GetBrowserContext())
           ->GetPrefs()
           ->GetList(prefs::kMediaRouterTabMirroringSources);
   return origins &&
          origins->Find(base::StringValue(origin.Serialize())) != origins->end();
 }
 #endif  // !defined(OS_ANDROID)
 
 }  // namespace media_router