Gav Wood
10 years ago
14 changed files with 478 additions and 82 deletions
File diff suppressed because one or more lines are too long
@ -0,0 +1,100 @@ |
|||
/*
|
|||
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 May 2015 |
|||
*/ |
|||
|
|||
#include <boost/test/unit_test.hpp> |
|||
#include <libwhisper/Message.h> |
|||
|
|||
using namespace std; |
|||
using namespace dev; |
|||
using namespace dev::shh; |
|||
|
|||
struct VerbosityHolder |
|||
{ |
|||
VerbosityHolder(int _temporaryValue) : oldLogVerbosity(g_logVerbosity) { g_logVerbosity = _temporaryValue; } |
|||
~VerbosityHolder() { g_logVerbosity = oldLogVerbosity; } |
|||
|
|||
int oldLogVerbosity; |
|||
}; |
|||
|
|||
FullTopic createRandomTopics(unsigned int i) |
|||
{ |
|||
FullTopic ret; |
|||
h256 t(i); |
|||
|
|||
for (int j = 0; j < 8; ++j) |
|||
{ |
|||
t = sha3(t); |
|||
ret.push_back(t); |
|||
} |
|||
|
|||
return move(ret); |
|||
} |
|||
|
|||
bytes createRandomPayload(unsigned int i) |
|||
{ |
|||
bytes ret; |
|||
srand(i); |
|||
int const sz = rand() % 1024; |
|||
for (int j = 0; j < sz; ++j) |
|||
ret.push_back(rand() % 256); |
|||
|
|||
return move(ret); |
|||
} |
|||
|
|||
void comparePayloads(Message const& m1, Message const& m2) |
|||
{ |
|||
bytes const& p1 = m1.payload(); |
|||
bytes const& p2 = m2.payload(); |
|||
BOOST_REQUIRE_EQUAL(p1.size(), p2.size()); |
|||
|
|||
for (size_t i = 0; i < p1.size(); ++i) |
|||
BOOST_REQUIRE_EQUAL(p1[i], p2[i]); |
|||
} |
|||
|
|||
void sealAndOpenSingleMessage(unsigned int i) |
|||
{ |
|||
Secret zero; |
|||
FullTopic topics = createRandomTopics(i); |
|||
bytes const payload = createRandomPayload(i); |
|||
Message m1(payload); |
|||
Envelope e = m1.seal(zero, topics, 1, 1); |
|||
|
|||
for (auto const& t: topics) |
|||
{ |
|||
FullTopic singleTopic; |
|||
singleTopic.push_back(t); |
|||
Message m2(e, singleTopic, zero); |
|||
comparePayloads(m1, m2); |
|||
} |
|||
} |
|||
|
|||
BOOST_AUTO_TEST_SUITE(whisperMessage) |
|||
|
|||
BOOST_AUTO_TEST_CASE(seal) |
|||
{ |
|||
VerbosityHolder setTemporaryLevel(10); |
|||
cnote << "Testing Envelope encryption..."; |
|||
|
|||
for (unsigned int i = 1; i < 10; ++i) |
|||
sealAndOpenSingleMessage(i); |
|||
} |
|||
|
|||
BOOST_AUTO_TEST_SUITE_END() |
|||
Loading…
Reference in new issue