Integrate external mount points to IsolatedContext

webkit/chromeos/fileapi/cros_mount_point_provider.cc

 // Copyright (c) 2012 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 "webkit/chromeos/fileapi/cros_mount_point_provider.h"
 
 #include "base/chromeos/chromeos_version.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/stringprintf.h"
 #include "base/synchronization/lock.h"
 #include "base/utf_string_conversions.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityOrigin.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebCString.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebFileSystem.h"
 #include "third_party/WebKit/Source/WebKit/chromium/public/platform/WebString.h"
 #include "webkit/chromeos/fileapi/file_access_permissions.h"
 #include "webkit/chromeos/fileapi/remote_file_stream_writer.h"
 #include "webkit/chromeos/fileapi/remote_file_system_operation.h"
 #include "webkit/fileapi/file_system_file_stream_reader.h"
 #include "webkit/fileapi/file_system_operation_context.h"
 #include "webkit/fileapi/file_system_url.h"
 #include "webkit/fileapi/file_system_util.h"
+#include "webkit/fileapi/isolated_context.h"
+#include "webkit/fileapi/isolated_file_util.h"
 #include "webkit/fileapi/local_file_stream_writer.h"
 #include "webkit/fileapi/local_file_system_operation.h"
 #include "webkit/glue/webkit_glue.h"
 
 namespace {
 
 const char kChromeUIScheme[] = "chrome";
 
 }  // namespace
 
 namespace chromeos {
 
-CrosMountPointProvider::MountPoint::MountPoint(
-    const FilePath& in_web_root_path,
-    const FilePath& in_local_root_path,
-    FileSystemLocation in_location,
-    fileapi::RemoteFileSystemProxyInterface* in_proxy)
-        : web_root_path(in_web_root_path), local_root_path(in_local_root_path),
-          location(in_location), remote_proxy(in_proxy) {
-}
-
-CrosMountPointProvider::MountPoint::~MountPoint() {
+// static
+bool CrosMountPointProvider::CanHandleURL(const fileapi::FileSystemURL& url) {
+  if (!url.is_valid())
+    return false;
+  return url.type() == fileapi::kFileSystemTypeNativeLocal ||
+         url.type() == fileapi::kFileSystemTypeDrive;
 }
 
 CrosMountPointProvider::CrosMountPointProvider(
     scoped_refptr<quota::SpecialStoragePolicy> special_storage_policy)
     : special_storage_policy_(special_storage_policy),
       file_access_permissions_(new FileAccessPermissions()),
-      local_file_util_(new fileapi::LocalFileUtil()) {
+      local_file_util_(new fileapi::IsolatedFileUtil()) {
   FilePath home_path;
   if (PathService::Get(base::DIR_HOME, &home_path))
     AddLocalMountPoint(home_path.AppendASCII("Downloads"));
   AddLocalMountPoint(FilePath(FILE_PATH_LITERAL("/media/archive")));
   AddLocalMountPoint(FilePath(FILE_PATH_LITERAL("/media/removable")));
 }
 
 CrosMountPointProvider::~CrosMountPointProvider() {
 }
 
-bool CrosMountPointProvider::GetRootForVirtualPath(
-    const FilePath& virtual_path, FilePath* root_path) {
-  const MountPoint* mount_point = GetMountPoint(virtual_path);
-  if (!mount_point)
-    return false;
-
-  DCHECK(root_path);
-  *root_path = mount_point->local_root_path;
-  return true;
-}
-
 void CrosMountPointProvider::ValidateFileSystemRoot(
     const GURL& origin_url,
     fileapi::FileSystemType type,
     bool create,
     const ValidateFileSystemCallback& callback) {
   // Nothing to validate for external filesystem.
   DCHECK(type == fileapi::kFileSystemTypeExternal);
   callback.Run(base::PLATFORM_FILE_OK);
 }
 
 FilePath CrosMountPointProvider::GetFileSystemRootPathOnFileThread(
     const GURL& origin_url,
     fileapi::FileSystemType type,
     const FilePath& virtual_path,
     bool create) {
   DCHECK(type == fileapi::kFileSystemTypeExternal);
-  FilePath root_path;
-  if (!GetRootForVirtualPath(virtual_path, &root_path))
+  fileapi::FileSystemURL url(origin_url, type, virtual_path);
+  if (!url.is_valid())
     return FilePath();
 
-  return root_path;
+  FilePath root_path;
+  if (!isolated_context()->GetRegisteredPath(url.filesystem_id(), &root_path))
+    return FilePath();
+
+  return root_path.DirName();
 }
 
 bool CrosMountPointProvider::IsAccessAllowed(const GURL& origin_url,
                                              fileapi::FileSystemType type,
                                              const FilePath& virtual_path) {
-  if (type != fileapi::kFileSystemTypeExternal)
+  // TODO(kinuko): this should call CanHandleURL() once
+  // http://crbug.com/142267 is fixed.
+  if (type != fileapi::kFileSystemTypeNativeLocal &&
+      type != fileapi::kFileSystemTypeDrive)
     return false;
 
   // Permit access to mount points from internal WebUI.
   if (origin_url.SchemeIs(kChromeUIScheme))
     return true;
 
   std::string extension_id = origin_url.host();
   // Check first to make sure this extension has fileBrowserHander permissions.
   if (!special_storage_policy_->IsFileHandler(extension_id))
     return false;
@@ skipping 15 matching lines @@
 void CrosMountPointProvider::DeleteFileSystem(
     const GURL& origin_url,
     fileapi::FileSystemType type,
     fileapi::FileSystemContext* context,
     const DeleteFileSystemCallback& callback) {
   NOTREACHED();
   callback.Run(base::PLATFORM_FILE_ERROR_INVALID_OPERATION);
 }
 
 bool CrosMountPointProvider::HasMountPoint(const FilePath& mount_point) {
-  base::AutoLock locker(mount_point_map_lock_);
-  MountPointMap::const_iterator iter = mount_point_map_.find(
-      mount_point.BaseName().value());
-  DCHECK(iter == mount_point_map_.end() ||
-         iter->second.local_root_path == mount_point.DirName());
-  return iter != mount_point_map_.end();
+  std::string mount_name = mount_point.BaseName().AsUTF8Unsafe();
+  FilePath path;
+  const bool valid = isolated_context()->GetRegisteredPath(mount_name, &path);
+  return valid && path == mount_point;
 }
 
 void CrosMountPointProvider::AddLocalMountPoint(const FilePath& mount_point) {
+  std::string mount_name = mount_point.BaseName().AsUTF8Unsafe();
+  isolated_context()->RevokeFileSystem(mount_name);
+  isolated_context()->RegisterExternalFileSystem(
+      mount_name,
+      fileapi::kFileSystemTypeNativeLocal,
+      mount_point);
   base::AutoLock locker(mount_point_map_lock_);
-  mount_point_map_.erase(mount_point.BaseName().value());
-  mount_point_map_.insert(std::make_pair(
-      mount_point.BaseName().value(),
-      MountPoint(mount_point.BaseName(),
-                 mount_point.DirName(),
-                 LOCAL,
-                 NULL)));
+  local_to_virtual_map_[mount_point] = mount_point.BaseName();
 }
 
 void CrosMountPointProvider::AddRemoteMountPoint(
     const FilePath& mount_point,
     fileapi::RemoteFileSystemProxyInterface* remote_proxy) {
   DCHECK(remote_proxy);
+  std::string mount_name = mount_point.BaseName().AsUTF8Unsafe();
+  isolated_context()->RevokeFileSystem(mount_name);
+  isolated_context()->RegisterExternalFileSystem(mount_name,
+                                                 fileapi::kFileSystemTypeDrive,
+                                                 mount_point);
   base::AutoLock locker(mount_point_map_lock_);
-  mount_point_map_.erase(mount_point.BaseName().value());
-  mount_point_map_.insert(std::make_pair(
-      mount_point.BaseName().value(),
-      MountPoint(mount_point.BaseName(),
-                 mount_point.DirName(),
-                 REMOTE,
-                 remote_proxy)));
+  remote_proxy_map_[mount_name] = remote_proxy;
+  local_to_virtual_map_[mount_point] = mount_point.BaseName();
 }
 
 void CrosMountPointProvider::RemoveMountPoint(const FilePath& mount_point) {
+  std::string mount_name = mount_point.BaseName().AsUTF8Unsafe();
+  isolated_context()->RevokeFileSystem(mount_name);
   base::AutoLock locker(mount_point_map_lock_);
-  mount_point_map_.erase(mount_point.BaseName().value());
+  remote_proxy_map_.erase(mount_name);
+  local_to_virtual_map_.erase(mount_point);
 }
 
 void CrosMountPointProvider::GrantFullAccessToExtension(
     const std::string& extension_id) {
   DCHECK(special_storage_policy_->IsFileHandler(extension_id));
   if (!special_storage_policy_->IsFileHandler(extension_id))
     return;
-  for (MountPointMap::const_iterator iter = mount_point_map_.begin();
-       iter != mount_point_map_.end();
-       ++iter) {
-    GrantFileAccessToExtension(extension_id, FilePath(iter->first));
+  std::vector<fileapi::IsolatedContext::FileInfo> files =
+      isolated_context()->GetExternalMountPoints();
+  for (size_t i = 0; i < files.size(); ++i) {
+    GrantFileAccessToExtension(extension_id,
+                               FilePath::FromUTF8Unsafe(files[i].name));
   }
 }
 
 void CrosMountPointProvider::GrantFileAccessToExtension(
     const std::string& extension_id, const FilePath& virtual_path) {
   // All we care about here is access from extensions for now.
   DCHECK(special_storage_policy_->IsFileHandler(extension_id));
   if (!special_storage_policy_->IsFileHandler(extension_id))
     return;
   file_access_permissions_->GrantAccessPermission(extension_id, virtual_path);
 }
 
 void CrosMountPointProvider::RevokeAccessForExtension(
       const std::string& extension_id) {
   file_access_permissions_->RevokePermissions(extension_id);
 }
 
 std::vector<FilePath> CrosMountPointProvider::GetRootDirectories() const {
+  std::vector<fileapi::IsolatedContext::FileInfo> files =
+      isolated_context()->GetExternalMountPoints();
   std::vector<FilePath> root_dirs;
-  for (MountPointMap::const_iterator iter = mount_point_map_.begin();
-       iter != mount_point_map_.end();
-       ++iter) {
-    root_dirs.push_back(iter->second.local_root_path.Append(iter->first));
-  }
+  for (size_t i = 0; i < files.size(); ++i)
+    root_dirs.push_back(files[i].path);
   return root_dirs;
 }
 
 fileapi::FileSystemFileUtil* CrosMountPointProvider::GetFileUtil(
     fileapi::FileSystemType type) {
+  DCHECK(type == fileapi::kFileSystemTypeNativeLocal);
   return local_file_util_.get();
 }
 
 FilePath CrosMountPointProvider::GetPathForPermissionsCheck(
     const FilePath& virtual_path) const {
-  const MountPoint* mount_point = GetMountPoint(virtual_path);
-  if (!mount_point)
-    return FilePath();
-
-  FilePath root_path = mount_point->local_root_path;
-
-  return root_path.Append(virtual_path);
-}
-
-const CrosMountPointProvider::MountPoint*
-CrosMountPointProvider::GetMountPoint(const FilePath& virtual_path) const {
-  std::vector<FilePath::StringType> components;
-  virtual_path.GetComponents(&components);
-  if (components.empty())
-    return NULL;
-
-  base::AutoLock locker(
-      const_cast<CrosMountPointProvider*>(this)->mount_point_map_lock_);
-  // Check if this root mount point is exposed by this provider.
-  MountPointMap::const_iterator iter = mount_point_map_.find(components[0]);
-  if (iter == mount_point_map_.end())
-    return NULL;
-
-  return &(iter->second);
+  return virtual_path;
 }
 
 fileapi::FileSystemOperationInterface*
 CrosMountPointProvider::CreateFileSystemOperation(
     const fileapi::FileSystemURL& url,
     fileapi::FileSystemContext* context) const {
-  const MountPoint* mount_point = GetMountPoint(url.path());
-  if (mount_point && mount_point->location == REMOTE)
-    return new chromeos::RemoteFileSystemOperation(mount_point->remote_proxy);
+  if (url.type() == fileapi::kFileSystemTypeDrive) {
+    base::AutoLock locker(mount_point_map_lock_);
+    RemoteProxyMap::const_iterator found = remote_proxy_map_.find(
+        url.filesystem_id());
+    // TODO(kinuko): we should handle not-found case gracefully.
+    // http://crbug.com/141617
+    if (found != remote_proxy_map_.end()) {
+      return new chromeos::RemoteFileSystemOperation(found->second);
+    }
+  }
 
+  DCHECK(url.type() == fileapi::kFileSystemTypeNativeLocal);
   scoped_ptr<fileapi::FileSystemOperationContext> operation_context(
       new fileapi::FileSystemOperationContext(context));
   return new fileapi::LocalFileSystemOperation(context,
                                                operation_context.Pass());
 }
 
 webkit_blob::FileStreamReader* CrosMountPointProvider::CreateFileStreamReader(
     const fileapi::FileSystemURL& url,
     int64 offset,
     fileapi::FileSystemContext* context) const {
   // For now we return a generic Reader implementation which utilizes
   // CreateSnapshotFile internally (i.e. will download everything first).
   // TODO(satorux,zel): implement more efficient reader for remote cases.
   return new fileapi::FileSystemFileStreamReader(context, url, offset);
 }
 
 fileapi::FileStreamWriter* CrosMountPointProvider::CreateFileStreamWriter(
     const fileapi::FileSystemURL& url,
     int64 offset,
     fileapi::FileSystemContext* context) const {
   if (!url.is_valid())
     return NULL;
-  const MountPoint* mount_point = GetMountPoint(url.path());
-  if (!mount_point)
-    return NULL;
-  if (mount_point->location == REMOTE) {
-    return new fileapi::RemoteFileStreamWriter(mount_point->remote_proxy,
-                                               url,
-                                               offset);
+  if (url.type() == fileapi::kFileSystemTypeDrive) {
+    base::AutoLock locker(mount_point_map_lock_);
+    RemoteProxyMap::const_iterator found = remote_proxy_map_.find(
+        url.filesystem_id());
+    if (found == remote_proxy_map_.end())
+      return NULL;
+    return new fileapi::RemoteFileStreamWriter(found->second, url, offset);
   }
-  FilePath root_path = mount_point->local_root_path;
-  return new fileapi::LocalFileStreamWriter(
-      root_path.Append(url.path()), offset);
+
+  DCHECK(url.type() == fileapi::kFileSystemTypeNativeLocal);
+  return new fileapi::LocalFileStreamWriter(url.path(), offset);
 }
 
 bool CrosMountPointProvider::GetVirtualPath(const FilePath& filesystem_path,
                                            FilePath* virtual_path) {
-  for (MountPointMap::const_iterator iter = mount_point_map_.begin();
-       iter != mount_point_map_.end();
-       ++iter) {
-    FilePath mount_prefix = iter->second.local_root_path.Append(iter->first);
-    *virtual_path = FilePath(iter->first);
-    if (mount_prefix == filesystem_path) {
-      return true;
-    } else if (mount_prefix.AppendRelativePath(filesystem_path, virtual_path)) {
-      return true;
-    }
+  base::AutoLock locker(mount_point_map_lock_);
+  std::map<FilePath, FilePath>::reverse_iterator iter(
+      local_to_virtual_map_.upper_bound(filesystem_path));
+  if (iter == local_to_virtual_map_.rend())
+    return false;
+  if (iter->first == filesystem_path) {
+    *virtual_path = iter->second;
+    return true;
   }
-  return false;
+  return iter->first.DirName().AppendRelativePath(
+      filesystem_path, virtual_path);
+}
+
+fileapi::IsolatedContext* CrosMountPointProvider::isolated_context() const {
+  return fileapi::IsolatedContext::GetInstance();
 }
 
 }  // namespace chromeos