Report and parse .img, .iso, and .smi as DMGs when downloading

chrome/common/safe_browsing/download_protection_util.cc

 // Copyright (c) 2012 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/common/safe_browsing/download_protection_util.h"
 
 #include <algorithm>
 
 #include "base/logging.h"
 #include "base/macros.h"
@@ skipping 250 matching lines @@
   EXTENSION_XAR,
   EXTENSION_XXE,
   EXTENSION_ZIPX,
   EXTENSION_ZPAQ,
   EXTENSION_RELS,
   EXTENSION_MSG,
   EXTENSION_EML,
   EXTENSION_RTF,
   EXTENSION_VHDX,
   EXTENSION_SEARCH_MS,
+  EXTENSION_IMG,
+  EXTENSION_SMI,
 
   // New values go above this one.
   EXTENSION_MAX
 };
 
 struct SafeBrowsingFiletype {
   const base::FilePath::CharType* const extension;
   int uma_value;
   bool is_supported_binary;
   bool is_archive;
@@ skipping 51 matching lines @@
     {FILE_PATH_LITERAL(".fxp"), EXTENSION_FXP, true, false},
     {FILE_PATH_LITERAL(".gadget"), EXTENSION_GADGET, true, false},
     {FILE_PATH_LITERAL(".grp"), EXTENSION_GRP, true, false},
     {FILE_PATH_LITERAL(".gz"), EXTENSION_GZ, true, true},
     {FILE_PATH_LITERAL(".gzip"), EXTENSION_GZIP, true, true},
     {FILE_PATH_LITERAL(".hfs"), EXTENSION_HFS, true, true},
     {FILE_PATH_LITERAL(".hlp"), EXTENSION_HLP, true, false},
     {FILE_PATH_LITERAL(".hqx"), EXTENSION_HQX, true, true},
     {FILE_PATH_LITERAL(".hta"), EXTENSION_HTA, true, false},
     {FILE_PATH_LITERAL(".htt"), EXTENSION_HTT, true, false},
+    {FILE_PATH_LITERAL(".img"), EXTENSION_IMG, true, false},
     {FILE_PATH_LITERAL(".inf"), EXTENSION_INF, true, false},
     {FILE_PATH_LITERAL(".ini"), EXTENSION_INI, true, false},
     {FILE_PATH_LITERAL(".ins"), EXTENSION_INS, true, false},
     {FILE_PATH_LITERAL(".inx"), EXTENSION_INX, false, false},  // UMA only.
     {FILE_PATH_LITERAL(".ipa"), EXTENSION_IPA, false, false},  // UMA only.
     {FILE_PATH_LITERAL(".iso"), EXTENSION_ISO, true, true},
     {FILE_PATH_LITERAL(".isp"), EXTENSION_ISP, true, false},
     {FILE_PATH_LITERAL(".isu"), EXTENSION_ISU, false, false},  // UMA only.
     {FILE_PATH_LITERAL(".jar"), EXTENSION_JAR, true, false},
     {FILE_PATH_LITERAL(".jnlp"), EXTENSION_JNLP, true, false},
@@ skipping 115 matching lines @@
     {FILE_PATH_LITERAL(".rtf"), EXTENSION_RTF, false, false},  // UMA only.
     {FILE_PATH_LITERAL(".scf"), EXTENSION_SCF, true, false},
     {FILE_PATH_LITERAL(".scr"), EXTENSION_SCR, true, false},
     {FILE_PATH_LITERAL(".sct"), EXTENSION_SCT, true, false},
     {FILE_PATH_LITERAL(".search-ms"), EXTENSION_SEARCH_MS, true, false},
     {FILE_PATH_LITERAL(".sh"), EXTENSION_SH, true, false},
     {FILE_PATH_LITERAL(".shar"), EXTENSION_SHAR, true, false},
     {FILE_PATH_LITERAL(".shb"), EXTENSION_SHB, true, false},
     {FILE_PATH_LITERAL(".shs"), EXTENSION_SHS, true, false},
     {FILE_PATH_LITERAL(".slp"), EXTENSION_SLP, true, true},
+    {FILE_PATH_LITERAL(".smi"), EXTENSION_SMI, true, false},
     {FILE_PATH_LITERAL(".spl"), EXTENSION_SPL, true, false},
     {FILE_PATH_LITERAL(".squashfs"), EXTENSION_SQUASHFS, true, true},
     {FILE_PATH_LITERAL(".svg"), EXTENSION_SVG, true, false},
     {FILE_PATH_LITERAL(".swf"), EXTENSION_SWF, true, false},
     {FILE_PATH_LITERAL(".swm"), EXTENSION_SWM, true, true},
     {FILE_PATH_LITERAL(".sys"), EXTENSION_SYS, true, false},
     {FILE_PATH_LITERAL(".tar"), EXTENSION_TAR, true, true},
     {FILE_PATH_LITERAL(".taz"), EXTENSION_TAZ, true, true},
     {FILE_PATH_LITERAL(".tbz"), EXTENSION_TBZ, true, true},
     {FILE_PATH_LITERAL(".tbz2"), EXTENSION_TBZ2, true, true},
@@ skipping 98 matching lines @@
     return ClientDownloadRequest::CHROME_EXTENSION;
   else if (file.MatchesExtension(FILE_PATH_LITERAL(".zip")))
     // DownloadProtectionService doesn't send a ClientDownloadRequest for ZIP
     // files unless they contain either executables or archives. The resulting
     // DownloadType is either ZIPPED_EXECUTABLE or ZIPPED_ARCHIVE respectively.
     // This function will return ZIPPED_EXECUTABLE for ZIP files as a
     // placeholder. The correct DownloadType will be determined based on the
     // result of analyzing the ZIP file.
     return ClientDownloadRequest::ZIPPED_EXECUTABLE;
   else if (file.MatchesExtension(FILE_PATH_LITERAL(".dmg")) ||
+           file.MatchesExtension(FILE_PATH_LITERAL(".img")) ||
+           file.MatchesExtension(FILE_PATH_LITERAL(".iso")) ||
            file.MatchesExtension(FILE_PATH_LITERAL(".pkg")) ||
+           file.MatchesExtension(FILE_PATH_LITERAL(".smi")) ||
            file.MatchesExtension(FILE_PATH_LITERAL(".osx")) ||
            file.MatchesExtension(FILE_PATH_LITERAL(".app")))
     return ClientDownloadRequest::MAC_EXECUTABLE;
   else if (IsArchiveFile(file))
     return ClientDownloadRequest::ARCHIVE;
   return ClientDownloadRequest::WIN_EXECUTABLE;
 }
 
 int GetSBClientDownloadExtensionValueForUMA(const base::FilePath& file) {
   return GetFileType(file).uma_value;
 }
 
 }  // namespace download_protection_util
 }  // namespace safe_browsing