Refactoring Me2MeNativeMessagingHost to implement NativeMessageHost interface

remoting/host/setup/me2me_native_messaging_host.cc

 // Copyright 2013 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.
 
 #include "remoting/host/setup/me2me_native_messaging_host.h"
 
 #include <cstdint>
 #include <memory>
 #include <string>
 #include <utility>
 
 #include "base/bind.h"
 #include "base/callback.h"
-#include "base/callback_helpers.h"
 #include "base/command_line.h"
+#include "base/json/json_reader.h"
+#include "base/json/json_writer.h"
 #include "base/logging.h"
 #include "base/macros.h"
+#include "base/single_thread_task_runner.h"
 #include "base/strings/stringize_macros.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "google_apis/gaia/gaia_oauth_client.h"
 #include "google_apis/google_api_keys.h"
 #include "net/base/network_interfaces.h"
+#include "remoting/base/auto_thread_task_runner.h"
 #include "remoting/base/rsa_key_pair.h"
+#include "remoting/host/chromoting_host_context.h"
+#include "remoting/host/native_messaging/log_message_handler.h"
 #include "remoting/host/native_messaging/pipe_messaging_channel.h"
 #include "remoting/host/pin_hash.h"
 #include "remoting/host/setup/oauth_client.h"
 #include "remoting/host/switches.h"
 #include "remoting/protocol/pairing_registry.h"
 
 #if defined(OS_WIN)
 #include "base/win/scoped_handle.h"
 #include "base/win/win_util.h"
 #include "remoting/host/win/launch_native_messaging_host_process.h"
@@ skipping 17 matching lines @@
 };
 
 // Helper to extract the "config" part of a message as a DictionaryValue.
 // Returns nullptr on failure, and logs an error message.
 std::unique_ptr<base::DictionaryValue> ConfigDictionaryFromMessage(
     std::unique_ptr<base::DictionaryValue> message) {
   std::unique_ptr<base::DictionaryValue> result;
   const base::DictionaryValue* config_dict;
   if (message->GetDictionary("config", &config_dict)) {
     result = config_dict->CreateDeepCopy();
-  } else {
-    LOG(ERROR) << "'config' dictionary not found";
   }
   return result;
 }
 
 }  // namespace
 
 namespace remoting {
 
 Me2MeNativeMessagingHost::Me2MeNativeMessagingHost(
     bool needs_elevation,
     intptr_t parent_window_handle,
-    std::unique_ptr<extensions::NativeMessagingChannel> channel,
+    std::unique_ptr<ChromotingHostContext> host_context,
     scoped_refptr<DaemonController> daemon_controller,
     scoped_refptr<protocol::PairingRegistry> pairing_registry,
     std::unique_ptr<OAuthClient> oauth_client)
     : needs_elevation_(needs_elevation),
 #if defined(OS_WIN)
       parent_window_handle_(parent_window_handle),
 #endif
-      channel_(std::move(channel)),
-      log_message_handler_(
-          base::Bind(&extensions::NativeMessagingChannel::SendMessage,
-                     base::Unretained(channel_.get()))),
+      host_context_(std::move(host_context)),
       daemon_controller_(daemon_controller),
       pairing_registry_(pairing_registry),
       oauth_client_(std::move(oauth_client)),
       weak_factory_(this) {
   weak_ptr_ = weak_factory_.GetWeakPtr();
 }
 
 Me2MeNativeMessagingHost::~Me2MeNativeMessagingHost() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 }
 
-void Me2MeNativeMessagingHost::Start(
-      const base::Closure& quit_closure) {
-  DCHECK(thread_checker_.CalledOnValidThread());
-  DCHECK(!quit_closure.is_null());
+void Me2MeNativeMessagingHost::OnMessage(const std::string& message) {
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
-  quit_closure_ = quit_closure;
-
-  channel_->Start(this);
-}
-
-void Me2MeNativeMessagingHost::OnMessage(std::unique_ptr<base::Value> message) {
-  DCHECK(thread_checker_.CalledOnValidThread());
-
-  if (!message->IsType(base::Value::TYPE_DICTIONARY)) {
-    LOG(ERROR) << "Received a message that's not a dictionary.";
-    channel_->SendMessage(nullptr);
+  std::unique_ptr<base::DictionaryValue> response(new base::DictionaryValue());
+  std::unique_ptr<base::Value> message_value = base::JSONReader::Read(message);
+  if (!message_value->IsType(base::Value::TYPE_DICTIONARY)) {
+    OnError("Received a message that's not a dictionary.");
     return;
   }
 
   std::unique_ptr<base::DictionaryValue> message_dict(
-      static_cast<base::DictionaryValue*>(message.release()));
-  std::unique_ptr<base::DictionaryValue> response(new base::DictionaryValue());
+      static_cast<base::DictionaryValue*>(message_value.release()));
 
   // If the client supplies an ID, it will expect it in the response. This
   // might be a string or a number, so cope with both.
   const base::Value* id;
   if (message_dict->Get("id", &id))
     response->Set("id", id->CreateDeepCopy());
 
   std::string type;
   if (!message_dict->GetString("type", &type)) {
-    LOG(ERROR) << "'type' not found";
-    channel_->SendMessage(nullptr);
+    OnError("'type' not found");
     return;
   }
 
   response->SetString("type", type + "Response");
 
   if (type == "hello") {
     ProcessHello(std::move(message_dict), std::move(response));
   } else if (type == "clearPairedClients") {
     ProcessClearPairedClients(std::move(message_dict), std::move(response));
   } else if (type == "deletePairedClient") {
@@ skipping 20 matching lines @@
     ProcessGetDaemonState(std::move(message_dict), std::move(response));
   } else if (type == "getHostClientId") {
     ProcessGetHostClientId(std::move(message_dict), std::move(response));
   } else if (type == "getCredentialsFromAuthCode") {
     ProcessGetCredentialsFromAuthCode(
         std::move(message_dict), std::move(response), true);
   } else if (type == "getRefreshTokenFromAuthCode") {
     ProcessGetCredentialsFromAuthCode(
         std::move(message_dict), std::move(response), false);
   } else {
-    LOG(ERROR) << "Unsupported request type: " << type;
-    OnError();
+    OnError("Unsupported request type: " + type);
   }
 }
 
-void Me2MeNativeMessagingHost::OnDisconnect() {
-  if (!quit_closure_.is_null())
-    base::ResetAndReturn(&quit_closure_).Run();
+void Me2MeNativeMessagingHost::Start(Client* client) {
+  DCHECK(task_runner()->BelongsToCurrentThread());
+  client_ = client;
+  log_message_handler_.reset(new LogMessageHandler(
+      base::Bind(&Me2MeNativeMessagingHost::SendMessageToClient, weak_ptr_)));
+}
+
+scoped_refptr<base::SingleThreadTaskRunner>
+Me2MeNativeMessagingHost::task_runner() const {
+  return host_context_->ui_task_runner();
 }
 
 void Me2MeNativeMessagingHost::ProcessHello(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->SetString("version", STRINGIZE(VERSION));
   std::unique_ptr<base::ListValue> supported_features_list(
       new base::ListValue());
   supported_features_list->AppendStrings(std::vector<std::string>(
       kSupportedFeatures, kSupportedFeatures + arraysize(kSupportedFeatures)));
   response->Set("supportedFeatures", supported_features_list.release());
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessClearPairedClients(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (needs_elevation_) {
     if (!DelegateToElevatedHost(std::move(message)))
       SendBooleanResult(std::move(response), false);
     return;
   }
 
   if (pairing_registry_.get()) {
     pairing_registry_->ClearAllPairings(
         base::Bind(&Me2MeNativeMessagingHost::SendBooleanResult, weak_ptr_,
                    base::Passed(&response)));
   } else {
     SendBooleanResult(std::move(response), false);
   }
 }
 
 void Me2MeNativeMessagingHost::ProcessDeletePairedClient(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (needs_elevation_) {
     if (!DelegateToElevatedHost(std::move(message)))
       SendBooleanResult(std::move(response), false);
     return;
   }
 
   std::string client_id;
   if (!message->GetString(protocol::PairingRegistry::kClientIdKey,
                           &client_id)) {
-    LOG(ERROR) << "'" << protocol::PairingRegistry::kClientIdKey
-               << "' string not found.";
-    OnError();
+    OnError("'" + std::string(protocol::PairingRegistry::kClientIdKey) +
+            "' string not found.");
     return;
   }
 
   if (pairing_registry_.get()) {
     pairing_registry_->DeletePairing(
         client_id, base::Bind(&Me2MeNativeMessagingHost::SendBooleanResult,
                               weak_ptr_, base::Passed(&response)));
   } else {
     SendBooleanResult(std::move(response), false);
   }
 }
 
 void Me2MeNativeMessagingHost::ProcessGetHostName(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->SetString("hostname", net::GetHostName());
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetPinHash(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   std::string host_id;
   if (!message->GetString("hostId", &host_id)) {
     LOG(ERROR) << "'hostId' not found: " << message.get();
-    OnError();
+    OnError(std::string());

[Sergey Ulanov, 2016/08/01 18:45:18]

OnError("'hostId' not found")?

[joedow, 2016/08/01 19:21:34]

Done.

     return;
   }
   std::string pin;
   if (!message->GetString("pin", &pin)) {
     LOG(ERROR) << "'pin' not found: " << message.get();
-    OnError();
+    OnError(std::string());

[Sergey Ulanov, 2016/08/01 18:45:19]

same here

[joedow, 2016/08/01 19:21:34]

Done.

     return;
   }
   response->SetString("hash", MakeHostPinHash(host_id, pin));
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessGenerateKeyPair(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::Generate();
   response->SetString("privateKey", key_pair->ToString());
   response->SetString("publicKey", key_pair->GetPublicKey());
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessUpdateDaemonConfig(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (needs_elevation_) {
     if (!DelegateToElevatedHost(std::move(message)))
       SendAsyncResult(std::move(response), DaemonController::RESULT_FAILED);
     return;
   }
 
   std::unique_ptr<base::DictionaryValue> config_dict =
       ConfigDictionaryFromMessage(std::move(message));
   if (!config_dict) {
-    OnError();
+    OnError("'config' dictionary not found");
     return;
   }
 
   daemon_controller_->UpdateConfig(
       std::move(config_dict),
       base::Bind(&Me2MeNativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetDaemonConfig(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   daemon_controller_->GetConfig(
       base::Bind(&Me2MeNativeMessagingHost::SendConfigResponse, weak_ptr_,
                  base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetPairedClients(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (pairing_registry_.get()) {
     pairing_registry_->GetAllPairings(
         base::Bind(&Me2MeNativeMessagingHost::SendPairedClientsResponse,
                    weak_ptr_, base::Passed(&response)));
   } else {
     std::unique_ptr<base::ListValue> no_paired_clients(new base::ListValue);
     SendPairedClientsResponse(std::move(response),
                               std::move(no_paired_clients));
   }
 }
 
 void Me2MeNativeMessagingHost::ProcessGetUsageStatsConsent(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   daemon_controller_->GetUsageStatsConsent(
       base::Bind(&Me2MeNativeMessagingHost::SendUsageStatsConsentResponse,
                  weak_ptr_, base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::ProcessStartDaemon(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (needs_elevation_) {
     if (!DelegateToElevatedHost(std::move(message)))
       SendAsyncResult(std::move(response), DaemonController::RESULT_FAILED);
     return;
   }
 
   bool consent;
   if (!message->GetBoolean("consent", &consent)) {
-    LOG(ERROR) << "'consent' not found.";
-    OnError();
+    OnError("'consent' not found.");
     return;
   }
 
   std::unique_ptr<base::DictionaryValue> config_dict =
       ConfigDictionaryFromMessage(std::move(message));
   if (!config_dict) {
-    OnError();
+    OnError("'config' dictionary not found");
     return;
   }
 
   daemon_controller_->SetConfigAndStart(
       std::move(config_dict), consent,
       base::Bind(&Me2MeNativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::ProcessStopDaemon(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (needs_elevation_) {
     if (!DelegateToElevatedHost(std::move(message)))
       SendAsyncResult(std::move(response), DaemonController::RESULT_FAILED);
     return;
   }
 
   daemon_controller_->Stop(
       base::Bind(&Me2MeNativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetDaemonState(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   DaemonController::State state = daemon_controller_->GetState();
   switch (state) {
     case DaemonController::STATE_NOT_IMPLEMENTED:
       response->SetString("state", "NOT_IMPLEMENTED");
       break;
     case DaemonController::STATE_STOPPED:
       response->SetString("state", "STOPPED");
       break;
     case DaemonController::STATE_STARTING:
       response->SetString("state", "STARTING");
       break;
     case DaemonController::STATE_STARTED:
       response->SetString("state", "STARTED");
       break;
     case DaemonController::STATE_STOPPING:
       response->SetString("state", "STOPPING");
       break;
     case DaemonController::STATE_UNKNOWN:
       response->SetString("state", "UNKNOWN");
       break;
   }
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetHostClientId(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->SetString("clientId", google_apis::GetOAuth2ClientID(
       google_apis::CLIENT_REMOTING_HOST));
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::ProcessGetCredentialsFromAuthCode(
     std::unique_ptr<base::DictionaryValue> message,
     std::unique_ptr<base::DictionaryValue> response,
     bool need_user_email) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   std::string auth_code;
   if (!message->GetString("authorizationCode", &auth_code)) {
-    LOG(ERROR) << "'authorizationCode' string not found.";
-    OnError();
+    OnError("'authorizationCode' string not found.");
     return;
   }
 
   gaia::OAuthClientInfo oauth_client_info = {
     google_apis::GetOAuth2ClientID(google_apis::CLIENT_REMOTING_HOST),
     google_apis::GetOAuth2ClientSecret(google_apis::CLIENT_REMOTING_HOST),
     kServiceAccountRedirectUri
   };
 
   oauth_client_->GetCredentialsFromAuthCode(
       oauth_client_info, auth_code, need_user_email, base::Bind(
           &Me2MeNativeMessagingHost::SendCredentialsResponse, weak_ptr_,
           base::Passed(&response)));
 }
 
 void Me2MeNativeMessagingHost::SendConfigResponse(
     std::unique_ptr<base::DictionaryValue> response,
     std::unique_ptr<base::DictionaryValue> config) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (config) {
     response->Set("config", config.release());
   } else {
     response->Set("config", base::Value::CreateNullValue());
   }
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::SendPairedClientsResponse(
     std::unique_ptr<base::DictionaryValue> response,
     std::unique_ptr<base::ListValue> pairings) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->Set("pairedClients", pairings.release());
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::SendUsageStatsConsentResponse(
     std::unique_ptr<base::DictionaryValue> response,
     const DaemonController::UsageStatsConsent& consent) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->SetBoolean("supported", consent.supported);
   response->SetBoolean("allowed", consent.allowed);
   response->SetBoolean("setByPolicy", consent.set_by_policy);
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::SendAsyncResult(
     std::unique_ptr<base::DictionaryValue> response,
     DaemonController::AsyncResult result) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   switch (result) {
     case DaemonController::RESULT_OK:
       response->SetString("result", "OK");
       break;
     case DaemonController::RESULT_FAILED:
       response->SetString("result", "FAILED");
       break;
     case DaemonController::RESULT_CANCELLED:
       response->SetString("result", "CANCELLED");
       break;
   }
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::SendBooleanResult(
     std::unique_ptr<base::DictionaryValue> response,
     bool result) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   response->SetBoolean("result", result);
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
 void Me2MeNativeMessagingHost::SendCredentialsResponse(
     std::unique_ptr<base::DictionaryValue> response,
     const std::string& user_email,
     const std::string& refresh_token) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   if (!user_email.empty()) {
     response->SetString("userEmail", user_email);
   }
   response->SetString("refreshToken", refresh_token);
-  channel_->SendMessage(std::move(response));
+  SendMessageToClient(std::move(response));
 }
 
-void Me2MeNativeMessagingHost::OnError() {
-  // Trigger a host shutdown by sending a nullptr message.
-  channel_->SendMessage(nullptr);
+void Me2MeNativeMessagingHost::SendMessageToClient(
+    std::unique_ptr<base::Value> message) const {
+  DCHECK(task_runner()->BelongsToCurrentThread());
+  std::string message_json;
+  base::JSONWriter::Write(*message, &message_json);
+  client_->PostMessageFromNativeHost(message_json);
 }
 
-void Me2MeNativeMessagingHost::Stop() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+void Me2MeNativeMessagingHost::OnError(const std::string& error_message) {
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
-  if (!quit_closure_.is_null())
-    base::ResetAndReturn(&quit_closure_).Run();
+  if (!error_message.empty()) {
+    LOG(ERROR) << error_message;
+  }
+
+  // Trigger a host shutdown by sending an empty message.
+  client_->CloseChannel(std::string());
 }
 
 #if defined(OS_WIN)
 Me2MeNativeMessagingHost::ElevatedChannelEventHandler::
-    ElevatedChannelEventHandler(Me2MeNativeMessagingHost* host)
-    : parent_(host) {
-}
+    ElevatedChannelEventHandler(extensions::NativeMessageHost::Client* client)
+    : client_(client) {}
 
 void Me2MeNativeMessagingHost::ElevatedChannelEventHandler::OnMessage(
     std::unique_ptr<base::Value> message) {
-  DCHECK(parent_->thread_checker_.CalledOnValidThread());
+  DCHECK(thread_checker_.CalledOnValidThread());
 
   // Simply pass along the response from the elevated host to the client.
-  parent_->channel_->SendMessage(std::move(message));
+  std::string message_json;
+  base::JSONWriter::Write(*message, &message_json);
+  client_->PostMessageFromNativeHost(message_json);
 }
 
 void Me2MeNativeMessagingHost::ElevatedChannelEventHandler::OnDisconnect() {
-  parent_->OnDisconnect();
+  DCHECK(thread_checker_.CalledOnValidThread());
+  client_->CloseChannel(std::string());
 }
 
 bool Me2MeNativeMessagingHost::DelegateToElevatedHost(
     std::unique_ptr<base::DictionaryValue> message) {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   EnsureElevatedHostCreated();
 
   // elevated_channel_ will be null if user rejects the UAC request.
   if (elevated_channel_)
     elevated_channel_->SendMessage(std::move(message));
 
   return elevated_channel_ != nullptr;
 }
 
 void Me2MeNativeMessagingHost::EnsureElevatedHostCreated() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
   DCHECK(needs_elevation_);
 
   if (elevated_channel_)
     return;
 
   base::win::ScopedHandle read_handle;
   base::win::ScopedHandle write_handle;
   // Get the name of the binary to launch.
   base::FilePath binary = base::CommandLine::ForCurrentProcess()->GetProgram();
   ProcessLaunchResult result = LaunchNativeMessagingHostProcess(
       binary, parent_window_handle_,
       /*elevate_process=*/true, &read_handle, &write_handle);
   if (result != PROCESS_LAUNCH_RESULT_SUCCESS) {
     if (result != PROCESS_LAUNCH_RESULT_CANCELLED) {
-      OnError();
+      OnError(std::string());
     }
     return;
   }
 
   // Set up the native messaging channel to talk to the elevated host.
   // Note that input for the elevated channel is output for the elevated host.
   elevated_channel_.reset(new PipeMessagingChannel(
       base::File(read_handle.Take()), base::File(write_handle.Take())));
 
   elevated_channel_event_handler_.reset(
-      new Me2MeNativeMessagingHost::ElevatedChannelEventHandler(this));
+      new Me2MeNativeMessagingHost::ElevatedChannelEventHandler(client_));
   elevated_channel_->Start(elevated_channel_event_handler_.get());
 
   elevated_host_timer_.Start(
       FROM_HERE, base::TimeDelta::FromSeconds(kElevatedHostTimeoutSeconds),
       this, &Me2MeNativeMessagingHost::DisconnectElevatedHost);
 }
 
 void Me2MeNativeMessagingHost::DisconnectElevatedHost() {
-  DCHECK(thread_checker_.CalledOnValidThread());
+  DCHECK(task_runner()->BelongsToCurrentThread());
 
   // This will send an EOF to the elevated host, triggering its shutdown.
   elevated_channel_.reset();
 }
 
 #else  // defined(OS_WIN)
 
 bool Me2MeNativeMessagingHost::DelegateToElevatedHost(
     std::unique_ptr<base::DictionaryValue> message) {
   NOTREACHED();
   return false;
 }
 
 #endif  // !defined(OS_WIN)
 
 }  // namespace remoting