File: //usr/lib/python3.6/site-packages/sshuttle/__pycache__/ssnet.cpython-36.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 |st �| j|�S )N)�AssertionErrorr@ )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 )NrB )r. rD 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? rD rF rI 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 rO 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 ZrecvrN )r2 r% rP r �s�vr r r rO s
zHandler.callback)NN)r rJ rK r3 rS rO r r r r rM s
rM c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Proxyc C s, t j| |j|j|j|jg� || _|| _d S )N)rM r3 r* r+ �wrap1�wrap2)r2 rX rY 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) rX r- rY r7 r/ r r* r. r> r+ r, )r2 rP rQ rR r r r rS ( 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) rX r1 rY rF rI r. r- r7 r, rN r6 )r2 r% r r r rO > s"
zProxy.callbackN)r rJ rK r3 rS rO r r r r rW s rW 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 rB FZchicken)rM 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 r[ 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 )�rangerd � MAX_CHANNELrc �get)r2 r r r r � next_channeld s
zMux.next_channelc C s$ d}x| j D ]}|t|�7 }q W |S )Nr )rg �len)r2 ZtotalZbyter r r �
amount_queuedm s zMux.amount_queuedc C s, | j tkr(| js"| jdttd�� d| _d S )Nr ZrttestT)rh �LATENCY_BUFFER_SIZEr> ri rj r )r2 r r r �check_fullnesss s
zMux.check_fullnessc C s� t |t�st�t|�dks t�tjdtd�td�||t|��| }| jj|� t d|t
j |t |��t|�| j
f � | j
t|�7 _
d S )Ni�� z!ccHHH�Sz* > channel=%d cmd=%s len=%d (fullness=%d)
)�
isinstance�bytesrA ro �struct�packr rg r r � cmd_to_namerm �hexrh )r2 �channel�cmd�data�pr r r ri ~ s zMux.sendc C s� t d|tj|t|��t|�f � |tkr<| jdt|� �nv|tkr\t d� d| _d| _ �nV|t
krnd| _ �nD|t kr�| j
j|� s�t�| jr�| j||� �n|tkr�| j
j|� s�t�| jr�| j||� n�|tkr�| j
j|� s�t�| jr�| j||� n�|tk�r"| j�r| j|� ntd��n�|tk�rJ| j�r@| j|� ntd��nh|tk�rr| j�rh| j|� ntd��n@| j
j|�}|�s�td|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 rx rm ry ro rj ri �CMD_PONGr> rh �CMD_EXITrN �CMD_TCP_CONNECTrc rA r] � CMD_DNS_REQr^ � CMD_UDP_OPENr` �
CMD_ROUTESr_ � Exception� CMD_HOST_REQra �
CMD_HOST_LISTrb r )r2 rz r{ r| rO r r r �
got_packet� sN
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_SETFLrg r r? r ro )r2 �flagsrH 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: %rrB )r: r� r[ r r r� r� r� r� r rC rq r r r r r rN rf )r2 r� rC r r r r r rF � s
zMux.fillc C s� | j � x�t| j�| jptkrhtjd| jd t� �\}}}}}|td�ksNt�|td�ks^t�|t | _| jr�t| j�| jkr�| jt| j� }| j| jd � | _d| _| j |||� q
P q
W d S )Nz!ccHHHrs r )
rF ro rf re �HDR_LENrv �unpackr rA r� )r2 �s1�s2rz r{ Zdatalenr| r r r �handle� s
z
Mux.handlec C s" t || j� | jr t || j� d S )N)r r[ rg r\ )r2 rP rQ rR r r r rS � s zMux.pre_selectc C sL t j | jg| jgg d�\}}}| j|kr0| j� | jrH| j|krH| j� d S )Nr )rT r[ r\ r� rg r� )r2 r% rP rQ r r r r rO � s
z Mux.callbackN)r rJ rK r3 rn rp rr ri r� r� rF r� rS rO r r r r rZ R s .rZ 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 r[ r\ �muxrz r� rc r r )r2 r� rz 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 rz )r2 r r r r5 s zMuxWrapper.__repr__c C s( | j s$| jj| jttd�� | j� d S )NrB )r, r� ri rz �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 )NrB )r- r� ri rz � 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� rc rz )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> ro ri rz � CMD_TCP_DATA)r2 r. r r r r@ + s zMuxWrapper.uwritec C s | j rtd�S d S d S )NrB )r, r )r2 r r r rD 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� ro )r2 r{ r| r r r r� 9 s
zMuxWrapper.got_packetN)r rJ rK r3 r4 r5 r7 r� r6 r� r� r> r@ rD 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 )rN )�.0rU 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) �removerS r ro r rh r> rT r rO r )
Zhandlersr� rP rQ rR Z to_remove�hrU ZreadyZdidr r r �runonceP s0
r� )9r rv r" r rT r: r� Zsshuttle.helpersr r r r r r rl rq ZSHUT_RDr= Z SHUT_RDWRr� r rj r~ r� r� r� r� r� r� r� r� ZCMD_DNS_RESPONSEr� Z CMD_UDP_DATAZ
CMD_UDP_CLOSErx Z ECONNREFUSEDZ ETIMEDOUTZ EHOSTUNREACHZ ENETUNREACHZ EHOSTDOWNZENETDOWNZ ECONNABORTEDZ
ECONNRESETr< r r r r& r( r' rM rW rZ r� r� r� r r r r �<module> s| 2 )M