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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/js_injection/common/web_message_mojom_traits.h"
#include <string>
#include <variant>
#include "components/js_injection/common/interfaces.mojom.h"
#include "mojo/public/cpp/base/big_buffer.h"
#include "mojo/public/cpp/bindings/union_traits.h"
#include "third_party/abseil-cpp/absl/functional/overload.h"
#include "third_party/blink/public/common/messaging/string_message_codec.h"
namespace mojo {
// static
bool StructTraits<js_injection::mojom::JsWebMessageArrayBufferValueDataView,
std::unique_ptr<blink::WebMessageArrayBufferPayload>>::
Read(js_injection::mojom::JsWebMessageArrayBufferValueDataView r,
std::unique_ptr<blink::WebMessageArrayBufferPayload>* out) {
mojo_base::BigBufferView big_buffer_view;
if (!r.ReadArrayBufferValue(&big_buffer_view)) {
return false;
}
std::optional<size_t> max_byte_length;
if (r.is_resizable_by_user_javascript()) {
max_byte_length = base::checked_cast<size_t>(r.max_byte_length());
}
*out = blink::WebMessageArrayBufferPayload::CreateFromBigBuffer(
mojo_base::BigBufferView::ToBigBuffer(std::move(big_buffer_view)),
max_byte_length);
return true;
}
// static
js_injection::mojom::JsWebMessageDataView::Tag UnionTraits<
js_injection::mojom::JsWebMessageDataView,
blink::WebMessagePayload>::GetTag(const blink::WebMessagePayload& payload) {
return std::visit(
absl::Overload{
[](const std::u16string&) {
return js_injection::mojom::JsWebMessageDataView::Tag::kStringValue;
},
[](const std::unique_ptr<blink::WebMessageArrayBufferPayload>&) {
return js_injection::mojom::JsWebMessageDataView::Tag::
kArrayBufferValue;
}},
payload);
}
// static
bool UnionTraits<
js_injection::mojom::JsWebMessageDataView,
blink::WebMessagePayload>::Read(js_injection::mojom::JsWebMessageDataView r,
blink::WebMessagePayload* out) {
if (r.is_string_value()) {
std::u16string string_value;
if (!r.ReadStringValue(&string_value))
return false;
out->emplace<std::u16string>(std::move(string_value));
} else if (r.is_array_buffer_value()) {
std::unique_ptr<blink::WebMessageArrayBufferPayload> array_buffer_value;
if (!r.ReadArrayBufferValue(&array_buffer_value)) {
return false;
}
out->emplace<std::unique_ptr<blink::WebMessageArrayBufferPayload>>(
std::move(array_buffer_value));
} else {
return false;
}
return true;
}
} // namespace mojo
|
