revert pruning flag for default servers (too much complexity)
[electrum-nvc.git] / lib / interface.py
index f4e434d..d7235a0 100644 (file)
 # 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']
@@ -64,12 +67,6 @@ class Interface(threading.Thread):
         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
@@ -81,40 +78,81 @@ class Interface(threading.Thread):
     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:
-                    print "received expected notification", method, params
-                    return
 
             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})
@@ -136,7 +174,7 @@ class Interface(threading.Thread):
         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
@@ -165,7 +203,7 @@ class Interface(threading.Thread):
         
         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()
@@ -226,7 +264,6 @@ class Interface(threading.Thread):
     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:
@@ -235,7 +272,7 @@ class Interface(threading.Thread):
             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)
@@ -257,12 +294,20 @@ class Interface(threading.Thread):
         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 == '': 
@@ -301,13 +346,15 @@ class Interface(threading.Thread):
                 out = out[sent:]
             except:
                 # this happens when we get disconnected
-                print "Not connected, cannot send"
+                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
@@ -326,6 +373,10 @@ class Interface(threading.Thread):
         self.callbacks = {}
         self.lock = threading.Lock()
 
+        self.servers = {} # actual list from IRC
+        self.rtime = 0
+        self.bytes_received = 0
+
 
 
     def init_interface(self):
@@ -333,7 +384,7 @@ class Interface(threading.Thread):
             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)
@@ -342,11 +393,16 @@ class Interface(threading.Thread):
                 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')
@@ -384,7 +440,9 @@ class Interface(threading.Thread):
             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:
@@ -432,12 +490,39 @@ class Interface(threading.Thread):
             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):
         q = self.responses.get(channel)
         if q: 
@@ -483,12 +568,13 @@ class Interface(threading.Thread):
     def run(self):
         while True:
             self.init_interface()
-            self.resend_subscriptions()
+            if self.is_connected:
+                self.resend_subscriptions()
+                self.run_tcp() if self.protocol in 'st' else self.run_http()
 
-            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)