Pieter Wuille
12 years ago
7 changed files with 121 additions and 107 deletions
@ -0,0 +1,116 @@
@@ -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