1// Copyright 2013 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "chrome/browser/media_galleries/fileapi/av_scanning_file_validator.h" 6 7#if defined(OS_WIN) 8#include <windows.h> 9#include <shlobj.h> 10#endif 11 12#include "base/bind.h" 13#include "base/callback.h" 14#include "base/location.h" 15#include "base/logging.h" 16#include "base/strings/utf_string_conversions.h" 17#include "chrome/common/chrome_constants.h" 18#include "content/public/browser/browser_thread.h" 19 20#if defined(OS_WIN) 21#include "base/win/scoped_comptr.h" 22#endif 23 24using content::BrowserThread; 25 26namespace chrome { 27 28namespace { 29 30#if defined(OS_WIN) 31base::PlatformFileError ScanFile( 32 const base::FilePath& dest_platform_path) { 33 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 34 35 base::win::ScopedComPtr<IAttachmentExecute> attachment_services; 36 HRESULT hr = attachment_services.CreateInstance(CLSID_AttachmentServices); 37 38 if (FAILED(hr)) { 39 // The thread must have COM initialized. 40 DCHECK_NE(CO_E_NOTINITIALIZED, hr); 41 return base::PLATFORM_FILE_ERROR_SECURITY; 42 } 43 44 hr = attachment_services->SetLocalPath(dest_platform_path.value().c_str()); 45 if (FAILED(hr)) 46 return base::PLATFORM_FILE_ERROR_SECURITY; 47 48 // A failure in the Save() call below could result in the downloaded file 49 // being deleted. 50 HRESULT scan_result = attachment_services->Save(); 51 if (scan_result == S_OK) 52 return base::PLATFORM_FILE_OK; 53 54 return base::PLATFORM_FILE_ERROR_SECURITY; 55} 56#endif 57 58} // namespace 59 60AVScanningFileValidator::~AVScanningFileValidator() {} 61 62void AVScanningFileValidator::StartPostWriteValidation( 63 const base::FilePath& dest_platform_path, 64 const ResultCallback& result_callback) { 65 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO)); 66 67#if defined(OS_WIN) 68 BrowserThread::PostTaskAndReplyWithResult( 69 BrowserThread::FILE, 70 FROM_HERE, 71 base::Bind(&ScanFile, dest_platform_path), 72 result_callback); 73#else 74 result_callback.Run(base::PLATFORM_FILE_OK); 75#endif 76} 77 78AVScanningFileValidator::AVScanningFileValidator() { 79} 80 81} // namespace chrome 82