simple_server.py

# test metls data transmission throughput
# vary the number of middleboxes on server to client path

import socket
import sys
from tlslite.api import *
from tlslite.utils.cryptomath import *
import random
import os


if __name__ == '__main__':
	if len(sys.argv) != 3:
		print 'usage: ' + sys.argv[0] + ' ip port'
	else:
		private_key_file = "tests/serverX509Key.pem"
		cert_file = "tests/serverX509Cert.pem"
		s = open(private_key_file, "rb").read()
		if sys.version_info[0] >= 3:
			s = str(s, 'utf-8')
		# OpenSSL/m2crypto does not support RSASSA-PSS certificates
		privateKey = parsePEMKey(s, private=True, implementations=["python"])

		s = open(cert_file, "rb").read()
		if sys.version_info[0] >= 3:
			s = str(s, 'utf-8')
		x509 = X509()
		x509.parse(s)
		cert_chain = X509CertChain([x509])

		ip = sys.argv[1]
		port = int(sys.argv[2])
		#number_of_middleboxes = int(sys.argv[3])

		settings = HandshakeSettings()
		# set to tls 1.2 
		settings.maxVersion = (3, 3)
		settings.versions = [(3, 3)]
		#settings.number_of_middleboxes = number_of_middleboxes
		settings.print_debug_info = True
		settings.middlebox_public_key = cert_chain.getEndEntityPublicKey()

		sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
		sock.bind((ip, port))
		sock.listen(5)
		print 'server socket listening on ' + ip + ':' + str(port)
		while True:
			client_sock, client_addr = sock.accept()
			conn = TLSConnection(client_sock)
			conn.middlebox_public_key = settings.middlebox_public_key
			print 'about to handshake'
			conn.handshakeServer(certChain=cert_chain, privateKey=privateKey, reqCert=False, settings=settings)
			print 'handshakeServer succeeded'
			# transmit data to client
			amout = 1024 * 1024 * 5 # 5 MB data
			cnt = 0
			while cnt < amout:
				conn.sendall(bytearray(4096))
				cnt += 4096
			conn.close()

		# # test data transfer
		# count = 0
		# while True:
		# 	data = conn.recv(20000)
		# 	if len(data) > 0:
		# 		count += len(data)
		# 		print 'received ' + str(count) + ' bytes data'
		# 		conn.sendall(data)
		# 	else:
		# 		break