feat: ajouter le forwarding dynamique SOCKS (-D)
Les redirections -L et -R existaient deja ; il manquait -D, ou le port local
ouvert se comporte comme un proxy SOCKS dont la destination est negociee par
chaque client (navigateur, curl...) plutot que fixee a l'avance.
- app/tunnels.py : implementation minimale d'un serveur SOCKS4/4a/5 (CONNECT
uniquement, sans authentification - largement suffisant pour un usage
local/personnel). Reutilise le meme _ForwardServer et le meme _relay() que
-L/-R ; seul le handshake initial differe (negocie la destination au lieu
de la recevoir en config). DIRECTION_DYNAMIC s'integre dans describe() et
Tunnel.stop() sans code specifique supplementaire.
- TunnelDialog : nouvelle option "Dynamique - SOCKS (-D)" ; les champs
Machine/Port de destination se masquent automatiquement pour ce mode
(QFormLayout.setRowVisible), puisque la destination n'est pas fixee ici.
tests/test_socks_tunnel.py : verifie SOCKS5 (adresse IPv4) et SOCKS4a (nom de
domaine) avec un vrai client SOCKS et un transport SSH mocke - a d'ailleurs
revele un bug reel (decode("idna", errors="replace") leve une exception, le
codec idna ne supportant pas errors="replace") sur le chemin nom-de-domaine,
corrige ici (decode("ascii", errors="replace"), les noms d'hote SOCKS etant
de l'ASCII simple).
Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
121
tests/test_socks_tunnel.py
Normal file
121
tests/test_socks_tunnel.py
Normal file
@@ -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")
|
||||
Reference in New Issue
Block a user