528 lines
18 KiB
Python
Executable File
528 lines
18 KiB
Python
Executable File
#!/usr/bin/env python2
|
|
# -*- mode: python -*-
|
|
#
|
|
# Electrum - lightweight Bitcoin client
|
|
# Copyright (C) 2011 thomasv@gitorious
|
|
#
|
|
# 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/>.
|
|
|
|
from decimal import Decimal
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
import time
|
|
import traceback
|
|
|
|
import threading
|
|
import socket
|
|
import Queue
|
|
from collections import defaultdict
|
|
|
|
DAEMON_SOCKET = 'daemon.sock'
|
|
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
is_bundle = getattr(sys, 'frozen', False)
|
|
is_local = not is_bundle and os.path.exists(os.path.join(script_dir, "setup-release.py"))
|
|
is_android = 'ANDROID_DATA' in os.environ
|
|
is_kivy = os.environ.get('PYTHONHOME','').find('kivy') != -1
|
|
|
|
if is_local or is_android:
|
|
sys.path.insert(0, os.path.join(script_dir, 'packages'))
|
|
elif is_bundle and sys.platform=='darwin':
|
|
sys.path.insert(0, os.getcwd() + "/lib/python2.7/packages")
|
|
|
|
# pure-python dependencies need to be imported here for pyinstaller
|
|
try:
|
|
import dns
|
|
import aes
|
|
import ecdsa
|
|
import requests
|
|
import six
|
|
import qrcode
|
|
import pbkdf2
|
|
import google.protobuf
|
|
except ImportError as e:
|
|
sys.exit("Error: %s. Try 'sudo pip install <module-name>'"%e.message)
|
|
|
|
# the following imports are for pyinstaller
|
|
from google.protobuf import descriptor
|
|
from google.protobuf import message
|
|
from google.protobuf import reflection
|
|
from google.protobuf import descriptor_pb2
|
|
|
|
|
|
# check that we have the correct version of ecdsa
|
|
try:
|
|
from ecdsa.ecdsa import curve_secp256k1, generator_secp256k1
|
|
except Exception:
|
|
sys.exit("cannot import ecdsa.curve_secp256k1. You probably need to upgrade ecdsa.\nTry: sudo pip install --upgrade ecdsa")
|
|
|
|
|
|
# load local module as electrum
|
|
if is_bundle or is_local or is_android:
|
|
import imp
|
|
imp.load_module('electrum', *imp.find_module('lib'))
|
|
imp.load_module('electrum_gui', *imp.find_module('gui'))
|
|
|
|
|
|
from electrum import util
|
|
from electrum import SimpleConfig, Network, Wallet, WalletStorage
|
|
from electrum.util import print_msg, print_error, print_stderr, json_encode, json_decode, set_verbosity, InvalidPassword
|
|
from electrum.plugins import Plugins, run_hook, always_hook
|
|
from electrum.commands import get_parser, known_commands, Commands, config_variables
|
|
|
|
|
|
# get password routine
|
|
def prompt_password(prompt, confirm=True):
|
|
import getpass
|
|
password = getpass.getpass(prompt, stream=None)
|
|
if password and confirm:
|
|
password2 = getpass.getpass("Confirm: ")
|
|
if password != password2:
|
|
sys.exit("Error: Passwords do not match.")
|
|
if not password:
|
|
password = None
|
|
return password
|
|
|
|
|
|
|
|
def init_gui(config, network, plugins):
|
|
gui_name = config.get('gui', 'qt')
|
|
if gui_name in ['lite', 'classic']:
|
|
gui_name = 'qt'
|
|
gui = __import__('electrum_gui.' + gui_name, fromlist=['electrum_gui'])
|
|
gui = gui.ElectrumGui(config, network, plugins)
|
|
return gui
|
|
|
|
|
|
|
|
def init_cmdline(config):
|
|
|
|
cmdname = config.get('cmd')
|
|
cmd = known_commands[cmdname]
|
|
|
|
if cmdname == 'signtransaction' and config.get('privkey'):
|
|
cmd.requires_wallet = False
|
|
cmd.requires_password = False
|
|
|
|
if cmdname in ['payto', 'paytomany'] and config.get('unsigned'):
|
|
cmd.requires_password = False
|
|
|
|
if cmdname in ['payto', 'paytomany'] and config.get('broadcast'):
|
|
cmd.requires_network = True
|
|
|
|
if cmdname in ['createrawtx'] and config.get('unsigned'):
|
|
cmd.requires_password = False
|
|
cmd.requires_wallet = False
|
|
|
|
if cmdname == 'listrequests' and config.get('status'):
|
|
cmd.requires_network = True
|
|
|
|
# instanciate wallet for command-line
|
|
storage = WalletStorage(config.get_wallet_path())
|
|
|
|
if cmd.name in ['create', 'restore']:
|
|
if storage.file_exists:
|
|
sys.exit("Error: Remove the existing wallet first!")
|
|
|
|
def password_dialog():
|
|
return prompt_password("Password (hit return if you do not wish to encrypt your wallet):")
|
|
|
|
if cmd.name == 'restore':
|
|
text = config.get('text')
|
|
password = password_dialog() if Wallet.is_seed(text) or Wallet.is_xprv(text) or Wallet.is_private_key(text) else None
|
|
try:
|
|
wallet = Wallet.from_text(text, password, storage)
|
|
except BaseException as e:
|
|
sys.exit(str(e))
|
|
if not config.get('offline'):
|
|
network = Network(config)
|
|
network.start()
|
|
wallet.start_threads(network)
|
|
print_msg("Recovering wallet...")
|
|
wallet.synchronize()
|
|
wallet.wait_until_synchronized()
|
|
msg = "Recovery successful" if wallet.is_found() else "Found no history for this wallet"
|
|
else:
|
|
msg = "This wallet was restored offline. It may contain more addresses than displayed."
|
|
print_msg(msg)
|
|
|
|
else:
|
|
password = password_dialog()
|
|
wallet = Wallet(storage)
|
|
seed = wallet.make_seed()
|
|
wallet.add_seed(seed, password)
|
|
wallet.create_master_keys(password)
|
|
wallet.create_main_account(password)
|
|
wallet.synchronize()
|
|
print_msg("Your wallet generation seed is:\n\"%s\"" % seed)
|
|
print_msg("Please keep it in a safe place; if you lose it, you will not be able to restore your wallet.")
|
|
|
|
print_msg("Wallet saved in '%s'" % wallet.storage.path)
|
|
sys.exit(0)
|
|
|
|
if cmd.requires_wallet and not storage.file_exists:
|
|
print_msg("Error: Wallet file not found.")
|
|
print_msg("Type 'electrum create' to create a new wallet, or provide a path to a wallet with the -w option")
|
|
sys.exit(0)
|
|
|
|
# create wallet instance
|
|
wallet = Wallet(storage) if cmd.requires_wallet else None
|
|
|
|
# notify plugins
|
|
always_hook('cmdline_load_wallet', wallet)
|
|
|
|
# important warning
|
|
if cmd.name in ['getprivatekeys']:
|
|
print_stderr("WARNING: ALL your private keys are secret.")
|
|
print_stderr("Exposing a single private key can compromise your entire wallet!")
|
|
print_stderr("In particular, DO NOT use 'redeem private key' services proposed by third parties.")
|
|
|
|
# commands needing password
|
|
if cmd.requires_password and wallet.use_encryption:
|
|
if config.get('password'):
|
|
password = config.get('password')
|
|
else:
|
|
password = prompt_password('Password:', False)
|
|
if not password:
|
|
print_msg("Error: Password required")
|
|
sys.exit(1)
|
|
# check password
|
|
try:
|
|
seed = wallet.check_password(password)
|
|
except InvalidPassword:
|
|
print_msg("Error: This password does not decode this wallet.")
|
|
sys.exit(1)
|
|
else:
|
|
password = None
|
|
|
|
# run the command
|
|
if cmd.name == 'deseed':
|
|
if not wallet.seed:
|
|
print_msg("Error: This wallet has no seed")
|
|
else:
|
|
ns = wallet.storage.path + '.seedless'
|
|
print_msg("Warning: you are going to create a seedless wallet'\nIt will be saved in '%s'" % ns)
|
|
if raw_input("Are you sure you want to continue? (y/n) ") in ['y', 'Y', 'yes']:
|
|
wallet.storage.path = ns
|
|
wallet.seed = ''
|
|
wallet.storage.put('seed', '', True)
|
|
wallet.use_encryption = False
|
|
wallet.storage.put('use_encryption', wallet.use_encryption, True)
|
|
for k in wallet.imported_keys.keys():
|
|
wallet.imported_keys[k] = ''
|
|
wallet.storage.put('imported_keys', wallet.imported_keys, True)
|
|
print_msg("Done.")
|
|
else:
|
|
print_msg("Action canceled.")
|
|
sys.exit(0)
|
|
|
|
elif cmd.name == 'password':
|
|
new_password = prompt_password('New password:')
|
|
wallet.update_password(password, new_password)
|
|
sys.exit(0)
|
|
|
|
return cmd, password
|
|
|
|
|
|
def run_command(config, network, password):
|
|
cmdname = config.get('cmd')
|
|
cmd = known_commands[cmdname]
|
|
# instanciate wallet for command-line
|
|
storage = WalletStorage(config.get_wallet_path())
|
|
# create wallet instance
|
|
wallet = Wallet(storage) if cmd.requires_wallet else None
|
|
# start threads
|
|
if wallet and network:
|
|
wallet.start_threads(network)
|
|
wallet.wait_until_synchronized()
|
|
# arguments passed to function
|
|
args = map(lambda x: config.get(x), cmd.params)
|
|
# decode json arguments
|
|
args = map(json_decode, args)
|
|
# options
|
|
args += map(lambda x: config.get(x), cmd.options)
|
|
cmd_runner = Commands(config, wallet, network)
|
|
cmd_runner.password = password
|
|
func = getattr(cmd_runner, cmd.name)
|
|
result = func(*args)
|
|
# stop threads
|
|
if wallet:
|
|
wallet.stop_threads()
|
|
return result
|
|
|
|
|
|
class ClientThread(util.DaemonThread):
|
|
|
|
def __init__(self, server, s):
|
|
util.DaemonThread.__init__(self)
|
|
self.server = server
|
|
self.client_pipe = util.SocketPipe(s)
|
|
self.response_queue = Queue.Queue()
|
|
self.server.add_client(self)
|
|
self.subscriptions = defaultdict(list)
|
|
self.network = self.server.network
|
|
|
|
def run(self):
|
|
config_options = self.client_pipe.get()
|
|
password = config_options.get('password')
|
|
config = SimpleConfig(config_options)
|
|
cmd = config.get('cmd')
|
|
if cmd == 'gui':
|
|
if self.server.gui:
|
|
if hasattr(server.gui, 'new_window'):
|
|
path = config.get_wallet_path()
|
|
self.server.gui.new_window(path, config.get('url'))
|
|
response = "ok"
|
|
else:
|
|
response = "error: current GUI does not support multiple windows"
|
|
else:
|
|
response = "Error: Electrum is running in daemon mode. Please stop the daemon first."
|
|
elif cmd == 'daemon':
|
|
sub = config.get('subcommand')
|
|
assert sub in ['start', 'stop', 'status']
|
|
if sub == 'start':
|
|
response = "Daemon already running"
|
|
elif sub == 'status':
|
|
p = self.network.get_parameters()
|
|
response = {
|
|
'path': self.network.config.path,
|
|
'server': p[0],
|
|
'blockchain_height': self.network.get_local_height(),
|
|
'server_height': self.network.get_server_height(),
|
|
'nodes': self.network.get_interfaces(),
|
|
'connected': self.network.is_connected(),
|
|
'auto_connect': p[4],
|
|
}
|
|
elif sub == 'stop':
|
|
self.server.stop()
|
|
response = "Daemon stopped"
|
|
else:
|
|
try:
|
|
response = run_command(config, self.network, password)
|
|
except BaseException as e:
|
|
err = traceback.format_exc()
|
|
response = {'error':err}
|
|
# send response and exit
|
|
self.client_pipe.send(response)
|
|
self.server.remove_client(self)
|
|
|
|
|
|
|
|
|
|
class NetworkServer(util.DaemonThread):
|
|
|
|
def __init__(self, config, network):
|
|
util.DaemonThread.__init__(self)
|
|
self.debug = False
|
|
self.config = config
|
|
self.pipe = util.QueuePipe()
|
|
self.network = network
|
|
self.lock = threading.RLock()
|
|
# each GUI is a client of the daemon
|
|
self.clients = []
|
|
# gui is None is we run as daemon
|
|
self.gui = None
|
|
|
|
def add_client(self, client):
|
|
for key in ['fee', 'status', 'banner', 'updated', 'servers', 'interfaces']:
|
|
value = self.network.get_status_value(key)
|
|
client.response_queue.put({'method':'network.status', 'params':[key, value]})
|
|
with self.lock:
|
|
self.clients.append(client)
|
|
print_error("new client:", len(self.clients))
|
|
|
|
def remove_client(self, client):
|
|
with self.lock:
|
|
self.clients.remove(client)
|
|
print_error("client quit:", len(self.clients))
|
|
|
|
def run(self):
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
|
daemon_socket = os.path.join(self.config.path, DAEMON_SOCKET)
|
|
if os.path.exists(daemon_socket):
|
|
os.remove(daemon_socket)
|
|
daemon_timeout = self.config.get('daemon_timeout', None)
|
|
s.bind(daemon_socket)
|
|
s.listen(5)
|
|
s.settimeout(0.1)
|
|
while self.is_running():
|
|
try:
|
|
connection, address = s.accept()
|
|
except socket.timeout:
|
|
continue
|
|
client = ClientThread(self, connection)
|
|
client.start()
|
|
print_error("Daemon exiting")
|
|
|
|
|
|
|
|
|
|
def get_daemon(config, start_daemon):
|
|
daemon_socket = os.path.join(config.path, DAEMON_SOCKET)
|
|
daemon_started = False
|
|
while True:
|
|
try:
|
|
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
|
|
s.connect(daemon_socket)
|
|
return s
|
|
except socket.error:
|
|
if not start_daemon:
|
|
return False
|
|
elif not daemon_started:
|
|
daemon_started = True
|
|
else:
|
|
time.sleep(0.1)
|
|
except:
|
|
# do not use daemon if AF_UNIX is not available (windows)
|
|
return False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
# make sure that certificates are here
|
|
assert os.path.exists(requests.utils.DEFAULT_CA_BUNDLE_PATH)
|
|
|
|
# on osx, delete Process Serial Number arg generated for apps launched in Finder
|
|
sys.argv = filter(lambda x: not x.startswith('-psn'), sys.argv)
|
|
|
|
# old 'help' syntax
|
|
if len(sys.argv)>1 and sys.argv[1] == 'help':
|
|
sys.argv.remove('help')
|
|
sys.argv.append('-h')
|
|
|
|
# read arguments from stdin pipe and prompt
|
|
for i, arg in enumerate(sys.argv):
|
|
if arg == '-':
|
|
if not sys.stdin.isatty():
|
|
sys.argv[i] = sys.stdin.read()
|
|
break
|
|
else:
|
|
raise BaseException('Cannot get argument from stdin')
|
|
elif arg == '?':
|
|
sys.argv[i] = raw_input("Enter argument:")
|
|
elif arg == ':':
|
|
sys.argv[i] = prompt_password('Enter argument (will not echo):', False)
|
|
|
|
# parse command line
|
|
parser = get_parser()
|
|
args = parser.parse_args()
|
|
|
|
# config is an object passed to the various constructors (wallet, interface, gui)
|
|
if is_android:
|
|
config_options = {
|
|
'verbose': True,
|
|
'cmd': 'gui',
|
|
'gui': 'kivy' if is_kivy else 'android',
|
|
#'auto_connect': True,
|
|
}
|
|
else:
|
|
config_options = args.__dict__
|
|
for k, v in config_options.items():
|
|
if v is None or (k in config_variables.get(args.cmd, {}).keys()):
|
|
config_options.pop(k)
|
|
if config_options.get('server'):
|
|
config_options['auto_connect'] = False
|
|
|
|
if config_options.get('portable'):
|
|
config_options['electrum_path'] = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'electrum_data')
|
|
|
|
set_verbosity(config_options.get('verbose'))
|
|
|
|
# check uri
|
|
uri = config_options.get('url')
|
|
if uri:
|
|
if not re.match('^bitcoin:', uri):
|
|
print_stderr('unknown command:', uri)
|
|
sys.exit(1)
|
|
config_options['url'] = uri
|
|
|
|
config = SimpleConfig(config_options)
|
|
cmd_name = config.get('cmd')
|
|
|
|
# initialize plugins.
|
|
gui_name = config.get('gui', 'qt') if cmd_name == 'gui' else 'cmdline'
|
|
plugins = Plugins(config, is_bundle or is_local or is_android, gui_name)
|
|
|
|
# run command offline
|
|
if cmd_name not in ['gui', 'daemon']:
|
|
cmd, password = init_cmdline(config)
|
|
if not cmd.requires_network or config.get('offline'):
|
|
result = run_command(config, None, password)
|
|
print_msg(json_encode(result))
|
|
sys.exit(0)
|
|
else:
|
|
config_options['password'] = password
|
|
|
|
# check if daemon is running
|
|
s = get_daemon(config, False)
|
|
if s:
|
|
p = util.SocketPipe(s)
|
|
p.set_timeout(1000000)
|
|
p.send(config_options)
|
|
result = p.get()
|
|
s.close()
|
|
if type(result) in [str, unicode]:
|
|
print_msg(result)
|
|
elif type(result) is dict and result.get('error'):
|
|
print_stderr(result.get('error'))
|
|
elif result is not None:
|
|
print_msg(json_encode(result))
|
|
sys.exit(0)
|
|
|
|
# daemon is not running
|
|
if cmd_name == 'gui':
|
|
if not config.get('offline'):
|
|
network = Network(config, plugins)
|
|
network.start()
|
|
else:
|
|
network = None
|
|
server = NetworkServer(config, network)
|
|
server.start()
|
|
server.gui = init_gui(config, network, plugins)
|
|
server.gui.main()
|
|
elif cmd_name == 'daemon':
|
|
subcommand = config.get('subcommand')
|
|
if subcommand in ['status', 'stop']:
|
|
print_msg("Daemon not running")
|
|
sys.exit(1)
|
|
elif subcommand == 'start':
|
|
p = os.fork()
|
|
if p == 0:
|
|
network = Network(config, plugins)
|
|
network.start()
|
|
server = NetworkServer(config, network)
|
|
if config.get('websocket_server'):
|
|
from electrum import websockets
|
|
websockets.WebSocketServer(config, server).start()
|
|
if config.get('requests_dir'):
|
|
util.check_www_dir(config.get('requests_dir'))
|
|
server.start()
|
|
server.join()
|
|
else:
|
|
print_stderr("starting daemon (PID %d)"%p)
|
|
sys.exit(0)
|
|
else:
|
|
print_msg("syntax: electrum daemon <start|status|stop>")
|
|
sys.exit(1)
|
|
else:
|
|
print_msg("Network daemon is not running. Try 'electrum daemon start'\nIf you want to run this command offline, use the -o flag.")
|
|
sys.exit(1)
|