scripts/peers

   1 #!/usr/bin/env python
   2
   3 import time, electrum, Queue
   4 from electrum import Interface, SimpleConfig
   5 from electrum.network import filter_protocol, parse_servers
   6 from collections import defaultdict
   7
   8 # 1. start interface and wait for connection
   9 interface = electrum.Interface('ecdsa.net:50002:s')
  10 interface.start(wait = True)
  11 if not interface.is_connected:
  12     print "not connected"
  13     exit()
  14
  15 # 2. get list of peers
  16 q = Queue.Queue()
  17 interface.send([('server.peers.subscribe',[])], lambda i,x: q.put(x))
  18 r = q.get(timeout=10000)
  19 peers = parse_servers(r.get('result'))
  20 peers = filter_protocol(peers,'s')
  21
  22 # start interfaces
  23 config = SimpleConfig()
  24 interfaces = map ( lambda server: Interface(server, config), peers )
  25 results_queue = Queue.Queue()
  26 reached_servers = []
  27 for i in interfaces: i.start(q)
  28
  29 while peers:
  30     i = q.get(timeout=10)
  31     peers.remove(i.server)
  32     if i.is_connected:
  33         i.send([('blockchain.headers.subscribe',[])], lambda i,x: results_queue.put((i,x)))
  34         reached_servers.append(i.server)
  35
  36 def analyze(results):
  37     out = {}
  38     dd = {}
  39     for k, v in results.items():
  40         height = v.get('block_height')
  41         merkle = v.get('merkle_root')
  42         utxo = v.get('utxo_root')
  43         d = dd.get(merkle, defaultdict(int))
  44         d[utxo] += 1
  45         dd[merkle] = d
  46
  47     refs = {}
  48     for merkle, d in dd.items():
  49         v = d.values()
  50         m = max(v)
  51         ref = d.keys()[v.index(m)]
  52         refs[merkle] = ref, m
  53
  54     for k, v in results.items():
  55         height = v.get('block_height')
  56         merkle = v.get('merkle_root')
  57         utxo = v.get('utxo_root')
  58         ref_utxo, num = refs.get(merkle)
  59
  60         if ref_utxo != utxo and num > 1:
  61             out[k] = height, merkle, utxo
  62
  63     return out
  64
  65
  66 results = {}
  67 while reached_servers:
  68     i, r = results_queue.get(timeout=10000)
  69     results[i.server] = r.get('result')
  70     reached_servers.remove(i.server)
  71
  72 errors = analyze(results).keys()
  73
  74 for n,v in sorted(results.items(), key=lambda x:x[1]['block_height']):
  75     print "%40s"%n, v['block_height'], v['utxo_root'], "error" if n in errors else "ok"
  76
  77
  78