Added new ReadFileToString API with a max_size argument

base/file_util.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 "base/file_util.h"
 
 #if defined(OS_WIN)
 #include <io.h>
 #endif
 #include <stdio.h>
 
 #include <fstream>
+#include <limits>
 
 #include "base/files/file_enumerator.h"
 #include "base/files/file_path.h"
 #include "base/logging.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_util.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 
 namespace base {
@@ skipping 96 matching lines @@
     else if (end2 + 1 < line2.length())
       line2.erase(end2 + 1);
 
     if (line1 != line2)
       return false;
   } while (!file1.eof() || !file2.eof());
 
   return true;
 }
 
-bool ReadFileToString(const FilePath& path, std::string* contents) {
+bool ReadFileToString(const FilePath& path,
+                      std::string* contents,
+                      size_t max_size) {
   if (path.ReferencesParent())
     return false;
   FILE* file = OpenFile(path, "rb");
   if (!file) {
     return false;
   }
 
   char buf[1 << 16];
   size_t len;
+  size_t size = 0;
+
+  // Many files supplied in path have incorrect size (proc files, streams etc)

[bartfab (slow), 2014/02/12 16:27:11]

Nit 1: s/path/|path|/
Nit 2: s/etc/etc./
Nit 3: End the sentence after the bracket. Then, start a new sentence by
capitalizing "Hence" on the next line.

[kaliamoorthi, 2014/02/17 11:09:41]

Done.

+  // hence the file is read sequentially as opposed to one-shot read.

[bartfab (slow), 2014/02/12 16:27:11]

Nit 1: Add comma after "hence."
Nit 2: s/one-shot/a one-shot/

[kaliamoorthi, 2014/02/17 11:09:41]

Done.

   while ((len = fread(buf, 1, sizeof(buf), file)) > 0) {
+    if ((max_size - size) < len) {
+      if (contents)
+        contents->clear();
+      CloseFile(file);
+      return false;
+    }
+    size += len;
     if (contents)
       contents->append(buf, len);
   }
   CloseFile(file);
 
   return true;
 }
 
+bool ReadFileToString(const FilePath& path, std::string* contents) {
+  return ReadFileToString(path, contents, std::numeric_limits<size_t>::max());
+}
+
 bool IsDirectoryEmpty(const FilePath& dir_path) {
   FileEnumerator files(dir_path, false,
       FileEnumerator::FILES | FileEnumerator::DIRECTORIES);
   if (files.Next().empty())
     return true;
   return false;
 }
 
 FILE* CreateAndOpenTemporaryFile(FilePath* path) {
   FilePath directory;
@@ skipping 82 matching lines @@
     if (!PathExists(new_path) &&
         (!have_suffix || !PathExists(FilePath(new_path.value() + suffix)))) {
       return count;
     }
   }
 
   return -1;
 }
 
 }  // namespace file_util