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
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifdef UNSAFE_BUFFERS_BUILD
// TODO(crbug.com/40285824): Remove this and convert code to safer constructs.
#pragma allow_unsafe_buffers
#endif
#include "chrome/install_static/install_modes.h"
#include <windows.h>
#include <cguid.h>
#include "base/strings/string_util.h"
#include "chrome/install_static/buildflags.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using ::testing::Eq;
using ::testing::Gt;
using ::testing::Le;
using ::testing::Ne;
using ::testing::Not;
using ::testing::ResultOf;
using ::testing::StrEq;
using ::testing::StrNe;
namespace install_static {
namespace {
// A matcher that returns true if |arg| contains a character that is neither
// alphanumeric nor a period.
MATCHER(ContainsIllegalProgIdChar, "") {
const wchar_t* scan = arg;
wchar_t c;
while ((c = *scan++) != 0) {
if (!base::IsAsciiAlphaNumeric(c) && c != L'.') {
return true;
}
}
return false;
}
} // namespace
TEST(InstallModes, VerifyModes) {
ASSERT_THAT(kInstallModes.size(), Gt(0));
for (size_t i = 0; i < kInstallModes.size(); ++i) {
const InstallConstants& mode = kInstallModes[i];
// The modes must be listed in order.
ASSERT_THAT(mode.index, Eq(i));
// The first mode must have no install switch; the rest must have one.
if (i == 0)
ASSERT_THAT(mode.install_switch, StrEq(""));
else
ASSERT_THAT(mode.install_switch, StrNe(""));
// The first mode must have no suffix; the rest must have one.
if (i == 0)
ASSERT_THAT(mode.install_suffix, StrEq(L""));
else
ASSERT_THAT(mode.install_suffix, StrNe(L""));
// The first mode must have no logo suffix; the rest must have one.
if (i == 0)
ASSERT_THAT(mode.logo_suffix, StrEq(L""));
else
ASSERT_THAT(mode.logo_suffix, StrNe(L""));
#if BUILDFLAG(USE_GOOGLE_UPDATE_INTEGRATION)
// The modes must have an appguid if Google Update integration is
// supported.
ASSERT_THAT(mode.app_guid, StrNe(L""));
#else
ASSERT_THAT(mode.app_guid, StrEq(L""));
#endif
// Every mode must have a base app name.
ASSERT_THAT(mode.base_app_name, StrNe(L""));
// Every mode must have a base app id, and it must not contain illegal
// ProgId characters.
ASSERT_THAT(mode.base_app_id, StrNe(L""));
ASSERT_THAT(mode.base_app_id, Not(ContainsIllegalProgIdChar()));
// The Browser ProgID prefix must not be empty, must be no greater than 11
// characters long, must contain no punctuation, and may not start with a
// digit (https://msdn.microsoft.com/library/windows/desktop/dd542719.aspx).
ASSERT_THAT(mode.browser_prog_id_prefix, StrNe(L""));
ASSERT_THAT(lstrlen(mode.browser_prog_id_prefix), Le(11));
ASSERT_THAT(mode.browser_prog_id_prefix, Not(ContainsIllegalProgIdChar()));
ASSERT_THAT(*mode.browser_prog_id_prefix, ResultOf(iswdigit, Eq(0)));
// Test the same things for PDF ProgID prefix.
ASSERT_THAT(mode.pdf_prog_id_prefix, StrNe(L""));
ASSERT_THAT(lstrlen(mode.pdf_prog_id_prefix), Le(11));
ASSERT_THAT(mode.pdf_prog_id_prefix, Not(ContainsIllegalProgIdChar()));
ASSERT_THAT(*mode.pdf_prog_id_prefix, ResultOf(iswdigit, Eq(0)));
// The ProgID description must not be empty.
ASSERT_THAT(mode.browser_prog_id_description, StrNe(L""));
// The PDFProgID description also must not be empty.
ASSERT_THAT(mode.pdf_prog_id_description, StrNe(L""));
// Every mode must have an Active Setup GUID.
ASSERT_THAT(mode.active_setup_guid, StrNe(L""));
// Every mode must have a toast activator CLSID.
ASSERT_THAT(mode.toast_activator_clsid, Ne(CLSID_NULL));
// Every mode must have an elevator CLSID.
ASSERT_THAT(mode.elevator_clsid, Ne(CLSID_NULL));
// Every mode must have an elevator IID.
ASSERT_THAT(mode.elevator_iid, Ne(CLSID_NULL));
// Assert that html_doc_icon_resource_index is set.
ASSERT_THAT(mode.html_doc_icon_resource_index, Ne(0));
// Assert that pdf_doc_icon_resource_index is set.
ASSERT_THAT(mode.pdf_doc_icon_resource_index, Ne(0));
// UNSUPPORTED and USE_GOOGLE_UPDATE_INTEGRATION are mutually exclusive.
#if !BUILDFLAG(USE_GOOGLE_UPDATE_INTEGRATION)
ASSERT_THAT(mode.channel_strategy, Eq(ChannelStrategy::UNSUPPORTED));
#endif
}
}
TEST(InstallModes, GetClientsKeyPath) {
constexpr wchar_t kAppGuid[] = L"test";
#if BUILDFLAG(USE_GOOGLE_UPDATE_INTEGRATION)
ASSERT_THAT(GetClientsKeyPath(kAppGuid),
StrEq(L"Software\\Google\\Update\\Clients\\test"));
#else
ASSERT_THAT(GetClientsKeyPath(kAppGuid),
StrEq(std::wstring(L"Software\\").append(kProductPathName)));
#endif
}
TEST(InstallModes, GetClientStateKeyPath) {
constexpr wchar_t kAppGuid[] = L"test";
#if BUILDFLAG(USE_GOOGLE_UPDATE_INTEGRATION)
ASSERT_THAT(GetClientStateKeyPath(kAppGuid),
StrEq(L"Software\\Google\\Update\\ClientState\\test"));
#else
ASSERT_THAT(GetClientStateKeyPath(kAppGuid),
StrEq(std::wstring(L"Software\\").append(kProductPathName)));
#endif
}
TEST(InstallModes, GetClientStateMediumKeyPath) {
constexpr wchar_t kAppGuid[] = L"test";
#if BUILDFLAG(USE_GOOGLE_UPDATE_INTEGRATION)
ASSERT_THAT(GetClientStateMediumKeyPath(kAppGuid),
StrEq(L"Software\\Google\\Update\\ClientStateMedium\\test"));
#else
ASSERT_THAT(GetClientStateMediumKeyPath(kAppGuid),
StrEq(std::wstring(L"Software\\").append(kProductPathName)));
#endif
}
} // namespace install_static
|
