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
|
// Copyright 2013 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/browser/install_verification/win/module_ids.h"
#include "base/strings/string_piece.h"
#include "chrome/browser/install_verification/win/module_verification_common.h"
#include "testing/gtest/include/gtest/gtest.h"
TEST(ModuleIDsTest, LoadModuleIDs) {
ModuleIDs module_ids;
LoadModuleIDs(&module_ids);
ASSERT_EQ(1u, module_ids[CalculateModuleNameDigest(L"chrome.dll")]);
}
TEST(ModuleIDsTest, ParseAdditionalModuleIDs) {
ModuleIDs module_ids;
ParseAdditionalModuleIDs(base::StringPiece(), &module_ids);
ASSERT_EQ(0u, module_ids.size());
// Invalid input.
ParseAdditionalModuleIDs("hello", &module_ids);
ASSERT_EQ(0u, module_ids.size());
ParseAdditionalModuleIDs("hello world", &module_ids);
ASSERT_EQ(0u, module_ids.size());
ParseAdditionalModuleIDs("hello world\nfoo bar", &module_ids);
ASSERT_EQ(0u, module_ids.size());
ParseAdditionalModuleIDs("123 world\nfoo bar", &module_ids);
ASSERT_EQ(0u, module_ids.size());
ParseAdditionalModuleIDs("hello 0123456789abcdef0123456789abcdef\nfoo bar",
&module_ids);
ASSERT_EQ(0u, module_ids.size());
// A single valid line followed by an invalid line.
ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\nfoo bar",
&module_ids);
ASSERT_EQ(1u, module_ids.size());
ASSERT_EQ(123u, module_ids.begin()->second);
ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
module_ids.clear();
// The same, but with \r\n.
ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\nfoo bar",
&module_ids);
ASSERT_EQ(1u, module_ids.size());
ASSERT_EQ(123u, module_ids.begin()->second);
ASSERT_EQ("0123456789abcdef0123456789abcdef", module_ids.begin()->first);
module_ids.clear();
// Several valid and invalid lines, with varying line terminations etc.
ParseAdditionalModuleIDs("123 0123456789abcdef0123456789abcdef\r\n"
"456 DEADBEEFDEADBEEF\r"
"789 DEADBEEFDEADBEEFDEADBEEFDEADBEEF\n"
"\n"
"\n"
"321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE\n",
&module_ids);
ASSERT_EQ(3u, module_ids.size());
ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
ASSERT_EQ(789u, module_ids["DEADBEEFDEADBEEFDEADBEEFDEADBEEF"]);
ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
module_ids.clear();
// Leading empty lines, no termination on final line.
ParseAdditionalModuleIDs("\n"
"123 0123456789abcdef0123456789abcdef\r\n"
"321 BAADCAFEBAADCAFEBAADCAFEBAADCAFE",
&module_ids);
ASSERT_EQ(2u, module_ids.size());
ASSERT_EQ(123u, module_ids["0123456789abcdef0123456789abcdef"]);
ASSERT_EQ(321u, module_ids["BAADCAFEBAADCAFEBAADCAFEBAADCAFE"]);
module_ids.clear();
}
|
