Media Remoting: Add remoting control logic for encrypted contents.

media/remoting/remoting_source_impl.h

Index: media/remoting/remoting_source_impl.h
diff --git a/media/remoting/remoting_source_impl.h b/media/remoting/remoting_source_impl.h
new file mode 100644
index 0000000000000000000000000000000000000000..40ffa105aff8689324c7b970b379d153b6846a9c
--- /dev/null
+++ b/media/remoting/remoting_source_impl.h
@@ -0,0 +1,157 @@
+// Copyright 2016 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 MEDIA_REMOTING_REMOTING_SOURCE_IMPL_H_
+#define MEDIA_REMOTING_REMOTING_SOURCE_IMPL_H_
+
+#include <vector>
+#include "base/callback.h"
+#include "base/memory/weak_ptr.h"
+#include "base/threading/thread_checker.h"
+#include "media/mojo/interfaces/remoting.mojom.h"
+#include "media/remoting/rpc/rpc_broker.h"
+#include "mojo/public/cpp/bindings/binding.h"
+
+namespace media {
+
+// State transition diagram:
+//
+// .--> SESSION_UNAVAILABLE
+// |          |      ^
+// |          V      |
+// |     SESSION_CAN_START
+// |          |
+// |          V
+// | .---SESSION_STARTING --.
+// | |        |             |
+// | |        V             |
+// | |   SESSION_STARTED----|
+// | |        |             |
+// | |        V             |
+// | '-> SESSION_STOPPING   |
+// '-----'    |             |
+//            V             V
+//    SESSION_PERMANENTLY_STOPPED
+
+enum RemotingSessionState {
+  // Remoting sink is not available. Can't start remoting.
+  SESSION_UNAVAILABLE,
+  // Remoting sink is available, Can start remoting.
+  SESSION_CAN_START,
+  // Starting a remoting session.
+  SESSION_STARTING,
+  // Remoting session is successively started.
+  SESSION_STARTED,
+  // Stopping the session.
+  SESSION_STOPPING,
+  // Remoting session is permanently stopped. This state indicates that the
+  // video stack cannot continue operation. For example, if a remoting session
+  // involving CDM content was stopped, there is no way to continue playback
+  // because the CDM is required but is no longer available.
+  SESSION_PERMANENTLY_STOPPED,
+};
+
+// Maintains a single remoting session for multiple clients. The session will
+// start remoting when receiving the first request. Once remoting is started,
+// it will be stopped when any of the following happens:
+// 1) Receives the request from any client to stop remoting.
+// 2) Remote sink is gone.
+// 3) Any client requests to permanently terminate the session.
+// 4) All clients are destroyed.
+//
+// This class is ref-counted because, in some cases, an instance will have
+// shared ownership between RemotingRendererController and
+// RemotingCdmController.
+class RemotingSourceImpl final
+    : public mojom::RemotingSource,
+      public base::RefCountedThreadSafe<RemotingSourceImpl> {
+ public:
+  class Client {
+   public:
+    // Get notified whether the remoting session is successively started.
+    virtual void OnStarted(bool success) = 0;
+    // Get notified when session state changes.
+    virtual void OnSessionStateChanged() = 0;
+  };
+
+  RemotingSourceImpl(mojom::RemotingSourceRequest source_request,
+                     mojom::RemoterPtr remoter);
+
+  // Get the current session state.
+  RemotingSessionState state() const {
+    DCHECK(thread_checker_.CalledOnValidThread());
+    return state_;
+  }
+
+  // RemotingSource implementations.
+  void OnSinkAvailable() override;
+  void OnSinkGone() override;
+  void OnStarted() override;
+  void OnStartFailed(mojom::RemotingStartFailReason reason) override;
+  void OnMessageFromSink(const std::vector<uint8_t>& message) override;
+  void OnStopped(mojom::RemotingStopReason reason) override;
+
+  using DataPipeStartCallback =
+      base::Callback<void(mojom::RemotingDataStreamSenderPtrInfo audio,
+                          mojom::RemotingDataStreamSenderPtrInfo video,
+                          mojo::ScopedDataPipeProducerHandle audio_handle,
+                          mojo::ScopedDataPipeProducerHandle video_handle)>;
+  void StartDataPipe(std::unique_ptr<mojo::DataPipe> audio_data_pipe,
+                     std::unique_ptr<mojo::DataPipe> video_data_pipe,
+                     const DataPipeStartCallback& done_callback);
+
+  // Requests to start remoting. Will try start a remoting session if not
+  // started yet. |client| will get informed whether the session is
+  // successifully started throught OnStarted().
+  void StartRemoting(Client* client);
+
+  // Requests to stop the current remoting session if started. When the session
+  // is stopping, all clients will get notified.
+  void StopRemoting(Client* client);
+
+  // Permanently terminates the current remoting session.
+  void Shutdown();
+
+  // Add/remove a client to/from |clients_|.
+  // Note: Clients can only added/removed through these methods.
+  // Remoting session will be stopped if all clients are gone.
+  void AddClient(Client* client);
+  void RemoveClient(Client* client);
+
+  remoting::RpcBroker* GetRpcBroker() const;
+
+ private:
+  friend class base::RefCountedThreadSafe<RemotingSourceImpl>;
+  ~RemotingSourceImpl() override;
+
+  // Updates the current session state and notifies all the clients if state
+  // changes.
+  void UpdateAndNotifyState(RemotingSessionState state);
+
+  // Callback from RpcBroker when sending message to remote sink.
+  void SendMessageToSink(std::unique_ptr<std::vector<uint8_t>> message);
+
+  // TODO(xjz): Might merge RpcBroker into RemotingSourceImpl.
+  // Handle incomging and outgoing RPC message.
+  remoting::RpcBroker rpc_broker_;
+
+  const mojo::Binding<mojom::RemotingSource> binding_;
+  const mojom::RemoterPtr remoter_;
+
+  // The current state.
+  RemotingSessionState state_ = RemotingSessionState::SESSION_UNAVAILABLE;
+
+  // Clients are added/removed to/from this list by calling Add/RemoveClient().
+  // All the clients are not belong to this class. They are supposed to call
+  // RemoveClient() before they are gone.
+  std::vector<Client*> clients_;
+
+  // This is used to check all the methods are called on the current thread in
+  // debug builds.
+  base::ThreadChecker thread_checker_;
+};
+
+}  // namespace media
+
+#endif  // MEDIA_REMOTING_REMOTING_SOURCE_IMPL_H_