create_dht.py

import os
import sys
import time
import socket
import random
from chord import *
from dht import *

nnodes = int(sys.argv[1])
print "Creating chord network with : %s nodes" % nnodes

# create ports
ports_list = [random.randrange(10000, 60000) for x in range(nnodes)]
print "Ports list : %s" % ports_list
# create addresses
address_list = map(lambda port: Address('127.0.0.1', port), ports_list)
# keep unique ones
address_list = sorted(set(address_list))
# hash the addresses
hash_list 	 = map(lambda addr: addr.__hash__(), address_list)
# create the nodes
locals_list   = create_dht(address_list)

print "Done creating peers, our pid %s is" % os.getpid()

def getid(address, offset = 0):
	return str((address.__hash__() + offset) % SIZE)

while 1:
	time.sleep(1)
	command = raw_input("Command: ")
	if command == "add_node":
		while 1:
			address = Address("127.0.0.1", random.randrange(10000, 60000))
			if not address.__hash__() in hash_list:
				ports_list.append(address.port)
				print "New node at port %s" % address.port
				address_list.append(address)
				locals_list.append(Local(address, locals_list[random.randrange(len(locals_list))].address))
				break
	elif command.split(' ')[0] == "shutdown":
		id = command.split(' ')[1]
		for address in address_list:
			if id == getid(address):
				break
		s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
		s.connect((address.ip, address.port))
		s.sendall(command + "\r\n")
		print "Response : '%s'" % s.recv(10000)
		s.close()
	else:
		connectFlag = False
		while not connectFlag:
			address = address_list[random.randrange(len(address_list))]
			s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
			try:
				s.connect((address.ip, address.port))
				connectFlag = True
			except socket.error:
				address_list.remove(address)
		s.sendall(command + "\r\n")
		print "Response : '%s'" % s.recv(10000)
		s.close()