| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2009 The Chromium OS 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 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ | 5 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ |
| 6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ | 6 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 #include <vector> | 9 #include <vector> |
| 10 |
| 10 #include <openssl/sha.h> | 11 #include <openssl/sha.h> |
| 11 #include "base/basictypes.h" | 12 #include <base/basictypes.h> |
| 12 #include "base/logging.h" | 13 #include <base/logging.h> |
| 13 | 14 |
| 14 // Omaha uses base64 encoded SHA-256 as the hash. This class provides a simple | 15 // Omaha uses base64 encoded SHA-256 as the hash. This class provides a simple |
| 15 // wrapper around OpenSSL providing such a formatted hash of data passed in. | 16 // wrapper around OpenSSL providing such a formatted hash of data passed in. |
| 16 // The methods of this class must be called in a very specific order: First the | 17 // The methods of this class must be called in a very specific order: First the |
| 17 // ctor (of course), then 0 or more calls to Update(), then Finalize(), then 0 | 18 // ctor (of course), then 0 or more calls to Update(), then Finalize(), then 0 |
| 18 // or more calls to hash(). | 19 // or more calls to hash(). |
| 19 | 20 |
| 20 namespace chromeos_update_engine { | 21 namespace chromeos_update_engine { |
| 21 | 22 |
| 22 class OmahaHashCalculator { | 23 class OmahaHashCalculator { |
| 23 public: | 24 public: |
| 24 OmahaHashCalculator(); | 25 OmahaHashCalculator(); |
| 25 | 26 |
| 26 // Update is called with all of the data that should be hashed in order. | 27 // Update is called with all of the data that should be hashed in order. |
| 27 // Update will read |length| bytes of |data|. | 28 // Update will read |length| bytes of |data|. |
| 28 // Returns true on success. | 29 // Returns true on success. |
| 29 bool Update(const char* data, size_t length); | 30 bool Update(const char* data, size_t length); |
| 30 | 31 |
| 32 // Updates the hash with up to |length| bytes of data from |file|. If |length| |
| 33 // is negative, reads in and updates with the whole file. Returns the number |
| 34 // of bytes that the hash was updated with, or -1 on error. |
| 35 off_t UpdateFile(const std::string& name, off_t length); |
| 36 |
| 31 // Call Finalize() when all data has been passed in. This method tells | 37 // Call Finalize() when all data has been passed in. This method tells |
| 32 // OpenSSl that no more data will come in and base64 encodes the resulting | 38 // OpenSSl that no more data will come in and base64 encodes the resulting |
| 33 // hash. | 39 // hash. |
| 34 // Returns true on success. | 40 // Returns true on success. |
| 35 bool Finalize(); | 41 bool Finalize(); |
| 36 | 42 |
| 37 // Gets the hash. Finalize() must have been called. | 43 // Gets the hash. Finalize() must have been called. |
| 38 const std::string& hash() const { | 44 const std::string& hash() const { |
| 39 DCHECK(!hash_.empty()) << "Call Finalize() first"; | 45 DCHECK(!hash_.empty()) << "Call Finalize() first"; |
| 40 return hash_; | 46 return hash_; |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 72 bool valid_; | 78 bool valid_; |
| 73 | 79 |
| 74 // The hash state used by OpenSSL | 80 // The hash state used by OpenSSL |
| 75 SHA256_CTX ctx_; | 81 SHA256_CTX ctx_; |
| 76 DISALLOW_COPY_AND_ASSIGN(OmahaHashCalculator); | 82 DISALLOW_COPY_AND_ASSIGN(OmahaHashCalculator); |
| 77 }; | 83 }; |
| 78 | 84 |
| 79 } // namespace chromeos_update_engine | 85 } // namespace chromeos_update_engine |
| 80 | 86 |
| 81 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ | 87 #endif // CHROMEOS_PLATFORM_UPDATE_ENGINE_OMAHA_HASH_CALCULATOR_H__ |
| OLD | NEW |