Comment out localhost
[electrum-nvc.git] / lib / blockchain.py
index ffa0623..aa4880c 100644 (file)
@@ -21,127 +21,101 @@ import threading, time, Queue, os, sys, shutil
 from util import user_dir, appdata_dir, print_error
 from bitcoin import *
 
+try:
+    from ltc_scrypt import getPoWHash
+except ImportError:
+    print_msg("Warning: scrypt not available, using fallback")
+    from scrypt import scrypt_1024_1_1_80 as getPoWHash
 
-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.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.interface = interface
-        interface.register_channel('verifier')
+        self.headers_url = 'http://novacoin.su/static/blockchain_headers'
         self.set_local_height()
+        self.queue = Queue.Queue()
 
-
-
-    def start_interfaces(self):
-        import interface
-        servers = interface.DEFAULT_SERVERS
-        servers = interface.filter_protocol(servers,'s')
-        print_error("using %d servers"% len(servers))
-        self.interfaces = map ( lambda server: interface.Interface({'server':server} ), servers )
-
-        for i in self.interfaces:
-            i.start()
-            # subscribe to block headers
-            i.register_channel('verifier')
-            i.register_channel('get_header')
-            i.send([ ('blockchain.headers.subscribe',[])], 'verifier')
-            # note: each interface should send its results directly to a queue, instead of channels
-            # pass the queue to the interface, so that several can share the same queue
-
-
-    def get_new_response(self):
-        # listen to interfaces, forward to verifier using the queue
-        while self.is_running():
-            for i in self.interfaces:
-                try:
-                    r = i.get_response('verifier',timeout=0)
-                except Queue.Empty:
-                    continue
-
-                result = r.get('result')
-                if result:
-                    return (i,result)
-
-            time.sleep(1)
-
-
+    
+    def height(self):
+        return self.local_height
 
 
     def stop(self):
         with self.lock: self.running = False
-        #self.interface.poke('verifier')
+
 
     def is_running(self):
         with self.lock: return self.running
 
 
-    def request_header(self, i, h):
-        print_error("requesting header %d from %s"%(h, i.server))
-        i.send([ ('blockchain.block.get_header',[h])], 'get_header')
+    def run(self):
+        self.init_headers_file()
+        self.set_local_height()
+        print_error( "blocks:", self.local_height )
+
+        with self.lock:
+            self.running = True
+
+        while self.is_running():
 
-    def retrieve_header(self, i):
-        while True:
             try:
-                r = i.get_response('get_header',timeout=1)
+                result = self.queue.get()
             except Queue.Empty:
-                print_error('timeout')
                 continue
 
-            if r.get('error'):
-                print_error('Verifier received an error:', r)
+            if not result: continue
+
+            i, header = result
+            if not header: continue
+            
+            height = header.get('block_height')
+
+            if height <= self.local_height:
                 continue
 
-            # 3. handle response
-            method = r['method']
-            params = r['params']
-            result = r['result']
+            if height > self.local_height + 50:
+                if not self.get_and_verify_chunks(i, header, height):
+                    continue
 
-            if method == 'blockchain.block.get_header':
-                return result
+            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: 
+                    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)
+                else:
+                    print_error("error", i.server)
+                    # todo: dismiss that server
+                    continue
 
-    def get_chain(self, interface, final_header):
 
-        header = final_header
-        chain = [ final_header ]
-        requested_header = False
-        
-        while self.is_running():
+            self.network.new_blockchain_height(height, i)
 
-            if requested_header:
-                header = self.retrieve_header(interface)
-                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)
-                requested_header = True
-                continue
+                    
+    def calculate_target(self, bits):
+        # convert to bignum
+        MM = 256*256*256
+        a = bits%MM
+        if a < 0x8000:
+            a *= 256
+        return (a) * pow(2, 8 * (bits/MM - 3))
 
-            # 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)
-                requested_header = True
-                continue
 
-            else:
-                # the chain is complete
-                return chain
-                    
-            
     def verify_chain(self, chain):
 
         first_header = chain[0]
@@ -152,13 +126,13 @@ class BlockchainVerifier(threading.Thread):
             height = header.get('block_height')
 
             prev_hash = self.hash_header(prev_header)
-            bits, target = self.get_target(height/2016)
-            _hash = self.hash_header(header)
+            header_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:
+                if header.get('nonce') != 0:
+                    assert int('0x'+_hash,16) < self.calculate_target(header.get('bits'))
+            except Exception:
                 return False
 
             prev_header = header
@@ -166,37 +140,6 @@ class BlockchainVerifier(threading.Thread):
         return True
 
 
-    def get_chunks(self, i, header, height):
-        requested_chunks = []
-        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])], 'get_header')
-            requested_chunks.append(n)
-            break
-
-        while requested_chunks:
-            try:
-                r = i.get_response('get_header',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
-            method = r['method']
-            params = r['params']
-            result = r['result']
-
-            if method == 'blockchain.block.get_chunk':
-                index = params[0]
-                self.verify_chunk(index, result)
-                requested_chunks.remove(index)
-
 
     def verify_chunk(self, index, hexdata):
         data = hexdata.decode('hex')
@@ -208,59 +151,26 @@ class BlockchainVerifier(threading.Thread):
         else:
             prev_header = self.read_header(index*2016-1)
             if prev_header is None: raise
-            previous_hash = self.hash_header(prev_header)
-
-        bits, target = self.get_target(index)
+            _hash = self.hash_header(prev_header)
 
         for i in range(num):
             height = index*2016 + i
             raw_header = data[i*80:(i+1)*80]
             header = self.header_from_string(raw_header)
-            _hash = self.hash_header(header)
+            header_hash = self.hash_header(header)
+
             assert previous_hash == header.get('prev_block_hash')
-            assert bits == header.get('bits')
-            assert eval('0x'+_hash) < target
+            if header.get('nonce') != 0:
+                assert int('0x'+header_hash,16) < calculate_target(header.get('bits'))
 
             previous_header = header
-            previous_hash = _hash 
+            previous_hash = header_hash 
 
         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):
         s = int_to_hex(res.get('version'),4) \
             + rev_hex(res.get('prev_block_hash')) \
@@ -272,7 +182,7 @@ class BlockchainVerifier(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])
@@ -283,7 +193,7 @@ class BlockchainVerifier(threading.Thread):
         return h
 
     def hash_header(self, header):
-        return rev_hex(Hash(self.header_to_string(header).decode('hex')).encode('hex'))
+        return rev_hex(getPoWHash(self.header_to_string(header).decode('hex')).encode('hex'))
 
     def path(self):
         return os.path.join( self.config.path, 'blockchain_headers')
@@ -298,7 +208,8 @@ class BlockchainVerifier(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()
 
@@ -328,7 +239,6 @@ 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):
@@ -342,106 +252,91 @@ class BlockchainVerifier(threading.Thread):
                 h = self.header_from_string(h)
                 return h 
 
+    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 get_target(self, index):
-
-        max_target = 0x00000000FFFF0000000000000000000000000000000000000000000000000000
-        if index == 0: return 0x1d00ffff, max_target
-
-        first = self.read_header((index-1)*2016)
-        last = self.read_header(index*2016-1)
-        
-        nActualTimespan = last.get('timestamp') - first.get('timestamp')
-        nTargetTimespan = 14*24*60*60
-        nActualTimespan = max(nActualTimespan, nTargetTimespan/4)
-        nActualTimespan = min(nActualTimespan, nTargetTimespan*4)
-
-        bits = last.get('bits') 
-        # convert to bignum
-        MM = 256*256*256
-        a = bits%MM
-        if a < 0x8000:
-            a *= 256
-        target = (a) * pow(2, 8 * (bits/MM - 3))
+    def retrieve_header(self, i, queue):
+        while True:
+            try:
+                ir = queue.get(timeout=1)
+            except Queue.Empty:
+                print_error('timeout')
+                continue
 
-        # new target
-        new_target = min( max_target, (target * nActualTimespan)/nTargetTimespan )
-        
-        # convert it to bits
-        c = ("%064X"%new_target)[2:]
-        i = 31
-        while c[0:2]=="00":
-            c = c[2:]
-            i -= 1
+            if not ir: 
+                continue
 
-        c = eval('0x'+c[0:6])
-        if c > 0x800000: 
-            c /= 256
-            i += 1
+            i, r = ir
 
-        new_bits = c + MM * i
-        return new_bits, new_target
+            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 run(self):
-        self.start_interfaces()
-        
-        self.init_headers_file()
-        self.set_local_height()
-        print_error( "blocks:", self.local_height )
+    def get_chain(self, interface, final_header):
 
-        with self.lock:
-            self.running = True
+        header = final_header
+        chain = [ final_header ]
+        requested_header = False
+        queue = Queue.Queue()
 
         while self.is_running():
 
-            i, header = self.get_new_response()
-            
-            height = header.get('block_height')
-
-            if height > self.local_height + 50:
-                self.get_chunks(i, header, height)
-
-
-            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
-                
-                # 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
-
-    
-
-
+            if requested_header:
+                header = self.retrieve_header(interface, queue)
+                if not header: return
+                chain = [ header ] + chain
+                requested_header = False
 
-if __name__ == "__main__":
-    import interface, simple_config
-    
-    config = simple_config.SimpleConfig({'verbose':True})
+            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
 
-    i0 = interface.Interface()
-    i0.start()
+            # 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
 
-    bv = BlockchainVerifier(i0, config)
-    bv.start()
+            else:
+                # the chain is complete
+                return chain
 
 
-    # listen to interfaces, forward to verifier using the queue
-    while 1:
-        time.sleep(1)
+    def get_and_verify_chunks(self, i, header, height):
 
+        queue = Queue.Queue()
+        min_index = (self.local_height + 1)/2016
+        max_index = (height + 1)/2016
+        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
 
+        return True