[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 42 matching lines @@
       base::Callback<void(int channel_id, ChannelError error_state)>;
 
   CastSocketService();
 
   // 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;
 
   // 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 of the remote host.
   // |channel_auth|: Authentication method used for connecting to a Cast
   //                 receiver. |channel_auth| must not be CHANNEL_AUTH_NONE.
   // |net_log|: Log of socket events.
   // |connect_timeout|: Connection timeout interval.
@@ skipping 12 matching lines @@
                  uint64_t device_capabilities,
                  const OnOpenCallback& open_cb);
 
   // 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 endpoint to be connected to.
   // |net_log|: Net log passed to cast socket.
   // |open_cb|: first parameter is channel id of newly created socket; second
   // parameter is channel error encountered during channel opening.
-  int OpenSocket(const net::IPEndPoint& ip_endpoint,
-                 net::NetLog* net_log,
-                 const OnOpenCallback& open_cb);
+  virtual int OpenSocket(const net::IPEndPoint& ip_endpoint,
+                         net::NetLog* net_log,
+                         const OnOpenCallback& open_cb);
 
   // Register read delegate to cast socket with |channel_id|. Returns false if
   // such socket does not exist.
   bool RegisterDelegate(int channel_id,
                         std::unique_ptr<CastTransport::Delegate> delegate);
 
   // Sets the CastSocket instance to be used for testing.
   void SetSocketForTest(
       std::unique_ptr<cast_channel::CastSocket> socket_for_test);
 
   // Sets injected ping timeout timer for testing.
   void SetPingTimeoutTimerForTest(std::unique_ptr<base::Timer> timer);
 
+ protected:
+  ~CastSocketService() override;
+
  private:
   FRIEND_TEST_ALL_PREFIXES(CastSocketServiceTest, TestOpenChannel);
   friend class CastSocketServiceTest;
 
   class CastSocketRecord {
    public:
     // Does not take ownership of |message_handler|.
     CastSocketRecord(std::unique_ptr<CastSocket> socket,
                      const OnOpenCallback& on_open_callback,
                      PassThroughMessageHandler* message_handler);
 
     ~CastSocketRecord();
 
     // Underlying cast socket.
     std::unique_ptr<CastSocket> cast_socket;
 
     // Pending callbacks to be invoked when cast socket opens.
     std::vector<OnOpenCallback> pending_on_open_callbacks;
 
     // Real read delegate for incoming errors and messages. Owned by |socket|'s
     // keep alive delegate.
     PassThroughMessageHandler* pass_through_message_handler;
   };
 
-  ~CastSocketService() override;
-
   // RefcountedKeyedService implementation.
   void ShutdownOnUIThread() override;
 
   // Creates a socket record, adds it to |socket_records_| and returns the
   // new channel_id. Takes ownership of |socket|.
   // |socket|: Underlying cast socket.
   // |on_open_callback|: Callback to be invoked when cast socket opens.
   // |message_handler|: Real read delegate for incoming errors and messages.
   // Does not take ownership of |message_handler|.
   int AddSocketRecord(std::unique_ptr<CastSocket> socket,
@@ skipping 27 matching lines @@
   std::unique_ptr<base::Timer> injected_timeout_timer_;
 
   THREAD_CHECKER(thread_checker_);
 
   DISALLOW_COPY_AND_ASSIGN(CastSocketService);
 };
 
 }  // namespace cast_channel
 
 #endif  // COMPONENTS_CAST_CHANNEL_CAST_CHANNEL_SERVICE_H_