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
12
keyring/CMakeLists.txt
Normal file
12
keyring/CMakeLists.txt
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
cmake_minimum_required(VERSION 3.0.2 FATAL_ERROR)
|
||||
|
||||
set(KEYRING_SOURCE
|
||||
keyring.h
|
||||
keyring.hpp
|
||||
keyring.cpp
|
||||
)
|
||||
|
||||
add_library(keyring STATIC ${KEYRING_SOURCE})
|
||||
|
||||
target_include_directories(keyring PUBLIC $<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/..>)
|
||||
target_link_libraries(keyring PUBLIC tdactor ton_crypto tl_api common )
|
||||
183
keyring/keyring.cpp
Normal file
183
keyring/keyring.cpp
Normal file
|
|
@ -0,0 +1,183 @@
|
|||
/*
|
||||
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
|
||||
*/
|
||||
#include "keyring.hpp"
|
||||
#include "common/errorcode.h"
|
||||
#include "common/io.hpp"
|
||||
#include "td/utils/port/path.h"
|
||||
#include "td/utils/filesystem.h"
|
||||
#include "td/utils/Random.h"
|
||||
|
||||
namespace ton {
|
||||
|
||||
namespace keyring {
|
||||
|
||||
void KeyringImpl::start_up() {
|
||||
td::mkdir(db_root_).ensure();
|
||||
}
|
||||
|
||||
td::Result<KeyringImpl::PrivateKeyDescr *> KeyringImpl::load_key(PublicKeyHash key_hash) {
|
||||
auto it = map_.find(key_hash);
|
||||
if (it != map_.end()) {
|
||||
return it->second.get();
|
||||
}
|
||||
|
||||
auto name = db_root_ + "/" + key_hash.bits256_value().to_hex();
|
||||
|
||||
auto R = td::read_file(td::CSlice{name});
|
||||
if (R.is_error()) {
|
||||
return R.move_as_error_prefix("key not in db: ");
|
||||
}
|
||||
auto data = R.move_as_ok();
|
||||
auto R2 = PrivateKey::import(td::SecureString(data));
|
||||
R2.ensure();
|
||||
|
||||
auto key = R2.move_as_ok();
|
||||
auto pub = key.compute_public_key();
|
||||
auto short_id = pub.compute_short_id();
|
||||
CHECK(short_id == key_hash);
|
||||
|
||||
auto D = key.create_decryptor_async();
|
||||
D.ensure();
|
||||
|
||||
return map_.emplace(short_id, std::make_unique<PrivateKeyDescr>(D.move_as_ok(), pub, false)).first->second.get();
|
||||
}
|
||||
|
||||
void KeyringImpl::add_key(PrivateKey key, bool is_temp, td::Promise<td::Unit> promise) {
|
||||
auto pub = key.compute_public_key();
|
||||
auto short_id = pub.compute_short_id();
|
||||
|
||||
if (map_.count(short_id)) {
|
||||
LOG(WARNING) << "duplicate key " << short_id;
|
||||
promise.set_value(td::Unit());
|
||||
return;
|
||||
}
|
||||
auto D = key.create_decryptor_async();
|
||||
D.ensure();
|
||||
|
||||
map_.emplace(short_id, std::make_unique<PrivateKeyDescr>(D.move_as_ok(), pub, is_temp));
|
||||
|
||||
if (!is_temp && key.exportable()) {
|
||||
auto S = key.export_as_slice();
|
||||
auto name = db_root_ + "/" + short_id.bits256_value().to_hex();
|
||||
|
||||
td::write_file(td::CSlice(name), S.as_slice()).ensure();
|
||||
}
|
||||
promise.set_value(td::Unit());
|
||||
}
|
||||
|
||||
void KeyringImpl::check_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
promise.set_value(td::Unit());
|
||||
}
|
||||
}
|
||||
|
||||
void KeyringImpl::add_key_short(PublicKeyHash key_hash, td::Promise<PublicKey> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
promise.set_result(map_[key_hash]->public_key);
|
||||
}
|
||||
}
|
||||
|
||||
void KeyringImpl::del_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) {
|
||||
map_.erase(key_hash);
|
||||
auto name = db_root_ + "/" + key_hash.bits256_value().to_hex();
|
||||
td::BufferSlice d{256};
|
||||
td::Random::secure_bytes(d.as_slice());
|
||||
td::write_file(name, d.as_slice()).ensure();
|
||||
td::unlink(name).ensure();
|
||||
promise.set_value(td::Unit());
|
||||
}
|
||||
|
||||
void KeyringImpl::get_public_key(PublicKeyHash key_hash, td::Promise<PublicKey> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
promise.set_result(map_[key_hash]->public_key);
|
||||
}
|
||||
}
|
||||
|
||||
void KeyringImpl::sign_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
td::actor::send_closure(S.move_as_ok()->decryptor, &DecryptorAsync::sign, std::move(data), std::move(promise));
|
||||
}
|
||||
}
|
||||
|
||||
void KeyringImpl::sign_add_get_public_key(PublicKeyHash key_hash, td::BufferSlice data,
|
||||
td::Promise<std::pair<td::BufferSlice, PublicKey>> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
return;
|
||||
}
|
||||
|
||||
auto D = S.move_as_ok();
|
||||
auto P = td::PromiseCreator::lambda(
|
||||
[promise = std::move(promise), id = D->public_key](td::Result<td::BufferSlice> R) mutable {
|
||||
if (R.is_error()) {
|
||||
promise.set_error(R.move_as_error());
|
||||
return;
|
||||
}
|
||||
promise.set_value(std::pair<td::BufferSlice, PublicKey>{R.move_as_ok(), id});
|
||||
});
|
||||
td::actor::send_closure(D->decryptor, &DecryptorAsync::sign, std::move(data), std::move(P));
|
||||
}
|
||||
|
||||
void KeyringImpl::sign_messages(PublicKeyHash key_hash, std::vector<td::BufferSlice> data,
|
||||
td::Promise<std::vector<td::Result<td::BufferSlice>>> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
td::actor::send_closure(S.move_as_ok()->decryptor, &DecryptorAsync::sign_batch, std::move(data),
|
||||
std::move(promise));
|
||||
}
|
||||
}
|
||||
|
||||
void KeyringImpl::decrypt_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) {
|
||||
auto S = load_key(key_hash);
|
||||
|
||||
if (S.is_error()) {
|
||||
promise.set_error(S.move_as_error());
|
||||
} else {
|
||||
td::actor::send_closure(S.move_as_ok()->decryptor, &DecryptorAsync::decrypt, std::move(data), std::move(promise));
|
||||
}
|
||||
}
|
||||
|
||||
td::actor::ActorOwn<Keyring> Keyring::create(std::string db_root) {
|
||||
return td::actor::create_actor<KeyringImpl>("keyring", db_root);
|
||||
}
|
||||
|
||||
} // namespace keyring
|
||||
|
||||
} // namespace ton
|
||||
52
keyring/keyring.h
Normal file
52
keyring/keyring.h
Normal file
|
|
@ -0,0 +1,52 @@
|
|||
/*
|
||||
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/actor/actor.h"
|
||||
#include "keys/keys.hpp"
|
||||
|
||||
namespace ton {
|
||||
|
||||
namespace keyring {
|
||||
|
||||
class Keyring : public td::actor::Actor {
|
||||
public:
|
||||
virtual ~Keyring() = default;
|
||||
|
||||
virtual void add_key(PrivateKey key, bool temp, td::Promise<td::Unit> promise) = 0;
|
||||
virtual void check_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) = 0;
|
||||
virtual void add_key_short(PublicKeyHash key_hash, td::Promise<PublicKey> promise) = 0;
|
||||
virtual void del_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) = 0;
|
||||
|
||||
//virtual void export_private_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) = 0;
|
||||
virtual void get_public_key(PublicKeyHash key_hash, td::Promise<PublicKey> promise) = 0;
|
||||
virtual void sign_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) = 0;
|
||||
virtual void sign_add_get_public_key(PublicKeyHash key_hash, td::BufferSlice data,
|
||||
td::Promise<std::pair<td::BufferSlice, PublicKey>> promise) = 0;
|
||||
virtual void sign_messages(PublicKeyHash key_hash, std::vector<td::BufferSlice> data,
|
||||
td::Promise<std::vector<td::Result<td::BufferSlice>>> promise) = 0;
|
||||
|
||||
virtual void decrypt_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) = 0;
|
||||
|
||||
static td::actor::ActorOwn<Keyring> create(std::string db_root);
|
||||
};
|
||||
|
||||
} // namespace keyring
|
||||
|
||||
} // namespace ton
|
||||
74
keyring/keyring.hpp
Normal file
74
keyring/keyring.hpp
Normal file
|
|
@ -0,0 +1,74 @@
|
|||
/*
|
||||
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 "keyring.h"
|
||||
#include "keys/encryptor.h"
|
||||
|
||||
#include <map>
|
||||
|
||||
namespace ton {
|
||||
|
||||
namespace keyring {
|
||||
|
||||
class KeyringImpl : public Keyring {
|
||||
private:
|
||||
struct PrivateKeyDescr {
|
||||
td::actor::ActorOwn<DecryptorAsync> decryptor;
|
||||
PublicKey public_key;
|
||||
bool is_temp;
|
||||
PrivateKeyDescr(td::actor::ActorOwn<DecryptorAsync> decryptor, PublicKey public_key, bool is_temp)
|
||||
: decryptor(std::move(decryptor)), public_key(public_key), is_temp(is_temp) {
|
||||
}
|
||||
};
|
||||
|
||||
public:
|
||||
void start_up() override;
|
||||
|
||||
td::Result<PrivateKeyDescr*> load_key(PublicKeyHash key_hash);
|
||||
|
||||
void add_key(PrivateKey key, bool is_temp, td::Promise<td::Unit> promise) override;
|
||||
void check_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) override;
|
||||
void add_key_short(PublicKeyHash key_hash, td::Promise<PublicKey> promise) override;
|
||||
void del_key(PublicKeyHash key_hash, td::Promise<td::Unit> promise) override;
|
||||
|
||||
void get_public_key(PublicKeyHash key_hash, td::Promise<PublicKey> promise) override;
|
||||
void sign_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) override;
|
||||
void sign_add_get_public_key(PublicKeyHash key_hash, td::BufferSlice data,
|
||||
td::Promise<std::pair<td::BufferSlice, PublicKey>> promise) override;
|
||||
void sign_messages(PublicKeyHash key_hash, std::vector<td::BufferSlice> data,
|
||||
td::Promise<std::vector<td::Result<td::BufferSlice>>> promise) override;
|
||||
|
||||
void decrypt_message(PublicKeyHash key_hash, td::BufferSlice data, td::Promise<td::BufferSlice> promise) override;
|
||||
|
||||
KeyringImpl(std::string db_root) : db_root_(db_root) {
|
||||
}
|
||||
|
||||
private:
|
||||
std::map<PublicKeyHash, std::unique_ptr<PrivateKeyDescr>> map_;
|
||||
std::unique_ptr<Decryptor> decryptor_;
|
||||
std::unique_ptr<Encryptor> encryptor_;
|
||||
|
||||
std::string db_root_;
|
||||
};
|
||||
|
||||
} // namespace keyring
|
||||
|
||||
} // namespace ton
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue