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 print "Contacting %d servers"%len(peers)
  23 # start interfaces
  24 config = SimpleConfig()
  25 interfaces = map ( lambda server: Interface(server, config), peers )
  26 results_queue = Queue.Queue()
  27 reached_servers = []
  28 for i in interfaces: i.start(q)
  29
  30 while peers:
  31     try:
  32         i = q.get(timeout=3)
  33     except:
  34         break
  35     peers.remove(i.server)
  36     if i.is_connected:
  37         i.send([('blockchain.headers.subscribe',[])], lambda i,x: results_queue.put((i,x)))
  38         reached_servers.append(i.server)
  39
  40 print "%d servers could be reached"%len(reached_servers)
  41
  42 def analyze(results):
  43     out = {}
  44     dd = {}
  45     for k, v in results.items():
  46         height = v.get('block_height')
  47         merkle = v.get('merkle_root')
  48         utxo = v.get('utxo_root')
  49         d = dd.get(merkle, defaultdict(int))
  50         d[utxo] += 1
  51         dd[merkle] = d
  52
  53     refs = {}
  54     for merkle, d in dd.items():
  55         v = d.values()
  56         m = max(v)
  57         ref = d.keys()[v.index(m)]
  58         refs[merkle] = ref, m
  59
  60     for k, v in results.items():
  61         height = v.get('block_height')
  62         merkle = v.get('merkle_root')
  63         utxo = v.get('utxo_root')
  64         ref_utxo, num = refs.get(merkle)
  65
  66         if ref_utxo != utxo and num > 1:
  67             out[k] = height, merkle, utxo
  68
  69     return out
  70
  71
  72 results = {}
  73 while reached_servers:
  74     i, r = results_queue.get(timeout=10000)
  75     results[i.server] = r.get('result')
  76     reached_servers.remove(i.server)
  77
  78 errors = analyze(results).keys()
  79
  80 for n,v in sorted(results.items(), key=lambda x:x[1]['block_height']):
  81     print "%40s"%n, v['block_height'], v['utxo_root'], "error" if n in errors else "ok"
  82
  83
  84