| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 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 "components/viz/common/server_gpu_memory_buffer_manager.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "base/threading/sequenced_task_runner_handle.h" | |
| 9 #include "gpu/ipc/client/gpu_memory_buffer_impl.h" | |
| 10 #include "gpu/ipc/client/gpu_memory_buffer_impl_shared_memory.h" | |
| 11 #include "gpu/ipc/common/gpu_memory_buffer_support.h" | |
| 12 #include "services/ui/gpu/interfaces/gpu_service.mojom.h" | |
| 13 | |
| 14 namespace viz { | |
| 15 | |
| 16 ServerGpuMemoryBufferManager::ServerGpuMemoryBufferManager( | |
| 17 ui::mojom::GpuService* gpu_service, | |
| 18 int client_id) | |
| 19 : gpu_service_(gpu_service), | |
| 20 client_id_(client_id), | |
| 21 native_configurations_(gpu::GetNativeGpuMemoryBufferConfigurations()), | |
| 22 task_runner_(base::SequencedTaskRunnerHandle::Get()), | |
| 23 weak_factory_(this) {} | |
| 24 | |
| 25 ServerGpuMemoryBufferManager::~ServerGpuMemoryBufferManager() {} | |
| 26 | |
| 27 void ServerGpuMemoryBufferManager::AllocateGpuMemoryBuffer( | |
| 28 gfx::GpuMemoryBufferId id, | |
| 29 int client_id, | |
| 30 const gfx::Size& size, | |
| 31 gfx::BufferFormat format, | |
| 32 gfx::BufferUsage usage, | |
| 33 gpu::SurfaceHandle surface_handle, | |
| 34 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback) { | |
| 35 DCHECK(task_runner_->RunsTasksInCurrentSequence()); | |
| 36 if (gpu::GetNativeGpuMemoryBufferType() != gfx::EMPTY_BUFFER) { | |
| 37 const bool is_native = native_configurations_.find(std::make_pair( | |
| 38 format, usage)) != native_configurations_.end(); | |
| 39 if (is_native) { | |
| 40 pending_buffers_.insert(client_id); | |
| 41 gpu_service_->CreateGpuMemoryBuffer( | |
| 42 id, size, format, usage, client_id, surface_handle, | |
| 43 base::Bind(&ServerGpuMemoryBufferManager::OnGpuMemoryBufferAllocated, | |
| 44 weak_factory_.GetWeakPtr(), client_id, | |
| 45 base::Passed(std::move(callback)))); | |
| 46 return; | |
| 47 } | |
| 48 } | |
| 49 | |
| 50 gfx::GpuMemoryBufferHandle buffer_handle; | |
| 51 // The requests are coming in from untrusted clients. So verify that it is | |
| 52 // possible to allocate shared memory buffer first. | |
| 53 if (gpu::GpuMemoryBufferImplSharedMemory::IsUsageSupported(usage) && | |
| 54 gpu::GpuMemoryBufferImplSharedMemory::IsSizeValidForFormat(size, | |
| 55 format)) { | |
| 56 buffer_handle = gpu::GpuMemoryBufferImplSharedMemory::CreateGpuMemoryBuffer( | |
| 57 id, size, format); | |
| 58 } | |
| 59 | |
| 60 task_runner_->PostTask(FROM_HERE, | |
| 61 base::BindOnce(std::move(callback), buffer_handle)); | |
| 62 } | |
| 63 | |
| 64 std::unique_ptr<gfx::GpuMemoryBuffer> | |
| 65 ServerGpuMemoryBufferManager::CreateGpuMemoryBuffer( | |
| 66 const gfx::Size& size, | |
| 67 gfx::BufferFormat format, | |
| 68 gfx::BufferUsage usage, | |
| 69 gpu::SurfaceHandle surface_handle) { | |
| 70 gfx::GpuMemoryBufferId id(next_gpu_memory_id_++); | |
| 71 gfx::GpuMemoryBufferHandle handle; | |
| 72 base::WaitableEvent wait_event( | |
| 73 base::WaitableEvent::ResetPolicy::MANUAL, | |
| 74 base::WaitableEvent::InitialState::NOT_SIGNALED); | |
| 75 DCHECK(!task_runner_->RunsTasksInCurrentSequence()); | |
| 76 auto reply_callback = base::BindOnce( | |
| 77 [](gfx::GpuMemoryBufferHandle* handle, base::WaitableEvent* wait_event, | |
| 78 const gfx::GpuMemoryBufferHandle& allocated_buffer_handle) { | |
| 79 *handle = allocated_buffer_handle; | |
| 80 wait_event->Signal(); | |
| 81 }, | |
| 82 &handle, &wait_event); | |
| 83 // We block with a WaitableEvent until the callback is run. So using | |
| 84 // base::Unretained() is safe here. | |
| 85 auto allocate_callback = | |
| 86 base::BindOnce(&ServerGpuMemoryBufferManager::AllocateGpuMemoryBuffer, | |
| 87 base::Unretained(this), id, client_id_, size, format, | |
| 88 usage, surface_handle, std::move(reply_callback)); | |
| 89 task_runner_->PostTask(FROM_HERE, std::move(allocate_callback)); | |
| 90 base::ThreadRestrictions::ScopedAllowWait allow_wait; | |
| 91 wait_event.Wait(); | |
| 92 if (handle.is_null()) | |
| 93 return nullptr; | |
| 94 return gpu::GpuMemoryBufferImpl::CreateFromHandle( | |
| 95 handle, size, format, usage, | |
| 96 base::Bind(&ServerGpuMemoryBufferManager::DestroyGpuMemoryBuffer, | |
| 97 weak_factory_.GetWeakPtr(), id, client_id_)); | |
| 98 } | |
| 99 | |
| 100 void ServerGpuMemoryBufferManager::SetDestructionSyncToken( | |
| 101 gfx::GpuMemoryBuffer* buffer, | |
| 102 const gpu::SyncToken& sync_token) { | |
| 103 static_cast<gpu::GpuMemoryBufferImpl*>(buffer)->set_destruction_sync_token( | |
| 104 sync_token); | |
| 105 } | |
| 106 | |
| 107 void ServerGpuMemoryBufferManager::DestroyGpuMemoryBuffer( | |
| 108 gfx::GpuMemoryBufferId id, | |
| 109 int client_id, | |
| 110 const gpu::SyncToken& sync_token) { | |
| 111 DCHECK(task_runner_->RunsTasksInCurrentSequence()); | |
| 112 if (native_buffers_[client_id].erase(id)) | |
| 113 gpu_service_->DestroyGpuMemoryBuffer(id, client_id, sync_token); | |
| 114 } | |
| 115 | |
| 116 void ServerGpuMemoryBufferManager::DestroyAllGpuMemoryBufferForClient( | |
| 117 int client_id) { | |
| 118 DCHECK(task_runner_->RunsTasksInCurrentSequence()); | |
| 119 for (gfx::GpuMemoryBufferId id : native_buffers_[client_id]) | |
| 120 gpu_service_->DestroyGpuMemoryBuffer(id, client_id, gpu::SyncToken()); | |
| 121 native_buffers_.erase(client_id); | |
| 122 pending_buffers_.erase(client_id); | |
| 123 } | |
| 124 | |
| 125 void ServerGpuMemoryBufferManager::OnGpuMemoryBufferAllocated( | |
| 126 int client_id, | |
| 127 base::OnceCallback<void(const gfx::GpuMemoryBufferHandle&)> callback, | |
| 128 const gfx::GpuMemoryBufferHandle& handle) { | |
| 129 DCHECK(task_runner_->RunsTasksInCurrentSequence()); | |
| 130 if (pending_buffers_.find(client_id) == pending_buffers_.end()) { | |
| 131 // The client has been destroyed since the allocation request was made. | |
| 132 if (!handle.is_null()) { | |
| 133 gpu_service_->DestroyGpuMemoryBuffer(handle.id, client_id, | |
| 134 gpu::SyncToken()); | |
| 135 } | |
| 136 std::move(callback).Run(gfx::GpuMemoryBufferHandle()); | |
| 137 return; | |
| 138 } | |
| 139 if (!handle.is_null()) | |
| 140 native_buffers_[client_id].insert(handle.id); | |
| 141 std::move(callback).Run(handle); | |
| 142 } | |
| 143 | |
| 144 } // namespace viz | |
| OLD | NEW |