v0.4: send position along merkle path
[electrum-server.git] / backends / abe / __init__.py
index 13cd73d..2e4e806 100644 (file)
@@ -10,6 +10,29 @@ from Queue import Queue
 import time, threading
 
 
+import hashlib
+encode = lambda x: x[::-1].encode('hex')
+decode = lambda x: x.decode('hex')[::-1]
+Hash = lambda x: hashlib.sha256(hashlib.sha256(x).digest()).digest()
+
+def rev_hex(s):
+    return s.decode('hex')[::-1].encode('hex')
+
+def int_to_hex(i, length=1):
+    s = hex(i)[2:].rstrip('L')
+    s = "0"*(2*length - len(s)) + s
+    return rev_hex(s)
+
+def header_to_string(res):
+    s = int_to_hex(res.get('version'),4) \
+        + rev_hex(res.get('prev_block_hash')) \
+        + rev_hex(res.get('merkle_root')) \
+        + int_to_hex(int(res.get('timestamp')),4) \
+        + int_to_hex(int(res.get('bits')),4) \
+        + int_to_hex(int(res.get('nonce')),4)
+    return s
+
+
 class AbeStore(Datastore_class):
 
     def __init__(self, config):
@@ -406,6 +429,40 @@ class AbeStore(Datastore_class):
         return out
         
 
+    def get_chunk(self, index):
+        sql = """
+            SELECT
+                block_hash,
+                block_version,
+                block_hashMerkleRoot,
+                block_nTime,
+                block_nBits,
+                block_nNonce,
+                block_height,
+                prev_block_hash,
+                block_height
+              FROM chain_summary
+             WHERE block_height >= %d AND block_height< %d AND in_longest = 1"""%(index*2016, (index+1)*2016)
+
+        out = self.safe_sql(sql)
+        msg = ''
+        for row in out:
+            (block_hash, block_version, hashMerkleRoot, nTime, nBits, nNonce, height, prev_block_hash, block_height) \
+                = ( self.hashout_hex(row[0]), int(row[1]), self.hashout_hex(row[2]), int(row[3]), int(row[4]), int(row[5]), int(row[6]), self.hashout_hex(row[7]), int(row[8]) )
+            h = {"block_height":block_height, "version":block_version, "prev_block_hash":prev_block_hash, 
+                   "merkle_root":hashMerkleRoot, "timestamp":nTime, "bits":nBits, "nonce":nNonce}
+
+            if h.get('block_height')==0: h['prev_block_hash'] = "0"*64
+            msg += header_to_string(h)
+
+            #print "hash", encode(Hash(msg.decode('hex')))
+            #if h.get('block_height')==1:break
+
+        print "get_chunk", index, len(msg)
+        return msg
+
+
+
     def get_tx_merkle(self, tx_hash):
 
         out = self.safe_sql("""
@@ -422,21 +479,20 @@ class AbeStore(Datastore_class):
         block_height = int(out[0][0])
 
         merkle = []
+        tx_pos = None
+
         # list all tx in block
         for row in self.safe_sql("""
             SELECT DISTINCT tx_id, tx_pos, tx_hash
               FROM txin_detail
              WHERE block_id = ?
              ORDER BY tx_pos""", (block_id,)):
-            tx_id, tx_pos, tx_h = row
-            merkle.append(tx_h)
+            _id, _pos, _hash = row
+            merkle.append(_hash)
+            if _hash == tx_hash: tx_pos = int(_pos)
 
         # find subset.
         # TODO: do not compute this on client request, better store the hash tree of each block in a database...
-        import hashlib
-        encode = lambda x: x[::-1].encode('hex')
-        decode = lambda x: x.decode('hex')[::-1]
-        Hash = lambda x: hashlib.sha256(hashlib.sha256(x).digest()).digest()
 
         merkle = map(decode, merkle)
         target_hash = decode(tx_hash)
@@ -446,19 +502,19 @@ class AbeStore(Datastore_class):
             if len(merkle)%2: merkle.append( merkle[-1] )
             n = []
             while merkle:
+                new_hash = Hash( merkle[0] + merkle[1] )
                 if merkle[0] == target_hash:
-                    s.append( "L" + encode(merkle[1]))
-                    n.append( target_hash )
+                    s.append( encode(merkle[1]))
+                    target_hash = new_hash
                 elif merkle[1] == target_hash:
-                    s.append( "R" + encode(merkle[0]))
-                    n.append( target_hash)
-                else:
-                    n.append( Hash( merkle[0] + merkle[1] ) )
+                    s.append( encode(merkle[0]))
+                    target_hash = new_hash
+                n.append( new_hash )
                 merkle = merkle[2:]
             merkle = n
 
         # send result
-        return {"block_height":block_height,"merkle":s}
+        return {"block_height":block_height, "merkle":s, "pos":tx_pos}
 
 
 
@@ -517,8 +573,10 @@ class AbeStore(Datastore_class):
         with store.dblock:
             store.catch_up()
             store.memorypool_update()
-            block_number = store.get_block_number(store.chain_id)
-            return block_number
+            height = store.get_block_number( store.chain_id )
+
+        block_header = store.get_block_header( height )
+        return block_header
 
 
 
@@ -543,12 +601,12 @@ class BlockchainProcessor(Processor):
     def __init__(self, config):
         Processor.__init__(self)
         self.store = AbeStore(config)
-        self.block_number = -1
         self.watched_addresses = []
 
         # catch_up first
-        n = self.store.main_iteration()
-        print "blockchain: %d blocks"%n
+        self.block_header = self.store.main_iteration()
+        self.block_number = self.block_header.get('block_height')
+        print "blockchain: %d blocks"%self.block_number
 
         threading.Timer(10, self.run_store_iteration).start()
 
@@ -564,6 +622,9 @@ class BlockchainProcessor(Processor):
         if method == 'blockchain.numblocks.subscribe':
             result = self.block_number
 
+        elif method == 'blockchain.headers.subscribe':
+            result = self.block_header
+
         elif method == 'blockchain.address.subscribe':
             try:
                 address = params[0]
@@ -589,6 +650,14 @@ class BlockchainProcessor(Processor):
                 error = str(e) + ': %d'% height
                 print "error:", error
 
+        elif method == 'blockchain.block.get_chunk':
+            try:
+                index = params[0]
+                result = self.store.get_chunk( index ) 
+            except BaseException, e:
+                error = str(e) + ': %d'% index
+                print "error:", error
+
         elif method == 'blockchain.transaction.broadcast':
             txo = self.store.send_tx(params[0])
             print "sent tx:", txo
@@ -622,7 +691,7 @@ class BlockchainProcessor(Processor):
     def run_store_iteration(self):
         
         try:
-            block_number = self.store.main_iteration()
+            block_header = self.store.main_iteration()
         except:
             traceback.print_exc(file=sys.stdout)
             print "terminating"
@@ -632,11 +701,15 @@ class BlockchainProcessor(Processor):
             print "exit timer"
             return
 
-        if self.block_number != block_number:
-            self.block_number = block_number
+        if self.block_number != block_header.get('block_height'):
+            self.block_number = block_header.get('block_height')
             print "block number:", self.block_number
             self.push_response({ 'id': None, 'method':'blockchain.numblocks.subscribe', 'params':[self.block_number] })
 
+        if self.block_header != block_header:
+            self.block_header = block_header
+            self.push_response({ 'id': None, 'method':'blockchain.headers.subscribe', 'params':[self.block_header] })
+
         while True:
             try:
                 addr = self.store.address_queue.get(False)