Gets presentation ID from route ID upon route creation, and overrides previous presentation ID with…

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 "base/containers/scoped_ptr_hash_map.h"
 #include "base/containers/small_map.h"
@@ skipping 605 matching lines @@
     DVLOG(1) << "No active route for: " << presentation_id;
     return;
   }
   router_->CloseRoute(route_id);
 }
 
 void PresentationServiceDelegateImpl::ListenForSessionMessages(
     int render_process_id,
     int render_frame_id,
     const PresentationSessionMessageCallback& message_cb) {
-  // BUG=464205
-  NOTIMPLEMENTED();
+  const std::vector<MediaRoute::Id>& route_ids = frame_manager_->GetRouteIds(
+      RenderFrameHostId(render_process_id, render_frame_id));
+  if (route_ids.empty()) {
+    DVLOG(1) << "No media routes found";
+    message_cb.Run(
+        scoped_ptr<ScopedVector<content::PresentationSessionMessage>>());
+    return;
+  }
+
+  router_->ListenForRouteMessages(route_ids, message_cb);
 }
 
 void PresentationServiceDelegateImpl::SendMessage(
     int render_process_id,
     int render_frame_id,
     scoped_ptr<content::PresentationSessionMessage> message_request,
     const SendMessageCallback& send_message_cb) {
-  // BUG=464205
-  NOTIMPLEMENTED();
+  if (message_request->is_binary()) {
+    NOTIMPLEMENTED();
+    send_message_cb.Run(false);
+    return;
+  }
+  const MediaRoute::Id& route_id = frame_manager_->GetRouteId(
+      RenderFrameHostId(render_process_id, render_frame_id),
+      message_request->presentation_id);
+  if (route_id.empty()) {
+    DVLOG(1) << "No active route for  " << message_request->presentation_id;
+    send_message_cb.Run(false);
+    return;
+  }
+  router_->SendRouteMessage(route_id, *(message_request->message),
+                            send_message_cb);
 }
 
 void PresentationServiceDelegateImpl::OnRouteCreated(const MediaRoute& route) {
   const MediaSource& source = route.media_source();
   DCHECK(!source.Empty());
   if (!default_source_.Equals(source))
     return;
   RenderFrameHost* main_frame = web_contents_->GetMainFrame();
   if (!main_frame)
     return;
   RenderFrameHostId render_frame_host_id(GetRenderFrameHostId(main_frame));
-  // TODO(imcheng): Pass in valid default presentation ID once it is
-  // available from MediaRoute URN. BUG=493365
-  std::string presentation_id;
+  std::string presentation_id =
+      GetPresentationIdAndUrl(route.media_route_id()).first;
   frame_manager_->OnPresentationSessionStarted(
       render_frame_host_id, true,
       content::PresentationSessionInfo(PresentationUrlFromMediaSource(source),
                                        presentation_id),
       route.media_route_id());
 }
 
 void PresentationServiceDelegateImpl::AddDefaultMediaSourceObserver(
     DefaultMediaSourceObserver* observer) {
   default_media_source_observers_.AddObserver(observer);
@@ skipping 18 matching lines @@
 bool PresentationServiceDelegateImpl::HasScreenAvailabilityListenerForTest(
     int render_process_id,
     int render_frame_id,
     const MediaSource::Id& source_id) const {
   RenderFrameHostId render_frame_host_id(render_process_id, render_frame_id);
   return frame_manager_->HasScreenAvailabilityListenerForTest(
       render_frame_host_id, source_id);
 }
 
 }  // namespace media_router