Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(367)

Side by Side Diff: components/viz/common/server_gpu_memory_buffer_manager.cc

Issue 2941423002: viz: Move ServerGpuMemoryBufferManager into host/gpu. (Closed)
Patch Set: fix tot merge Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698