OLD | NEW |
(Empty) | |
| 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/safe_audio_video_checker.h" |
| 6 |
| 7 #include "base/bind.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" |
| 18 |
| 19 SafeAudioVideoChecker::SafeAudioVideoChecker( |
| 20 const base::PlatformFile& file, |
| 21 const fileapi::CopyOrMoveFileValidator::ResultCallback& callback) |
| 22 : state_(INITIAL_STATE), |
| 23 file_(file), |
| 24 file_closer_(&file_), |
| 25 callback_(callback) { |
| 26 DCHECK(!callback.is_null()); |
| 27 } |
| 28 |
| 29 void SafeAudioVideoChecker::Start() { |
| 30 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); |
| 31 if (state_ != INITIAL_STATE) |
| 32 return; |
| 33 state_ = PINGED_STATE; |
| 34 |
| 35 DCHECK(file_closer_); |
| 36 if (*file_closer_.get() == base::kInvalidPlatformFileValue) { |
| 37 callback_.Run(base::PLATFORM_FILE_ERROR_SECURITY); |
| 38 state_ = FINISHED_STATE; |
| 39 return; |
| 40 } |
| 41 |
| 42 utility_process_host_ = content::UtilityProcessHost::Create( |
| 43 this, base::MessageLoopProxy::current())->AsWeakPtr(); |
| 44 utility_process_host_->EnableZygote(); |
| 45 utility_process_host_->Send(new ChromeUtilityMsg_StartupPing); |
| 46 } |
| 47 |
| 48 SafeAudioVideoChecker::~SafeAudioVideoChecker() {} |
| 49 |
| 50 void SafeAudioVideoChecker::OnProcessStarted() { |
| 51 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); |
| 52 if (state_ != PINGED_STATE) |
| 53 return; |
| 54 state_ = STARTED_STATE; |
| 55 |
| 56 if (utility_process_host_->GetData().handle == base::kNullProcessHandle) |
| 57 DLOG(ERROR) << "Child process handle is null"; |
| 58 IPC::PlatformFileForTransit file_for_transit = |
| 59 IPC::GetFileHandleForProcess(*file_closer_.release(), |
| 60 utility_process_host_->GetData().handle, |
| 61 true /* close_source_handle */); |
| 62 const int64 kFileDecodeTimeInMS = 250; |
| 63 utility_process_host_->Send(new ChromeUtilityMsg_CheckMediaFile( |
| 64 kFileDecodeTimeInMS, file_for_transit)); |
| 65 } |
| 66 |
| 67 void SafeAudioVideoChecker::OnCheckingFinished(bool valid) { |
| 68 DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::IO)); |
| 69 if (state_ != STARTED_STATE) |
| 70 return; |
| 71 state_ = FINISHED_STATE; |
| 72 |
| 73 callback_.Run(valid ? base::PLATFORM_FILE_OK |
| 74 : base::PLATFORM_FILE_ERROR_SECURITY); |
| 75 } |
| 76 |
| 77 void SafeAudioVideoChecker::OnProcessCrashed(int exit_code) { |
| 78 OnCheckingFinished(false); |
| 79 } |
| 80 |
| 81 bool SafeAudioVideoChecker::OnMessageReceived(const IPC::Message& message) { |
| 82 bool handled = true; |
| 83 IPC_BEGIN_MESSAGE_MAP(SafeAudioVideoChecker, message) |
| 84 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_ProcessStarted, |
| 85 OnProcessStarted) |
| 86 IPC_MESSAGE_HANDLER(ChromeUtilityHostMsg_CheckMediaFile_Finished, |
| 87 OnCheckingFinished) |
| 88 IPC_MESSAGE_UNHANDLED(handled = false) |
| 89 IPC_END_MESSAGE_MAP() |
| 90 return handled; |
| 91 } |
OLD | NEW |