blob: ecfa018b5e02ba8712df8c17259f77c1c1f9355b (
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
|
// 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 <vector>
#include "base/memory/scoped_ptr.h"
#include "mojo/public/c/environment/async_waiter.h"
#include "mojo/public/cpp/system/core.h"
namespace IPC {
namespace internal {
// 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.
//
// MessageReader has to be used in IO thread. It isn't thread-safe.
//
class MessagePipeReader {
public:
// Delay the object deletion using the current message loop.
// This is intended to used by MessagePipeReader owners.
class DelayedDeleter {
public:
typedef base::DefaultDeleter<MessagePipeReader> DefaultType;
static void DeleteNow(MessagePipeReader* ptr) { delete ptr; }
DelayedDeleter() {}
DelayedDeleter(const DefaultType&) {}
DelayedDeleter& operator=(const DefaultType&) { return *this; }
void operator()(MessagePipeReader* ptr) const;
};
explicit MessagePipeReader(mojo::ScopedMessagePipeHandle handle);
virtual ~MessagePipeReader();
MojoHandle handle() const { return pipe_.get().value(); }
// Returns received bytes.
const std::vector<char>& data_buffer() const {
return data_buffer_;
}
// Delegate received handles ownership. The subclass should take the
// ownership over in its OnMessageReceived(). They will leak otherwise.
void TakeHandleBuffer(std::vector<MojoHandle>* handle_buffer) {
handle_buffer_.swap(*handle_buffer);
}
// Close and destroy the MessagePipe.
void Close();
// Close the mesage pipe with notifying the client with the error.
void CloseWithError(MojoResult error);
// Return true if the MessagePipe is alive.
bool IsValid() { return pipe_.is_valid(); }
//
// The client have to implment these callback to get the readiness
// event from the reader
//
virtual void OnMessageReceived() = 0;
virtual void OnPipeClosed() = 0;
virtual void OnPipeError(MojoResult error) = 0;
private:
static void InvokePipeIsReady(void* closure, MojoResult result);
MojoResult ReadMessageBytes();
void PipeIsReady(MojoResult wait_result);
void StartWaiting();
void StopWaiting();
std::vector<char> data_buffer_;
std::vector<MojoHandle> handle_buffer_;
MojoAsyncWaitID pipe_wait_id_;
mojo::ScopedMessagePipeHandle pipe_;
DISALLOW_COPY_AND_ASSIGN(MessagePipeReader);
};
} // namespace internal
} // namespace IPC
#endif // IPC_IPC_MESSAGE_PIPE_READER_H_
|