Pieter Wuille
12 years ago
committed by
Pieter Wuille
7 changed files with 121 additions and 107 deletions
@ -0,0 +1,116 @@ |
|||
// Copyright (c) 2009-2010 Satoshi Nakamoto
|
|||
// Copyright (c) 2009-2012 The Bitcoin developers
|
|||
// Distributed under the MIT/X11 software license, see the accompanying
|
|||
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|||
#ifndef BITCOIN_HASH_H |
|||
#define BITCOIN_HASH_H |
|||
|
|||
#include "uint256.h" |
|||
#include "serialize.h" |
|||
|
|||
#include <openssl/sha.h> |
|||
#include <openssl/ripemd.h> |
|||
|
|||
template<typename T1> |
|||
inline uint256 Hash(const T1 pbegin, const T1 pend) |
|||
{ |
|||
static unsigned char pblank[1]; |
|||
uint256 hash1; |
|||
SHA256((pbegin == pend ? pblank : (unsigned char*)&pbegin[0]), (pend - pbegin) * sizeof(pbegin[0]), (unsigned char*)&hash1); |
|||
uint256 hash2; |
|||
SHA256((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2); |
|||
return hash2; |
|||
} |
|||
|
|||
class CHashWriter |
|||
{ |
|||
private: |
|||
SHA256_CTX ctx; |
|||
|
|||
public: |
|||
int nType; |
|||
int nVersion; |
|||
|
|||
void Init() { |
|||
SHA256_Init(&ctx); |
|||
} |
|||
|
|||
CHashWriter(int nTypeIn, int nVersionIn) : nType(nTypeIn), nVersion(nVersionIn) { |
|||
Init(); |
|||
} |
|||
|
|||
CHashWriter& write(const char *pch, size_t size) { |
|||
SHA256_Update(&ctx, pch, size); |
|||
return (*this); |
|||
} |
|||
|
|||
// invalidates the object
|
|||
uint256 GetHash() { |
|||
uint256 hash1; |
|||
SHA256_Final((unsigned char*)&hash1, &ctx); |
|||
uint256 hash2; |
|||
SHA256((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2); |
|||
return hash2; |
|||
} |
|||
|
|||
template<typename T> |
|||
CHashWriter& operator<<(const T& obj) { |
|||
// Serialize to this stream
|
|||
::Serialize(*this, obj, nType, nVersion); |
|||
return (*this); |
|||
} |
|||
}; |
|||
|
|||
|
|||
template<typename T1, typename T2> |
|||
inline uint256 Hash(const T1 p1begin, const T1 p1end, |
|||
const T2 p2begin, const T2 p2end) |
|||
{ |
|||
static unsigned char pblank[1]; |
|||
uint256 hash1; |
|||
SHA256_CTX ctx; |
|||
SHA256_Init(&ctx); |
|||
SHA256_Update(&ctx, (p1begin == p1end ? pblank : (unsigned char*)&p1begin[0]), (p1end - p1begin) * sizeof(p1begin[0])); |
|||
SHA256_Update(&ctx, (p2begin == p2end ? pblank : (unsigned char*)&p2begin[0]), (p2end - p2begin) * sizeof(p2begin[0])); |
|||
SHA256_Final((unsigned char*)&hash1, &ctx); |
|||
uint256 hash2; |
|||
SHA256((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2); |
|||
return hash2; |
|||
} |
|||
|
|||
template<typename T1, typename T2, typename T3> |
|||
inline uint256 Hash(const T1 p1begin, const T1 p1end, |
|||
const T2 p2begin, const T2 p2end, |
|||
const T3 p3begin, const T3 p3end) |
|||
{ |
|||
static unsigned char pblank[1]; |
|||
uint256 hash1; |
|||
SHA256_CTX ctx; |
|||
SHA256_Init(&ctx); |
|||
SHA256_Update(&ctx, (p1begin == p1end ? pblank : (unsigned char*)&p1begin[0]), (p1end - p1begin) * sizeof(p1begin[0])); |
|||
SHA256_Update(&ctx, (p2begin == p2end ? pblank : (unsigned char*)&p2begin[0]), (p2end - p2begin) * sizeof(p2begin[0])); |
|||
SHA256_Update(&ctx, (p3begin == p3end ? pblank : (unsigned char*)&p3begin[0]), (p3end - p3begin) * sizeof(p3begin[0])); |
|||
SHA256_Final((unsigned char*)&hash1, &ctx); |
|||
uint256 hash2; |
|||
SHA256((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2); |
|||
return hash2; |
|||
} |
|||
|
|||
template<typename T> |
|||
uint256 SerializeHash(const T& obj, int nType=SER_GETHASH, int nVersion=PROTOCOL_VERSION) |
|||
{ |
|||
CHashWriter ss(nType, nVersion); |
|||
ss << obj; |
|||
return ss.GetHash(); |
|||
} |
|||
|
|||
inline uint160 Hash160(const std::vector<unsigned char>& vch) |
|||
{ |
|||
uint256 hash1; |
|||
SHA256(&vch[0], vch.size(), (unsigned char*)&hash1); |
|||
uint160 hash2; |
|||
RIPEMD160((unsigned char*)&hash1, sizeof(hash1), (unsigned char*)&hash2); |
|||
return hash2; |
|||
} |
|||
|
|||
#endif |
Loading…
Reference in new issue