[Media Router] Add CastMediaSinkService

chrome/browser/media/router/discovery/mdns/cast_media_sink_service.cc

Index: chrome/browser/media/router/discovery/mdns/cast_media_sink_service.cc
diff --git a/chrome/browser/media/router/discovery/mdns/cast_media_sink_service.cc b/chrome/browser/media/router/discovery/mdns/cast_media_sink_service.cc
new file mode 100644
index 0000000000000000000000000000000000000000..fc2de0046599b810484209d7cc5c80ae710afc0e
--- /dev/null
+++ b/chrome/browser/media/router/discovery/mdns/cast_media_sink_service.cc
@@ -0,0 +1,207 @@
+// Copyright 2017 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/discovery/mdns/cast_media_sink_service.h"
+
+#include "chrome/browser/browser_process.h"
+#include "chrome/common/media_router/discovery/media_sink_internal.h"
+#include "components/cast_channel/cast_socket_service.h"
+#include "components/cast_channel/cast_socket_service_factory.h"
+#include "components/net_log/chrome_net_log.h"
+#include "content/public/common/content_client.h"
+#include "net/base/host_port_pair.h"
+#include "net/base/ip_address.h"
+
+namespace {
+// mDNS service types.
+const char kCastServiceType[] = "_googlecast._tcp.local";
+
+enum ErrorType {
+  NONE,
+  NOT_CAST_DEVICE,
+  MISSING_ID,
+  MISSING_FRIENDLY_NAME,
+  MISSING_OR_INVALID_IP_ADDRESS,
+  MISSING_OR_INVALID_PORT,
+};
+
+ErrorType CreateCastMediaSink(const media_router::DnsSdService& service,
+                              int channel_id,
+                              bool audio_only,
+                              media_router::MediaSinkInternal* cast_sink) {
+  DCHECK(cast_sink);
+  if (service.service_name.find(kCastServiceType) == std::string::npos)
+    return ErrorType::NOT_CAST_DEVICE;
+
+  net::IPAddress ip_address;
+  if (!ip_address.AssignFromIPLiteral(service.ip_address))
+    return ErrorType::MISSING_OR_INVALID_IP_ADDRESS;
+
+  std::map<std::string, std::string> service_data;
+  for (const auto& item : service.service_data) {
+    // |item| format should be "id=xxxxxx", etc.
+    size_t split_idx = item.find('=');
+    if (split_idx == std::string::npos)
+      continue;
+
+    std::string key = item.substr(0, split_idx);
+    std::string val = item.substr(split_idx + 1);
+    service_data[key] = val;
+  }
+
+  // When use this "sink" within browser, please note it will have a different
+  // ID when it is sent to the extension, because it derives a different sink ID
+  // using the given sink ID.
+  std::string unique_id = service_data["id"];
+  if (unique_id.empty())
+    return ErrorType::MISSING_ID;
+  std::string friendly_name = service_data["fn"];
+  if (friendly_name.empty())
+    return ErrorType::MISSING_FRIENDLY_NAME;
+  media_router::MediaSink sink(unique_id, friendly_name,
+                               media_router::MediaSink::IconType::CAST);
+
+  media_router::CastSinkExtraData extra_data;
+  extra_data.ip_address = ip_address;
+  extra_data.model_name = service_data["md"];
+  extra_data.capabilities = cast_channel::CastDeviceCapability::AUDIO_OUT;
+  if (!audio_only)
+    extra_data.capabilities |= cast_channel::CastDeviceCapability::VIDEO_OUT;
+  extra_data.cast_channel_id = channel_id;
+
+  cast_sink->set_sink(sink);
+  cast_sink->set_cast_data(extra_data);
+
+  return ErrorType::NONE;
+}
+
+}  // namespace
+
+namespace media_router {
+
+CastMediaSinkService::CastMediaSinkService(
+    const OnSinksDiscoveredCallback& callback,
+    content::BrowserContext* browser_context)
+    : MediaSinkServiceBase(callback) {
+  DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+  cast_socket_service_ = cast_channel::CastSocketServiceFactory::GetInstance()

[imcheng, 2017/06/15 00:50:59]

Would we need to add a DependsOn from MediaRouterFactory to
CastSocketServiceFactory (until we make CastSocketService a singleton)?

[zhaobin, 2017/06/20 02:30:28]

It seems fine since we call
cast_channel::CastSocketServiceFactory::GetInstance() explicitly here?

+                             ->GetForBrowserContext(browser_context);
+  DCHECK(cast_socket_service_);
+}
+
+CastMediaSinkService::~CastMediaSinkService() {}
+
+void CastMediaSinkService::Start() {
+  DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+  if (dns_sd_registry_)
+    return;
+
+  dns_sd_registry_ = test_dns_sd_registry_ ? test_dns_sd_registry_
+                                           : DnsSdRegistry::GetInstance();
+  dns_sd_registry_->AddObserver(this);
+  dns_sd_registry_->RegisterDnsSdListener(kCastServiceType);
+  MediaSinkServiceBase::StartTimer();
+}
+
+void CastMediaSinkService::Stop() {
+  DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+  if (!dns_sd_registry_)
+    return;
+
+  dns_sd_registry_->UnregisterDnsSdListener(kCastServiceType);
+  dns_sd_registry_->RemoveObserver(this);
+  dns_sd_registry_ = nullptr;
+  MediaSinkServiceBase::StopTimer();
+}
+
+void CastMediaSinkService::SetDnsSdRegistryForTesting(DnsSdRegistry* registry) {
+  DCHECK(!test_dns_sd_registry_);
+  test_dns_sd_registry_ = registry;
+}
+
+void CastMediaSinkService::SetCastSocketServiceForTesting(
+    cast_channel::CastSocketService* cast_socket_service) {
+  cast_socket_service_ = cast_socket_service;

[imcheng, 2017/06/15 00:50:56]

Add a DCHECK to be consistent with above.

[zhaobin, 2017/06/20 02:30:28]

Moved this to a ctor used by test.

+}
+
+void CastMediaSinkService::OnDnsSdEvent(
+    const std::string& service_type,
+    const DnsSdRegistry::DnsSdServiceList& services) {
+  DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
+  DVLOG(2) << "CastMediaSinkService::OnDnsSdEvent found " << services.size()
+           << " services";
+
+  current_sinks_.clear();
+  current_services_ = services;
+
+  for (const auto& service : services) {
+    net::IPAddress ip_address;
+    if (!ip_address.AssignFromIPLiteral(service.ip_address)) {
+      DVLOG(2) << "Invalid ip_addresss: " << service.ip_address;
+      continue;
+    }
+    net::HostPortPair host_port_pair =
+        net::HostPortPair::FromString(service.service_host_port);
+
+    content::BrowserThread::PostTask(
+        content::BrowserThread::IO, FROM_HERE,
+        base::Bind(&CastMediaSinkService::OpenChannelOnIOThread, this, service,
+                   net::IPEndPoint(ip_address, host_port_pair.port())));
+  }
+}
+
+void CastMediaSinkService::OpenChannelOnIOThread(
+    const DnsSdService& service,
+    const net::IPEndPoint& ip_endpoint) {
+  cast_socket_service_->OpenSocket(

[imcheng, 2017/06/15 00:50:57]

What if the socket already exists (opened by browser or the extension before)?

[zhaobin, 2017/06/20 02:30:28]

Handled inside CastSocketService::OpenSocket().

+      ip_endpoint, g_browser_process->net_log(),
+      base::Bind(&CastMediaSinkService::OnChannelOpenedOnIOThread, this,
+                 service));
+}
+
+void CastMediaSinkService::OnChannelOpenedOnIOThread(
+    const DnsSdService& service,
+    int channel_id,

[imcheng, 2017/06/15 00:50:56]

Would it be possible to pass back the CastSocket in the callback?

[zhaobin, 2017/06/20 02:30:28]

DISALLOW_COPY_AND_ASSIGN(CastSocketImpl); 

Do not feel like changing this...

+    cast_channel::ChannelError channel_error) {
+  if (channel_error != cast_channel::ChannelError::NONE) {
+    DVLOG(2) << "Fail to open channel " << service.ip_address << ": "
+             << service.service_host_port
+             << " [ChannelError]: " << (int)channel_error;
+    return;
+  }
+
+  auto* socket = cast_socket_service_->GetSocket(channel_id);
+  if (!socket) {
+    DVLOG(2) << "Fail to find socket with [channel_id]: " << channel_id;
+    return;
+  }
+
+  content::BrowserThread::PostTask(
+      content::BrowserThread::UI, FROM_HERE,
+      base::Bind(&CastMediaSinkService::OnChannelOpenedOnUIThread, this,
+                 service, channel_id, socket->audio_only()));
+}
+
+void CastMediaSinkService::OnChannelOpenedOnUIThread(
+    const DnsSdService& service,
+    int channel_id,
+    bool audio_only) {
+  MediaSinkInternal sink;

[imcheng, 2017/06/15 00:50:56]

DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);

[zhaobin, 2017/06/20 02:30:28]

Done.

+  ErrorType error = CreateCastMediaSink(service, channel_id, audio_only, &sink);
+  if (error != ErrorType::NONE) {
+    DVLOG(2) << "Fail to create Cast device [error]: " << error;
+    return;
+  }
+
+  if (!base::ContainsValue(current_services_, service)) {

[imcheng, 2017/06/15 00:50:56]

This check means we would drop the sink if we somehow had a OnDnsSdEvent not
containing |service|. If we didn't do that, would the sink be eventually removed
anyway if the channel errors out?

[zhaobin, 2017/06/20 02:30:28]

DialMediaSinkService has similar logic so copied it here.

Yes, if we remove sinks when observing any socket error (not sure if we want to
do that, seems a little aggressive)

+    DVLOG(2) << "Service data not found in current service data list...";
+    return;
+  }
+
+  DVLOG(2) << "Ading sink to current_sinks_ [id]: " << sink.sink().id();
+  current_sinks_.insert(sink);
+  MediaSinkServiceBase::RestartTimer();
+}
+
+}  // namespace media_router