Register Chrome in Active Setup to be called by Windows at user login.

chrome/installer/setup/setup_util.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.
 //
 // This file declares util functions for setup project.
 
 #include "chrome/installer/setup/setup_util.h"
 
+#include <windows.h>
+
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/string_util.h"
 #include "chrome/installer/util/installer_state.h"
 #include "chrome/installer/util/util_constants.h"
 #include "courgette/courgette.h"
 #include "third_party/bspatch/mbspatch.h"
 
 namespace installer {
 
@@ skipping 112 matching lines @@
       PLOG(ERROR) << "VirtualAllocEx";
       ::TerminateProcess(pi.hProcess, ~0);
     }
     ::CloseHandle(pi.hThread);
     ::CloseHandle(pi.hProcess);
   }
 
   return ok != FALSE;
 }
 
+string16 GetActiveSetupPath(BrowserDistribution* dist) {
+  static const wchar_t* kInstalledComponentsPath =

[tommi (sloooow) - chröme, 2012/07/20 09:59:25]

static const wchar_t kInstalledComponentsPath[] =

[gab, 2012/07/20 14:45:47]

Done.

+      L"Software\\Microsoft\\Active Setup\\Installed Components\\";
+  return kInstalledComponentsPath + dist->GetAppGuid();
+}
+
+ScopedTokenPrivilege::ScopedTokenPrivilege(const wchar_t* privilege_name)
+    : is_enabled_(true) {

[tommi (sloooow) - chröme, 2012/07/20 09:59:25]

nit: initialize to false and just set to true when everything has been done
successfully (see below).

[gab, 2012/07/20 14:45:47]

Done.

+  if (!::OpenProcessToken(::GetCurrentProcess(),
+      TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, token_.Receive())) {
+    is_enabled_ = false;
+    return;
+  }
+
+  LUID privilege_luid;
+  if (!::LookupPrivilegeValue(NULL, privilege_name, &privilege_luid)) {
+    is_enabled_ = false;
+    token_.Close();
+    return;
+  }
+
+  // Adjust the token's privileges to enable |privilege_name|. If this privilege
+  // was already enabled, |previous_privileges_|.PrivilegeCount will be set to 0
+  // and we then know not to disable this privilege upon destruction.
+  TOKEN_PRIVILEGES tp;
+  tp.PrivilegeCount = 1;
+  tp.Privileges[0].Luid = privilege_luid;
+  tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
+  DWORD return_length;
+  if (!::AdjustTokenPrivileges(token_, FALSE, &tp, sizeof(TOKEN_PRIVILEGES),
+                               &previous_privileges_, &return_length)) {
+    is_enabled_ = false;
+    token_.Close();
+  }

[tommi (sloooow) - chröme, 2012/07/20 09:59:25]

} else {
  is_enabled_ = true;
}

[gab, 2012/07/20 14:45:47]

Done.

+}
+
+ScopedTokenPrivilege::~ScopedTokenPrivilege() {
+  if (is_enabled_ && previous_privileges_.PrivilegeCount != 0) {
+    ::AdjustTokenPrivileges(token_, FALSE, &previous_privileges_,
+                            sizeof(TOKEN_PRIVILEGES), NULL, NULL);
+  }
+}
+
 }  // namespace installer