Me2me Native Messaging host on Windows: restructure NativeMessagingHost and NativeMessagingChannel.…

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 <string>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/callback.h"
+#include "base/callback_helpers.h"
 #include "base/logging.h"
 #include "base/strings/stringize_macros.h"
 #include "base/threading/thread.h"
 #include "base/values.h"
 #include "google_apis/gaia/gaia_oauth_client.h"
 #include "google_apis/google_api_keys.h"
 #include "net/base/net_util.h"
 #include "remoting/base/rsa_key_pair.h"
 #include "remoting/host/pin_hash.h"
 #include "remoting/host/setup/oauth_client.h"
@@ skipping 34 matching lines @@
     scoped_refptr<protocol::PairingRegistry> pairing_registry,
     scoped_ptr<OAuthClient> oauth_client)
     : daemon_controller_(daemon_controller),
       pairing_registry_(pairing_registry),
       oauth_client_(oauth_client.Pass()),
       weak_factory_(this) {
   weak_ptr_ = weak_factory_.GetWeakPtr();
 }
 
 NativeMessagingHost::~NativeMessagingHost() {
+  DCHECK(thread_checker_.CalledOnValidThread());
 }
 
-void NativeMessagingHost::SetSendMessageCallback(
-    const SendMessageCallback& send_message) {
-  send_message_ = send_message;
+void NativeMessagingHost::Start(
+      base::PlatformFile input,
+      base::PlatformFile output,
+      const base::Closure& quit_closure) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  // Set up the native messaging channel.
+  channel_.reset(
+      new NativeMessagingChannel(
+          base::Bind(&NativeMessagingHost::ProcessMessage, weak_ptr_),
+          input,
+          output));
+
+  quit_closure_ = quit_closure;
+  channel_->Start(base::Bind(&NativeMessagingHost::ShutDown, weak_ptr_));
+}
+
+void NativeMessagingHost::ShutDown() {

[Sergey Ulanov, 2013/12/12 19:10:28]

Don't need this method. Just pass quit_closure_ to the channel when starting it.

[weitao, 2013/12/12 23:33:48]

Done.

+  DCHECK(thread_checker_.CalledOnValidThread());
+
+  if (!quit_closure_.is_null())
+    base::ResetAndReturn(&quit_closure_).Run();
 }
 
 void NativeMessagingHost::ProcessMessage(
     scoped_ptr<base::DictionaryValue> message) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   scoped_ptr<base::DictionaryValue> response(new base::DictionaryValue());
 
   // 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->Get("id", &id))
     response->Set("id", id->DeepCopy());
 
   std::string type;
   if (!message->GetString("type", &type)) {
     LOG(ERROR) << "'type' not found";
-    send_message_.Run(scoped_ptr<base::DictionaryValue>());
+    channel_->SendMessage(scoped_ptr<base::DictionaryValue>());
     return;
   }
 
   response->SetString("type", type + "Response");
 
   bool success = false;
   if (type == "hello") {
     success = ProcessHello(*message, response.Pass());
   } else if (type == "clearPairedClients") {
     success = ProcessClearPairedClients(*message, response.Pass());
@@ skipping 21 matching lines @@
     success = ProcessGetDaemonState(*message, response.Pass());
   } else if (type == "getHostClientId") {
     success = ProcessGetHostClientId(*message, response.Pass());
   } else if (type == "getCredentialsFromAuthCode") {
     success = ProcessGetCredentialsFromAuthCode(*message, response.Pass());
   } else {
     LOG(ERROR) << "Unsupported request type: " << type;
   }
 
   if (!success)
-    send_message_.Run(scoped_ptr<base::DictionaryValue>());
+    channel_->SendMessage(scoped_ptr<base::DictionaryValue>());
 }
 
 bool NativeMessagingHost::ProcessHello(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetString("version", STRINGIZE(VERSION));
   scoped_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());
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessClearPairedClients(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   if (pairing_registry_) {
     pairing_registry_->ClearAllPairings(
         base::Bind(&NativeMessagingHost::SendBooleanResult, weak_ptr_,
                    base::Passed(&response)));
   } else {
     SendBooleanResult(response.Pass(), false);
   }
   return true;
 }
 
 bool NativeMessagingHost::ProcessDeletePairedClient(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   std::string client_id;
   if (!message.GetString(protocol::PairingRegistry::kClientIdKey, &client_id)) {
     LOG(ERROR) << "'" << protocol::PairingRegistry::kClientIdKey
                << "' string not found.";
     return false;
   }
 
   if (pairing_registry_) {
     pairing_registry_->DeletePairing(
         client_id, base::Bind(&NativeMessagingHost::SendBooleanResult,
                               weak_ptr_, base::Passed(&response)));
   } else {
     SendBooleanResult(response.Pass(), false);
   }
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetHostName(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetString("hostname", net::GetHostName());
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetPinHash(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   std::string host_id;
   if (!message.GetString("hostId", &host_id)) {
     LOG(ERROR) << "'hostId' not found: " << message;
     return false;
   }
   std::string pin;
   if (!message.GetString("pin", &pin)) {
     LOG(ERROR) << "'pin' not found: " << message;
     return false;
   }
   response->SetString("hash", MakeHostPinHash(host_id, pin));
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessGenerateKeyPair(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::Generate();
   response->SetString("privateKey", key_pair->ToString());
   response->SetString("publicKey", key_pair->GetPublicKey());
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessUpdateDaemonConfig(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   scoped_ptr<base::DictionaryValue> config_dict =
       ConfigDictionaryFromMessage(message);
   if (!config_dict)
     return false;
 
   daemon_controller_->UpdateConfig(
       config_dict.Pass(),
       base::Bind(&NativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetDaemonConfig(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   daemon_controller_->GetConfig(
       base::Bind(&NativeMessagingHost::SendConfigResponse, weak_ptr_,
                  base::Passed(&response)));
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetPairedClients(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   if (pairing_registry_) {
     pairing_registry_->GetAllPairings(
         base::Bind(&NativeMessagingHost::SendPairedClientsResponse, weak_ptr_,
                    base::Passed(&response)));
   } else {
     scoped_ptr<base::ListValue> no_paired_clients(new base::ListValue);
     SendPairedClientsResponse(response.Pass(), no_paired_clients.Pass());
   }
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetUsageStatsConsent(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   daemon_controller_->GetUsageStatsConsent(
       base::Bind(&NativeMessagingHost::SendUsageStatsConsentResponse,
                  weak_ptr_, base::Passed(&response)));
   return true;
 }
 
 bool NativeMessagingHost::ProcessStartDaemon(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   bool consent;
   if (!message.GetBoolean("consent", &consent)) {
     LOG(ERROR) << "'consent' not found.";
     return false;
   }
 
   scoped_ptr<base::DictionaryValue> config_dict =
       ConfigDictionaryFromMessage(message);
   if (!config_dict)
     return false;
 
   daemon_controller_->SetConfigAndStart(
       config_dict.Pass(), consent,
       base::Bind(&NativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
   return true;
 }
 
 bool NativeMessagingHost::ProcessStopDaemon(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   daemon_controller_->Stop(
       base::Bind(&NativeMessagingHost::SendAsyncResult, weak_ptr_,
                  base::Passed(&response)));
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetDaemonState(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   DaemonController::State state = daemon_controller_->GetState();
   switch (state) {
     case DaemonController::STATE_NOT_IMPLEMENTED:
       response->SetString("state", "NOT_IMPLEMENTED");
       break;
     case DaemonController::STATE_NOT_INSTALLED:
       response->SetString("state", "NOT_INSTALLED");
       break;
     case DaemonController::STATE_INSTALLING:
       response->SetString("state", "INSTALLING");
       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;
   }
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetHostClientId(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetString("clientId", google_apis::GetOAuth2ClientID(
       google_apis::CLIENT_REMOTING_HOST));
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
   return true;
 }
 
 bool NativeMessagingHost::ProcessGetCredentialsFromAuthCode(
     const base::DictionaryValue& message,
     scoped_ptr<base::DictionaryValue> response) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   std::string auth_code;
   if (!message.GetString("authorizationCode", &auth_code)) {
     LOG(ERROR) << "'authorizationCode' string not found.";
     return false;
   }
 
   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, base::Bind(
           &NativeMessagingHost::SendCredentialsResponse, weak_ptr_,
           base::Passed(&response)));
 
   return true;
 }
 
 void NativeMessagingHost::SendConfigResponse(
     scoped_ptr<base::DictionaryValue> response,
     scoped_ptr<base::DictionaryValue> config) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   if (config) {
     response->Set("config", config.release());
   } else {
     response->Set("config", Value::CreateNullValue());
   }
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 void NativeMessagingHost::SendPairedClientsResponse(
     scoped_ptr<base::DictionaryValue> response,
     scoped_ptr<base::ListValue> pairings) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->Set("pairedClients", pairings.release());
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 void NativeMessagingHost::SendUsageStatsConsentResponse(
     scoped_ptr<base::DictionaryValue> response,
     const DaemonController::UsageStatsConsent& consent) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetBoolean("supported", consent.supported);
   response->SetBoolean("allowed", consent.allowed);
   response->SetBoolean("setByPolicy", consent.set_by_policy);
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 void NativeMessagingHost::SendAsyncResult(
     scoped_ptr<base::DictionaryValue> response,
     DaemonController::AsyncResult result) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   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;
     case DaemonController::RESULT_FAILED_DIRECTORY:
       response->SetString("result", "FAILED_DIRECTORY");
       break;
   }
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 void NativeMessagingHost::SendBooleanResult(
     scoped_ptr<base::DictionaryValue> response, bool result) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetBoolean("result", result);
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 void NativeMessagingHost::SendCredentialsResponse(
     scoped_ptr<base::DictionaryValue> response,
     const std::string& user_email,
     const std::string& refresh_token) {
+  DCHECK(thread_checker_.CalledOnValidThread());
+
   response->SetString("userEmail", user_email);
   response->SetString("refreshToken", refresh_token);
-  send_message_.Run(response.Pass());
+  channel_->SendMessage(response.Pass());
 }
 
 }  // namespace remoting