11320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Copyright 2014 The Chromium Authors. All rights reserved. 21320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// Use of this source code is governed by a BSD-style license that can be 31320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci// found in the LICENSE file. 41320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 51320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome/browser/safe_browsing/incident_reporting/blacklist_load_analyzer.h" 61320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include <vector> 81320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 91320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_path.h" 101320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/path_service.h" 111320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/scoped_native_library.h" 121320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/strings/string_util.h" 131320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "chrome_elf/blacklist/blacklist.h" 141320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "testing/gtest/include/gtest/gtest.h" 151320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 161320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace { 171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucciconst wchar_t kTestDllName[] = L"blacklist_test_dll_1.dll"; 181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 201320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tuccinamespace safe_browsing { 211320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 221320f92c476a1ad9d19dba2a48c72b75566198e9Primiano TucciTEST(BlacklistLoadAnalyzer, TestBlacklistBypass) { 231320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::FilePath current_dir; 241320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ASSERT_TRUE(PathService::Get(base::DIR_EXE, ¤t_dir)); 251320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 261320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // Load test dll. 271320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::ScopedNativeLibrary dll1(current_dir.Append(kTestDllName)); 281320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 291320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // No blacklisted dlls should be found. 301320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci std::vector<base::string16> module_names; 311320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci EXPECT_TRUE(GetLoadedBlacklistedModules(&module_names)); 321320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci EXPECT_TRUE(module_names.empty()); 331320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 341320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci std::vector<base::string16>::const_iterator module_iter(module_names.begin()); 351320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci for (; module_iter != module_names.end(); ++module_iter) { 361320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci LOG(ERROR) << "Found blacklisted module: " << *module_iter; 371320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci } 381320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 391320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // Add test dll to blacklist 401320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci blacklist::AddDllToBlacklist(kTestDllName); 411320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 421320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci // Check that the test dll appears in list. 431320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci module_names.clear(); 441320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci EXPECT_TRUE(GetLoadedBlacklistedModules(&module_names)); 451320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ASSERT_EQ(1, module_names.size()); 461320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci EXPECT_STREQ(kTestDllName, 471320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::StringToLowerASCII( 481320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::FilePath(module_names[0]).BaseName().value()).c_str()); 491320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} 501320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 511320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci} // namespace safe_browsing 52