ipc/mojo/ipc_mojo_bootstrap.h

   1 // Copyright 2014 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 IPC_MOJO_IPC_MOJO_BOOTSTRAP_H_
   6 #define IPC_MOJO_IPC_MOJO_BOOTSTRAP_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/process/process_handle.h"
  10 #include "ipc/ipc_channel.h"
  11 #include "ipc/ipc_listener.h"
  12 #include "third_party/mojo/src/mojo/edk/embedder/scoped_platform_handle.h"
  13
  14 namespace IPC {
  15
  16 // MojoBootstrap establishes a bootstrap pipe between two processes in
  17 // Chrome. It creates a native IPC::Channel first, then sends one
  18 // side of a newly created pipe to peer process. The pipe is intended
  19 // to be wrapped by Mojo MessagePipe.
  20 //
  21 // Clients should implement MojoBootstrapDelegate to get the pipe
  22 // from MojoBootstrap object. It should also tell the client process handle
  23 // using OnClientLaunched().
  24 //
  25 // This lives on IO thread other than Create(), which can be called from
  26 // UI thread as Channel::Create() can be.
  27 class IPC_MOJO_EXPORT MojoBootstrap : public Listener {
  28  public:
  29   class Delegate {
  30    public:
  31     virtual void OnPipeAvailable(
  32         mojo::embedder::ScopedPlatformHandle handle) = 0;
  33     virtual void OnBootstrapError() = 0;
  34   };
  35
  36   // Create the MojoBootstrap instance.
  37   // Instead of creating IPC::Channel, passs its ChannelHandle as |handle|,
  38   // mode as |mode|. The result is notified to passed |delegate|.
  39   static scoped_ptr<MojoBootstrap> Create(ChannelHandle handle,
  40                                           Channel::Mode mode,
  41                                           Delegate* delegate);
  42
  43   MojoBootstrap();
  44   ~MojoBootstrap() override;
  45
  46   // Start the handshake over the underlying platform channel.
  47   bool Connect();
  48
  49   // GetSelfPID returns the PID associated with |channel_|.
  50   base::ProcessId GetSelfPID() const;
  51
  52   // Each client should call this once the process handle becomes known.
  53   virtual void OnClientLaunched(base::ProcessHandle process) = 0;
  54
  55 #if defined(OS_POSIX) && !defined(OS_NACL)
  56   int GetClientFileDescriptor() const;
  57   base::ScopedFD TakeClientFileDescriptor();
  58 #endif  // defined(OS_POSIX) && !defined(OS_NACL)
  59
  60  protected:
  61   // On MojoServerBootstrap: INITIALIZED -> WAITING_ACK -> READY
  62   // On MojoClientBootstrap: INITIALIZED -> READY
  63   // STATE_ERROR is a catch-all state that captures any observed error.
  64   enum State { STATE_INITIALIZED, STATE_WAITING_ACK, STATE_READY, STATE_ERROR };
  65
  66   Delegate* delegate() const { return delegate_; }
  67   bool Send(Message* message);
  68   void Fail();
  69   bool HasFailed() const;
  70
  71   State state() const { return state_; }
  72   void set_state(State state) { state_ = state; }
  73
  74  private:
  75   void Init(scoped_ptr<Channel> channel, Delegate* delegate);
  76
  77   // Listener implementations
  78   void OnBadMessageReceived(const Message& message) override;
  79   void OnChannelError() override;
  80
  81   scoped_ptr<Channel> channel_;
  82   Delegate* delegate_;
  83   State state_;
  84
  85   DISALLOW_COPY_AND_ASSIGN(MojoBootstrap);
  86 };
  87
  88 }  // namespace IPC
  89
  90 #endif  // IPC_MOJO_IPC_MOJO_BOOTSTRAP_H_