69 lines
1.6 KiB
Python
69 lines
1.6 KiB
Python
|
#!/usr/bin/python
|
||
|
|
|
||
|
|
import socket
|
||
|
|
import sys
|
||
|
|
from types import StringType, IntType, LongType, DictType, ListType, TupleType
|
||
|
|
import random
|
||
|
|
|
||
|
|
port = int(sys.argv[1])
|
||
|
|
|
||
|
|
# from BitTorrent 4.3.0
|
||
|
|
def encode_bencached(x,r):
|
||
|
|
r.append(x.bencoded)
|
||
|
|
|
||
|
|
def encode_int(x, r):
|
||
|
|
r.extend(('i', str(x), 'e'))
|
||
|
|
|
||
|
|
def encode_string(x, r):
|
||
|
|
r.extend((str(len(x)), ':', x))
|
||
|
|
|
||
|
|
def encode_list(x, r):
|
||
|
|
r.append('l')
|
||
|
|
for i in x:
|
||
|
|
encode_func[type(i)](i, r)
|
||
|
|
r.append('e')
|
||
|
|
|
||
|
|
def encode_dict(x,r):
|
||
|
|
r.append('d')
|
||
|
|
ilist = x.items()
|
||
|
|
ilist.sort()
|
||
|
|
for k, v in ilist:
|
||
|
|
r.extend((str(len(k)), ':', k))
|
||
|
|
encode_func[type(v)](v, r)
|
||
|
|
r.append('e')
|
||
|
|
|
||
|
|
encode_func = {}
|
||
|
|
encode_func[IntType] = encode_int
|
||
|
|
encode_func[LongType] = encode_int
|
||
|
|
encode_func[StringType] = encode_string
|
||
|
|
encode_func[ListType] = encode_list
|
||
|
|
encode_func[TupleType] = encode_list
|
||
|
|
encode_func[DictType] = encode_dict
|
||
|
|
|
||
|
|
def bencode(x):
|
||
|
|
r = []
|
||
|
|
encode_func[type(x)](x, r)
|
||
|
|
return ''.join(r)
|
||
|
|
|
||
|
|
def send_dht_message(msg):
|
||
|
|
s.sendto(bencode(msg), 0, ('127.0.0.1', port))
|
||
|
|
|
||
|
|
def random_key():
|
||
|
|
ret = ''
|
||
|
|
for i in range(0, 20):
|
||
|
|
ret += chr(random.randint(0, 255))
|
||
|
|
return ret
|
||
|
|
|
||
|
|
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||
|
|
node_id = '1' * 20;
|
||
|
|
query = 'get_peers'
|
||
|
|
|
||
|
|
print 'test random info-hashes'
|
||
|
|
for i in xrange(1, 30000):
|
||
|
|
send_dht_message({'a': {'id': node_id, 'info_hash': random_key()}, 'q': query, 'y': 'q', 't': '%d' % i})
|
||
|
|
|
||
|
|
print 'test random peer-ids'
|
||
|
|
for i in xrange(1, 30000):
|
||
|
|
send_dht_message({'a': {'id': random_key(), 'info_hash': random_key()}, 'q': query, 'y': 'q', 't': '%d' % i})
|
||
|
|
|