From 296ae46ed8f8f6e5f986f978febad302e3df231a Mon Sep 17 00:00:00 2001 From: Antonio Juarez Date: Mon, 3 Mar 2014 22:07:58 +0000 Subject: moved all stuff to github --- tests/io.h | 107 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 107 insertions(+) create mode 100644 tests/io.h (limited to 'tests/io.h') diff --git a/tests/io.h b/tests/io.h new file mode 100644 index 000000000..64f2bae77 --- /dev/null +++ b/tests/io.h @@ -0,0 +1,107 @@ +// Copyright (c) 2012-2013 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 +#include +#include +#include +#include + +inline bool hexdecode(const char *from, std::size_t length, void *to) { + std::size_t i; + for (i = 0; i < length; i++) { + int v = 0; + if (from[2 * i] >= '0' && from[2 * i] <= '9') { + v = from[2 * i] - '0'; + } else if (from[2 * i] >= 'a' && from[2 * i] <= 'f') { + v = from[2 * i] - 'a' + 10; + } else { + return false; + } + v <<= 4; + if (from[2 * i + 1] >= '0' && from[2 * i + 1] <= '9') { + v |= from[2 * i + 1] - '0'; + } else if (from[2 * i + 1] >= 'a' && from[2 * i + 1] <= 'f') { + v |= from[2 * i + 1] - 'a' + 10; + } else { + return false; + } + *(reinterpret_cast(to) + i) = v; + } + return true; +} + +inline void get(std::istream &input, bool &res) { + std::string sres; + input >> sres; + if (sres == "false") { + res = false; + } else if (sres == "true") { + res = true; + } else { + input.setstate(std::ios_base::failbit); + } +} + +template +typename std::enable_if::value, void>::type +get(std::istream &input, T &res) { + input >> res; +} + +inline void getvar(std::istream &input, std::size_t length, void *res) { + std::string sres; + input >> sres; + if (sres.length() != 2 * length || !hexdecode(sres.data(), length, res)) { + input.setstate(std::ios_base::failbit); + } +} + +template +typename std::enable_if::value && !std::is_scalar::value, void>::type +get(std::istream &input, T &res) { + getvar(input, sizeof(T), &res); +} + +inline void get(std::istream &input, std::vector &res) { + std::string sres; + input >> sres; + if (sres == "x") { + res.clear(); + } else if (sres.length() % 2 != 0) { + input.setstate(std::ios_base::failbit); + } else { + std::size_t length = sres.length() / 2; + res.resize(length); + if (!hexdecode(sres.data(), length, res.data())) { + input.setstate(std::ios_base::failbit); + } + } +} + +#if !defined(_MSC_VER) + +template +typename std::enable_if<(sizeof...(TT) > 0), void>::type +get(std::istream &input, T &res, TT &... resres) { + get(input, res); + get(input, resres...); +} + +#else +#include +#include +#include +#include +#include + +#define NESTED_GET(z, n, data) get(input, BOOST_PP_CAT(res, n)); +#define GET(z, n, data) \ +template \ +void get(std::istream &input, BOOST_PP_ENUM_BINARY_PARAMS(n, T, &res)) { \ + BOOST_PP_REPEAT(n, NESTED_GET, ~) \ +} +BOOST_PP_REPEAT_FROM_TO(2, 5, GET, ~) + +#endif -- cgit v1.2.3