2015-05-27 12:08:46 +00:00
|
|
|
// Copyright (c) 2012-2015, The CryptoNote developers, The Bytecoin developers
|
2014-09-15 17:33:47 +00:00
|
|
|
//
|
|
|
|
// This file is part of Bytecoin.
|
|
|
|
//
|
|
|
|
// Bytecoin is free software: you can redistribute it and/or modify
|
|
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
|
|
// (at your option) any later version.
|
|
|
|
//
|
|
|
|
// Bytecoin 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 Lesser General Public License for more details.
|
|
|
|
//
|
|
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
|
|
// along with Bytecoin. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
#include "BinaryInputStreamSerializer.h"
|
|
|
|
#include "SerializationOverloads.h"
|
|
|
|
|
|
|
|
#include <algorithm>
|
|
|
|
#include <cassert>
|
|
|
|
#include <stdexcept>
|
|
|
|
|
|
|
|
namespace {
|
|
|
|
|
2015-04-06 16:13:07 +00:00
|
|
|
template<typename T, int bits = std::numeric_limits<T>::digits>
|
|
|
|
typename std::enable_if<std::is_integral<T>::value && std::is_unsigned<T>::value, size_t>::type
|
|
|
|
readVarint(std::istream& s, T &i) {
|
|
|
|
size_t read = 0;
|
|
|
|
i = 0;
|
|
|
|
for (int shift = 0;; shift += 7) {
|
|
|
|
if (s.eof()) {
|
|
|
|
return read;
|
|
|
|
}
|
|
|
|
uint8_t byte = s.get();
|
|
|
|
if (!s) {
|
|
|
|
throw std::runtime_error("Stream read error");
|
|
|
|
}
|
|
|
|
++read;
|
|
|
|
if (shift + 7 >= bits && byte >= 1 << (bits - shift)) {
|
|
|
|
throw std::runtime_error("Varint overflow");
|
|
|
|
}
|
|
|
|
if (byte == 0 && shift != 0) {
|
|
|
|
throw std::runtime_error("Non-canonical varint representation");
|
|
|
|
}
|
|
|
|
i |= static_cast<T>(byte & 0x7f) << shift;
|
|
|
|
if ((byte & 0x80) == 0) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return read;
|
2014-09-15 17:33:47 +00:00
|
|
|
}
|
|
|
|
|
2015-04-06 16:13:07 +00:00
|
|
|
template<typename StorageType, typename T>
|
|
|
|
void readVarintAs(std::istream& s, T &i) {
|
|
|
|
StorageType v;
|
|
|
|
readVarint(s, v);
|
|
|
|
i = static_cast<T>(v);
|
2014-09-15 17:33:47 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2015-05-27 12:08:46 +00:00
|
|
|
namespace CryptoNote {
|
2014-09-15 17:33:47 +00:00
|
|
|
|
|
|
|
ISerializer::SerializerType BinaryInputStreamSerializer::type() const {
|
|
|
|
return ISerializer::INPUT;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::beginObject(const std::string& name) {
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::endObject() {
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::beginArray(std::size_t& size, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarintAs<uint64_t>(stream, size);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::endArray() {
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(uint8_t& value, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarint(stream, value);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(uint32_t& value, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarint(stream, value);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(int32_t& value, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarintAs<uint32_t>(stream, value);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(int64_t& value, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarintAs<uint64_t>(stream, value);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(uint64_t& value, const std::string& name) {
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarint(stream, value);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(bool& value, const std::string& name) {
|
2015-05-27 12:08:46 +00:00
|
|
|
value = stream.get() != 0;
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(std::string& value, const std::string& name) {
|
|
|
|
uint64_t size;
|
2015-04-06 16:13:07 +00:00
|
|
|
readVarint(stream, size);
|
|
|
|
|
|
|
|
if (size > 0) {
|
|
|
|
std::vector<char> temp;
|
|
|
|
temp.resize(size);
|
|
|
|
checkedRead(&temp[0], size);
|
|
|
|
value.reserve(size);
|
|
|
|
value.assign(&temp[0], size);
|
|
|
|
} else {
|
|
|
|
value.clear();
|
|
|
|
}
|
2014-09-15 17:33:47 +00:00
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2015-04-06 16:13:07 +00:00
|
|
|
ISerializer& BinaryInputStreamSerializer::binary(void* value, std::size_t size, const std::string& name) {
|
|
|
|
stream.read(static_cast<char*>(value), size);
|
2014-09-15 17:33:47 +00:00
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2015-04-06 16:13:07 +00:00
|
|
|
ISerializer& BinaryInputStreamSerializer::binary(std::string& value, const std::string& name) {
|
|
|
|
return (*this)(value, name);
|
2014-09-15 17:33:47 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool BinaryInputStreamSerializer::hasObject(const std::string& name) {
|
|
|
|
assert(false); //the method is not supported for this type of serialization
|
|
|
|
throw std::runtime_error("hasObject method is not supported in BinaryInputStreamSerializer");
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
ISerializer& BinaryInputStreamSerializer::operator()(double& value, const std::string& name) {
|
|
|
|
assert(false); //the method is not supported for this type of serialization
|
|
|
|
throw std::runtime_error("double serialization is not supported in BinaryInputStreamSerializer");
|
|
|
|
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2015-04-06 16:13:07 +00:00
|
|
|
void BinaryInputStreamSerializer::checkedRead(char* buf, size_t size) {
|
|
|
|
stream.read(buf, size);
|
|
|
|
if (!stream) {
|
|
|
|
throw std::runtime_error("Stream read error");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-09-15 17:33:47 +00:00
|
|
|
}
|