mojo filesystem: Simplify full file reading/writing.

components/filesystem/directory_impl.cc

 // 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.
 
 #include "components/filesystem/directory_impl.h"
 
 #include <utility>
 
 #include "base/files/file.h"
 #include "base/files/file_enumerator.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "build/build_config.h"
 #include "components/filesystem/file_impl.h"
 #include "components/filesystem/util.h"
+#include "mojo/common/common_type_converters.h"
 
 namespace filesystem {
 
 DirectoryImpl::DirectoryImpl(mojo::InterfaceRequest<Directory> request,
                              base::FilePath directory_path,
                              scoped_ptr<base::ScopedTempDir> temp_dir)
     : binding_(this, std::move(request)),
       directory_path_(directory_path),
       temp_dir_(std::move(temp_dir)) {}
 
@@ skipping 29 matching lines @@
                              const OpenFileCallback& callback) {
   base::FilePath path;
   FileError error = ValidatePath(raw_path, directory_path_, &path);
   if (error != FileError::OK) {
     callback.Run(error);
     return;
   }
 
 #if defined(OS_WIN)
   // On Windows, FILE_FLAG_BACKUP_SEMANTICS is needed to open a directory.
-  if (DirectoryExists(path))
+  if (base::DirectoryExists(path))
     open_flags |= base::File::FLAG_BACKUP_SEMANTICS;
 #endif  // OS_WIN
 
   base::File base_file(path, open_flags);
   if (!base_file.IsValid()) {
     callback.Run(GetError(base_file));
     return;
   }
 
   base::File::Info info;
@@ skipping 130 matching lines @@
   }
 
   if (!file.Flush()) {
     callback.Run(FileError::FAILED);
     return;
   }
 
   callback.Run(FileError::OK);
 }
 
+void DirectoryImpl::ReadEntireFile(const mojo::String& raw_path,
+                                   const ReadEntireFileCallback& callback) {
+  base::FilePath path;
+  FileError error = ValidatePath(raw_path, directory_path_, &path);
+  if (error != FileError::OK) {
+    callback.Run(error, mojo::Array<uint8_t>(0));
+    return;
+  }
+
+  if (base::DirectoryExists(path)) {
+    callback.Run(FileError::NOT_A_FILE, mojo::Array<uint8_t>(0));
+    return;
+  }
+
+  base::File base_file(path, base::File::FLAG_OPEN | base::File::FLAG_READ);
+  if (!base_file.IsValid()) {
+    callback.Run(GetError(base_file), mojo::Array<uint8_t>(0));
+    return;
+  }
+
+  std::string contents;
+  const int kBufferSize = 1 << 16;
+  scoped_ptr<char[]> buf(new char[kBufferSize]);
+  size_t len;
+  while ((len = base_file.ReadAtCurrentPos(buf.get(), kBufferSize)) > 0)

[sky, 2016/01/27 01:08:40]

ReadAtCurrentPos returns an int.

+    contents.append(buf.get(), len);
+
+  callback.Run(FileError::OK, mojo::Array<uint8_t>::From(contents));
+}
+
+void DirectoryImpl::WriteFile(const mojo::String& raw_path,
+                              mojo::Array<uint8_t> data,
+                              const WriteFileCallback& callback) {
+  base::FilePath path;
+  FileError error = ValidatePath(raw_path, directory_path_, &path);
+  if (error != FileError::OK) {
+    callback.Run(error);
+    return;
+  }
+
+  if (base::DirectoryExists(path)) {
+    callback.Run(FileError::NOT_A_FILE);
+    return;
+  }
+
+  base::File base_file(path,
+                       base::File::FLAG_CREATE_ALWAYS | base::File::FLAG_WRITE);
+  if (!base_file.IsValid()) {
+    callback.Run(GetError(base_file));
+    return;
+  }
+
+  // If we're given empty data, we don't write and just truncate the file.
+  if (data.size()) {
+    if (base_file.Write(0, reinterpret_cast<char*>(&data.front()),
+                        data.size()) == -1) {
+      callback.Run(GetError(base_file));
+      return;
+    }
+  }
+
+  callback.Run(FileError::OK);
+}
+
 }  // namespace filesystem