| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/file_util.h" | |
| 6 | |
| 7 #if defined(OS_WIN) | |
| 8 #include <io.h> | |
| 9 #endif | |
| 10 #include <stdio.h> | |
| 11 | |
| 12 #include <fstream> | |
| 13 #include <limits> | |
| 14 | |
| 15 #include "base/files/file_enumerator.h" | |
| 16 #include "base/files/file_path.h" | |
| 17 #include "base/logging.h" | |
| 18 #include "base/strings/string_piece.h" | |
| 19 #include "base/strings/string_util.h" | |
| 20 #include "base/strings/stringprintf.h" | |
| 21 #include "base/strings/utf_string_conversions.h" | |
| 22 | |
| 23 namespace base { | |
| 24 | |
| 25 namespace { | |
| 26 | |
| 27 // The maximum number of 'uniquified' files we will try to create. | |
| 28 // This is used when the filename we're trying to download is already in use, | |
| 29 // so we create a new unique filename by appending " (nnn)" before the | |
| 30 // extension, where 1 <= nnn <= kMaxUniqueFiles. | |
| 31 // Also used by code that cleans up said files. | |
| 32 static const int kMaxUniqueFiles = 100; | |
| 33 | |
| 34 } // namespace | |
| 35 | |
| 36 int64 ComputeDirectorySize(const FilePath& root_path) { | |
| 37 int64 running_size = 0; | |
| 38 FileEnumerator file_iter(root_path, true, FileEnumerator::FILES); | |
| 39 while (!file_iter.Next().empty()) | |
| 40 running_size += file_iter.GetInfo().GetSize(); | |
| 41 return running_size; | |
| 42 } | |
| 43 | |
| 44 bool Move(const FilePath& from_path, const FilePath& to_path) { | |
| 45 if (from_path.ReferencesParent() || to_path.ReferencesParent()) | |
| 46 return false; | |
| 47 return internal::MoveUnsafe(from_path, to_path); | |
| 48 } | |
| 49 | |
| 50 bool CopyFile(const FilePath& from_path, const FilePath& to_path) { | |
| 51 if (from_path.ReferencesParent() || to_path.ReferencesParent()) | |
| 52 return false; | |
| 53 return internal::CopyFileUnsafe(from_path, to_path); | |
| 54 } | |
| 55 | |
| 56 bool ContentsEqual(const FilePath& filename1, const FilePath& filename2) { | |
| 57 // We open the file in binary format even if they are text files because | |
| 58 // we are just comparing that bytes are exactly same in both files and not | |
| 59 // doing anything smart with text formatting. | |
| 60 std::ifstream file1(filename1.value().c_str(), | |
| 61 std::ios::in | std::ios::binary); | |
| 62 std::ifstream file2(filename2.value().c_str(), | |
| 63 std::ios::in | std::ios::binary); | |
| 64 | |
| 65 // Even if both files aren't openable (and thus, in some sense, "equal"), | |
| 66 // any unusable file yields a result of "false". | |
| 67 if (!file1.is_open() || !file2.is_open()) | |
| 68 return false; | |
| 69 | |
| 70 const int BUFFER_SIZE = 2056; | |
| 71 char buffer1[BUFFER_SIZE], buffer2[BUFFER_SIZE]; | |
| 72 do { | |
| 73 file1.read(buffer1, BUFFER_SIZE); | |
| 74 file2.read(buffer2, BUFFER_SIZE); | |
| 75 | |
| 76 if ((file1.eof() != file2.eof()) || | |
| 77 (file1.gcount() != file2.gcount()) || | |
| 78 (memcmp(buffer1, buffer2, file1.gcount()))) { | |
| 79 file1.close(); | |
| 80 file2.close(); | |
| 81 return false; | |
| 82 } | |
| 83 } while (!file1.eof() || !file2.eof()); | |
| 84 | |
| 85 file1.close(); | |
| 86 file2.close(); | |
| 87 return true; | |
| 88 } | |
| 89 | |
| 90 bool TextContentsEqual(const FilePath& filename1, const FilePath& filename2) { | |
| 91 std::ifstream file1(filename1.value().c_str(), std::ios::in); | |
| 92 std::ifstream file2(filename2.value().c_str(), std::ios::in); | |
| 93 | |
| 94 // Even if both files aren't openable (and thus, in some sense, "equal"), | |
| 95 // any unusable file yields a result of "false". | |
| 96 if (!file1.is_open() || !file2.is_open()) | |
| 97 return false; | |
| 98 | |
| 99 do { | |
| 100 std::string line1, line2; | |
| 101 getline(file1, line1); | |
| 102 getline(file2, line2); | |
| 103 | |
| 104 // Check for mismatched EOF states, or any error state. | |
| 105 if ((file1.eof() != file2.eof()) || | |
| 106 file1.bad() || file2.bad()) { | |
| 107 return false; | |
| 108 } | |
| 109 | |
| 110 // Trim all '\r' and '\n' characters from the end of the line. | |
| 111 std::string::size_type end1 = line1.find_last_not_of("\r\n"); | |
| 112 if (end1 == std::string::npos) | |
| 113 line1.clear(); | |
| 114 else if (end1 + 1 < line1.length()) | |
| 115 line1.erase(end1 + 1); | |
| 116 | |
| 117 std::string::size_type end2 = line2.find_last_not_of("\r\n"); | |
| 118 if (end2 == std::string::npos) | |
| 119 line2.clear(); | |
| 120 else if (end2 + 1 < line2.length()) | |
| 121 line2.erase(end2 + 1); | |
| 122 | |
| 123 if (line1 != line2) | |
| 124 return false; | |
| 125 } while (!file1.eof() || !file2.eof()); | |
| 126 | |
| 127 return true; | |
| 128 } | |
| 129 | |
| 130 bool ReadFileToString(const FilePath& path, | |
| 131 std::string* contents, | |
| 132 size_t max_size) { | |
| 133 if (contents) | |
| 134 contents->clear(); | |
| 135 if (path.ReferencesParent()) | |
| 136 return false; | |
| 137 FILE* file = OpenFile(path, "rb"); | |
| 138 if (!file) { | |
| 139 return false; | |
| 140 } | |
| 141 | |
| 142 char buf[1 << 16]; | |
| 143 size_t len; | |
| 144 size_t size = 0; | |
| 145 bool read_status = true; | |
| 146 | |
| 147 // Many files supplied in |path| have incorrect size (proc files etc). | |
| 148 // Hence, the file is read sequentially as opposed to a one-shot read. | |
| 149 while ((len = fread(buf, 1, sizeof(buf), file)) > 0) { | |
| 150 if (contents) | |
| 151 contents->append(buf, std::min(len, max_size - size)); | |
| 152 | |
| 153 if ((max_size - size) < len) { | |
| 154 read_status = false; | |
| 155 break; | |
| 156 } | |
| 157 | |
| 158 size += len; | |
| 159 } | |
| 160 read_status = read_status && !ferror(file); | |
| 161 CloseFile(file); | |
| 162 | |
| 163 return read_status; | |
| 164 } | |
| 165 | |
| 166 bool ReadFileToString(const FilePath& path, std::string* contents) { | |
| 167 return ReadFileToString(path, contents, std::numeric_limits<size_t>::max()); | |
| 168 } | |
| 169 | |
| 170 bool IsDirectoryEmpty(const FilePath& dir_path) { | |
| 171 FileEnumerator files(dir_path, false, | |
| 172 FileEnumerator::FILES | FileEnumerator::DIRECTORIES); | |
| 173 if (files.Next().empty()) | |
| 174 return true; | |
| 175 return false; | |
| 176 } | |
| 177 | |
| 178 FILE* CreateAndOpenTemporaryFile(FilePath* path) { | |
| 179 FilePath directory; | |
| 180 if (!GetTempDir(&directory)) | |
| 181 return NULL; | |
| 182 | |
| 183 return CreateAndOpenTemporaryFileInDir(directory, path); | |
| 184 } | |
| 185 | |
| 186 bool CreateDirectory(const FilePath& full_path) { | |
| 187 return CreateDirectoryAndGetError(full_path, NULL); | |
| 188 } | |
| 189 | |
| 190 bool GetFileSize(const FilePath& file_path, int64* file_size) { | |
| 191 File::Info info; | |
| 192 if (!GetFileInfo(file_path, &info)) | |
| 193 return false; | |
| 194 *file_size = info.size; | |
| 195 return true; | |
| 196 } | |
| 197 | |
| 198 bool TouchFile(const FilePath& path, | |
| 199 const Time& last_accessed, | |
| 200 const Time& last_modified) { | |
| 201 int flags = File::FLAG_OPEN | File::FLAG_WRITE_ATTRIBUTES; | |
| 202 | |
| 203 #if defined(OS_WIN) | |
| 204 // On Windows, FILE_FLAG_BACKUP_SEMANTICS is needed to open a directory. | |
| 205 if (DirectoryExists(path)) | |
| 206 flags |= File::FLAG_BACKUP_SEMANTICS; | |
| 207 #endif // OS_WIN | |
| 208 | |
| 209 File file(path, flags); | |
| 210 if (!file.IsValid()) | |
| 211 return false; | |
| 212 | |
| 213 return file.SetTimes(last_accessed, last_modified); | |
| 214 } | |
| 215 | |
| 216 bool CloseFile(FILE* file) { | |
| 217 if (file == NULL) | |
| 218 return true; | |
| 219 return fclose(file) == 0; | |
| 220 } | |
| 221 | |
| 222 bool TruncateFile(FILE* file) { | |
| 223 if (file == NULL) | |
| 224 return false; | |
| 225 long current_offset = ftell(file); | |
| 226 if (current_offset == -1) | |
| 227 return false; | |
| 228 #if defined(OS_WIN) | |
| 229 int fd = _fileno(file); | |
| 230 if (_chsize(fd, current_offset) != 0) | |
| 231 return false; | |
| 232 #else | |
| 233 int fd = fileno(file); | |
| 234 if (ftruncate(fd, current_offset) != 0) | |
| 235 return false; | |
| 236 #endif | |
| 237 return true; | |
| 238 } | |
| 239 | |
| 240 int GetUniquePathNumber(const FilePath& path, | |
| 241 const FilePath::StringType& suffix) { | |
| 242 bool have_suffix = !suffix.empty(); | |
| 243 if (!PathExists(path) && | |
| 244 (!have_suffix || !PathExists(FilePath(path.value() + suffix)))) { | |
| 245 return 0; | |
| 246 } | |
| 247 | |
| 248 FilePath new_path; | |
| 249 for (int count = 1; count <= kMaxUniqueFiles; ++count) { | |
| 250 new_path = path.InsertBeforeExtensionASCII(StringPrintf(" (%d)", count)); | |
| 251 if (!PathExists(new_path) && | |
| 252 (!have_suffix || !PathExists(FilePath(new_path.value() + suffix)))) { | |
| 253 return count; | |
| 254 } | |
| 255 } | |
| 256 | |
| 257 return -1; | |
| 258 } | |
| 259 | |
| 260 } // namespace base | |
| OLD | NEW |