sockets

3 files changed, 11 insertions, 7 deletions

diff --git a/chat_client.py b/chat_client.py
index 88bb7ae..affc1d3 100644
--- a/chat_client.py
+++ b/chat_client.py
@@ -1,16 +1,18 @@
 #!/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.connect((host,port))
         while True:
-            message = input()
+            message = input("moi > ")
+            message = whoami+" > "+message # à commenter pour le bindshell
             s.sendall(message.encode())
             data = s.recv(1024)
-            print(data.decode())
+            print(data.decode().strip())
     except KeyboardInterrupt:
         s.close()
     finally:
diff --git a/chat_server.py b/chat_server.py
index 13ad880..aecf8c9 100644
--- a/chat_server.py
+++ b/chat_server.py
@@ -1,8 +1,9 @@
 #!/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.bind(('',port))
@@ -12,8 +13,9 @@ def chat_server(port):
         while True:
             data = conn.recv(1024)
             print(data.decode())
-            reponse = input()
-            conn.sendall(reponse.encode())
+            reponse = input("moi > ")
+            reponse = whoami+" > "+reponse
+            conn.sendall(reponse.encode().strip())
     except KeyboardInterrupt:
         s.close()
     finally:
diff --git a/reverseshell_listener.py b/reverseshell_listener.py
index 23eb39a..b391282 100644
--- a/reverseshell_listener.py
+++ b/reverseshell_listener.py
@@ -14,7 +14,7 @@ def chat_server(port):
             print(data.decode(),end="")
             command = input()
             command += "\n"
-            conn.send(command.encode())
+            conn.send(command.encode().strip())
             time.sleep(0.1)
     except KeyboardInterrupt:
         s.close()