Update Python Ofensivo - Más ejercicios con conexiones socket

python-ofensivo/07_conexiones_red_protocolos/04_socket/client.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+
+import socket
+
+
+def start_client():
+
+    host = 'localhost'
+    port = 1234
+
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        s.connect((host, port))
+
+        while True:
+            msg = input("[+] Introduce un mensaje: ")
+
+            s.sendall(msg.encode('utf-8'))
+
+            if msg.strip() == 'bye':
+                break
+
+            data = s.recv(1024)
+
+            print(f"\n[+] Mensaje del servidor: {data.decode().strip()}")
+
+
+start_client()

python-ofensivo/07_conexiones_red_protocolos/04_socket/server.py

@@ -0,0 +1,58 @@
+#!/usr/bin/env python3
+
+import socket
+import threading
+import pdb
+
+
+class ClientThread(threading.Thread):
+
+    def __init__(self, conn, addr):
+        super().__init__()
+        self.conn = conn
+        self.addr = addr
+
+        print(f"\nConexión establecida desde {self.addr}")
+
+    def run(self):
+        msg = ''
+
+        while True:
+
+            data = self.conn.recv(1024)
+            msg = data.decode()
+
+            # pdb.set_trace() # Breakpoint
+
+            if msg.strip() == 'bye':
+                break
+
+            print(f"\n[+] Recibido del cliente {self.addr}: {msg.strip()}")
+            self.conn.send(data)
+
+        print(f"\n[-] Cliente {self.addr} desconectado")
+        self.conn.close()
+
+
+HOST = 'localhost'
+PORT = 1234
+
+with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+
+    # Alterar propiedades del socket
+    # Reutilizar direcciones
+    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    # Otras propiedades: https://docs.python.org/3/library/socket.html#socket.socket.setsockopt
+    # socket.IPPROTO_IP, socket.IPPROTO_IPV6, socket.IPPROTO_TCP, socket.IPPROTO_UDP...
+
+    # Enlazar socket a dirección y puerto
+    s.bind((HOST, PORT))
+
+    print(f"Servidor TCP escuchando en {HOST}:{PORT}")
+
+    while True:
+
+        s.listen()
+        conn, addr = s.accept()
+        new_thread = ClientThread(conn, addr)
+        new_thread.start()

python-ofensivo/07_conexiones_red_protocolos/05_socket/client.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+
+import socket
+
+
+def start_chat_client():
+
+    host = 'localhost'
+    port = 1234
+
+    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    client_socket.connect((host, port))
+
+    while True:
+
+        client_msg = input(f"\n[+] Mensaje para enviar al servidor: ")
+        client_socket.send(client_msg.encode())
+
+        if client_msg == "bye":
+            break
+
+        server_msg = client_socket.recv(1024).strip().decode()
+        print(f"\n[+] Mensaje del servidor: {server_msg}")
+
+    client_socket.close()
+
+
+start_chat_client()

python-ofensivo/07_conexiones_red_protocolos/05_socket/server.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python3
+
+import socket
+
+
+def start_chat_server():
+    host = 'localhost'
+    port = 1234
+
+    server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+    # reuse socket TIME_WAIT
+    server_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    server_socket.bind((host, port))
+    server_socket.listen(1)
+
+    print(f"\n[+] Servidor listo para aceptar una conexión en {host}:{port}")
+
+    conn, addr = server_socket.accept()
+
+    print(f"\n[+] Conexión establecida con {addr[0]}:{addr[1]}")
+
+    while True:
+
+        client_msg = conn.recv(1024).strip().decode()
+        print(f"\n[+] Mensaje del cliente: {client_msg}")
+
+        if client_msg == "bye":
+            break
+
+        server_msg = input(f"\n[+] Mensaje para el cliente: ")
+        conn.send(server_msg.encode())
+
+    print(f"\n[+] Conexión finalizada con {addr[0]}:{addr[1]}")
+    conn.close()
+
+
+start_chat_server()