| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 "media/gpu/ipc/client/gpu_jpeg_decode_accelerator_host.h" | |
| 6 | |
| 7 #include <stddef.h> | |
| 8 | |
| 9 #include "base/bind.h" | |
| 10 #include "base/logging.h" | |
| 11 #include "base/macros.h" | |
| 12 #include "base/memory/shared_memory_handle.h" | |
| 13 #include "base/memory/weak_ptr.h" | |
| 14 #include "base/synchronization/waitable_event.h" | |
| 15 #include "build/build_config.h" | |
| 16 #include "gpu/ipc/client/gpu_channel_host.h" | |
| 17 #include "ipc/ipc_listener.h" | |
| 18 #include "ipc/ipc_message_macros.h" | |
| 19 #include "ipc/ipc_message_utils.h" | |
| 20 #include "media/gpu/ipc/common/media_messages.h" | |
| 21 | |
| 22 namespace media { | |
| 23 | |
| 24 // Class to receive AcceleratedJpegDecoderHostMsg_DecodeAck IPC message on IO | |
| 25 // thread. This does very similar what MessageFilter usually does. It is not | |
| 26 // MessageFilter because GpuChannelHost doesn't support AddFilter. | |
| 27 class GpuJpegDecodeAcceleratorHost::Receiver : public IPC::Listener { | |
| 28 public: | |
| 29 Receiver(Client* client, | |
| 30 const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner) | |
| 31 : client_(client), | |
| 32 io_task_runner_(io_task_runner), | |
| 33 weak_factory_for_io_( | |
| 34 base::MakeUnique<base::WeakPtrFactory<Receiver>>(this)), | |
| 35 weak_ptr_for_io_(weak_factory_for_io_->GetWeakPtr()) { | |
| 36 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 37 } | |
| 38 | |
| 39 ~Receiver() override { | |
| 40 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 41 // If |io_task_runner_| no longer accepts tasks, |weak_factory_for_io_| | |
| 42 // will leak. This is acceptable, because that should only happen on | |
| 43 // Browser shutdown. | |
| 44 io_task_runner_->DeleteSoon(FROM_HERE, weak_factory_for_io_.release()); | |
| 45 } | |
| 46 | |
| 47 void InvalidateWeakPtrOnIOThread(base::WaitableEvent* event) { | |
| 48 DCHECK(io_task_runner_->BelongsToCurrentThread()); | |
| 49 weak_factory_for_io_->InvalidateWeakPtrs(); | |
| 50 event->Signal(); | |
| 51 } | |
| 52 | |
| 53 // IPC::Listener implementation. | |
| 54 void OnChannelError() override { | |
| 55 DCHECK(io_task_runner_->BelongsToCurrentThread()); | |
| 56 | |
| 57 OnDecodeAck(kInvalidBitstreamBufferId, PLATFORM_FAILURE); | |
| 58 } | |
| 59 | |
| 60 bool OnMessageReceived(const IPC::Message& msg) override { | |
| 61 DCHECK(io_task_runner_->BelongsToCurrentThread()); | |
| 62 | |
| 63 bool handled = true; | |
| 64 IPC_BEGIN_MESSAGE_MAP(GpuJpegDecodeAcceleratorHost::Receiver, msg) | |
| 65 IPC_MESSAGE_HANDLER(AcceleratedJpegDecoderHostMsg_DecodeAck, OnDecodeAck) | |
| 66 IPC_MESSAGE_UNHANDLED(handled = false) | |
| 67 IPC_END_MESSAGE_MAP() | |
| 68 DCHECK(handled); | |
| 69 return handled; | |
| 70 } | |
| 71 | |
| 72 base::WeakPtr<IPC::Listener> AsWeakPtrForIO() { return weak_ptr_for_io_; } | |
| 73 | |
| 74 private: | |
| 75 void OnDecodeAck(int32_t bitstream_buffer_id, Error error) { | |
| 76 DCHECK(io_task_runner_->BelongsToCurrentThread()); | |
| 77 | |
| 78 if (!client_) | |
| 79 return; | |
| 80 | |
| 81 if (error == JpegDecodeAccelerator::NO_ERRORS) { | |
| 82 client_->VideoFrameReady(bitstream_buffer_id); | |
| 83 } else { | |
| 84 // Only NotifyError once. | |
| 85 // Client::NotifyError() may trigger deletion of |this| (on another | |
| 86 // thread), so calling it needs to be the last thing done on this stack! | |
| 87 JpegDecodeAccelerator::Client* client = nullptr; | |
| 88 std::swap(client, client_); | |
| 89 client->NotifyError(bitstream_buffer_id, error); | |
| 90 } | |
| 91 } | |
| 92 | |
| 93 Client* client_; | |
| 94 | |
| 95 // GPU IO task runner. | |
| 96 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | |
| 97 | |
| 98 SEQUENCE_CHECKER(sequence_checker_); | |
| 99 | |
| 100 // Weak pointers will be invalidated on IO thread. | |
| 101 std::unique_ptr<base::WeakPtrFactory<Receiver>> weak_factory_for_io_; | |
| 102 base::WeakPtr<Receiver> weak_ptr_for_io_; | |
| 103 | |
| 104 DISALLOW_COPY_AND_ASSIGN(Receiver); | |
| 105 }; | |
| 106 | |
| 107 GpuJpegDecodeAcceleratorHost::GpuJpegDecodeAcceleratorHost( | |
| 108 scoped_refptr<gpu::GpuChannelHost> channel, | |
| 109 int32_t route_id, | |
| 110 const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner) | |
| 111 : channel_(std::move(channel)), | |
| 112 decoder_route_id_(route_id), | |
| 113 io_task_runner_(io_task_runner) { | |
| 114 DCHECK(channel_); | |
| 115 DCHECK_NE(decoder_route_id_, MSG_ROUTING_NONE); | |
| 116 } | |
| 117 | |
| 118 GpuJpegDecodeAcceleratorHost::~GpuJpegDecodeAcceleratorHost() { | |
| 119 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 120 Send(new AcceleratedJpegDecoderMsg_Destroy(decoder_route_id_)); | |
| 121 | |
| 122 if (receiver_) { | |
| 123 channel_->RemoveRoute(decoder_route_id_); | |
| 124 | |
| 125 // Invalidate weak ptr of |receiver_|. After that, no more messages will be | |
| 126 // routed to |receiver_| on IO thread. | |
| 127 base::WaitableEvent event(base::WaitableEvent::ResetPolicy::AUTOMATIC, | |
| 128 base::WaitableEvent::InitialState::NOT_SIGNALED); | |
| 129 // Use of Unretained() is safe, because if the task executes, we block | |
| 130 // until it is finished by waiting on |event| below. | |
| 131 bool task_expected_to_run = io_task_runner_->PostTask( | |
| 132 FROM_HERE, base::Bind(&Receiver::InvalidateWeakPtrOnIOThread, | |
| 133 base::Unretained(receiver_.get()), | |
| 134 base::Unretained(&event))); | |
| 135 // If the current call is happening during the browser shutdown, the | |
| 136 // |io_task_runner_| may no longer be accepting tasks. | |
| 137 if (task_expected_to_run) | |
| 138 event.Wait(); | |
| 139 } | |
| 140 } | |
| 141 | |
| 142 bool GpuJpegDecodeAcceleratorHost::Initialize( | |
| 143 JpegDecodeAccelerator::Client* client) { | |
| 144 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 145 | |
| 146 bool succeeded = false; | |
| 147 // This cannot be on IO thread because the msg is synchronous. | |
| 148 Send(new GpuChannelMsg_CreateJpegDecoder(decoder_route_id_, &succeeded)); | |
| 149 | |
| 150 if (!succeeded) { | |
| 151 DLOG(ERROR) << "Send(GpuChannelMsg_CreateJpegDecoder()) failed"; | |
| 152 return false; | |
| 153 } | |
| 154 | |
| 155 receiver_.reset(new Receiver(client, io_task_runner_)); | |
| 156 | |
| 157 return true; | |
| 158 } | |
| 159 | |
| 160 void GpuJpegDecodeAcceleratorHost::Decode( | |
| 161 const BitstreamBuffer& bitstream_buffer, | |
| 162 const scoped_refptr<VideoFrame>& video_frame) { | |
| 163 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 164 | |
| 165 DCHECK( | |
| 166 base::SharedMemory::IsHandleValid(video_frame->shared_memory_handle())); | |
| 167 | |
| 168 AcceleratedJpegDecoderMsg_Decode_Params decode_params; | |
| 169 decode_params.input_buffer = bitstream_buffer; | |
| 170 base::SharedMemoryHandle input_handle = | |
| 171 channel_->ShareToGpuProcess(bitstream_buffer.handle()); | |
| 172 if (!base::SharedMemory::IsHandleValid(input_handle)) { | |
| 173 DLOG(ERROR) << "Failed to duplicate handle of BitstreamBuffer"; | |
| 174 return; | |
| 175 } | |
| 176 decode_params.input_buffer.set_handle(input_handle); | |
| 177 base::SharedMemoryHandle output_handle = | |
| 178 channel_->ShareToGpuProcess(video_frame->shared_memory_handle()); | |
| 179 if (!base::SharedMemory::IsHandleValid(output_handle)) { | |
| 180 DLOG(ERROR) << "Failed to duplicate handle of VideoFrame"; | |
| 181 #if defined(OS_POSIX) && !defined(OS_MACOSX) | |
| 182 if (input_handle.OwnershipPassesToIPC()) { | |
| 183 input_handle.Close(); | |
| 184 } | |
| 185 #else | |
| 186 // TODO(kcwu) fix the handle leak after crbug.com/493414 resolved. | |
| 187 #endif | |
| 188 return; | |
| 189 } | |
| 190 | |
| 191 size_t output_buffer_size = VideoFrame::AllocationSize( | |
| 192 video_frame->format(), video_frame->coded_size()); | |
| 193 | |
| 194 decode_params.coded_size = video_frame->coded_size(); | |
| 195 decode_params.output_video_frame_handle = output_handle; | |
| 196 decode_params.output_buffer_size = | |
| 197 base::checked_cast<uint32_t>(output_buffer_size); | |
| 198 Send(new AcceleratedJpegDecoderMsg_Decode(decoder_route_id_, decode_params)); | |
| 199 } | |
| 200 | |
| 201 bool GpuJpegDecodeAcceleratorHost::IsSupported() { | |
| 202 return channel_->gpu_info().jpeg_decode_accelerator_supported; | |
| 203 } | |
| 204 | |
| 205 void GpuJpegDecodeAcceleratorHost::Send(IPC::Message* message) { | |
| 206 DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | |
| 207 | |
| 208 if (!channel_->Send(message)) { | |
| 209 DLOG(ERROR) << "Send(" << message->type() << ") failed"; | |
| 210 } | |
| 211 } | |
| 212 | |
| 213 base::WeakPtr<IPC::Listener> GpuJpegDecodeAcceleratorHost::GetReceiver() { | |
| 214 return receiver_->AsWeakPtrForIO(); | |
| 215 } | |
| 216 | |
| 217 } // namespace media | |
| OLD | NEW |