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.
5 #ifndef MOJO_COMMON_MESSAGE_PUMP_MOJO_H_
6 #define MOJO_COMMON_MESSAGE_PUMP_MOJO_H_
10 #include "base/memory/scoped_ptr.h"
11 #include "base/message_loop/message_pump.h"
12 #include "base/synchronization/lock.h"
13 #include "base/time/time.h"
14 #include "mojo/common/mojo_common_export.h"
15 #include "mojo/public/cpp/system/core.h"
20 class MessagePumpMojoHandler;
22 // Mojo implementation of MessagePump.
23 class MOJO_COMMON_EXPORT MessagePumpMojo : public base::MessagePump {
26 virtual ~MessagePumpMojo();
28 // Static factory function (for using with |base::Thread::Options|, wrapped
29 // using |base::Bind()|).
30 static scoped_ptr<base::MessagePump> Create();
32 // Registers a MessagePumpMojoHandler for the specified handle. Only one
33 // handler can be registered for a specified handle.
34 void AddHandler(MessagePumpMojoHandler* handler,
36 MojoWaitFlags wait_flags,
37 base::TimeTicks deadline);
39 void RemoveHandler(const Handle& handle);
42 virtual void Run(Delegate* delegate) OVERRIDE;
43 virtual void Quit() OVERRIDE;
44 virtual void ScheduleWork() OVERRIDE;
45 virtual void ScheduleDelayedWork(
46 const base::TimeTicks& delayed_work_time) OVERRIDE;
52 // Contains the data needed to track a request to AddHandler().
54 Handler() : handler(NULL), wait_flags(MOJO_WAIT_FLAG_NONE), id(0) {}
56 MessagePumpMojoHandler* handler;
57 MojoWaitFlags wait_flags;
58 base::TimeTicks deadline;
59 // See description of |MessagePumpMojo::next_handler_id_| for details.
63 typedef std::map<Handle, Handler> HandleToHandler;
65 // Implementation of Run().
66 void DoRunLoop(RunState* run_state, Delegate* delegate);
68 // Services the set of handles ready. If |block| is true this waits for a
69 // handle to become ready, otherwise this does not block.
70 void DoInternalWork(const RunState& run_state, bool block);
72 // Removes the first invalid handle. This is called if MojoWaitMany finds an
74 void RemoveFirstInvalidHandle(const WaitState& wait_state);
76 void SignalControlPipe(const RunState& run_state);
78 WaitState GetWaitState(const RunState& run_state) const;
80 // Returns the deadline for the call to MojoWaitMany().
81 MojoDeadline GetDeadlineForWait(const RunState& run_state) const;
83 // If non-NULL we're running (inside Run()). Member is reference to value on
87 // Lock for accessing |run_state_|. In general the only method that we have to
88 // worry about is ScheduleWork(). All other methods are invoked on the same
90 base::Lock run_state_lock_;
92 HandleToHandler handlers_;
94 // An ever increasing value assigned to each Handler::id. Used to detect
95 // uniqueness while notifying. That is, while notifying expired timers we copy
96 // |handlers_| and only notify handlers whose id match. If the id does not
97 // match it means the handler was removed then added so that we shouldn't
101 DISALLOW_COPY_AND_ASSIGN(MessagePumpMojo);
104 } // namespace common
107 #endif // MOJO_COMMON_MESSAGE_PUMP_MOJO_H_