1 files changed, 81 insertions, 0 deletions
diff --git a/widget/nsDragServiceProxy.cpp b/widget/nsDragServiceProxy.cpp
new file mode 100644
index 000000000..7005a3e0b
--- /dev/null
+++ b/widget/nsDragServiceProxy.cpp
@@ -0,0 +1,81 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "nsDragServiceProxy.h"
+#include "nsIDocument.h"
+#include "nsISupportsPrimitives.h"
+#include "mozilla/dom/TabChild.h"
+#include "mozilla/gfx/2D.h"
+#include "mozilla/UniquePtr.h"
+#include "mozilla/Unused.h"
+#include "nsContentUtils.h"
+
+using mozilla::ipc::Shmem;
+using mozilla::dom::TabChild;
+using mozilla::dom::OptionalShmem;
+
+NS_IMPL_ISUPPORTS_INHERITED0(nsDragServiceProxy, nsBaseDragService)
+
+nsDragServiceProxy::nsDragServiceProxy()
+{
+}
+
+nsDragServiceProxy::~nsDragServiceProxy()
+{
+}
+
+nsresult
+nsDragServiceProxy::InvokeDragSessionImpl(nsIArray* aArrayTransferables,
+                                          nsIScriptableRegion* aRegion,
+                                          uint32_t aActionType)
+{
+  nsCOMPtr<nsIDocument> doc = do_QueryInterface(mSourceDocument);
+  NS_ENSURE_STATE(doc->GetDocShell());
+  TabChild* child = TabChild::GetFrom(doc->GetDocShell());
+  NS_ENSURE_STATE(child);
+  nsTArray<mozilla::dom::IPCDataTransfer> dataTransfers;
+  nsContentUtils::TransferablesToIPCTransferables(aArrayTransferables,
+                                                  dataTransfers,
+                                                  false,
+                                                  child->Manager(),
+                                                  nullptr);
+
+  LayoutDeviceIntRect dragRect;
+  if (mHasImage || mSelection) {
+    nsPresContext* pc;
+    RefPtr<mozilla::gfx::SourceSurface> surface;
+    DrawDrag(mSourceNode, aRegion, mScreenPosition, &dragRect, &surface, &pc);
+
+    if (surface) {
+      RefPtr<mozilla::gfx::DataSourceSurface> dataSurface =
+        surface->GetDataSurface();
+      if (dataSurface) {
+        size_t length;
+        int32_t stride;
+        Shmem surfaceData;
+        nsContentUtils::GetSurfaceData(dataSurface, &length, &stride, child,
+                                       &surfaceData);
+        // Save the surface data to shared memory.
+        if (!surfaceData.IsReadable() || !surfaceData.get<char>()) {
+          NS_WARNING("Failed to create shared memory for drag session.");
+          return NS_ERROR_FAILURE;
+        }
+
+        mozilla::Unused <<
+          child->SendInvokeDragSession(dataTransfers, aActionType, surfaceData,
+                                       stride, static_cast<uint8_t>(dataSurface->GetFormat()),
+                                       dragRect);
+        StartDragSession();
+        return NS_OK;
+      }
+    }
+  }
+
+  mozilla::Unused << child->SendInvokeDragSession(dataTransfers, aActionType,
+                                                  mozilla::void_t(), 0, 0, dragRect);
+  StartDragSession();
+  return NS_OK;
+}