Browse Source
* gui and daemon are in the same process * commands that require network are sent to the daemon * open only one gui window per wallet283
ThomasV
9 years ago
11 changed files with 299 additions and 468 deletions
@ -1,254 +0,0 @@ |
|||
#!/usr/bin/env python |
|||
# |
|||
# Electrum - lightweight Bitcoin client |
|||
# Copyright (C) 2014 Thomas Voegtlin |
|||
# |
|||
# 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 socket |
|||
import time |
|||
import sys |
|||
import os |
|||
import threading |
|||
import traceback |
|||
import json |
|||
import Queue |
|||
from collections import defaultdict |
|||
|
|||
import util |
|||
from network import Network |
|||
from util import print_error, print_stderr, parse_json |
|||
from simple_config import SimpleConfig |
|||
|
|||
DAEMON_SOCKET = 'daemon.sock' |
|||
|
|||
|
|||
def do_start_daemon(config): |
|||
import subprocess |
|||
args = [sys.executable, __file__, config.path] |
|||
logfile = open(os.path.join(config.path, 'daemon.log'),'w') |
|||
p = subprocess.Popen(args, stderr=logfile, stdout=logfile, close_fds=(os.name=="posix")) |
|||
print_stderr("starting daemon (PID %d)"%p.pid) |
|||
|
|||
|
|||
|
|||
def get_daemon(config, start_daemon): |
|||
import socket |
|||
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: |
|||
do_start_daemon(config) |
|||
daemon_started = True |
|||
else: |
|||
time.sleep(0.1) |
|||
except: |
|||
# do not use daemon if AF_UNIX is not available (windows) |
|||
return False |
|||
|
|||
|
|||
|
|||
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) |
|||
|
|||
def reading_thread(self): |
|||
while self.is_running(): |
|||
try: |
|||
request = self.client_pipe.get() |
|||
except util.timeout: |
|||
continue |
|||
if request is None: |
|||
self.running = False |
|||
break |
|||
method = request.get('method') |
|||
params = request.get('params') |
|||
if method == 'daemon.stop': |
|||
self.server.stop() |
|||
continue |
|||
if method[-10:] == '.subscribe': |
|||
self.subscriptions[method].append(params) |
|||
self.server.send_request(self, request) |
|||
|
|||
def run(self): |
|||
threading.Thread(target=self.reading_thread).start() |
|||
while self.is_running(): |
|||
try: |
|||
response = self.response_queue.get(timeout=0.1) |
|||
except Queue.Empty: |
|||
continue |
|||
try: |
|||
self.client_pipe.send(response) |
|||
except socket.error: |
|||
self.running = False |
|||
break |
|||
self.server.remove_client(self) |
|||
|
|||
|
|||
|
|||
|
|||
|
|||
class NetworkServer(util.DaemonThread): |
|||
|
|||
def __init__(self, config): |
|||
util.DaemonThread.__init__(self) |
|||
self.debug = False |
|||
self.config = config |
|||
self.pipe = util.QueuePipe() |
|||
self.network = Network(self.pipe, config) |
|||
self.lock = threading.RLock() |
|||
# each GUI is a client of the daemon |
|||
self.clients = [] |
|||
self.request_id = 0 |
|||
self.requests = {} |
|||
|
|||
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 send_request(self, client, request): |
|||
with self.lock: |
|||
self.request_id += 1 |
|||
self.requests[self.request_id] = (request['id'], client) |
|||
request['id'] = self.request_id |
|||
if self.debug: |
|||
print_error("-->", request) |
|||
self.pipe.send(request) |
|||
|
|||
def run(self): |
|||
self.network.start() |
|||
while self.is_running(): |
|||
try: |
|||
response = self.pipe.get() |
|||
except util.timeout: |
|||
continue |
|||
if self.debug: |
|||
print_error("<--", response) |
|||
response_id = response.get('id') |
|||
if response_id: |
|||
with self.lock: |
|||
client_id, client = self.requests.pop(response_id) |
|||
response['id'] = client_id |
|||
client.response_queue.put(response) |
|||
else: |
|||
# notification |
|||
m = response.get('method') |
|||
v = response.get('params') |
|||
for client in self.clients: |
|||
if m == 'network.status' or v in client.subscriptions.get(m, []): |
|||
client.response_queue.put(response) |
|||
self.network.stop() |
|||
print_error("server exiting") |
|||
|
|||
|
|||
def daemon_loop(server): |
|||
s = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) |
|||
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) |
|||
daemon_socket = os.path.join(server.config.path, DAEMON_SOCKET) |
|||
if os.path.exists(daemon_socket): |
|||
os.remove(daemon_socket) |
|||
daemon_timeout = server.config.get('daemon_timeout', None) |
|||
s.bind(daemon_socket) |
|||
s.listen(5) |
|||
s.settimeout(1) |
|||
t = time.time() |
|||
while server.running: |
|||
try: |
|||
connection, address = s.accept() |
|||
except socket.timeout: |
|||
if daemon_timeout is None: |
|||
continue |
|||
if not server.clients: |
|||
if time.time() - t > daemon_timeout: |
|||
print_error("Daemon timeout") |
|||
break |
|||
else: |
|||
t = time.time() |
|||
continue |
|||
t = time.time() |
|||
client = ClientThread(server, connection) |
|||
client.start() |
|||
server.stop() |
|||
# sleep so that other threads can terminate cleanly |
|||
time.sleep(0.5) |
|||
print_error("Daemon exiting") |
|||
|
|||
|
|||
|
|||
def check_www_dir(rdir): |
|||
# rewrite index.html every time |
|||
import urllib, urlparse, shutil, os |
|||
if not os.path.exists(rdir): |
|||
os.mkdir(rdir) |
|||
index = os.path.join(rdir, 'index.html') |
|||
src = os.path.join(os.path.dirname(__file__), 'www', 'index.html') |
|||
shutil.copy(src, index) |
|||
files = [ |
|||
"https://code.jquery.com/jquery-1.9.1.min.js", |
|||
"https://raw.githubusercontent.com/davidshimjs/qrcodejs/master/qrcode.js", |
|||
"https://code.jquery.com/ui/1.10.3/jquery-ui.js", |
|||
"https://code.jquery.com/ui/1.10.3/themes/smoothness/jquery-ui.css" |
|||
] |
|||
for URL in files: |
|||
path = urlparse.urlsplit(URL).path |
|||
filename = os.path.basename(path) |
|||
path = os.path.join(rdir, filename) |
|||
if not os.path.exists(path): |
|||
print_error("downloading ", URL) |
|||
urllib.urlretrieve(URL, path) |
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
import simple_config, util |
|||
_config = {} |
|||
if len(sys.argv) > 1: |
|||
_config['electrum_path'] = sys.argv[1] |
|||
config = simple_config.SimpleConfig(_config) |
|||
util.set_verbosity(True) |
|||
server = NetworkServer(config) |
|||
server.start() |
|||
if config.get('websocket_server'): |
|||
import websockets |
|||
websockets.WebSocketServer(config, server).start() |
|||
if config.get('requests_dir'): |
|||
check_www_dir(config.get('requests_dir')) |
|||
|
|||
try: |
|||
daemon_loop(server) |
|||
except KeyboardInterrupt: |
|||
print "Ctrl C - Stopping daemon" |
|||
server.stop() |
|||
sys.exit(1) |
|||
Loading…
Reference in new issue