Implementation of MessagePump for Mojo.

mojo/common/message_pump_mojo.h

+// 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/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 {
+ public:
+  MessagePumpMojo();
+  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;
+
+ private:
+  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_;
+
+  DISALLOW_COPY_AND_ASSIGN(MessagePumpMojo);
+};
+
+}  // namespace common
+}  // namespace mojo
+
+#endif  // MOJO_COMMON_MESSAGE_PUMP_MOJO_H_