1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ui/surface/transport_dib.h"
#include <stddef.h>
#include <memory>
#include "base/logging.h"
#include "base/memory/ptr_util.h"
#include "base/memory/shared_memory_mapping.h"
#include "base/memory/unsafe_shared_memory_region.h"
#include "base/numerics/checked_math.h"
#include "build/build_config.h"
#include "skia/ext/platform_canvas.h"
TransportDIB::TransportDIB(base::UnsafeSharedMemoryRegion region)
: shm_region_(std::move(region)) {}
TransportDIB::~TransportDIB() = default;
// static
std::unique_ptr<TransportDIB> TransportDIB::Map(
base::UnsafeSharedMemoryRegion region) {
std::unique_ptr<TransportDIB> dib = CreateWithHandle(std::move(region));
if (!dib->Map())
return nullptr;
return dib;
}
// static
std::unique_ptr<TransportDIB> TransportDIB::CreateWithHandle(
base::UnsafeSharedMemoryRegion region) {
return base::WrapUnique(new TransportDIB(std::move(region)));
}
std::unique_ptr<SkCanvas> TransportDIB::GetPlatformCanvas(int w,
int h,
bool opaque) {
if (!shm_region_.IsValid())
return nullptr;
// Calculate the size for the memory region backing the canvas. If not valid
// then fail gracefully.
size_t canvas_size;
// 32-bit RGB data. A size_t causes a type change from int when multiplying.
const size_t bpp = 4;
if (!base::CheckMul(h, base::CheckMul(w, bpp)).AssignIfValid(&canvas_size))
return nullptr;
#if BUILDFLAG(IS_WIN)
// This DIB already mapped the file into this process, but PlatformCanvas
// will map it again.
DCHECK(!memory()) << "Mapped file twice in the same process.";
// We can't check the canvas size before mapping, but it's safe because
// Windows will fail to map the section if the dimensions of the canvas
// are too large.
std::unique_ptr<SkCanvas> canvas =
skia::CreatePlatformCanvasWithSharedSection(
w, h, opaque, shm_region_.GetPlatformHandle(),
skia::RETURN_NULL_ON_FAILURE);
if (canvas)
size_ = canvas_size;
return canvas;
#else
if ((!memory() && !Map()) || !VerifyCanvasSize(w, h))
return nullptr;
return skia::CreatePlatformCanvasWithPixels(w, h, opaque,
static_cast<uint8_t*>(memory()),
skia::RETURN_NULL_ON_FAILURE);
#endif
}
bool TransportDIB::Map() {
if (!shm_region_.IsValid())
return false;
if (memory())
return true;
shm_mapping_ = shm_region_.Map();
if (!shm_mapping_.IsValid()) {
PLOG(ERROR) << "Failed to map transport DIB";
return false;
}
size_ = shm_mapping_.size();
return true;
}
void* TransportDIB::memory() const {
return shm_mapping_.IsValid() ? shm_mapping_.memory() : nullptr;
}
base::UnsafeSharedMemoryRegion* TransportDIB::shared_memory_region() {
return &shm_region_;
}
// static
bool TransportDIB::VerifyCanvasSize(int w, int h) {
if (w <= 0 || h <= 0)
return false;
size_t canvas_size;
// 32-bit RGB data. A size_t causes a type change from int when multiplying.
const size_t bpp = 4;
if (!base::CheckMul(h, base::CheckMul(w, bpp)).AssignIfValid(&canvas_size))
return false;
return canvas_size <= size_;
}
|
