base/test/test_reg_util_win.cc

   1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/test/test_reg_util_win.h"
   6
   7 #include "base/guid.h"
   8 #include "base/logging.h"
   9 #include "base/strings/string_number_conversions.h"
  10 #include "base/strings/string_util.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace registry_util {
  15
  16 namespace {
  17
  18 const wchar_t kTimestampDelimiter[] = L"$";
  19 const wchar_t kTempTestKeyPath[] = L"Software\\Chromium\\TempTestKeys";
  20
  21 void DeleteStaleTestKeys(const base::Time& now,
  22                          const base::string16& test_key_root) {
  23   base::win::RegKey test_root_key;
  24   if (test_root_key.Open(HKEY_CURRENT_USER,
  25                          test_key_root.c_str(),
  26                          KEY_ALL_ACCESS) != ERROR_SUCCESS) {
  27     // This will occur on first-run, but is harmless.
  28     return;
  29   }
  30
  31   base::win::RegistryKeyIterator iterator_test_root_key(HKEY_CURRENT_USER,
  32                                                         test_key_root.c_str());
  33   for (; iterator_test_root_key.Valid(); ++iterator_test_root_key) {
  34     base::string16 key_name = iterator_test_root_key.Name();
  35     std::vector<base::string16> tokens;
  36     if (!Tokenize(key_name, base::string16(kTimestampDelimiter), &tokens))
  37       continue;
  38     int64 key_name_as_number = 0;
  39
  40     if (!base::StringToInt64(tokens[0], &key_name_as_number)) {
  41       test_root_key.DeleteKey(key_name.c_str());
  42       continue;
  43     }
  44
  45     base::Time key_time = base::Time::FromInternalValue(key_name_as_number);
  46     base::TimeDelta age = now - key_time;
  47
  48     if (age > base::TimeDelta::FromHours(24))
  49       test_root_key.DeleteKey(key_name.c_str());
  50   }
  51 }
  52
  53 base::string16 GenerateTempKeyPath(const base::string16& test_key_root,
  54                                    const base::Time& timestamp) {
  55   base::string16 key_path = test_key_root;
  56   key_path += L"\\" + base::Int64ToString16(timestamp.ToInternalValue());
  57   key_path += kTimestampDelimiter + base::ASCIIToUTF16(base::GenerateGUID());
  58
  59   return key_path;
  60 }
  61
  62 }  // namespace
  63
  64 RegistryOverrideManager::ScopedRegistryKeyOverride::ScopedRegistryKeyOverride(
  65     HKEY override,
  66     const base::string16& key_path)
  67     : override_(override) {
  68   EXPECT_EQ(
  69       ERROR_SUCCESS,
  70       temp_key_.Create(HKEY_CURRENT_USER, key_path.c_str(), KEY_ALL_ACCESS));
  71   EXPECT_EQ(ERROR_SUCCESS,
  72             ::RegOverridePredefKey(override_, temp_key_.Handle()));
  73 }
  74
  75 RegistryOverrideManager::
  76     ScopedRegistryKeyOverride::~ScopedRegistryKeyOverride() {
  77   ::RegOverridePredefKey(override_, NULL);
  78   temp_key_.DeleteKey(L"");
  79 }
  80
  81 RegistryOverrideManager::RegistryOverrideManager()
  82     : timestamp_(base::Time::Now()), test_key_root_(kTempTestKeyPath) {
  83   DeleteStaleTestKeys(timestamp_, test_key_root_);
  84 }
  85
  86 RegistryOverrideManager::RegistryOverrideManager(
  87     const base::Time& timestamp,
  88     const base::string16& test_key_root)
  89     : timestamp_(timestamp), test_key_root_(test_key_root) {
  90   DeleteStaleTestKeys(timestamp_, test_key_root_);
  91 }
  92
  93 RegistryOverrideManager::~RegistryOverrideManager() {}
  94
  95 void RegistryOverrideManager::OverrideRegistry(HKEY override) {
  96   base::string16 key_path = GenerateTempKeyPath(test_key_root_, timestamp_);
  97   overrides_.push_back(new ScopedRegistryKeyOverride(override, key_path));
  98 }
  99
 100 base::string16 GenerateTempKeyPath() {
 101   return GenerateTempKeyPath(base::string16(kTempTestKeyPath),
 102                              base::Time::Now());
 103 }
 104
 105 }  // namespace registry_util