Enable pairing registry in me2me native messaging host on Windows

remoting/host/setup/me2me_native_messaging_host_main.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 "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "net/url_request/url_fetcher.h"
 #include "remoting/host/host_exit_codes.h"
 #include "remoting/host/logging.h"
 #include "remoting/host/pairing_registry_delegate.h"
 #include "remoting/host/setup/me2me_native_messaging_host.h"
 
 #if defined(OS_WIN)
+#include "base/win/registry.h"
 #include "base/win/windows_version.h"
+#include "remoting/host/pairing_registry_delegate_win.h"
 #endif  // defined(OS_WIN)
 
+using remoting::protocol::PairingRegistry;
+
 namespace {
 
 const char kParentWindowSwitchName[] = "parent-window";
 
 }  // namespace
 
 namespace remoting {
 
 #if defined(OS_WIN)
 bool IsProcessElevated() {
@@ skipping 103 matching lines @@
 #endif
 
   // OAuth client (for credential requests).
   scoped_refptr<net::URLRequestContextGetter> url_request_context_getter(
       new URLRequestContextGetter(io_thread.message_loop_proxy()));
   scoped_ptr<OAuthClient> oauth_client(
       new OAuthClient(url_request_context_getter));
 
   net::URLFetcher::SetIgnoreCertificateRequests(true);
 
-  // Create the pairing registry and native messaging host.
-  scoped_refptr<protocol::PairingRegistry> pairing_registry =
+  // Create the pairing registry.
+  scoped_refptr<PairingRegistry> pairing_registry;
+
+#if defined(OS_WIN)
+  base::win::RegKey root;
+  LONG result = root.Open(HKEY_LOCAL_MACHINE, kPairingRegistryKeyName,
+                          KEY_READ);
+  if (result != ERROR_SUCCESS) {
+    SetLastError(result);
+    PLOG(ERROR) << "Failed to open HKLM\\" << kPairingRegistryKeyName;
+    return kInitializationFailed;
+  }
+
+  base::win::RegKey unprivileged;
+  result = unprivileged.Open(root.Handle(), kPairingRegistrySecretsKeyName,
+                             needs_elevation ? KEY_READ : KEY_READ | KEY_WRITE);
+  if (result != ERROR_SUCCESS) {
+    SetLastError(result);
+    PLOG(ERROR) << "Failed to open HKLM\\" << kPairingRegistrySecretsKeyName
+                << "\\" << kPairingRegistrySecretsKeyName;
+    return kInitializationFailed;
+  }
+
+  // Only try to open the privileged key if the current process is elevated.
+  base::win::RegKey privileged;
+  if (!needs_elevation) {
+    result = privileged.Open(root.Handle(), kPairingRegistryClientsKeyName,
+                             KEY_READ | KEY_WRITE);
+    if (result != ERROR_SUCCESS) {
+      SetLastError(result);
+      PLOG(ERROR) << "Failed to open HKLM\\" << kPairingRegistryKeyName << "\\"
+                  << kPairingRegistryClientsKeyName;
+      return kInitializationFailed;
+    }
+  }
+
+  // Initialize the pairing registry delegate and set the root keys.
+  scoped_ptr<PairingRegistryDelegateWin> delegate(
+      new PairingRegistryDelegateWin());
+  if (!delegate->SetRootKeys(privileged.Take(), unprivileged.Take()))
+    return kInitializationFailed;
+
+  pairing_registry = new PairingRegistry(
+      io_thread.message_loop_proxy(),
+      delegate.PassAs<PairingRegistry::Delegate>());
+#else  // defined(OS_WIN)
+  pairing_registry =
       CreatePairingRegistry(io_thread.message_loop_proxy());
+#endif  // !defined(OS_WIN)
 
   // Set up the native messaging channel.
   scoped_ptr<NativeMessagingChannel> channel(
       new NativeMessagingChannel(read_file, write_file));
 
+  // Create the native messaging host.
   scoped_ptr<Me2MeNativeMessagingHost> host(
       new Me2MeNativeMessagingHost(
           needs_elevation,
           channel.Pass(),
           daemon_controller,
           pairing_registry,
           oauth_client.Pass()));
   host->Start(run_loop.QuitClosure());
 
   // Run the loop until channel is alive.
   run_loop.Run();
   return kSuccessExitCode;
 }
 
 }  // namespace remoting
 
 int main(int argc, char** argv) {
   // This object instance is required by Chrome code (such as MessageLoop).
   base::AtExitManager exit_manager;
 
   CommandLine::Init(argc, argv);
   remoting::InitHostLogging();
 
   return remoting::Me2MeNativeMessagingHostMain();
 }