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 |