259 lines
12 KiB
C++
259 lines
12 KiB
C++
/*
|
|
* Copyright (c) 2015 Cryptonomex, Inc., and contributors.
|
|
*
|
|
* The MIT License
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
#include <algorithm>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include <iterator>
|
|
|
|
#include <fc/io/fstream.hpp>
|
|
#include <fc/io/json.hpp>
|
|
#include <fc/io/stdio.hpp>
|
|
|
|
#include <graphene/app/api.hpp>
|
|
#include <graphene/chain/protocol/address.hpp>
|
|
#include <graphene/chain/protocol/protocol.hpp>
|
|
#include <graphene/egenesis/egenesis.hpp>
|
|
#include <graphene/utilities/key_conversion.hpp>
|
|
|
|
#include <boost/filesystem.hpp>
|
|
|
|
#ifndef WIN32
|
|
#include <csignal>
|
|
#endif
|
|
|
|
using namespace graphene::app;
|
|
using namespace graphene::chain;
|
|
using namespace graphene::utilities;
|
|
using namespace std;
|
|
namespace bpo = boost::program_options;
|
|
|
|
// hack: import create_example_genesis() even though it's a way, way
|
|
// specific internal detail
|
|
namespace graphene { namespace app { namespace detail {
|
|
genesis_state_type create_example_genesis();
|
|
} } } // graphene::app::detail
|
|
|
|
int main( int argc, char** argv )
|
|
{
|
|
try
|
|
{
|
|
bpo::options_description cli_options("Graphene empty blocks");
|
|
cli_options.add_options()
|
|
("help,h", "Print this help message and exit.")
|
|
("genesis-json,g", bpo::value<boost::filesystem::path>(), "File to read genesis state from")
|
|
("out,o", bpo::value<boost::filesystem::path>(), "File to output new genesis to")
|
|
("dev-account-prefix", bpo::value<std::string>()->default_value("devacct"), "Prefix for dev accounts")
|
|
("dev-key-prefix", bpo::value<std::string>()->default_value("devkey-"), "Prefix for dev key")
|
|
("dev-account-count", bpo::value<uint32_t>()->default_value(0), "Prefix for dev accounts")
|
|
("dev-balance-count", bpo::value<uint32_t>()->default_value(0), "Prefix for dev balances")
|
|
("dev-balance-amount", bpo::value<uint64_t>()->default_value(uint64_t(1000)*uint64_t(1000)*uint64_t(100000)), "Amount in each dev balance")
|
|
("nop", "just write the genesis file out after reading it in, do not alter any keys or add accounts or balances. used to pretty-print a genesis file")
|
|
("replace-all-keys", bpo::value<boost::filesystem::path>(), "Replace all keys/addresses in the genesis files with dev keys based on dev-key-prefix and dump the new keys to this filename.")
|
|
;
|
|
|
|
bpo::variables_map options;
|
|
try
|
|
{
|
|
boost::program_options::store( boost::program_options::parse_command_line(argc, argv, cli_options), options );
|
|
}
|
|
catch (const boost::program_options::error& e)
|
|
{
|
|
std::cerr << "empty_blocks: error parsing command line: " << e.what() << "\n";
|
|
return 1;
|
|
}
|
|
|
|
if( options.count("help") )
|
|
{
|
|
std::cout << cli_options << "\n";
|
|
return 1;
|
|
}
|
|
|
|
if( !options.count( "genesis-json" ) )
|
|
{
|
|
std::cerr << "--genesis-json option is required\n";
|
|
return 1;
|
|
}
|
|
|
|
if( !options.count( "out" ) )
|
|
{
|
|
std::cerr << "--out option is required\n";
|
|
return 1;
|
|
}
|
|
|
|
genesis_state_type genesis;
|
|
if( options.count("genesis-json") )
|
|
{
|
|
fc::path genesis_json_filename = options["genesis-json"].as<boost::filesystem::path>();
|
|
std::cerr << "update_genesis: Reading genesis from file " << genesis_json_filename.preferred_string() << "\n";
|
|
std::string genesis_json;
|
|
read_file_contents( genesis_json_filename, genesis_json );
|
|
genesis = fc::json::from_string( genesis_json ).as< genesis_state_type >(20);
|
|
}
|
|
else
|
|
{
|
|
std::cerr << "update_genesis: Using example genesis\n";
|
|
genesis = graphene::app::detail::create_example_genesis();
|
|
}
|
|
|
|
if (!options.count("nop"))
|
|
{
|
|
const std::string dev_key_prefix = options["dev-key-prefix"].as<std::string>();
|
|
auto get_dev_key = [&dev_key_prefix]( std::string prefix, uint32_t i ) -> public_key_type
|
|
{
|
|
return fc::ecc::private_key::regenerate( fc::sha256::hash( dev_key_prefix + prefix + std::to_string(i) ) ).get_public_key();
|
|
};
|
|
|
|
if (options.count("replace-all-keys"))
|
|
{
|
|
unsigned dev_keys_used = 0;
|
|
std::map<std::string, fc::ecc::private_key> replacement_keys;
|
|
auto get_replacement_key = [&](const std::string& original_key) -> fc::ecc::private_key {
|
|
auto iter = replacement_keys.find(original_key);
|
|
if (iter != replacement_keys.end())
|
|
return iter->second;
|
|
fc::ecc::private_key new_private_key = fc::ecc::private_key::regenerate(fc::sha256::hash(dev_key_prefix + std::to_string(dev_keys_used++)));
|
|
replacement_keys[original_key] = new_private_key;
|
|
return new_private_key;
|
|
};
|
|
|
|
for (genesis_state_type::initial_balance_type& initial_balance : genesis.initial_balances)
|
|
{
|
|
std::string address_string = (std::string)initial_balance.owner;
|
|
initial_balance.owner = address(get_replacement_key(address_string).get_public_key());
|
|
}
|
|
|
|
for (genesis_state_type::initial_vesting_balance_type& initial_balance : genesis.initial_vesting_balances)
|
|
{
|
|
std::string address_string = (std::string)initial_balance.owner;
|
|
initial_balance.owner = address(get_replacement_key(address_string).get_public_key());
|
|
}
|
|
|
|
for (genesis_state_type::initial_witness_type& initial_witness : genesis.initial_witness_candidates)
|
|
{
|
|
std::string public_key_string = (std::string)initial_witness.block_signing_key;
|
|
initial_witness.block_signing_key = get_replacement_key(public_key_string).get_public_key();
|
|
}
|
|
|
|
for (genesis_state_type::initial_account_type& initial_account : genesis.initial_accounts)
|
|
{
|
|
std::string public_key_string = (std::string)initial_account.owner_key;
|
|
initial_account.owner_key = get_replacement_key(public_key_string).get_public_key();
|
|
public_key_string = (std::string)initial_account.active_key;
|
|
initial_account.active_key = get_replacement_key(public_key_string).get_public_key();
|
|
}
|
|
|
|
for (genesis_state_type::initial_bts_account_type& initial_account : genesis.initial_bts_accounts)
|
|
{
|
|
for (auto iter = initial_account.owner_authority.key_auths.begin();
|
|
iter != initial_account.owner_authority.key_auths.end(); ++iter)
|
|
{
|
|
std::string public_key_string = (std::string)iter->first;
|
|
iter->first = get_replacement_key(public_key_string).get_public_key();
|
|
}
|
|
for (auto iter = initial_account.active_authority.key_auths.begin();
|
|
iter != initial_account.active_authority.key_auths.end(); ++iter)
|
|
{
|
|
std::string public_key_string = (std::string)iter->first;
|
|
iter->first = get_replacement_key(public_key_string).get_public_key();
|
|
}
|
|
for (auto iter = initial_account.owner_authority.address_auths.begin();
|
|
iter != initial_account.owner_authority.address_auths.end(); ++iter)
|
|
{
|
|
std::string address_string = (std::string)iter->first;
|
|
iter->first = address(get_replacement_key(address_string).get_public_key());
|
|
}
|
|
for (auto iter = initial_account.active_authority.address_auths.begin();
|
|
iter != initial_account.active_authority.address_auths.end(); ++iter)
|
|
{
|
|
std::string address_string = (std::string)iter->first;
|
|
iter->first = address(get_replacement_key(address_string).get_public_key());
|
|
}
|
|
}
|
|
fc::path keys_csv_path = options["replace-all-keys"].as<boost::filesystem::path>();
|
|
std::ofstream keys_csv(keys_csv_path.string());
|
|
keys_csv << "wif_private_key,public_key,address\n";
|
|
for (const auto& value : replacement_keys)
|
|
keys_csv << graphene::utilities::key_to_wif(value.second) << "," << std::string(public_key_type(value.second.get_public_key()))
|
|
<< "," << std::string(address(value.second.get_public_key())) << "\n";
|
|
|
|
|
|
}
|
|
else
|
|
{
|
|
uint32_t dev_account_count = options["dev-account-count"].as<uint32_t>();
|
|
std::string dev_account_prefix = options["dev-account-prefix"].as<std::string>();
|
|
for(uint32_t i=0;i<dev_account_count;i++)
|
|
{
|
|
genesis_state_type::initial_account_type acct(
|
|
dev_account_prefix+std::to_string(i),
|
|
get_dev_key( "owner-", i ),
|
|
get_dev_key( "active-", i ),
|
|
false );
|
|
|
|
genesis.initial_accounts.push_back( acct );
|
|
}
|
|
|
|
uint32_t dev_balance_count = options["dev-balance-count"].as<uint32_t>();
|
|
uint64_t dev_balance_amount = options["dev-balance-amount"].as<uint64_t>();
|
|
for(uint32_t i=0;i<dev_balance_count;i++)
|
|
{
|
|
genesis_state_type::initial_balance_type bal;
|
|
bal.owner = address( get_dev_key( "balance-", i ) );
|
|
bal.asset_symbol = "CORE";
|
|
bal.amount = dev_balance_amount;
|
|
genesis.initial_balances.push_back( bal );
|
|
}
|
|
|
|
std::map< std::string, size_t > name2index;
|
|
size_t num_accounts = genesis.initial_accounts.size();
|
|
for( size_t i=0; i<num_accounts; i++ )
|
|
name2index[ genesis.initial_accounts[i].name ] = i;
|
|
|
|
for( uint32_t i=0; i<genesis.initial_active_witnesses; i++ )
|
|
{
|
|
genesis_state_type::initial_witness_type& wit = genesis.initial_witness_candidates[ i ];
|
|
genesis_state_type::initial_account_type& wit_acct = genesis.initial_accounts[ name2index[ wit.owner_name ] ];
|
|
if( wit.owner_name.substr(0, 4) != "init" )
|
|
{
|
|
std::cerr << "need " << genesis.initial_active_witnesses << " init accounts as first entries in initial_active_witnesses\n";
|
|
return 1;
|
|
}
|
|
wit.block_signing_key = get_dev_key( "wit-block-signing-", i );
|
|
wit_acct.owner_key = get_dev_key( "wit-owner-", i );
|
|
wit_acct.active_key = get_dev_key( "wit-active-", i );
|
|
}
|
|
}
|
|
}
|
|
|
|
fc::path output_filename = options["out"].as<boost::filesystem::path>();
|
|
fc::json::save_to_file( genesis, output_filename );
|
|
}
|
|
catch ( const fc::exception& e )
|
|
{
|
|
std::cout << e.to_detail_string() << "\n";
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|