gcapi_reactivation.cc revision eb525c5499e34cc9c4b825d6d9e75bb07cc06ace
1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver// Copyright (c) 2012 The Chromium Authors. All rights reserved. 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver// Use of this source code is governed by a BSD-style license that can be 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver// found in the LICENSE file. 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "chrome/installer/gcapi/gcapi_reactivation.h" 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "base/time/time.h" 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "base/win/registry.h" 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "chrome/installer/gcapi/gcapi.h" 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#include "chrome/installer/util/google_update_constants.h" 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverusing base::Time; 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverusing base::win::RegKey; 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruvernamespace { 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverconst wchar_t kReactivationHistoryKey[] = L"reactivation"; 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverstd::wstring GetReactivationHistoryKeyPath() { 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver std::wstring reactivation_path(google_update::kRegPathClientState); 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reactivation_path += L"\\"; 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reactivation_path += google_update::kChromeUpgradeCode; 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reactivation_path += L"\\"; 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reactivation_path += kReactivationHistoryKey; 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return reactivation_path; 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} // namespace 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverbool HasBeenReactivated() { 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver RegKey reactivation_key(HKEY_CURRENT_USER, 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver GetReactivationHistoryKeyPath().c_str(), 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver KEY_QUERY_VALUE); 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return reactivation_key.Valid(); 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruverbool SetReactivationBrandCode(const std::wstring& brand_code, int shell_mode) { 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver bool success = false; 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // This function currently only should be run in a non-elevated shell, 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // so we return "true" if it is being invoked from an elevated shell. 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (shell_mode == GCAPI_INVOKED_UAC_ELEVATION) 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return true; 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver std::wstring path(google_update::kRegPathClientState); 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver path += L"\\"; 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver path += google_update::kChromeUpgradeCode; 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver RegKey client_state_key(HKEY_CURRENT_USER, path.c_str(), KEY_SET_VALUE); 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (client_state_key.Valid()) { 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver success = client_state_key.WriteValue( 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver google_update::kRegRLZReactivationBrandField, 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver brand_code.c_str()) == ERROR_SUCCESS; 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (success) { 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // Store this brand code in the reactivation history. Store it with a 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver // a currently un-used timestamp for future proofing. 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver RegKey reactivation_key(HKEY_CURRENT_USER, 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver GetReactivationHistoryKeyPath().c_str(), 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver KEY_WRITE); 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (reactivation_key.Valid()) { 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver int64 timestamp = Time::Now().ToInternalValue(); 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reactivation_key.WriteValue(brand_code.c_str(), 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver ×tamp, 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver sizeof(timestamp), 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver REG_QWORD); 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return success; 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver