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.
5 #include "chrome/browser/media_galleries/fileapi/safe_audio_video_checker.h"
8 #include "base/callback.h"
9 #include "base/location.h"
10 #include "base/logging.h"
11 #include "base/process/process_handle.h"
12 #include "chrome/common/chrome_utility_messages.h"
13 #include "content/public/browser/child_process_data.h"
14 #include "content/public/browser/utility_process_host.h"
15 #include "content/public/browser/browser_thread.h"
16 #include "ipc/ipc_message_macros.h"
17 #include "ipc/ipc_platform_file.h"
19 SafeAudioVideoChecker::SafeAudioVideoChecker(
21 const fileapi::CopyOrMoveFileValidator::ResultCallback& callback)
22 : state_(INITIAL_STATE),
25 DCHECK(!callback.is_null());
28 void SafeAudioVideoChecker::Start() {
29 DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
30 if (state_ != INITIAL_STATE)
32 state_ = PINGED_STATE;
34 if (!file_.IsValid()) {
35 callback_.Run(base::File::FILE_ERROR_SECURITY);
36 state_ = FINISHED_STATE;
40 utility_process_host_ = content::UtilityProcessHost::Create(
41 this, base::MessageLoopProxy::current())->AsWeakPtr();
42 utility_process_host_->Send(new ChromeUtilityMsg_StartupPing);
45 SafeAudioVideoChecker::~SafeAudioVideoChecker() {}
47 void SafeAudioVideoChecker::OnProcessStarted() {
48 DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
49 if (state_ != PINGED_STATE)
51 state_ = STARTED_STATE;
53 if (utility_process_host_->GetData().handle == base::kNullProcessHandle)
54 DLOG(ERROR) << "Child process handle is null";
55 IPC::PlatformFileForTransit file_for_transit =
56 IPC::TakeFileHandleForProcess(file_.Pass(),
57 utility_process_host_->GetData().handle);
58 const int64 kFileDecodeTimeInMS = 250;
59 utility_process_host_->Send(new ChromeUtilityMsg_CheckMediaFile(
60 kFileDecodeTimeInMS, file_for_transit));
63 void SafeAudioVideoChecker::OnCheckingFinished(bool valid) {
64 DCHECK_CURRENTLY_ON(content::BrowserThread::IO);
65 if (state_ != STARTED_STATE)
67 state_ = FINISHED_STATE;
69 callback_.Run(valid ? base::File::FILE_OK :
70 base::File::FILE_ERROR_SECURITY);
73 void SafeAudioVideoChecker::OnProcessCrashed(int exit_code) {
74 OnCheckingFinished(false);
77 bool SafeAudioVideoChecker::OnMessageReceived(const IPC::Message& message) {
79 IPC_BEGIN_MESSAGE_MAP(SafeAudioVideoChecker, message)
80 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_ProcessStarted,
82 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_CheckMediaFile_Finished,
84 IPC_MESSAGE_UNHANDLED(handled = false)