15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/strings/string_number_conversions.h" 6868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 7868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/enumerate_modules_model_win.h" 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)typedef testing::Test EnumerateModulesTest; 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set up some constants to use as default when creating the structs. 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const ModuleEnumerator::ModuleType kType = 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::LOADED_MODULE; 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const ModuleEnumerator::ModuleStatus kStatus = 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED; 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const ModuleEnumerator::RecommendedAction kAction = 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NONE; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 23868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)static const ModuleEnumerator::OperatingSystem kOs = 24868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ModuleEnumerator::ALL; 25868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This is a list of test cases to normalize. 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const struct NormalizationEntryList { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module test_case; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module expected; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} kNormalizationTestCases[] = { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only path normalization needed. 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"c:\\foo\\bar.dll", L"", L"Prod", L"Desc", L"1.0", 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"Sig", kAction}, 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"c:\\foo\\", L"bar.dll", L"Prod", L"Desc", L"1.0", 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"Sig", kAction}, 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Lower case normalization. 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"C:\\Foo\\Bar.dll", L"", L"", L"", L"1.0", 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"", kAction}, 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"c:\\foo\\", L"bar.dll", L"", L"", L"1.0", 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"", kAction}, 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Version can include strings after the version number. Strip that away. 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"c:\\foo.dll", L"", L"", L"", L"1.0 asdf", 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"", kAction}, 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"c:\\", L"foo.dll", L"", L"", L"1.0", 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) L"", kAction}, 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Corner case: No path (not sure this will ever happen). 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"bar.dll", L"", L"", L"", L"", L"", kAction}, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"", L"bar.dll", L"", L"", L"", L"", kAction}, 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Error case: Missing filename (not sure this will ever happen). 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"", L"", L"", L"", L"1.0", L"", kAction}, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) {kType, kStatus, L"", L"", L"", L"", L"1.0", L"", kAction}, 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(EnumerateModulesTest, NormalizeEntry) { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kNormalizationTestCases); ++i) { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module test = kNormalizationTestCases[i].test_case; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(test.normalized); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NormalizeModule(&test); 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module expected = kNormalizationTestCases[i].expected; 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SCOPED_TRACE("Test case no: " + base::IntToString(i)); 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected.type, test.type); 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected.status, test.status); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.location.c_str(), test.location.c_str()); 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.name.c_str(), test.name.c_str()); 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.product_name.c_str(), test.product_name.c_str()); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.description.c_str(), test.description.c_str()); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.version.c_str(), test.version.c_str()); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_STREQ(expected.digital_signer.c_str(), test.digital_signer.c_str()); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected.recommended_action, test.recommended_action); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test.normalized); 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ModuleEnumerator::Module kStandardModule = 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kType, kStatus, L"c:\\foo\\bar.dll", L"", L"Prod", L"Desc", L"1.0", L"Sig", 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NONE }; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ModuleEnumerator::Module kStandardModuleNoDescription = 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kType, kStatus, L"c:\\foo\\bar.dll", L"", L"Prod", L"", L"1.0", L"Sig", 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NONE }; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const ModuleEnumerator::Module kStandardModuleNoSignature = 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kType, kStatus, L"c:\\foo\\bar.dll", L"", L"Prod", L"Desc", L"1.0", L"", 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NONE }; 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Name, location, description and signature are compared by hashing. 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kMatchName[] = "88e8c9e0"; // "bar.dll". 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kNoMatchName[] = "barfoo.dll"; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kMatchLocation[] = "e6ca7b1c"; // "c:\\foo\\". 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kNoMatchLocation[] = "c:\\foobar\\"; 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kMatchDesc[] = "5c4419a6"; // "Desc". 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kNoMatchDesc[] = "NoDesc"; 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kVersionHigh[] = "2.0"; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kVersionLow[] = "0.5"; 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kMatchSignature[] = "7bfd87e1"; // "Sig". 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kNoMatchSignature[] = "giS"; 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static const char kEmpty[] = ""; 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const struct MatchingEntryList { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::ModuleStatus expected_result; 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module test_case; 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::BlacklistEntry blacklist; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} kMatchineEntryList[] = { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Each BlacklistEntry is: 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Filename, location, desc_or_signer, version from, version to, help_tip. 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { // Matches: Name (location doesn't match) => Not enough for a match. 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 115868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kNoMatchLocation, kEmpty, kEmpty, kEmpty, kOs, 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name (location not given) => Suspected match. 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SUSPECTED_BAD, 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 120868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kEmpty, kEmpty, kEmpty, kEmpty, kOs, 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, not version (location not given) => Not a match. 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 125868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kEmpty, kEmpty, kVersionHigh, kVersionHigh, kOs, 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location => Suspected match. 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SUSPECTED_BAD, 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 130868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kEmpty, kEmpty, kEmpty, kOs, 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, (description not given) => Confirmed match. 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModuleNoDescription, // Note: No description. 135868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kEmpty, kEmpty, kEmpty, kOs, 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, (signature not given) => Confirmed match. 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModuleNoSignature, // Note: No signature. 140868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kEmpty, kEmpty, kEmpty, kOs, 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location (not version) => Not a match. 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 145868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kEmpty, kVersionHigh, kVersionLow, kOs, 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, signature => Confirmed match. 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 150868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, kEmpty, kEmpty, kOs, 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, signature (not version) => No match. 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, 156868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kVersionLow, kVersionLow, kOs, ModuleEnumerator::SEE_LINK } 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, description => Confirmed match. 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 160868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchDesc, kEmpty, kEmpty, kOs, 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::SEE_LINK } 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, description (not version) => No match. 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchDesc, 166868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kVersionHigh, kVersionHigh, kOs, ModuleEnumerator::SEE_LINK } 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, signature, version => Confirmed match. 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, 171868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kVersionLow, kVersionHigh, kOs, ModuleEnumerator::SEE_LINK } 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, signature, version (lower) => Confirmed. 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, 176868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kVersionLow, kEmpty, kOs, ModuleEnumerator::SEE_LINK } 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, location, signature, version (upper) => Confirmed. 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, 181868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) kEmpty, kVersionHigh, kOs, ModuleEnumerator::SEE_LINK } 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, Location, Version lower is inclusive => Confirmed. 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::CONFIRMED_BAD, 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kMatchSignature, 186868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) "1.0", "2.0", kOs, ModuleEnumerator::SEE_LINK } 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // Matches: Name, Location, Version higher is exclusive => No match. 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kStandardModule, 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kMatchName, kMatchLocation, kEmpty, 191868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) "0.0", "1.0", kOs, ModuleEnumerator::SEE_LINK } 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, { // All empty fields doesn't produce a match. 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NOT_MATCHED, 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { kType, kStatus, L"", L"", L"", L"", L""}, 195868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) { "a.dll", "", "", "", "", kOs, ModuleEnumerator::SEE_LINK } 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }, 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(EnumerateModulesTest, MatchFunction) { 2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kMatchineEntryList); ++i) { 2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module test = kMatchineEntryList[i].test_case; 2025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::NormalizeModule(&test); 2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::BlacklistEntry blacklist = 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kMatchineEntryList[i].blacklist; 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SCOPED_TRACE("Test case no " + base::IntToString(i) + 207a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) ": '" + base::UTF16ToASCII(test.name) + "'"); 2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kMatchineEntryList[i].expected_result, 2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Match(test, blacklist)); 2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const struct CollapsePathList { 214a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 expected_result; 215a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 test_case; 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} kCollapsePathList[] = { 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Negative testing (should not collapse this path). 2185d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { base::ASCIIToUTF16("c:\\a\\a.dll"), base::ASCIIToUTF16("c:\\a\\a.dll") }, 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // These two are to test that we select the maximum collapsed path. 2205d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { base::ASCIIToUTF16("%foo%\\a.dll"), base::ASCIIToUTF16("c:\\foo\\a.dll") }, 2215d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) { base::ASCIIToUTF16("%x%\\a.dll"), 2225d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) base::ASCIIToUTF16("c:\\foo\\bar\\a.dll") }, 2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(EnumerateModulesTest, CollapsePath) { 2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<ModuleEnumerator> module_enumerator(new ModuleEnumerator(NULL)); 2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) module_enumerator->path_mapping_.clear(); 2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) module_enumerator->path_mapping_.push_back( 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::make_pair(L"c:\\foo\\", L"%foo%")); 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) module_enumerator->path_mapping_.push_back( 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::make_pair(L"c:\\foo\\bar\\", L"%x%")); 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < arraysize(kCollapsePathList); ++i) { 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ModuleEnumerator::Module module; 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) module.location = kCollapsePathList[i].test_case; 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) module_enumerator->CollapsePath(&module); 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 238a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) SCOPED_TRACE("Test case no " + base::IntToString(i) + ": '" + 239a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) base::UTF16ToASCII(kCollapsePathList[i].expected_result) + 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "'"); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kCollapsePathList[i].expected_result, module.location); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 244