Gav Wood
10 years ago
4 changed files with 266 additions and 16 deletions
@ -0,0 +1,69 @@ |
|||
/*
|
|||
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 WhisperDB.cpp
|
|||
* @author Vladislav Gluhovsky <vlad@ethdev.com> |
|||
* @date July 2015 |
|||
*/ |
|||
|
|||
#include "WhisperDB.h" |
|||
#include <boost/filesystem.hpp> |
|||
#include <libdevcore/FileSystem.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev; |
|||
using namespace dev::shh; |
|||
|
|||
WhisperDB::WhisperDB() |
|||
{ |
|||
string path = dev::getDataDir("shh"); |
|||
boost::filesystem::create_directories(path); |
|||
leveldb::Options op; |
|||
op.create_if_missing = true; |
|||
op.max_open_files = 256; |
|||
leveldb::DB* p = nullptr; |
|||
leveldb::Status status = leveldb::DB::Open(op, path + "/messages", &p); |
|||
m_db.reset(p); |
|||
if (!status.ok()) |
|||
BOOST_THROW_EXCEPTION(FailedToOpenLevelDB(status.ToString())); |
|||
} |
|||
|
|||
string WhisperDB::lookup(dev::h256 const& _key) const |
|||
{ |
|||
string ret; |
|||
leveldb::Slice slice((char const*)_key.data(), _key.size); |
|||
leveldb::Status status = m_db->Get(m_readOptions, slice, &ret); |
|||
if (!status.ok() && !status.IsNotFound()) |
|||
BOOST_THROW_EXCEPTION(FailedLookupInLevelDB(status.ToString())); |
|||
|
|||
return ret; |
|||
} |
|||
|
|||
void WhisperDB::insert(dev::h256 const& _key, string const& _value) |
|||
{ |
|||
leveldb::Slice slice((char const*)_key.data(), _key.size); |
|||
leveldb::Status status = m_db->Put(m_writeOptions, slice, _value); |
|||
if (!status.ok()) |
|||
BOOST_THROW_EXCEPTION(FailedInsertInLevelDB(status.ToString())); |
|||
} |
|||
|
|||
void WhisperDB::kill(dev::h256 const& _key) |
|||
{ |
|||
leveldb::Slice slice((char const*)_key.data(), _key.size); |
|||
leveldb::Status status = m_db->Delete(m_writeOptions, slice); |
|||
if (!status.ok()) |
|||
BOOST_THROW_EXCEPTION(FailedDeleteInLevelDB(status.ToString())); |
|||
} |
@ -0,0 +1,55 @@ |
|||
/*
|
|||
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 WhisperDB.h
|
|||
* @author Vladislav Gluhovsky <vlad@ethdev.com> |
|||
* @date July 2015 |
|||
*/ |
|||
|
|||
#pragma once |
|||
|
|||
#include <libdevcore/db.h> |
|||
#include <libdevcore/FixedHash.h> |
|||
#include "Common.h" |
|||
|
|||
namespace dev |
|||
{ |
|||
namespace shh |
|||
{ |
|||
|
|||
struct FailedToOpenLevelDB: virtual Exception { FailedToOpenLevelDB(std::string const& _message): Exception(_message) {} }; |
|||
struct FailedInsertInLevelDB: virtual Exception { FailedInsertInLevelDB(std::string const& _message): Exception(_message) {} }; |
|||
struct FailedLookupInLevelDB: virtual Exception { FailedLookupInLevelDB(std::string const& _message): Exception(_message) {} }; |
|||
struct FailedDeleteInLevelDB: virtual Exception { FailedDeleteInLevelDB(std::string const& _message): Exception(_message) {} }; |
|||
|
|||
class WhisperDB |
|||
{ |
|||
public: |
|||
WhisperDB(); |
|||
~WhisperDB() {} |
|||
|
|||
std::string lookup(dev::h256 const& _key) const; |
|||
void insert(dev::h256 const& _key, std::string const& _value); |
|||
void kill(dev::h256 const& _key); |
|||
|
|||
private: |
|||
leveldb::ReadOptions m_readOptions; |
|||
leveldb::WriteOptions m_writeOptions; |
|||
std::unique_ptr<leveldb::DB> m_db; |
|||
}; |
|||
|
|||
} |
|||
} |
@ -0,0 +1,122 @@ |
|||
/*
|
|||
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 whisperMessage.cpp
|
|||
* @author Vladislav Gluhovsky <vlad@ethdev.com> |
|||
* @date July 2015 |
|||
*/ |
|||
|
|||
#include <thread> |
|||
#include <boost/test/unit_test.hpp> |
|||
#include <libwhisper/WhisperDB.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev; |
|||
using namespace dev::shh; |
|||
|
|||
BOOST_AUTO_TEST_SUITE(whisperDB) |
|||
|
|||
BOOST_AUTO_TEST_CASE(basic) |
|||
{ |
|||
VerbosityHolder setTemporaryLevel(10); |
|||
cnote << "Testing Whisper DB..."; |
|||
|
|||
string s; |
|||
string const text1 = "lorem"; |
|||
string const text2 = "ipsum"; |
|||
h256 h1(0xBEEF); |
|||
h256 h2(0xC0FFEE); |
|||
WhisperDB db; |
|||
|
|||
db.kill(h1); |
|||
db.kill(h2); |
|||
|
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(s.empty()); |
|||
|
|||
db.insert(h1, text2); |
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(s.empty()); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(!s.compare(text2)); |
|||
|
|||
db.insert(h1, text1); |
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(s.empty()); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(!s.compare(text1)); |
|||
|
|||
db.insert(h2, text2); |
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(!s.compare(text2)); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(!s.compare(text1)); |
|||
|
|||
db.kill(h1); |
|||
db.kill(h2); |
|||
|
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(s.empty()); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(s.empty()); |
|||
} |
|||
|
|||
BOOST_AUTO_TEST_CASE(persistence) |
|||
{ |
|||
VerbosityHolder setTemporaryLevel(10); |
|||
cnote << "Testing persistence of Whisper DB..."; |
|||
|
|||
string s; |
|||
string const text1 = "sator"; |
|||
string const text2 = "arepo"; |
|||
h256 const h1(0x12345678); |
|||
h256 const h2(0xBADD00DE); |
|||
|
|||
{ |
|||
WhisperDB db; |
|||
db.kill(h1); |
|||
db.kill(h2); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(s.empty()); |
|||
db.insert(h1, text2); |
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(s.empty()); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(!s.compare(text2)); |
|||
} |
|||
|
|||
this_thread::sleep_for(chrono::milliseconds(20)); |
|||
|
|||
{ |
|||
WhisperDB db; |
|||
db.insert(h1, text1); |
|||
db.insert(h2, text2); |
|||
} |
|||
|
|||
this_thread::sleep_for(chrono::milliseconds(20)); |
|||
|
|||
{ |
|||
WhisperDB db; |
|||
s = db.lookup(h2); |
|||
BOOST_REQUIRE(!s.compare(text2)); |
|||
s = db.lookup(h1); |
|||
BOOST_REQUIRE(!s.compare(text1)); |
|||
db.kill(h1); |
|||
db.kill(h2); |
|||
} |
|||
} |
|||
|
|||
BOOST_AUTO_TEST_SUITE_END() |
Loading…
Reference in new issue