3 files changed, 22 insertions, 22 deletions
diff --git a/chat_client.py b/chat_client.py
index 6de5b88..3b3c082 100644
--- a/chat_client.py
+++ b/chat_client.py
@@ -1,23 +1,26 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
-import sys, socket
+import sys,socket,os
 
 def chat_client(host,port):
+    whoami = os.getenv("USER")
     try:
-        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) 
         s.connect((host,port))
         while True:
-            message = input()
+            message = input("moi > ")
+            message = whoami+" > "+message
             s.sendall(message.encode())
-            data=s.recv(1024)
-            print(data.decode().strip())
+            data = s.recv(1024)
+            print(data.decode())
     except KeyboardInterrupt:
         s.close()
     finally:
         print("bye")
 
-if __name__ == "__main__":
+if __name__ ==  "__main__":
     try:
         chat_client(sys.argv[1],int(sys.argv[2]))
     except IndexError:
-        print(f"{sys.argv[0]} demande un hôte et un port en arguments")
+        print(f"{sys.argv[0]} demande un hôte où se connecter et un numéro de port")
+
diff --git a/chat_server.py b/chat_server.py
index 33f6b59..7f553a9 100644
--- a/chat_server.py
+++ b/chat_server.py
@@ -1,18 +1,21 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
-import sys,socket
+import sys,socket,os
 
 def chat_server(port):
+    whoami = os.getenv("USER")
     try:
-        s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
+        # Enabling SO_REUSEADDR option
         s.bind(('',port))
         s.listen()
         conn,addr = s.accept()
-        print(addr)
-        while 1:
+        print(f"Connexion depuis {addr[0]} sur le port {addr[1]}")
+        while True:
             data = conn.recv(1024)
             print(data.decode())
-            reponse = input()
+            reponse = input("moi > ")
+            reponse = whoami+" > "+reponse
             conn.sendall(reponse.encode())
     except KeyboardInterrupt:
         s.close()
@@ -23,11 +26,5 @@ if __name__ == "__main__":
     try:
         chat_server(int(sys.argv[1]))
     except IndexError:
-        print(f"{sys.argv[0]} demande un port en  agument")
-
-
-
-
-
-
+        print(f"{sys.argv[0]} demande un numéro de port en argument")
 
diff --git a/reverseshell_listener.py b/reverseshell_listener.py
index bbf81b4..23eb39a 100644
--- a/reverseshell_listener.py
+++ b/reverseshell_listener.py
@@ -8,7 +8,7 @@ def chat_server(port):
         s.bind(('',port))
         s.listen()
         conn,addr = s.accept()
-        print(addr)
+        print(f"Connexion depuis {addr[0]} sur le port {addr[1]}")
         while True:
             data = conn.recv(4096)
             print(data.decode(),end="")