[Media Router] Support dual discovery

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

 // 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.
 
 #ifndef CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_MDNS_CAST_MEDIA_SINK_SERVICE_H_
 #define CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_MDNS_CAST_MEDIA_SINK_SERVICE_H_
 
 #include <memory>
 #include <set>
 
 #include "base/gtest_prod_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/threading/thread_checker.h"
+#include "chrome/browser/media/router/discovery/dial/dial_media_sink_service_impl.h"
 #include "chrome/browser/media/router/discovery/mdns/dns_sd_delegate.h"
 #include "chrome/browser/media/router/discovery/mdns/dns_sd_registry.h"
 #include "chrome/browser/media/router/discovery/media_sink_service_base.h"
 #include "components/cast_channel/cast_channel_enum.h"
 #include "components/cast_channel/cast_socket.h"
 #include "net/base/ip_endpoint.h"
 
 namespace cast_channel {
 class CastSocketService;
 }  // namespace cast_channel
 
 namespace content {
 class BrowserContext;
 }  // namespace content
 
 namespace media_router {
 
 // A service which can be used to start background discovery and resolution of
 // Cast devices.
 // Public APIs should be invoked on the UI thread.
-class CastMediaSinkService
-    : public MediaSinkServiceBase,
-      public DnsSdRegistry::DnsSdObserver,
-      public base::RefCountedThreadSafe<CastMediaSinkService> {
+class CastMediaSinkService : public MediaSinkServiceBase,
+                             public DialMediaSinkServiceDelegate,
+                             public DnsSdRegistry::DnsSdObserver {
  public:
   CastMediaSinkService(const OnSinksDiscoveredCallback& callback,
                        content::BrowserContext* browser_context);
 
   // Used by unit tests.
   CastMediaSinkService(const OnSinksDiscoveredCallback& callback,
                        cast_channel::CastSocketService* cast_socket_service);
 
   // mDNS service types.
   static const char kCastServiceType[];
 
+  // Default Cast control port to open Cast Socket from DIAL sink.
+  static const int kCastControlPort;
+
   // MediaSinkService implementation
   void Start() override;
   void Stop() override;
 
+  // DialMediaSinkServiceDelegate implementation.
+  void OnDialSinkAdded(const MediaSinkInternal& sink) override;
+  void OnDialSinksRemoved() override;
+
  protected:
   // Used to mock out the DnsSdRegistry for testing.
   void SetDnsSdRegistryForTest(DnsSdRegistry* registry);
 
   ~CastMediaSinkService() override;
 
  private:
   // Receives incoming messages and errors and provides additional API context.
   class CastSocketObserver : public cast_channel::CastSocket::Observer {
    public:
@@ skipping 15 matching lines @@
 
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestReStartAfterStop);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestMultipleStartAndStop);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest,
                            TestOnChannelOpenedOnIOThread);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest,
                            TestMultipleOnChannelOpenedOnIOThread);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestOnDnsSdEvent);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestMultipleOnDnsSdEvent);
   FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestTimer);
+  FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestOnDialSinkAdded);
+  FRIEND_TEST_ALL_PREFIXES(CastMediaSinkServiceTest, TestOnFetchCompleted);
+
+  // MediaSinkServiceBase implementation
+  void OnFetchCompleted() override;
+
+  // Returns cast sinks from |current_sinks_by_mdns_map_| and
+  // |current_sinks_by_dial_map_|.
+  std::set<MediaSinkInternal> GetCastSinksOnIOThread();
+
+  // Stores |cast_sinks| in |current_sinks_| and invokes
+  // MediaSinkServiceBase::OnFetchCompleted on the UI thread.
+  void OnFetchCompletedOnUIThread(std::set<MediaSinkInternal> cast_sinks);
 
   // DnsSdRegistry::DnsSdObserver implementation
   void OnDnsSdEvent(const std::string& service_type,
                     const DnsSdRegistry::DnsSdServiceList& services) override;
 
+  // Opens cast channels on the IO thread.
+  void OpenChannelsOnIOThread(std::vector<MediaSinkInternal> cast_sinks);
+
   // Opens cast channel on IO thread.
-  // |service|: mDNS service description.
   // |ip_endpoint|: cast channel's target IP endpoint.
-  void OpenChannelOnIOThread(const DnsSdService& service,
-                             const net::IPEndPoint& ip_endpoint);
+  // |cast_sink|: Cast sink created from mDNS service description or DIAL sink.
+  void OpenChannelOnIOThread(const net::IPEndPoint& ip_endpoint,
+                             MediaSinkInternal cast_sink);
 
   // Invoked when opening cast channel on IO thread completes.
-  // |service|: mDNS service description.
+  // |cast_sink|: Cast sink created from mDNS service description or DIAL sink.
   // |channel_id|: channel id of newly created cast channel.
   // |channel_error|: error encounted when opending cast channel.
-  void OnChannelOpenedOnIOThread(const DnsSdService& service,
+  void OnChannelOpenedOnIOThread(MediaSinkInternal cast_sink,
                                  int channel_id,
                                  cast_channel::ChannelError channel_error);
 
-  // Invoked by |OnChannelOpenedOnIOThread| to post task on UI thread.
-  // |service|: mDNS service description.
-  // |channel_id|: channel id of newly created cast channel.
-  // |audio_only|: if cast channel is audio only or not.
-  void OnChannelOpenedOnUIThread(const DnsSdService& service,
-                                 int channel_id,
-                                 bool audio_only);
-
   // Raw pointer to DnsSdRegistry instance, which is a global leaky singleton
   // and lives as long as the browser process.
   DnsSdRegistry* dns_sd_registry_ = nullptr;
 
-  // Service list from current round of discovery.
-  DnsSdRegistry::DnsSdServiceList current_services_;
+  // Set of mDNS service ip addresses from current round of discovery.
+  std::set<net::IPEndPoint> current_service_ip_endpoints_;
+
+  using MediaSinkInternalMap = std::map<net::IPEndPoint, MediaSinkInternal>;
+
+  // Map of sinks from current round of mDNS discovery keyed by IP address.
+  MediaSinkInternalMap current_sinks_by_mdns_;
+
+  // Map of sinks from current round of DIAL discovery keyed by IP address.
+  MediaSinkInternalMap current_sinks_by_dial_;
 
   // Service managing creating and removing cast channels.
   scoped_refptr<cast_channel::CastSocketService> cast_socket_service_;
 
-  THREAD_CHECKER(thread_checker_);
-
   DISALLOW_COPY_AND_ASSIGN(CastMediaSinkService);
 };
 
 }  // namespace media_router
 
 #endif  // CHROME_BROWSER_MEDIA_ROUTER_DISCOVERY_MDNS_CAST_MEDIA_SINK_SERVICE_H_