a ƹDf@sddZddgZddlZddlZddlZddlZddlmZGdddeZ Gddde Z e dkr`dS) a This module contains the base class for secure tunnel connections, and describes the pathos tunnel interface. See the following for an example. Usage ===== A typical call to a pathos 'tunnel' will roughly follow this example: >>> # instantiate the tunnel, providing it with a unique identifier >>> tunnel = Tunnel('tunnel') >>> >>> # establish a tunnel to the remote host and port >>> remotehost = 'remote.host.edu' >>> remoteport = 12345 >>> localport = tunnel.connect(remotehost, remoteport) >>> print("Tunnel connected at local port: %s" % tunnel._lport) >>> >>> # pause script execution to maintain the tunnel (i.e. do something) >>> sys.stdin.readline() >>> >>> # tear-down the tunneled connection >>> tunnel.disconnect() TunnelTunnelExceptionN)Pipec@seZdZdZdS)rz-Exception for failure to establish ssh tunnelN)__name__ __module__ __qualname____doc__r r a/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/pathos/secure/tunnel.pyr,sc@sReZdZdZdZdZdZdddZdd Zd d Z dd d Z ddZ dddZ dS)rz=a ssh-tunnel launcher for parallel and distributed computing.iiTNc Csddlm}|dur4ddlm}|r,||n||}||j|j}|}|dkrXtdz|j||||dWnXty}z@|jddkr| WYd}~qBn| tdWYd}~n d}~00d |_ |S) aestablish a secure shell tunnel between local and remote host Input: host -- remote hostname [user@host:path is also valid] port -- remote port number Additional Input: through -- 'tunnel-through' hostname [default = None] r) portnumberN) randomportzNo available local port)throughbindzConnection failedT) Zpathos.portpickerr Z pathos.corer MINPORTMAXPORTr_connectargs disconnect_Tunnel__disconnect connected) selfhostportr r r Zpick localporter r r connect7s$  zTunnel.connectcCsH|jdkrD|jrtd|jt|jtjt|jd|dS)zdestroy the ssh tunnelrzKill ssh pid=%dN) _pidverboseprintoskillsignalSIGTERMwaitpidrrr r r r[s  zTunnel.disconnectcCs"d|_d|_d|_d|_d|_dS)zdisconnect tunnel internalsrFN)rr_lport_rport_hostr$r r r Z __disconnectes zTunnel.__disconnectc sZtj|dur*dfddtdDn||_td|_||rV|jfi|dS)zocreate a ssh tunnel launcher Inputs: name -- a unique identifier (string) for the launcher Nc3s|]}tVqdS)N)randomchoice).0iZxyzr r uz"Tunnel.__init__..launcher) string ascii_lettersjoinrangenamer _launcherrr)rr6kwdsr r-r __init__ns zTunnel.__init__cCsP|jsd|jSz"|jjdddd}Wn|jj}Yn0d|S)Nz Tunnel('%s') ")rr6r7messagesplitrstrip)rmsgr r r __repr__}s  "zTunnel.__repr__c Csd|||f}d}|r|}n|}|j|||dd|j||_||_||_|j|_|j}|r|dr|t dnt |t ddS)Nz-q -N -L %d:%s:%dr(T)rcommandoptionsZ backgroundrZfailure) r7Zlaunchr%r&r'pidrresponse startswithrr) rrZ remotehostZ remoteportr rDrCZrhostliner r r rs&      zTunnel._connect)NN)N)N) rrrrrrrrrrr9rBrr r r r r0s $  __main__) r__all__rr!r)r2Z pathos.securer Exceptionrobjectrrr r r r  s k