Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(63)

Side by Side Diff: mojo/edk/system/dispatcher.h

Issue 814543006: Move //mojo/{public, edk} underneath //third_party (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebase Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « mojo/edk/system/data_pipe_unittest.cc ('k') | mojo/edk/system/dispatcher.cc » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #ifndef MOJO_EDK_SYSTEM_DISPATCHER_H_
6 #define MOJO_EDK_SYSTEM_DISPATCHER_H_
7
8 #include <stddef.h>
9 #include <stdint.h>
10
11 #include <vector>
12
13 #include "base/macros.h"
14 #include "base/memory/ref_counted.h"
15 #include "base/memory/scoped_ptr.h"
16 #include "base/synchronization/lock.h"
17 #include "mojo/edk/embedder/platform_handle_vector.h"
18 #include "mojo/edk/system/handle_signals_state.h"
19 #include "mojo/edk/system/memory.h"
20 #include "mojo/edk/system/system_impl_export.h"
21 #include "mojo/public/c/system/buffer.h"
22 #include "mojo/public/c/system/data_pipe.h"
23 #include "mojo/public/c/system/message_pipe.h"
24 #include "mojo/public/c/system/types.h"
25
26 namespace mojo {
27
28 namespace embedder {
29 class PlatformSharedBufferMapping;
30 }
31
32 namespace system {
33
34 class Channel;
35 class Core;
36 class Dispatcher;
37 class DispatcherTransport;
38 class HandleTable;
39 class LocalMessagePipeEndpoint;
40 class ProxyMessagePipeEndpoint;
41 class TransportData;
42 class Awakable;
43
44 typedef std::vector<scoped_refptr<Dispatcher>> DispatcherVector;
45
46 namespace test {
47
48 // Test helper. We need to declare it here so we can friend it.
49 MOJO_SYSTEM_IMPL_EXPORT DispatcherTransport
50 DispatcherTryStartTransport(Dispatcher* dispatcher);
51
52 } // namespace test
53
54 // A |Dispatcher| implements Mojo primitives that are "attached" to a particular
55 // handle. This includes most (all?) primitives except for |MojoWait...()|. This
56 // object is thread-safe, with its state being protected by a single lock
57 // |lock_|, which is also made available to implementation subclasses (via the
58 // |lock()| method).
59 class MOJO_SYSTEM_IMPL_EXPORT Dispatcher
60 : public base::RefCountedThreadSafe<Dispatcher> {
61 public:
62 enum Type {
63 kTypeUnknown = 0,
64 kTypeMessagePipe,
65 kTypeDataPipeProducer,
66 kTypeDataPipeConsumer,
67 kTypeSharedBuffer,
68
69 // "Private" types (not exposed via the public interface):
70 kTypePlatformHandle = -1
71 };
72 virtual Type GetType() const = 0;
73
74 // These methods implement the various primitives named |Mojo...()|. These
75 // take |lock_| and handle races with |Close()|. Then they call out to
76 // subclasses' |...ImplNoLock()| methods (still under |lock_|), which actually
77 // implement the primitives.
78 // NOTE(vtl): This puts a big lock around each dispatcher (i.e., handle), and
79 // prevents the various |...ImplNoLock()|s from releasing the lock as soon as
80 // possible. If this becomes an issue, we can rethink this.
81 MojoResult Close();
82
83 // |transports| may be non-null if and only if there are handles to be
84 // written; not that |this| must not be in |transports|. On success, all the
85 // dispatchers in |transports| must have been moved to a closed state; on
86 // failure, they should remain in their original state.
87 MojoResult WriteMessage(UserPointer<const void> bytes,
88 uint32_t num_bytes,
89 std::vector<DispatcherTransport>* transports,
90 MojoWriteMessageFlags flags);
91 // |dispatchers| must be non-null but empty, if |num_dispatchers| is non-null
92 // and nonzero. On success, it will be set to the dispatchers to be received
93 // (and assigned handles) as part of the message.
94 MojoResult ReadMessage(UserPointer<void> bytes,
95 UserPointer<uint32_t> num_bytes,
96 DispatcherVector* dispatchers,
97 uint32_t* num_dispatchers,
98 MojoReadMessageFlags flags);
99 MojoResult WriteData(UserPointer<const void> elements,
100 UserPointer<uint32_t> elements_num_bytes,
101 MojoWriteDataFlags flags);
102 MojoResult BeginWriteData(UserPointer<void*> buffer,
103 UserPointer<uint32_t> buffer_num_bytes,
104 MojoWriteDataFlags flags);
105 MojoResult EndWriteData(uint32_t num_bytes_written);
106 MojoResult ReadData(UserPointer<void> elements,
107 UserPointer<uint32_t> num_bytes,
108 MojoReadDataFlags flags);
109 MojoResult BeginReadData(UserPointer<const void*> buffer,
110 UserPointer<uint32_t> buffer_num_bytes,
111 MojoReadDataFlags flags);
112 MojoResult EndReadData(uint32_t num_bytes_read);
113 // |options| may be null. |new_dispatcher| must not be null, but
114 // |*new_dispatcher| should be null (and will contain the dispatcher for the
115 // new handle on success).
116 MojoResult DuplicateBufferHandle(
117 UserPointer<const MojoDuplicateBufferHandleOptions> options,
118 scoped_refptr<Dispatcher>* new_dispatcher);
119 MojoResult MapBuffer(
120 uint64_t offset,
121 uint64_t num_bytes,
122 MojoMapBufferFlags flags,
123 scoped_ptr<embedder::PlatformSharedBufferMapping>* mapping);
124
125 // Gets the current handle signals state. (The default implementation simply
126 // returns a default-constructed |HandleSignalsState|, i.e., no signals
127 // satisfied or satisfiable.) Note: The state is subject to change from other
128 // threads.
129 HandleSignalsState GetHandleSignalsState() const;
130
131 // Adds an awakable to this dispatcher, which will be woken up when this
132 // object changes state to satisfy |signals| with context |context|. It will
133 // also be woken up when it becomes impossible for the object to ever satisfy
134 // |signals| with a suitable error status.
135 //
136 // If |signals_state| is non-null, on *failure* |*signals_state| will be set
137 // to the current handle signals state (on success, it is left untouched).
138 //
139 // Returns:
140 // - |MOJO_RESULT_OK| if the awakable was added;
141 // - |MOJO_RESULT_ALREADY_EXISTS| if |signals| is already satisfied;
142 // - |MOJO_RESULT_INVALID_ARGUMENT| if the dispatcher has been closed; and
143 // - |MOJO_RESULT_FAILED_PRECONDITION| if it is not (or no longer) possible
144 // that |signals| will ever be satisfied.
145 MojoResult AddAwakable(Awakable* awakable,
146 MojoHandleSignals signals,
147 uint32_t context,
148 HandleSignalsState* signals_state);
149 // Removes an awakable from this dispatcher. (It is valid to call this
150 // multiple times for the same |awakable| on the same object, so long as
151 // |AddAwakable()| was called at most once.) If |signals_state| is non-null,
152 // |*signals_state| will be set to the current handle signals state.
153 void RemoveAwakable(Awakable* awakable, HandleSignalsState* signals_state);
154
155 // A dispatcher must be put into a special state in order to be sent across a
156 // message pipe. Outside of tests, only |HandleTableAccess| is allowed to do
157 // this, since there are requirements on the handle table (see below).
158 //
159 // In this special state, only a restricted set of operations is allowed.
160 // These are the ones available as |DispatcherTransport| methods. Other
161 // |Dispatcher| methods must not be called until |DispatcherTransport::End()|
162 // has been called.
163 class HandleTableAccess {
164 private:
165 friend class Core;
166 friend class HandleTable;
167 // Tests also need this, to avoid needing |Core|.
168 friend DispatcherTransport test::DispatcherTryStartTransport(Dispatcher*);
169
170 // This must be called under the handle table lock and only if the handle
171 // table entry is not marked busy. The caller must maintain a reference to
172 // |dispatcher| until |DispatcherTransport::End()| is called.
173 static DispatcherTransport TryStartTransport(Dispatcher* dispatcher);
174 };
175
176 // A |TransportData| may serialize dispatchers that are given to it (and which
177 // were previously attached to the |MessageInTransit| that is creating it) to
178 // a given |Channel| and then (probably in a different process) deserialize.
179 // Note that the |MessageInTransit| "owns" (i.e., has the only ref to) these
180 // dispatchers, so there are no locking issues. (There's no lock ordering
181 // issue, and in fact no need to take dispatcher locks at all.)
182 // TODO(vtl): Consider making another wrapper similar to |DispatcherTransport|
183 // (but with an owning, unique reference), and having
184 // |CreateEquivalentDispatcherAndCloseImplNoLock()| return that wrapper (and
185 // |MessageInTransit|, etc. only holding on to such wrappers).
186 class TransportDataAccess {
187 private:
188 friend class TransportData;
189
190 // Serialization API. These functions may only be called on such
191 // dispatchers. (|channel| is the |Channel| to which the dispatcher is to be
192 // serialized.) See the |Dispatcher| methods of the same names for more
193 // details.
194 static void StartSerialize(Dispatcher* dispatcher,
195 Channel* channel,
196 size_t* max_size,
197 size_t* max_platform_handles);
198 static bool EndSerializeAndClose(
199 Dispatcher* dispatcher,
200 Channel* channel,
201 void* destination,
202 size_t* actual_size,
203 embedder::PlatformHandleVector* platform_handles);
204
205 // Deserialization API.
206 // Note: This "clears" (i.e., reset to the invalid handle) any platform
207 // handles that it takes ownership of.
208 static scoped_refptr<Dispatcher> Deserialize(
209 Channel* channel,
210 int32_t type,
211 const void* source,
212 size_t size,
213 embedder::PlatformHandleVector* platform_handles);
214 };
215
216 protected:
217 friend class base::RefCountedThreadSafe<Dispatcher>;
218
219 Dispatcher();
220 virtual ~Dispatcher();
221
222 // These are to be overridden by subclasses (if necessary). They are called
223 // exactly once -- first |CancelAllAwakablesNoLock()|, then
224 // |CloseImplNoLock()|,
225 // when the dispatcher is being closed. They are called under |lock_|.
226 virtual void CancelAllAwakablesNoLock();
227 virtual void CloseImplNoLock();
228 virtual scoped_refptr<Dispatcher>
229 CreateEquivalentDispatcherAndCloseImplNoLock() = 0;
230
231 // These are to be overridden by subclasses (if necessary). They are never
232 // called after the dispatcher has been closed. They are called under |lock_|.
233 // See the descriptions of the methods without the "ImplNoLock" for more
234 // information.
235 virtual MojoResult WriteMessageImplNoLock(
236 UserPointer<const void> bytes,
237 uint32_t num_bytes,
238 std::vector<DispatcherTransport>* transports,
239 MojoWriteMessageFlags flags);
240 virtual MojoResult ReadMessageImplNoLock(UserPointer<void> bytes,
241 UserPointer<uint32_t> num_bytes,
242 DispatcherVector* dispatchers,
243 uint32_t* num_dispatchers,
244 MojoReadMessageFlags flags);
245 virtual MojoResult WriteDataImplNoLock(UserPointer<const void> elements,
246 UserPointer<uint32_t> num_bytes,
247 MojoWriteDataFlags flags);
248 virtual MojoResult BeginWriteDataImplNoLock(
249 UserPointer<void*> buffer,
250 UserPointer<uint32_t> buffer_num_bytes,
251 MojoWriteDataFlags flags);
252 virtual MojoResult EndWriteDataImplNoLock(uint32_t num_bytes_written);
253 virtual MojoResult ReadDataImplNoLock(UserPointer<void> elements,
254 UserPointer<uint32_t> num_bytes,
255 MojoReadDataFlags flags);
256 virtual MojoResult BeginReadDataImplNoLock(
257 UserPointer<const void*> buffer,
258 UserPointer<uint32_t> buffer_num_bytes,
259 MojoReadDataFlags flags);
260 virtual MojoResult EndReadDataImplNoLock(uint32_t num_bytes_read);
261 virtual MojoResult DuplicateBufferHandleImplNoLock(
262 UserPointer<const MojoDuplicateBufferHandleOptions> options,
263 scoped_refptr<Dispatcher>* new_dispatcher);
264 virtual MojoResult MapBufferImplNoLock(
265 uint64_t offset,
266 uint64_t num_bytes,
267 MojoMapBufferFlags flags,
268 scoped_ptr<embedder::PlatformSharedBufferMapping>* mapping);
269 virtual HandleSignalsState GetHandleSignalsStateImplNoLock() const;
270 virtual MojoResult AddAwakableImplNoLock(Awakable* awakable,
271 MojoHandleSignals signals,
272 uint32_t context,
273 HandleSignalsState* signals_state);
274 virtual void RemoveAwakableImplNoLock(Awakable* awakable,
275 HandleSignalsState* signals_state);
276
277 // These implement the API used to serialize dispatchers to a |Channel|
278 // (described below). They will only be called on a dispatcher that's attached
279 // to and "owned" by a |MessageInTransit|. See the non-"impl" versions for
280 // more information.
281 //
282 // Note: |StartSerializeImplNoLock()| is actually called with |lock_| NOT
283 // held, since the dispatcher should only be accessible to the calling thread.
284 // On Debug builds, |EndSerializeAndCloseImplNoLock()| is called with |lock_|
285 // held, to satisfy any |lock_.AssertAcquired()| (e.g., in |CloseImplNoLock()|
286 // -- and anything it calls); disentangling those assertions is
287 // difficult/fragile, and would weaken our general checking of invariants.
288 //
289 // TODO(vtl): Consider making these pure virtual once most things support
290 // being passed over a message pipe.
291 virtual void StartSerializeImplNoLock(Channel* channel,
292 size_t* max_size,
293 size_t* max_platform_handles);
294 virtual bool EndSerializeAndCloseImplNoLock(
295 Channel* channel,
296 void* destination,
297 size_t* actual_size,
298 embedder::PlatformHandleVector* platform_handles);
299
300 // Available to subclasses. (Note: Returns a non-const reference, just like
301 // |base::AutoLock|'s constructor takes a non-const reference.)
302 base::Lock& lock() const { return lock_; }
303
304 private:
305 friend class DispatcherTransport;
306
307 // This should be overridden to return true if/when there's an ongoing
308 // operation (e.g., two-phase read/writes on data pipes) that should prevent a
309 // handle from being sent over a message pipe (with status "busy").
310 virtual bool IsBusyNoLock() const;
311
312 // Closes the dispatcher. This must be done under lock, and unlike |Close()|,
313 // the dispatcher must not be closed already. (This is the "equivalent" of
314 // |CreateEquivalentDispatcherAndCloseNoLock()|, for situations where the
315 // dispatcher must be disposed of instead of "transferred".)
316 void CloseNoLock();
317
318 // Creates an equivalent dispatcher -- representing the same resource as this
319 // dispatcher -- and close (i.e., disable) this dispatcher. I.e., this
320 // dispatcher will look as though it was closed, but the resource it
321 // represents will be assigned to the new dispatcher. This must be called
322 // under the dispatcher's lock.
323 scoped_refptr<Dispatcher> CreateEquivalentDispatcherAndCloseNoLock();
324
325 // API to serialize dispatchers to a |Channel|, exposed to only
326 // |TransportData| (via |TransportData|). They may only be called on a
327 // dispatcher attached to a |MessageInTransit| (and in particular not in
328 // |CoreImpl|'s handle table).
329 //
330 // Starts the serialization. Returns (via the two "out" parameters) the
331 // maximum amount of space that may be needed to serialize this dispatcher to
332 // the given |Channel| (no more than
333 // |TransportData::kMaxSerializedDispatcherSize|) and the maximum number of
334 // |PlatformHandle|s that may need to be attached (no more than
335 // |TransportData::kMaxSerializedDispatcherPlatformHandles|). If this
336 // dispatcher cannot be serialized to the given |Channel|, |*max_size| and
337 // |*max_platform_handles| should be set to zero. A call to this method will
338 // ALWAYS be followed by a call to |EndSerializeAndClose()| (even if this
339 // dispatcher cannot be serialized to the given |Channel|).
340 void StartSerialize(Channel* channel,
341 size_t* max_size,
342 size_t* max_platform_handles);
343 // Completes the serialization of this dispatcher to the given |Channel| and
344 // closes it. (This call will always follow an earlier call to
345 // |StartSerialize()|, with the same |Channel|.) This does so by writing to
346 // |destination| and appending any |PlatformHandle|s needed to
347 // |platform_handles| (which may be null if no platform handles were indicated
348 // to be required to |StartSerialize()|). This may write no more than the
349 // amount indicated by |StartSerialize()|. (WARNING: Beware of races, e.g., if
350 // something can be mutated between the two calls!) Returns true on success,
351 // in which case |*actual_size| is set to the amount it actually wrote to
352 // |destination|. On failure, |*actual_size| should not be modified; however,
353 // the dispatcher will still be closed.
354 bool EndSerializeAndClose(Channel* channel,
355 void* destination,
356 size_t* actual_size,
357 embedder::PlatformHandleVector* platform_handles);
358
359 // This protects the following members as well as any state added by
360 // subclasses.
361 mutable base::Lock lock_;
362 bool is_closed_;
363
364 DISALLOW_COPY_AND_ASSIGN(Dispatcher);
365 };
366
367 // Wrapper around a |Dispatcher| pointer, while it's being processed to be
368 // passed in a message pipe. See the comment about
369 // |Dispatcher::HandleTableAccess| for more details.
370 //
371 // Note: This class is deliberately "thin" -- no more expensive than a
372 // |Dispatcher*|.
373 class MOJO_SYSTEM_IMPL_EXPORT DispatcherTransport {
374 public:
375 DispatcherTransport() : dispatcher_(nullptr) {}
376
377 void End();
378
379 Dispatcher::Type GetType() const { return dispatcher_->GetType(); }
380 bool IsBusy() const { return dispatcher_->IsBusyNoLock(); }
381 void Close() { dispatcher_->CloseNoLock(); }
382 scoped_refptr<Dispatcher> CreateEquivalentDispatcherAndClose() {
383 return dispatcher_->CreateEquivalentDispatcherAndCloseNoLock();
384 }
385
386 bool is_valid() const { return !!dispatcher_; }
387
388 protected:
389 Dispatcher* dispatcher() { return dispatcher_; }
390
391 private:
392 friend class Dispatcher::HandleTableAccess;
393
394 explicit DispatcherTransport(Dispatcher* dispatcher)
395 : dispatcher_(dispatcher) {}
396
397 Dispatcher* dispatcher_;
398
399 // Copy and assign allowed.
400 };
401
402 } // namespace system
403 } // namespace mojo
404
405 #endif // MOJO_EDK_SYSTEM_DISPATCHER_H_
OLDNEW
« no previous file with comments | « mojo/edk/system/data_pipe_unittest.cc ('k') | mojo/edk/system/dispatcher.cc » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698