Home | History | Annotate | Download | only in client 
      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 #include "remoting/client/chromoting_client.h"
      6 
      7 #include "base/bind.h"
      8 #include "remoting/base/capabilities.h"
      9 #include "remoting/client/audio_decode_scheduler.h"
     10 #include "remoting/client/audio_player.h"
     11 #include "remoting/client/client_context.h"
     12 #include "remoting/client/client_user_interface.h"
     13 #include "remoting/client/rectangle_update_decoder.h"
     14 #include "remoting/proto/audio.pb.h"
     15 #include "remoting/proto/video.pb.h"
     16 #include "remoting/protocol/authentication_method.h"
     17 #include "remoting/protocol/connection_to_host.h"
     18 #include "remoting/protocol/host_stub.h"
     19 #include "remoting/protocol/negotiating_client_authenticator.h"
     20 #include "remoting/protocol/session_config.h"
     21 #include "remoting/protocol/transport.h"
     22 
     23 namespace remoting {
     24 
     25 using protocol::AuthenticationMethod;
     26 
     27 ChromotingClient::ChromotingClient(
     28     const ClientConfig& config,
     29     ClientContext* client_context,
     30     protocol::ConnectionToHost* connection,
     31     ClientUserInterface* user_interface,
     32     scoped_refptr<FrameConsumerProxy> frame_consumer,
     33     scoped_ptr<AudioPlayer> audio_player)
     34     : config_(config),
     35       task_runner_(client_context->main_task_runner()),
     36       connection_(connection),
     37       user_interface_(user_interface),
     38       host_capabilities_received_(false),
     39       weak_factory_(this) {
     40   rectangle_decoder_ =
     41       new RectangleUpdateDecoder(client_context->main_task_runner(),
     42                                  client_context->decode_task_runner(),
     43                                  frame_consumer);
     44   if (audio_player) {
     45     audio_decode_scheduler_.reset(new AudioDecodeScheduler(
     46         client_context->main_task_runner(),
     47         client_context->audio_decode_task_runner(),
     48         audio_player.Pass()));
     49   }
     50 }
     51 
     52 ChromotingClient::~ChromotingClient() {
     53 }
     54 
     55 void ChromotingClient::Start(
     56     SignalStrategy* signal_strategy,
     57     scoped_ptr<protocol::TransportFactory> transport_factory) {
     58   DCHECK(task_runner_->BelongsToCurrentThread());
     59 
     60   scoped_ptr<protocol::Authenticator> authenticator(
     61       new protocol::NegotiatingClientAuthenticator(
     62           config_.client_pairing_id,
     63           config_.client_paired_secret,
     64           config_.authentication_tag,
     65           config_.fetch_secret_callback,
     66           user_interface_->GetTokenFetcher(config_.host_public_key),
     67           config_.authentication_methods));
     68 
     69   // Create a WeakPtr to ourself for to use for all posted tasks.
     70   weak_ptr_ = weak_factory_.GetWeakPtr();
     71 
     72   connection_->Connect(signal_strategy,
     73                        config_.host_jid,
     74                        config_.host_public_key,
     75                        transport_factory.Pass(),
     76                        authenticator.Pass(),
     77                        this,
     78                        this,
     79                        this,
     80                        rectangle_decoder_.get(),
     81                        audio_decode_scheduler_.get());
     82 }
     83 
     84 FrameProducer* ChromotingClient::GetFrameProducer() {
     85   return rectangle_decoder_.get();
     86 }
     87 
     88 ChromotingStats* ChromotingClient::GetStats() {
     89   DCHECK(task_runner_->BelongsToCurrentThread());
     90   return rectangle_decoder_->GetStats();
     91 }
     92 
     93 void ChromotingClient::SetCapabilities(
     94     const protocol::Capabilities& capabilities) {
     95   DCHECK(task_runner_->BelongsToCurrentThread());
     96 
     97   // Only accept the first |protocol::Capabilities| message.
     98   if (host_capabilities_received_) {
     99     LOG(WARNING) << "protocol::Capabilities has been received already.";
    100     return;
    101   }
    102 
    103   host_capabilities_received_ = true;
    104   if (capabilities.has_capabilities())
    105     host_capabilities_ = capabilities.capabilities();
    106 
    107   VLOG(1) << "Host capabilities: " << host_capabilities_;
    108 
    109   // Calculate the set of capabilities enabled by both client and host and pass
    110   // it to the webapp.
    111   user_interface_->SetCapabilities(
    112       IntersectCapabilities(config_.capabilities, host_capabilities_));
    113 }
    114 
    115 void ChromotingClient::SetPairingResponse(
    116     const protocol::PairingResponse& pairing_response) {
    117   DCHECK(task_runner_->BelongsToCurrentThread());
    118 
    119   user_interface_->SetPairingResponse(pairing_response);
    120 }
    121 
    122 void ChromotingClient::DeliverHostMessage(
    123     const protocol::ExtensionMessage& message) {
    124   DCHECK(task_runner_->BelongsToCurrentThread());
    125 
    126   user_interface_->DeliverHostMessage(message);
    127 }
    128 
    129 void ChromotingClient::InjectClipboardEvent(
    130     const protocol::ClipboardEvent& event) {
    131   DCHECK(task_runner_->BelongsToCurrentThread());
    132 
    133   user_interface_->GetClipboardStub()->InjectClipboardEvent(event);
    134 }
    135 
    136 void ChromotingClient::SetCursorShape(
    137     const protocol::CursorShapeInfo& cursor_shape) {
    138   DCHECK(task_runner_->BelongsToCurrentThread());
    139 
    140   user_interface_->GetCursorShapeStub()->SetCursorShape(cursor_shape);
    141 }
    142 
    143 void ChromotingClient::OnConnectionState(
    144     protocol::ConnectionToHost::State state,
    145     protocol::ErrorCode error) {
    146   DCHECK(task_runner_->BelongsToCurrentThread());
    147   VLOG(1) << "ChromotingClient::OnConnectionState(" << state << ")";
    148 
    149   if (state == protocol::ConnectionToHost::AUTHENTICATED) {
    150     OnAuthenticated();
    151   } else if (state == protocol::ConnectionToHost::CONNECTED) {
    152     OnChannelsConnected();
    153   }
    154   user_interface_->OnConnectionState(state, error);
    155 }
    156 
    157 void ChromotingClient::OnConnectionReady(bool ready) {
    158   VLOG(1) << "ChromotingClient::OnConnectionReady(" << ready << ")";
    159   user_interface_->OnConnectionReady(ready);
    160 }
    161 
    162 void ChromotingClient::OnAuthenticated() {
    163   DCHECK(task_runner_->BelongsToCurrentThread());
    164 
    165   // Initialize the decoder.
    166   rectangle_decoder_->Initialize(connection_->config());
    167   if (connection_->config().is_audio_enabled())
    168     audio_decode_scheduler_->Initialize(connection_->config());
    169 
    170   // Do not negotiate capabilities with the host if the host does not support
    171   // them.
    172   if (!connection_->config().SupportsCapabilities()) {
    173     VLOG(1) << "The host does not support any capabilities.";
    174 
    175     host_capabilities_received_ = true;
    176     user_interface_->SetCapabilities(host_capabilities_);
    177   }
    178 }
    179 
    180 void ChromotingClient::OnChannelsConnected() {
    181   DCHECK(task_runner_->BelongsToCurrentThread());
    182 
    183   // Negotiate capabilities with the host.
    184   if (connection_->config().SupportsCapabilities()) {
    185     VLOG(1) << "Client capabilities: " << config_.capabilities;
    186 
    187     protocol::Capabilities capabilities;
    188     capabilities.set_capabilities(config_.capabilities);
    189     connection_->host_stub()->SetCapabilities(capabilities);
    190   }
    191 }
    192 
    193 }  // namespace remoting
    194