128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
#!/usr/bin/env 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/>.
|
|
|
|
import time, thread, sys, socket
|
|
|
|
# see http://code.google.com/p/jsonrpclib/
|
|
import jsonrpclib
|
|
from wallet import Wallet
|
|
|
|
"""
|
|
Simple wallet daemon for webservers.
|
|
- generates new addresses on request
|
|
- private keys are not needed in order to generate new addresses. A neutralized wallet can be used (seed removed)
|
|
- no gap limit: use 'getnum' to know how many addresses have been created.
|
|
|
|
todo:
|
|
- return the max gap
|
|
- add expiration date
|
|
|
|
"""
|
|
|
|
|
|
host = 'ecdsa.org'
|
|
port = 8444
|
|
wallet_path = 'wallet_path'
|
|
username = 'foo'
|
|
password = 'bar'
|
|
wallet = Wallet()
|
|
stopping = False
|
|
|
|
|
|
|
|
from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCRequestHandler
|
|
import SimpleXMLRPCServer
|
|
|
|
class authHandler(SimpleJSONRPCRequestHandler):
|
|
def parse_request(self):
|
|
if SimpleXMLRPCServer.SimpleXMLRPCRequestHandler.parse_request(self):
|
|
if self.authenticate(self.headers):
|
|
return True
|
|
else:
|
|
self.send_error(401, 'Authentication failed')
|
|
return False
|
|
|
|
def authenticate(self, headers):
|
|
from base64 import b64decode
|
|
basic, _, encoded = headers.get('Authorization').partition(' ')
|
|
assert basic == 'Basic', 'Only basic authentication supported'
|
|
x_username, _, x_password = b64decode(encoded).partition(':')
|
|
return username == x_username and password == x_password
|
|
|
|
|
|
def do_stop():
|
|
global stopping
|
|
stopping = True
|
|
|
|
def get_new_address():
|
|
a = wallet.create_new_address(False)
|
|
wallet.save()
|
|
return a
|
|
|
|
def get_num():
|
|
return len(wallet.addresses)
|
|
|
|
def get_mpk():
|
|
return wallet.master_public_key.encode('hex')
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if len(sys.argv)>1:
|
|
import jsonrpclib
|
|
server = jsonrpclib.Server('http://%s:%s@%s:%d'%(username, password, host, port))
|
|
cmd = sys.argv[1]
|
|
|
|
try:
|
|
if cmd == 'getnum':
|
|
out = server.getnum()
|
|
elif cmd == 'getkey':
|
|
out = server.getkey()
|
|
elif cmd == 'getnewaddress':
|
|
out = server.getnewaddress()
|
|
elif cmd == 'stop':
|
|
out = server.stop()
|
|
except socket.error:
|
|
print "server not running"
|
|
sys.exit(1)
|
|
print out
|
|
sys.exit(0)
|
|
|
|
else:
|
|
|
|
wallet.set_path(wallet_path)
|
|
wallet.read()
|
|
|
|
def server_thread():
|
|
from SocketServer import ThreadingMixIn
|
|
from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCServer
|
|
server = SimpleJSONRPCServer(( host, port), requestHandler=authHandler)
|
|
server.register_function(get_new_address, 'getnewaddress')
|
|
server.register_function(get_num, 'getnum')
|
|
server.register_function(get_mpk, 'getkey')
|
|
server.register_function(do_stop, 'stop')
|
|
server.serve_forever()
|
|
|
|
thread.start_new_thread(server_thread, ())
|
|
while not stopping: time.sleep(0.1)
|
|
|
|
|
|
|
|
|