1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
// Copyright (c) 2010 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 "chrome/installer/util/package.h"
#include "base/file_util.h"
#include "base/logging.h"
#include "base/utf_string_conversions.h"
#include "base/win/registry.h"
#include "chrome/installer/util/delete_tree_work_item.h"
#include "chrome/installer/util/google_update_constants.h"
#include "chrome/installer/util/package_properties.h"
#include "chrome/installer/util/product.h"
using base::win::RegKey;
namespace installer {
Package::Package(bool system_level, const FilePath& path,
PackageProperties* properties)
: system_level_(system_level), path_(path), properties_(properties) {
DCHECK(properties_);
}
Package::~Package() {
}
const FilePath& Package::path() const {
return path_;
}
const Products& Package::products() const {
return products_;
}
PackageProperties* Package::properties() const {
return properties_;
}
bool Package::IsEqual(const FilePath& path) const {
return FilePath::CompareEqualIgnoreCase(path_.value(), path.value());
}
void Package::AssociateProduct(const Product* product) {
#ifndef NDEBUG
for (size_t i = 0; i < products_.size(); ++i) {
DCHECK_NE(product->distribution()->GetType(),
products_[i]->distribution()->GetType());
}
#endif
products_.push_back(product);
}
bool Package::system_level() const {
return system_level_;
}
FilePath Package::GetInstallerDirectory(
const Version& version) const {
return path_.Append(UTF8ToWide(version.GetString()))
.Append(installer::kInstallerDir);
}
Version* Package::GetCurrentVersion() const {
scoped_ptr<Version> current_version;
// Be aware that there might be a pending "new_chrome.exe" already in the
// installation path.
FilePath new_chrome_exe(path_.Append(installer::kChromeNewExe));
bool new_chrome_exists = file_util::PathExists(new_chrome_exe);
HKEY root = system_level_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
for (size_t i = 0; i < products_.size(); ++i) {
const Product* product = products_[i];
RegKey chrome_key(root, product->distribution()->GetVersionKey().c_str(),
KEY_READ);
std::wstring version;
if (new_chrome_exists)
chrome_key.ReadValue(google_update::kRegOldVersionField, &version);
if (version.empty())
chrome_key.ReadValue(google_update::kRegVersionField, &version);
if (!version.empty()) {
scoped_ptr<Version> this_version(Version::GetVersionFromString(version));
if (this_version.get()) {
if (!current_version.get() ||
(current_version->CompareTo(*this_version) > 0)) {
current_version.reset(this_version.release());
} else if (current_version.get()) {
DCHECK_EQ(current_version->GetString(), this_version->GetString())
<< "found distributions of different versions in the same "
"installation folder!";
}
}
}
}
return current_version.release();
}
void Package::RemoveOldVersionDirectories(
const Version& latest_version) const {
file_util::FileEnumerator version_enum(path_, false,
file_util::FileEnumerator::DIRECTORIES);
scoped_ptr<Version> version;
// We try to delete all directories whose versions are lower than
// latest_version.
FilePath next_version = version_enum.Next();
while (!next_version.empty()) {
file_util::FileEnumerator::FindInfo find_data = {0};
version_enum.GetFindInfo(&find_data);
VLOG(1) << "directory found: " << find_data.cFileName;
version.reset(Version::GetVersionFromString(find_data.cFileName));
if (version.get() && (latest_version.CompareTo(*version) > 0)) {
std::vector<FilePath> key_files;
for (Products::const_iterator it = products_.begin();
it != products_.end(); ++it) {
BrowserDistribution* dist = it->get()->distribution();
std::vector<FilePath> dist_key_files(dist->GetKeyFiles());
std::vector<FilePath>::const_iterator key_file_iter(
dist_key_files.begin());
for (; key_file_iter != dist_key_files.end(); ++key_file_iter) {
key_files.push_back(next_version.Append(*key_file_iter));
}
}
VLOG(1) << "Deleting directory: " << next_version.value();
scoped_ptr<DeleteTreeWorkItem> item(
WorkItem::CreateDeleteTreeWorkItem(next_version, key_files));
if (!item->Do())
item->Rollback();
}
next_version = version_enum.Next();
}
}
} // namespace installer
|
