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
268
dht/dht-types.h
Normal file
268
dht/dht-types.h
Normal file
|
|
@ -0,0 +1,268 @@
|
|||
/*
|
||||
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/int_types.h"
|
||||
#include "td/utils/SharedSlice.h"
|
||||
#include "adnl/adnl-node-id.hpp"
|
||||
#include "tl-utils/tl-utils.hpp"
|
||||
#include "common/io.hpp"
|
||||
|
||||
namespace ton {
|
||||
|
||||
namespace dht {
|
||||
|
||||
using DhtKeyName = std::string;
|
||||
|
||||
class DhtKeyId {
|
||||
public:
|
||||
explicit DhtKeyId(td::Bits256 value) : value_(value) {
|
||||
}
|
||||
explicit DhtKeyId(adnl::AdnlNodeIdShort value) : value_(value.bits256_value()) {
|
||||
}
|
||||
DhtKeyId() {
|
||||
}
|
||||
td::Bits256 tl() const {
|
||||
return value_;
|
||||
}
|
||||
|
||||
bool operator<(const DhtKeyId &with) const {
|
||||
return value_ < with.value_;
|
||||
}
|
||||
bool operator==(const DhtKeyId &with) const {
|
||||
return value_ == with.value_;
|
||||
}
|
||||
bool operator!=(const DhtKeyId &with) const {
|
||||
return value_ != with.value_;
|
||||
}
|
||||
DhtKeyId operator^(const DhtKeyId &with) const {
|
||||
return DhtKeyId{value_ ^ with.value_};
|
||||
}
|
||||
DhtKeyId operator^(const adnl::AdnlNodeIdShort &with) const {
|
||||
return DhtKeyId{value_ ^ with.bits256_value()};
|
||||
}
|
||||
|
||||
bool get_bit(td::uint32 bit) const {
|
||||
return value_[bit];
|
||||
}
|
||||
|
||||
td::uint32 count_leading_zeroes() const {
|
||||
return value_.count_leading_zeroes();
|
||||
}
|
||||
adnl::AdnlNodeIdShort to_adnl() const {
|
||||
return adnl::AdnlNodeIdShort{value_};
|
||||
}
|
||||
|
||||
static DhtKeyId zero() {
|
||||
return DhtKeyId{td::Bits256::zero()};
|
||||
}
|
||||
|
||||
private:
|
||||
td::Bits256 value_;
|
||||
};
|
||||
|
||||
using DhtXoredKeyId = DhtKeyId;
|
||||
|
||||
class DhtKey {
|
||||
public:
|
||||
static constexpr td::uint32 max_name_length() {
|
||||
return 127;
|
||||
}
|
||||
static constexpr td::uint32 max_index() {
|
||||
return 15;
|
||||
}
|
||||
DhtKey(PublicKeyHash id, DhtKeyName namestr, td::uint32 idx)
|
||||
: id_(std::move(id)), namestr_(std::move(namestr)), idx_(idx) {
|
||||
CHECK(namestr.size() <= max_name_length());
|
||||
}
|
||||
static td::Result<DhtKey> create(tl_object_ptr<ton_api::dht_key> key);
|
||||
td::Status check() const;
|
||||
const auto &public_key_hash() const {
|
||||
return id_;
|
||||
}
|
||||
const auto &name() const {
|
||||
return namestr_;
|
||||
}
|
||||
td::uint32 idx() const {
|
||||
return idx_;
|
||||
}
|
||||
tl_object_ptr<ton_api::dht_key> tl() const;
|
||||
DhtKeyId compute_key_id() const;
|
||||
DhtKey clone() const;
|
||||
|
||||
private:
|
||||
PublicKeyHash id_;
|
||||
DhtKeyName namestr_;
|
||||
td::uint32 idx_;
|
||||
};
|
||||
|
||||
class DhtValue;
|
||||
|
||||
class DhtUpdateRule {
|
||||
public:
|
||||
virtual ~DhtUpdateRule() = default;
|
||||
virtual td::Status check_value(const DhtValue &value) = 0;
|
||||
virtual td::Status update_value(DhtValue &value, DhtValue &&new_value) = 0;
|
||||
virtual bool need_republish() const = 0;
|
||||
virtual tl_object_ptr<ton_api::dht_UpdateRule> tl() const = 0;
|
||||
static td::Result<std::shared_ptr<DhtUpdateRule>> create(tl_object_ptr<ton_api::dht_UpdateRule> obj);
|
||||
};
|
||||
|
||||
class DhtKeyDescription {
|
||||
public:
|
||||
DhtKeyDescription(DhtKey key, PublicKey public_key, std::shared_ptr<DhtUpdateRule> update_rule,
|
||||
td::BufferSlice signature)
|
||||
: key_(std::move(key))
|
||||
, public_key_(std::move(public_key))
|
||||
, update_rule_(std::move(update_rule))
|
||||
, signature_(signature.as_slice()) {
|
||||
}
|
||||
DhtKeyDescription(DhtKey key, PublicKey public_key, std::shared_ptr<DhtUpdateRule> update_rule,
|
||||
td::SharedSlice signature)
|
||||
: key_(std::move(key))
|
||||
, public_key_(std::move(public_key))
|
||||
, update_rule_(std::move(update_rule))
|
||||
, signature_(std::move(signature)) {
|
||||
}
|
||||
const auto &key() const {
|
||||
return key_;
|
||||
}
|
||||
const auto &public_key() const {
|
||||
return public_key_;
|
||||
}
|
||||
const auto &update_rule() const {
|
||||
return update_rule_;
|
||||
}
|
||||
void update_signature(td::BufferSlice signature);
|
||||
void update_signature(td::SharedSlice signature);
|
||||
td::BufferSlice to_sign() const;
|
||||
td::Status check() const;
|
||||
tl_object_ptr<ton_api::dht_keyDescription> tl() const;
|
||||
DhtKeyDescription clone() const;
|
||||
static td::Result<DhtKeyDescription> create(tl_object_ptr<ton_api::dht_keyDescription> desc, bool check_signature);
|
||||
static td::Result<DhtKeyDescription> create(DhtKey key, PublicKey public_key,
|
||||
std::shared_ptr<DhtUpdateRule> update_rule, td::BufferSlice signature);
|
||||
static td::Result<DhtKeyDescription> create(DhtKey key, PublicKey public_key,
|
||||
std::shared_ptr<DhtUpdateRule> update_rule, td::SharedSlice signature);
|
||||
|
||||
private:
|
||||
DhtKey key_;
|
||||
PublicKey public_key_;
|
||||
std::shared_ptr<DhtUpdateRule> update_rule_;
|
||||
td::SharedSlice signature_;
|
||||
};
|
||||
|
||||
class DhtValue {
|
||||
public:
|
||||
static constexpr td::uint32 max_value_size() {
|
||||
return 768;
|
||||
}
|
||||
|
||||
DhtValue(DhtKeyDescription key, td::BufferSlice value, td::uint32 ttl, td::BufferSlice signature)
|
||||
: key_(std::move(key)), value_(value.as_slice()), ttl_(ttl), signature_(signature.as_slice()) {
|
||||
}
|
||||
DhtValue(DhtKeyDescription key, td::SharedSlice value, td::uint32 ttl, td::SharedSlice signature)
|
||||
: key_(std::move(key)), value_(std::move(value)), ttl_(ttl), signature_(std::move(signature)) {
|
||||
}
|
||||
|
||||
static td::Result<DhtValue> create(tl_object_ptr<ton_api::dht_value> obj, bool check_signature);
|
||||
static td::Result<DhtValue> create(DhtKeyDescription key, td::BufferSlice value, td::uint32 ttl,
|
||||
td::BufferSlice signature);
|
||||
static td::Result<DhtValue> create(DhtKeyDescription key, td::SharedSlice value, td::uint32 ttl,
|
||||
td::SharedSlice signature);
|
||||
const auto &key() const {
|
||||
return key_;
|
||||
}
|
||||
const auto &value() const {
|
||||
return value_;
|
||||
}
|
||||
const auto &signature() const {
|
||||
return signature_;
|
||||
}
|
||||
td::uint32 ttl() const {
|
||||
return ttl_;
|
||||
}
|
||||
bool expired() const {
|
||||
return ttl_ < td::Clocks::system();
|
||||
}
|
||||
DhtValue clone() const;
|
||||
|
||||
tl_object_ptr<ton_api::dht_value> tl() const;
|
||||
td::BufferSlice to_sign() const;
|
||||
td::Status update(DhtValue &&value);
|
||||
void set(td::BufferSlice value, td::uint32 ttl, td::BufferSlice signature);
|
||||
void set(td::SharedSlice value, td::uint32 ttl, td::SharedSlice signature);
|
||||
void update_signature(td::BufferSlice signature);
|
||||
void update_signature(td::SharedSlice signature);
|
||||
td::Status check() const;
|
||||
|
||||
DhtKeyId key_id() const;
|
||||
|
||||
private:
|
||||
DhtKeyDescription key_;
|
||||
td::SharedSlice value_;
|
||||
td::uint32 ttl_;
|
||||
td::SharedSlice signature_;
|
||||
};
|
||||
|
||||
class DhtUpdateRuleSignature : public DhtUpdateRule {
|
||||
public:
|
||||
td::Status check_value(const DhtValue &value) override;
|
||||
td::Status update_value(DhtValue &value, DhtValue &&new_value) override;
|
||||
bool need_republish() const override {
|
||||
return true;
|
||||
}
|
||||
tl_object_ptr<ton_api::dht_UpdateRule> tl() const override;
|
||||
static td::Result<std::shared_ptr<DhtUpdateRule>> create();
|
||||
};
|
||||
|
||||
class DhtUpdateRuleAnybody : public DhtUpdateRule {
|
||||
public:
|
||||
td::Status check_value(const DhtValue &value) override;
|
||||
td::Status update_value(DhtValue &value, DhtValue &&new_value) override;
|
||||
bool need_republish() const override {
|
||||
return false;
|
||||
}
|
||||
tl_object_ptr<ton_api::dht_UpdateRule> tl() const override;
|
||||
static td::Result<std::shared_ptr<DhtUpdateRule>> create();
|
||||
};
|
||||
|
||||
class DhtUpdateRuleOverlayNodes : public DhtUpdateRule {
|
||||
public:
|
||||
td::Status check_value(const DhtValue &value) override;
|
||||
td::Status update_value(DhtValue &value, DhtValue &&new_value) override;
|
||||
bool need_republish() const override {
|
||||
return false;
|
||||
}
|
||||
tl_object_ptr<ton_api::dht_UpdateRule> tl() const override;
|
||||
static td::Result<std::shared_ptr<DhtUpdateRule>> create();
|
||||
};
|
||||
|
||||
} // namespace dht
|
||||
|
||||
} // namespace ton
|
||||
|
||||
namespace td {
|
||||
|
||||
inline td::StringBuilder &operator<<(td::StringBuilder &sb, const ton::dht::DhtKeyId &dht) {
|
||||
sb << dht.tl();
|
||||
return sb;
|
||||
}
|
||||
|
||||
} // namespace td
|
||||
Loading…
Add table
Add a link
Reference in a new issue