1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
// 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 {
#if defined(OS_WIN)
base::File::Error ScanFile(const base::FilePath& dest_platform_path) {
DCHECK_CURRENTLY_ON(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::File::FILE_ERROR_SECURITY;
}
hr = attachment_services->SetLocalPath(dest_platform_path.value().c_str());
if (FAILED(hr))
return base::File::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::File::FILE_OK;
return base::File::FILE_ERROR_SECURITY;
}
#endif
} // namespace
AVScanningFileValidator::~AVScanningFileValidator() {}
void AVScanningFileValidator::StartPostWriteValidation(
const base::FilePath& dest_platform_path,
const ResultCallback& result_callback) {
DCHECK_CURRENTLY_ON(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::File::FILE_OK);
#endif
}
AVScanningFileValidator::AVScanningFileValidator() {
}
|
