1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
# 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()
|
