Index: chrome/browser/media_galleries/fileapi/scanning_file_validator.cc |
diff --git a/chrome/browser/media_galleries/fileapi/supported_image_type_validator.cc b/chrome/browser/media_galleries/fileapi/scanning_file_validator.cc |
similarity index 55% |
copy from chrome/browser/media_galleries/fileapi/supported_image_type_validator.cc |
copy to chrome/browser/media_galleries/fileapi/scanning_file_validator.cc |
index 9ae76253c3bb4db6d4584970202449ed06684405..e37c751b64210e54f5c8e71eb3503e83907a6369 100644 |
--- a/chrome/browser/media_galleries/fileapi/supported_image_type_validator.cc |
+++ b/chrome/browser/media_galleries/fileapi/scanning_file_validator.cc |
@@ -2,7 +2,7 @@ |
// 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/supported_image_type_validator.h" |
+#include "chrome/browser/media_galleries/fileapi/scanning_file_validator.h" |
#include "base/bind.h" |
#include "base/callback.h" |
@@ -13,9 +13,12 @@ |
#include "base/memory/weak_ptr.h" |
#include "base/stl_util.h" |
#include "base/threading/thread_restrictions.h" |
-#include "chrome/browser/image_decoder.h" |
#include "content/public/browser/browser_thread.h" |
+#if defined(OS_WIN) |
+#include "base/file_util.h" |
+#endif |
+ |
using content::BrowserThread; |
namespace chrome { |
@@ -51,57 +54,30 @@ scoped_ptr<std::string> ReadOnFileThread(const base::FilePath& path) { |
return result.Pass(); |
} |
-class ImageDecoderDelegateAdapter : public ImageDecoder::Delegate { |
- public: |
- ImageDecoderDelegateAdapter( |
- scoped_ptr<std::string> data, |
- const fileapi::CopyOrMoveFileValidator::ResultCallback& callback) |
- : data_(data.Pass()), |
- callback_(callback) { |
- DCHECK(data_); |
- } |
- |
- const std::string& data() { |
- return *data_; |
- } |
- |
- // ImageDecoder::Delegate methods. |
- virtual void OnImageDecoded(const ImageDecoder* /*decoder*/, |
- const SkBitmap& /*decoded_image*/) OVERRIDE { |
- callback_.Run(base::PLATFORM_FILE_OK); |
- delete this; |
- } |
- |
- virtual void OnDecodeImageFailed(const ImageDecoder* /*decoder*/) OVERRIDE { |
- callback_.Run(base::PLATFORM_FILE_ERROR_SECURITY); |
- delete this; |
- } |
- |
- private: |
- scoped_ptr<std::string> data_; |
- fileapi::CopyOrMoveFileValidator::ResultCallback callback_; |
+#if defined(OS_WIN) |
+base::PlatformFileError ScanFile( |
+ const base::FilePath& dest_platform_path) { |
+ DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); |
+ base::PlatformFileError result = base::PLATFORM_FILE_OK; |
+ |
+ // Execute an AV check of the transferred file. S_OK means |
+ // the file was OK. Other results may indicate the file was |
+ // deleted. Any other results, including E_FAIL and |
+ // INET_E_SECURITY_PROBLEM, are notified as security errors. |
+ HRESULT scan_result = |
+ base::ScanAndSaveDownloadedFile(dest_platform_path, std::string()); |
+ if (scan_result != S_OK) |
+ result = base::PLATFORM_FILE_ERROR_SECURITY; |
+ |
+ return result; |
+} |
+#endif |
- DISALLOW_COPY_AND_ASSIGN(ImageDecoderDelegateAdapter); |
-}; |
} // namespace |
SupportedImageTypeValidator::~SupportedImageTypeValidator() {} |
Lei Zhang
2013/08/05 22:42:40
This isn't right. Shouldn't it be ScanningFileVali
Greg Billock
2013/08/05 23:06:12
Yes. I'm updating it in the next patchset. This wa
|
-// static |
-bool SupportedImageTypeValidator::SupportsFileType(const base::FilePath& path) { |
- base::FilePath::StringType extension = path.Extension(); |
- return extension == FILE_PATH_LITERAL(".bmp") || |
- extension == FILE_PATH_LITERAL(".gif") || |
- extension == FILE_PATH_LITERAL(".jfif") || |
- extension == FILE_PATH_LITERAL(".jpeg") || |
- extension == FILE_PATH_LITERAL(".jpg") || |
- extension == FILE_PATH_LITERAL(".pjp") || |
- extension == FILE_PATH_LITERAL(".pjpeg") || |
- extension == FILE_PATH_LITERAL(".png") || |
- extension == FILE_PATH_LITERAL(".webp"); |
-} |
- |
void SupportedImageTypeValidator::StartPreWriteValidation( |
const ResultCallback& result_callback) { |
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
@@ -120,13 +96,19 @@ void SupportedImageTypeValidator::StartPostWriteValidation( |
const base::FilePath& dest_platform_path, |
const ResultCallback& result_callback) { |
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); |
- post_write_callback_ = result_callback; |
- // TODO(gbillock): Insert AV call here in the right validator. |
+#if defined(OS_WIN) |
+ BrowserThread::PostTaskAndReplyWithResult( |
+ BrowserThread::FILE, |
+ FROM_HERE, |
+ base::Bind(&ScanFile, dest_platform_path), |
+ result_callback); |
+#else |
BrowserThread::PostTask( |
BrowserThread::IO, |
FROM_HERE, |
- base::Bind(post_write_callback_, base::PLATFORM_FILE_OK)); |
+ base::Bind(result_callback, base::PLATFORM_FILE_OK)); |
+#endif |
} |
SupportedImageTypeValidator::SupportedImageTypeValidator( |
@@ -142,13 +124,10 @@ void SupportedImageTypeValidator::OnFileOpen(scoped_ptr<std::string> data) { |
return; |
} |
- // |adapter| will delete itself after a completion message is received. |
- ImageDecoderDelegateAdapter* adapter = |
- new ImageDecoderDelegateAdapter(data.Pass(), callback_); |
- decoder_ = new ImageDecoder(adapter, adapter->data(), |
- ImageDecoder::DEFAULT_CODEC); |
- decoder_->Start(content::BrowserThread::GetMessageLoopProxyForThread( |
- BrowserThread::IO)); |
+ BrowserThread::PostTask( |
+ BrowserThread::IO, |
+ FROM_HERE, |
+ base::Bind(callback_, base::PLATFORM_FILE_OK)); |
} |
} // namespace chrome |