mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
initial commit
This commit is contained in:
initial commit
vvaltman
commit
c2da007f40
1610 changed files with 398047 additions and 0 deletions
276
tdutils/td/utils/tl_helpers.h
Normal file
276
tdutils/td/utils/tl_helpers.h
Normal file
|
|
@ -0,0 +1,276 @@
|
|||
/*
|
||||
This file is part of TON Blockchain Library.
|
||||
|
||||
TON Blockchain Library 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 2 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
TON Blockchain Library 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 TON Blockchain Library. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
Copyright 2017-2019 Telegram Systems LLP
|
||||
*/
|
||||
#pragma once
|
||||
|
||||
#include "td/utils/common.h"
|
||||
#include "td/utils/logging.h"
|
||||
#include "td/utils/misc.h"
|
||||
#include "td/utils/Slice.h"
|
||||
#include "td/utils/SharedSlice.h"
|
||||
#include "td/utils/StackAllocator.h"
|
||||
#include "td/utils/Status.h"
|
||||
#include "td/utils/tl_parsers.h"
|
||||
#include "td/utils/tl_storers.h"
|
||||
#include "td/utils/Variant.h"
|
||||
|
||||
#include <type_traits>
|
||||
#include <unordered_set>
|
||||
|
||||
#define BEGIN_STORE_FLAGS() \
|
||||
do { \
|
||||
uint32 flags_store = 0; \
|
||||
uint32 bit_offset_store = 0
|
||||
|
||||
#define STORE_FLAG(flag) \
|
||||
flags_store |= (flag) << bit_offset_store; \
|
||||
bit_offset_store++
|
||||
|
||||
#define END_STORE_FLAGS() \
|
||||
CHECK(bit_offset_store < 31); \
|
||||
td::store(flags_store, storer); \
|
||||
} \
|
||||
while (false)
|
||||
|
||||
#define BEGIN_PARSE_FLAGS() \
|
||||
do { \
|
||||
uint32 flags_parse; \
|
||||
uint32 bit_offset_parse = 0; \
|
||||
td::parse(flags_parse, parser)
|
||||
|
||||
#define PARSE_FLAG(flag) \
|
||||
flag = ((flags_parse >> bit_offset_parse) & 1) != 0; \
|
||||
bit_offset_parse++
|
||||
|
||||
#define END_PARSE_FLAGS() \
|
||||
CHECK(bit_offset_parse < 31); \
|
||||
if ((flags_parse & ~((1 << bit_offset_parse) - 1)) != 0) { \
|
||||
parser.set_error(PSTRING() << "Invalid flags " << flags_parse << " left, current bit is " << bit_offset_parse); \
|
||||
} \
|
||||
} \
|
||||
while (false)
|
||||
|
||||
namespace td {
|
||||
|
||||
template <class StorerT>
|
||||
void store(bool x, StorerT &storer) {
|
||||
storer.store_binary(static_cast<int32>(x));
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(bool &x, ParserT &parser) {
|
||||
x = parser.fetch_int() != 0;
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void store(int32 x, StorerT &storer) {
|
||||
storer.store_binary(x);
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(int32 &x, ParserT &parser) {
|
||||
x = parser.fetch_int();
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void store(uint32 x, StorerT &storer) {
|
||||
storer.store_binary(x);
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(uint32 &x, ParserT &parser) {
|
||||
x = static_cast<uint32>(parser.fetch_int());
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void store(int64 x, StorerT &storer) {
|
||||
storer.store_binary(x);
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(int64 &x, ParserT &parser) {
|
||||
x = parser.fetch_long();
|
||||
}
|
||||
template <class StorerT>
|
||||
void store(uint64 x, StorerT &storer) {
|
||||
storer.store_binary(x);
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(uint64 &x, ParserT &parser) {
|
||||
x = static_cast<uint64>(parser.fetch_long());
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void store(double x, StorerT &storer) {
|
||||
storer.store_binary(x);
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(double &x, ParserT &parser) {
|
||||
x = parser.fetch_double();
|
||||
}
|
||||
|
||||
template <class StorerT>
|
||||
void store(Slice x, StorerT &storer) {
|
||||
storer.store_string(x);
|
||||
}
|
||||
template <class StorerT>
|
||||
void store(const string &x, StorerT &storer) {
|
||||
storer.store_string(x);
|
||||
}
|
||||
template <class StorerT>
|
||||
void store(const SecureString &x, StorerT &storer) {
|
||||
storer.store_string(x.as_slice());
|
||||
}
|
||||
template <class ParserT>
|
||||
void parse(string &x, ParserT &parser) {
|
||||
x = parser.template fetch_string<string>();
|
||||
}
|
||||
|
||||
template <class ParserT>
|
||||
void parse(SecureString &x, ParserT &parser) {
|
||||
x = parser.template fetch_string<SecureString>();
|
||||
}
|
||||
|
||||
template <class T, class StorerT>
|
||||
void store(const vector<T> &vec, StorerT &storer) {
|
||||
storer.store_binary(narrow_cast<int32>(vec.size()));
|
||||
for (auto &val : vec) {
|
||||
store(val, storer);
|
||||
}
|
||||
}
|
||||
template <class T, class ParserT>
|
||||
void parse(vector<T> &vec, ParserT &parser) {
|
||||
uint32 size = parser.fetch_int();
|
||||
if (parser.get_left_len() < size) {
|
||||
parser.set_error("Wrong vector length");
|
||||
return;
|
||||
}
|
||||
vec = vector<T>(size);
|
||||
for (auto &val : vec) {
|
||||
parse(val, parser);
|
||||
}
|
||||
}
|
||||
|
||||
template <class Key, class Hash, class KeyEqual, class Allocator, class StorerT>
|
||||
void store(const std::unordered_set<Key, Hash, KeyEqual, Allocator> &s, StorerT &storer) {
|
||||
storer.store_binary(narrow_cast<int32>(s.size()));
|
||||
for (auto &val : s) {
|
||||
store(val, storer);
|
||||
}
|
||||
}
|
||||
template <class Key, class Hash, class KeyEqual, class Allocator, class ParserT>
|
||||
void parse(std::unordered_set<Key, Hash, KeyEqual, Allocator> &s, ParserT &parser) {
|
||||
uint32 size = parser.fetch_int();
|
||||
if (parser.get_left_len() < size) {
|
||||
parser.set_error("Wrong set length");
|
||||
return;
|
||||
}
|
||||
s.clear();
|
||||
for (uint32 i = 0; i < size; i++) {
|
||||
Key val;
|
||||
parse(val, parser);
|
||||
s.insert(std::move(val));
|
||||
}
|
||||
}
|
||||
|
||||
template <class T, class StorerT>
|
||||
std::enable_if_t<std::is_enum<T>::value> store(const T &val, StorerT &storer) {
|
||||
store(static_cast<int32>(val), storer);
|
||||
}
|
||||
template <class T, class ParserT>
|
||||
std::enable_if_t<std::is_enum<T>::value> parse(T &val, ParserT &parser) {
|
||||
int32 result;
|
||||
parse(result, parser);
|
||||
val = static_cast<T>(result);
|
||||
}
|
||||
|
||||
template <class T, class StorerT>
|
||||
std::enable_if_t<!std::is_enum<T>::value> store(const T &val, StorerT &storer) {
|
||||
val.store(storer);
|
||||
}
|
||||
template <class T, class ParserT>
|
||||
std::enable_if_t<!std::is_enum<T>::value> parse(T &val, ParserT &parser) {
|
||||
val.parse(parser);
|
||||
}
|
||||
|
||||
template <class... Types, class StorerT>
|
||||
void store(const Variant<Types...> &variant, StorerT &storer) {
|
||||
store(variant.get_offset(), storer);
|
||||
variant.visit([&storer](auto &&value) {
|
||||
using td::store;
|
||||
store(value, storer);
|
||||
});
|
||||
}
|
||||
template <class... Types, class ParserT>
|
||||
void parse(Variant<Types...> &variant, ParserT &parser) {
|
||||
auto type_offset = parser.fetch_int();
|
||||
if (type_offset < 0 || type_offset >= static_cast<int32>(sizeof...(Types))) {
|
||||
return parser.set_error("Invalid type");
|
||||
}
|
||||
variant.for_each([type_offset, &parser, &variant](int offset, auto *ptr) {
|
||||
using T = std::decay_t<decltype(*ptr)>;
|
||||
if (offset == type_offset) {
|
||||
variant = T();
|
||||
parse(variant.template get<T>(), parser);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
template <class T>
|
||||
string serialize(const T &object) {
|
||||
TlStorerCalcLength calc_length;
|
||||
store(object, calc_length);
|
||||
size_t length = calc_length.get_length();
|
||||
|
||||
string key(length, '\0');
|
||||
if (!is_aligned_pointer<4>(key.data())) {
|
||||
auto ptr = StackAllocator::alloc(length);
|
||||
MutableSlice data = ptr.as_slice();
|
||||
TlStorerUnsafe storer(data.ubegin());
|
||||
store(object, storer);
|
||||
CHECK(storer.get_buf() == data.uend());
|
||||
key.assign(data.begin(), data.size());
|
||||
} else {
|
||||
MutableSlice data = key;
|
||||
TlStorerUnsafe storer(data.ubegin());
|
||||
store(object, storer);
|
||||
CHECK(storer.get_buf() == data.uend());
|
||||
}
|
||||
return key;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
SecureString serialize_secure(const T &object) {
|
||||
TlStorerCalcLength calc_length;
|
||||
store(object, calc_length);
|
||||
size_t length = calc_length.get_length();
|
||||
|
||||
SecureString key(length, '\0');
|
||||
CHECK(is_aligned_pointer<4>(key.data()));
|
||||
MutableSlice data = key.as_mutable_slice();
|
||||
TlStorerUnsafe storer(data.ubegin());
|
||||
store(object, storer);
|
||||
CHECK(storer.get_buf() == data.uend());
|
||||
return key;
|
||||
}
|
||||
|
||||
template <class T>
|
||||
TD_WARN_UNUSED_RESULT Status unserialize(T &object, Slice data) {
|
||||
TlParser parser(data);
|
||||
parse(object, parser);
|
||||
parser.fetch_end();
|
||||
return parser.get_status();
|
||||
}
|
||||
|
||||
} // namespace td
|
||||
Loading…
Add table
Add a link
Reference in a new issue