blob: c76110678c183b02da1b8ac0608358564eedc540 (
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
|
// 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_COMMON_MESSAGE_PUMP_MOJO_H_
#define MOJO_COMMON_MESSAGE_PUMP_MOJO_H_
#include <map>
#include "base/memory/scoped_ptr.h"
#include "base/message_loop/message_pump.h"
#include "base/synchronization/lock.h"
#include "base/time/time.h"
#include "mojo/common/mojo_common_export.h"
#include "mojo/public/cpp/system/core.h"
namespace mojo {
namespace common {
class MessagePumpMojoHandler;
// Mojo implementation of MessagePump.
class MOJO_COMMON_EXPORT MessagePumpMojo : public base::MessagePump {
public:
MessagePumpMojo();
virtual ~MessagePumpMojo();
// Static factory function (for using with |base::Thread::Options|, wrapped
// using |base::Bind()|).
static scoped_ptr<base::MessagePump> Create();
// Registers a MessagePumpMojoHandler for the specified handle. Only one
// handler can be registered for a specified handle.
void AddHandler(MessagePumpMojoHandler* handler,
const Handle& handle,
MojoWaitFlags wait_flags,
base::TimeTicks deadline);
void RemoveHandler(const Handle& handle);
// MessagePump:
virtual void Run(Delegate* delegate) OVERRIDE;
virtual void Quit() OVERRIDE;
virtual void ScheduleWork() OVERRIDE;
virtual void ScheduleDelayedWork(
const base::TimeTicks& delayed_work_time) OVERRIDE;
private:
struct RunState;
struct WaitState;
// Contains the data needed to track a request to AddHandler().
struct Handler {
Handler() : handler(NULL), wait_flags(MOJO_WAIT_FLAG_NONE), id(0) {}
MessagePumpMojoHandler* handler;
MojoWaitFlags wait_flags;
base::TimeTicks deadline;
// See description of |MessagePumpMojo::next_handler_id_| for details.
int id;
};
typedef std::map<Handle, Handler> HandleToHandler;
// Implementation of Run().
void DoRunLoop(RunState* run_state, Delegate* delegate);
// Services the set of handles ready. If |block| is true this waits for a
// handle to become ready, otherwise this does not block.
void DoInternalWork(const RunState& run_state, bool block);
// Removes the first invalid handle. This is called if MojoWaitMany finds an
// invalid handle.
void RemoveFirstInvalidHandle(const WaitState& wait_state);
void SignalControlPipe(const RunState& run_state);
WaitState GetWaitState(const RunState& run_state) const;
// Returns the deadline for the call to MojoWaitMany().
MojoDeadline GetDeadlineForWait(const RunState& run_state) const;
// If non-NULL we're running (inside Run()). Member is reference to value on
// stack.
RunState* run_state_;
// Lock for accessing |run_state_|. In general the only method that we have to
// worry about is ScheduleWork(). All other methods are invoked on the same
// thread.
base::Lock run_state_lock_;
HandleToHandler handlers_;
// An ever increasing value assigned to each Handler::id. Used to detect
// uniqueness while notifying. That is, while notifying expired timers we copy
// |handlers_| and only notify handlers whose id match. If the id does not
// match it means the handler was removed then added so that we shouldn't
// notify it.
int next_handler_id_;
DISALLOW_COPY_AND_ASSIGN(MessagePumpMojo);
};
} // namespace common
} // namespace mojo
#endif // MOJO_COMMON_MESSAGE_PUMP_MOJO_H_
|
