File: //lib/python3.6/site-packages/sshuttle/__pycache__/client.cpython-36.pyc
3
� L_�d � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl j
Z
d dl j Z d dl
jZd dljZd dljZd dl mZmZmZmZmZ d dl mZmZmZmZm Z m Z m Z d dl m Z m!Z! yd dl"m#Z# W n e$k
r� dZ#Y nX y d dl%Z&e'e&j%d� e&Z%W nR e(k
�rZ y d dl)Z)e'e)j%d� e)Z%W n e$k
�rT d dl%Z%Y nX Y nX ej*ej+ej,�a-dd� Z.da/d d
� Z0d d � Z1d
d� Z2G dd� d�Z3G dd� d�Z4i Z5i Z6dd� Z7dd� Z8dd� Z9dd� Z:dd � Z;d d � Z<d d � Z=d!d"� Z>dS )#� N)� SockWrapper�Handler�Proxy�Mux�
MuxWrapper)�log�debug1�debug2�debug3�Fatal�islocal�resolvconf_nameservers)�
get_method�Features)�getpwnamZrecvmsgc C s t d| � tjd� d S )Nzexiting on signal %d
� )r �sys�exit)Zsignum�frame� r � /usr/lib/python3.6/client.py�
got_signal) s r c C s t jj| �aytt�jd�}W nB tk
r` } z&|jtjkr@d S t dt|f ��W Y d d }~X nX |stt j
t� d S t |j � p�d�}|dkr�t j
t� d S yt j
|d� W nL tk
r� } z0|jtjkr�t j
t� d S |jtjkr�n� W Y d d }~X nX t dt|f ��d S )Ni zcan't read %s: %sr z(%s: sshuttle is already running (pid=%d))�os�path�abspath�_pidname�open�read�IOError�errno�ENOENTr �unlink�int�strip�kill�OSErrorZESRCHZEPERM)�pidfileZoldpid�er r r � check_daemon1 s2 "
r( c C s� t j� rt jd� t j� t j� r,t jd� t jtt jt jB t jB d�} zt j | dt j
� � W d t j | � X t j d� t
j
t
jt� tt jd�}t j|j� d� t j|j� d� |j � d S )Nr i� s %d
�/zr+r )r �fork�_exit�setsidr r �O_WRONLY�O_CREAT�O_EXCL�write�getpid�close�chdir�signal�SIGTERMr �devnull�dup2�fileno)ZoutfdZsir r r � daemonizeP s
r9 c C sF yt jt� W n2 tk
r@ } z| jtjkr.n� W Y d d } ~ X nX d S )N)r r! r r% r r )r' r r r �daemon_cleanuph s r: c @ sD e Zd Zejdfdd�Zdd� Zdd� Zdd � Zd
d � Z d d
� Z
dS )�
MultiListenerr c C s" || _ || _d | _d | _d| _d S )NF)�type�proto�v6�v4� bind_called)�selfZkindr= r r r �__init__t s
zMultiListener.__init__c C s: | j s
t�| jr | jj|||� | jr6| jj|||� d S )N)r@ �AssertionErrorr> �
setsockoptr? )rA �levelZoptname�valuer r r rD { s
zMultiListener.setsockoptc sT | j s
t�g }| jr |j| j� | jr2|j| j� �jt|� ���fdd��� d S )Nc s � | ����S )Nr )�sock)�callback�handlers�method�muxr r �<lambda>� s z+MultiListener.add_handler.<locals>.<lambda>)r@ rC r> �appendr? r )rA rI rH rJ rK Zsocksr )rH rI rJ rK r � add_handler� s
zMultiListener.add_handlerc C sz | j s
t�| jr | jj|� | jrvy| jj|� W nB tjk
rt } z$|jtjkr`| jr`d | _n|�W Y d d }~X nX d S )N) r@ rC r> �listenr? �socket�errorr �
EADDRINUSE)rA Zbacklogr' r r r rO � s
zMultiListener.listenc C sz | j s t�d| _ |d k r>tjtj| j| j�| _| jj|� nd | _|d k rptjtj| j| j�| _ | j j|� nd | _ d S )NT)
r@ rC rP �AF_INET6r< r= r> �bind�AF_INETr? )rA Z
address_v6Z
address_v4r r r rT � s zMultiListener.bindc C sr | j s
t�| jr<| jj� }td||f � td|| jf � | jrn| jj� }td||f � td|| jf � d S )Nz%s listening on %r.
z%s listening with %r.
)r@ rC r> � getsocknamer r r? )rA ZwhatZlistenipr r r �print_listening� s
z MultiListener.print_listeningN) �__name__�
__module__� __qualname__rP Z SOCK_STREAMrB rD rN rO rT rW r r r r r; r s r; c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�FirewallClientc s� g | _ tjjtjjt��}tjtjd gdgtj p4d d|g dg }t
j rX|dg7 }t j � j
d�rndg}n
dd d
g}|r�|d d | g7 }|| |g}tj� \}� � fd
d�}d } tj� dkr�|dd � }xX|D ]P}
y4|
d dkr�tjjd� tj|
||d�| _d } P W q� tk
�r Y q�X q�W |
| _|j� � jd�| _| �r\td| j � t| ��| jj � } | j � | dd� dk�r�td| j| f ��| dd� }t |j d��| _ | j j!| � d S ) Nr z-vz--methodz
--firewallz--syslogZOpenBSDZdoasZsudoz-pz[local sudo] Password: z /usr/bin/envz
PYTHONPATH=%sc s � j � d S )N)r2 r )�s2r r �setup� s z&FirewallClient.__init__.<locals>.setupr Zsuz [local su] )�stdoutZ
preexec_fnZrwbz Spawning firewall manager: %r
� s READYz%r expected READY, got %r� �ASCII���rb )"� auto_netsr r �dirname�__file__r �
executable�argv�helpers�verbose�ssyslog�_p�platform�
startswithrP Z
socketpair�getuid�stderrr0 � ssubprocess�Popen�pr% r2 �makefile�pfiler r �readline�checkr �decoderJ Z set_firewall) rA � method_name�sudo_pythonpathZ python_pathZargvbaseZ elev_prefixZ
argv_tries�s1r] r' rg �liner )r\ r rB � sP &
zFirewallClient.__init__c
C s: || _ || _|| _|| _|| _|| _|| _|| _| | _d S )N) �subnets_include�subnets_exclude�nslist�redirectport_v6�redirectport_v4�
dnsport_v6�
dnsport_v4�udp�user)
rA r| r} r~ r r� r� r� r� r� r r r r] � s zFirewallClient.setupc C s$ | j j� }|r td| j|f ��d S )Nz%r returned %d)rr �pollr rg )rA �rvr r r rv s
zFirewallClient.checkc C st | j jd� x>| j| j D ].\}}}}}| j jd|||jd�||f � qW x8| jD ].\}}}}}| j jd|||jd�||f � qTW | j jd� x,| jD ]"\}}| j jd||jd�f � q�W | j jd| j| j| j | j
f � d}| j r�d }| j d kr�d
}n&t
| j t��rt| j d �}n
d | j }| j jd
||f � | j j� | j j� }| j� |dk�rptd| j|f ��d S )Ns ROUTES
s %d,%d,0,%s,%d,%d
ra s %d,%d,1,%s,%d,%d
s NSLIST
s %d,%s
s PORTS %d,%d,%d,%d
r r � -zutf-8s %ds GO %d %s
s STARTED
z%r expected STARTED, got %r)rt r0 r| rc �encoder} r~ r r� r� r� r� r� �
isinstance�str�bytes�flushru rv r rg ) rA �family�ip�widthZfportZlportr� r� r{ r r r �start s@
zFirewallClient.startc C sF t jd|� st�t jd|� s$t�| jjd||f � | jj� d S )Ns [^-\w\.]s [^0-9.]s HOST %s,%s
)�re�searchrC rt r0 r� )rA Zhostnamer� r r r � sethostip- s zFirewallClient.sethostipc C s. | j j� | jj� }|r*td| j|f ��d S )Nzcleanup: %r returned %d)rt r2 rr �waitr rg )rA r� r r r �done3 s
zFirewallClient.doneN) rX rY rZ rB r] rv r� r� r� r r r r r[ � s 7
(r[ c C s� g }x:t j� D ].\}}|| k rtd| � |j|� |j|= qW x|D ]
}t |= qFW tdtt � � g }xRtj� D ]F\}\}}|| k rrtd||f � |j|tj d� |j|� |j|= qrW x|D ]
}t|= q�W tdtt� � d S )Nz expiring dnsreqs channel=%d
zRemaining DNS requests: %d
z(expiring UDP channel channel=%d peer=%r
� zRemaining UDP channels: %d
)
�dnsreqs�itemsr
rM �channels�len�
udp_by_src�send�ssnetZ
CMD_UDP_CLOSE)�nowrK �remove�chanZtimeoutZpeerr r r �expire_connections> s&
r� c
C st y| j � \}}W n~ tjk
r� } z`|jd tjtjgkr|td� tj t
� z| j � \}}|j � W d tj tj tj
�a
X d S � W Y d d }~X nX |j|�}td|d |d |d |d f � |d |j� d kr�t|d |j�r�td� |j � d S |j� }|�std� |j � d S |j|tjd|j|d jd�|d f � t||�} |jtt||�| �� t t j � |� d S ) Nr z3Rejected incoming connection: too many open files!
z Accept TCP: %s:%r -> %s:%r.
r z -- ignored: that's my address!
z8warning: too many open channels. Discarded connection.
s %d,%s,%dra ) ZacceptrP rQ �argsr ZEMFILEZENFILEr r r2 � _extra_fdr r6 �O_RDONLYZ
get_tcp_dstiprV r r� � next_channelr r� r� ZCMD_TCP_CONNECTr� r rM r r r� �time)
�listenerrJ rK rI rG �srcipr' �dstipr� Zoutwrapr r r � onaccept_tcpU s:
$
r� c C sB |j dd�\}}}|t|�f}td||f � |j||||� d S )N� ,� zdoing send from %r to %r
)�splitr" r
�send_udp)r� �datarJ rG r� �srcZsrcportr� r r r �udp_donez s r� c
s� t j � }�j�d�}|d kr d S |\�}}td�|f � �tkrPt� \� }n6|j� � � ���fdd�|j� < |j� tjd�j � � |d ft�< d|d j
d �|d
f } |j� tj | | � t ||� d S ) Ni zAccept UDP: %r -> %r.
c s t � |���d�S )N)r� )r� )�cmdr� )r� r� rJ r� r r rL � s z onaccept_udp.<locals>.<lambda>s %d� s %s,%d,r ra r )
r� �recv_udpr r� r� r� r� r� Z CMD_UDP_OPENr� r� Z CMD_UDP_DATAr� )
r� rJ rK rI r� �tr� r� �_Zhdrr )r� r� rJ r� r � onaccept_udp� s
r� c C s4 t d| ||f � |j| = t| = |j||||� d S )Nz#dns_done: channel=%d src=%r dst=%r
)r
r� r� r� )r� r� rJ rG r� r� rK r r r �dns_done� s r� c s� t j � }�j�d�}|d kr d S |\��}td��t|�f � �j� � |d t� < �j� tj|� � �����fdd��j � < t
|�� d S )Ni z$DNS request from %r to %r: %d bytes
r� c s t � |�����d�S )N)r� r� rK )r� )r� r� )r� r� r� rJ rK r� r r rL � s zondns.<locals>.<lambda>) r� r� r r� r� r� r� r� Z CMD_DNS_REQr� r� )r� rJ rK rI r� r� r� r )r� r� r� rJ rK r� r �ondns� s
r� c
" s� t dtj� � �j}
g }tjdkr,dt_ndt_t d� y2tj|||t j
oRt j
j t || | � d�d�\}}W n@ t
jk
r� } z"|jd tjkr�td ��n� W Y d d }~X nX t|jd
�|jd ���|j�� d }yVd
}x|r�|dkr�|jd�}q�W d
}x |�r|dk�r|jd�}q�W |jt|��}W nD t
jk
�rl } z$|jd tjk�rZtd��n� W Y d d }~X nX |j� }|�r�td| ��||k�r�td||f ��td� tj j � | �r�t � tdt � � ����fdd�}|�_ �fdd���fdd�}|�_!�j"|t#|
�� |�r*|j"|t$|
�� |�r@|j"|t%|
�� |d k �rrt d| � �j&dt'j(t)j*dj+|��� x:|j� }|�r�td| ��t'j,|�� |�rt�j-� �qtW d S ) Nz'Starting client with Python version %s
r zc : zclient: zconnecting to server...
)�latency_control�
auto_hosts�
to_nameserverrc )ro Zoptionsr z#failed to establish ssh session (1)�rb�wbs SSHUTTLE0001�x� z#failed to establish ssh session (2)z server died with error code %dz&expected server init string %r; got %rz Connected.
zdaemonizing (%s).
c s� � r�x�| j � jd�D ]�}|s q|jdd�\}}}t|�}t|�}|jd�}|tjkrp�jd krptd|||f � |tjkr��j d kr�td|||f � qtd|||f � �j
j |||ddf� qW d �_ �� d S )N�
r� r� ra zIgnored auto net %d/%s/%d
zAdding auto net %d/%s/%d
r )
r# r� r" rw rP rS r> r rU r? rc rM �
got_routes)Zroutestrr{ r� r� r� )rc �fwrK � serverready� tcp_listenerr r �onroutes� s
z_main.<locals>.onroutesc s" � j � tjtj� tjd�� d S )NZ Connected)r� �sdnotifyr� ZreadyZstatusr )r� r r r� s z_main.<locals>.serverreadyc sF t d| � x4| j� j� D ]$}|r|jdd�\}}� j||� qW d S )Nzgot host list: %r
r� r )r r# r� r� )Zhostlistr{ �namer� )r� r r �
onhostlist s
z_main.<locals>.onhostlistzseed_hosts: %r
�
).r rl Zpython_versionrJ rh ri Z logprefix�sshZconnectrj rk �stdin�dictrP rQ r� r ZEPIPEr r rs rM Zrecvr� Z
ECONNRESETr� r r r^ r� r9 r r� Z
got_host_listrN r� r� r� r� r� Z CMD_HOST_REQr� r� �joinZrunonceZcheck_fullness)r� � udp_listenerr� �ssh_cmd�
remotename�pythonr� � dns_listener�
seed_hostsr� rc �daemonr� rJ rI Z
serverprocZ
serversockr' Zexpected�vZ
initstringr� r� r� r )rc r� rK r� r� r �_main� s�
r� c * * s� |s t d��|rJy t|� W n, t k
rH } ztd| � dS d }~X nX td� t||�}|r�|t� 7 }|d k r�dt|dd � � }nd }| |
}dd� |D �}d d� |D �}d
d� |D �}d d� |D �}|jj� }t � }�d k� r�|j
r�d-�nd �|d k �rDt d k�rt d��yt |�j }W n" t
k
�rB t d| ��Y nX |jjdk�r�t|�dk�pf�d k |_
t|�dk�p~� d k |_n d |_
d |_|j|_t|�dk|_|d k�r�dnd|_|j
�s�g }|}|jj|� |j
�r�d k�r�t d��td|j
� td|j � td|j � td|j � � d k�r6d.� |j�rrt� fdd�|D �� �rr|
jtj� d d ddf� |j
�r�t�fd d�|D �� �r�|
jtj�d d ddf� ��rֈd �rֈ �rֈ d �r�dg} ntd d d/�} g } d } d} d} d}!td!� �x2| D �](}"td"|" � t � }#|j�r2t tj �}$nd }$��rT�d �rT�}%|%d } n ��rl�d |"f}%|"} nd }%d} � �r�� d �r�� }&|&d } n � �r�� d |"f}&|"} nd }&d} y2|#j |%|&� |$�r�|$j |%|&� d}!| j|"� P W nH tj k
�r, } z(|j t j!k�r|} | j|"� n|�W Y d d }~X nX �qW td#� |!�sP| �sLt"�| �|#j#d$� |#j$d%� |$�rt|$j$d&� d}!|j�r�td'� td d d0�} x�| D ]�}"td"|" � |"| k�r��q�t tj �}'��rڈd |"f}%|"}(nd }%d}(� �r�� d |"f}&|"})nd }&d})y |'j |%|&� d}!| j|"� P W nH tj k
�rj } z(|j t j!k�rV|} | j|"� n|�W Y d d }~X nX �q�W td#� |'j$d(� |!�s�| �s�t"�| �n d}(d})d }'|�r�|j
�s�t"�| dk�r�t d)��|�r�|j�s�t"�|j
�s�t"�|(dk�r�t d*��|�r| dk�rt d+��|�r*|)dk�r*t d,��|jj%|#� |$�rH|jj&|$� |'�rZ|jj&|'� |j'| |
|| | |(|)|j|� z t(|#|$||||||'| |
| ||�
S z(|�r�d|j)_*|j+� t,j-t,j.� � W d |�r�t/� X X d S )1Nz You must specify -r/--remote.z%s
r_ zStarting sshuttle proxy.
z%s@%sr c S s g | ]}|d t jkr|�qS )r )rP rS )�.0�ir r r �
<listcomp>I s zmain.<locals>.<listcomp>c S s g | ]}|d t jkr|�qS )r )rP rS )r� r� r r r r� J s c S s g | ]}|d t jkr|�qS )r )rP rU )r� r� r r r r� K s c S s g | ]}|d t jkr|�qS )r )rP rU )r� r� r r r r� L s �auto�::1r z-Routing by user not available on this system.zUser %s does not exist.ZnatFTz IPv6 required but not listening.zIPv6 enabled: %r
zUDP enabled: %r
zDNS enabled: %r
zUser enabled: %r
� 127.0.0.1c 3 s | ]}� d |d kV qdS )r r Nr )r� �sex)� listenip_v4r r � <genexpr>� s zmain.<locals>.<genexpr>� c 3 s | ]}� d |d kV qdS )r r Nr )r� r� )� listenip_v6r r r� � s � i 0 i(# zBinding redirector:z %dr� �
zTCP redirectorzUDP redirectorz Binding DNS:ZDNSz&IPv6 subnets defined but not listeningz)IPv6 ns servers defined but not listeningz&IPv4 subnets defined but not listeningz)IPv4 ns servers defined but not listening)r� r )r� r rb rb )0r r( r r r[ r
�tuplerJ Zget_supported_featuresr Zipv6r Zpw_uid�KeyErrorr� r� Zipv4r� �dnsr� Zassert_features�anyrM rP rU rS �ranger r; Z
SOCK_DGRAMrT rQ r rR rC rO rW Zsetup_tcp_listenerZsetup_udp_listenerr] r� rr �
returncoder� r� r� �stopr: )*r� r� r� r� r� r� r� r~ rx r� r� rc r| r} r� r� r& r� ry r' r� ZsubnetsZ
subnets_v6Z nslist_v6Z
subnets_v4Z nslist_v4ZavailZrequiredZportsZ
used_portsZlast_er r� ZboundZportr� r� Zlv6Zlv4r� r� r� r )r� r� r �main* sj
r� )?r r� r4 r� �
subprocessrp r r rl Zsshuttle.helpersrh Zsshuttle.ssnetr� Z sshuttle.sshr� Zsshuttle.ssyslogrj Zsshuttle.sdnotifyr� r r r r r r r r r
r r r
Zsshuttle.methodsr r �pwdr � ImportErrorrP Z pythonsocket�getattr�AttributeErrorZ
socket_extr r6 r� r� r r r( r9 r: r; r[ r� r� r� r� r� r� r� r� r� r� r r r r �<module> sb
$
H %}