Index: chrome/browser/media_galleries/fileapi/av_scanning_file_validator.cc |
diff --git a/chrome/browser/media_galleries/fileapi/av_scanning_file_validator.cc b/chrome/browser/media_galleries/fileapi/av_scanning_file_validator.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a6c71e0c1fad1fe341d21174a0a422810518bfbe |
--- /dev/null |
+++ b/chrome/browser/media_galleries/fileapi/av_scanning_file_validator.cc |
@@ -0,0 +1,81 @@ |
+// Copyright 2013 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/browser/media_galleries/fileapi/av_scanning_file_validator.h" |
+ |
+#if defined(OS_WIN) |
+#include <windows.h> |
+#include <shlobj.h> |
+#endif |
+ |
+#include "base/bind.h" |
+#include "base/callback.h" |
+#include "base/location.h" |
+#include "base/logging.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "chrome/common/chrome_constants.h" |
+#include "content/public/browser/browser_thread.h" |
+ |
+#if defined(OS_WIN) |
+#include "base/win/scoped_comptr.h" |
+#endif |
+ |
+using content::BrowserThread; |
+ |
+namespace chrome { |
+ |
+namespace { |
+ |
+#if defined(OS_WIN) |
+base::PlatformFileError ScanFile( |
+ const base::FilePath& dest_platform_path) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
+ |
+ base::win::ScopedComPtr<IAttachmentExecute> attachment_services; |
+ HRESULT hr = attachment_services.CreateInstance(CLSID_AttachmentServices); |
+ |
+ if (FAILED(hr)) { |
+ // The thread must have COM initialized. |
+ DCHECK_NE(CO_E_NOTINITIALIZED, hr); |
+ return base::PLATFORM_FILE_ERROR_SECURITY; |
+ } |
+ |
+ hr = attachment_services->SetLocalPath(dest_platform_path.value().c_str()); |
+ if (FAILED(hr)) |
+ return base::PLATFORM_FILE_ERROR_SECURITY; |
+ |
+ // A failure in the Save() call below could result in the downloaded file |
+ // being deleted. |
+ HRESULT scan_result = attachment_services->Save(); |
+ if (scan_result == S_OK) |
+ return base::PLATFORM_FILE_OK; |
+ |
+ return base::PLATFORM_FILE_ERROR_SECURITY; |
+} |
+#endif |
+ |
+} // namespace |
+ |
+AVScanningFileValidator::~AVScanningFileValidator() {} |
+ |
+void AVScanningFileValidator::StartPostWriteValidation( |
+ const base::FilePath& dest_platform_path, |
+ const ResultCallback& result_callback) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
+ |
+#if defined(OS_WIN) |
+ BrowserThread::PostTaskAndReplyWithResult( |
+ BrowserThread::FILE, |
+ FROM_HERE, |
+ base::Bind(&ScanFile, dest_platform_path), |
+ result_callback); |
+#else |
+ result_callback.Run(base::PLATFORM_FILE_OK); |
+#endif |
+} |
+ |
+AVScanningFileValidator::AVScanningFileValidator() { |
+} |
+ |
+} // namespace chrome |