apt_dht_Khashmir/knet.py

   1 #
   2 #  knet.py
   3 #  create a network of khashmir nodes
   4 # usage: knet.py <num_nodes> <start_port> <ip_address>
   5
   6 from random import randrange
   7 import sys, os
   8
   9 from twisted.internet import reactor
  10
  11 from khashmir import Khashmir
  12
  13 class Network:
  14     def __init__(self, size=0, startport=5555, localip='127.0.0.1'):
  15         self.num = size
  16         self.startport = startport
  17         self.localip = localip
  18
  19     def _done(self, val):
  20         self.done = 1
  21
  22     def setUp(self):
  23         self.kfiles()
  24         self.l = []
  25         for i in range(self.num):
  26             self.l.append(Khashmir('', self.startport + i, '/tmp/kh%s.db' % (self.startport + i)))
  27         reactor.iterate()
  28         reactor.iterate()
  29
  30         for i in self.l:
  31             i.addContact(self.localip, self.l[randrange(0,self.num)].port)
  32             i.addContact(self.localip, self.l[randrange(0,self.num)].port)
  33             i.addContact(self.localip, self.l[randrange(0,self.num)].port)
  34             reactor.iterate()
  35             reactor.iterate()
  36             reactor.iterate()
  37
  38         for i in self.l:
  39             self.done = 0
  40             i.findCloseNodes(self._done)
  41             while not self.done:
  42                 reactor.iterate()
  43         for i in self.l:
  44             self.done = 0
  45             i.findCloseNodes(self._done)
  46             while not self.done:
  47                 reactor.iterate()
  48
  49     def tearDown(self):
  50         for i in self.l:
  51             i.listenport.stopListening()
  52         self.kfiles()
  53
  54     def kfiles(self):
  55         for i in range(self.startport, self.startport+self.num):
  56             try:
  57                 os.unlink('/tmp/kh%s.db' % i)
  58             except:
  59                 pass
  60
  61         reactor.iterate()
  62
  63 if __name__ == "__main__":
  64     n = Network(int(sys.argv[1]), int(sys.argv[2]), sys.argv[3])
  65     n.setUp()
  66     try:
  67         reactor.run()
  68     finally:
  69         n.tearDown()