DLP: Add source for drag-and-drop data

- Added source for drag-and-drop data the
same as clipboard data to be able to apply
the rules of data leak prevention policy.

Bug: 1139886
Change-Id: I321c9ab7527cb1b16561973b1e57420e5a4b9e36
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2489634Reviewed-by: Sadrul Chowdhury <sadrul@chromium.org>
Reviewed-by: Darwin Huang <huangdarwin@chromium.org>
Commit-Queue: Aya Elsayed <ayaelattar@chromium.org>
Cr-Commit-Position: refs/heads/master@{#822163}

ui/base/dragdrop/os_exchange_data.cc

@@ -10,6 +10,7 @@
 #include "base/callback.h"
 #include "base/pickle.h"
 #include "ui/base/clipboard/clipboard_format_type.h"
+#include "ui/base/data_transfer_policy/data_transfer_endpoint.h"
 #include "ui/base/dragdrop/os_exchange_data_provider_factory.h"
 #include "url/gurl.h"
 
@@ -160,4 +161,13 @@ bool OSExchangeData::GetHtml(base::string16* html, GURL* base_url) const {
 }
 #endif
 
+void OSExchangeData::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {
+  provider_->SetSource(std::move(data_source));
+}
+
+DataTransferEndpoint* OSExchangeData::GetSource() const {
+  return provider_->GetSource();
+}
+
 }  // namespace ui

ui/base/dragdrop/os_exchange_data.h

@@ -32,6 +32,7 @@ class Pickle;
 namespace ui {
 
 class ClipboardFormatType;
+class DataTransferEndpoint;
 struct FileInfo;
 
 ///////////////////////////////////////////////////////////////////////////////
@@ -186,6 +187,12 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeData {
   bool HasHtml() const;
 #endif
 
+  // Adds a DataTransferEndpoint to represent the source of the data.
+  // TODO(crbug.com/1142406): Update all drag-and-drop references to set the
+  // source of the data.
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source);
+  DataTransferEndpoint* GetSource() const;
+
  private:
   // Provides the actual data.
   std::unique_ptr<OSExchangeDataProvider> provider_;

ui/base/dragdrop/os_exchange_data_provider.h

@@ -31,6 +31,8 @@
 
 namespace ui {
 
+class DataTransferEndpoint;
+
 // Controls whether or not filenames should be converted to file: URLs when
 // getting a URL.
 enum class FilenameToURLPolicy {
@@ -105,6 +107,10 @@ class COMPONENT_EXPORT(UI_BASE_DATA_EXCHANGE) OSExchangeDataProvider {
   virtual gfx::ImageSkia GetDragImage() const = 0;
   virtual gfx::Vector2d GetDragImageOffset() const = 0;
 #endif
+
+  // These functions are only implemented on Chrome OS currently.
+  virtual void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) = 0;
+  virtual DataTransferEndpoint* GetSource() const = 0;
 };
 
 }  // namespace ui

ui/base/dragdrop/os_exchange_data_provider_mac.h

@@ -76,6 +76,9 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderMac
   // OSExchangeData.
   static NSArray* SupportedPasteboardTypes();
 
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
+  DataTransferEndpoint* GetSource() const override;
+
  protected:
   OSExchangeDataProviderMac();
   OSExchangeDataProviderMac(const OSExchangeDataProviderMac&);

ui/base/dragdrop/os_exchange_data_provider_mac.mm

@@ -17,6 +17,7 @@
 #include "ui/base/clipboard/clipboard_format_type.h"
 #import "ui/base/clipboard/clipboard_util_mac.h"
 #include "ui/base/clipboard/custom_data_helper.h"
+#include "ui/base/data_transfer_policy/data_transfer_policy_controller.h"
 #import "ui/base/dragdrop/cocoa_dnd_util.h"
 #include "ui/base/dragdrop/file_info/file_info.h"
 #include "url/gurl.h"
@@ -337,4 +338,11 @@ NSArray* OSExchangeDataProviderMac::SupportedPasteboardTypes() {
   ];
 }
 
+void OSExchangeDataProviderMac::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {}
+
+DataTransferEndpoint* OSExchangeDataProviderMac::GetSource() const {
+  return nullptr;
+}
+
 }  // namespace ui

ui/base/dragdrop/os_exchange_data_provider_non_backed.cc

@@ -12,6 +12,7 @@
 #include "base/strings/utf_string_conversions.h"
 #include "net/base/filename_util.h"
 #include "ui/base/clipboard/clipboard_format_type.h"
+#include "ui/base/data_transfer_policy/data_transfer_endpoint.h"
 #include "ui/base/dragdrop/file_info/file_info.h"
 #include "ui/base/dragdrop/os_exchange_data.h"
 #include "url/gurl.h"
@@ -238,4 +239,13 @@ bool OSExchangeDataProviderNonBacked::GetPlainTextURL(GURL* url) const {
   return true;
 }
 
+void OSExchangeDataProviderNonBacked::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {
+  source_ = std::move(data_source);
+}
+
+DataTransferEndpoint* OSExchangeDataProviderNonBacked::GetSource() const {
+  return source_.get();
+}
+
 }  // namespace ui

ui/base/dragdrop/os_exchange_data_provider_non_backed.h

@@ -6,6 +6,7 @@
 #define UI_BASE_DRAGDROP_OS_EXCHANGE_DATA_PROVIDER_NON_BACKED_H_
 
 #include <map>
+#include <memory>
 
 #include "base/component_export.h"
 #include "base/pickle.h"
@@ -72,6 +73,9 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderNonBacked
   gfx::ImageSkia GetDragImage() const override;
   gfx::Vector2d GetDragImageOffset() const override;
 
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
+  DataTransferEndpoint* GetSource() const override;
+
  private:
   // Returns true if |formats_| contains a file format and the file name can be
   // parsed as a URL.
@@ -107,6 +111,9 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderNonBacked
   GURL base_url_;
 
   bool originated_from_renderer_ = false;
+
+  // Data source.
+  std::unique_ptr<DataTransferEndpoint> source_;
 };
 
 }  // namespace ui

ui/base/dragdrop/os_exchange_data_provider_win.cc

@@ -34,6 +34,7 @@
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/base/clipboard/clipboard_format_type.h"
 #include "ui/base/clipboard/clipboard_util_win.h"
+#include "ui/base/data_transfer_policy/data_transfer_policy_controller.h"
 #include "ui/base/dragdrop/file_info/file_info.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/l10n/l10n_util_win.h"
@@ -754,6 +755,13 @@ gfx::Vector2d OSExchangeDataProviderWin::GetDragImageOffset() const {
   return gfx::Vector2d();
 }
 
+void OSExchangeDataProviderWin::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {}
+
+DataTransferEndpoint* OSExchangeDataProviderWin::GetSource() const {
+  return nullptr;
+}
+
 ///////////////////////////////////////////////////////////////////////////////
 // DataObjectImpl, IDataObject implementation:
 

ui/base/dragdrop/os_exchange_data_provider_win.h

@@ -191,6 +191,9 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderWin
   gfx::ImageSkia GetDragImage() const override;
   gfx::Vector2d GetDragImageOffset() const override;
 
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
+  DataTransferEndpoint* GetSource() const override;
+
  private:
   void SetVirtualFileContentAtIndexForTesting(base::span<const uint8_t> data,
                                               DWORD tymed,

ui/base/dragdrop/os_exchange_data_provider_x11.cc

@@ -8,6 +8,7 @@
 
 #include "base/check.h"
 #include "ui/base/clipboard/clipboard_constants.h"
+#include "ui/base/data_transfer_policy/data_transfer_policy_controller.h"
 #include "ui/base/x/selection_utils.h"
 #include "ui/events/platform/platform_event_source.h"
 #include "ui/gfx/x/x11_atom_cache.h"
@@ -46,4 +47,11 @@ bool OSExchangeDataProviderX11::DispatchXEvent(x11::Event* xev) {
   return false;
 }
 
+void OSExchangeDataProviderX11::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {}
+
+DataTransferEndpoint* OSExchangeDataProviderX11::GetSource() const {
+  return nullptr;
+}
+
 }  // namespace ui

ui/base/dragdrop/os_exchange_data_provider_x11.h

@@ -38,6 +38,9 @@ class COMPONENT_EXPORT(UI_BASE) OSExchangeDataProviderX11
   // XEventDispatcher:
   bool DispatchXEvent(x11::Event* xev) override;
 
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
+  DataTransferEndpoint* GetSource() const override;
+
  private:
   friend class OSExchangeDataProviderX11Test;
 };

ui/base/x/x11_os_exchange_data_provider.cc

@@ -13,6 +13,7 @@
 #include "net/base/filename_util.h"
 #include "ui/base/clipboard/clipboard_constants.h"
 #include "ui/base/clipboard/clipboard_format_type.h"
+#include "ui/base/data_transfer_policy/data_transfer_endpoint.h"
 #include "ui/base/dragdrop/file_info/file_info.h"
 #include "ui/base/x/selection_utils.h"
 #include "ui/base/x/x11_util.h"
@@ -475,4 +476,11 @@ void XOSExchangeDataProvider::InsertData(
   format_map_.Insert(format, data);
 }
 
+void XOSExchangeDataProvider::SetSource(
+    std::unique_ptr<DataTransferEndpoint> data_source) {}
+
+DataTransferEndpoint* XOSExchangeDataProvider::GetSource() const {
+  return nullptr;
+}
+
 }  // namespace ui

ui/base/x/x11_os_exchange_data_provider.h

@@ -92,6 +92,9 @@ class COMPONENT_EXPORT(UI_BASE_X) XOSExchangeDataProvider
   gfx::ImageSkia GetDragImage() const override;
   gfx::Vector2d GetDragImageOffset() const override;
 
+  void SetSource(std::unique_ptr<DataTransferEndpoint> data_source) override;
+  DataTransferEndpoint* GetSource() const override;
+
  protected:
   friend class OSExchangeDataProviderX11Test;
   using PickleData = std::map<ClipboardFormatType, base::Pickle>;