Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "content/public/common/quarantine.h" | 5 #include "content/public/common/quarantine.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 | 8 |
| 9 #include <cguid.h> | 9 #include <cguid.h> |
| 10 #include <objbase.h> | 10 #include <objbase.h> |
| 11 #include <shellapi.h> | 11 #include <shellapi.h> |
| 12 #include <shlobj.h> | 12 #include <shlobj.h> |
| 13 #include <shobjidl.h> | 13 #include <shobjidl.h> |
| 14 #include <wininet.h> | 14 #include <wininet.h> |
| 15 | 15 |
| 16 #include <vector> | 16 #include <vector> |
| 17 | 17 |
| 18 #include "base/files/file_util.h" | 18 #include "base/files/file_util.h" |
| 19 #include "base/guid.h" | 19 #include "base/guid.h" |
| 20 #include "base/logging.h" | 20 #include "base/logging.h" |
| 21 #include "base/macros.h" | 21 #include "base/macros.h" |
| 22 #include "base/metrics/histogram_macros.h" | 22 #include "base/metrics/histogram_macros.h" |
| 23 #include "base/metrics/sparse_histogram.h" | 23 #include "base/metrics/sparse_histogram.h" |
| 24 #include "base/strings/string_piece.h" | 24 #include "base/strings/string_piece.h" |
| 25 #include "base/strings/string_split.h" | 25 #include "base/strings/string_split.h" |
| 26 #include "base/strings/utf_string_conversions.h" | 26 #include "base/strings/utf_string_conversions.h" |
| 27 #include "base/threading/thread_restrictions.h" | 27 #include "base/threading/thread_restrictions.h" |
| 28 #include "base/win/scoped_com_initializer.h" | |
| 28 #include "base/win/scoped_comptr.h" | 29 #include "base/win/scoped_comptr.h" |
| 29 #include "base/win/scoped_handle.h" | 30 #include "base/win/scoped_handle.h" |
| 30 #include "url/gurl.h" | 31 #include "url/gurl.h" |
| 31 | 32 |
| 32 namespace content { | 33 namespace content { |
| 33 namespace { | 34 namespace { |
| 34 | 35 |
| 35 // [MS-FSCC] Section 5.6.1 | 36 // [MS-FSCC] Section 5.6.1 |
| 36 const base::FilePath::CharType kZoneIdentifierStreamSuffix[] = | 37 const base::FilePath::CharType kZoneIdentifierStreamSuffix[] = |
| 37 FILE_PATH_LITERAL(":Zone.Identifier"); | 38 FILE_PATH_LITERAL(":Zone.Identifier"); |
| (...skipping 257 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 295 } | 296 } |
| 296 | 297 |
| 297 } // namespace | 298 } // namespace |
| 298 | 299 |
| 299 QuarantineFileResult QuarantineFile(const base::FilePath& file, | 300 QuarantineFileResult QuarantineFile(const base::FilePath& file, |
| 300 const GURL& source_url, | 301 const GURL& source_url, |
| 301 const GURL& referrer_url, | 302 const GURL& referrer_url, |
| 302 const std::string& client_guid) { | 303 const std::string& client_guid) { |
| 303 base::ThreadRestrictions::AssertIOAllowed(); | 304 base::ThreadRestrictions::AssertIOAllowed(); |
| 304 | 305 |
| 306 // TODO(siggi): As this is a rare operation, this seems better than putting | |
|
Sigurður Ásgeirsson
2017/07/20 12:36:34
Gab: there are two places where downloading needs
robliao
2017/07/20 18:29:17
I was actually thinking about making ScopedCOMInit
Sigurður Ásgeirsson
2017/07/20 19:48:15
I'm not sure, but I'm pretty sure it's out of scop
robliao
2017/07/20 20:15:53
Dropping all of downloads in a COM STA sounds fine
| |
| 307 // all download activity on a COM STA sequence. | |
| 308 base::win::ScopedCOMInitializer com_initializer; | |
| 309 | |
| 305 int64_t file_size = 0; | 310 int64_t file_size = 0; |
| 306 if (!base::PathExists(file) || !base::GetFileSize(file, &file_size)) | 311 if (!base::PathExists(file) || !base::GetFileSize(file, &file_size)) |
| 307 return QuarantineFileResult::FILE_MISSING; | 312 return QuarantineFileResult::FILE_MISSING; |
| 308 | 313 |
| 309 std::string braces_guid = "{" + client_guid + "}"; | 314 std::string braces_guid = "{" + client_guid + "}"; |
| 310 GUID guid = GUID_NULL; | 315 GUID guid = GUID_NULL; |
| 311 if (base::IsValidGUID(client_guid)) { | 316 if (base::IsValidGUID(client_guid)) { |
| 312 HRESULT hr = CLSIDFromString(base::UTF8ToUTF16(braces_guid).c_str(), &guid); | 317 HRESULT hr = CLSIDFromString(base::UTF8ToUTF16(braces_guid).c_str(), &guid); |
| 313 if (FAILED(hr)) | 318 if (FAILED(hr)) |
| 314 guid = GUID_NULL; | 319 guid = GUID_NULL; |
| (...skipping 28 matching lines...) Expand all Loading... | |
| 343 return QuarantineFileResult::OK; | 348 return QuarantineFileResult::OK; |
| 344 } | 349 } |
| 345 | 350 |
| 346 bool IsFileQuarantined(const base::FilePath& file, | 351 bool IsFileQuarantined(const base::FilePath& file, |
| 347 const GURL& source_url, | 352 const GURL& source_url, |
| 348 const GURL& referrer_url) { | 353 const GURL& referrer_url) { |
| 349 return ZoneIdentifierPresentForFile(file); | 354 return ZoneIdentifierPresentForFile(file); |
| 350 } | 355 } |
| 351 | 356 |
| 352 } // namespace content | 357 } // namespace content |
| OLD | NEW |