Changes to write various registry keys as required by Omaha for Windows Cloud Print Virtual Driver.

cloud_print/virtual_driver/win/install/setup.cc

 // Copyright (c) 2011 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 <windows.h>
 #include <winspool.h>
 
 #include "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/file_version_info_win.h"
 #include "base/logging.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
 #include "base/string16.h"
+#include "base/win/registry.h"
 #include "base/win/scoped_handle.h"
 #include "cloud_print/virtual_driver/win/virtual_driver_consts.h"
 #include "cloud_print/virtual_driver/win/virtual_driver_helpers.h"
 #include "grit/virtual_driver_setup_resources.h"
 
 namespace {
+const wchar_t* kVersionKey = L"pv";

[sanjeevr, 2011/08/03 11:19:29]

It is the norm in Chrome to use const wchar_t[] instead of const wchar_t*.

[abeera, 2011/08/03 19:50:30]

Done.

+const wchar_t* kNameKey = L"name";
+const wchar_t* kLangKey = L"lang";
+
+void SetRegistryKeys() {
+  base::win::RegKey key;
+  if (key.Create(HKEY_LOCAL_MACHINE, cloud_print::kKeyLocation,
+                 KEY_ALL_ACCESS) != ERROR_SUCCESS) {

[sanjeevr, 2011/08/03 11:19:29]

You don't need KEY_ALL_ACCESS here. You just need KEY_SET_VALUE.

[abeera, 2011/08/03 19:50:30]

Done.

+    LOG(ERROR) << "Unable to open key";
+  }
+
+  // Get the version from the resource file.
+  std::wstring version_string;
+  scoped_ptr<FileVersionInfo> version_info(
+      FileVersionInfo::CreateFileVersionInfoForCurrentModule());
+
+  if (version_info.get()) {
+    FileVersionInfoWin* version_info_win =
+        static_cast<FileVersionInfoWin*>(version_info.get());
+    version_string = version_info_win->product_version();
+  } else {
+    LOG(ERROR) << "Unable to get version string";
+    // Use a random version string so that Omaha has something to go by.
+    version_string = L"0.0.0.99";
+  }
+
+  if (key.WriteValue(kVersionKey, version_string.c_str()) != ERROR_SUCCESS ||
+      key.WriteValue(kNameKey, L"GCP Virtual Driver") != ERROR_SUCCESS ||

[sanjeevr, 2011/08/03 11:19:29]

Why don't you create constant strings for these as well?

[abeera, 2011/08/03 19:50:30]

Done.

+      key.WriteValue(kLangKey, L"en") != ERROR_SUCCESS) {
+    LOG(ERROR) << "Unable to set registry keys";
+  }
+}
+
+void DeleteRegistryKeys() {
+  base::win::RegKey key;
+  if (key.Create(HKEY_LOCAL_MACHINE, cloud_print::kKeyLocation,
+                 KEY_ALL_ACCESS) != ERROR_SUCCESS) {

[sanjeevr, 2011/08/03 11:19:29]

Again you only need DELETE access. And you don't need to call Create, just Open.

[abeera, 2011/08/03 19:50:30]

Done.

+    LOG(ERROR) << "Unable to open key to delete";
+  }
+  if (key.DeleteKey(L"") != ERROR_SUCCESS) {
+    LOG(ERROR) << "Unable to delete key";
+  }
+}
 
 HRESULT GetGpdPath(FilePath* path) {
   if (!PathService::Get(base::DIR_EXE, path)) {
     LOG(ERROR) << "Unable to get install path.";
     return ERROR_PATH_NOT_FOUND;
   }
   *path = path->Append(L"gcp.gpd");
   return S_OK;
 }
 
@@ skipping 239 matching lines @@
   result = InstallGpd();
   if (!SUCCEEDED(result)) {
     LOG(ERROR) << "Unable to install gpd.";
     return result;
   }
   result = InstallPrinter();
   if (!SUCCEEDED(result)) {
     LOG(ERROR) << "Unable to install printer.";
     return result;
   }
+  SetRegistryKeys();
   return S_OK;
 }
 
 HRESULT UninstallVirtualDriver(void) {
   HRESULT result = S_OK;
   result = UninstallPrinter();
   if (!SUCCEEDED(result)) {
     LOG(ERROR) << "Unable to uninstall gpd.";
     return result;
   }
   result = UninstallGpd();
   if (!SUCCEEDED(result)) {
     LOG(ERROR) << "Unable to remove gpd.";
     return result;
   }
   result = RegisterPortMonitor(false);
   if (!SUCCEEDED(result)) {
     LOG(ERROR) << "Unable to remove port monitor.";
     return result;
   }
+  DeleteRegistryKeys();
   return S_OK;
 }
 
 }  // namespace
 
 int WINAPI WinMain(__in  HINSTANCE hInstance,
             __in  HINSTANCE hPrevInstance,
             __in  LPSTR lpCmdLine,
             __in  int nCmdShow) {
   base::AtExitManager at_exit_manager;
   CommandLine::Init(0, NULL);
   HRESULT retval = S_OK;
   if (CommandLine::ForCurrentProcess()->HasSwitch("uninstall")) {
     retval = UninstallVirtualDriver();
   } else {
     retval = InstallVirtualDriver();
   }
-  if (!CommandLine::ForCurrentProcess()->HasSwitch("silent")) {
+  // Installer is silent by default as required by Omaha.
+  if (CommandLine::ForCurrentProcess()->HasSwitch("verbose")) {
     cloud_print::DisplayWindowsMessage(NULL, retval,
         cloud_print::LoadLocalString(IDS_DRIVER_NAME));
   }
   return retval;
 }