[Media Router] Add CastMediaSinkService

components/cast_channel/cast_socket_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 COMPONENTS_CAST_CHANNEL_CAST_CHANNEL_SERVICE_H_
 #define COMPONENTS_CAST_CHANNEL_CAST_CHANNEL_SERVICE_H_
 
 #include <map>
 #include <memory>
 
@@ skipping 19 matching lines @@
   // Adds |socket| to |sockets_| and returns raw pointer of |socket|. Takes
   // ownership of |socket|.
   CastSocket* AddSocket(std::unique_ptr<CastSocket> socket);
 
   // Removes the CastSocket corresponding to |channel_id| from the
   // CastSocketRegistry. Returns nullptr if no such CastSocket exists.
   std::unique_ptr<CastSocket> RemoveSocket(int channel_id);
 
   // Returns the socket corresponding to |channel_id| if one exists, or nullptr
   // otherwise.
-  CastSocket* GetSocket(int channel_id) const;
+  virtual CastSocket* GetSocket(int channel_id) const;
 
   CastSocket* GetSocket(const net::IPEndPoint& ip_endpoint) const;
 
   // Opens cast socket with |ip_endpoint| and invokes |open_cb| when opening
   // operation finishes. If cast socket with |ip_endpoint| already exists,
   // invoke |open_cb| directly with existing socket's channel ID.
   // Parameters:
   // |ip_endpoint|: IP address and port of the remote host.
   // |net_log|: Log of socket events.
   // |connect_timeout|: Connection timeout interval.
@@ skipping 14 matching lines @@
                  CastSocket::Observer* observer);
 
   // Opens cast socket with |ip_endpoint| and invokes |open_cb| when opening
   // operation finishes. If cast socket with |ip_endpoint| already exists,
   // invoke |open_cb| directly with existing socket's channel ID.
   // |ip_endpoint|: IP address and port of the remote host.
   // |net_log|: Log of socket events.
   // |open_cb|: OnOpenCallback invoked when cast socket is opened.
   // |observer|: Observer handles messages and errors on newly opened socket.
   // Does not take ownership of |observer|.
-  int OpenSocket(const net::IPEndPoint& ip_endpoint,
-                 net::NetLog* net_log,
-                 const CastSocket::OnOpenCallback& open_cb,
-                 CastSocket::Observer* observer);
+  virtual int OpenSocket(const net::IPEndPoint& ip_endpoint,
+                         net::NetLog* net_log,
+                         const CastSocket::OnOpenCallback& open_cb,
+                         CastSocket::Observer* observer);
 
   // Returns an observer corresponding to |id|.
   CastSocket::Observer* GetObserver(const std::string& id);
 
   // Adds |observer| to |socket_observer_map_| keyed by |id|. Return raw pointer
   // of the newly added observer.
   CastSocket::Observer* AddObserver(
       const std::string& id,
       std::unique_ptr<CastSocket::Observer> observer);
 
   // Allow test to inject a mock cast socket.
   void SetSocketForTest(std::unique_ptr<CastSocket> socket_for_test);
 
- private:
+ protected:
   ~CastSocketService() override;
 
+ private:
   // RefcountedKeyedService implementation.
   void ShutdownOnUIThread() override;
 
   // Used to generate CastSocket id.
   static int last_channel_id_;
 
   // The collection of CastSocket keyed by channel_id.
   std::map<int, std::unique_ptr<CastSocket>> sockets_;
 
   // Map of CastSocket::Observer keyed by observer id. For extension side
   // observers, id is extension_id; For browser side observers, id is a hard
   // coded string.
   std::map<std::string, std::unique_ptr<CastSocket::Observer>>
       socket_observer_map_;
 
   scoped_refptr<cast_channel::Logger> logger_;
 
   std::unique_ptr<CastSocket> socket_for_test_;
 
   THREAD_CHECKER(thread_checker_);
 
   DISALLOW_COPY_AND_ASSIGN(CastSocketService);
 };
 
 }  // namespace cast_channel
 
 #endif  // COMPONENTS_CAST_CHANNEL_CAST_CHANNEL_SERVICE_H_