Do not use mutables as default values!
[electrum-nvc.git] / lib / blockchain.py
index ee6d99e..ac5ea14 100644 (file)
@@ -30,16 +30,17 @@ class Blockchain(threading.Thread):
         self.config = config
         self.network = network
         self.lock = threading.Lock()
-        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 height(self):
+        return self.local_height
+
+
     def stop(self):
         with self.lock: self.running = False
 
@@ -69,40 +70,35 @@ class Blockchain(threading.Thread):
             if not header: continue
             
             height = header.get('block_height')
-            self.servers_height[i.server] = height
+
+            if height <= self.local_height:
+                continue
 
             if height > self.local_height + 50:
-                self.get_chunks(i, header, height)
-                self.network.trigger_callback('updated')
+                if not self.get_and_verify_chunks(i, header, height):
+                    continue
 
             if height > self.local_height:
                 # get missing parts from interface (until it connects to my chain)
                 chain = self.get_chain( i, header )
 
                 # skip that server if the result is not consistent
-                if not chain: continue
+                if not chain: 
+                    print_error('e')
+                    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
+                    continue
 
-            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.network.is_lagging = True
-            else:
-                self.network.is_lagging = False
-                
-            self.network.trigger_callback('updated')
+
+            self.network.new_blockchain_height(height, i)
 
 
                     
@@ -117,13 +113,13 @@ class Blockchain(threading.Thread):
             height = header.get('block_height')
 
             prev_hash = self.hash_header(prev_header)
-            bits, target = self.get_target(height/2016)
+            bits, target = self.get_target(height/2016, chain)
             _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:
+                assert int('0x'+_hash,16) < target
+            except Exception:
                 return False
 
             prev_header = header
@@ -153,7 +149,7 @@ class Blockchain(threading.Thread):
             _hash = self.hash_header(header)
             assert previous_hash == header.get('prev_block_hash')
             assert bits == header.get('bits')
-            assert eval('0x'+_hash) < target
+            assert int('0x'+_hash,16) < target
 
             previous_header = header
             previous_hash = _hash 
@@ -161,36 +157,6 @@ class Blockchain(threading.Thread):
         self.save_chunk(index, data)
         print_error("validated chunk %d"%height)
 
-
-    def verify_header(self, header):
-        # add header to the blockchain file
-        # if there is a reorg, push it in a stack
-
-        height = header.get('block_height')
-
-        prev_header = self.read_header(height -1)
-        if not prev_header:
-            # return False to request previous header
-            return False
-
-        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:
-            # this can be caused by a reorg.
-            print_error("verify header failed"+ repr(header))
-            verifier.undo_verifications()
-
-            # return False to request previous header.
-            return False
-
-        self.save_header(header)
-        print_error("verify header:", _hash, height)
-        return True
         
 
     def header_to_string(self, res):
@@ -204,7 +170,7 @@ class Blockchain(threading.Thread):
 
 
     def header_from_string(self, s):
-        hex_to_int = lambda s: eval('0x' + s[::-1].encode('hex'))
+        hex_to_int = lambda s: int('0x' + s[::-1].encode('hex'), 16)
         h = {}
         h['version'] = hex_to_int(s[0:4])
         h['prev_block_hash'] = hash_encode(s[4:36])
@@ -230,7 +196,8 @@ class Blockchain(threading.Thread):
             socket.setdefaulttimeout(30)
             print_error("downloading ", self.headers_url )
             urllib.urlretrieve(self.headers_url, filename)
-        except:
+            print_error("done.")
+        except Exception:
             print_error( "download failed. creating file", filename )
             open(filename,'wb+').close()
 
@@ -260,7 +227,6 @@ class Blockchain(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):
@@ -275,14 +241,20 @@ class Blockchain(threading.Thread):
                 return h 
 
 
-    def get_target(self, index):
+    def get_target(self, index, chain=None):
+        if chain is None:
+            chain = []  # Do not use mutables as default values!
 
         max_target = 0x00000000FFFF0000000000000000000000000000000000000000000000000000
         if index == 0: return 0x1d00ffff, max_target
 
         first = self.read_header((index-1)*2016)
         last = self.read_header(index*2016-1)
-        
+        if last is None:
+            for h in chain:
+                if h.get('block_height') == index*2016-1:
+                    last = h
         nActualTimespan = last.get('timestamp') - first.get('timestamp')
         nTargetTimespan = 14*24*60*60
         nActualTimespan = max(nActualTimespan, nTargetTimespan/4)
@@ -306,8 +278,8 @@ class Blockchain(threading.Thread):
             c = c[2:]
             i -= 1
 
-        c = eval('0x'+c[0:6])
-        if c > 0x800000: 
+        c = int('0x'+c[0:6],16)
+        if c >= 0x800000: 
             c /= 256
             i += 1
 
@@ -381,39 +353,25 @@ class Blockchain(threading.Thread):
                 return chain
 
 
-    def get_chunks(self, i, header, height):
-        requested_chunks = []
+    def get_and_verify_chunks(self, i, header, height):
+
         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)
+        n = min_index
+        while n < max_index + 1:
+            print_error( "Requesting chunk:", n )
+            r = i.synchronous_get([ ('blockchain.block.get_chunk',[n])])[0]
+            if not r: 
                 continue
+            try:
+                self.verify_chunk(n, r)
+                n = n + 1
+            except Exception:
+                print_error('Verify chunk failed!')
+                n = n - 1
+                if n < 0:
+                    return False
 
-            # 3. handle response
-            params = r['params']
-            result = r['result']
-
-            index = params[0]
-            self.verify_chunk(index, result)
-            requested_chunks.remove(index)
-
-
-
-
-
-
+        return True