Return Windows error code when create-process fails.

chrome/installer/mini_installer/mini_installer.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.
 
 // mini_installer.exe is the first exe that is run when chrome is being
 // installed or upgraded. It is designed to be extremely small (~5KB with no
 // extra resources linked) and it has two main jobs:
 //   1) unpack the resources (possibly decompressing some)
 //   2) run the real installer (setup.exe) with appropriate flags.
 //
@@ skipping 14 matching lines @@
 #include <shellapi.h>
 #include <stdlib.h>
 
 #include "chrome/installer/mini_installer/appid.h"
 #include "chrome/installer/mini_installer/configuration.h"
 #include "chrome/installer/mini_installer/decompress.h"
 #include "chrome/installer/mini_installer/exit_code.h"
 #include "chrome/installer/mini_installer/mini_installer_constants.h"
 #include "chrome/installer/mini_installer/mini_string.h"
 #include "chrome/installer/mini_installer/pe_resource.h"
+#include "chrome/installer/mini_installer/regkey.h"
 
 namespace mini_installer {
 
-typedef DWORD ProcessExitCode;
 typedef StackString<MAX_PATH> PathString;
 typedef StackString<MAX_PATH * 4> CommandString;
 
+struct ProcessExitCode {

[robertshield, 2015/08/04 03:11:36]

This isn't a just an exit code anymore, suggest calling it ProcessExitResults or
similar.

[bcwhite, 2015/08/04 19:34:30]

Done.

+  DWORD exit_code;
+  DWORD windows_code;

[robertshield, 2015/08/04 03:11:36]

It isn't clear to a Windows dev what "windows_code" means. If this is always the
Last Error Code then last_error_code would be clearer. If it is a generic holder
for error related codes, then "error_code" would be clearer.

[bcwhite, 2015/08/04 19:34:31]

It's usually the result of ::GetLastError() but sometimes the return code from
Windows registry functions.  It's always Windows specific, though. 
"windows_error"?

[robertshield, 2015/08/05 18:39:48]

Ok.

+
+  explicit ProcessExitCode(DWORD exit) : exit_code(exit), windows_code(0) {}
+  ProcessExitCode(DWORD exit, DWORD win) : exit_code(exit), windows_code(win) {
+  }
+
+  bool IsSuccess() {
+    return exit_code == ERROR_SUCCESS;

[robertshield, 2015/08/04 03:11:36]

Nit: ERROR_SUCCESS is a windows error code, not a process exit code. Do you mean
SUCCESS_EXIT_CODE?

[bcwhite, 2015/08/04 19:34:30]

Done.

+  }
+};
+
 // This structure passes data back and forth for the processing
 // of resource callbacks.
 struct Context {
   // Input to the call back method. Specifies the dir to save resources.
   const wchar_t* base_path;
   // First output from call back method. Full path of Chrome archive.
   PathString* chrome_resource_path;
   // Second output from call back method. Full path of Setup archive/exe.
   PathString* setup_resource_path;
 };
 
-// A helper class used to manipulate the Windows registry.  Typically, members
-// return Windows last-error codes a la the Win32 registry API.
-class RegKey {
- public:
-  RegKey() : key_(NULL) { }
-  ~RegKey() { Close(); }
 
-  // Opens the key named |sub_key| with given |access| rights.  Returns
-  // ERROR_SUCCESS or some other error.
-  LONG Open(HKEY key, const wchar_t* sub_key, REGSAM access);
+// Opens the Google Update ClientState key for the current install
+// configuration.  This includes locating the correct key in the face of
+// multi-install.  The flag will by default be written to HKCU, but if
+// --system-level is included in the command line, it will be written to
+// HKLM instead.
+bool OpenInstallStateKey(const Configuration& configuration, RegKey* key) {
+  const HKEY root_key =
+      configuration.is_system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
+  const wchar_t* app_guid = configuration.chrome_app_guid();
+  const REGSAM key_access = KEY_QUERY_VALUE | KEY_SET_VALUE;
+  LONG ret = ERROR_SUCCESS;
 
-  // Returns true if a key is open.
-  bool is_valid() const { return key_ != NULL; }
-
-  // Read a REG_SZ value from the registry into the memory indicated by |value|
-  // (of |value_size| wchar_t units).  Returns ERROR_SUCCESS,
-  // ERROR_FILE_NOT_FOUND, ERROR_MORE_DATA, or some other error.  |value| is
-  // guaranteed to be null-terminated on success.
-  LONG ReadValue(const wchar_t* value_name,
-                 wchar_t* value,
-                 size_t value_size) const;
-
-  // Write a REG_SZ value to the registry.  |value| must be null-terminated.
-  // Returns ERROR_SUCCESS or an error code.
-  LONG WriteValue(const wchar_t* value_name, const wchar_t* value);
-
-  // Closes the key if it was open.
-  void Close();
-
- private:
-  RegKey(const RegKey&);
-  RegKey& operator=(const RegKey&);
-
-  HKEY key_;
-};  // class RegKey
-
-LONG RegKey::Open(HKEY key, const wchar_t* sub_key, REGSAM access) {
-  Close();
-  return ::RegOpenKeyEx(key, sub_key, NULL, access, &key_);
+  return RegKey::OpenClientStateKey(root_key, app_guid, key_access, key);
 }
 
-LONG RegKey::ReadValue(const wchar_t* value_name,
-                       wchar_t* value,
-                       size_t value_size) const {
-  DWORD type;
-  DWORD byte_length = static_cast<DWORD>(value_size * sizeof(wchar_t));
-  LONG result = ::RegQueryValueEx(key_, value_name, NULL, &type,
-                                  reinterpret_cast<BYTE*>(value),
-                                  &byte_length);
-  if (result == ERROR_SUCCESS) {
-    if (type != REG_SZ) {
-      result = ERROR_NOT_SUPPORTED;
-    } else if (byte_length == 0) {
-      *value = L'\0';
-    } else if (value[byte_length/sizeof(wchar_t) - 1] != L'\0') {
-      if ((byte_length / sizeof(wchar_t)) < value_size)
-        value[byte_length / sizeof(wchar_t)] = L'\0';
-      else
-        result = ERROR_MORE_DATA;
+// Writes install results into registry where it is read by Google Update.
+// Don't write anything if there is already a result present, likely
+// written by setup.exe.
+void WriteInstallResults(const Configuration& configuration,
+                         ProcessExitCode result) {
+#if defined(GOOGLE_CHROME_BUILD)
+  // Calls to setup.exe will write a "success" result if everything was good
+  // so we don't need to write anything from here.
+  if (result.IsSuccess())
+    return;
+
+  RegKey key;
+  DWORD value;
+  if (OpenInstallStateKey(configuration, &key)) {
+    if (key.ReadDWValue(kInstallerResultRegistryValue, &value)
+            != ERROR_SUCCESS || value == 0) {
+      key.WriteDWValue(kInstallerResultRegistryValue,
+                       result.exit_code ? 1 /* FAILED_CUSTOM_ERROR */
+                                        : 0 /* SUCCESS */);
+      key.WriteDWValue(kInstallerErrorRegistryValue, result.exit_code);
+      key.WriteDWValue(kInstallerExtraCode1RegistryValue, result.windows_code);
     }
+    key.Close();
   }
-  return result;
-}
-
-LONG RegKey::WriteValue(const wchar_t* value_name, const wchar_t* value) {
-  return ::RegSetValueEx(key_, value_name, 0, REG_SZ,
-                         reinterpret_cast<const BYTE*>(value),
-                         (lstrlen(value) + 1) * sizeof(wchar_t));
-}
-
-void RegKey::Close() {
-  if (key_ != NULL) {
-    ::RegCloseKey(key_);
-    key_ = NULL;
-  }
-}
-
-// Helper function to read a value from registry. Returns true if value
-// is read successfully and stored in parameter value. Returns false otherwise.
-// |size| is measured in wchar_t units.
-bool ReadValueFromRegistry(HKEY root_key, const wchar_t *sub_key,
-                           const wchar_t *value_name, wchar_t *value,
-                           size_t size) {
-  RegKey key;
-
-  if (key.Open(root_key, sub_key, KEY_QUERY_VALUE) == ERROR_SUCCESS &&
-      key.ReadValue(value_name, value, size) == ERROR_SUCCESS) {
-    return true;
-  }
-  return false;
-}
-
-// Opens the Google Update ClientState key for a product.  This finds only
-// registry entries for Chrome; it does not support the Chromium registry
-// layout.
-bool OpenClientStateKey(HKEY root_key, const wchar_t* app_guid, REGSAM access,
-                        RegKey* key) {
-  PathString client_state_key;
-  return client_state_key.assign(kClientStateKeyBase) &&
-         client_state_key.append(app_guid) &&
-         (key->Open(root_key,
-                    client_state_key.get(),
-                    access | KEY_WOW64_32KEY) == ERROR_SUCCESS);
+#endif
 }
 
 // This function sets the flag in registry to indicate that Google Update
 // should try full installer next time. If the current installer works, this
-// flag is cleared by setup.exe at the end of install. The flag will by default
-// be written to HKCU, but if --system-level is included in the command line,
-// it will be written to HKLM instead.
-// TODO(grt): Write a unit test for this that uses registry virtualization.
+// flag is cleared by setup.exe at the end of install.
 void SetInstallerFlags(const Configuration& configuration) {
   RegKey key;
-  const REGSAM key_access = KEY_QUERY_VALUE | KEY_SET_VALUE;
-  const HKEY root_key =
-      configuration.is_system_level() ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
-  // This is ignored if multi-install is true.
-  const wchar_t* app_guid =
-      configuration.has_chrome_frame() ?
-          google_update::kChromeFrameAppGuid :
-          configuration.chrome_app_guid();
   StackString<128> value;
   LONG ret = ERROR_SUCCESS;
 
-  // When multi_install is true, we are potentially:
-  // 1. Performing a multi-install of some product(s) on a clean machine.
-  //    Neither the product(s) nor the multi-installer will have a ClientState
-  //    key in the registry, so there is nothing to be done.
-  // 2. Upgrading an existing multi-install.  The multi-installer will have a
-  //    ClientState key in the registry.  Only it need be modified.
-  // 3. Migrating a single-install into a multi-install.  The product will have
-  //    a ClientState key in the registry.  Only it need be modified.
-  // To handle all cases, we inspect the product's ClientState to see if it
-  // exists and its "ap" value does not contain "-multi".  This is case 3, so we
-  // modify the product's ClientState.  Otherwise, we check the
-  // multi-installer's ClientState and modify it if it exists.
-  if (configuration.is_multi_install()) {
-    if (OpenClientStateKey(root_key, app_guid, key_access, &key)) {
-      // The product has a client state key.  See if it's a single-install.
-      ret = key.ReadValue(kApRegistryValue, value.get(), value.capacity());
-      if (ret != ERROR_FILE_NOT_FOUND &&
-          (ret != ERROR_SUCCESS ||
-           FindTagInStr(value.get(), kMultiInstallTag, NULL))) {
-        // Error or case 2: modify the multi-installer's value.
-        key.Close();
-        app_guid = google_update::kMultiInstallAppGuid;
-      }  // else case 3: modify this value.
-    } else {
-      // case 1 or 2: modify the multi-installer's value.
-      key.Close();
-      app_guid = google_update::kMultiInstallAppGuid;
-    }
-  }
+  if (!OpenInstallStateKey(configuration, &key))
+    return;
 
-  if (!key.is_valid()) {
-    if (!OpenClientStateKey(root_key, app_guid, key_access, &key))
-      return;
-
-    value.clear();
-    ret = key.ReadValue(kApRegistryValue, value.get(), value.capacity());
-  }
+  ret = key.ReadSZValue(kApRegistryValue, value.get(), value.capacity());
 
   // The conditions below are handling two cases:
   // 1. When ap value is present, we want to add the required tag only if it is
   //    not present.
   // 2. When ap value is missing, we are going to create it with the required
   //    tag.
   if ((ret == ERROR_SUCCESS) || (ret == ERROR_FILE_NOT_FOUND)) {
     if (ret == ERROR_FILE_NOT_FOUND)
       value.clear();
 
     if (!StrEndsWith(value.get(), kFullInstallerSuffix) &&
         value.append(kFullInstallerSuffix)) {
-      key.WriteValue(kApRegistryValue, value.get());
+      key.WriteSZValue(kApRegistryValue, value.get());
     }
   }
 }
 
 // Gets the setup.exe path from Registry by looking at the value of Uninstall
 // string.  |size| is measured in wchar_t units.
 ProcessExitCode GetSetupExePathForAppGuid(bool system_level,
                                           const wchar_t* app_guid,
                                           const wchar_t* previous_version,
                                           wchar_t* path,
                                           size_t size) {
   const HKEY root_key = system_level ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
   RegKey key;
-  if (!OpenClientStateKey(root_key, app_guid, KEY_QUERY_VALUE, &key) ||
-      (key.ReadValue(kUninstallRegistryValue, path, size) != ERROR_SUCCESS)) {
-    return UNABLE_TO_FIND_REGISTRY_KEY;
-  }
+  if (!RegKey::OpenClientStateKey(root_key, app_guid, KEY_QUERY_VALUE, &key))
+    return ProcessExitCode(UNABLE_TO_FIND_REGISTRY_KEY);
+  DWORD result = key.ReadSZValue(kUninstallRegistryValue, path, size);
+  if (result != ERROR_SUCCESS)
+    return ProcessExitCode(UNABLE_TO_FIND_REGISTRY_KEY, result);
 
   // Check that the path to the existing installer includes the expected
   // version number.  It's not necessary for accuracy to verify before/after
   // delimiters.
   if (!SearchStringI(path, previous_version))
-    return PATCH_NOT_FOR_INSTALLED_VERSION;
+    return ProcessExitCode(PATCH_NOT_FOR_INSTALLED_VERSION);
 
-  return SUCCESS_EXIT_CODE;
+  return ProcessExitCode(SUCCESS_EXIT_CODE);
 }
 
 // Gets the path to setup.exe of the previous version. The overall path is found
 // in the Uninstall string in the registry. A previous version number specified
 // in |configuration| is used if available. |size| is measured in wchar_t units.
 ProcessExitCode GetPreviousSetupExePath(const Configuration& configuration,
                                         wchar_t* path,
                                         size_t size) {
   bool system_level = configuration.is_system_level();
   const wchar_t* previous_version = configuration.previous_version();
-  ProcessExitCode exit_code = GENERIC_ERROR;
+  ProcessExitCode exit_code = ProcessExitCode(GENERIC_ERROR);
 
   // If this is a multi install, first try looking in the binaries for the path.
   if (configuration.is_multi_install()) {
     exit_code = GetSetupExePathForAppGuid(
         system_level, google_update::kMultiInstallAppGuid, previous_version,
         path, size);
   }
 
   // Failing that, look in Chrome Frame's client state key if --chrome-frame was
   // specified.
-  if (exit_code != SUCCESS_EXIT_CODE && configuration.has_chrome_frame()) {
+  if (!exit_code.IsSuccess() && configuration.has_chrome_frame()) {
     exit_code = GetSetupExePathForAppGuid(
         system_level, google_update::kChromeFrameAppGuid, previous_version,
         path, size);
   }
 
   // Make a last-ditch effort to look in the Chrome client state key.
-  if (exit_code != SUCCESS_EXIT_CODE) {
+  if (!exit_code.IsSuccess()) {
     exit_code = GetSetupExePathForAppGuid(
         system_level, configuration.chrome_app_guid(), previous_version,
         path, size);
   }
 
   return exit_code;
 }
 
 // Calls CreateProcess with good default parameters and waits for the process to
 // terminate returning the process exit code. |exit_code|, if non-NULL, is
 // populated with the process exit code.
 ProcessExitCode RunProcessAndWait(const wchar_t* exe_path, wchar_t* cmdline) {
   STARTUPINFOW si = {sizeof(si)};
   PROCESS_INFORMATION pi = {0};
   if (!::CreateProcess(exe_path, cmdline, NULL, NULL, FALSE, CREATE_NO_WINDOW,
                        NULL, NULL, &si, &pi)) {
-    return COULD_NOT_CREATE_PROCESS;
+    return ProcessExitCode(COULD_NOT_CREATE_PROCESS, ::GetLastError());
   }
 
   ::CloseHandle(pi.hThread);
 
-  ProcessExitCode exit_code = SUCCESS_EXIT_CODE;
+  DWORD exit_code = SUCCESS_EXIT_CODE;
   DWORD wr = ::WaitForSingleObject(pi.hProcess, INFINITE);
-  if (WAIT_OBJECT_0 != wr || !::GetExitCodeProcess(pi.hProcess, &exit_code))
-    exit_code = WAIT_FOR_PROCESS_FAILED;
+  if (WAIT_OBJECT_0 != wr || !::GetExitCodeProcess(pi.hProcess, &exit_code)) {
+    // Note:  We've assumed that WAIT_OBJCT_0 != wr means a failure.  The call
+    // could return a different object but since we never spawn more than one
+    // sub-process at a time that case should never happen.
+    return ProcessExitCode(WAIT_FOR_PROCESS_FAILED, ::GetLastError());
+  }
 
   ::CloseHandle(pi.hProcess);
 
-  return exit_code;
+  return ProcessExitCode(exit_code);
 }
 
 // Appends any command line params passed to mini_installer to the given buffer
 // so that they can be passed on to setup.exe.
 // |buffer| is unchanged in case of error.
 void AppendCommandLineFlags(const Configuration& configuration,
                             CommandString* buffer) {
   PathString full_exe_path;
   size_t len = ::GetModuleFileName(NULL, full_exe_path.get(),
                                    full_exe_path.capacity());
@@ skipping 109 matching lines @@
 // uncompressed 'BN' resources are also extracted. This is generally the set of
 // DLLs/resources needed by setup.exe to run.
 ProcessExitCode UnpackBinaryResources(const Configuration& configuration,
                                       HMODULE module, const wchar_t* base_path,
                                       PathString* archive_path,
                                       PathString* setup_path) {
   // Generate the setup.exe path where we patch/uncompress setup resource.
   PathString setup_dest_path;
   if (!setup_dest_path.assign(base_path) ||
       !setup_dest_path.append(kSetupExe))
-    return PATH_STRING_OVERFLOW;
+    return ProcessExitCode(PATH_STRING_OVERFLOW);
 
   // Prepare the input to OnResourceFound method that needs a location where
   // it will write all the resources.
   Context context = {
     base_path,
     archive_path,
     setup_path,
   };
 
   // Get the resources of type 'B7' (7zip archive).
   // We need a chrome archive to do the installation. So if there
   // is a problem in fetching B7 resource, just return an error.
   if (!::EnumResourceNames(module, kLZMAResourceType, OnResourceFound,
-                           reinterpret_cast<LONG_PTR>(&context)) ||
-      archive_path->length() == 0)
-    return UNABLE_TO_EXTRACT_CHROME_ARCHIVE;
+                           reinterpret_cast<LONG_PTR>(&context))) {
+    return ProcessExitCode(UNABLE_TO_EXTRACT_CHROME_ARCHIVE, ::GetLastError());
+  }
+  if (archive_path->length() == 0) {
+    return ProcessExitCode(UNABLE_TO_EXTRACT_CHROME_ARCHIVE);
+  }
 
-  ProcessExitCode exit_code = SUCCESS_EXIT_CODE;
+  ProcessExitCode exit_code = ProcessExitCode(SUCCESS_EXIT_CODE);
 
   // If we found setup 'B7' resource (used for differential updates), handle
   // it.  Note that this is only for Chrome; Chromium installs are always
   // "full" installs.
   if (setup_path->length() > 0) {
     CommandString cmd_line;
     PathString exe_path;
     // Get the path to setup.exe first.
     exit_code = GetPreviousSetupExePath(configuration, exe_path.get(),
                                         exe_path.capacity());
-    if (exit_code == SUCCESS_EXIT_CODE) {
+    if (exit_code.IsSuccess()) {
       if (!cmd_line.append(exe_path.get()) ||
           !cmd_line.append(L" --") ||
           !cmd_line.append(kCmdUpdateSetupExe) ||
           !cmd_line.append(L"=\"") ||
           !cmd_line.append(setup_path->get()) ||
           !cmd_line.append(L"\" --") ||
           !cmd_line.append(kCmdNewSetupExe) ||
           !cmd_line.append(L"=\"") ||
           !cmd_line.append(setup_dest_path.get()) ||
           !cmd_line.append(L"\"")) {
-        exit_code = COMMAND_STRING_OVERFLOW;
+        exit_code = ProcessExitCode(COMMAND_STRING_OVERFLOW);
       }
     }
 
     // Get any command line option specified for mini_installer and pass them
     // on to setup.exe.  This is important since switches such as
     // --multi-install and --chrome-frame affect where setup.exe will write
     // installer results for consumption by Google Update.
     AppendCommandLineFlags(configuration, &cmd_line);
 
-    if (exit_code == SUCCESS_EXIT_CODE)
+    if (exit_code.IsSuccess())
       exit_code = RunProcessAndWait(exe_path.get(), cmd_line.get());
 
-    if (exit_code != SUCCESS_EXIT_CODE)
+    if (!exit_code.IsSuccess())
       DeleteFile(setup_path->get());
     else if (!setup_path->assign(setup_dest_path.get()))
-      exit_code = PATH_STRING_OVERFLOW;
+      exit_code = ProcessExitCode(PATH_STRING_OVERFLOW);
 
     return exit_code;
   }
 
   // setup.exe wasn't sent as 'B7', lets see if it was sent as 'BL'
   // (compressed setup).
   if (!::EnumResourceNames(module, kLZCResourceType, OnResourceFound,
                            reinterpret_cast<LONG_PTR>(&context)) &&
-      ::GetLastError() != ERROR_RESOURCE_TYPE_NOT_FOUND)
-    return UNABLE_TO_EXTRACT_SETUP_B7;
+      ::GetLastError() != ERROR_RESOURCE_TYPE_NOT_FOUND) {
+    return ProcessExitCode(UNABLE_TO_EXTRACT_SETUP_B7, ::GetLastError());
+  }
 
   if (setup_path->length() > 0) {
     // Uncompress LZ compressed resource. Setup is packed with 'MSCF'
     // as opposed to old DOS way of 'SZDD'. Hence we don't use LZCopy.
     bool success = mini_installer::Expand(setup_path->get(),
                                           setup_dest_path.get());
     ::DeleteFile(setup_path->get());
     if (success) {
       if (!setup_path->assign(setup_dest_path.get())) {
         ::DeleteFile(setup_dest_path.get());
-        exit_code = PATH_STRING_OVERFLOW;
+        exit_code = ProcessExitCode(PATH_STRING_OVERFLOW);
       }
     } else {
-      exit_code = UNABLE_TO_EXTRACT_SETUP_EXE;
+      exit_code = ProcessExitCode(UNABLE_TO_EXTRACT_SETUP_EXE);
     }
 
 #if defined(COMPONENT_BUILD)
     // Extract the (uncompressed) modules required by setup.exe.
     if (!::EnumResourceNames(module, kBinResourceType, WriteResourceToDirectory,
-                             reinterpret_cast<LONG_PTR>(base_path)))
-      return UNABLE_TO_EXTRACT_SETUP;
+                             reinterpret_cast<LONG_PTR>(base_path))) {
+      return ProcessExitCode(UNABLE_TO_EXTRACT_SETUP, ::GetLastError());
+    }
 #endif
 
     return exit_code;
   }
 
   // setup.exe still not found. So finally check if it was sent as 'BN'
   // (uncompressed setup).
   // TODO(tommi): We don't need BN anymore so let's remove it (and remove
   // it from create_installer_archive.py).
   if (!::EnumResourceNames(module, kBinResourceType, OnResourceFound,
                            reinterpret_cast<LONG_PTR>(&context)) &&
-      ::GetLastError() != ERROR_RESOURCE_TYPE_NOT_FOUND)
-    return UNABLE_TO_EXTRACT_SETUP_BN;
+      ::GetLastError() != ERROR_RESOURCE_TYPE_NOT_FOUND) {
+    return ProcessExitCode(UNABLE_TO_EXTRACT_SETUP_BN, ::GetLastError());
+  }
 
   if (setup_path->length() > 0) {
     if (setup_path->comparei(setup_dest_path.get()) != 0) {
       if (!::MoveFileEx(setup_path->get(), setup_dest_path.get(),
                         MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING)) {
         ::DeleteFile(setup_path->get());
         setup_path->clear();
       } else if (!setup_path->assign(setup_dest_path.get())) {
         ::DeleteFile(setup_dest_path.get());
       }
     }
   }
 
   if (setup_path->length() == 0)
-    exit_code = UNABLE_TO_EXTRACT_SETUP;
+    exit_code = ProcessExitCode(UNABLE_TO_EXTRACT_SETUP);
 
   return exit_code;
 }
 
 // Executes setup.exe, waits for it to finish and returns the exit code.
 ProcessExitCode RunSetup(const Configuration& configuration,
                          const wchar_t* archive_path,
                          const wchar_t* setup_path) {
   // There could be three full paths in the command line for setup.exe (path
   // to exe itself, path to archive and path to log file), so we declare
   // total size as three + one additional to hold command line options.
   CommandString cmd_line;
 
   // Get the path to setup.exe first.
   if (::lstrlen(setup_path) > 0) {
     if (!cmd_line.assign(L"\"") ||
         !cmd_line.append(setup_path) ||
         !cmd_line.append(L"\""))
-      return COMMAND_STRING_OVERFLOW;
+      return ProcessExitCode(COMMAND_STRING_OVERFLOW);
   } else {
     ProcessExitCode exit_code = GetPreviousSetupExePath(
         configuration, cmd_line.get(), cmd_line.capacity());
-    if (exit_code != SUCCESS_EXIT_CODE)
+    if (!exit_code.IsSuccess())
       return exit_code;
   }
 
   // Append the command line param for chrome archive file.
   if (!cmd_line.append(L" --") ||
 #if defined(COMPONENT_BUILD)
       // For faster developer turnaround, the component build generates
       // uncompressed archives.
       !cmd_line.append(kCmdUncompressedArchive) ||
 #else
       !cmd_line.append(kCmdInstallArchive) ||
 #endif
       !cmd_line.append(L"=\"") ||
       !cmd_line.append(archive_path) ||
       !cmd_line.append(L"\""))
-    return COMMAND_STRING_OVERFLOW;
+    return ProcessExitCode(COMMAND_STRING_OVERFLOW);
 
   // Append the command line param for chrome previous version.
   if (configuration.previous_version() &&
       (!cmd_line.append(L" --") ||
        !cmd_line.append(kCmdPreviousVersion) ||
        !cmd_line.append(L"=\"") ||
        !cmd_line.append(configuration.previous_version()) ||
        !cmd_line.append(L"\""))) {
-    return COMMAND_STRING_OVERFLOW;
+    return ProcessExitCode(COMMAND_STRING_OVERFLOW);
   }
 
   // Get any command line option specified for mini_installer and pass them
   // on to setup.exe
   AppendCommandLineFlags(configuration, &cmd_line);
 
   return RunProcessAndWait(NULL, cmd_line.get());
 }
 
 // Deletes given files and working dir.
@@ skipping 190 matching lines @@
 // Checks the command line for specific mini installer flags.
 // If the function returns true, the command line has been processed and all
 // required actions taken.  The installer must exit and return the returned
 // |exit_code|.
 bool ProcessNonInstallOperations(const Configuration& configuration,
                                  ProcessExitCode* exit_code) {
   switch (configuration.operation()) {
     case Configuration::CLEANUP:
       // Cleanup has already taken place in DeleteOldChromeTempDirectories at
       // this point, so just tell our caller to exit early.
-      *exit_code = SUCCESS_EXIT_CODE;
+      *exit_code = ProcessExitCode(SUCCESS_EXIT_CODE);
       return true;
 
     default:
       return false;
   }
 }
 
 // Returns true if we should delete the temp files we create (default).
 // Returns false iff the user has manually created a ChromeInstallerCleanup
 // string value in the registry under HKCU\\Software\\[Google|Chromium]
 // and set its value to "0".  That explicitly forbids the mini installer from
 // deleting these files.
 // Support for this has been publicly mentioned in troubleshooting tips so
 // we continue to support it.
 bool ShouldDeleteExtractedFiles() {
   wchar_t value[2] = {0};
-  if (ReadValueFromRegistry(HKEY_CURRENT_USER, kCleanupRegistryKey,
-                            kCleanupRegistryValue, value, _countof(value)) &&
+  if (RegKey::ReadSZValue(HKEY_CURRENT_USER, kCleanupRegistryKey,
+                          kCleanupRegistryValue, value, _countof(value)) &&
       value[0] == L'0') {
     return false;
   }
 
   return true;
 }
 
 // Main function. First gets a working dir, unpacks the resources and finally
 // executes setup.exe to do the install/upgrade.
 ProcessExitCode WMain(HMODULE module) {
   // Always start with deleting potential leftovers from previous installations.
   // This can make the difference between success and failure.  We've seen
   // many installations out in the field fail due to out of disk space problems
   // so this could buy us some space.
   DeleteOldChromeTempDirectories();
 
-  ProcessExitCode exit_code = SUCCESS_EXIT_CODE;
+  ProcessExitCode exit_code = ProcessExitCode(SUCCESS_EXIT_CODE);
 
   // Parse configuration from the command line and resources.
   Configuration configuration;
   if (!configuration.Initialize(module))
-    return GENERIC_INITIALIZATION_FAILURE;
+    return ProcessExitCode(GENERIC_INITIALIZATION_FAILURE);
 
   // If the --cleanup switch was specified on the command line, then that means
   // we should only do the cleanup and then exit.
   if (ProcessNonInstallOperations(configuration, &exit_code))
     return exit_code;
 
   // First get a path where we can extract payload
   PathString base_path;
   if (!GetWorkDir(module, &base_path))
-    return UNABLE_TO_GET_WORK_DIRECTORY;
+    return ProcessExitCode(UNABLE_TO_GET_WORK_DIRECTORY);
 
 #if defined(GOOGLE_CHROME_BUILD)
   // Set the magic suffix in registry to try full installer next time. We ignore
   // any errors here and we try to set the suffix for user level unless
   // --system-level is on the command line in which case we set it for system
   // level instead. This only applies to the Google Chrome distribution.
   SetInstallerFlags(configuration);
 #endif
 
   PathString archive_path;
   PathString setup_path;
   exit_code = UnpackBinaryResources(configuration, module, base_path.get(),
                                     &archive_path, &setup_path);
 
   // While unpacking the binaries, we paged in a whole bunch of memory that
   // we don't need anymore.  Let's give it back to the pool before running
   // setup.
   ::SetProcessWorkingSetSize(::GetCurrentProcess(), -1, -1);
 
-  if (exit_code == SUCCESS_EXIT_CODE)
+  if (exit_code.IsSuccess())
     exit_code = RunSetup(configuration, archive_path.get(), setup_path.get());
 
   if (ShouldDeleteExtractedFiles())
     DeleteExtractedFiles(base_path.get(), archive_path.get(), setup_path.get());
 
+  WriteInstallResults(configuration, exit_code);
   return exit_code;
 }
 
 }  // namespace mini_installer
 
 int MainEntryPoint() {
   mini_installer::ProcessExitCode result =
       mini_installer::WMain(::GetModuleHandle(NULL));
-  ::ExitProcess(result);
+
+  ::ExitProcess(result.exit_code);
 }
 
 // VC Express editions don't come with the memset CRT obj file and linking to
 // the obj files between versions becomes a bit problematic. Therefore,
 // simply implement memset.
 //
 // This also avoids having to explicitly set the __sse2_available hack when
 // linking with both the x64 and x86 obj files which is required when not
 // linking with the std C lib in certain instances (including Chromium) with
 // MSVC.  __sse2_available determines whether to use SSE2 intructions with
 // std C lib routines, and is set by MSVC's std C lib implementation normally.
 extern "C" {
 #pragma function(memset)
 void* memset(void* dest, int c, size_t count) {
   void* start = dest;
   while (count--) {
     *reinterpret_cast<char*>(dest) = static_cast<char>(c);
     dest = reinterpret_cast<char*>(dest) + 1;
   }
   return start;
 }
 }  // extern "C"