blob: 4eba660294630be80082c6c0fd5c558ca4b1ce66 [file] [log] [blame]
// 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.
#include <map>
#include "base/message_loop/message_pump.h"
#include "mojo/common/mojo_common_export.h"
#include "mojo/public/system/core.h"
namespace mojo {
namespace common {
class MessagePumpMojoHandler;
// Mojo implementation of MessagePump.
class MOJO_COMMON_EXPORT MessagePumpMojo : public base::MessagePump {
virtual ~MessagePumpMojo();
// Registers a MessagePumpMojoHandler for the specified handle. Only one
// handler can be registered for a specified handle. If there is an existing
// handler registered it is clobbered and silently removed.
// The handler is notified either when the handle is ready, or when it becomes
// invalid. If the handle becomes invalid the handler is removed and notified.
void AddHandler(MessagePumpMojoHandler* handler,
MojoHandle handle,
MojoWaitFlags wait_flags);
void RemoveHandler(MojoHandle 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;
struct RunState;
struct WaitState;
// Creates a MessagePumpMojoHandler and the set of MojoWaitFlags it was
// registered with.
struct Handler {
Handler() : handler(NULL), wait_flags(MOJO_WAIT_FLAG_NONE) {}
MessagePumpMojoHandler* handler;
MojoWaitFlags wait_flags;
typedef std::map<MojoHandle, Handler> HandleToHandler;
// 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(bool block);
// Removes the first invalid handle. This is called if MojoWaitMany finds an
// invalid handle.
void RemoveFirstInvalidHandle(const WaitState& wait_state);
void SignalControlPipe();
WaitState GetWaitState() const;
// If non-NULL we're running (inside Run()). Member is reference to value on
// stack.
RunState* run_state_;
HandleToHandler handlers_;
} // namespace common
} // namespace mojo