# along with this program. If not, see <http://www.gnu.org/licenses/>.
-import random, socket, ast, re
+import random, socket, ast, re, ssl
import threading, traceback, sys, time, json, Queue
-from version import ELECTRUM_VERSION
+from version import ELECTRUM_VERSION, PROTOCOL_VERSION
from util import print_error
DEFAULT_TIMEOUT = 5
DEFAULT_SERVERS = [
+ #'electrum.bitcoins.sk:50001:t',
+ #'uncle-enzo.info:50001:t',
+ #'electrum.bitfoo.org:50001:t',
+ #'webbtc.net:50001:t',
+ 'electrum.bitcoin.cz:50001:t',
'electrum.novit.ro:50001:t',
+ 'electrum.be:50001:t',
+ 'electrum.bysh.me:50001:t',
'electrum.pdmc.net:50001:t',
- 'ecdsa.org:50001:t',
- 'electrum.bitcoins.sk:50001:t',
- 'uncle-enzo.info:50001:t',
- 'electrum.bytesized-hosting.com:50001:t',
- 'electrum.bitcoin.cz:50001:t',
- 'electrum.bitfoo.org:50001:t'
+ 'electrum.no-ip.org:50001:t',
+ 'ecdsa.org:50001:t'
]
proxy_modes = ['socks4', 'socks5', 'http']
def register_callback(self, event, callback):
with self.lock:
- self.callbacks[event] = callback
+ if not self.callbacks.get(event):
+ self.callbacks[event] = []
+ self.callbacks[event].append(callback)
def trigger_callback(self, event):
with self.lock:
- callback = self.callbacks.get(event)
- if callback:
- callback()
-
-
+ callbacks = self.callbacks.get(event,[])[:]
+ if callbacks:
+ [callback() for callback in callbacks]
def init_server(self, host, port, proxy=None, use_ssl=True):
self.host = host
self.port = port
self.proxy = proxy
self.use_ssl = use_ssl
-
- self.servers = [] # actual list from IRC
- self.rtime = 0
- self.bytes_received = 0
-
- self.is_connected = True
self.poll_interval = 1
#json
def queue_json_response(self, c):
# uncomment to debug
- # print "<--",c
+ # print_error( "<--",c )
msg_id = c.get('id')
error = c.get('error')
if error:
- print "received error:", c
+ print_error("received error:", c)
+ if msg_id is not None:
+ with self.lock:
+ method, params, channel = self.unanswered_requests.pop(msg_id)
+ response_queue = self.responses[channel]
+ response_queue.put({'method':method, 'params':params, 'error':error, 'id':msg_id})
+
return
if msg_id is not None:
with self.lock:
method, params, channel = self.unanswered_requests.pop(msg_id)
result = c.get('result')
+
+ if method == 'server.version':
+ self.server_version = result
+
+ elif method == 'server.peers.subscribe':
+ servers = {}
+ for item in result:
+ s = []
+ host = item[1]
+ ports = []
+ version = None
+ pruning = False
+ if len(item) > 2:
+ for v in item[2]:
+ if re.match("[stgh]\d+", v):
+ ports.append((v[0], v[1:]))
+ elif re.match("v(.?)+", v):
+ version = v[1:]
+ elif v == 'p':
+ pruning = True
+ try:
+ is_recent = float(version)>=float(PROTOCOL_VERSION)
+ except:
+ is_recent = False
+ if ports and is_recent:
+ servers[host] = {'ports':ports, 'pruning':pruning}
+ self.servers = servers
+ self.trigger_callback('peers')
+
else:
- # notification. we should find the channel(s)..
+ # notification: find the channel(s)
method = c.get('method')
params = c.get('params')
- with self.lock:
- for k,v in self.subscriptions.items():
- if (method, params) in v:
- channel = k
- else:
- raise
if method == 'blockchain.numblocks.subscribe':
result = params[0]
params = []
+ elif method == 'blockchain.headers.subscribe':
+ result = params[0]
+ params = []
+
elif method == 'blockchain.address.subscribe':
addr = params[0]
result = params[1]
params = [addr]
+ with self.lock:
+ for k,v in self.subscriptions.items():
+ if (method, params) in v:
+ channel = k
+ break
+ else:
+ print_error( "received unexpected notification", method, params)
+ print_error( self.subscriptions )
+ return
response_queue = self.responses[channel]
response_queue.put({'method':method, 'params':params, 'result':result, 'id':msg_id})
self.init_server(host, port, proxy, use_ssl)
self.session_id = None
self.connection_msg = ('https' if self.use_ssl else 'http') + '://%s:%d'%( self.host, self.port )
-
+ self.is_connected = True
def run_http(self):
self.is_connected = True
if self.proxy:
import socks
- socks.setdefaultproxy(proxy_modes.index(self.proxy["mode"]), self.proxy["host"], int(self.proxy["port"]) )
+ socks.setdefaultproxy(proxy_modes.index(self.proxy["mode"]) + 1, self.proxy["host"], int(self.proxy["port"]) )
socks.wrapmodule(urllib2)
cj = cookielib.CookieJar()
def init_tcp(self, host, port, proxy=None, use_ssl=True):
self.init_server(host, port, proxy, use_ssl)
- import ssl
global proxy_modes
self.connection_msg = "%s:%d"%(self.host,self.port)
if self.proxy is None:
self.connection_msg += " using proxy %s:%s:%s"%(self.proxy.get('mode'), self.proxy.get('host'), self.proxy.get('port'))
import socks
s = socks.socksocket()
- s.setproxy(proxy_modes.index(self.proxy["mode"]), self.proxy["host"], int(self.proxy["port"]) )
+ s.setproxy(proxy_modes.index(self.proxy["mode"]) + 1, self.proxy["host"], int(self.proxy["port"]) )
if self.use_ssl:
s = ssl.wrap_socket(s, ssl_version=ssl.PROTOCOL_SSLv23)
try:
out = ''
while self.is_connected:
- try: msg = self.s.recv(1024)
+
+ try:
+ timeout = False
+ msg = self.s.recv(1024)
except socket.timeout:
- print "timeout"
+ timeout = True
+ except ssl.SSLError:
+ timeout = True
+
+ if timeout:
# ping the server with server.version, as a real ping does not exist yet
- self.send([('server.version', [ELECTRUM_VERSION])])
+ self.send([('server.version', [ELECTRUM_VERSION, PROTOCOL_VERSION])])
continue
+
out += msg
self.bytes_received += len(msg)
if msg == '':
self.is_connected = False
- print "Disconnected."
while True:
s = out.find('\n')
self.message_id += 1
out += request + '\n'
while out:
- sent = self.s.send( out )
- out = out[sent:]
+ try:
+ sent = self.s.send( out )
+ out = out[sent:]
+ except:
+ # this happens when we get disconnected
+ print_error( "Not connected, cannot send" )
+ return None
return ids
def __init__(self, config=None, loop=False):
+ self.server = None
+ self.proxy = None
if config is None:
from simple_config import SimpleConfig
self.daemon = True
self.loop = loop
self.config = config
+ self.connect_event = threading.Event()
self.subscriptions = {}
self.responses = {}
self.callbacks = {}
self.lock = threading.Lock()
- self.init_interface()
+
+ self.servers = {} # actual list from IRC
+ self.rtime = 0
+ self.bytes_received = 0
self.init_with_server(self.config)
else:
print "Using random server..."
- servers = DEFAULT_SERVERS
+ servers = DEFAULT_SERVERS[:]
while servers:
server = random.choice( servers )
servers.remove(server)
if self.is_connected: break
if not servers:
- raise BaseException('no server available')
-
+ print 'no server available'
+ self.is_connected = False
+ self.connect_event.set() # to finish start
+ self.server = 'ecdsa.org:50001:t'
+ self.proxy = None
+ return
+
+ self.connect_event.set()
if self.is_connected:
- self.send([('server.version', [ELECTRUM_VERSION])])
+ self.send([('server.version', [ELECTRUM_VERSION, PROTOCOL_VERSION])])
self.trigger_callback('connected')
else:
self.trigger_callback('notconnected')
with self.lock:
if self.subscriptions.get(channel) is None:
self.subscriptions[channel] = []
- self.subscriptions[channel] += sub
+ for message in sub:
+ if message not in self.subscriptions[channel]:
+ self.subscriptions[channel].append(message)
if self.protocol in 'st':
with self.lock:
print "changing server:", server, proxy
self.server = server
self.proxy = proxy
- if self.protocol in 'st':
+ if self.protocol in 'st' and self.s:
self.s.shutdown(socket.SHUT_RDWR)
self.s.close()
self.is_connected = False # this exits the polling loop
+ self.trigger_callback('disconnecting') # for actively disconnecting
+
+ def stop(self):
+ if self.protocol in 'st' and self.s:
+ self.s.shutdown(socket.SHUT_RDWR)
+ self.s.close()
+
+
+ def get_servers_list(self):
+ plist = {}
+ if not self.servers:
+ servers_list = {}
+ for x in DEFAULT_SERVERS:
+ h,port,protocol = x.split(':')
+ servers_list[h] = {'ports':[(protocol,port)]}
+ else:
+ servers_list = self.servers
+
+ for _host, v in servers_list.items():
+ pp = v['ports']
+ z = {}
+ for item2 in pp:
+ _protocol, _port = item2
+ z[_protocol] = _port
+ plist[_host] = z
+
+ return plist, servers_list
def is_empty(self, channel):
return out
+ def start(self):
+ threading.Thread.start(self)
+ # wait until connection is established
+ self.connect_event.wait()
def run(self):
while True:
- self.run_tcp() if self.protocol in 'st' else self.run_http()
+ self.init_interface()
+ if self.is_connected:
+ self.resend_subscriptions()
+ self.run_tcp() if self.protocol in 'st' else self.run_http()
+
self.trigger_callback('disconnected')
- if not self.loop: break
+ if not self.loop: break
time.sleep(5)
- self.init_interface()
- self.resend_subscriptions()