blob: cd88149801465a10a1be4981905d3a91d10e2009 (
plain)
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
117
118
119
120
|
// Copyright 2014 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.
#ifndef IPC_IPC_MESSAGE_PIPE_READER_H_
#define IPC_IPC_MESSAGE_PIPE_READER_H_
#include <stdint.h>
#include <memory>
#include <vector>
#include "base/atomicops.h"
#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/process/process_handle.h"
#include "base/threading/thread_checker.h"
#include "ipc/ipc.mojom.h"
#include "ipc/ipc_export.h"
#include "ipc/ipc_message.h"
#include "mojo/public/cpp/bindings/associated_binding.h"
#include "mojo/public/cpp/bindings/scoped_interface_endpoint_handle.h"
#include "mojo/public/cpp/system/core.h"
#include "mojo/public/cpp/system/message_pipe.h"
namespace IPC {
namespace internal {
class AsyncHandleWaiter;
// A helper class to handle bytestream directly over mojo::MessagePipe
// in template-method pattern. MessagePipeReader manages the lifetime
// of given MessagePipe and participates the event loop, and
// read the stream and call the client when it is ready.
//
// Each client has to:
//
// * Provide a subclass implemenation of a specific use of a MessagePipe
// and implement callbacks.
// * Create the subclass instance with a MessagePipeHandle.
// The constructor automatically start listening on the pipe.
//
// All functions must be called on the IO thread, except for Send(), which can
// be called on any thread. All |Delegate| functions will be called on the IO
// thread.
//
class IPC_EXPORT MessagePipeReader : public NON_EXPORTED_BASE(mojom::Channel) {
public:
class Delegate {
public:
virtual void OnPeerPidReceived() = 0;
virtual void OnMessageReceived(const Message& message) = 0;
virtual void OnPipeError() = 0;
virtual void OnAssociatedInterfaceRequest(
const std::string& name,
mojo::ScopedInterfaceEndpointHandle handle) = 0;
};
// Builds a reader that reads messages from |receive_handle| and lets
// |delegate| know.
//
// |pipe| is the message pipe handle corresponding to the channel's master
// interface. This is the message pipe underlying both |sender| and
// |receiver|.
//
// Both |sender| and |receiver| must be non-null.
//
// Note that MessagePipeReader doesn't delete |delegate|.
MessagePipeReader(mojo::MessagePipeHandle pipe,
mojom::ChannelAssociatedPtr sender,
mojo::AssociatedInterfaceRequest<mojom::Channel> receiver,
Delegate* delegate);
~MessagePipeReader() override;
// Close and destroy the MessagePipe.
void Close();
// Return true if the MessagePipe is alive.
bool IsValid() { return sender_; }
// Sends an IPC::Message to the other end of the pipe. Safe to call from any
// thread.
bool Send(std::unique_ptr<Message> message);
// Requests an associated interface from the other end of the pipe.
void GetRemoteInterface(const std::string& name,
mojo::ScopedInterfaceEndpointHandle handle);
base::ProcessId GetPeerPid() const { return peer_pid_; }
mojom::Channel* sender() const { return sender_.get(); }
protected:
void OnPipeClosed();
void OnPipeError(MojoResult error);
private:
// mojom::Channel:
void SetPeerPid(int32_t peer_pid) override;
void Receive(
const std::vector<uint8_t>& data,
base::Optional<std::vector<mojom::SerializedHandlePtr>> handles) override;
void GetAssociatedInterface(
const std::string& name,
mojom::GenericInterfaceAssociatedRequest request) override;
// |delegate_| is null once the message pipe is closed.
Delegate* delegate_;
base::ProcessId peer_pid_ = base::kNullProcessId;
mojom::ChannelAssociatedPtr sender_;
mojo::AssociatedBinding<mojom::Channel> binding_;
base::ThreadChecker thread_checker_;
DISALLOW_COPY_AND_ASSIGN(MessagePipeReader);
};
} // namespace internal
} // namespace IPC
#endif // IPC_IPC_MESSAGE_PIPE_READER_H_
|
