Add built-in native message host for DriveFS

This allows the Drive Web interface to talk to DriveFS on Chrome OS via
the native messaging api.

Bug: 1020954
Change-Id: I450e8ad46cc601e0b0d8df2c19dd4e255f963391
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/1910982
Commit-Queue: Austin Tankiang <austinct@chromium.org>
Reviewed-by: Sergey Ulanov <sergeyu@chromium.org>
Reviewed-by: Sergei Datsenko <dats@chromium.org>
Reviewed-by: Sam McNally <sammc@chromium.org>
Cr-Commit-Position: refs/heads/master@{#716064}

chrome/browser/chromeos/BUILD.gn

@@ -907,6 +907,8 @@ source_set("chromeos") {
     "display/quirks_manager_delegate_impl.h",
     "drive/drive_integration_service.cc",
     "drive/drive_integration_service.h",
+    "drive/drivefs_native_message_host.cc",
+    "drive/drivefs_native_message_host.h",
     "drive/file_system_util.cc",
     "drive/file_system_util.h",
     "drive/fileapi/drivefs_async_file_util.cc",

chrome/browser/chromeos/drive/drivefs_native_message_host.cc

+// Copyright 2019 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.
+
+#include "chrome/browser/chromeos/drive/drivefs_native_message_host.h"
+
+#include "base/bind.h"
+#include "base/logging.h"
+#include "base/stl_util.h"
+#include "base/task/post_task.h"
+#include "base/task/task_traits.h"
+#include "chrome/browser/chromeos/drive/drive_integration_service.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/common/extensions/extension_constants.h"
+#include "components/drive/file_errors.h"
+
+namespace drive {
+
+const char kDriveFsNativeMessageHostName[] = "com.google.drive.nativeproxy";
+
+const char* const kDriveFsNativeMessageHostOrigins[] = {
+    "chrome-extension://lmjegmlicamnimmfhcmpkclmigmmcbeh/",
+};
+
+constexpr size_t kDriveFsNativeMessageHostOriginsSize =
+    base::size(kDriveFsNativeMessageHostOrigins);
+
+class DriveFsNativeMessageHost : public extensions::NativeMessageHost {
+ public:
+  explicit DriveFsNativeMessageHost(Profile* profile)
+      : drive_service_(DriveIntegrationServiceFactory::GetForProfile(profile)) {
+  }
+  ~DriveFsNativeMessageHost() override = default;
+
+  void OnMessage(const std::string& message) override {
+    if (!drive_service_ || !drive_service_->GetDriveFsInterface()) {
+      OnDriveFsResponse(FILE_ERROR_SERVICE_UNAVAILABLE, "");
+      return;
+    }
+
+    drive_service_->GetDriveFsInterface()->SendNativeMessageRequest(
+        message, base::Bind(&DriveFsNativeMessageHost::OnDriveFsResponse,
+                            weak_ptr_factory_.GetWeakPtr()));
+  }
+
+  void Start(Client* client) override { client_ = client; }
+
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner() const override {
+    return task_runner_;
+  }
+
+ private:
+  void OnDriveFsResponse(FileError error, const std::string& response) {
+    if (error == FILE_ERROR_OK) {
+      client_->PostMessageFromNativeHost(response);
+    } else {
+      LOG(WARNING) << "DriveFS returned error " << FileErrorToString(error);
+      client_->CloseChannel(FileErrorToString(error));
+    }
+  }
+
+  DriveIntegrationService* drive_service_;
+
+  Client* client_ = nullptr;
+
+  const scoped_refptr<base::SingleThreadTaskRunner> task_runner_ =
+      base::CreateSingleThreadTaskRunner({base::CurrentThread()});
+
+  base::WeakPtrFactory<DriveFsNativeMessageHost> weak_ptr_factory_{this};
+
+  DISALLOW_COPY_AND_ASSIGN(DriveFsNativeMessageHost);
+};
+
+std::unique_ptr<extensions::NativeMessageHost> CreateDriveFsNativeMessageHost(
+    content::BrowserContext* browser_context) {
+  return std::make_unique<DriveFsNativeMessageHost>(
+      Profile::FromBrowserContext(browser_context));
+}
+
+}  // namespace drive

chrome/browser/chromeos/drive/drivefs_native_message_host.h

+// Copyright 2019 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 CHROME_BROWSER_CHROMEOS_DRIVE_DRIVEFS_NATIVE_MESSAGE_HOST_H_
+#define CHROME_BROWSER_CHROMEOS_DRIVE_DRIVEFS_NATIVE_MESSAGE_HOST_H_
+
+#include <memory>
+
+#include "content/public/browser/browser_context.h"
+#include "extensions/browser/api/messaging/native_message_host.h"
+
+namespace drive {
+
+extern const char kDriveFsNativeMessageHostName[];
+
+extern const char* const kDriveFsNativeMessageHostOrigins[];
+
+extern const size_t kDriveFsNativeMessageHostOriginsSize;
+
+std::unique_ptr<extensions::NativeMessageHost> CreateDriveFsNativeMessageHost(
+    content::BrowserContext* browser_context);
+
+}  // namespace drive
+
+#endif  //  CHROME_BROWSER_CHROMEOS_DRIVE_DRIVEFS_NATIVE_MESSAGE_HOST_H_

chrome/browser/extensions/api/messaging/native_message_host_chromeos.cc

@@ -20,6 +20,7 @@
 #include "base/values.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/arc/extensions/arc_support_message_host.h"
+#include "chrome/browser/chromeos/drive/drivefs_native_message_host.h"
 #include "chrome/browser/chromeos/wilco_dtc_supportd/wilco_dtc_supportd_messaging.h"
 #include "content/public/browser/browser_task_traits.h"
 #include "content/public/browser/browser_thread.h"
@@ -132,6 +133,10 @@ static const BuiltInHost kBuiltInHost[] = {
      chromeos::kWilcoDtcSupportdHostOrigins,
      chromeos::kWilcoDtcSupportdHostOriginsSize,
      &chromeos::CreateExtensionOwnedWilcoDtcSupportdMessageHost},
+    {drive::kDriveFsNativeMessageHostName,
+     drive::kDriveFsNativeMessageHostOrigins,
+     drive::kDriveFsNativeMessageHostOriginsSize,
+     &drive::CreateDriveFsNativeMessageHost},
 };
 
 bool MatchesSecurityOrigin(const BuiltInHost& host,

chromeos/components/drivefs/fake_drivefs.cc

@@ -416,4 +416,10 @@ void FakeDriveFs::FetchAllChangeLogs() {}
 void FakeDriveFs::FetchChangeLog(
     std::vector<mojom::FetchChangeLogOptionsPtr> options) {}
 
+void FakeDriveFs::SendNativeMessageRequest(
+    const std::string& request,
+    SendNativeMessageRequestCallback callback) {
+  std::move(callback).Run(drive::FILE_ERROR_SERVICE_UNAVAILABLE, "");
+}
+
 }  // namespace drivefs

chromeos/components/drivefs/fake_drivefs.h

@@ -98,6 +98,10 @@ class FakeDriveFs : public drivefs::mojom::DriveFs,
   void FetchChangeLog(
       std::vector<mojom::FetchChangeLogOptionsPtr> options) override;
 
+  void SendNativeMessageRequest(
+      const std::string& request,
+      SendNativeMessageRequestCallback callback) override;
+
   const base::FilePath mount_path_;
 
   std::map<base::FilePath, FileMetadata> metadata_;

chromeos/components/drivefs/mojom/drivefs.mojom

@@ -57,6 +57,16 @@ interface DriveFs {
 
   // Fetches the changelog(s) using the provided |options|.
   FetchChangeLog(array<FetchChangeLogOptions> options);
+
+  // Send a native message |request| to DriveFS. |request| is a DriveFS
+  // specific message that looks like "native_opener/v2/$command_id/$message",
+  // where $command_id is an integer and $message is the base64 encoding of a
+  // JSPB encoded proto message.
+  //
+  // |response| is only filled out if |error| is kOk, and returns a JSPB
+  // encoded proto message.
+  SendNativeMessageRequest(string request) => (
+      FileError error, string response);
 };
 
 // Implemented by Chrome, used from DriveFS.