Paweł Bylica
8 years ago
5 changed files with 3 additions and 711 deletions
@ -1,183 +0,0 @@ |
|||||
/*
|
|
||||
This file is part of cpp-ethereum. |
|
||||
|
|
||||
cpp-ethereum is free software: you can redistribute it and/or modify |
|
||||
it under the terms of the GNU General Public License as published by |
|
||||
the Free Software Foundation, either version 3 of the License, or |
|
||||
(at your option) any later version. |
|
||||
|
|
||||
cpp-ethereum is distributed in the hope that it will be useful, |
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
GNU General Public License for more details. |
|
||||
|
|
||||
You should have received a copy of the GNU General Public License |
|
||||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
||||
*/ |
|
||||
/** @file MemoryDB.cpp
|
|
||||
* @author Gav Wood <i@gavwood.com> |
|
||||
* @date 2014 |
|
||||
*/ |
|
||||
|
|
||||
#include "Common.h" |
|
||||
#include "MemoryDB.h" |
|
||||
using namespace std; |
|
||||
using namespace dev; |
|
||||
|
|
||||
namespace dev |
|
||||
{ |
|
||||
|
|
||||
const char* DBChannel::name() { return "TDB"; } |
|
||||
const char* DBWarn::name() { return "TDB"; } |
|
||||
|
|
||||
std::unordered_map<h256, std::string> MemoryDB::get() const |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
std::unordered_map<h256, std::string> ret; |
|
||||
for (auto const& i: m_main) |
|
||||
if (!m_enforceRefs || i.second.second > 0) |
|
||||
ret.insert(make_pair(i.first, i.second.first)); |
|
||||
return ret; |
|
||||
} |
|
||||
|
|
||||
MemoryDB& MemoryDB::operator=(MemoryDB const& _c) |
|
||||
{ |
|
||||
if (this == &_c) |
|
||||
return *this; |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(_c.x_this); |
|
||||
WriteGuard l2(x_this); |
|
||||
#endif |
|
||||
m_main = _c.m_main; |
|
||||
m_aux = _c.m_aux; |
|
||||
return *this; |
|
||||
} |
|
||||
|
|
||||
std::string MemoryDB::lookup(h256 const& _h) const |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
auto it = m_main.find(_h); |
|
||||
if (it != m_main.end()) |
|
||||
{ |
|
||||
if (!m_enforceRefs || it->second.second > 0) |
|
||||
return it->second.first; |
|
||||
else |
|
||||
cwarn << "Lookup required for value with refcount == 0. This is probably a critical trie issue" << _h; |
|
||||
} |
|
||||
return std::string(); |
|
||||
} |
|
||||
|
|
||||
bool MemoryDB::exists(h256 const& _h) const |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
auto it = m_main.find(_h); |
|
||||
if (it != m_main.end() && (!m_enforceRefs || it->second.second > 0)) |
|
||||
return true; |
|
||||
return false; |
|
||||
} |
|
||||
|
|
||||
void MemoryDB::insert(h256 const& _h, bytesConstRef _v) |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
WriteGuard l(x_this); |
|
||||
#endif |
|
||||
auto it = m_main.find(_h); |
|
||||
if (it != m_main.end()) |
|
||||
{ |
|
||||
it->second.first = _v.toString(); |
|
||||
it->second.second++; |
|
||||
} |
|
||||
else |
|
||||
m_main[_h] = make_pair(_v.toString(), 1); |
|
||||
#if ETH_PARANOIA |
|
||||
dbdebug << "INST" << _h << "=>" << m_main[_h].second; |
|
||||
#endif |
|
||||
} |
|
||||
|
|
||||
bool MemoryDB::kill(h256 const& _h) |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
if (m_main.count(_h)) |
|
||||
{ |
|
||||
if (m_main[_h].second > 0) |
|
||||
{ |
|
||||
m_main[_h].second--; |
|
||||
return true; |
|
||||
} |
|
||||
#if ETH_PARANOIA |
|
||||
else |
|
||||
{ |
|
||||
// If we get to this point, then there was probably a node in the level DB which we need to remove and which we have previously
|
|
||||
// used as part of the memory-based MemoryDB. Nothing to be worried about *as long as the node exists in the DB*.
|
|
||||
dbdebug << "NOKILL-WAS" << _h; |
|
||||
} |
|
||||
dbdebug << "KILL" << _h << "=>" << m_main[_h].second; |
|
||||
} |
|
||||
else |
|
||||
{ |
|
||||
dbdebug << "NOKILL" << _h; |
|
||||
#endif |
|
||||
} |
|
||||
return false; |
|
||||
} |
|
||||
|
|
||||
bytes MemoryDB::lookupAux(h256 const& _h) const |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
auto it = m_aux.find(_h); |
|
||||
if (it != m_aux.end() && (!m_enforceRefs || it->second.second)) |
|
||||
return it->second.first; |
|
||||
return bytes(); |
|
||||
} |
|
||||
|
|
||||
void MemoryDB::removeAux(h256 const& _h) |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
WriteGuard l(x_this); |
|
||||
#endif |
|
||||
m_aux[_h].second = false; |
|
||||
} |
|
||||
|
|
||||
void MemoryDB::insertAux(h256 const& _h, bytesConstRef _v) |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
WriteGuard l(x_this); |
|
||||
#endif |
|
||||
m_aux[_h] = make_pair(_v.toBytes(), true); |
|
||||
} |
|
||||
|
|
||||
void MemoryDB::purge() |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
WriteGuard l(x_this); |
|
||||
#endif |
|
||||
for (auto it = m_main.begin(); it != m_main.end(); ) |
|
||||
if (it->second.second) |
|
||||
++it; |
|
||||
else |
|
||||
it = m_main.erase(it); |
|
||||
} |
|
||||
|
|
||||
h256Hash MemoryDB::keys() const |
|
||||
{ |
|
||||
#if DEV_GUARDED_DB |
|
||||
ReadGuard l(x_this); |
|
||||
#endif |
|
||||
h256Hash ret; |
|
||||
for (auto const& i: m_main) |
|
||||
if (i.second.second) |
|
||||
ret.insert(i.first); |
|
||||
return ret; |
|
||||
} |
|
||||
|
|
||||
} |
|
@ -1,99 +0,0 @@ |
|||||
/*
|
|
||||
This file is part of cpp-ethereum. |
|
||||
|
|
||||
cpp-ethereum is free software: you can redistribute it and/or modify |
|
||||
it under the terms of the GNU General Public License as published by |
|
||||
the Free Software Foundation, either version 3 of the License, or |
|
||||
(at your option) any later version. |
|
||||
|
|
||||
cpp-ethereum is distributed in the hope that it will be useful, |
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
GNU General Public License for more details. |
|
||||
|
|
||||
You should have received a copy of the GNU General Public License |
|
||||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
||||
*/ |
|
||||
/** @file MemoryDB.h
|
|
||||
* @author Gav Wood <i@gavwood.com> |
|
||||
* @date 2014 |
|
||||
*/ |
|
||||
|
|
||||
#pragma once |
|
||||
|
|
||||
#include <unordered_map> |
|
||||
#include <libdevcore/Common.h> |
|
||||
#include <libdevcore/Guards.h> |
|
||||
#include <libdevcore/FixedHash.h> |
|
||||
#include <libdevcore/Log.h> |
|
||||
#include <libdevcore/RLP.h> |
|
||||
#include "SHA3.h" |
|
||||
|
|
||||
namespace dev |
|
||||
{ |
|
||||
|
|
||||
struct DBChannel: public LogChannel { static const char* name(); static const int verbosity = 18; }; |
|
||||
struct DBWarn: public LogChannel { static const char* name(); static const int verbosity = 1; }; |
|
||||
|
|
||||
#define dbdebug clog(DBChannel) |
|
||||
#define dbwarn clog(DBWarn) |
|
||||
|
|
||||
class MemoryDB |
|
||||
{ |
|
||||
friend class EnforceRefs; |
|
||||
|
|
||||
public: |
|
||||
MemoryDB() {} |
|
||||
MemoryDB(MemoryDB const& _c) { operator=(_c); } |
|
||||
|
|
||||
MemoryDB& operator=(MemoryDB const& _c); |
|
||||
|
|
||||
void clear() { m_main.clear(); } // WARNING !!!! didn't originally clear m_refCount!!!
|
|
||||
std::unordered_map<h256, std::string> get() const; |
|
||||
|
|
||||
std::string lookup(h256 const& _h) const; |
|
||||
bool exists(h256 const& _h) const; |
|
||||
void insert(h256 const& _h, bytesConstRef _v); |
|
||||
bool kill(h256 const& _h); |
|
||||
void purge(); |
|
||||
|
|
||||
bytes lookupAux(h256 const& _h) const; |
|
||||
void removeAux(h256 const& _h); |
|
||||
void insertAux(h256 const& _h, bytesConstRef _v); |
|
||||
|
|
||||
h256Hash keys() const; |
|
||||
|
|
||||
protected: |
|
||||
#if DEV_GUARDED_DB |
|
||||
mutable SharedMutex x_this; |
|
||||
#endif |
|
||||
std::unordered_map<h256, std::pair<std::string, unsigned>> m_main; |
|
||||
std::unordered_map<h256, std::pair<bytes, bool>> m_aux; |
|
||||
|
|
||||
mutable bool m_enforceRefs = false; |
|
||||
}; |
|
||||
|
|
||||
class EnforceRefs |
|
||||
{ |
|
||||
public: |
|
||||
EnforceRefs(MemoryDB const& _o, bool _r): m_o(_o), m_r(_o.m_enforceRefs) { _o.m_enforceRefs = _r; } |
|
||||
~EnforceRefs() { m_o.m_enforceRefs = m_r; } |
|
||||
|
|
||||
private: |
|
||||
MemoryDB const& m_o; |
|
||||
bool m_r; |
|
||||
}; |
|
||||
|
|
||||
inline std::ostream& operator<<(std::ostream& _out, MemoryDB const& _m) |
|
||||
{ |
|
||||
for (auto const& i: _m.get()) |
|
||||
{ |
|
||||
_out << i.first << ": "; |
|
||||
_out << RLP(i.second); |
|
||||
_out << " " << toHex(i.second); |
|
||||
_out << std::endl; |
|
||||
} |
|
||||
return _out; |
|
||||
} |
|
||||
|
|
||||
} |
|
@ -1,60 +0,0 @@ |
|||||
/*
|
|
||||
This file is part of cpp-ethereum. |
|
||||
|
|
||||
cpp-ethereum is free software: you can redistribute it and/or modify |
|
||||
it under the terms of the GNU General Public License as published by |
|
||||
the Free Software Foundation, either version 3 of the License, or |
|
||||
(at your option) any later version. |
|
||||
|
|
||||
cpp-ethereum is distributed in the hope that it will be useful, |
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
||||
GNU General Public License for more details. |
|
||||
|
|
||||
You should have received a copy of the GNU General Public License |
|
||||
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
||||
*/ |
|
||||
#pragma once |
|
||||
#include <utility> |
|
||||
#include <queue> |
|
||||
#include <condition_variable> |
|
||||
#include <mutex> |
|
||||
|
|
||||
|
|
||||
namespace dev |
|
||||
{ |
|
||||
|
|
||||
/// Concurrent queue.
|
|
||||
/// You can push and pop elements to/from the queue. Pop will block until the queue is not empty.
|
|
||||
/// The default backend (_QueueT) is std::queue. It can be changed to any type that has
|
|
||||
/// proper push(), pop(), empty() and front() methods.
|
|
||||
template<typename _T, typename _QueueT = std::queue<_T>> |
|
||||
class concurrent_queue |
|
||||
{ |
|
||||
public: |
|
||||
template<typename _U> |
|
||||
void push(_U&& _elem) |
|
||||
{ |
|
||||
{ |
|
||||
std::lock_guard<decltype(x_mutex)> guard{x_mutex}; |
|
||||
m_queue.push(std::forward<_U>(_elem)); |
|
||||
} |
|
||||
m_cv.notify_one(); |
|
||||
} |
|
||||
|
|
||||
_T pop() |
|
||||
{ |
|
||||
std::unique_lock<std::mutex> lock{x_mutex}; |
|
||||
m_cv.wait(lock, [this]{ return !m_queue.empty(); }); |
|
||||
auto item = std::move(m_queue.front()); |
|
||||
m_queue.pop(); |
|
||||
return item; |
|
||||
} |
|
||||
|
|
||||
private: |
|
||||
_QueueT m_queue; |
|
||||
std::mutex x_mutex; |
|
||||
std::condition_variable m_cv; |
|
||||
}; |
|
||||
|
|
||||
} |
|
@ -1,360 +0,0 @@ |
|||||
/*
|
|
||||
The MIT License (MIT) |
|
||||
|
|
||||
Copyright (C) 2014 okdshin |
|
||||
|
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy |
|
||||
of this software and associated documentation files (the "Software"), to deal |
|
||||
in the Software without restriction, including without limitation the rights |
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|
||||
copies of the Software, and to permit persons to whom the Software is |
|
||||
furnished to do so, subject to the following conditions: |
|
||||
|
|
||||
The above copyright notice and this permission notice shall be included in |
|
||||
all copies or substantial portions of the Software. |
|
||||
|
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
|
||||
THE SOFTWARE. |
|
||||
*/ |
|
||||
#ifndef PICOSHA2_H |
|
||||
#define PICOSHA2_H |
|
||||
//picosha2:20140213
|
|
||||
#include <cstdint> |
|
||||
#include <iostream> |
|
||||
#include <vector> |
|
||||
#include <iterator> |
|
||||
#include <cassert> |
|
||||
#include <sstream> |
|
||||
#include <algorithm> |
|
||||
|
|
||||
namespace picosha2 |
|
||||
{ |
|
||||
|
|
||||
namespace detail |
|
||||
{ |
|
||||
|
|
||||
inline uint8_t mask_8bit(uint8_t x){ |
|
||||
return x&0xff; |
|
||||
} |
|
||||
|
|
||||
inline uint32_t mask_32bit(uint32_t x){ |
|
||||
return x&0xffffffff; |
|
||||
} |
|
||||
|
|
||||
static const uint32_t add_constant[64] = { |
|
||||
0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, |
|
||||
0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5, |
|
||||
0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, |
|
||||
0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174, |
|
||||
0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, |
|
||||
0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da, |
|
||||
0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, |
|
||||
0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967, |
|
||||
0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, |
|
||||
0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85, |
|
||||
0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, |
|
||||
0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070, |
|
||||
0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, |
|
||||
0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3, |
|
||||
0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, |
|
||||
0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2 |
|
||||
}; |
|
||||
|
|
||||
static const uint32_t initial_message_digest[8] = { |
|
||||
0x6a09e667, 0xbb67ae85, 0x3c6ef372, 0xa54ff53a, |
|
||||
0x510e527f, 0x9b05688c, 0x1f83d9ab, 0x5be0cd19 |
|
||||
}; |
|
||||
|
|
||||
inline uint32_t ch(uint32_t x, uint32_t y, uint32_t z){ |
|
||||
return (x&y)^((~x)&z); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t maj(uint32_t x, uint32_t y, uint32_t z){ |
|
||||
return (x&y)^(x&z)^(y&z); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t rotr(uint32_t x, std::size_t n){ |
|
||||
assert(n < 32); |
|
||||
return mask_32bit((x>>n)|(x<<(32-n))); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t bsig0(uint32_t x){ |
|
||||
return rotr(x, 2)^rotr(x, 13)^rotr(x, 22); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t bsig1(uint32_t x){ |
|
||||
return rotr(x, 6)^rotr(x, 11)^rotr(x, 25); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t shr(uint32_t x, std::size_t n){ |
|
||||
assert(n < 32); |
|
||||
return x >> n; |
|
||||
} |
|
||||
|
|
||||
inline uint32_t ssig0(uint32_t x){ |
|
||||
return rotr(x, 7)^rotr(x, 18)^shr(x, 3); |
|
||||
} |
|
||||
|
|
||||
inline uint32_t ssig1(uint32_t x){ |
|
||||
return rotr(x, 17)^rotr(x, 19)^shr(x, 10); |
|
||||
} |
|
||||
|
|
||||
template<typename RaIter1, typename RaIter2> |
|
||||
void hash256_block(RaIter1 message_digest, RaIter2 first, RaIter2 last){ |
|
||||
(void)last; // FIXME: check this is valid
|
|
||||
uint32_t w[64]; |
|
||||
std::fill(w, w+64, 0); |
|
||||
for(std::size_t i = 0; i < 16; ++i){ |
|
||||
w[i] = (static_cast<uint32_t>(mask_8bit(*(first+i*4)))<<24) |
|
||||
|(static_cast<uint32_t>(mask_8bit(*(first+i*4+1)))<<16) |
|
||||
|(static_cast<uint32_t>(mask_8bit(*(first+i*4+2)))<<8) |
|
||||
|(static_cast<uint32_t>(mask_8bit(*(first+i*4+3)))); |
|
||||
} |
|
||||
for(std::size_t i = 16; i < 64; ++i){ |
|
||||
w[i] = mask_32bit(ssig1(w[i-2])+w[i-7]+ssig0(w[i-15])+w[i-16]); |
|
||||
} |
|
||||
|
|
||||
uint32_t a = *message_digest; |
|
||||
uint32_t b = *(message_digest+1); |
|
||||
uint32_t c = *(message_digest+2); |
|
||||
uint32_t d = *(message_digest+3); |
|
||||
uint32_t e = *(message_digest+4); |
|
||||
uint32_t f = *(message_digest+5); |
|
||||
uint32_t g = *(message_digest+6); |
|
||||
uint32_t h = *(message_digest+7); |
|
||||
|
|
||||
for(std::size_t i = 0; i < 64; ++i){ |
|
||||
uint32_t temp1 = h+bsig1(e)+ch(e,f,g)+add_constant[i]+w[i]; |
|
||||
uint32_t temp2 = bsig0(a)+maj(a,b,c); |
|
||||
h = g; |
|
||||
g = f; |
|
||||
f = e; |
|
||||
e = mask_32bit(d+temp1); |
|
||||
d = c; |
|
||||
c = b; |
|
||||
b = a; |
|
||||
a = mask_32bit(temp1+temp2); |
|
||||
} |
|
||||
*message_digest += a; |
|
||||
*(message_digest+1) += b; |
|
||||
*(message_digest+2) += c; |
|
||||
*(message_digest+3) += d; |
|
||||
*(message_digest+4) += e; |
|
||||
*(message_digest+5) += f; |
|
||||
*(message_digest+6) += g; |
|
||||
*(message_digest+7) += h; |
|
||||
for(std::size_t i = 0; i < 8; ++i){ |
|
||||
*(message_digest+i) = mask_32bit(*(message_digest+i)); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
}//namespace detail
|
|
||||
|
|
||||
template<typename InIter> |
|
||||
void output_hex(InIter first, InIter last, std::ostream& os){ |
|
||||
os.setf(std::ios::hex, std::ios::basefield); |
|
||||
while(first != last){ |
|
||||
os.width(2); |
|
||||
os.fill('0'); |
|
||||
os << static_cast<unsigned int>(*first); |
|
||||
++first; |
|
||||
} |
|
||||
os.setf(std::ios::dec, std::ios::basefield); |
|
||||
} |
|
||||
|
|
||||
template<typename InIter> |
|
||||
void bytes_to_hex_string(InIter first, InIter last, std::string& hex_str){ |
|
||||
std::ostringstream oss; |
|
||||
output_hex(first, last, oss); |
|
||||
hex_str.assign(oss.str()); |
|
||||
} |
|
||||
|
|
||||
template<typename InContainer> |
|
||||
void bytes_to_hex_string(const InContainer& bytes, std::string& hex_str){ |
|
||||
bytes_to_hex_string(bytes.begin(), bytes.end(), hex_str); |
|
||||
} |
|
||||
|
|
||||
template<typename InIter> |
|
||||
std::string bytes_to_hex_string(InIter first, InIter last){ |
|
||||
std::string hex_str; |
|
||||
bytes_to_hex_string(first, last, hex_str); |
|
||||
return hex_str; |
|
||||
} |
|
||||
|
|
||||
template<typename InContainer> |
|
||||
std::string bytes_to_hex_string(const InContainer& bytes){ |
|
||||
std::string hex_str; |
|
||||
bytes_to_hex_string(bytes, hex_str); |
|
||||
return hex_str; |
|
||||
} |
|
||||
|
|
||||
class hash256_one_by_one { |
|
||||
public: |
|
||||
hash256_one_by_one(){ |
|
||||
init(); |
|
||||
} |
|
||||
|
|
||||
void init(){ |
|
||||
buffer_.clear(); |
|
||||
std::fill(data_length_digits_, data_length_digits_+4, 0); |
|
||||
std::copy(detail::initial_message_digest, detail::initial_message_digest+8, h_); |
|
||||
} |
|
||||
|
|
||||
template<typename RaIter> |
|
||||
void process(RaIter first, RaIter last){ |
|
||||
add_to_data_length(std::distance(first, last)); |
|
||||
std::copy(first, last, std::back_inserter(buffer_)); |
|
||||
std::size_t i = 0; |
|
||||
for(;i+64 <= buffer_.size(); i+=64){ |
|
||||
detail::hash256_block(h_, buffer_.begin()+i, buffer_.begin()+i+64); |
|
||||
} |
|
||||
buffer_.erase(buffer_.begin(), buffer_.begin()+i); |
|
||||
} |
|
||||
|
|
||||
void finish(){ |
|
||||
uint8_t temp[64]; |
|
||||
std::fill(temp, temp+64, 0); |
|
||||
std::size_t remains = buffer_.size(); |
|
||||
std::copy(buffer_.begin(), buffer_.end(), temp); |
|
||||
temp[remains] = 0x80; |
|
||||
|
|
||||
if(remains > 55){ |
|
||||
std::fill(temp+remains+1, temp+64, 0); |
|
||||
detail::hash256_block(h_, temp, temp+64); |
|
||||
std::fill(temp, temp+64-4, 0); |
|
||||
} |
|
||||
else { |
|
||||
std::fill(temp+remains+1, temp+64-4, 0); |
|
||||
} |
|
||||
|
|
||||
write_data_bit_length(&(temp[56])); |
|
||||
detail::hash256_block(h_, temp, temp+64); |
|
||||
} |
|
||||
|
|
||||
template<typename OutIter> |
|
||||
void get_hash_bytes(OutIter first, OutIter last)const{ |
|
||||
for(const uint32_t* iter = h_; iter != h_+8; ++iter){ |
|
||||
for(std::size_t i = 0; i < 4 && first != last; ++i){ |
|
||||
*(first++) = detail::mask_8bit(static_cast<uint8_t>((*iter >> (24-8*i)))); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private: |
|
||||
void add_to_data_length(uint32_t n) { |
|
||||
uint32_t carry = 0; |
|
||||
data_length_digits_[0] += n; |
|
||||
for(std::size_t i = 0; i < 4; ++i) { |
|
||||
data_length_digits_[i] += carry; |
|
||||
if(data_length_digits_[i] >= 65536u) { |
|
||||
data_length_digits_[i] -= 65536u; |
|
||||
carry = 1; |
|
||||
} |
|
||||
else { |
|
||||
break; |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
void write_data_bit_length(uint8_t* begin) { |
|
||||
uint32_t data_bit_length_digits[4]; |
|
||||
std::copy( |
|
||||
data_length_digits_, data_length_digits_+4, |
|
||||
data_bit_length_digits |
|
||||
); |
|
||||
|
|
||||
// convert byte length to bit length (multiply 8 or shift 3 times left)
|
|
||||
uint32_t carry = 0; |
|
||||
for(std::size_t i = 0; i < 4; ++i) { |
|
||||
uint32_t before_val = data_bit_length_digits[i]; |
|
||||
data_bit_length_digits[i] <<= 3; |
|
||||
data_bit_length_digits[i] |= carry; |
|
||||
data_bit_length_digits[i] &= 65535u; |
|
||||
carry = (before_val >> (16-3)) & 65535u; |
|
||||
} |
|
||||
|
|
||||
// write data_bit_length
|
|
||||
for(int i = 3; i >= 0; --i) { |
|
||||
(*begin++) = static_cast<uint8_t>(data_bit_length_digits[i] >> 8); |
|
||||
(*begin++) = static_cast<uint8_t>(data_bit_length_digits[i]); |
|
||||
} |
|
||||
} |
|
||||
std::vector<uint8_t> buffer_; |
|
||||
uint32_t data_length_digits_[4]; //as 64bit integer (16bit x 4 integer)
|
|
||||
uint32_t h_[8]; |
|
||||
}; |
|
||||
|
|
||||
inline void get_hash_hex_string(const hash256_one_by_one& hasher, std::string& hex_str){ |
|
||||
uint8_t hash[32]; |
|
||||
hasher.get_hash_bytes(hash, hash+32); |
|
||||
return bytes_to_hex_string(hash, hash+32, hex_str); |
|
||||
} |
|
||||
|
|
||||
inline std::string get_hash_hex_string(const hash256_one_by_one& hasher){ |
|
||||
std::string hex_str; |
|
||||
get_hash_hex_string(hasher, hex_str); |
|
||||
return hex_str; |
|
||||
} |
|
||||
|
|
||||
template<typename RaIter, typename OutIter> |
|
||||
void hash256(RaIter first, RaIter last, OutIter first2, OutIter last2){ |
|
||||
hash256_one_by_one hasher; |
|
||||
//hasher.init();
|
|
||||
hasher.process(first, last); |
|
||||
hasher.finish(); |
|
||||
hasher.get_hash_bytes(first2, last2); |
|
||||
} |
|
||||
|
|
||||
template<typename RaIter, typename OutContainer> |
|
||||
void hash256(RaIter first, RaIter last, OutContainer& dst){ |
|
||||
hash256(first, last, dst.begin(), dst.end()); |
|
||||
} |
|
||||
|
|
||||
template<typename RaContainer, typename OutIter> |
|
||||
void hash256(const RaContainer& src, OutIter first, OutIter last){ |
|
||||
hash256(src.begin(), src.end(), first, last); |
|
||||
} |
|
||||
|
|
||||
template<typename RaContainer, typename OutContainer> |
|
||||
void hash256(const RaContainer& src, OutContainer& dst){ |
|
||||
hash256(src.begin(), src.end(), dst.begin(), dst.end()); |
|
||||
} |
|
||||
|
|
||||
|
|
||||
template<typename RaIter> |
|
||||
void hash256_hex_string(RaIter first, RaIter last, std::string& hex_str){ |
|
||||
uint8_t hashed[32]; |
|
||||
hash256(first, last, hashed, hashed+32); |
|
||||
std::ostringstream oss; |
|
||||
output_hex(hashed, hashed+32, oss); |
|
||||
hex_str.assign(oss.str()); |
|
||||
} |
|
||||
|
|
||||
template<typename RaIter> |
|
||||
std::string hash256_hex_string(RaIter first, RaIter last){ |
|
||||
std::string hex_str; |
|
||||
hash256_hex_string(first, last, hex_str); |
|
||||
return hex_str; |
|
||||
} |
|
||||
|
|
||||
inline void hash256_hex_string(const std::string& src, std::string& hex_str){ |
|
||||
hash256_hex_string(src.begin(), src.end(), hex_str); |
|
||||
} |
|
||||
|
|
||||
template<typename RaContainer> |
|
||||
void hash256_hex_string(const RaContainer& src, std::string& hex_str){ |
|
||||
hash256_hex_string(src.begin(), src.end(), hex_str); |
|
||||
} |
|
||||
|
|
||||
template<typename RaContainer> |
|
||||
std::string hash256_hex_string(const RaContainer& src){ |
|
||||
return hash256_hex_string(src.begin(), src.end()); |
|
||||
} |
|
||||
|
|
||||
}//namespace picosha2
|
|
||||
|
|
||||
#endif //PICOSHA2_H
|
|
Loading…
Reference in new issue