Server:
import socket
import sys
import argparse
host = 'localhost'
data_payload = 2048
backlog = 5
def echo_server(port):
'''A simple echo server'''
#Create a TCP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#Enable reuse address /port
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
#Bind the socket to the port
server_address = (host, port)
print "Starting up echo server on %s port %s " %server_address
sock.bind(server_address)
#Listen to clients, backlog argument specifies the max number of queued connections
sock.listen(5)
while True:
print "Waiting to receive message from client"
client, address = sock.accept()
data = client.recv(data_payload)
if data:
print "Data: %s" %data
client.send(data)
print "send %s bytes back to %s" %(len(data), address)
#End connection
client.close()
if __name__=="__main__":
parser = argparse.ArgumentParser(description='Socket Server Example')
parser.add_argument('--port', action='store', dest='port', type=int, required=True)
given_args = parser.parse_args()
port = given_args.port
echo_server(port)
Client:
import socket
import sys
import argparse
host = 'localhost'
def echo_client(port):
'''A simple echo client'''
#Create a TCP/IP socket
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
#Connect the socket to the server
server_address = (host, port)
print "Connecting to %s port %s" %server_address
sock.connect(server_address)
#Send data
try:
#Send data
message = "Test message. This will be echoed."
print "Sending %s" %message
sock.sendall(message)
#Look for the response
amount_received = 0
amount_expected = len(message)
while amount_received < amount_expected:
data = sock.recv(16)
amount_received +=len(data)
print "Receviced :%s " %data
except socket.error, e:
print "Socket error: %s" %str(e)
except Exception, e:
print "Other exception: %s" %str(e)
finally:
print "Closing connection to the server"
sock.close()
if __name__=="__main__":
parser = argparse.ArgumentParser(description='Socket Client Example')
parser.add_argument('--port', action='store', dest='port', type=int, required=True)
given_args = parser.parse_args()
port = given_args.port
echo_client(port)