Enable WebRTC support in the remoting host.

remoting/host/remoting_me2me_host.cc

 // Copyright (c) 2012 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.
 //
 // This file implements a standalone host process for Me2Me.
 
 #include <string>
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/debug/alias.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringize_macros.h"
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "ipc/ipc_channel.h"
 #include "ipc/ipc_channel_proxy.h"
 #include "ipc/ipc_listener.h"
+#include "jingle/glue/thread_wrapper.h"
 #include "net/base/net_util.h"
 #include "net/base/network_change_notifier.h"
 #include "net/socket/client_socket_factory.h"
 #include "net/socket/ssl_server_socket.h"
 #include "net/url_request/url_fetcher.h"
 #include "policy/policy_constants.h"
 #include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/base/breakpad.h"
 #include "remoting/base/constants.h"
 #include "remoting/base/logging.h"
@@ skipping 29 matching lines @@
 #include "remoting/host/session_manager_factory.h"
 #include "remoting/host/shutdown_watchdog.h"
 #include "remoting/host/signaling_connector.h"
 #include "remoting/host/single_window_desktop_environment.h"
 #include "remoting/host/switches.h"
 #include "remoting/host/third_party_auth_config.h"
 #include "remoting/host/token_validator_factory_impl.h"
 #include "remoting/host/usage_stats_consent.h"
 #include "remoting/host/username.h"
 #include "remoting/host/video_frame_recorder_host_extension.h"
+#include "remoting/protocol/authenticator.h"
+#include "remoting/protocol/channel_authenticator.h"
+#include "remoting/protocol/chromium_port_allocator_factory.h"
+#include "remoting/protocol/jingle_session_manager.h"
 #include "remoting/protocol/me2me_host_authenticator_factory.h"
 #include "remoting/protocol/network_settings.h"
 #include "remoting/protocol/pairing_registry.h"
 #include "remoting/protocol/port_range.h"
 #include "remoting/protocol/token_validator.h"
+#include "remoting/protocol/webrtc_transport.h"
 #include "remoting/signaling/push_notification_subscriber.h"
 #include "remoting/signaling/xmpp_signal_strategy.h"
+#include "third_party/webrtc/base/scoped_ref_ptr.h"
 
 #if defined(OS_POSIX)
 #include <signal.h>
 #include <sys/types.h>
 #include <unistd.h>
 #include "base/file_descriptor_posix.h"
 #include "remoting/host/pam_authorization_factory_posix.h"
 #include "remoting/host/posix/signal_handler.h"
 #endif  // defined(OS_POSIX)
 
@@ skipping 49 matching lines @@
 const char kSignalParentSwitchName[] = "signal-parent";
 
 // Command line switch used to enable VP9 encoding.
 const char kEnableVp9SwitchName[] = "enable-vp9";
 
 // Command line switch used to enable and configure the frame-recorder.
 const char kFrameRecorderBufferKbName[] = "frame-recorder-buffer-kb";
 
 const char kWindowIdSwitchName[] = "window-id";
 
+// Command line switch used to enable WebRTC-based protocol.
+const char kEnableWebrtc[] = "enable-webrtc";
+
 // Maximum time to wait for clean shutdown to occur, before forcing termination
 // of the process.
 const int kShutdownTimeoutSeconds = 15;
 
 // Maximum time to wait for reporting host-offline-reason to the service,
 // before continuing normal process shutdown.
 const int kHostOfflineReasonTimeoutSeconds = 10;
 
 // Host offline reasons not associated with shutting down the host process
 // and therefore not expressible through HostExitCodes enum.
 const char kHostOfflineReasonPolicyReadError[] = "POLICY_READ_ERROR";
 const char kHostOfflineReasonPolicyChangeRequiresRestart[] =
     "POLICY_CHANGE_REQUIRES_RESTART";
 
 }  // namespace
 
 namespace remoting {
 
+#if !defined(NDEBUG)
+
+// Authenticator that accepts all connections. Use only for testing.
+class NoopAuthenticator : public protocol::Authenticator {
+ public:
+  NoopAuthenticator() {}
+  ~NoopAuthenticator() override {}
+
+  // protocol::Authenticator interface.
+  State state() const override { return done_ ? ACCEPTED : WAITING_MESSAGE; }
+  bool started() const override { return done_; }
+  RejectionReason rejection_reason() const override {
+    NOTREACHED();
+    return INVALID_CREDENTIALS;
+  }
+  void ProcessMessage(const buzz::XmlElement* message,
+                      const base::Closure& resume_callback) override {
+    done_ = true;
+    resume_callback.Run();
+  }
+  scoped_ptr<buzz::XmlElement> GetNextMessage() override {
+    NOTREACHED();
+    return nullptr;
+  }
+  const std::string& GetAuthKey() const override { return auth_key_; }
+  scoped_ptr<protocol::ChannelAuthenticator> CreateChannelAuthenticator()
+      const override {
+    NOTREACHED();
+    return nullptr;
+  };
+
+ private:
+  bool done_ = false;
+  std::string auth_key_ = "NOKEY";
+};
+
+// Factory for Authenticator instances.
+class NoopAuthenticatorFactory : public protocol::AuthenticatorFactory {
+ public:
+  NoopAuthenticatorFactory() {}
+  ~NoopAuthenticatorFactory() override {}
+
+  scoped_ptr<protocol::Authenticator> CreateAuthenticator(
+      const std::string& local_jid,
+      const std::string& remote_jid,
+      const buzz::XmlElement* first_message) override {
+    return make_scoped_ptr(new NoopAuthenticator());
+  }
+};
+
+#endif  // !defined(NDEBUG)
+
 class HostProcess : public ConfigWatcher::Delegate,
                     public HostChangeNotificationListener::Listener,
                     public IPC::Listener,
                     public base::RefCountedThreadSafe<HostProcess> {
  public:
   // |shutdown_watchdog| is armed when shutdown is started, and should be kept
   // alive as long as possible until the process exits (since destroying the
   // watchdog disarms it).
   HostProcess(scoped_ptr<ChromotingHostContext> context,
               int* exit_code_out,
@@ skipping 498 matching lines @@
   ShutdownHost(kSuccessExitCode);
 }
 #endif  // OS_POSIX
 
 void HostProcess::CreateAuthenticatorFactory() {
   DCHECK(context_->network_task_runner()->BelongsToCurrentThread());
 
   if (state_ != HOST_STARTED)
     return;
 
+#if !defined(NDEBUG)
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(kEnableWebrtc)) {
+    host_->SetAuthenticatorFactory(
+        make_scoped_ptr(new NoopAuthenticatorFactory()));
+    return;
+  }
+#endif  // !defined(NDEBUG)
+
   std::string local_certificate = key_pair_->GenerateCertificate();
   if (local_certificate.empty()) {
     LOG(ERROR) << "Failed to generate host certificate.";
     ShutdownHost(kInitializationFailed);
     return;
   }
 
   scoped_ptr<protocol::AuthenticatorFactory> factory;
 
   if (third_party_auth_config_.is_null()) {
@@ skipping 722 matching lines @@
   if (!udp_port_range_.is_null()) {
     network_settings.port_range = udp_port_range_;
   } else if (!allow_nat_traversal_) {
     // For legacy reasons we have to restrict the port range to a set of default
     // values when nat traversal is disabled, even if the port range was not
     // set in policy.
     network_settings.port_range.min_port = NetworkSettings::kDefaultMinPort;
     network_settings.port_range.max_port = NetworkSettings::kDefaultMaxPort;
   }
 
-  scoped_ptr<protocol::SessionManager> session_manager =
-      CreateHostSessionManager(signal_strategy_.get(), network_settings,
-                               context_->url_request_context_getter());
+  scoped_ptr<protocol::SessionManager> session_manager;
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(kEnableWebrtc)) {
+#if !defined(NDEBUG)
+    network_settings.flags = protocol::NetworkSettings::NAT_TRAVERSAL_OUTGOING;
+
+    rtc::scoped_refptr<webrtc::PortAllocatorFactoryInterface>
+        port_allocator_factory = protocol::ChromiumPortAllocatorFactory::Create(
+            network_settings, context_->url_request_context_getter());
+
+    jingle_glue::JingleThreadWrapper::EnsureForCurrentMessageLoop();
+
+    // The network thread is also used as worker thread for webrtc.
+    //
+    // TODO(sergeyu): Figure out if we would benefit from using a separate
+    // thread as a worker thread.
+    scoped_ptr<protocol::TransportFactory> transport_factory(
+        new protocol::WebrtcTransportFactory(
+            jingle_glue::JingleThreadWrapper::current(), signal_strategy_.get(),
+            port_allocator_factory, protocol::TransportRole::SERVER));
+
+    session_manager.reset(
+        new protocol::JingleSessionManager(transport_factory.Pass()));
+#else  // !defined(NDEBUG)
+    LOG(ERROR) << "WebRTC is enabled only in debug builds.";
+    ShutdownHost(kUsageExitCode);
+    return;
+#endif  // defined(NDEBUG)
+  } else {
+    session_manager =
+        CreateHostSessionManager(signal_strategy_.get(), network_settings,
+                                 context_->url_request_context_getter());
+  }
 
   scoped_ptr<protocol::CandidateSessionConfig> protocol_config =
       protocol::CandidateSessionConfig::CreateDefault();
   if (!desktop_environment_factory_->SupportsAudioCapture())
     protocol_config->DisableAudioChannel();
   if (enable_vp9_)
     protocol_config->set_vp9_experiment_enabled(true);
+#if !defined(NDEBUG)
+  if (base::CommandLine::ForCurrentProcess()->HasSwitch(kEnableWebrtc)) {
+    protocol_config->set_webrtc_supported(true);
+  }
+#endif  // !defined(NDEBUG)
   session_manager->set_protocol_config(protocol_config.Pass());
 
   host_.reset(new ChromotingHost(
       signal_strategy_.get(), desktop_environment_factory_.get(),
       session_manager.Pass(), context_->audio_task_runner(),
       context_->input_task_runner(), context_->video_capture_task_runner(),
       context_->video_encode_task_runner(), context_->network_task_runner(),
       context_->ui_task_runner()));
 
   if (frame_recorder_buffer_size_ > 0) {
@@ skipping 197 matching lines @@
       base::TimeDelta::FromSeconds(kShutdownTimeoutSeconds));
   new HostProcess(context.Pass(), &exit_code, &shutdown_watchdog);
 
   // Run the main (also UI) message loop until the host no longer needs it.
   message_loop.Run();
 
   return exit_code;
 }
 
 }  // namespace remoting