Add files via upload

Author: narjesno

Description.pdf

@@ -0,0 +1,90 @@
+
+import socket 
+import select
+import pickle
+import sys 
+from socket import error as SocketError
+import errno
+
+LIST = "1"
+SEND = "2"
+RECIEVE = "3"
+EXIT = "4"
+OK = "OK"
+
+PORT = 9000
+
+HEADER_LENGTH = 1024
+
+
+server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+
+IP_address = 'localhost'
+print(IP_address)	
+server.connect((IP_address, PORT))
+print("Connected to server ... ")
+
+print("Enter client's name: ")
+name = input()
+server.send(name.encode())
+
+
+while True: 
+
+	try :
+
+		print("Choose one of these options : 1.List 	 2.Send 	3.Receive 	4.Exit")
+		
+		command = input()
+
+		server.send(command.encode())
+				
+		if command == LIST :
+
+			data = server.recv(HEADER_LENGTH)
+			client_list = pickle.loads(data)
+			print(client_list)
+
+		elif command == SEND :
+
+			while True :
+				print("Enter the name of your receiver:")
+				receiver = input()
+				server.send(receiver.encode())
+				existance = (server.recv(HEADER_LENGTH)).decode()
+				
+				if existance != OK:
+					print("Receiver not found, please try another receiver.")
+					break	
+
+				print("Enter the message: ")
+				message = input()
+				server.send(message.encode())
+				break;
+
+
+		elif command == RECIEVE :
+
+			data = server.recv(HEADER_LENGTH)
+			message_list = pickle.loads(data)
+			if len(message_list) == 0 :
+				print("Your inbox in empty")
+			print(message_list)
+
+		elif command == EXIT :
+			print("Okay bye bye")
+			exit()
+
+		else: 
+			print("Please try again ")	
+
+	except Exception as e:
+		print("Exiting")
+		break
+		# if e.errno != errno.ECONNRESET :
+		# 	raise # Not error we are looking for
+		# pass # Handle error here.
+
+
+
+server.close() 

client.py

@@ -0,0 +1,169 @@
+
+import socket
+import pickle
+import select 
+import sys
+import threading 
+from _thread import start_new_thread
+
+
+HEADER_LENGTH = 1024
+
+LIST = "1"
+SEND = "2"
+RECEIVE = "3"
+EXIT = "4"
+OK = "OK"
+OH = "NOT OK"
+client_list = []
+
+class Message:
+
+	def	__init__(self, sender, message):
+		self.sender = sender
+		self.message = message
+
+class Client:
+
+	def __init__(self,client_address,client_socket):
+		global client_list
+		self.socket=client_socket
+		self.address = client_address
+		self.messages = []
+		client_list.append(self)
+		print (f"{client_address} connected")
+
+	def set_name(self, name):
+		self.client_name = name
+
+	def kill_session(self):
+		global client_list
+		client_list.remove(self)
+
+	def send(self, data):
+		self.socket.send(data)
+
+	def recv(self):
+		return self.socket.recv(HEADER_LENGTH).decode()
+
+	def get_name(self):
+		return self.client_name
+
+	def insert_message(self, message):
+		self.messages.append(message)
+
+	def get_messages(self):
+		message_list = []
+		for message in self.messages:
+			message_str = message.sender + " : " + message.message
+			message_list.append(message_str)
+		return message_list
+
+
+def run(client):
+
+	global client_list
+
+	client_name = client.socket.recv(HEADER_LENGTH).decode()
+	if not client_name:
+		print("Client closed connection")
+		client.kill_session()
+		return
+
+	print(client_name)
+	client.set_name(client_name)
+
+
+	while True:
+
+		command = client.recv()
+		print(command)
+		if not command:
+			client.kill_session()
+			print("client closed connection")
+			break
+
+		if command == LIST :
+
+			client_names = []
+			for client in client_list:
+				client_names.append(client.get_name())
+			data = pickle.dumps(client_names)
+			client.send(data)
+
+		elif command == SEND:
+
+			receiver = client.recv()
+			receiver_client = None
+
+			for cl in client_list:
+				if cl.get_name() == receiver:
+					receiver_client = cl
+			if receiver_client is None:
+				client.send(OH.encode())
+				continue
+			else:
+				client.send(OK.encode())
+
+			message = client.recv()
+			print(message)
+			client_message = Message(client.get_name(), message)
+			receiver_client.insert_message(client_message)
+
+			print("message inserted")
+
+		elif command == RECEIVE:
+			message_list = client.get_messages()
+			print(message_list)
+			data = pickle.dumps(message_list)
+			client.send(data)
+
+
+		elif command == EXIT:
+			print (str(client_name) + " disconnected.")
+			client.kill_session()
+			break
+		else :
+			print("None of the above :/ puta ")
+
+
+#reserve a port on your computer, in our case :9000 . can be anyhting
+
+server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+#server.TCPServer.allow_reuse_address = True
+print ("Server Socket successfully created")
+
+
+
+IP_address = 'localhost'
+Port = '9000' 
+
+
+server.bind((IP_address, int(Port)))	
+
+#put socket into listening mode
+
+server.listen(5) 
+
+print(f"Listening for connections on {IP_address}:{Port}...")
+
+#forever loop until client wants to exit
+
+while True: 
+	try:
+		c, address = server.accept()
+		#print(client)
+		#client.settimeout(60)
+
+		start_new_thread(run, (Client(address, c),))
+
+
+	except Exception as e:
+		print("Exiting")
+		break
+
+print("Closing server socket")
+server.close()
+
+
+