mirror of
https://github.com/jwinterm/grim.git
synced 2026-10-17 14:21:24 +02:00
1592 lines
60 KiB
Rust
1592 lines
60 KiB
Rust
// Copyright 2023 The Grim Developers
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
use std::{fs, thread};
|
|
use std::fs::File;
|
|
use std::io::Write;
|
|
use std::net::{SocketAddr, TcpListener};
|
|
use std::path::PathBuf;
|
|
use std::sync::{Arc, mpsc};
|
|
use parking_lot::RwLock;
|
|
use std::sync::atomic::{AtomicBool, AtomicU8, Ordering};
|
|
use std::thread::Thread;
|
|
use std::time::Duration;
|
|
use futures::channel::oneshot;
|
|
use serde_json::{json, Value};
|
|
use rand::Rng;
|
|
|
|
use grin_api::{ApiServer, Router};
|
|
use grin_chain::SyncStatus;
|
|
use grin_core::global;
|
|
use grin_core::global::ChainTypes;
|
|
use grin_keychain::{ExtKeychain, Identifier, Keychain};
|
|
use grin_util::{Mutex, ToHex};
|
|
use grin_util::secp::SecretKey;
|
|
use grin_util::types::ZeroingString;
|
|
use grin_wallet_api::Owner;
|
|
use grin_wallet_controller::command::parse_slatepack;
|
|
use grin_wallet_controller::controller;
|
|
use grin_wallet_controller::controller::ForeignAPIHandlerV2;
|
|
use grin_wallet_impls::{DefaultLCProvider, DefaultWalletImpl, HTTPNodeClient};
|
|
use grin_wallet_libwallet::{address, Error, InitTxArgs, IssueInvoiceTxArgs, NodeClient, RetrieveTxQueryArgs, RetrieveTxQuerySortField, RetrieveTxQuerySortOrder, Slate, SlatepackAddress, SlateState, SlateVersion, StatusMessage, TxLogEntry, TxLogEntryType, VersionedSlate, WalletInst, WalletLCProvider};
|
|
use grin_wallet_libwallet::api_impl::owner::{cancel_tx, retrieve_summary_info, retrieve_txs};
|
|
use grin_wallet_util::OnionV3Address;
|
|
|
|
use crate::AppConfig;
|
|
use crate::node::{Node, NodeConfig};
|
|
use crate::tor::Tor;
|
|
use crate::wallet::{ConnectionsConfig, ExternalConnection, WalletConfig};
|
|
use crate::wallet::store::TxHeightStore;
|
|
use crate::wallet::types::{ConnectionMethod, WalletAccount, WalletData, WalletInstance, WalletTransaction};
|
|
|
|
/// Contains wallet instance, configuration and state, handles wallet commands.
|
|
#[derive(Clone)]
|
|
pub struct Wallet {
|
|
/// Wallet configuration.
|
|
config: Arc<RwLock<WalletConfig>>,
|
|
/// Wallet instance, initializing on wallet opening and clearing on wallet closing.
|
|
instance: Option<WalletInstance>,
|
|
/// [`WalletInstance`] external connection URL.
|
|
external_connection: Arc<RwLock<Option<ExternalConnection>>>,
|
|
|
|
/// Wallet Slatepack address to receive txs at transport.
|
|
slatepack_address: Arc<RwLock<Option<String>>>,
|
|
|
|
/// Wallet sync thread.
|
|
sync_thread: Arc<RwLock<Option<Thread>>>,
|
|
|
|
/// Running wallet foreign API server and port.
|
|
foreign_api_server: Arc<RwLock<Option<(ApiServer, u16)>>>,
|
|
|
|
/// Flag to check if wallet reopening is needed.
|
|
reopen: Arc<AtomicBool>,
|
|
/// Flag to check if wallet is open.
|
|
is_open: Arc<AtomicBool>,
|
|
/// Flag to check if wallet is closing.
|
|
closing: Arc<AtomicBool>,
|
|
|
|
/// Flag to check if wallet was deleted to remove it from the list.
|
|
deleted: Arc<AtomicBool>,
|
|
|
|
/// Error on wallet loading.
|
|
sync_error: Arc<AtomicBool>,
|
|
/// Info loading progress in percents.
|
|
info_sync_progress: Arc<AtomicU8>,
|
|
|
|
/// Wallet accounts.
|
|
accounts: Arc<RwLock<Vec<WalletAccount>>>,
|
|
|
|
/// Wallet info to show at ui.
|
|
data: Arc<RwLock<Option<WalletData>>>,
|
|
/// Attempts amount to update wallet data.
|
|
sync_attempts: Arc<AtomicU8>,
|
|
/// Flag to check if wallet is syncing.
|
|
syncing: Arc<AtomicBool>,
|
|
|
|
/// Flag to check if wallet repairing and restoring missing outputs is needed.
|
|
repair_needed: Arc<AtomicBool>,
|
|
/// Wallet repair progress in percents.
|
|
repair_progress: Arc<AtomicU8>
|
|
}
|
|
|
|
impl Wallet {
|
|
/// Create new [`Wallet`] instance with provided [`WalletConfig`].
|
|
fn new(config: WalletConfig) -> Self {
|
|
Self {
|
|
config: Arc::new(RwLock::new(config)),
|
|
instance: None,
|
|
external_connection: Arc::new(RwLock::new(None)),
|
|
slatepack_address: Arc::new(RwLock::new(None)),
|
|
sync_thread: Arc::from(RwLock::new(None)),
|
|
foreign_api_server: Arc::new(RwLock::new(None)),
|
|
reopen: Arc::new(AtomicBool::new(false)),
|
|
is_open: Arc::from(AtomicBool::new(false)),
|
|
closing: Arc::new(AtomicBool::new(false)),
|
|
deleted: Arc::new(AtomicBool::new(false)),
|
|
sync_error: Arc::from(AtomicBool::new(false)),
|
|
info_sync_progress: Arc::from(AtomicU8::new(0)),
|
|
accounts: Arc::new(RwLock::new(vec![])),
|
|
data: Arc::new(RwLock::new(None)),
|
|
sync_attempts: Arc::new(AtomicU8::new(0)),
|
|
syncing: Arc::new(AtomicBool::new(false)),
|
|
repair_needed: Arc::new(AtomicBool::new(false)),
|
|
repair_progress: Arc::new(AtomicU8::new(0))
|
|
}
|
|
}
|
|
|
|
/// Create new wallet.
|
|
pub fn create(
|
|
name: String,
|
|
password: String,
|
|
mnemonic: String,
|
|
conn_method: &ConnectionMethod
|
|
) -> Result<Wallet, Error> {
|
|
let mut config = WalletConfig::create(name, conn_method);
|
|
let w = Wallet::new(config.clone());
|
|
{
|
|
let instance = Self::create_wallet_instance(&mut config)?;
|
|
let mut w_lock = instance.lock();
|
|
let p = w_lock.lc_provider()?;
|
|
p.create_wallet(None,
|
|
Some(ZeroingString::from(mnemonic.clone())),
|
|
mnemonic.len(),
|
|
ZeroingString::from(password),
|
|
false,
|
|
)?;
|
|
}
|
|
Ok(w)
|
|
}
|
|
|
|
/// Initialize [`Wallet`] from provided data path.
|
|
pub fn init(data_path: PathBuf) -> Option<Wallet> {
|
|
let wallet_config = WalletConfig::load(data_path.clone());
|
|
if let Some(config) = wallet_config {
|
|
return Some(Wallet::new(config));
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Create [`WalletInstance`] from provided [`WalletConfig`].
|
|
fn create_wallet_instance(config: &mut WalletConfig) -> Result<WalletInstance, Error> {
|
|
// Assume global chain type has already been initialized.
|
|
let chain_type = config.chain_type;
|
|
if !global::GLOBAL_CHAIN_TYPE.is_init() {
|
|
global::init_global_chain_type(chain_type);
|
|
} else {
|
|
global::set_global_chain_type(chain_type);
|
|
global::set_local_chain_type(chain_type);
|
|
}
|
|
|
|
// Setup node client.
|
|
let integrated = || {
|
|
let api_url = format!("http://{}", NodeConfig::get_api_address());
|
|
let api_secret = NodeConfig::get_foreign_api_secret();
|
|
(api_url, api_secret)
|
|
};
|
|
let (node_api_url, node_secret) = if let Some(id) = config.ext_conn_id {
|
|
if let Some(conn) = ConnectionsConfig::ext_conn(id) {
|
|
(conn.url, conn.secret)
|
|
} else {
|
|
if chain_type == ChainTypes::Mainnet {
|
|
// Save default external connection if previous was deleted.
|
|
let default = ExternalConnection::default_main();
|
|
config.ext_conn_id = Some(default.id);
|
|
config.save();
|
|
|
|
(default.url, default.secret)
|
|
} else {
|
|
integrated()
|
|
}
|
|
}
|
|
} else {
|
|
integrated()
|
|
};
|
|
let node_client = HTTPNodeClient::new(&node_api_url, node_secret)?;
|
|
|
|
// Create wallet instance.
|
|
let wallet = Self::inst_wallet::<
|
|
DefaultLCProvider<HTTPNodeClient, ExtKeychain>,
|
|
HTTPNodeClient,
|
|
ExtKeychain,
|
|
>(config, node_client)?;
|
|
Ok(wallet)
|
|
}
|
|
|
|
/// Instantiate [`WalletInstance`] from provided node client and [`WalletConfig`].
|
|
fn inst_wallet<L, C, K>(
|
|
config: &mut WalletConfig,
|
|
node_client: C,
|
|
) -> Result<Arc<Mutex<Box<dyn WalletInst<'static, L, C, K>>>>, Error>
|
|
where
|
|
DefaultWalletImpl<'static, C>: WalletInst<'static, L, C, K>,
|
|
L: WalletLCProvider<'static, C, K>,
|
|
C: NodeClient + 'static,
|
|
K: Keychain + 'static,
|
|
{
|
|
let mut wallet = Box::new(DefaultWalletImpl::<'static, C>::new(node_client).unwrap())
|
|
as Box<dyn WalletInst<'static, L, C, K>>;
|
|
let lc = wallet.lc_provider()?;
|
|
lc.set_top_level_directory(config.get_data_path().as_str())?;
|
|
Ok(Arc::new(Mutex::new(wallet)))
|
|
}
|
|
|
|
/// Get parent key identifier for current account.
|
|
pub fn get_parent_key_id(&self) -> Result<Identifier, Error> {
|
|
let instance = self.instance.clone().unwrap();
|
|
let mut w_lock = instance.lock();
|
|
let lc = w_lock.lc_provider()?;
|
|
let w_inst = lc.wallet_inst()?;
|
|
Ok(w_inst.parent_key_id())
|
|
}
|
|
|
|
/// Get wallet [`SecretKey`] for transports.
|
|
pub fn secret_key(&self) -> Result<SecretKey, Error> {
|
|
let instance = self.instance.clone().unwrap();
|
|
let mut w_lock = instance.lock();
|
|
let lc = w_lock.lc_provider()?;
|
|
let w_inst = lc.wallet_inst()?;
|
|
let k = w_inst.keychain((&None).as_ref())?;
|
|
let parent_key_id = w_inst.parent_key_id();
|
|
let sec_key = address::address_from_derivation_path(&k, &parent_key_id, 0)
|
|
.map_err(|e| Error::TorConfig(format!("{:?}", e)))?;
|
|
Ok(sec_key)
|
|
}
|
|
|
|
/// Get unique opened wallet identifier, including current account.
|
|
pub fn identifier(&self) -> String {
|
|
let config = self.get_config();
|
|
format!("wallet_{}_{}", config.id, config.account.to_hex())
|
|
}
|
|
|
|
/// Get Slatepack address to receive txs at transport.
|
|
pub fn slatepack_address(&self) -> Option<String> {
|
|
let r_address = self.slatepack_address.read();
|
|
if r_address.is_some() {
|
|
let addr = r_address.clone();
|
|
return addr
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Get wallet config.
|
|
pub fn get_config(&self) -> WalletConfig {
|
|
self.config.read().clone()
|
|
}
|
|
|
|
/// Change wallet name.
|
|
pub fn change_name(&self, name: String) {
|
|
let mut w_config = self.config.write();
|
|
w_config.name = name;
|
|
w_config.save();
|
|
}
|
|
|
|
/// Check if start of Tor listener on wallet opening is needed.
|
|
pub fn auto_start_tor_listener(&self) -> bool {
|
|
let r_config = self.config.read();
|
|
r_config.enable_tor_listener.unwrap_or(true)
|
|
}
|
|
|
|
/// Update start of Tor listener on wallet opening.
|
|
pub fn update_auto_start_tor_listener(&self, start: bool) {
|
|
let mut w_config = self.config.write();
|
|
w_config.enable_tor_listener = Some(start);
|
|
w_config.save();
|
|
}
|
|
|
|
/// Check if Dandelion usage is needed to post transactions.
|
|
pub fn can_use_dandelion(&self) -> bool {
|
|
let r_config = self.config.read();
|
|
r_config.use_dandelion.unwrap_or(true)
|
|
}
|
|
|
|
/// Update usage of Dandelion to post transactions.
|
|
pub fn update_use_dandelion(&self, use_dandelion: bool) {
|
|
let mut w_config = self.config.write();
|
|
w_config.use_dandelion = Some(use_dandelion);
|
|
w_config.save();
|
|
}
|
|
|
|
/// Update minimal amount of confirmations.
|
|
pub fn update_min_confirmations(&self, min_confirmations: u64) {
|
|
let mut w_config = self.config.write();
|
|
w_config.min_confirmations = min_confirmations;
|
|
w_config.save();
|
|
}
|
|
|
|
/// Update external connection identifier.
|
|
pub fn update_ext_conn_id(&self, id: Option<i64>) {
|
|
let mut w_config = self.config.write();
|
|
w_config.ext_conn_id = id;
|
|
w_config.save();
|
|
}
|
|
|
|
/// Open the wallet and start [`WalletData`] sync at separate thread.
|
|
pub fn open(&mut self, password: String) -> Result<(), Error> {
|
|
if self.is_open() {
|
|
return Err(Error::GenericError("Already opened".to_string()));
|
|
}
|
|
|
|
// Create new wallet instance if sync thread was stopped or instance was not created.
|
|
if self.sync_thread.read().is_none() || self.instance.is_none() {
|
|
let mut config = self.get_config();
|
|
let new_instance = Self::create_wallet_instance(&mut config)?;
|
|
self.instance = Some(new_instance);
|
|
// Setup current external connection.
|
|
{
|
|
let mut w_conn = self.external_connection.write();
|
|
*w_conn = self.get_current_ext_conn();
|
|
}
|
|
}
|
|
|
|
// Open the wallet.
|
|
{
|
|
let instance = self.instance.clone().unwrap();
|
|
let mut wallet_lock = instance.lock();
|
|
let lc = wallet_lock.lc_provider()?;
|
|
match lc.open_wallet(None, ZeroingString::from(password), false, false) {
|
|
Ok(_) => {
|
|
// Reset an error on opening.
|
|
self.set_sync_error(false);
|
|
self.reset_sync_attempts();
|
|
|
|
// Set current account.
|
|
let wallet_inst = lc.wallet_inst()?;
|
|
let label = self.get_config().account.to_owned();
|
|
wallet_inst.set_parent_key_id_by_name(label.as_str())?;
|
|
|
|
// Start new synchronization thread or wake up existing one.
|
|
let mut thread_w = self.sync_thread.write();
|
|
if thread_w.is_none() {
|
|
let thread = start_sync(self.clone());
|
|
*thread_w = Some(thread);
|
|
} else {
|
|
thread_w.clone().unwrap().unpark();
|
|
}
|
|
self.is_open.store(true, Ordering::Relaxed);
|
|
}
|
|
Err(e) => {
|
|
self.instance = None;
|
|
return Err(e)
|
|
}
|
|
}
|
|
}
|
|
|
|
// Set slatepack address.
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
let mut w_address = self.slatepack_address.write();
|
|
*w_address = Some(api.get_slatepack_address(m, 0)?.to_string());
|
|
Ok(())
|
|
})?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
/// Get external connection URL applied to [`WalletInstance`]
|
|
/// after wallet opening if sync is running or get it from config.
|
|
pub fn get_current_ext_conn(&self) -> Option<ExternalConnection> {
|
|
if self.sync_thread.read().is_some() {
|
|
let r_conn = self.external_connection.read();
|
|
r_conn.clone()
|
|
} else {
|
|
let config = self.get_config();
|
|
if let Some(id) = config.ext_conn_id {
|
|
return match ConnectionsConfig::ext_conn(id) {
|
|
None => {
|
|
if config.chain_type == ChainTypes::Mainnet {
|
|
Some(ExternalConnection::default_main())
|
|
} else {
|
|
None
|
|
}
|
|
},
|
|
Some(ext_conn) => Some(ext_conn)
|
|
}
|
|
}
|
|
None
|
|
}
|
|
}
|
|
|
|
/// Check if wallet is open.
|
|
pub fn is_open(&self) -> bool {
|
|
self.is_open.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Check if wallet is closing.
|
|
pub fn is_closing(&self) -> bool {
|
|
self.closing.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Close the wallet.
|
|
pub fn close(&self) {
|
|
if !self.is_open() || self.instance.is_none() {
|
|
return;
|
|
}
|
|
self.closing.store(true, Ordering::Relaxed);
|
|
|
|
// Close wallet at separate thread.
|
|
let wallet_close = self.clone();
|
|
let service_id = wallet_close.identifier();
|
|
thread::spawn(move || {
|
|
// Stop running API server.
|
|
let api_server_exists = {
|
|
wallet_close.foreign_api_server.read().is_some()
|
|
};
|
|
if api_server_exists {
|
|
let mut w_api_server = wallet_close.foreign_api_server.write();
|
|
w_api_server.as_mut().unwrap().0.stop();
|
|
*w_api_server = None;
|
|
}
|
|
// Stop running Tor service.
|
|
Tor::stop_service(&service_id);
|
|
// Close the wallet.
|
|
let instance = wallet_close.instance.clone().unwrap();
|
|
Self::close_wallet(&instance);
|
|
// Mark wallet as not opened.
|
|
wallet_close.closing.store(false, Ordering::Relaxed);
|
|
wallet_close.is_open.store(false, Ordering::Relaxed);
|
|
// Wake up thread to exit.
|
|
wallet_close.sync(true);
|
|
});
|
|
}
|
|
|
|
/// Close wallet for provided [`WalletInstance`].
|
|
fn close_wallet(instance: &WalletInstance) {
|
|
let mut wallet_lock = instance.lock();
|
|
let lc = wallet_lock.lc_provider().unwrap();
|
|
let _ = lc.close_wallet(None);
|
|
}
|
|
|
|
/// Create account into wallet.
|
|
pub fn create_account(&self, label: &String) -> Result<(), Error> {
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
api.create_account_path(m, label)?;
|
|
|
|
// Update account list at separate thread.
|
|
if let Some(data) = self.get_data() {
|
|
let last_height = data.info.last_confirmed_height;
|
|
let wallet = self.clone();
|
|
thread::spawn(move || {
|
|
update_accounts(&wallet, last_height, None);
|
|
});
|
|
}
|
|
|
|
// Sync wallet data.
|
|
self.sync(false);
|
|
Ok(())
|
|
})
|
|
}
|
|
|
|
/// Set active account from provided label.
|
|
pub fn set_active_account(&mut self, label: &String) -> Result<(), Error> {
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
api.set_active_account(m, label)?;
|
|
// Set Slatepack address.
|
|
let mut w_address = self.slatepack_address.write();
|
|
*w_address = Some(api.get_slatepack_address(m, 0)?.to_string());
|
|
Ok(())
|
|
})?;
|
|
|
|
// Stop service from previous account.
|
|
let cur_service_id = self.identifier();
|
|
Tor::stop_service(&cur_service_id);
|
|
|
|
// Save account label into config.
|
|
let mut w_config = self.config.write();
|
|
w_config.account = label.to_owned();
|
|
w_config.save();
|
|
|
|
// Clear wallet info.
|
|
let mut w_data = self.data.write();
|
|
*w_data = None;
|
|
|
|
// Reset progress values.
|
|
self.info_sync_progress.store(0, Ordering::Relaxed);
|
|
|
|
// Sync wallet data.
|
|
self.sync(false);
|
|
Ok(())
|
|
}
|
|
|
|
/// Get list of accounts for the wallet.
|
|
pub fn accounts(&self) -> Vec<WalletAccount> {
|
|
self.accounts.read().clone()
|
|
}
|
|
|
|
/// Set wallet reopen status.
|
|
pub fn set_reopen(&self, reopen: bool) {
|
|
self.reopen.store(reopen, Ordering::Relaxed);
|
|
}
|
|
|
|
/// Check if wallet reopen is needed.
|
|
pub fn reopen_needed(&self) -> bool {
|
|
self.reopen.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Get wallet info synchronization progress.
|
|
pub fn info_sync_progress(&self) -> u8 {
|
|
self.info_sync_progress.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Check if wallet had an error on synchronization.
|
|
pub fn sync_error(&self) -> bool {
|
|
self.sync_error.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Set an error for wallet on synchronization.
|
|
pub fn set_sync_error(&self, error: bool) {
|
|
self.sync_error.store(error, Ordering::Relaxed);
|
|
}
|
|
|
|
/// Get current wallet synchronization attempts before setting an error.
|
|
fn get_sync_attempts(&self) -> u8 {
|
|
self.sync_attempts.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Increment wallet synchronization attempts before setting an error.
|
|
fn increment_sync_attempts(&self) {
|
|
let mut attempts = self.get_sync_attempts();
|
|
attempts += 1;
|
|
self.sync_attempts.store(attempts, Ordering::Relaxed);
|
|
}
|
|
|
|
/// Reset wallet synchronization attempts.
|
|
fn reset_sync_attempts(&self) {
|
|
self.sync_attempts.store(0, Ordering::Relaxed);
|
|
}
|
|
|
|
/// Get wallet data.
|
|
pub fn get_data(&self) -> Option<WalletData> {
|
|
let r_data = self.data.read();
|
|
r_data.clone()
|
|
}
|
|
|
|
/// Sync wallet data from node or locally.
|
|
pub fn sync(&self, from_node: bool) {
|
|
if from_node {
|
|
let thread_r = self.sync_thread.read();
|
|
if let Some(thread) = thread_r.as_ref() {
|
|
thread.unpark();
|
|
}
|
|
} else {
|
|
let wallet = self.clone();
|
|
thread::spawn(move || {
|
|
sync_wallet_data(&wallet, false);
|
|
});
|
|
}
|
|
}
|
|
|
|
/// Check if wallet is syncing.
|
|
pub fn syncing(&self) -> bool {
|
|
self.syncing.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Get running Foreign API server port.
|
|
pub fn foreign_api_port(&self) -> Option<u16> {
|
|
let r_api = self.foreign_api_server.read();
|
|
if r_api.is_some() {
|
|
let api = r_api.as_ref().unwrap();
|
|
return Some(api.1);
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Parse Slatepack message into [`Slate`].
|
|
pub fn parse_slatepack(&self, text: &String) -> Result<Slate, grin_wallet_controller::Error> {
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
return match parse_slatepack(&mut api, None, None, Some(text.clone())) {
|
|
Ok(s) => Ok(s.0),
|
|
Err(e) => Err(e)
|
|
}
|
|
}
|
|
|
|
/// Create Slatepack message from provided slate.
|
|
fn create_slatepack_message(&self, slate: &Slate) -> Result<String, Error> {
|
|
let mut message = "".to_string();
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
message = api.create_slatepack_message(m, &slate, Some(0), vec![])?;
|
|
Ok(())
|
|
})?;
|
|
|
|
// Write Slatepack message to file.
|
|
let slatepack_dir = self.get_config().get_slatepack_path(&slate);
|
|
let mut output = File::create(slatepack_dir)?;
|
|
output.write_all(message.as_bytes())?;
|
|
output.sync_all()?;
|
|
Ok(message)
|
|
}
|
|
|
|
/// Read slatepack from file.
|
|
pub fn read_slatepack(&self, slate: &Slate) -> Option<String> {
|
|
let slatepack_path = self.get_config().get_slatepack_path(slate);
|
|
match fs::read_to_string(slatepack_path) {
|
|
Ok(s) => Some(s),
|
|
Err(_) => None
|
|
}
|
|
}
|
|
|
|
/// Get last stored [`Slate`] for transaction.
|
|
pub fn read_slate_by_tx(&self, tx: &WalletTransaction) -> Option<(Slate, String)> {
|
|
let mut slate = None;
|
|
if let Some(slate_id) = tx.data.tx_slate_id {
|
|
// Get slate state based on tx state and status.
|
|
let state = if tx.posting {
|
|
if tx.data.tx_type == TxLogEntryType::TxSent {
|
|
Some(SlateState::Standard3)
|
|
} else {
|
|
Some(SlateState::Invoice3)
|
|
}
|
|
} else if !tx.data.confirmed && (tx.data.tx_type == TxLogEntryType::TxSent ||
|
|
tx.data.tx_type == TxLogEntryType::TxReceived) {
|
|
if tx.can_finalize {
|
|
if tx.data.tx_type == TxLogEntryType::TxSent {
|
|
Some(SlateState::Standard1)
|
|
} else {
|
|
Some(SlateState::Invoice1)
|
|
}
|
|
} else {
|
|
if tx.data.tx_type == TxLogEntryType::TxReceived {
|
|
Some(SlateState::Standard2)
|
|
} else {
|
|
Some(SlateState::Invoice2)
|
|
}
|
|
}
|
|
} else {
|
|
None
|
|
};
|
|
// Get slate from state by reading Slatepack message file.
|
|
if let Some(st) = state {
|
|
let mut s = Slate::blank(0, false);
|
|
s.id = slate_id;
|
|
s.state = st;
|
|
if let Some(m) = self.read_slatepack(&s) {
|
|
if let Ok(s) = self.parse_slatepack(&m) {
|
|
slate = Some((s, m));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
slate
|
|
}
|
|
|
|
/// Get transaction for [`Slate`] id.
|
|
pub fn tx_by_slate(&self, slate: &Slate) -> Option<WalletTransaction> {
|
|
if let Some(data) = self.get_data() {
|
|
let data_txs = data.txs.unwrap();
|
|
let txs = data_txs.iter().map(|tx| tx.clone()).filter(|tx| {
|
|
tx.data.tx_slate_id == Some(slate.id)
|
|
}).collect::<Vec<WalletTransaction>>();
|
|
return if let Some(tx) = txs.get(0) {
|
|
Some(tx.clone())
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
None
|
|
}
|
|
|
|
/// Initialize a transaction to send amount, return request for funds receiver.
|
|
pub fn send(&self, amount: u64) -> Result<(Slate, String), Error> {
|
|
let config = self.get_config();
|
|
let args = InitTxArgs {
|
|
src_acct_name: Some(config.account),
|
|
amount,
|
|
minimum_confirmations: config.min_confirmations,
|
|
num_change_outputs: 1,
|
|
selection_strategy_is_use_all: false,
|
|
..Default::default()
|
|
};
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
let slate = api.init_send_tx(None, args)?;
|
|
|
|
// Lock outputs to for this transaction.
|
|
api.tx_lock_outputs(None, &slate)?;
|
|
|
|
// Create Slatepack message response.
|
|
let message_resp = self.create_slatepack_message(&slate)?;
|
|
|
|
// Sync wallet info.
|
|
self.sync(false);
|
|
|
|
Ok((slate, message_resp))
|
|
}
|
|
|
|
/// Send amount to provided address with Tor transport.
|
|
pub async fn send_tor(&mut self, amount: u64, addr: &SlatepackAddress) -> Option<Slate> {
|
|
// Initialize transaction.
|
|
let send_res = self.send(amount);
|
|
|
|
if send_res.is_err() {
|
|
return None;
|
|
}
|
|
let slate = send_res.unwrap().0;
|
|
|
|
// Function to cancel initialized tx in case of error.
|
|
let cancel_tx = || {
|
|
let instance = self.instance.clone().unwrap();
|
|
let id = slate.clone().id;
|
|
cancel_tx(instance, None, &None, None, Some(id.clone())).unwrap();
|
|
// Setup posting flag, and ability to finalize.
|
|
{
|
|
let mut w_data = self.data.write();
|
|
let mut data = w_data.clone().unwrap();
|
|
let txs = data.txs.clone().unwrap().iter_mut().map(|tx| {
|
|
if tx.data.tx_slate_id == Some(id) {
|
|
tx.cancelling = false;
|
|
tx.posting = false;
|
|
tx.can_finalize = false;
|
|
tx.data.tx_type = if tx.data.tx_type == TxLogEntryType::TxReceived {
|
|
TxLogEntryType::TxReceivedCancelled
|
|
} else {
|
|
TxLogEntryType::TxSentCancelled
|
|
};
|
|
}
|
|
tx.clone()
|
|
}).collect::<Vec<WalletTransaction>>();
|
|
data.txs = Some(txs);
|
|
*w_data = Some(data);
|
|
}
|
|
// Refresh wallet info to update statuses.
|
|
self.sync(false);
|
|
};
|
|
|
|
// Initialize parameters.
|
|
let tor_addr = OnionV3Address::try_from(addr).unwrap().to_http_str();
|
|
let url = format!("{}/v2/foreign", tor_addr);
|
|
let slate_send = VersionedSlate::into_version(slate.clone(), SlateVersion::V4).unwrap();
|
|
let body = json!({
|
|
"jsonrpc": "2.0",
|
|
"method": "receive_tx",
|
|
"id": 1,
|
|
"params": [
|
|
slate_send,
|
|
null,
|
|
null
|
|
]
|
|
}).to_string();
|
|
|
|
// Send request to receiver.
|
|
let req_res = Tor::post(body, url).await;
|
|
if req_res.is_none() {
|
|
cancel_tx();
|
|
return None;
|
|
}
|
|
|
|
// Parse response and finalize transaction.
|
|
let res: Value = serde_json::from_str(&req_res.unwrap()).unwrap();
|
|
if res["error"] != json!(null) {
|
|
cancel_tx();
|
|
return None;
|
|
}
|
|
|
|
// Slatepack message json value.
|
|
let slate_value = res["result"]["Ok"].clone();
|
|
|
|
let mut ret_slate = None;
|
|
match Slate::deserialize_upgrade(&serde_json::to_string(&slate_value).unwrap()) {
|
|
Ok(s) => {
|
|
let mut api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
// Finalize transaction.
|
|
return if let Ok(slate) = api.finalize_tx(m, &s) {
|
|
ret_slate = Some(slate.clone());
|
|
// Save Slatepack message to file.
|
|
let _ = self.create_slatepack_message(&slate).unwrap_or("".to_string());
|
|
// Post transaction to blockchain.
|
|
let result = self.post(&slate, self.can_use_dandelion());
|
|
match result {
|
|
Ok(_) => {
|
|
Ok(())
|
|
}
|
|
Err(e) => {
|
|
Err(e)
|
|
}
|
|
}
|
|
} else {
|
|
Err(Error::GenericError("TX finalization error".to_string()))
|
|
};
|
|
}).unwrap();
|
|
}
|
|
Err(_) => {}
|
|
};
|
|
|
|
if ret_slate.is_none() {
|
|
cancel_tx();
|
|
}
|
|
ret_slate
|
|
}
|
|
|
|
/// Initialize an invoice transaction to receive amount, return request for funds sender.
|
|
pub fn issue_invoice(&self, amount: u64) -> Result<(Slate, String), Error> {
|
|
let args = IssueInvoiceTxArgs {
|
|
dest_acct_name: None,
|
|
amount,
|
|
target_slate_version: None,
|
|
};
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
let slate = api.issue_invoice_tx(None, args)?;
|
|
|
|
// Create Slatepack message response.
|
|
let response = self.create_slatepack_message(&slate.clone())?;
|
|
|
|
// Sync wallet info.
|
|
self.sync(false);
|
|
|
|
Ok((slate, response))
|
|
}
|
|
|
|
/// Handle message from the invoice issuer to send founds, return response for funds receiver.
|
|
pub fn pay(&self, message: &String) -> Result<String, Error> {
|
|
if let Ok(slate) = self.parse_slatepack(message) {
|
|
let config = self.get_config();
|
|
let args = InitTxArgs {
|
|
src_acct_name: None,
|
|
amount: slate.amount,
|
|
minimum_confirmations: config.min_confirmations,
|
|
selection_strategy_is_use_all: false,
|
|
..Default::default()
|
|
};
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
let slate = api.process_invoice_tx(None, &slate, args)?;
|
|
api.tx_lock_outputs(None, &slate)?;
|
|
|
|
// Create Slatepack message response.
|
|
let response = self.create_slatepack_message(&slate)?;
|
|
|
|
// Sync wallet info.
|
|
self.sync(false);
|
|
|
|
Ok(response)
|
|
} else {
|
|
Err(Error::SlatepackDeser("Slatepack parsing error".to_string()))
|
|
}
|
|
}
|
|
|
|
/// Handle message to receive funds, return response to sender.
|
|
pub fn receive(&self, message: &String) -> Result<String, Error> {
|
|
if let Ok(mut slate) = self.parse_slatepack(message) {
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
controller::foreign_single_use(api.wallet_inst.clone(), None, |api| {
|
|
slate = api.receive_tx(&slate, Some(self.get_config().account.as_str()), None)?;
|
|
Ok(())
|
|
})?;
|
|
// Create Slatepack message response.
|
|
let response = self.create_slatepack_message(&slate)?;
|
|
|
|
// Sync wallet info.
|
|
self.sync(false);
|
|
|
|
Ok(response)
|
|
} else {
|
|
Err(Error::SlatepackDeser("Slatepack parsing error".to_string()))
|
|
}
|
|
}
|
|
|
|
/// Finalize transaction from provided message as sender or invoice issuer with Dandelion.
|
|
pub fn finalize(&self, message: &String, dandelion: bool) -> Result<Slate, Error> {
|
|
if let Ok(mut slate) = self.parse_slatepack(message) {
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
slate = api.finalize_tx(None, &slate)?;
|
|
// Save Slatepack message to file.
|
|
let _ = self.create_slatepack_message(&slate)?;
|
|
// Post transaction to blockchain.
|
|
let _ = self.post(&slate, dandelion);
|
|
Ok(slate)
|
|
} else {
|
|
Err(Error::SlatepackDeser("Slatepack parsing error".to_string()))
|
|
}
|
|
}
|
|
|
|
/// Post transaction to blockchain.
|
|
pub fn post(&self, slate: &Slate, dandelion: bool) -> Result<(), Error> {
|
|
// Post transaction to blockchain.
|
|
let api = Owner::new(self.instance.clone().unwrap(), None);
|
|
api.post_tx(None, slate, dandelion)?;
|
|
// Setup transaction repost height, posting flag and ability to finalize.
|
|
let mut slate = slate.clone();
|
|
if slate.state == SlateState::Invoice2 {
|
|
slate.state = SlateState::Invoice3
|
|
} else if slate.state == SlateState::Standard2 {
|
|
slate.state = SlateState::Standard3
|
|
};
|
|
if let Some(tx) = self.tx_by_slate(&slate) {
|
|
let mut w_data = self.data.write();
|
|
let mut data = w_data.clone().unwrap();
|
|
let mut data_txs = data.txs.unwrap();
|
|
for t in &mut data_txs {
|
|
if t.data.id == tx.data.id {
|
|
t.repost_height = Some(data.info.last_confirmed_height);
|
|
t.posting = true;
|
|
t.can_finalize = false;
|
|
}
|
|
}
|
|
data.txs = Some(data_txs);
|
|
*w_data = Some(data);
|
|
}
|
|
// Sync local wallet info.
|
|
self.sync(false);
|
|
Ok(())
|
|
}
|
|
|
|
/// Cancel transaction.
|
|
pub fn cancel(&mut self, id: u32) {
|
|
// Setup cancelling status.
|
|
{
|
|
let mut w_data = self.data.write();
|
|
let mut data = w_data.clone().unwrap();
|
|
let txs = data.txs.clone().unwrap().iter_mut().map(|tx| {
|
|
if tx.data.id == id {
|
|
tx.cancelling = true;
|
|
tx.can_finalize = false;
|
|
}
|
|
tx.clone()
|
|
}).collect::<Vec<WalletTransaction>>();
|
|
data.txs = Some(txs);
|
|
*w_data = Some(data);
|
|
}
|
|
|
|
let wallet = self.clone();
|
|
thread::spawn(move || {
|
|
// Wait sync to finish.
|
|
if wallet.syncing() {
|
|
thread::sleep(Duration::from_millis(1000));
|
|
}
|
|
let instance = wallet.instance.clone().unwrap();
|
|
let _ = cancel_tx(instance, None, &None, Some(id), None);
|
|
// Setup tx status, cancelling, posting flag, and ability to finalize.
|
|
{
|
|
let mut w_data = wallet.data.write();
|
|
let mut data = w_data.clone().unwrap();
|
|
let mut data_txs = data.txs.unwrap();
|
|
let txs = data_txs.iter_mut().map(|tx| {
|
|
if tx.data.id == id {
|
|
tx.cancelling = false;
|
|
tx.posting = false;
|
|
tx.can_finalize = false;
|
|
tx.data.tx_type = if tx.data.tx_type == TxLogEntryType::TxReceived {
|
|
TxLogEntryType::TxReceivedCancelled
|
|
} else {
|
|
TxLogEntryType::TxSentCancelled
|
|
};
|
|
}
|
|
tx.clone()
|
|
}).collect::<Vec<WalletTransaction>>();
|
|
data.txs = Some(txs);
|
|
*w_data = Some(data);
|
|
}
|
|
// Refresh wallet info to update statuses.
|
|
sync_wallet_data(&wallet, false);
|
|
});
|
|
}
|
|
|
|
/// Change wallet password.
|
|
pub fn change_password(&self, old: String, new: String) -> Result<(), Error> {
|
|
let instance = self.instance.clone().unwrap();
|
|
let mut wallet_lock = instance.lock();
|
|
let lc = wallet_lock.lc_provider()?;
|
|
lc.change_password(None, ZeroingString::from(old), ZeroingString::from(new))
|
|
}
|
|
|
|
/// Initiate wallet repair by scanning its outputs.
|
|
pub fn repair(&self) {
|
|
self.repair_needed.store(true, Ordering::Relaxed);
|
|
self.sync(true);
|
|
}
|
|
|
|
/// Check if wallet is repairing.
|
|
pub fn is_repairing(&self) -> bool {
|
|
self.repair_needed.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Get wallet repairing progress.
|
|
pub fn repairing_progress(&self) -> u8 {
|
|
self.repair_progress.load(Ordering::Relaxed)
|
|
}
|
|
|
|
/// Deleting wallet database files.
|
|
pub fn delete_db(&self, reopen: bool) {
|
|
let wallet_delete = self.clone();
|
|
// Close wallet if open.
|
|
if self.is_open() {
|
|
self.close();
|
|
}
|
|
thread::spawn(move || {
|
|
// Wait wallet to be closed.
|
|
if wallet_delete.is_open() {
|
|
thread::sleep(Duration::from_millis(300));
|
|
}
|
|
// Remove wallet db files.
|
|
let _ = fs::remove_dir_all(wallet_delete.get_config().get_db_path());
|
|
// Start sync to close thread.
|
|
wallet_delete.sync(true);
|
|
// Mark wallet to reopen.
|
|
wallet_delete.set_reopen(reopen);
|
|
});
|
|
}
|
|
|
|
/// Get recovery phrase.
|
|
pub fn get_recovery(&self, password: String) -> Result<ZeroingString, Error> {
|
|
let instance = self.instance.clone().unwrap();
|
|
let mut wallet_lock = instance.lock();
|
|
let lc = wallet_lock.lc_provider().unwrap();
|
|
lc.get_mnemonic(None, ZeroingString::from(password))
|
|
}
|
|
|
|
/// Close the wallet, delete its files and mark it as deleted.
|
|
pub fn delete_wallet(&self) {
|
|
if self.is_open() {
|
|
self.close();
|
|
}
|
|
// Mark wallet as deleted.
|
|
let wallet_delete = self.clone();
|
|
wallet_delete.deleted.store(true, Ordering::Relaxed);
|
|
|
|
thread::spawn(move || {
|
|
// Wait wallet to be closed.
|
|
if wallet_delete.is_open() {
|
|
thread::sleep(Duration::from_millis(100));
|
|
}
|
|
// Remove wallet files.
|
|
let _ = fs::remove_dir_all(wallet_delete.get_config().get_data_path());
|
|
// Mark wallet as deleted.
|
|
wallet_delete.deleted.store(true, Ordering::Relaxed);
|
|
// Start sync to close thread.
|
|
wallet_delete.sync(true);
|
|
});
|
|
}
|
|
|
|
/// Check if wallet was deleted to remove it from list.
|
|
pub fn is_deleted(&self) -> bool {
|
|
self.deleted.load(Ordering::Relaxed)
|
|
}
|
|
}
|
|
|
|
/// Delay in seconds to sync [`WalletData`] (60 seconds as average block time).
|
|
const SYNC_DELAY: Duration = Duration::from_millis(60 * 1000);
|
|
|
|
/// Delay in seconds for sync thread to wait before start of new attempt.
|
|
const ATTEMPT_DELAY: Duration = Duration::from_millis(3 * 1000);
|
|
|
|
/// Number of attempts to sync [`WalletData`] before setting an error.
|
|
const SYNC_ATTEMPTS: u8 = 10;
|
|
|
|
/// Launch thread to sync wallet data from node.
|
|
fn start_sync(wallet: Wallet) -> Thread {
|
|
// Reset progress values.
|
|
wallet.info_sync_progress.store(0, Ordering::Relaxed);
|
|
wallet.repair_progress.store(0, Ordering::Relaxed);
|
|
|
|
// To call on sync thread stop.
|
|
let on_thread_stop = |wallet: Wallet| {
|
|
// Clear thread instance.
|
|
let mut thread_w = wallet.sync_thread.write();
|
|
*thread_w = None;
|
|
|
|
// Clear wallet info.
|
|
let mut w_data = wallet.data.write();
|
|
*w_data = None;
|
|
|
|
// Clear syncing status.
|
|
wallet.syncing.store(false, Ordering::Relaxed);
|
|
};
|
|
|
|
thread::spawn(move || loop {
|
|
// Set syncing status.
|
|
wallet.syncing.store(true, Ordering::Relaxed);
|
|
|
|
// Close wallet on chain type change.
|
|
if wallet.get_config().chain_type != AppConfig::chain_type() {
|
|
wallet.close();
|
|
}
|
|
|
|
// Stop syncing if wallet was closed.
|
|
if !wallet.is_open() || wallet.is_closing() {
|
|
on_thread_stop(wallet);
|
|
return;
|
|
}
|
|
|
|
// Check integrated node state.
|
|
if wallet.get_current_ext_conn().is_none() {
|
|
let not_enabled = !Node::is_running() || Node::is_stopping();
|
|
if not_enabled {
|
|
// Reset loading progress.
|
|
wallet.info_sync_progress.store(0, Ordering::Relaxed);
|
|
}
|
|
// Set an error when required integrated node is not enabled.
|
|
wallet.set_sync_error(not_enabled);
|
|
// Skip cycle when node sync is not finished.
|
|
if !Node::is_running() || Node::get_sync_status() != Some(SyncStatus::NoSync) {
|
|
thread::park_timeout(ATTEMPT_DELAY);
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// Scan outputs if repair is needed or sync data if there is no error.
|
|
if !wallet.sync_error() {
|
|
if wallet.is_repairing() {
|
|
repair_wallet(&wallet);
|
|
// Stop sync if wallet was closed.
|
|
if !wallet.is_open() || wallet.is_closing() {
|
|
on_thread_stop(wallet);
|
|
return;
|
|
}
|
|
}
|
|
// Retrieve data from local database if current data is empty.
|
|
if wallet.get_data().is_none() {
|
|
sync_wallet_data(&wallet, false);
|
|
}
|
|
|
|
// Start Foreign API listener if API server is not running.
|
|
let mut api_server_running = {
|
|
wallet.foreign_api_server.read().is_some()
|
|
};
|
|
if !api_server_running && wallet.is_open() && !wallet.is_closing() {
|
|
match start_api_server(&wallet) {
|
|
Ok(api_server) => {
|
|
let mut api_server_w = wallet.foreign_api_server.write();
|
|
*api_server_w = Some(api_server);
|
|
api_server_running = true;
|
|
}
|
|
Err(_) => {}
|
|
}
|
|
}
|
|
|
|
// Start Tor service if API server is running and wallet is open.
|
|
if wallet.auto_start_tor_listener() && wallet.is_open() && !wallet.is_closing() &&
|
|
api_server_running && !Tor::is_service_running(&wallet.identifier()) {
|
|
let r_foreign_api = wallet.foreign_api_server.read();
|
|
let api = r_foreign_api.as_ref().unwrap();
|
|
if let Ok(sec_key) = wallet.secret_key() {
|
|
Tor::start_service(api.1, sec_key, &wallet.identifier());
|
|
}
|
|
}
|
|
|
|
// Sync wallet from node.
|
|
sync_wallet_data(&wallet, true);
|
|
}
|
|
|
|
// Stop sync if wallet was closed.
|
|
if !wallet.is_open() || wallet.is_closing() {
|
|
on_thread_stop(wallet);
|
|
return;
|
|
}
|
|
|
|
// Setup flag to check if sync was failed.
|
|
let failed_sync = wallet.sync_error() || wallet.get_sync_attempts() != 0;
|
|
|
|
// Clear syncing status.
|
|
if !failed_sync {
|
|
wallet.syncing.store(false, Ordering::Relaxed);
|
|
}
|
|
|
|
// Repeat after default or attempt delay if synchronization was not successful.
|
|
let delay = if failed_sync {
|
|
ATTEMPT_DELAY
|
|
} else {
|
|
SYNC_DELAY
|
|
};
|
|
thread::park_timeout(delay);
|
|
}).thread().clone()
|
|
}
|
|
|
|
/// Retrieve [`WalletData`] from local base or node.
|
|
fn sync_wallet_data(wallet: &Wallet, from_node: bool) {
|
|
let fresh_sync = wallet.get_data().is_none();
|
|
|
|
// Update info sync progress at separate thread.
|
|
let wallet_info = wallet.clone();
|
|
let (info_tx, info_rx) = mpsc::channel::<StatusMessage>();
|
|
thread::spawn(move || {
|
|
while let Ok(m) = info_rx.recv() {
|
|
match m {
|
|
StatusMessage::UpdatingOutputs(_) => {}
|
|
StatusMessage::UpdatingTransactions(_) => {}
|
|
StatusMessage::FullScanWarn(_) => {}
|
|
StatusMessage::Scanning(_, progress) => {
|
|
wallet_info.info_sync_progress.store(progress, Ordering::Relaxed);
|
|
}
|
|
StatusMessage::ScanningComplete(_) => {
|
|
wallet_info.info_sync_progress.store(100, Ordering::Relaxed);
|
|
}
|
|
StatusMessage::UpdateWarning(_) => {}
|
|
}
|
|
}
|
|
});
|
|
|
|
let config = wallet.get_config();
|
|
|
|
// Retrieve wallet info.
|
|
if let Some(instance) = &wallet.instance {
|
|
if let Ok(info) = retrieve_summary_info(
|
|
instance.clone(),
|
|
None,
|
|
&Some(info_tx),
|
|
from_node,
|
|
config.min_confirmations
|
|
) {
|
|
// Do not retrieve txs if wallet was closed or its first sync.
|
|
if !wallet.is_open() || wallet.is_closing() ||
|
|
(!from_node && info.1.last_confirmed_height == 0) {
|
|
return;
|
|
}
|
|
|
|
if wallet.info_sync_progress() == 100 || !from_node {
|
|
// Setup accounts data.
|
|
let last_height = info.1.last_confirmed_height;
|
|
let spendable = if wallet.get_data().is_none() {
|
|
None
|
|
} else {
|
|
Some(info.1.amount_currently_spendable)
|
|
};
|
|
update_accounts(wallet, last_height, spendable);
|
|
|
|
// Update wallet info.
|
|
{
|
|
let mut w_data = wallet.data.write();
|
|
let txs = if w_data.is_some() {
|
|
w_data.clone().unwrap().txs
|
|
} else {
|
|
None
|
|
};
|
|
*w_data = Some(WalletData { info: info.1.clone(), txs });
|
|
}
|
|
|
|
// Retrieve txs from local database.
|
|
let txs_args = RetrieveTxQueryArgs {
|
|
exclude_cancelled: Some(false),
|
|
sort_field: Some(RetrieveTxQuerySortField::CreationTimestamp),
|
|
sort_order: Some(RetrieveTxQuerySortOrder::Desc),
|
|
..Default::default()
|
|
};
|
|
if let Ok(txs) = retrieve_txs(instance.clone(),
|
|
None,
|
|
&None,
|
|
false,
|
|
None,
|
|
None,
|
|
Some(txs_args)) {
|
|
// Exit if wallet was closed.
|
|
if !wallet.is_open() {
|
|
return;
|
|
}
|
|
// Reset sync attempts.
|
|
wallet.reset_sync_attempts();
|
|
|
|
// Filter transactions for current account.
|
|
let filter_txs = txs.1.iter().map(|v| v.clone()).filter(|tx| {
|
|
match wallet.get_parent_key_id() {
|
|
Ok(key) => {
|
|
tx.parent_key_id == key
|
|
}
|
|
Err(_) => {
|
|
true
|
|
}
|
|
}
|
|
}).collect::<Vec<TxLogEntry>>();
|
|
|
|
// Initialize tx confirmation height storage.
|
|
let tx_height_store = TxHeightStore::new(config.get_extra_db_path());
|
|
let data = wallet.get_data().unwrap();
|
|
let data_txs = data.txs.unwrap_or(vec![]);
|
|
|
|
// Create wallet txs.
|
|
let mut new_txs: Vec<WalletTransaction> = vec![];
|
|
for tx in &filter_txs {
|
|
// Setup transaction amount.
|
|
let amount = if tx.amount_debited > tx.amount_credited {
|
|
tx.amount_debited - tx.amount_credited
|
|
} else {
|
|
tx.amount_credited - tx.amount_debited
|
|
};
|
|
|
|
let unconfirmed_sent_or_received = tx.tx_slate_id.is_some() &&
|
|
!tx.confirmed && (tx.tx_type == TxLogEntryType::TxSent ||
|
|
tx.tx_type == TxLogEntryType::TxReceived);
|
|
|
|
// Setup transaction posting status based on slate state.
|
|
let posting = if unconfirmed_sent_or_received {
|
|
// Create slate to check existing file.
|
|
let is_invoice = tx.tx_type == TxLogEntryType::TxReceived;
|
|
let mut slate = Slate::blank(0, is_invoice);
|
|
slate.id = tx.tx_slate_id.unwrap();
|
|
slate.state = match is_invoice {
|
|
true => SlateState::Invoice3,
|
|
_ => SlateState::Standard3
|
|
};
|
|
|
|
// Setup posting status if we have other tx with same slate id.
|
|
let mut same_tx_posting = false;
|
|
for t in &mut new_txs {
|
|
if t.data.tx_slate_id == tx.tx_slate_id &&
|
|
tx.tx_type != t.data.tx_type {
|
|
same_tx_posting = t.posting ||
|
|
wallet.read_slatepack(&slate).is_some();
|
|
if same_tx_posting && !t.posting {
|
|
t.posting = true;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
same_tx_posting || wallet.read_slatepack(&slate).is_some()
|
|
} else {
|
|
false
|
|
};
|
|
|
|
// Setup flag for ability to finalize transaction.
|
|
let can_finalize = if !posting && unconfirmed_sent_or_received {
|
|
// Check existing file.
|
|
let mut slate = Slate::blank(1, false);
|
|
slate.id = tx.tx_slate_id.unwrap();
|
|
slate.state = match tx.tx_type {
|
|
TxLogEntryType::TxReceived => SlateState::Invoice1,
|
|
_ => SlateState::Standard1
|
|
};
|
|
wallet.read_slatepack(&slate).is_some()
|
|
} else {
|
|
false
|
|
};
|
|
|
|
// Setup confirmation, reposting height and cancelling status.
|
|
let mut conf_height = None;
|
|
let mut setup_conf_height = |t: &TxLogEntry, current_empty: bool| -> bool {
|
|
if current_empty && t.kernel_lookup_min_height.is_some() &&
|
|
t.kernel_excess.is_some() && t.confirmed {
|
|
// Get tx height from database or from node.
|
|
if let Some(height) = tx_height_store.read_tx_height(t.id) {
|
|
conf_height = Some(height);
|
|
} else {
|
|
let mut w_lock = wallet.instance.as_ref().unwrap().lock();
|
|
let w = w_lock.lc_provider()
|
|
.unwrap()
|
|
.wallet_inst()
|
|
.unwrap();
|
|
if let Ok(res) = w.w2n_client().get_kernel(
|
|
t.kernel_excess.as_ref().unwrap(),
|
|
t.kernel_lookup_min_height,
|
|
None
|
|
) {
|
|
if res.is_some() {
|
|
let h = res.unwrap().1;
|
|
conf_height = Some(h);
|
|
tx_height_store.write_tx_height(t.id, h);
|
|
} else {
|
|
conf_height = Some(0);
|
|
}
|
|
} else {
|
|
conf_height = None;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
false
|
|
};
|
|
|
|
let mut repost_height = None;
|
|
let mut cancelling = false;
|
|
if data_txs.is_empty() {
|
|
setup_conf_height(tx, true);
|
|
} else {
|
|
for t in &data_txs {
|
|
if t.data.id == tx.id {
|
|
if !setup_conf_height(tx, t.conf_height.is_none() ||
|
|
t.conf_height.unwrap() == 0) {
|
|
conf_height = t.conf_height;
|
|
}
|
|
repost_height = t.repost_height;
|
|
if t.cancelling &&
|
|
tx.tx_type != TxLogEntryType::TxReceivedCancelled &&
|
|
tx.tx_type != TxLogEntryType::TxSentCancelled {
|
|
cancelling = true;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Add transaction to the list.
|
|
new_txs.push(WalletTransaction {
|
|
data: tx.clone(),
|
|
amount,
|
|
cancelling,
|
|
posting,
|
|
can_finalize,
|
|
conf_height,
|
|
repost_height,
|
|
from_node: !fresh_sync || from_node
|
|
});
|
|
}
|
|
|
|
// Update wallet txs.
|
|
let mut w_data = wallet.data.write();
|
|
let info = if w_data.is_some() {
|
|
w_data.clone().unwrap().info
|
|
} else {
|
|
info.1
|
|
};
|
|
*w_data = Some(WalletData { info, txs: Some(new_txs) });
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Reset progress.
|
|
wallet.info_sync_progress.store(0, Ordering::Relaxed);
|
|
|
|
// Exit if wallet was closed.
|
|
if !wallet.is_open() {
|
|
return;
|
|
}
|
|
|
|
// Set an error if data was not loaded after opening or increment attempts count.
|
|
if wallet.get_data().is_none() {
|
|
wallet.set_sync_error(true);
|
|
} else {
|
|
wallet.increment_sync_attempts();
|
|
}
|
|
|
|
// Set an error if maximum number of attempts was reached.
|
|
if wallet.get_sync_attempts() >= SYNC_ATTEMPTS {
|
|
wallet.reset_sync_attempts();
|
|
wallet.set_sync_error(true);
|
|
}
|
|
}
|
|
|
|
/// Start Foreign API server to receive txs over transport and mining rewards.
|
|
fn start_api_server(wallet: &Wallet) -> Result<(ApiServer, u16), Error> {
|
|
let host = "127.0.0.1";
|
|
// Find free port.
|
|
let port = if wallet.get_config().chain_type == ChainTypes::Mainnet {
|
|
rand::thread_rng().gen_range(37000..40000)
|
|
} else {
|
|
rand::thread_rng().gen_range(47000..50000)
|
|
};
|
|
let free_port = (port..).find(|port| {
|
|
return match TcpListener::bind((host, port.to_owned())) {
|
|
Ok(_) => {
|
|
let node_p2p_port = NodeConfig::get_p2p_port();
|
|
let node_api_port = NodeConfig::get_api_ip_port().1;
|
|
port.to_string() != node_p2p_port && port.to_string() != node_api_port
|
|
},
|
|
Err(_) => false
|
|
}
|
|
}).unwrap();
|
|
|
|
// Setup API server address.
|
|
let api_addr = format!("{}:{}", host, free_port);
|
|
|
|
// Start Foreign API server thread.
|
|
let instance = wallet.instance.clone().unwrap();
|
|
let api_handler_v2 = ForeignAPIHandlerV2::new(instance,
|
|
Arc::new(Mutex::new(None)),
|
|
false,
|
|
Mutex::new(None));
|
|
let mut router = Router::new();
|
|
router
|
|
.add_route("/v2/foreign", Arc::new(api_handler_v2))
|
|
.map_err(|_| Error::GenericError("Router failed to add route".to_string()))?;
|
|
|
|
let api_chan: &'static mut (oneshot::Sender<()>, oneshot::Receiver<()>) =
|
|
Box::leak(Box::new(oneshot::channel::<()>()));
|
|
|
|
let mut apis = ApiServer::new();
|
|
let socket_addr: SocketAddr = api_addr.parse().unwrap();
|
|
let _ = apis.start(socket_addr, router, None, api_chan)
|
|
.map_err(|_| Error::GenericError("API thread failed to start".to_string()))?;
|
|
Ok((apis, free_port))
|
|
}
|
|
|
|
/// Update wallet accounts data.
|
|
fn update_accounts(wallet: &Wallet, current_height: u64, current_spendable: Option<u64>) {
|
|
// Update only current account if list is not empty.
|
|
if current_spendable.is_some() {
|
|
let mut accounts = wallet.accounts.read().clone();
|
|
for a in accounts.iter_mut() {
|
|
if a.label == wallet.get_config().account {
|
|
a.spendable_amount = current_spendable.unwrap();
|
|
}
|
|
}
|
|
// Save accounts data.
|
|
let mut w_data = wallet.accounts.write();
|
|
*w_data = accounts;
|
|
} else {
|
|
let mut api = Owner::new(wallet.instance.clone().unwrap(), None);
|
|
let _ = controller::owner_single_use(None, None, Some(&mut api), |api, m| {
|
|
let mut accounts = vec![];
|
|
for a in api.accounts(m)? {
|
|
api.set_active_account(m, a.label.as_str())?;
|
|
// Calculate wallet account balance.
|
|
let outputs = api.retrieve_outputs(m, true, false, None)?;
|
|
let min_confirmations = wallet.get_config().min_confirmations;
|
|
let mut spendable_amount = 0;
|
|
for out_mapping in outputs.1 {
|
|
let out = out_mapping.output;
|
|
if out.status == grin_wallet_libwallet::OutputStatus::Unspent {
|
|
if !out.is_coinbase || out.lock_height <= current_height
|
|
|| out.num_confirmations(current_height) >= min_confirmations {
|
|
spendable_amount += out.value;
|
|
}
|
|
}
|
|
}
|
|
|
|
// Add account to list.
|
|
accounts.push(WalletAccount {
|
|
spendable_amount,
|
|
label: a.label,
|
|
path: a.path.to_bip_32_string(),
|
|
});
|
|
}
|
|
// Sort in reverse.
|
|
accounts.reverse();
|
|
|
|
// Save accounts data.
|
|
let mut w_data = wallet.accounts.write();
|
|
*w_data = accounts;
|
|
|
|
// Set current active account from config.
|
|
api.set_active_account(m, wallet.get_config().account.as_str())?;
|
|
|
|
Ok(())
|
|
});
|
|
}
|
|
}
|
|
|
|
/// Scan wallet's outputs, repairing and restoring missing outputs if required.
|
|
fn repair_wallet(wallet: &Wallet) {
|
|
let (info_tx, info_rx) = mpsc::channel::<StatusMessage>();
|
|
// Update scan progress at separate thread.
|
|
let wallet_scan = wallet.clone();
|
|
thread::spawn(move || {
|
|
while let Ok(m) = info_rx.recv() {
|
|
match m {
|
|
StatusMessage::UpdatingOutputs(_) => {}
|
|
StatusMessage::UpdatingTransactions(_) => {}
|
|
StatusMessage::FullScanWarn(_) => {}
|
|
StatusMessage::Scanning(_, progress) => {
|
|
wallet_scan.repair_progress.store(progress, Ordering::Relaxed);
|
|
}
|
|
StatusMessage::ScanningComplete(_) => {
|
|
wallet_scan.repair_progress.store(100, Ordering::Relaxed);
|
|
}
|
|
StatusMessage::UpdateWarning(_) => {}
|
|
}
|
|
}
|
|
});
|
|
|
|
// Start wallet scanning.
|
|
let api = Owner::new(wallet.instance.clone().unwrap(), Some(info_tx));
|
|
match api.scan(None, Some(1), false) {
|
|
Ok(()) => {
|
|
// Set sync error if scanning was not complete and wallet is open.
|
|
if wallet.is_open() && wallet.repair_progress.load(Ordering::Relaxed) != 100 {
|
|
wallet.set_sync_error(true);
|
|
} else {
|
|
wallet.repair_needed.store(false, Ordering::Relaxed);
|
|
}
|
|
}
|
|
Err(_) => {
|
|
// Set sync error if wallet is open.
|
|
if wallet.is_open() {
|
|
wallet.set_sync_error(true);
|
|
} else {
|
|
wallet.repair_needed.store(false, Ordering::Relaxed);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Reset repair progress.
|
|
wallet.repair_progress.store(0, Ordering::Relaxed);
|
|
} |