Update Python Ofensivo - Más ejercicios con conexiones socket

This commit is contained in:
Manuel Vergara 2024-01-11 23:08:41 +01:00
parent f09fa0cbf5
commit 8109c2fe30
4 changed files with 150 additions and 0 deletions

View File

@ -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()

View File

@ -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()

View File

@ -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()

View File

@ -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()