File: //usr/lib/python3.6/site-packages/sshuttle/__pycache__/ssnet.cpython-36.opt-1.pyc
3
xR<_4M � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z m
Z
m Z m Z m
Z
dZdZd ZdZdZdZdZd Zd
Zd Zd Zd
ZdZdZdZ dZ dZ dZ dZ dZ!dZ"edededededed ed ed e d e d e d!e d"e d#e!d$e"d%iZ#ej$ej%ej&ej'ej(ej)ej'ej*ej+g Z,d&d'� Z-d(d)� Z.d*d+� Z/d,d-� Z0d a1G d.d/� d/�Z2G d0d1� d1�Z3G d2d3� d3e3�Z4G d4d5� d5e3�Z5G d6d7� d7e2�Z6d8d9� Z7d:d;� Z8dS )<� N)�b�log�debug1�debug2�debug3�Fatali�� i � � � � i B iB iB iB iB iB iB iB iB i B i
B i B i B i
B iB ZEXITZPINGZPONGZ TCP_CONNECTZTCP_STOP_SENDINGZTCP_EOFZTCP_DATAZROUTESZHOST_REQZ HOST_LISTZDNS_REQZ DNS_RESPONSEZUDP_OPENZUDP_DATAZ UDP_CLOSEc C s || kr| j |� d S )N)�append)�socks�elem� r �/usr/lib/python3.6/ssnet.py�_add> s r c C sP g }x>| D ]6}y|j |j� � W q
tk
r> |j |� Y q
X q
W |j� |S )N)r �fileno�AttributeError�sort)r �out�ir r r �_fdsC s
r c G sb y| |� S t k
r\ tj� d d� \}}|jtjtjfkrB� ntd| j|f � d S Y nX d S )Nr z%s: err was: %s
)�OSError�sys�exc_info�errnoZ EWOULDBLOCKZEAGAINr �__name__)�func�args�_�er r r � _nb_cleanN s r c C s� y$| j � }|r"d|d |d f S W nf tjk
rx tj� d d� \}}|jd tjkr\n|jd tjtj fkrt� Y n t
k
r� Y nX dS )Nz%s:%sr r r �unknown) Z getpeername�socket�errorr r r r �EINVALZENOTCONNZENOTSOCKr )�sockZpnr r r r r �
_try_peernameZ s r& c @ sz e Zd Zd dd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z e
dd� �Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS ) � SockWrapperNc C sZ t d7 a tdt � d | _|| _|| _d | _| _g | _|| _|pJt | j�| _
| j � d S )Nr z(creating new SockWrapper (%d now exist)
F) �_swcountr �exc�rsock�wsock� shut_read�
shut_write�buf�
connect_tor& �peername� try_connect)�selfr* r+ r/ r0 r r r �__init__o s zSockWrapper.__init__c C s4 t d8 a td| t f � | jr0td| | jf � d S )Nr z%r: deleting (%d remain)
z%r: error was: %s
)r( r r) )r2 r r r �__del__| s zSockWrapper.__del__c C sB | j | jkr d| j j� }nd| j j� | jj� f }d|| jf S )Nz#%dz#%d,%dzSW%s:%s)r* r+ r r0 )r2 Zfdsr r r �__repr__� s zSockWrapper.__repr__c C s | j s || _ | j� | j� d S )N)r) �nowrite�noread)r2 r r r r �seterr� s zSockWrapper.seterrc C sZ | j r| jr| j� d | _ | j s$d S | jjd� td| | j f � y| jj| j � d | _ W n� tjk
�rT t j
� d d� \}}td| |f � |j d t j
kr�| jjtjtj�}tj|tj|��}td| |f � |j d t jt jgkr�nb|j d dk�rd | _ nJ|j d t jk�r d | _ n0|j d tt jt jg k�rNd | _ | j|� n� Y nX d S )NFz%r: trying connect to %r
r z%r: connect result: %s
r z %r: fixed connect result: %s
)r/ r- r7 r* � setblockingr Zconnectr" r# r r r r r$ Z
getsockoptZ
SOL_SOCKETZSO_ERROR�os�strerrorZ EINPROGRESSZEALREADYZEISCONN�NET_ERRSZEACCESZEPERMr8 )r2 r r Zrealerrr r r r1 � s8
zSockWrapper.try_connectc C s | j std| � d| _ d S )Nz%r: done reading
T)r, r )r2 r r r r7 � s zSockWrapper.noreadc C sf | j sbtd| � d| _ y| jjt� W n8 tjk
r` tj� d d� \}}| j d| � Y nX d S )Nz%r: done writing
Tr z nowrite: %s)
r- r r+ Zshutdown�SHUT_WRr" r# r r r8 )r2 r r r r r r6 � s zSockWrapper.nowritec C s dS )NFr r r r r �too_full� s zSockWrapper.too_fullc C s� | j r
dS | jjd� yttj| jj� |�S tk
r� tj � d d� \}}|j
t
j krrt d| � | j
� dS | jd| � dS Y nX d S )Nr Fr z%r: uwrite: got EPIPE
z
uwrite: %s)r/ r+ r9 r r: �writer r r r r ZEPIPEr r6 r8 )r2 r. r r r r r �uwrite� s zSockWrapper.uwritec C s
| j |�S )N)r@ )r2 r. r r r r? � s zSockWrapper.writec C st | j r
d S | jrd S | jjd� yttj| jj� d�S tk
rn t j
� d d� \}}| j d| � t d�S X d S )NFi r z uread: %s� )
r/ r, r* r9 r r: �readr r r r r8 r )r2 r r r r r �uread� s zSockWrapper.ureadc C s: | j r
d S | j� }|r"| j j|� |td�kr6| j� d S )NrA )r. rC r r r7 )r2 �rbr r r �fill� s zSockWrapper.fillc C sv | j r8| j d r8|j| j d �}| j d |d � | j d< x"| j rZ| j d rZ| j jd� q:W | j rr| jrr|j� d S )Nr )r. r? �popr, r6 )r2 Zoutwrap�wroter r r �copy_to� s zSockWrapper.copy_to)NN)r �
__module__� __qualname__r3 r4 r5 r8 r1 r7 r6 � staticmethodr> r@ r? rC rE rH r r r r r' m s
0
r' c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) �HandlerNc C s d| _ |p g | _|r|| _d S )NT)�okr �callback)r2 r rN r r r r3 s
zHandler.__init__c C s x| j D ]}t||� qW d S )N)r r )r2 �r�w�xr r r r �
pre_select s zHandler.pre_selectc C s\ t d| � tj| jg g d�\}}}x2|D ]*}|jd�}|s*t d| � g | _d| _q*W d S )Nz--no callback defined-- %r
r i z--closed-- %r
F)r �selectr ZrecvrM )r2 r% rO r �s�vr r r rN s
zHandler.callback)NN)r rI rJ r3 rR rN r r r r rL s
rL c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Proxyc C s, t j| |j|j|j|jg� || _|| _d S )N)rL r3 r* r+ �wrap1�wrap2)r2 rW rX r r r r3 " s zProxy.__init__c C s� | j jr| jj� | jjr$| j j� | j jr<t|| j j� n8| j jr^| jj� stt|| jj � n| j j
stt|| j j� | jjr�t|| jj� n8| jjr�| j j� s�t|| j j � n| jj
s�t|| jj� d S )N) rW r- rX r7 r/ r r* r. r> r+ r, )r2 rO rP rQ r r r rR ( s$
zProxy.pre_selectc C s� | j j� | jj� | j j� | jj� | j j| j� | jj| j � | j jrf| jjrfg | j _| j j� | jjr�| j jr�g | j_| jj� | j jr�| jjr�| j j r�| jj r�d| _ | j j
� | jj
� d S )NF) rW r1 rX rE rH r. r- r7 r, rM r6 )r2 r% r r r rN > s"
zProxy.callbackN)r rI rJ r3 rR rN r r r r rV s rV c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�Muxc C s� t j| ||g� || _|| _d | _ | _| _d | _ | _| _ d | _
| _ i | _ d| _
d| _td�| _g | _d| _d| _| jdttd�� d S )Nr rA FZchicken)rL r3 �rfile�wfile� new_channel� got_dns_req�
got_routes� got_udp_openZ got_udp_dataZ
got_udp_close� got_host_req�
got_host_list�channels�chani�wantr �inbuf�outbuf�fullnessr> �send�CMD_PING)r2 rZ r[ r r r r3 T s
z Mux.__init__c C sH xBt d�D ]6}| jd7 _| jtkr,d| _| jj| j�s
| jS q
W d S )Ni r )�rangerc � MAX_CHANNELrb �get)r2 r r r r � next_channeld s
zMux.next_channelc C s$ d}x| j D ]}|t|�7 }q W |S )Nr )rf �len)r2 ZtotalZbyter r r �
amount_queuedm s zMux.amount_queuedc C s, | j tkr(| js"| jdttd�� d| _d S )Nr ZrttestT)rg �LATENCY_BUFFER_SIZEr> rh ri r )r2 r r r �check_fullnesss s
zMux.check_fullnessc C sl t jdtd�td�||t|��| }| jj|� td|tj|t |��t|�| j
f � | j
t|�7 _
d S )Nz!ccHHH�Sz* > channel=%d cmd=%s len=%d (fullness=%d)
) �struct�packr rn rf r r � cmd_to_namerl �hexrg )r2 �channel�cmd�data�pr r r rh ~ s zMux.sendc C sz t d|tj|t|��t|�f � |tkr<| jdt|� �n:|tkr\t d� d| _d| _ �n|t
krnd| _ �n|t kr�| j
r�| j
||� n�|tkr�| jr�| j||� n�|tkr�| jr�| j||� n�|tkr�| jr�| j|� ntd��n�|tk�r| j�r| j|� ntd��nh|tk�r6| j�r,| j|� ntd��n@| jj|�}|�sltd|tj|t|��t|�f � n
|||� d S ) Nz < channel=%d cmd=%s len=%d
r zreceived PING response
Fz"got CMD_ROUTES without got_routes?z&got CMD_HOST_REQ without got_host_req?z(got CMD_HOST_LIST without got_host_list?z-warning: closed channel %d got cmd=%s len=%d
)r ru rl rv rn ri rh �CMD_PONGr> rg �CMD_EXITrM �CMD_TCP_CONNECTr\ � CMD_DNS_REQr] � CMD_UDP_OPENr_ �
CMD_ROUTESr^ � Exception� CMD_HOST_REQr` �
CMD_HOST_LISTra rb r )r2 rw rx ry rN r r r �
got_packet� sH
zMux.got_packetc C s� yt j| jj� d� W nH tk
r^ tj| jj� tj�}|t jO }tj| jj� tj|�}Y nX | j r�| j d r�t
t j | jj� | j d �}t d|t
| j d �f � |r�| j d |d � | j d< x$| j r�| j d r�g | j dd�< q�W d S )NFr zmux wrote: %r/%d
r )r: � set_blockingr[ r r �fcntl�F_GETFL�
O_NONBLOCK�F_SETFLrf r r? r rn )r2 �flagsrG r r r �flush� s
z Mux.flushc C s� yt j| jj� d� W nH tk
r^ tj| jj� tj�}|t jO }tj| jj� tj|�}Y nX yt t j
| jj� t �}W n4 t k
r� t
j� d d� \}}td| ��Y nX |td�kr�d| _|r�| j|7 _d S )NFr z
other end: %rrA )r: r� rZ r r r� r� r� r� r rB rp r r r r r rM re )r2 r� rB r r r r r rE � s
zMux.fillc C s� | j � x�t| j�| jptkrHtjd| jd t� �\}}}}}|t | _| jr�t| j�| jkr�| jt| j� }| j| jd � | _d| _| j|||� q
P q
W d S )Nz!ccHHHr )rE rn re rd �HDR_LENrs �unpackr� )r2 �s1�s2rw rx Zdatalenry r r r �handle� s
z
Mux.handlec C s" t || j� | jr t || j� d S )N)r rZ rf r[ )r2 rO rP rQ r r r rR � s zMux.pre_selectc C sL t j | jg| jgg d�\}}}| j|kr0| j� | jrH| j|krH| j� d S )Nr )rS rZ r[ r� rf r� )r2 r% rO rP r r r r rN � s
z Mux.callbackN)r rI rJ r3 rm ro rq rh r� r� rE r� rR rN r r r r rY R s .rY c @ sl e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�
MuxWrapperc C sB t j| |j|j� || _|| _| j| jj|< g | _t d| � d S )Nznew channel: %d
)
r' r3 rZ r[ �muxrw r� rb r r )r2 r� rw r r r r3 � s zMuxWrapper.__init__c C s | j � tj| � d S )N)r6 r' r4 )r2 r r r r4 s zMuxWrapper.__del__c C s d| j | jf S )Nz SW%r:Mux#%d)r0 rw )r2 r r r r5 s zMuxWrapper.__repr__c C s( | j s$| jj| jttd�� | j� d S )NrA )r, r� rh rw �CMD_TCP_STOP_SENDINGr � setnoread)r2 r r r r7 s zMuxWrapper.noreadc C s$ | j s td| � d| _ | j� d S )Nz%r: done reading
T)r, r � maybe_close)r2 r r r r� s zMuxWrapper.setnoreadc C s( | j s$| jj| jttd�� | j� d S )NrA )r- r� rh rw � CMD_TCP_EOFr �
setnowrite)r2 r r r r6 s zMuxWrapper.nowritec C s$ | j s td| � d| _ | j� d S )Nz%r: done writing
T)r- r r� )r2 r r r r� s zMuxWrapper.setnowritec C s* | j r&| jr&td| � d | jj| j< d S )Nz%r: closing connection
)r, r- r r� rb rw )r2 r r r r� ! s zMuxWrapper.maybe_closec C s | j jS )N)r� r> )r2 r r r r> ( s zMuxWrapper.too_fullc C s> | j jr dS t|�dkr$|d d� }| j j| jt|� t|�S )Nr i )r� r> rn rh rw � CMD_TCP_DATA)r2 r. r r r r@ + s zMuxWrapper.uwritec C s | j rtd�S d S d S )NrA )r, r )r2 r r r rC 3 s zMuxWrapper.ureadc C sR |t kr| j� n<|tkr$| j� n*|tkr:| jj|� ntd|t|�f ��d S )Nz unknown command %d (%d bytes)) r� r� r� r� r� r. r r� rn )r2 rx ry r r r r� 9 s
zMuxWrapper.got_packetN)r rI rJ r3 r4 r5 r7 r� r6 r� r� r> r@ rC r� r r r r r� � s r� c C sH t d||f � tj| �}|jtjtjd� t||||fd||f d�S )NzConnecting to %s:%d
�? z%s:%d)r/ r0 )r r" Z
setsockoptZSOL_IPZIP_TTLr' )ZfamilyZipZportZoutsockr r r � connect_dstG s
r� c
C s$ g }g }g }dd� | D �}x|D ]}| j |� q W x| D ]}|j|||� q8W tdt| �t|�t|�t|�|j|jf � tj|||�\}}}tdt| �t|�t|�t|�f � || | }i } x6| D ].}x(|jD ] }||kr�|j |� d| |< q�W q�W x$|D ] }|| k�r t
d| ���q W d S )Nc S s g | ]}|j s|�qS r )rM )�.0rT r r r �
<listcomp>T s zrunonce.<locals>.<listcomp>z,Waiting: %d r=%r w=%r x=%r (fullness=%d/%d)
z Ready: %d r=%r w=%r x=%r
r z%socket %r was not used by any handler) �removerR r rn r rg r> rS r rN r )
Zhandlersr� rO rP rQ Z to_remove�hrT ZreadyZdidr r r �runonceP s0
r� )9r rs r" r rS r: r� Zsshuttle.helpersr r r r r r rk rp ZSHUT_RDr= Z SHUT_RDWRr� r| ri r{ r} r� r� r� r� r� r� r~ ZCMD_DNS_RESPONSEr Z CMD_UDP_DATAZ
CMD_UDP_CLOSEru Z ECONNREFUSEDZ ETIMEDOUTZ EHOSTUNREACHZ ENETUNREACHZ EHOSTDOWNZENETDOWNZ ECONNABORTEDZ
ECONNRESETr< r r r r& r( r' rL rV rY r� r� r� r r r r �<module> s| 2 )M