| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h" | 5 #include "chrome/browser/safe_browsing/safe_browsing_store_file.h" |
| 6 | 6 |
| 7 #include "base/callback.h" | 7 #include "base/callback.h" |
| 8 #include "base/md5.h" | 8 #include "base/md5.h" |
| 9 #include "base/metrics/histogram.h" | 9 #include "base/metrics/histogram.h" |
| 10 | 10 |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 | 51 |
| 52 // Read an array of |nmemb| items from |fp| into |ptr|, and fold the | 52 // Read an array of |nmemb| items from |fp| into |ptr|, and fold the |
| 53 // input data into the checksum in |context|, if non-NULL. Return | 53 // input data into the checksum in |context|, if non-NULL. Return |
| 54 // true on success. | 54 // true on success. |
| 55 template <class T> | 55 template <class T> |
| 56 bool ReadArray(T* ptr, size_t nmemb, FILE* fp, base::MD5Context* context) { | 56 bool ReadArray(T* ptr, size_t nmemb, FILE* fp, base::MD5Context* context) { |
| 57 const size_t ret = fread(ptr, sizeof(T), nmemb, fp); | 57 const size_t ret = fread(ptr, sizeof(T), nmemb, fp); |
| 58 if (ret != nmemb) | 58 if (ret != nmemb) |
| 59 return false; | 59 return false; |
| 60 | 60 |
| 61 if (context) | 61 if (context) { |
| 62 base::MD5Update(context, ptr, sizeof(T) * nmemb); | 62 base::MD5Update(context, |
| 63 base::StringPiece(reinterpret_cast<char*>(ptr), |
| 64 sizeof(T) * nmemb)); |
| 65 } |
| 63 return true; | 66 return true; |
| 64 } | 67 } |
| 65 | 68 |
| 66 // Write an array of |nmemb| items from |ptr| to |fp|, and fold the | 69 // Write an array of |nmemb| items from |ptr| to |fp|, and fold the |
| 67 // output data into the checksum in |context|, if non-NULL. Return | 70 // output data into the checksum in |context|, if non-NULL. Return |
| 68 // true on success. | 71 // true on success. |
| 69 template <class T> | 72 template <class T> |
| 70 bool WriteArray(const T* ptr, size_t nmemb, FILE* fp, | 73 bool WriteArray(const T* ptr, size_t nmemb, FILE* fp, |
| 71 base::MD5Context* context) { | 74 base::MD5Context* context) { |
| 72 const size_t ret = fwrite(ptr, sizeof(T), nmemb, fp); | 75 const size_t ret = fwrite(ptr, sizeof(T), nmemb, fp); |
| 73 if (ret != nmemb) | 76 if (ret != nmemb) |
| 74 return false; | 77 return false; |
| 75 | 78 |
| 76 if (context) | 79 if (context) { |
| 77 base::MD5Update(context, ptr, sizeof(T) * nmemb); | 80 base::MD5Update(context, |
| 81 base::StringPiece(reinterpret_cast<const char*>(ptr), |
| 82 sizeof(T) * nmemb)); |
| 83 } |
| 78 | 84 |
| 79 return true; | 85 return true; |
| 80 } | 86 } |
| 81 | 87 |
| 82 // Expand |values| to fit |count| new items, read those items from | 88 // Expand |values| to fit |count| new items, read those items from |
| 83 // |fp| and fold them into the checksum in |context|. Returns true on | 89 // |fp| and fold them into the checksum in |context|. Returns true on |
| 84 // success. | 90 // success. |
| 85 template <class T> | 91 template <class T> |
| 86 bool ReadToVector(std::vector<T>* values, size_t count, FILE* fp, | 92 bool ReadToVector(std::vector<T>* values, size_t count, FILE* fp, |
| 87 base::MD5Context* context) { | 93 base::MD5Context* context) { |
| (...skipping 634 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 722 out->insert(out->end(), sub_chunks_cache_.begin(), sub_chunks_cache_.end()); | 728 out->insert(out->end(), sub_chunks_cache_.begin(), sub_chunks_cache_.end()); |
| 723 } | 729 } |
| 724 | 730 |
| 725 void SafeBrowsingStoreFile::DeleteAddChunk(int32 chunk_id) { | 731 void SafeBrowsingStoreFile::DeleteAddChunk(int32 chunk_id) { |
| 726 add_del_cache_.insert(chunk_id); | 732 add_del_cache_.insert(chunk_id); |
| 727 } | 733 } |
| 728 | 734 |
| 729 void SafeBrowsingStoreFile::DeleteSubChunk(int32 chunk_id) { | 735 void SafeBrowsingStoreFile::DeleteSubChunk(int32 chunk_id) { |
| 730 sub_del_cache_.insert(chunk_id); | 736 sub_del_cache_.insert(chunk_id); |
| 731 } | 737 } |
| OLD | NEW |