1 files changed, 75 insertions, 0 deletions
diff --git a/chromium/extensions/browser/api/display_source/wifi_display/wifi_display_session_service_impl.h b/chromium/extensions/browser/api/display_source/wifi_display/wifi_display_session_service_impl.h
new file mode 100644
index 00000000000..6639871b116
--- /dev/null
+++ b/chromium/extensions/browser/api/display_source/wifi_display/wifi_display_session_service_impl.h
@@ -0,0 +1,75 @@
+// Copyright 2015 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 EXTENSIONS_BROWSER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_SESSION_SERVICE_IMPL_H_
+#define EXTENSIONS_BROWSER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_SESSION_SERVICE_IMPL_H_
+
+#include "components/keyed_service/content/browser_context_keyed_service_factory.h"
+#include "extensions/browser/api/display_source/display_source_connection_delegate.h"
+#include "extensions/common/mojo/wifi_display_session_service.mojom.h"
+#include "mojo/public/cpp/bindings/interface_request.h"
+#include "mojo/public/cpp/bindings/strong_binding.h"
+
+namespace content {
+class BrowserContext;
+}  // namespace content
+
+namespace extensions {
+
+// This class provides access to the network transport for the Wi-Fi Display
+// session (which is itself hosted in the sandboxed renderer process).
+class WiFiDisplaySessionServiceImpl
+    : public WiFiDisplaySessionService,
+      public DisplaySourceConnectionDelegate::Observer {
+ public:
+  ~WiFiDisplaySessionServiceImpl() override;
+  static void BindToRequest(
+      content::BrowserContext* context,
+      mojo::InterfaceRequest<WiFiDisplaySessionService> request);
+
+ private:
+  // WiFiDisplaySessionService overrides.
+  void SetClient(WiFiDisplaySessionServiceClientPtr client) override;
+  void Connect(int32_t sink_id,
+               int32_t auth_method,
+               const mojo::String& auth_data) override;
+  void Disconnect() override;
+  void SendMessage(const mojo::String& message) override;
+
+  // DisplaySourceConnectionDelegate::Observer overrides.
+  void OnSinksUpdated(const DisplaySourceSinkInfoList& sinks) override;
+  void OnConnectionError(int sink_id,
+                         DisplaySourceErrorType type,
+                         const std::string& description) override;
+
+  explicit WiFiDisplaySessionServiceImpl(
+      DisplaySourceConnectionDelegate* delegate,
+      mojo::InterfaceRequest<WiFiDisplaySessionService> request);
+
+  // Called if a message is received from the connected sink.
+  void OnSinkMessage(const std::string& message);
+
+  // Failure callbacks for Connect and Disconnect methods.
+  void OnConnectFailed(int sink_id, const std::string& reason);
+  void OnDisconnectFailed(int sink_id, const std::string& reason);
+
+  // Mojo error callback.
+  void OnClientConnectionError();
+
+  mojo::StrongBinding<WiFiDisplaySessionService> binding_;
+  WiFiDisplaySessionServiceClientPtr client_;
+  DisplaySourceConnectionDelegate* delegate_;
+
+  api::display_source::SinkState sink_state_;
+  // Id of the sink of the session this object is associated with.
+  int sink_id_;
+
+  base::WeakPtrFactory<WiFiDisplaySessionServiceImpl> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(WiFiDisplaySessionServiceImpl);
+};
+
+}  // namespace extensions
+
+#endif  // EXTENSIONS_BROWSER_API_DISPLAY_SOURCE_WIFI_DISPLAY_WIFI_DISPLAY_SESSION_SERVICE_IMPL_H_