455 lines
13 KiB
C++
Executable file
455 lines
13 KiB
C++
Executable file
// Copyright (c) 2011-2016 The Cryptonote developers
|
|
// Distributed under the MIT/X11 software license, see the accompanying
|
|
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
#include "Common/StdInputStream.h"
|
|
#include "Common/StdOutputStream.h"
|
|
#include "Serialization/BinaryInputStreamSerializer.h"
|
|
#include "Serialization/BinaryOutputStreamSerializer.h"
|
|
#include "Serialization/BinarySerializationTools.h"
|
|
|
|
using namespace Common;
|
|
using namespace CryptoNote;
|
|
|
|
TEST(BinarySerializer, uint16) {
|
|
|
|
std::stringstream ss;
|
|
|
|
uint16_t u16 = 0xfeff;
|
|
uint32_t u32 = 0x3fddfd48;
|
|
|
|
{
|
|
StdOutputStream os(ss);
|
|
BinaryOutputStreamSerializer s(os);
|
|
s(u32, "u32");
|
|
s(u16, "u16");
|
|
}
|
|
|
|
{
|
|
StdInputStream is(ss);
|
|
BinaryInputStreamSerializer s(is);
|
|
|
|
uint32_t t32 = 0;
|
|
uint16_t t16 = 0;
|
|
|
|
s(t32, "u32");
|
|
s(t16, "u16");
|
|
|
|
ASSERT_EQ(u32, t32);
|
|
ASSERT_EQ(u16, t16);
|
|
}
|
|
}
|
|
|
|
|
|
//#include <cstring>
|
|
//#include <cstdint>
|
|
//#include <cstdio>
|
|
//#include <iostream>
|
|
//#include <vector>
|
|
//#include <boost/foreach.hpp>
|
|
//#include "CryptoNoteCore/CryptoNoteBasic.h"
|
|
//#include "CryptoNoteCore/CryptoNoteBasicImpl.h"
|
|
//#include "Serialization/serialization.h"
|
|
//#include "Serialization/binary_archive.h"
|
|
//#include "Serialization/json_archive.h"
|
|
//#include "Serialization/variant.h"
|
|
//#include "Serialization/vector.h"
|
|
//#include "Serialization/binary_utils.h"
|
|
//#include "gtest/gtest.h"
|
|
//using namespace std;
|
|
//
|
|
//struct Struct
|
|
//{
|
|
// int32_t a;
|
|
// int32_t b;
|
|
// char blob[8];
|
|
//};
|
|
//
|
|
//template <class Archive>
|
|
//struct serializer<Archive, Struct>
|
|
//{
|
|
// static bool serialize(Archive &ar, Struct &s) {
|
|
// ar.begin_object();
|
|
// ar.tag("a");
|
|
// ar.serialize_int(s.a);
|
|
// ar.tag("b");
|
|
// ar.serialize_int(s.b);
|
|
// ar.tag("blob");
|
|
// ar.serialize_blob(s.blob, sizeof(s.blob));
|
|
// ar.end_object();
|
|
// return true;
|
|
// }
|
|
//};
|
|
//
|
|
//struct Struct1
|
|
//{
|
|
// vector<boost::variant<Struct, int32_t>> si;
|
|
// vector<int16_t> vi;
|
|
//
|
|
// BEGIN_SERIALIZE_OBJECT()
|
|
// FIELD(si)
|
|
// FIELD(vi)
|
|
// END_SERIALIZE()
|
|
// /*template <bool W, template <bool> class Archive>
|
|
// bool do_serialize(Archive<W> &ar)
|
|
// {
|
|
// ar.begin_object();
|
|
// ar.tag("si");
|
|
// ::do_serialize(ar, si);
|
|
// ar.tag("vi");
|
|
// ::do_serialize(ar, vi);
|
|
// ar.end_object();
|
|
// }*/
|
|
//};
|
|
//
|
|
//struct Blob
|
|
//{
|
|
// uint64_t a;
|
|
// uint32_t b;
|
|
//
|
|
// bool operator==(const Blob& rhs) const
|
|
// {
|
|
// return a == rhs.a;
|
|
// }
|
|
//};
|
|
//
|
|
//VARIANT_TAG(binary_archive, Struct, 0xe0);
|
|
//VARIANT_TAG(binary_archive, int, 0xe1);
|
|
//VARIANT_TAG(json_archive, Struct, "struct");
|
|
//VARIANT_TAG(json_archive, int, "int");
|
|
//
|
|
//BLOB_SERIALIZER(Blob);
|
|
//
|
|
//bool try_parse(const string &blob)
|
|
//{
|
|
// Struct1 s1;
|
|
// return serialization::parse_binary(blob, s1);
|
|
//}
|
|
//
|
|
//TEST(Serialization, BinaryArchiveInts) {
|
|
// uint64_t x = 0xff00000000, x1;
|
|
//
|
|
// ostringstream oss;
|
|
// binary_archive<true> oar(oss);
|
|
// oar.serialize_int(x);
|
|
// ASSERT_TRUE(oss.good());
|
|
// ASSERT_EQ(8, oss.str().size());
|
|
// ASSERT_EQ(string("\0\0\0\0\xff\0\0\0", 8), oss.str());
|
|
//
|
|
// istringstream iss(oss.str());
|
|
// binary_archive<false> iar(iss);
|
|
// iar.serialize_int(x1);
|
|
// ASSERT_EQ(8, iss.tellg());
|
|
// ASSERT_TRUE(iss.good());
|
|
//
|
|
// ASSERT_EQ(x, x1);
|
|
//}
|
|
//
|
|
//TEST(Serialization, BinaryArchiveVarInts) {
|
|
// uint64_t x = 0xff00000000, x1;
|
|
//
|
|
// ostringstream oss;
|
|
// binary_archive<true> oar(oss);
|
|
// oar.serialize_varint(x);
|
|
// ASSERT_TRUE(oss.good());
|
|
// ASSERT_EQ(6, oss.str().size());
|
|
// ASSERT_EQ(string("\x80\x80\x80\x80\xF0\x1F", 6), oss.str());
|
|
//
|
|
// istringstream iss(oss.str());
|
|
// binary_archive<false> iar(iss);
|
|
// iar.serialize_varint(x1);
|
|
// ASSERT_TRUE(iss.good());
|
|
// ASSERT_EQ(x, x1);
|
|
//}
|
|
//
|
|
//TEST(Serialization, Test1) {
|
|
// ostringstream str;
|
|
// binary_archive<true> ar(str);
|
|
//
|
|
// Struct1 s1;
|
|
// s1.si.push_back(0);
|
|
// {
|
|
// Struct s;
|
|
// s.a = 5;
|
|
// s.b = 65539;
|
|
// std::memcpy(s.blob, "12345678", 8);
|
|
// s1.si.push_back(s);
|
|
// }
|
|
// s1.si.push_back(1);
|
|
// s1.vi.push_back(10);
|
|
// s1.vi.push_back(22);
|
|
//
|
|
// string blob;
|
|
// ASSERT_TRUE(serialization::dump_binary(s1, blob));
|
|
// ASSERT_TRUE(try_parse(blob));
|
|
//
|
|
// ASSERT_EQ('\xE0', blob[6]);
|
|
// blob[6] = '\xE1';
|
|
// ASSERT_FALSE(try_parse(blob));
|
|
// blob[6] = '\xE2';
|
|
// ASSERT_FALSE(try_parse(blob));
|
|
//}
|
|
//
|
|
//TEST(Serialization, Overflow) {
|
|
// Blob x = { 0xff00000000 };
|
|
// Blob x1;
|
|
//
|
|
// string blob;
|
|
// ASSERT_TRUE(serialization::dump_binary(x, blob));
|
|
// ASSERT_EQ(sizeof(Blob), blob.size());
|
|
//
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, x1));
|
|
// ASSERT_EQ(x, x1);
|
|
//
|
|
// vector<Blob> bigvector;
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, bigvector));
|
|
// ASSERT_EQ(0, bigvector.size());
|
|
//}
|
|
//
|
|
//TEST(Serialization, serializes_vector_uint64_as_varint)
|
|
//{
|
|
// std::vector<uint64_t> v;
|
|
// string blob;
|
|
//
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(1, blob.size());
|
|
//
|
|
// // +1 byte
|
|
// v.push_back(0);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(2, blob.size());
|
|
//
|
|
// // +1 byte
|
|
// v.push_back(1);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(3, blob.size());
|
|
//
|
|
// // +2 bytes
|
|
// v.push_back(0x80);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(5, blob.size());
|
|
//
|
|
// // +2 bytes
|
|
// v.push_back(0xFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(7, blob.size());
|
|
//
|
|
// // +2 bytes
|
|
// v.push_back(0x3FFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(9, blob.size());
|
|
//
|
|
// // +3 bytes
|
|
// v.push_back(0x40FF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(12, blob.size());
|
|
//
|
|
// // +10 bytes
|
|
// v.push_back(0xFFFFFFFFFFFFFFFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(22, blob.size());
|
|
//}
|
|
//
|
|
//TEST(Serialization, serializes_vector_int64_as_fixed_int)
|
|
//{
|
|
// std::vector<int64_t> v;
|
|
// string blob;
|
|
//
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(1, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(9, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(1);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(17, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0x80);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(25, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0xFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(33, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0x3FFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(41, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0x40FF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(49, blob.size());
|
|
//
|
|
// // +8 bytes
|
|
// v.push_back(0xFFFFFFFFFFFFFFFF);
|
|
// ASSERT_TRUE(serialization::dump_binary(v, blob));
|
|
// ASSERT_EQ(57, blob.size());
|
|
//}
|
|
//
|
|
//namespace
|
|
//{
|
|
// template<typename T>
|
|
// std::vector<T> linearize_vector2(const std::vector< std::vector<T> >& vec_vec)
|
|
// {
|
|
// std::vector<T> res;
|
|
// BOOST_FOREACH(const auto& vec, vec_vec)
|
|
// {
|
|
// res.insert(res.end(), vec.begin(), vec.end());
|
|
// }
|
|
// return res;
|
|
// }
|
|
//}
|
|
//
|
|
//TEST(Serialization, serializes_transacion_signatures_correctly)
|
|
//{
|
|
// using namespace CryptoNote;
|
|
//
|
|
// Transaction tx;
|
|
// Transaction tx1;
|
|
// string blob;
|
|
//
|
|
// // Empty tx
|
|
// tx.clear();
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_EQ(5, blob.size()); // 5 bytes + 0 bytes extra + 0 bytes signatures
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Miner tx without signatures
|
|
// TransactionInputGenerate txin_gen1;
|
|
// txin_gen1.height = 0;
|
|
// tx.clear();
|
|
// tx.vin.push_back(txin_gen1);
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_EQ(7, blob.size()); // 5 bytes + 2 bytes vin[0] + 0 bytes extra + 0 bytes signatures
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Miner tx with empty signatures 2nd vector
|
|
// tx.signatures.resize(1);
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_EQ(7, blob.size()); // 5 bytes + 2 bytes vin[0] + 0 bytes extra + 0 bytes signatures
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Miner tx with one signature
|
|
// tx.signatures[0].resize(1);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Miner tx with 2 empty vectors
|
|
// tx.signatures.resize(2);
|
|
// tx.signatures[0].resize(0);
|
|
// tx.signatures[1].resize(0);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Miner tx with 2 signatures
|
|
// tx.signatures[0].resize(1);
|
|
// tx.signatures[1].resize(1);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Two TransactionInputGenerate, no signatures
|
|
// tx.vin.push_back(txin_gen1);
|
|
// tx.signatures.resize(0);
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_EQ(9, blob.size()); // 5 bytes + 2 * 2 bytes vins + 0 bytes extra + 0 bytes signatures
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Two TransactionInputGenerate, signatures vector contains only one empty element
|
|
// tx.signatures.resize(1);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Two TransactionInputGenerate, signatures vector contains two empty elements
|
|
// tx.signatures.resize(2);
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_EQ(9, blob.size()); // 5 bytes + 2 * 2 bytes vins + 0 bytes extra + 0 bytes signatures
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Two TransactionInputGenerate, signatures vector contains three empty elements
|
|
// tx.signatures.resize(3);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Two TransactionInputGenerate, signatures vector contains two non empty elements
|
|
// tx.signatures.resize(2);
|
|
// tx.signatures[0].resize(1);
|
|
// tx.signatures[1].resize(1);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // A few bytes instead of signature
|
|
// tx.vin.clear();
|
|
// tx.vin.push_back(txin_gen1);
|
|
// tx.signatures.clear();
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// blob.append(std::string(sizeof(Crypto::Signature) / 2, 'x'));
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, tx1));
|
|
//
|
|
// // blob contains one signature
|
|
// blob.append(std::string(sizeof(Crypto::Signature) / 2, 'y'));
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, tx1));
|
|
//
|
|
// // Not enough signature vectors for all inputs
|
|
// TransactionInputToKey txin_to_key1;
|
|
// txin_to_key1.keyOffsets.resize(2);
|
|
// tx.vin.clear();
|
|
// tx.vin.push_back(txin_to_key1);
|
|
// tx.vin.push_back(txin_to_key1);
|
|
// tx.signatures.resize(1);
|
|
// tx.signatures[0].resize(2);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // Too much signatures for two inputs
|
|
// tx.signatures.resize(3);
|
|
// tx.signatures[0].resize(2);
|
|
// tx.signatures[1].resize(2);
|
|
// tx.signatures[2].resize(2);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // First signatures vector contains too little elements
|
|
// tx.signatures.resize(2);
|
|
// tx.signatures[0].resize(1);
|
|
// tx.signatures[1].resize(2);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // First signatures vector contains too much elements
|
|
// tx.signatures.resize(2);
|
|
// tx.signatures[0].resize(3);
|
|
// tx.signatures[1].resize(2);
|
|
// ASSERT_FALSE(serialization::dump_binary(tx, blob));
|
|
//
|
|
// // There are signatures for each input
|
|
// tx.signatures.resize(2);
|
|
// tx.signatures[0].resize(2);
|
|
// tx.signatures[1].resize(2);
|
|
// ASSERT_TRUE(serialization::dump_binary(tx, blob));
|
|
// ASSERT_TRUE(serialization::parse_binary(blob, tx1));
|
|
// ASSERT_EQ(tx, tx1);
|
|
// ASSERT_EQ(linearize_vector2(tx.signatures), linearize_vector2(tx1.signatures));
|
|
//
|
|
// // Blob doesn't contain enough data
|
|
// blob.resize(blob.size() - sizeof(Crypto::Signature) / 2);
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, tx1));
|
|
//
|
|
// // Blob contains too much data
|
|
// blob.resize(blob.size() + sizeof(Crypto::Signature));
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, tx1));
|
|
//
|
|
// // Blob contains one excess signature
|
|
// blob.resize(blob.size() + sizeof(Crypto::Signature) / 2);
|
|
// ASSERT_FALSE(serialization::parse_binary(blob, tx1));
|
|
//}
|