show 'server lagging' icon if server lags and client does not use auto-connect
[electrum-nvc.git] / lib / blockchain.py
index c48c7fb..ee6d99e 100644 (file)
@@ -22,114 +22,113 @@ from util import user_dir, appdata_dir, print_error
 from bitcoin import *
 
 
-class BlockchainVerifier(threading.Thread):
-    """ Simple Payment Verification """
+class Blockchain(threading.Thread):
 
-    def __init__(self, interface, config):
+    def __init__(self, config, network):
         threading.Thread.__init__(self)
         self.daemon = True
         self.config = config
-        self.interface = interface
-        self.interface.register_channel('verifier')
+        self.network = network
         self.lock = threading.Lock()
-        self.pending_headers = [] # headers that have not been verified
         self.height = 0
         self.local_height = 0
         self.running = False
         self.headers_url = 'http://headers.electrum.org/blockchain_headers'
+        self.set_local_height()
+        self.queue = Queue.Queue()
+        self.servers_height = {}
+        self.is_lagging = False
 
-
+    
     def stop(self):
         with self.lock: self.running = False
-        self.interface.poke('verifier')
+
 
     def is_running(self):
         with self.lock: return self.running
 
-    def run(self):
 
+    def run(self):
         self.init_headers_file()
         self.set_local_height()
+        print_error( "blocks:", self.local_height )
 
         with self.lock:
             self.running = True
-        requested_chunks = []
-        requested_headers = []
-        all_chunks = False
-        
-        # subscribe to block headers
-        self.interface.send([ ('blockchain.headers.subscribe',[])], 'verifier')
 
         while self.is_running():
-            # request missing chunks
-            if not all_chunks and self.height and not requested_chunks:
-
-                if self.local_height + 50 < self.height:
-                    min_index = (self.local_height + 1)/2016
-                    max_index = (self.height + 1)/2016
-                    for i in range(min_index, max_index + 1):
-                        print_error( "requesting chunk", i )
-                        self.interface.send([ ('blockchain.block.get_chunk',[i])], 'verifier')
-                        requested_chunks.append(i)
-                        break
-                else:
-                    all_chunks = True
-                    print_error("downloaded all chunks")
-
-
-            # process pending headers
-            if self.pending_headers and all_chunks:
-                done = []
-                for header in self.pending_headers:
-                    if self.verify_header(header):
-                        done.append(header)
-                    else:
-                        # request previous header
-                        i = header.get('block_height') - 1
-                        if i not in requested_headers:
-                            print_error("requesting header %d"%i)
-                            self.interface.send([ ('blockchain.block.get_header',[i])], 'verifier')
-                            requested_headers.append(i)
-                        # no point continuing
-                        break
-                if done:
-                    self.interface.trigger_callback('updated')
-                    for header in done: 
-                        self.pending_headers.remove(header)
 
             try:
-                r = self.interface.get_response('verifier',timeout=1)
+                result = self.queue.get()
             except Queue.Empty:
                 continue
-            if not r: continue
 
-            if r.get('error'):
-                print_error('Verifier received an error:', r)
-                continue
+            if not result: continue
 
-            # 3. handle response
-            method = r['method']
-            params = r['params']
-            result = r['result']
+            i, header = result
+            if not header: continue
+            
+            height = header.get('block_height')
+            self.servers_height[i.server] = height
 
-            if method == 'blockchain.block.get_chunk':
-                index = params[0]
-                self.verify_chunk(index, result)
-                requested_chunks.remove(index)
+            if height > self.local_height + 50:
+                self.get_chunks(i, header, height)
+                self.network.trigger_callback('updated')
 
-            elif method in ['blockchain.headers.subscribe', 'blockchain.block.get_header']:
+            if height > self.local_height:
+                # get missing parts from interface (until it connects to my chain)
+                chain = self.get_chain( i, header )
 
-                self.pending_headers.append(result)
-                if method == 'blockchain.block.get_header':
-                    requested_headers.remove(result.get('block_height'))
+                # skip that server if the result is not consistent
+                if not chain: continue
+                
+                # verify the chain
+                if self.verify_chain( chain ):
+                    print_error("height:", height, i.server)
+                    for header in chain:
+                        self.save_header(header)
+                        self.height = height
+                else:
+                    print_error("error", i.server)
+                    # todo: dismiss that server
+
+            h = self.servers_height.get(self.network.interface.server)
+            if h is not None and h < height - 1:
+                print_error( "Server is lagging", height, h)
+                if self.config.get('auto_cycle'):
+                    self.network.set_server(i.server)
                 else:
-                    self.height = result.get('block_height')
-                    ## fixme # self.interface.poke('synchronizer')
+                    self.network.is_lagging = True
+            else:
+                self.network.is_lagging = False
                 
-                self.pending_headers.sort(key=lambda x: x.get('block_height'))
-                # print "pending headers", map(lambda x: x.get('block_height'), self.pending_headers)
+            self.network.trigger_callback('updated')
+
+
+                    
+            
+    def verify_chain(self, chain):
+
+        first_header = chain[0]
+        prev_header = self.read_header(first_header.get('block_height') -1)
+        
+        for header in chain:
 
+            height = header.get('block_height')
 
+            prev_hash = self.hash_header(prev_header)
+            bits, target = self.get_target(height/2016)
+            _hash = self.hash_header(header)
+            try:
+                assert prev_hash == header.get('prev_block_hash')
+                assert bits == header.get('bits')
+                assert eval('0x'+_hash) < target
+            except:
+                return False
+
+            prev_header = header
+
+        return True
 
 
 
@@ -137,7 +136,6 @@ class BlockchainVerifier(threading.Thread):
         data = hexdata.decode('hex')
         height = index*2016
         num = len(data)/80
-        print_error("validating headers %d"%height)
 
         if index == 0:  
             previous_hash = ("0"*64)
@@ -161,6 +159,7 @@ class BlockchainVerifier(threading.Thread):
             previous_hash = _hash 
 
         self.save_chunk(index, data)
+        print_error("validated chunk %d"%height)
 
 
     def verify_header(self, header):
@@ -184,17 +183,8 @@ class BlockchainVerifier(threading.Thread):
         except:
             # this can be caused by a reorg.
             print_error("verify header failed"+ repr(header))
-            # undo verifications
-            with self.lock:
-                items = self.verified_tx.items()[:]
-            for tx_hash, item in items:
-                tx_height, timestamp, pos = item
-                if tx_height >= height:
-                    print_error("redoing", tx_hash)
-                    with self.lock:
-                        self.verified_tx.pop(tx_hash)
-                        if tx_hash in self.merkle_roots:
-                            self.merkle_roots.pop(tx_hash)
+            verifier.undo_verifications()
+
             # return False to request previous header.
             return False
 
@@ -203,8 +193,6 @@ class BlockchainVerifier(threading.Thread):
         return True
         
 
-            
-
     def header_to_string(self, res):
         s = int_to_hex(res.get('version'),4) \
             + rev_hex(res.get('prev_block_hash')) \
@@ -230,9 +218,7 @@ class BlockchainVerifier(threading.Thread):
         return rev_hex(Hash(self.header_to_string(header).decode('hex')).encode('hex'))
 
     def path(self):
-        wdir = self.config.get('blockchain_headers_path', user_dir())
-        if wdir and not os.path.exists( wdir ): os.mkdir(wdir)
-        return os.path.join( wdir, 'blockchain_headers')
+        return os.path.join( self.config.path, 'blockchain_headers')
 
     def init_headers_file(self):
         filename = self.path()
@@ -274,6 +260,7 @@ class BlockchainVerifier(threading.Thread):
             h = os.path.getsize(name)/80 - 1
             if self.local_height != h:
                 self.local_height = h
+                self.height = self.local_height
 
 
     def read_header(self, block_height):
@@ -327,3 +314,106 @@ class BlockchainVerifier(threading.Thread):
         new_bits = c + MM * i
         return new_bits, new_target
 
+
+    def request_header(self, i, h, queue):
+        print_error("requesting header %d from %s"%(h, i.server))
+        i.send([ ('blockchain.block.get_header',[h])], lambda i,r: queue.put((i,r)))
+
+    def retrieve_header(self, i, queue):
+        while True:
+            try:
+                ir = queue.get(timeout=1)
+            except Queue.Empty:
+                print_error('timeout')
+                continue
+
+            if not ir: 
+                continue
+
+            i, r = ir
+
+            if r.get('error'):
+                print_error('Verifier received an error:', r)
+                continue
+
+            # 3. handle response
+            method = r['method']
+            params = r['params']
+            result = r['result']
+
+            if method == 'blockchain.block.get_header':
+                return result
+                
+
+
+    def get_chain(self, interface, final_header):
+
+        header = final_header
+        chain = [ final_header ]
+        requested_header = False
+        queue = Queue.Queue()
+
+        while self.is_running():
+
+            if requested_header:
+                header = self.retrieve_header(interface, queue)
+                if not header: return
+                chain = [ header ] + chain
+                requested_header = False
+
+            height = header.get('block_height')
+            previous_header = self.read_header(height -1)
+            if not previous_header:
+                self.request_header(interface, height - 1, queue)
+                requested_header = True
+                continue
+
+            # verify that it connects to my chain
+            prev_hash = self.hash_header(previous_header)
+            if prev_hash != header.get('prev_block_hash'):
+                print_error("reorg")
+                self.request_header(interface, height - 1, queue)
+                requested_header = True
+                continue
+
+            else:
+                # the chain is complete
+                return chain
+
+
+    def get_chunks(self, i, header, height):
+        requested_chunks = []
+        queue = Queue.Queue()
+        min_index = (self.local_height + 1)/2016
+        max_index = (height + 1)/2016
+        for n in range(min_index, max_index + 1):
+            print_error( "requesting chunk", n )
+            i.send([ ('blockchain.block.get_chunk',[n])], lambda i,r:queue.put(r))
+            requested_chunks.append(n)
+            break
+
+        while requested_chunks:
+            try:
+                r = queue.get(timeout=1)
+            except Queue.Empty:
+                continue
+            if not r: continue
+
+            if r.get('error'):
+                print_error('Verifier received an error:', r)
+                continue
+
+            # 3. handle response
+            params = r['params']
+            result = r['result']
+
+            index = params[0]
+            self.verify_chunk(index, result)
+            requested_chunks.remove(index)
+
+
+
+
+
+
+