Peerplays_Blockchain/peerplays_migrated
1
0
Fork 0
Code Issues 35 Pull requests 4 Projects Releases 20 Packages Wiki Activity Actions

Initial work on BTS1 key import in CLI wallet

Browse source
This commit is contained in:
Vikram Rajkumar 2015-08-14 16:51:04 -04:00
parent 03175707ff
commit be55df25d0
2 changed files with 108 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
libraries/wallet/include/graphene/wallet/wallet.hpp
View file

@ -523,6 +523,10 @@ class wallet_api
*/
bool import_key(string account_name_or_id, string wif_key);
map<string, bool> import_accounts( string filename, string password );
bool import_account_keys( string filename, string password, string src_account_name, string dest_account_name );
/**
* This call will construct a transaction that will claim all balances controled
* by wif_keys and deposit them into the given account.
@ -1255,6 +1259,8 @@ FC_API( graphene::wallet::wallet_api,
(list_account_balances)
(list_assets)
(import_key)
(import_accounts)
(import_account_keys)
(import_balance)
(suggest_brain_key)
(register_account)

102
libraries/wallet/wallet.cpp
View file

@ -2225,6 +2225,108 @@ bool wallet_api::import_key(string account_name_or_id, string wif_key)
return false;
}
map<string, bool> wallet_api::import_accounts( string filename, string password )
{
FC_ASSERT( !is_locked() );
FC_ASSERT( fc::exists( filename ) );
const auto imported_keys = fc::json::from_file<exported_keys>( filename );
const auto password_hash = fc::sha512::hash( password );
FC_ASSERT( fc::sha512::hash( password_hash ) == imported_keys.password_checksum );
map<string, bool> result;
for( const auto& item : imported_keys.account_keys )
{
const auto import_this_account = [ & ]() -> bool
{
try
{
const account_object account = get_account( item.account_name );
const auto& owner_keys = account.owner.get_keys();
const auto& active_keys = account.active.get_keys();
for( const auto& public_key : item.public_keys )
{
if( std::find( owner_keys.begin(), owner_keys.end(), public_key ) != owner_keys.end() )
return true;
if( std::find( active_keys.begin(), active_keys.end(), public_key ) != active_keys.end() )
return true;
}
}
catch( ... )
{
}
return false;
};
const auto should_proceed = import_this_account();
result[ item.account_name ] = should_proceed;
if( should_proceed )
{
// TODO: First check that all private keys match public keys
for( const auto& encrypted_key : item.encrypted_private_keys )
{
const auto plain_text = fc::aes_decrypt( password_hash, encrypted_key );
const auto private_key = fc::raw::unpack<private_key_type>( plain_text );
import_key( item.account_name, string( graphene::utilities::key_to_wif( private_key ) ) );
}
}
}
return result;
}
bool wallet_api::import_account_keys( string filename, string password, string src_account_name, string dest_account_name )
{
FC_ASSERT( !is_locked() );
FC_ASSERT( fc::exists( filename ) );
bool is_my_account = false;
const auto accounts = list_my_accounts();
for( const auto& account : accounts )
{
if( account.name == dest_account_name )
{
is_my_account = true;
break;
}
}
FC_ASSERT( is_my_account );
const auto imported_keys = fc::json::from_file<exported_keys>( filename );
const auto password_hash = fc::sha512::hash( password );
FC_ASSERT( fc::sha512::hash( password_hash ) == imported_keys.password_checksum );
bool found_account = false;
for( const auto& item : imported_keys.account_keys )
{
if( item.account_name != src_account_name )
continue;
found_account = true;
for( const auto& encrypted_key : item.encrypted_private_keys )
{
const auto plain_text = fc::aes_decrypt( password_hash, encrypted_key );
const auto private_key = fc::raw::unpack<private_key_type>( plain_text );
import_key( dest_account_name, string( graphene::utilities::key_to_wif( private_key ) ) );
}
return true;
}
FC_ASSERT( found_account );
return false;
}
string wallet_api::normalize_brain_key(string s) const
{
return detail::normalize_brain_key( s );