diff --git a/app/dialogs/tunnel_dialog.py b/app/dialogs/tunnel_dialog.py index 766e8f1..e816555 100644 --- a/app/dialogs/tunnel_dialog.py +++ b/app/dialogs/tunnel_dialog.py @@ -7,9 +7,7 @@ from PySide6.QtWidgets import ( QPushButton, QComboBox, QSpinBox, QLineEdit, QLabel, QMessageBox, QAbstractItemView, ) -from app.tunnels import ( - DIRECTION_LOCAL, DIRECTION_REMOTE, DIRECTION_LABELS, TunnelManager, describe, -) +from app.tunnels import DIRECTION_DYNAMIC, DIRECTION_LABELS, TunnelManager, describe class TunnelDialog(QDialog): @@ -26,7 +24,10 @@ class TunnelDialog(QDialog): "la connexion SSH, vers une machine joignable depuis le serveur — pratique pour " "atteindre un service qui n'ecoute que sur la boucle locale du serveur.
" "Une redirection distante fait l'inverse : un port ouvert sur le serveur est " - "relaye vers une machine joignable depuis votre poste." + "relaye vers une machine joignable depuis votre poste.
" + "Une redirection dynamique ouvre un proxy SOCKS sur votre poste : chaque " + "application qui s'y connecte choisit elle-meme sa destination, atteinte a travers " + "la connexion SSH (pratique pour naviguer comme si vous etiez sur le reseau du serveur)." ) info.setWordWrap(True) layout.addWidget(info) @@ -52,6 +53,10 @@ class TunnelDialog(QDialog): form.addRow("Machine de destination :", self.dest_host) form.addRow("Port de destination :", self.dest_port) layout.addLayout(form) + self.form = form + + self.direction.currentIndexChanged.connect(self._update_dest_fields) + self._update_dest_fields() buttons = QHBoxLayout() self.add_btn = QPushButton("Ouvrir le tunnel") @@ -69,6 +74,13 @@ class TunnelDialog(QDialog): self.remove_btn.clicked.connect(self.remove_tunnel) self.refresh() + def _update_dest_fields(self) -> None: + """La destination d'une redirection dynamique (-D) est negociee par + chaque client SOCKS : les champs n'ont pas de sens dans ce mode.""" + has_dest = self.direction.currentData() != DIRECTION_DYNAMIC + self.form.setRowVisible(self.dest_host, has_dest) + self.form.setRowVisible(self.dest_port, has_dest) + def refresh(self) -> None: self.tree.clear() for tunnel in self.manager.tunnels: @@ -83,15 +95,14 @@ class TunnelDialog(QDialog): self.remove_btn.setEnabled(bool(self.manager.tunnels)) def add_tunnel(self) -> None: - if not self.dest_host.text().strip(): + direction = self.direction.currentData() + if direction != DIRECTION_DYNAMIC and not self.dest_host.text().strip(): QMessageBox.warning(self, "Destination manquante", "Indiquez la machine de destination.") return - config = { - "direction": self.direction.currentData(), - "listen_port": self.listen_port.value(), - "dest_host": self.dest_host.text().strip(), - "dest_port": self.dest_port.value(), - } + config = {"direction": direction, "listen_port": self.listen_port.value()} + if direction != DIRECTION_DYNAMIC: + config["dest_host"] = self.dest_host.text().strip() + config["dest_port"] = self.dest_port.value() tunnel = self.manager.start(config) if tunnel.error: QMessageBox.critical( diff --git a/app/tunnels.py b/app/tunnels.py index dbeefa4..7a34ce6 100644 --- a/app/tunnels.py +++ b/app/tunnels.py @@ -1,4 +1,4 @@ -"""Tunnels SSH (redirection de ports), facon `ssh -L` et `ssh -R`. +"""Tunnels SSH (redirection de ports), facon `ssh -L`, `ssh -R` et `ssh -D`. Redirection locale (-L) : un port ouvert sur votre poste est relaye, a travers la connexion SSH, vers un hote joignable depuis le serveur. Typiquement pour atteindre @@ -7,6 +7,10 @@ l'interface web d'un service qui n'ecoute que sur la boucle locale du serveur. Redirection distante (-R) : l'inverse, un port ouvert sur le serveur est relaye vers une machine joignable depuis votre poste. +Redirection dynamique (-D) : un port ouvert sur votre poste se comporte comme un +proxy SOCKS ; la destination n'est pas fixee a l'avance mais negociee par chaque +client SOCKS (navigateur, curl...), puis atteinte a travers la connexion SSH. + Chaque tunnel tourne dans son propre thread : un relais bloque ne doit gener ni l'interface ni les autres tunnels. """ @@ -21,10 +25,12 @@ import paramiko DIRECTION_LOCAL = "local" DIRECTION_REMOTE = "remote" +DIRECTION_DYNAMIC = "dynamic" DIRECTION_LABELS = { DIRECTION_LOCAL: "Locale (-L)", DIRECTION_REMOTE: "Distante (-R)", + DIRECTION_DYNAMIC: "Dynamique - SOCKS (-D)", } @@ -33,6 +39,8 @@ def describe(tunnel: dict) -> str: if tunnel.get("direction") == DIRECTION_REMOTE: return (f"serveur:{tunnel['listen_port']} -> " f"{tunnel['dest_host']}:{tunnel['dest_port']} (via ce poste)") + if tunnel.get("direction") == DIRECTION_DYNAMIC: + return f"SOCKS sur 127.0.0.1:{tunnel['listen_port']} (destination negociee par le client)" return (f"localhost:{tunnel['listen_port']} -> " f"{tunnel['dest_host']}:{tunnel['dest_port']} (via le serveur)") @@ -84,6 +92,110 @@ class _ForwardHandler(socketserver.BaseRequestHandler): _relay(self.request, chan) +# ---------------------------------------------------------------------------- +# Redirection dynamique (-D) : proxy SOCKS local +# ---------------------------------------------------------------------------- +def _recv_exact(sock: socket.socket, n: int) -> bytes: + """recv() ne garantit pas de renvoyer les n octets demandes en un seul appel.""" + data = b"" + while len(data) < n: + chunk = sock.recv(n - len(data)) + if not chunk: + raise ConnectionError("Connexion fermee pendant la negociation SOCKS") + data += chunk + return data + + +def _read_c_string(sock: socket.socket) -> bytes: + out = bytearray() + while True: + b = _recv_exact(sock, 1) + if b == b"\x00": + return bytes(out) + out += b + + +def _socks5_handshake(sock: socket.socket) -> tuple[str, int] | None: + nmethods = _recv_exact(sock, 1)[0] + methods = _recv_exact(sock, nmethods) if nmethods else b"" + if 0x00 not in methods: + sock.sendall(b"\x05\xff") # aucune methode acceptable : on ne gere que "sans authentification" + return None + sock.sendall(b"\x05\x00") + + _, cmd, _, atyp = _recv_exact(sock, 4) + if cmd != 0x01: # seul CONNECT est pris en charge (pas BIND / UDP ASSOCIATE) + sock.sendall(b"\x05\x07\x00\x01" + b"\x00" * 6) + return None + + if atyp == 0x01: + host = socket.inet_ntoa(_recv_exact(sock, 4)) + elif atyp == 0x03: + length = _recv_exact(sock, 1)[0] + host = _recv_exact(sock, length).decode("ascii", errors="replace") + elif atyp == 0x04: + host = socket.inet_ntop(socket.AF_INET6, _recv_exact(sock, 16)) + else: + sock.sendall(b"\x05\x08\x00\x01" + b"\x00" * 6) # type d'adresse non supporte + return None + port = int.from_bytes(_recv_exact(sock, 2), "big") + + # Adresse/port de retour a 0 : sans interet pour un CONNECT, la plupart des + # clients ne les utilisent pas au-dela de verifier le code de succes. + sock.sendall(b"\x05\x00\x00\x01" + b"\x00\x00\x00\x00\x00\x00") + return host, port + + +def _socks4_handshake(sock: socket.socket) -> tuple[str, int] | None: + cmd = _recv_exact(sock, 1)[0] + port = int.from_bytes(_recv_exact(sock, 2), "big") + ip_bytes = _recv_exact(sock, 4) + _read_c_string(sock) # identifiant utilisateur, ignore + if cmd != 0x01: + sock.sendall(b"\x00\x5b" + b"\x00" * 6) + return None + # SOCKS4a : une IP de la forme 0.0.0.x (x != 0) signale un nom de domaine a + # lire ensuite plutot qu'une adresse IP directement utilisable. + if ip_bytes[:3] == b"\x00\x00\x00" and ip_bytes[3] != 0: + host = _read_c_string(sock).decode("ascii", errors="replace") + else: + host = socket.inet_ntoa(ip_bytes) + sock.sendall(b"\x00\x5a" + port.to_bytes(2, "big") + ip_bytes) + return host, port + + +def _socks_handshake(sock: socket.socket) -> tuple[str, int] | None: + """Negocie une requete SOCKS4/4a ou SOCKS5 (CONNECT uniquement, sans + authentification) et renvoie (hote, port) demande par le client, ou None si + la negociation echoue ou porte sur une commande non prise en charge.""" + version = _recv_exact(sock, 1) + if version == b"\x05": + return _socks5_handshake(sock) + if version == b"\x04": + return _socks4_handshake(sock) + return None + + +class _SocksHandler(socketserver.BaseRequestHandler): + transport: paramiko.Transport + + def handle(self): + sock = self.request + try: + dest = _socks_handshake(sock) + except (OSError, ConnectionError, IndexError): + return + if dest is None: + return + try: + chan = self.transport.open_channel("direct-tcpip", dest, sock.getpeername()) + except Exception: + return + if chan is None: + return + _relay(sock, chan) + + class Tunnel: """Un tunnel actif, avec de quoi l'arreter.""" @@ -103,9 +215,20 @@ class Tunnel: self._transport = transport if self.direction == DIRECTION_REMOTE: self._start_remote(transport) + elif self.direction == DIRECTION_DYNAMIC: + self._start_dynamic(transport) else: self._start_local(transport) + # ---------- redirection dynamique (-D) : proxy SOCKS ---------- + def _start_dynamic(self, transport: paramiko.Transport) -> None: + config = self.config + handler = type("SocksHandler", (_SocksHandler,), {"transport": transport}) + bind = config.get("listen_host") or "127.0.0.1" + self._server = _ForwardServer((bind, int(config["listen_port"])), handler) + self._thread = threading.Thread(target=self._server.serve_forever, daemon=True) + self._thread.start() + # ---------- redirection locale (-L) ---------- def _start_local(self, transport: paramiko.Transport) -> None: config = self.config diff --git a/tests/test_socks_tunnel.py b/tests/test_socks_tunnel.py new file mode 100644 index 0000000..75dbafe --- /dev/null +++ b/tests/test_socks_tunnel.py @@ -0,0 +1,121 @@ +"""Tunnel dynamique (-D) : verifie que le proxy SOCKS5 / SOCKS4a negocie +correctement la destination demandee par le client puis relaie le trafic a +travers un (faux) transport SSH -- sans connexion SSH reelle, comme le fait +deja test_tunnels.py pour la redirection locale.""" +import sys, pathlib +sys.path.insert(0, str(pathlib.Path(__file__).resolve().parent.parent)) + +import socket +import threading +import time + +from app.tunnels import TunnelManager, DIRECTION_DYNAMIC + + +class EchoServer(threading.Thread): + """Serveur de destination : renvoie ce qu'il recoit, prefixe.""" + + def __init__(self): + super().__init__(daemon=True) + self.sock = socket.socket() + self.sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) + self.sock.bind(("127.0.0.1", 0)) + self.sock.listen(5) + self.port = self.sock.getsockname()[1] + + def run(self): + while True: + try: + conn, _ = self.sock.accept() + except OSError: + return + data = conn.recv(1024) + conn.sendall(b"echo:" + data) + conn.close() + + +class FakeTransport: + """Imite paramiko.Transport : ouvre un vrai socket vers la destination negociee.""" + + def open_channel(self, kind, dest, src): + assert kind == "direct-tcpip" + return socket.create_connection(dest, timeout=5) + + +def socks5_connect(proxy_port: int, host: str, port: int) -> socket.socket: + sock = socket.create_connection(("127.0.0.1", proxy_port), timeout=5) + sock.sendall(b"\x05\x01\x00") # version 5, 1 methode proposee : "sans authentification" + assert sock.recv(2) == b"\x05\x00", "methode 'sans authentification' attendue" + sock.sendall(b"\x05\x01\x00\x01" + socket.inet_aton(host) + port.to_bytes(2, "big")) + reply = sock.recv(10) + assert reply[0:2] == b"\x05\x00", f"CONNECT refuse : {reply}" + return sock + + +def socks4a_connect(proxy_port: int, host: str, port: int) -> socket.socket: + sock = socket.create_connection(("127.0.0.1", proxy_port), timeout=5) + # IP 0.0.0.1 = extension SOCKS4a : le nom suit l'identifiant utilisateur + sock.sendall( + b"\x04\x01" + port.to_bytes(2, "big") + b"\x00\x00\x00\x01" + + b"user\x00" + host.encode("ascii") + b"\x00" + ) + reply = sock.recv(8) + assert reply[0:2] == b"\x00\x5a", f"CONNECT refuse : {reply}" + return sock + + +echo = EchoServer() +echo.start() +print(f"serveur de destination (echo) sur le port {echo.port}") + +manager = TunnelManager(FakeTransport()) +tunnel = manager.start({"direction": DIRECTION_DYNAMIC, "listen_port": 0}) +assert tunnel.error is None, tunnel.error +proxy_port = tunnel._server.server_address[1] +print("proxy SOCKS ouvert sur le port", proxy_port) +time.sleep(0.2) + +print() +print("=== 1. SOCKS5, adresse IPv4 ===") +client = socks5_connect(proxy_port, "127.0.0.1", echo.port) +client.sendall(b"bonjour-socks5") +reponse = client.recv(1024) +client.close() +print(" recu:", reponse) +assert reponse == b"echo:bonjour-socks5", reponse + +print() +print("=== 2. SOCKS4a, nom de domaine ===") +client = socks4a_connect(proxy_port, "localhost", echo.port) +client.sendall(b"bonjour-socks4a") +reponse = client.recv(1024) +client.close() +print(" recu:", reponse) +assert reponse == b"echo:bonjour-socks4a", reponse + +print() +print("=== 3. SOCKS5 : commande non supportee (BIND) rejetee proprement ===") +sock = socket.create_connection(("127.0.0.1", proxy_port), timeout=5) +sock.sendall(b"\x05\x01\x00") +assert sock.recv(2) == b"\x05\x00" +sock.sendall(b"\x05\x02\x00\x01" + socket.inet_aton("127.0.0.1") + (1234).to_bytes(2, "big")) # BIND +reply = sock.recv(10) +print(" reponse:", reply) +assert reply[1] == 0x07, reply # 0x07 = commande non supportee +sock.close() +print(" OK") + +print() +print("=== 4. Fermeture du tunnel : le port SOCKS n'accepte plus de connexion ===") +manager.stop_all() +time.sleep(0.3) +try: + socket.create_connection(("127.0.0.1", proxy_port), timeout=1).close() + ferme = False +except OSError: + ferme = True +assert ferme, "le port doit etre libere apres fermeture" +print(" OK") + +print() +print("SOCKS DYNAMIQUE (-D) : TOUS LES TESTS OK")