OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Md5sum implementation for Android. This version handles files as well as | 5 // Md5sum implementation for Android. This version handles files as well as |
6 // directories. Its output is sorted by file path. | 6 // directories. Its output is sorted by file path. |
7 | 7 |
8 #include <fstream> | 8 #include <fstream> |
9 #include <iostream> | 9 #include <iostream> |
10 #include <set> | 10 #include <set> |
11 #include <string> | 11 #include <string> |
12 | 12 |
13 #include "base/files/file_enumerator.h" | 13 #include "base/files/file_enumerator.h" |
14 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
15 #include "base/files/file_util.h" | 15 #include "base/files/file_util.h" |
16 #include "base/logging.h" | 16 #include "base/logging.h" |
17 #include "base/md5.h" | 17 #include "base/md5.h" |
18 | 18 |
19 namespace { | 19 namespace { |
20 | 20 |
21 const int kBufferSize = 1024; | |
22 | |
23 // Returns whether |path|'s MD5 was successfully written to |digest_string|. | 21 // Returns whether |path|'s MD5 was successfully written to |digest_string|. |
24 bool MD5Sum(const char* path, std::string* digest_string) { | 22 bool MD5Sum(const char* path, std::string* digest_string) { |
25 std::ifstream stream(path); | 23 base::ScopedFILE file(fopen(path, "rb")); |
26 if (!stream.good()) { | 24 if (!file) { |
27 LOG(ERROR) << "Could not open file " << path; | 25 LOG(ERROR) << "Could not open file " << path; |
28 return false; | 26 return false; |
29 } | 27 } |
30 base::MD5Context ctx; | 28 base::MD5Context ctx; |
31 base::MD5Init(&ctx); | 29 base::MD5Init(&ctx); |
32 char buf[kBufferSize]; | 30 const size_t kBufferSize = 1 << 16; |
33 while (stream.good()) { | 31 scoped_ptr<char[]> buf(new char[kBufferSize]); |
34 std::streamsize bytes_read = stream.readsome(buf, sizeof(buf)); | 32 size_t len; |
35 if (bytes_read == 0) | 33 while ((len = fread(buf.get(), 1, kBufferSize, file.get())) > 0) |
36 break; | 34 base::MD5Update(&ctx, base::StringPiece(buf.get(), len)); |
37 base::MD5Update(&ctx, base::StringPiece(buf, bytes_read)); | 35 if (ferror(file.get())) { |
38 } | |
39 if (stream.fail()) { | |
40 LOG(ERROR) << "Error reading file " << path; | 36 LOG(ERROR) << "Error reading file " << path; |
41 return false; | 37 return false; |
42 } | 38 } |
43 base::MD5Digest digest; | 39 base::MD5Digest digest; |
44 base::MD5Final(&digest, &ctx); | 40 base::MD5Final(&digest, &ctx); |
45 *digest_string = base::MD5DigestToBase16(digest); | 41 *digest_string = base::MD5DigestToBase16(digest); |
46 return true; | 42 return true; |
47 } | 43 } |
48 | 44 |
49 // Returns the set of all files contained in |files|. This handles directories | 45 // Returns the set of all files contained in |files|. This handles directories |
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
84 for (std::set<std::string>::const_iterator it = files.begin(); | 80 for (std::set<std::string>::const_iterator it = files.begin(); |
85 it != files.end(); ++it) { | 81 it != files.end(); ++it) { |
86 if (!MD5Sum(it->c_str(), &digest)) | 82 if (!MD5Sum(it->c_str(), &digest)) |
87 failed = true; | 83 failed = true; |
88 base::FilePath file_path(*it); | 84 base::FilePath file_path(*it); |
89 std::cout << digest << " " | 85 std::cout << digest << " " |
90 << base::MakeAbsoluteFilePath(file_path).value() << std::endl; | 86 << base::MakeAbsoluteFilePath(file_path).value() << std::endl; |
91 } | 87 } |
92 return failed; | 88 return failed; |
93 } | 89 } |
OLD | NEW |