• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4 
5 #include "chrome/common/safe_browsing/download_protection_util.h"
6 
7 #include "base/files/file_path.h"
8 #include "base/logging.h"
9 
10 namespace safe_browsing {
11 namespace download_protection_util {
12 
IsArchiveFile(const base::FilePath & file)13 bool IsArchiveFile(const base::FilePath& file) {
14   return file.MatchesExtension(FILE_PATH_LITERAL(".zip"));
15 }
16 
IsBinaryFile(const base::FilePath & file)17 bool IsBinaryFile(const base::FilePath& file) {
18   return (
19       // Executable extensions for MS Windows.
20       file.MatchesExtension(FILE_PATH_LITERAL(".bas")) ||
21       file.MatchesExtension(FILE_PATH_LITERAL(".bat")) ||
22       file.MatchesExtension(FILE_PATH_LITERAL(".cab")) ||
23       file.MatchesExtension(FILE_PATH_LITERAL(".cmd")) ||
24       file.MatchesExtension(FILE_PATH_LITERAL(".com")) ||
25       file.MatchesExtension(FILE_PATH_LITERAL(".exe")) ||
26       file.MatchesExtension(FILE_PATH_LITERAL(".hta")) ||
27       file.MatchesExtension(FILE_PATH_LITERAL(".msi")) ||
28       file.MatchesExtension(FILE_PATH_LITERAL(".pif")) ||
29       file.MatchesExtension(FILE_PATH_LITERAL(".reg")) ||
30       file.MatchesExtension(FILE_PATH_LITERAL(".scr")) ||
31       file.MatchesExtension(FILE_PATH_LITERAL(".vb")) ||
32       file.MatchesExtension(FILE_PATH_LITERAL(".vbs")) ||
33       // Chrome extensions and android APKs are also reported.
34       file.MatchesExtension(FILE_PATH_LITERAL(".crx")) ||
35       file.MatchesExtension(FILE_PATH_LITERAL(".apk")) ||
36       // Archives _may_ contain binaries, we'll check in ExtractFileFeatures.
37       IsArchiveFile(file));
38 }
39 
GetDownloadType(const base::FilePath & file)40 ClientDownloadRequest::DownloadType GetDownloadType(
41     const base::FilePath& file) {
42   DCHECK(IsBinaryFile(file));
43   if (file.MatchesExtension(FILE_PATH_LITERAL(".apk")))
44     return ClientDownloadRequest::ANDROID_APK;
45   else if (file.MatchesExtension(FILE_PATH_LITERAL(".crx")))
46     return ClientDownloadRequest::CHROME_EXTENSION;
47   // For zip files, we use the ZIPPED_EXECUTABLE type since we will only send
48   // the pingback if we find an executable inside the zip archive.
49   else if (file.MatchesExtension(FILE_PATH_LITERAL(".zip")))
50     return ClientDownloadRequest::ZIPPED_EXECUTABLE;
51   return ClientDownloadRequest::WIN_EXECUTABLE;
52 }
53 
54 }  // namespace download_protection_util
55 }  // namespace safe_browsing
56