src/remoting/protocol/connection_to_client.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef REMOTING_PROTOCOL_CONNECTION_TO_CLIENT_H_
   6 #define REMOTING_PROTOCOL_CONNECTION_TO_CLIENT_H_
   7
   8 #include <deque>
   9 #include <string>
  10 #include <vector>
  11
  12 #include "base/memory/scoped_ptr.h"
  13 #include "base/synchronization/lock.h"
  14 #include "base/threading/non_thread_safe.h"
  15 #include "remoting/protocol/audio_writer.h"
  16 #include "remoting/protocol/session.h"
  17 #include "remoting/protocol/video_writer.h"
  18
  19 namespace net {
  20 class IPEndPoint;
  21 }  // namespace net
  22
  23 namespace remoting {
  24 namespace protocol {
  25
  26 class ClientStub;
  27 class ClipboardStub;
  28 class HostStub;
  29 class InputStub;
  30 class HostControlDispatcher;
  31 class HostEventDispatcher;
  32
  33 // This class represents a remote viewer connection to the chromoting
  34 // host. It sets up all protocol channels and connects them to the
  35 // stubs.
  36 class ConnectionToClient : public base::NonThreadSafe,
  37                            public Session::EventHandler {
  38  public:
  39   class EventHandler {
  40    public:
  41     // Called when the network connection is authenticated.
  42     virtual void OnConnectionAuthenticated(ConnectionToClient* connection) = 0;
  43
  44     // Called when the network connection is authenticated and all
  45     // channels are connected.
  46     virtual void OnConnectionChannelsConnected(
  47         ConnectionToClient* connection) = 0;
  48
  49     // Called when the network connection is closed or failed.
  50     virtual void OnConnectionClosed(ConnectionToClient* connection,
  51                                     ErrorCode error) = 0;
  52
  53     // Called when sequence number is updated.
  54     virtual void OnSequenceNumberUpdated(ConnectionToClient* connection,
  55                                          int64 sequence_number) = 0;
  56
  57     // Called on notification of a route change event, which happens when a
  58     // channel is connected.
  59     virtual void OnRouteChange(ConnectionToClient* connection,
  60                                const std::string& channel_name,
  61                                const TransportRoute& route) = 0;
  62
  63    protected:
  64     virtual ~EventHandler() {}
  65   };
  66
  67   // Constructs a ConnectionToClient object for the |session|. Takes
  68   // ownership of |session|.
  69   explicit ConnectionToClient(Session* session);
  70   virtual ~ConnectionToClient();
  71
  72   // Set |event_handler| for connection events. Must be called once when this
  73   // object is created.
  74   void SetEventHandler(EventHandler* event_handler);
  75
  76   // Returns the connection in use.
  77   virtual Session* session();
  78
  79   // Disconnect the client connection.
  80   virtual void Disconnect();
  81
  82   // Update the sequence number when received from the client. EventHandler
  83   // will be called.
  84   virtual void UpdateSequenceNumber(int64 sequence_number);
  85
  86   // Get the stubs used by the host to transmit messages to the client.
  87   // The stubs must not be accessed before OnConnectionAuthenticated(), or
  88   // after OnConnectionClosed().
  89   // Note that the audio stub will be NULL if audio is not enabled.
  90   virtual VideoStub* video_stub();
  91   virtual AudioStub* audio_stub();
  92   virtual ClientStub* client_stub();
  93
  94   // Set/get the stubs which will handle messages we receive from the client.
  95   // All stubs MUST be set before the session's channels become connected.
  96   virtual void set_clipboard_stub(ClipboardStub* clipboard_stub);
  97   virtual ClipboardStub* clipboard_stub();
  98   virtual void set_host_stub(HostStub* host_stub);
  99   virtual HostStub* host_stub();
 100   virtual void set_input_stub(InputStub* input_stub);
 101   virtual InputStub* input_stub();
 102
 103   // Session::EventHandler interface.
 104   virtual void OnSessionStateChange(Session::State state) OVERRIDE;
 105   virtual void OnSessionRouteChange(const std::string& channel_name,
 106                                     const TransportRoute& route) OVERRIDE;
 107
 108  private:
 109   // Callback for channel initialization.
 110   void OnChannelInitialized(bool successful);
 111
 112   void NotifyIfChannelsReady();
 113
 114   void Close(ErrorCode error);
 115
 116   // Stops writing in the channels.
 117   void CloseChannels();
 118
 119   // Event handler for handling events sent from this object.
 120   EventHandler* handler_;
 121
 122   // Stubs that are called for incoming messages.
 123   ClipboardStub* clipboard_stub_;
 124   HostStub* host_stub_;
 125   InputStub* input_stub_;
 126
 127   // The libjingle channel used to send and receive data from the remote client.
 128   scoped_ptr<Session> session_;
 129
 130   scoped_ptr<HostControlDispatcher> control_dispatcher_;
 131   scoped_ptr<HostEventDispatcher> event_dispatcher_;
 132   scoped_ptr<VideoWriter> video_writer_;
 133   scoped_ptr<AudioWriter> audio_writer_;
 134
 135   DISALLOW_COPY_AND_ASSIGN(ConnectionToClient);
 136 };
 137
 138 }  // namespace protocol
 139 }  // namespace remoting
 140
 141 #endif  // REMOTING_PROTOCOL_CONNECTION_TO_CLIENT_H_