mirror of
https://github.com/ton-blockchain/ton
synced 2025-03-09 15:40:10 +00:00
107 lines
4.1 KiB
C++
107 lines
4.1 KiB
C++
/*
|
|
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-2020 Telegram Systems LLP
|
|
*/
|
|
#include "HighloadWalletV2.h"
|
|
#include "GenericAccount.h"
|
|
#include "SmartContractCode.h"
|
|
|
|
#include "vm/boc.h"
|
|
#include "vm/cells/CellString.h"
|
|
#include "td/utils/base64.h"
|
|
|
|
#include <limits>
|
|
|
|
namespace ton {
|
|
|
|
td::Result<td::Ref<vm::Cell>> HighloadWalletV2::get_init_message(const td::Ed25519::PrivateKey& private_key,
|
|
td::uint32 valid_until) const noexcept {
|
|
TRY_RESULT(wallet_id, get_wallet_id());
|
|
td::uint32 id = -1;
|
|
auto append_message = [&](auto&& cb) -> vm::CellBuilder& {
|
|
cb.store_long(wallet_id, 32).store_long(valid_until, 32).store_long(id, 32);
|
|
CHECK(cb.store_maybe_ref({}));
|
|
return cb;
|
|
};
|
|
auto signature = private_key.sign(append_message(vm::CellBuilder()).finalize()->get_hash().as_slice()).move_as_ok();
|
|
|
|
return append_message(vm::CellBuilder().store_bytes(signature)).finalize();
|
|
}
|
|
|
|
td::Result<td::Ref<vm::Cell>> HighloadWalletV2::make_a_gift_message(const td::Ed25519::PrivateKey& private_key,
|
|
td::uint32 valid_until,
|
|
td::Span<Gift> gifts) const {
|
|
TRY_RESULT(wallet_id, get_wallet_id());
|
|
CHECK(gifts.size() <= get_max_gifts_size());
|
|
vm::Dictionary messages(16);
|
|
for (size_t i = 0; i < gifts.size(); i++) {
|
|
auto& gift = gifts[i];
|
|
td::int32 send_mode = 3;
|
|
if (gift.gramms == -1) {
|
|
send_mode += 128;
|
|
}
|
|
vm::CellBuilder cb;
|
|
cb.store_long(send_mode, 8).store_ref(create_int_message(gift));
|
|
auto key = messages.integer_key(td::make_refint(i), 16, false);
|
|
messages.set_builder(key.bits(), 16, cb);
|
|
}
|
|
std::string hash;
|
|
{
|
|
vm::CellBuilder cb;
|
|
CHECK(cb.store_maybe_ref(messages.get_root_cell()));
|
|
hash = cb.finalize()->get_hash().as_slice().substr(28, 4).str();
|
|
}
|
|
|
|
vm::CellBuilder cb;
|
|
cb.store_long(wallet_id, 32).store_long(valid_until, 32).store_bytes(hash);
|
|
CHECK(cb.store_maybe_ref(messages.get_root_cell()));
|
|
auto message_outer = cb.finalize();
|
|
auto signature = private_key.sign(message_outer->get_hash().as_slice()).move_as_ok();
|
|
return vm::CellBuilder().store_bytes(signature).append_cellslice(vm::load_cell_slice(message_outer)).finalize();
|
|
}
|
|
|
|
td::Ref<vm::Cell> HighloadWalletV2::get_init_data(const InitData& init_data) noexcept {
|
|
vm::CellBuilder cb;
|
|
cb.store_long(init_data.wallet_id, 32).store_long(init_data.seqno, 64).store_bytes(init_data.public_key);
|
|
CHECK(cb.store_maybe_ref({}));
|
|
return cb.finalize();
|
|
}
|
|
|
|
td::Result<td::uint32> HighloadWalletV2::get_wallet_id() const {
|
|
return TRY_VM([&]() -> td::Result<td::uint32> {
|
|
if (state_.data.is_null()) {
|
|
return 0;
|
|
}
|
|
auto cs = vm::load_cell_slice(state_.data);
|
|
return static_cast<td::uint32>(cs.fetch_ulong(32));
|
|
}());
|
|
}
|
|
|
|
td::Result<td::Ed25519::PublicKey> HighloadWalletV2::get_public_key() const {
|
|
return TRY_VM([&]() -> td::Result<td::Ed25519::PublicKey> {
|
|
if (state_.data.is_null()) {
|
|
return td::Status::Error("data is null");
|
|
}
|
|
auto cs = vm::load_cell_slice(state_.data);
|
|
cs.skip_first(96);
|
|
td::SecureString res(td::Ed25519::PublicKey::LENGTH);
|
|
cs.fetch_bytes(res.as_mutable_slice().ubegin(), td::narrow_cast<td::int32>(res.size()));
|
|
return td::Ed25519::PublicKey(std::move(res));
|
|
}());
|
|
}
|
|
|
|
} // namespace ton
|