[Presentation API] (5th) (1-UA) integrate controller and receiver side for 1-UA messaging

chrome/browser/media/router/browser_presentation_connection_proxy.cc

Index: chrome/browser/media/router/browser_presentation_connection_proxy.cc
diff --git a/chrome/browser/media/router/browser_presentation_connection_proxy.cc b/chrome/browser/media/router/browser_presentation_connection_proxy.cc
new file mode 100644
index 0000000000000000000000000000000000000000..9434d22db22b3fdee1ed7930059f078b3d57c532
--- /dev/null
+++ b/chrome/browser/media/router/browser_presentation_connection_proxy.cc
@@ -0,0 +1,151 @@
+// 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 "chrome/browser/media/router/browser_presentation_connection_proxy.h"
+
+#include "chrome/browser/media/router/media_route.h"
+#include "chrome/browser/media/router/media_router.h"
+#include "content/public/common/presentation_constants.h"
+
+namespace {
+
+std::unique_ptr<content::PresentationSessionMessage>
+GetPresentationSessionMessage(blink::mojom::SessionMessagePtr input) {
+  std::unique_ptr<content::PresentationSessionMessage> output;
+  if (input.is_null())
+    return output;
+
+  switch (input->type) {
+    case blink::mojom::PresentationMessageType::TEXT: {
+      // Return nullptr content::PresentationSessionMessage if invalid (unset
+      // |message|,
+      // set |data|, or size too large).

[mark a. foltz, 2016/12/02 23:06:48]

Rewrap comment.

[zhaobin, 2017/01/13 00:05:49]

Done.

+      if (input->data || !input->message ||
+          input->message->size() > content::kMaxPresentationSessionMessageSize)
+        return output;
+
+      output.reset(new content::PresentationSessionMessage(
+          content::PresentationMessageType::TEXT));
+      output->message = std::move(input->message.value());
+      return output;
+    }
+    case blink::mojom::PresentationMessageType::ARRAY_BUFFER: {
+      // Return nullptr content::PresentationSessionMessage if invalid (unset
+      // |data|, set
+      // |message|, or size too large).
+      if (!input->data || input->message ||
+          input->data->size() > content::kMaxPresentationSessionMessageSize)
+        return output;
+
+      output.reset(new content::PresentationSessionMessage(
+          content::PresentationMessageType::ARRAY_BUFFER));
+      output->data.reset(
+          new std::vector<uint8_t>(std::move(input->data.value())));
+      return output;
+    }
+    case blink::mojom::PresentationMessageType::BLOB: {
+      // Return nullptr content::PresentationSessionMessage if invalid (unset
+      // |data|, set
+      // |message|, or size too large).
+      if (!input->data || input->message ||
+          input->data->size() > content::kMaxPresentationSessionMessageSize)
+        return output;
+
+      output.reset(new content::PresentationSessionMessage(
+          content::PresentationMessageType::BLOB));
+      output->data.reset(
+          new std::vector<uint8_t>(std::move(input->data.value())));
+      return output;
+    }
+  }
+
+  NOTREACHED() << "Invalid presentation message type " << input->type;
+  return output;
+}
+
+}  // namespace
+
+namespace media_router {
+
+BrowserPresentationConnectionProxy::BrowserPresentationConnectionProxy(
+    const content::PresentationSessionInfo& session_info,
+    MediaRouter* router,
+    MediaRoute* route)
+    : session_info_(session_info),
+      router_(router),
+      route_(route),
+      binding_(this) {
+  DVLOG(2) << __FUNCTION__ << " [id]: " << session_info.presentation_id;
+}
+
+BrowserPresentationConnectionProxy::~BrowserPresentationConnectionProxy() {
+  DVLOG(2) << __FUNCTION__;
+  if (!on_message_callback_)
+    return;
+  // Run the callback with false, indicating the renderer to stop sending
+  // the requests and invalidate all pending requests.
+  on_message_callback_->Run(false);
+  on_message_callback_.reset();
+}
+
+blink::mojom::PresentationConnectionPtr
+BrowserPresentationConnectionProxy::Bind() {
+  return binding_.CreateInterfacePtrAndBind();
+}
+
+void BrowserPresentationConnectionProxy::SetTargetConnection(
+    blink::mojom::PresentationConnectionPtr connection) {
+  DVLOG(2) << __FUNCTION__;
+  target_connection_ = std::move(connection);
+  target_connection_->DidChangeState(
+      blink::mojom::PresentationConnectionState::CONNECTED);
+}
+
+void BrowserPresentationConnectionProxy::OnMessage(
+    blink::mojom::SessionMessagePtr session_message,
+    const OnMessageCallback& on_message_callback) {
+  DVLOG(2) << __FUNCTION__ << " [id]: " << session_info_.presentation_id;
+
+  DCHECK(!session_message.is_null());
+  // on_message_callback_ should be null by now, otherwise resetting of
+  // on_message_callback_ with new callback will drop the old callback.
+  if (!router_ || on_message_callback_) {
+    on_message_callback_->Run(false);
+    return;
+  }
+  on_message_callback_.reset(new OnMessageCallback(on_message_callback));
+
+  std::unique_ptr<content::PresentationSessionMessage> message =
+      GetPresentationSessionMessage(std::move(session_message));
+
+  if (message->is_binary()) {
+    router_->SendRouteBinaryMessage(
+        route_->media_route_id(), std::move(message->data),
+        base::Bind(
+            &BrowserPresentationConnectionProxy::OnConnectionMessageCallback,
+            base::Unretained(this)));
+  } else {
+    router_->SendRouteMessage(
+        route_->media_route_id(), message->message,
+        base::Bind(
+            &BrowserPresentationConnectionProxy::OnConnectionMessageCallback,
+            base::Unretained(this)));
+  }
+}
+
+void BrowserPresentationConnectionProxy::DidChangeState(
+    blink::mojom::PresentationConnectionState state) {}
+
+void BrowserPresentationConnectionProxy::OnConnectionMessageCallback(
+    bool sent) {
+  DVLOG(2) << __FUNCTION__ << " [sent]: " << sent;
+  // It is possible that Reset() is invoked before receiving this callback.
+  // So, always check on_message_callback_ for non-null.
+  if (on_message_callback_) {
+    on_message_callback_->Run(sent);
+    on_message_callback_.reset();
+  }
+}
+
+}  // namespace media_router