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
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "mojo/public/cpp/bindings/lib/control_message_proxy.h"
#include <stddef.h>
#include <stdint.h>
#include <utility>
#include "mojo/public/cpp/bindings/lib/message_builder.h"
#include "mojo/public/cpp/bindings/message.h"
#include "mojo/public/cpp/system/macros.h"
#include "mojo/public/interfaces/bindings/interface_control_messages.mojom.h"
namespace mojo {
namespace internal {
namespace {
using RunCallback = Callback<void(QueryVersionResultPtr)>;
class RunResponseForwardToCallback : public MessageReceiver {
public:
RunResponseForwardToCallback(const RunCallback& callback)
: callback_(callback) {}
bool Accept(Message* message) override;
private:
RunCallback callback_;
MOJO_DISALLOW_COPY_AND_ASSIGN(RunResponseForwardToCallback);
};
bool RunResponseForwardToCallback::Accept(Message* message) {
RunResponseMessageParams_Data* params =
reinterpret_cast<RunResponseMessageParams_Data*>(
message->mutable_payload());
params->DecodePointersAndHandles(message->mutable_handles());
RunResponseMessageParamsPtr params_ptr;
Deserialize_(params, ¶ms_ptr, nullptr);
callback_.Run(std::move(params_ptr->query_version_result));
return true;
}
void SendRunMessage(MessageReceiverWithResponder* receiver,
QueryVersionPtr query_version,
const RunCallback& callback) {
RunMessageParamsPtr params_ptr(RunMessageParams::New());
params_ptr->reserved0 = 16u;
params_ptr->reserved1 = 0u;
params_ptr->query_version = std::move(query_version);
size_t size = GetSerializedSize_(params_ptr);
RequestMessageBuilder builder(kRunMessageId, size);
RunMessageParams_Data* params = nullptr;
Serialize_(std::move(params_ptr), builder.buffer(), ¶ms);
params->EncodePointersAndHandles(builder.message()->mutable_handles());
MessageReceiver* responder = new RunResponseForwardToCallback(callback);
if (!receiver->AcceptWithResponder(builder.message(), responder))
delete responder;
}
void SendRunOrClosePipeMessage(MessageReceiverWithResponder* receiver,
RequireVersionPtr require_version) {
RunOrClosePipeMessageParamsPtr params_ptr(RunOrClosePipeMessageParams::New());
params_ptr->reserved0 = 16u;
params_ptr->reserved1 = 0u;
params_ptr->require_version = std::move(require_version);
size_t size = GetSerializedSize_(params_ptr);
MessageBuilder builder(kRunOrClosePipeMessageId, size);
RunOrClosePipeMessageParams_Data* params = nullptr;
Serialize_(std::move(params_ptr), builder.buffer(), ¶ms);
params->EncodePointersAndHandles(builder.message()->mutable_handles());
bool ok = receiver->Accept(builder.message());
MOJO_ALLOW_UNUSED_LOCAL(ok);
}
} // namespace
ControlMessageProxy::ControlMessageProxy(MessageReceiverWithResponder* receiver)
: receiver_(receiver) {
}
void ControlMessageProxy::QueryVersion(
const Callback<void(uint32_t)>& callback) {
auto run_callback = [callback](QueryVersionResultPtr query_version_result) {
callback.Run(query_version_result->version);
};
SendRunMessage(receiver_, QueryVersion::New(), run_callback);
}
void ControlMessageProxy::RequireVersion(uint32_t version) {
RequireVersionPtr require_version(RequireVersion::New());
require_version->version = version;
SendRunOrClosePipeMessage(receiver_, std::move(require_version));
}
} // namespace internal
} // namespace mojo
|