Current File : //usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyc
�
+�{Wc @ s@ d Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
m
Z
m Z d d l m
Z
y? e j � �- e j d e � d d l Z e j j Z Wd QXWn- e k
r� d Z d e f d � � YZ n Xd d
l m Z y d d l Z Wn e k
r8d Z n Xd d l m Z e � Z d
� Z e j e d � e j � Z! d d � Z" d � Z# d � Z$ d d d d d � Z% d d d d d � Z& d d d d d � Z' d � Z( d � Z) d d d d d � Z* d d d � Z+ e j, d k r!e* Z- n e' Z- d d d d d g Z. d S(! sX Basic ssh tunnel utilities, and convenience functions for tunneling
zeromq connections.
i����( t print_functionN( t getpasst getuser( t Processt ignoret SSHExceptionc B s e Z RS( ( t __name__t
__module__( ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyR s i ( t forward_tunneli ( t bc C s� g } xz t | � D]l } t j � } | j d � x= | j � d t k rq | j � t j � } | j d � q5 W| j | � q WxK t | � D]= \ } } | j � d } | j � | | | <t j | � q� W| S( s5 Selects and return n random ports that are available.t i i ( R
i ( R
i ( t ranget sockett bindt getsocknamet
_random_portst closet appendt enumeratet add( t nt portst it sockt port( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyt select_random_ports- s
s pass(word|phrase):c C s@ | d k r t j d k } n | s- t } n t } | | | � S( s� Attempt to make an ssh connection without a password.
This is mainly used for requiring password input only once
when many tunnels may be connected to the same server.
If paramiko is None, the default for the platform is chosen.
t win32N( t Nonet syst platformt _try_passwordless_openssht _try_passwordless_paramiko( t servert keyfilet paramikot f( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyt try_passwordless_sshE s c C s� t d k r t d � � n d | } | r< | d | 7} n | d 7} t j j � } | j d d � d } t j | d | �} xv t r� y: | j | t
g d d �} | d
k r� t d � � n Wn+ t j k
r� q� q� t j
k
r� t SXt Sq� Wd S(
s. Try passwordless login with shell ssh command.s! pexpect unavailable, use paramikos ssh -f s -i s exitt SSH_ASKPASSs, Are you sure you want to continue connectingt envt timeoutg�������?i s2 The authenticity of the host can't be established.N( t pexpectR t ImportErrort ost environt copyt popt spawnt Truet expectt
_password_patR t TIMEOUTt EOFt False( R R! t cmdR& t
ssh_newkeyt pR ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyR T s(
c C s� t d k rG d } t j d k r. | d 7} n
| d 7} t | � � n t | � \ } } } t j � } | j � | j t j � � y& | j
| | d | d | d t �Wn t j k
r� t
SX| j � t Sd S( s% Try passwordless login with paramiko.s Paramiko unavaliable, R s= Paramiko is required for ssh tunneled connections on Windows.s use OpenSSH.t usernamet key_filenamet
look_for_keysN( R" R R R R) t
_split_servert SSHClientt load_system_host_keyst set_missing_host_key_policyt
WarningPolicyt connectR/ t AuthenticationExceptionR4 R ( R R! t msgR8 R t client( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyR o s"
i< c C s> t | | d | d | d | d | �\ } } | j | � | S( s\ Connect a socket to an address via an ssh tunnel.
This is a wrapper for socket.connect(addr), when addr is not accessible
from the local machine. It simply creates an ssh tunnel using the remaining args,
and calls socket.connect('tcp://localhost:lport') where lport is the randomly
selected local port of the tunnel.
R! t passwordR" R' ( t open_tunnelR@ ( R t addrR R! RD R" R' t new_urlt tunnel( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyt tunnel_connection� s -
c C s� t d � d } | j d � \ } } | j d � \ } } t | � } | d k rd t j d k } n | rs t }
n t }
|
| | | d | d | d | d | �} d
| | f S( s� Open a tunneled connection from a 0MQ url.
For use inside tunnel_connection.
Returns
-------
(url, tunnel) : (str, object)
The 0MQ url that has been forwarded, and the tunnel object
i i s ://t :R t remoteipR! RD R' s tcp://127.0.0.1:%iN( R t splitt intR R R t paramiko_tunnelt openssh_tunnel( RF R R! RD R" R' t lportt transportt ipt rportt tunnelfRH ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyRE � s *s 127.0.0.1c C s� t d k r t d � � n d } | r8 | d | 7} n d | k rj | j d � \ } } | d | 7} n d | | f } t j | d t �\ }
} | s$t |
|
j d � d |
j d
� !� } d | | | | | f } t j | d t �\ }
} | s$t j t
| j d d
d � � | Sn d | | | | | | f } t j
j � }
|
j d d � d } t j | d |
�} t } xt r�y: | j | t g d d �} | d k r�t d � � n Wnt t j k
r�q�q�t j k
r9| j r/t | j � t | j � t | j � t d | � � q�| j Sq�X| rSt d � d } n | d k rrt d | � } n | j | � t } q�Wd S( s4 Create an ssh tunnel using command-line ssh that connects port lport
on this machine to localhost:rport on server. The tunnel
will automatically close when not in use, remaining open
for a minimum of timeout seconds for an initial connection.
This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
as seen from `server`.
keyfile and password may be specified, but ssh config is checked for defaults.
Parameters
----------
lport : int
local port for connecting to the tunnel from this machine.
rport : int
port on the remote machine to connect to.
server : str
The ssh server to connect to. The full ssh server string will be parsed.
user@server:port
remoteip : str [Default: 127.0.0.1]
The remote ip, specifying the destination of the tunnel.
Default is localhost, which means that the tunnel would redirect
localhost:lport on this machine to localhost:rport on the *server*.
keyfile : str; path to public key file
This specifies a key to be used in ssh login, default None.
Regular default ssh keys will be used without specifying this argument.
password : str;
Your ssh password to the ssh server. Note that if this is left None,
you will be prompted for it if passwordless key based login is unavailable.
timeout : int [default: 60]
The time (in seconds) after which no activity will result in the tunnel
closing. This prevents orphaned tunnels from running forever.
s( pexpect unavailable, use paramiko_tunnels ssh s -i RJ s -p %ss %s -O check %st withexitstatuss (pid=i t )s& %s -O forward -L 127.0.0.1:%i:%s:%i %ss
-O forwards -O canceli s/ %s -f -S none -L 127.0.0.1:%i:%s:%i %s sleep %iR% s, Are you sure you want to continue connectingR& R' g�������?i s2 The authenticity of the host can't be established.s tunnel '%s' failed to starts Password rejected, try agains %s's password: N( R( R R) RL t runR/ RM t findt atexitt registert _stop_tunnelt replaceR* R+ R, R- R. R4 R0 R1 R R2 R3 t
exitstatust printt beforet aftert RuntimeErrort pidR t sendline( RP RS R RK R! RD R' t sshR R5 t outputR] Rb R& R6 RH t failedR ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyRO � s\ $ )
c C s t j | � d S( N( R( RW ( R5 ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyR[ s c C ss d | k r' | j d d � \ } } n t � } d | k r` | j d � \ } } t | � } n d } | | | f S( Nt @i RJ i ( RL R RM ( R R8 R ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyR; s c C s� t d k r t d � � n | d k rL t | | � sL t d | � } qL n t d t d | | | | f d t d | d | � � } t | _ | j
� | S( s6 launch a tunner with paramiko in a subprocess. This should only be used
when shell ssh is unavailable (e.g. Windows).
This creates a tunnel redirecting `localhost:lport` to `remoteip:rport`,
as seen from `server`.
If you are familiar with ssh tunnels, this creates the tunnel:
ssh server -L localhost:lport:remoteip:rport
keyfile and password may be specified, but ssh config is checked for defaults.
Parameters
----------
lport : int
local port for connecting to the tunnel from this machine.
rport : int
port on the remote machine to connect to.
server : str
The ssh server to connect to. The full ssh server string will be parsed.
user@server:port
remoteip : str [Default: 127.0.0.1]
The remote ip, specifying the destination of the tunnel.
Default is localhost, which means that the tunnel would redirect
localhost:lport on this machine to localhost:rport on the *server*.
keyfile : str; path to public key file
This specifies a key to be used in ssh login, default None.
Regular default ssh keys will be used without specifying this argument.
password : str;
Your ssh password to the ssh server. Note that if this is left None,
you will be prompted for it if passwordless key based login is unavailable.
timeout : int [default: 60]
The time (in seconds) after which no activity will result in the tunnel
closing. This prevents orphaned tunnels from running forever.
s Paramiko not availables %s's password: t targett argst kwargsR! RD N( R" R R) R R R t _paramiko_tunnelt dictR/ t daemont start( RP RS R RK R! RD R' R7 ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyRN s (
c
C s3 t | � \ } } } t j � } | j � | j t j � � y, | j | | d | d | d t d | �Wn7 t k
r� } t d | | | f � t
j d � n Xt j t j
t j � y t | | | | j � � WnU t k
rt d � t
j d � n. t k
r.} t d | � t
j d
� n Xd S( s� Function for actually starting a paramiko tunnel, to be passed
to multiprocessing.Process(target=this), and not called directly.
R8 R9 R: RD s" *** Failed to connect to %s:%d: %ri s' SIGINT: Port forwarding stopped cleanlyi s% Port forwarding stopped uncleanly: %si� N( R; R" R< R= R> R? R@ R/ t ExceptionR^ R t exitt signalt SIGINTt SIG_IGNR t
get_transportt KeyboardInterrupt(
RP RS R RK R! RD R8 R RC t e( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyRk N s&
R RI t
ssh_tunnelRO RN R$ (/ t __doc__t
__future__R RY R* t reRq R R t warningsR R t multiprocessingR t catch_warningst simplefiltert DeprecationWarningR" t
ssh_exceptionR R) R Ro t forwardR R( t utils.strtypesR t setR R t compilet
IGNORECASER1 R$ R R RI RE RO R[ R; RN Rk R Rw t __all__( ( ( s4 /usr/lib64/python2.7/site-packages/zmq/ssh/tunnel.pyt <module> sR
Z 7"
Mini Shell