| 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
|
|
|