77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
# The contents of this file are subject to the BitTorrent Open Source License
|
|
# Version 1.1 (the License). You may not copy or use this file, in either
|
|
# source code or executable form, except in compliance with the License. You
|
|
# may obtain a copy of the License at http://www.bittorrent.com/license/.
|
|
#
|
|
# Software distributed under the License is distributed on an AS IS basis,
|
|
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
|
|
# for the specific language governing rights and limitations under the
|
|
# License.
|
|
|
|
#
|
|
# knet.py
|
|
# create a network of khashmir nodes
|
|
# usage: knet.py <num_nodes> <start_port> <ip_address>
|
|
|
|
from khashmir import Khashmir
|
|
from random import randrange
|
|
import sys, os
|
|
|
|
class Network:
|
|
def __init__(self, size=0, startport=5555, localip='127.0.0.1'):
|
|
self.num = size
|
|
self.startport = startport
|
|
self.localip = localip
|
|
|
|
def _done(self, val):
|
|
self.done = 1
|
|
|
|
def setUp(self):
|
|
self.kfiles()
|
|
self.l = []
|
|
for i in range(self.num):
|
|
self.l.append(Khashmir('', self.startport + i, '/tmp/kh%s.db' % (self.startport + i)))
|
|
reactor.iterate()
|
|
reactor.iterate()
|
|
|
|
for i in self.l:
|
|
i.addContact(self.localip, self.l[randrange(0,self.num)].port)
|
|
i.addContact(self.localip, self.l[randrange(0,self.num)].port)
|
|
i.addContact(self.localip, self.l[randrange(0,self.num)].port)
|
|
reactor.iterate()
|
|
reactor.iterate()
|
|
reactor.iterate()
|
|
|
|
for i in self.l:
|
|
self.done = 0
|
|
i.findCloseNodes(self._done)
|
|
while not self.done:
|
|
reactor.iterate()
|
|
for i in self.l:
|
|
self.done = 0
|
|
i.findCloseNodes(self._done)
|
|
while not self.done:
|
|
reactor.iterate()
|
|
|
|
def tearDown(self):
|
|
for i in self.l:
|
|
i.listenport.stopListening()
|
|
self.kfiles()
|
|
|
|
def kfiles(self):
|
|
for i in range(self.startport, self.startport+self.num):
|
|
try:
|
|
os.unlink('/tmp/kh%s.db' % i)
|
|
except:
|
|
pass
|
|
|
|
reactor.iterate()
|
|
|
|
if __name__ == "__main__":
|
|
n = Network(int(sys.argv[1]), int(sys.argv[2]), sys.argv[3])
|
|
n.setUp()
|
|
try:
|
|
reactor.run()
|
|
finally:
|
|
n.tearDown()
|