123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187 |
- #!/usr/bin/env python
- #
- # mmgen = Multi-Mode GENerator, command-line Bitcoin cold storage solution
- # Copyright (C) 2013-2014 by philemon <mmgen-py@yandex.com>
- #
- # This program is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, either version 3 of the License, or
- # (at your option) any later version.
- #
- # This program 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 General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- """
- mmgen-addrgen: Generate a list or range of addresses from a mmgen
- deterministic wallet.
- Call as 'btc-keygen' to allow key generation as well.
- """
- import sys
- import mmgen.config as g
- from mmgen.Opts import *
- from mmgen.license import *
- from mmgen.util import *
- from mmgen.addr import *
- from mmgen.tx import make_addr_data_chksum
- what = "keys" if sys.argv[0].split("-")[-1] == "keygen" else "addresses"
- help_data = {
- 'prog_name': g.prog_name,
- 'desc': """Generate a list or range of {} from an {g.proj_name} wallet,
- mnemonic, seed or password""".format(what,g=g),
- 'usage':"[opts] [infile] <address list>",
- 'options': """
- -h, --help Print this help message{}
- -d, --outdir= d Specify an alternate directory 'd' for output
- -c, --save-checksum Save address list checksum to file
- -e, --echo-passphrase Echo passphrase or mnemonic to screen upon entry{}
- -H, --show-hash-presets Show information on available hash presets
- -K, --no-keyconv Use internal libraries for address generation
- instead of 'keyconv'
- -l, --seed-len= N Length of seed. Options: {seed_lens}
- (default: {g.seed_len})
- -p, --hash-preset= p Use scrypt.hash() parameters from preset 'p' when
- hashing password (default: '{g.hash_preset}')
- -P, --passwd-file= f Get passphrase from file 'f'
- -q, --quiet Suppress warnings; overwrite files without
- prompting
- -S, --stdout Print {what} to stdout
- -v, --verbose Produce more verbose output{}
- -b, --from-brain= l,p Generate {what} from a user-created password,
- i.e. a "brainwallet", using seed length 'l' and
- hash preset 'p' (comma-separated)
- -g, --from-incog Generate {what} from an incognito wallet
- -X, --from-incog-hex Generate {what} from incognito hexadecimal wallet
- -G, --from-incog-hidden=f,o,l Generate {what} from incognito data in file
- 'f' at offset 'o', with seed length of 'l'
- -m, --from-mnemonic Generate {what} from an electrum-like mnemonic
- -s, --from-seed Generate {what} from a seed in .{g.seed_ext} format
- """.format(
- *(
- (
- "\n-A, --no-addresses Print only secret keys, no addresses",
- "\n-f, --flat-list Produce a flat list of keys suitable for use with" +
- "\n '{}-txsign'".format(g.proj_name.lower()),
- "\n-x, --b16 Print secret keys in hexadecimal too"
- )
- if what == "keys" else ("","","")),
- seed_lens=", ".join([str(i) for i in g.seed_lens]),
- what=what, g=g
- ),
- 'notes': """
- Addresses are given in a comma-separated list. Hyphen-separated ranges are
- also allowed.{}
- If available, the external 'keyconv' program will be used for address
- generation.
- Data for the --from-<what> options will be taken from <infile> if <infile>
- is specified. Otherwise, the user will be prompted to enter the data.
- For passphrases all combinations of whitespace are equal, and leading and
- trailing space are ignored. This permits reading passphrase data from a
- multi-line file with free spacing and indentation. This is particularly
- convenient for long brainwallet passphrases, for example.
- BRAINWALLET NOTE:
- As brainwallets require especially strong hashing to thwart dictionary
- attacks, the brainwallet hash preset must be specified by the user, using
- the 'p' parameter of the '--from-brain' option
- The '--from-brain' option also requires the user to specify a seed length
- (the 'l' parameter)
- For a brainwallet passphrase to always generate the same keys and addresses,
- the same 'l' and 'p' parameters to '--from-brain' must be used in all future
- invocations with that passphrase
- """.format("\n\nBy default, both addresses and secret keys are generated."
- if what == "keys" else "")
- }
- opts,cmd_args = parse_opts(sys.argv,help_data)
- if 'show_hash_presets' in opts: show_hash_presets()
- if 'verbose' in opts: g.verbose = True
- if 'quiet' in opts: g.quiet = True
- if 'from_incog_hex' in opts or 'from_incog_hidden' in opts:
- opts['from_incog'] = True
- if g.debug: show_opts_and_cmd_args(opts,cmd_args)
- if len(cmd_args) == 1 and (
- 'from_mnemonic' in opts
- or 'from_brain' in opts
- or 'from_seed' in opts
- or 'from_incog_hidden' in opts
- ):
- infile,addr_idx_arg = "",cmd_args[0]
- elif len(cmd_args) == 2:
- infile,addr_idx_arg = cmd_args
- check_infile(infile)
- else: usage(help_data)
- addr_idxs = parse_address_list(addr_idx_arg)
- if not addr_idxs: sys.exit(2)
- do_license_msg()
- # Interact with user:
- if what == "keys" and not g.quiet:
- confirm_or_exit(cmessages['unencrypted_secret_keys'], 'continue')
- # Generate data:
- seed = get_seed_retry(infile,opts)
- seed_id = make_chksum_8(seed)
- for l in (
- ('flat_list', 'no_addresses'),
- ('flat_list', 'b16'),
- ): check_incompatible_opts(opts,l)
- opts['gen_what'] = \
- ("addrs") if what == "addresses" else (
- ("keys") if 'no_addresses' in opts else ("addrs","keys"))
- addr_data = generate_addrs(seed, addr_idxs, opts)
- addr_data_chksum = make_addr_data_chksum([(a.num,a.addr)
- for a in addr_data]) if 'addrs' in opts['gen_what'] else ""
- addr_data_str = format_addr_data(
- addr_data, addr_data_chksum, seed_id, addr_idxs, opts)
- outfile_base = "{}[{}]".format(seed_id, fmt_addr_idxs(addr_idxs))
- # Output data:
- if 'stdout' in opts:
- confirm = True if (what == "keys" and not g.quiet) else False
- write_to_stdout(addr_data_str,what,confirm)
- elif not sys.stdout.isatty():
- write_to_stdout(addr_data_str,what,confirm=False)
- else:
- confirm = False if g.quiet else True
- outfile = outfile_base + "." + (
- g.keylist_ext if 'flat_list' in opts else (
- g.keyfile_ext if opts['gen_what'] == ("keys") else (
- g.addrfile_ext if opts['gen_what'] == ("addrs") else "akeys")))
- write_to_file(outfile,addr_data_str,opts,what,confirm,True)
- if 'addrs' in opts['gen_what']:
- msg("Checksum for address data {}: {}".format(outfile_base,addr_data_chksum))
- if 'save_checksum' in opts:
- a = "address data checksum"
- write_to_file(outfile_base+".chk",addr_data_chksum,opts,a,confirm,True)
- else:
- qmsg("This checksum will be used to verify the address file in the future.")
- qmsg("Record it to a safe location.")
|