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
|
// Copyright 2013 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 MOJO_SYSTEM_MESSAGE_PIPE_DISPATCHER_H_
#define MOJO_SYSTEM_MESSAGE_PIPE_DISPATCHER_H_
#include "base/basictypes.h"
#include "base/compiler_specific.h"
#include "base/memory/ref_counted.h"
#include "mojo/system/dispatcher.h"
#include "mojo/system/system_impl_export.h"
namespace mojo {
namespace system {
class MessagePipe;
class MessagePipeDispatcherTransport;
// This is the |Dispatcher| implementation for message pipes (created by the
// Mojo primitive |MojoCreateMessagePipe()|). This class is thread-safe.
class MOJO_SYSTEM_IMPL_EXPORT MessagePipeDispatcher : public Dispatcher {
public:
MessagePipeDispatcher();
// Must be called before any other methods. (This method is not thread-safe.)
void Init(scoped_refptr<MessagePipe> message_pipe, unsigned port);
virtual Type GetType() const OVERRIDE;
private:
friend class MessagePipeDispatcherTransport;
friend class base::RefCountedThreadSafe<MessagePipeDispatcher>;
virtual ~MessagePipeDispatcher();
// Gets a dumb pointer to |message_pipe_|. This must be called under the
// |Dispatcher| lock (that it's a dumb pointer is okay since it's under lock).
// This is needed when sending handles across processes, where nontrivial,
// invasive work needs to be done.
MessagePipe* GetMessagePipeNoLock() const;
// Similarly for the port.
unsigned GetPortNoLock() const;
// |Dispatcher| implementation/overrides:
virtual void CancelAllWaitersNoLock() OVERRIDE;
virtual void CloseImplNoLock() OVERRIDE;
virtual scoped_refptr<Dispatcher>
CreateEquivalentDispatcherAndCloseImplNoLock() OVERRIDE;
virtual MojoResult WriteMessageImplNoLock(
const void* bytes,
uint32_t num_bytes,
std::vector<DispatcherTransport>* transports,
MojoWriteMessageFlags flags) OVERRIDE;
virtual MojoResult ReadMessageImplNoLock(
void* bytes,
uint32_t* num_bytes,
std::vector<scoped_refptr<Dispatcher> >* dispatchers,
uint32_t* num_dispatchers,
MojoReadMessageFlags flags) OVERRIDE;
virtual MojoResult AddWaiterImplNoLock(Waiter* waiter,
MojoWaitFlags flags,
MojoResult wake_result) OVERRIDE;
virtual void RemoveWaiterImplNoLock(Waiter* waiter) OVERRIDE;
// Protected by |lock()|:
scoped_refptr<MessagePipe> message_pipe_; // This will be null if closed.
unsigned port_;
DISALLOW_COPY_AND_ASSIGN(MessagePipeDispatcher);
};
class MessagePipeDispatcherTransport : public DispatcherTransport {
public:
explicit MessagePipeDispatcherTransport(DispatcherTransport transport);
MessagePipe* GetMessagePipe() {
return message_pipe_dispatcher()->GetMessagePipeNoLock();
}
unsigned GetPort() { return message_pipe_dispatcher()->GetPortNoLock(); }
private:
MessagePipeDispatcher* message_pipe_dispatcher() {
return static_cast<MessagePipeDispatcher*>(dispatcher());
}
// Copy and assign allowed.
};
} // namespace system
} // namespace mojo
#endif // MOJO_SYSTEM_MESSAGE_PIPE_DISPATCHER_H_
|