File: //lib/python3.6/site-packages/sshuttle/__pycache__/server.cpython-36.opt-1.pyc
3
� L_�3 � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZejd$krXd dl m
Z
nej ej B dfdd�Z
d dl
jZd dljZd dljZd dlZd dl
mZmZmZmZ d dlmZmZmZm Z m Z m Z m Z dd � Z d
d � Z!d d
� Z"dd� Z#dd� Z$dd� Z%dd� Z&dd� Z'dd� Z(dd� Z)G d d � d �Z*G d d � d e�Z+G d d!� d!e�Z,d"d#� Z-dS )%� N� )�whichc C sx |d k r|g}n,dt jkr.t jd jt j�}nt jjt j�}x6|D ].}t jj|| �}t jj|�rBt j||�rB|S qBW d S )N�PATH) �os�environ�split�pathsep�defpath�path�join�exists�access)�file�moder
Z search_paths�p�filepath� r � /usr/lib/python3.6/server.pyr s
r )�Handler�Proxy�Mux�
MuxWrapper)�b�log�debug1�debug2�debug3�Fatal� resolvconf_random_nameserverc C s� | dkr d} t jd| �}|r�|j� }|d }t|d p8d�}|d d kr\|d7 }t|d �}n>|d
d kr||d 7 }t|d �}n |d
d kr�|d7 }t|d�}|}tjdtj|��d |fS d S )N�defaultz 0.0.0.0/0z*^(\d+(\.\d+(\.\d+(\.\d+)?)?)?)(?:/(\d+))?$r � � � z.0.0.0� � z.0.0� r z.0� z!I) �re�match�groups�int�min�struct�unpack�socketZ inet_aton)Zipstr�m�gZips�widthr r r �_ipmatch& s$
r2 c C s |dkr | S d| |f S d S )Nr! z%s/%dr )�ipr1 r r r �_ipstr< s r4 c C s8 | sdS x*t d�D ] }| d td|�@ rd| S qW dS )Nr! r r" )�range�_shl)Znetmask�ir r r � _maskbitsD s r8 c C s | t d| � S )Nr$ )r* )�n�bitsr r r r6 N s r6 c C sB | j d �}t|�dk rdS t|d �}t|d �}t|�}||fS )Nr r r$ )NN)r �lenr2 r8 )�lineZcols�ipwZmaskw�maskr r r �_route_netstatR s
r? c C s> | j d d�d }d|kr dS |j d�\}}t|�}|t|�fS )Nr" r �/)NN)r r2 r* )r< Zipmr3 r> r= r r r �_route_iproute\ s rA c C s� t jd dd�}tj| tj|d�}g }x~|jD ]t}|j� s<q.||jd��\}}|sTq.t|d |�}|d t t d|�d d| �@ } |j
t j t j
tjd | ��|f� q.W |j� }
|
dkr�td
| |
f � td � |S ) Nr �C)r �LC_ALL)�stdout�env�ASCIIr" r r! z!IzWARNING: %r returned %d
z1WARNING: That prevents --auto-nets from working.
)r r � ssubprocess�Popen�PIPErD �strip�decoder+ r6 �appendr. �AF_INETZ inet_ntoar, �pack�waitr ) �argvZ
extract_routerE r �routesr< r= r> r1 r3 �rvr r r � _list_routese s& rS c c sx t d�rtddgt�} n$t d�r0tddgt�} n td� g } x6| D ].\}}}|jd� rB|jd� rB|||fV qBW d S )Nr3 ZrouteZnetstatz-rnz:WARNING: Neither ip nor netstat were found on the server.
z0.z127.)r rS rA r? r �
startswith)rQ �familyr3 r1 r r r � list_routes s rV c C s t j� } djtj| � �S )N� )�sys�exc_infor � traceback�format_exception)rY r r r � _exc_dump� s r\ c C s� t j� \}}tj� }|s�d}zpyD|j� tj|j� d� tj|j� d� |j� tj| |�p^d}W n& t k
r� t
dt � � d}Y nX W d tj |� X |j� ||fS )N�c r" r z%s
�b )
r. Z
socketpairr �fork�close�dup2�fileno� hostwatchZhw_main� Exceptionr r\ �_exit)Z
seed_hosts�
auto_hosts�s1�s2�pidrR r r r �start_hostwatch� s" rj c @ s e Zd Zdd� ZdS )� Hostwatchc C s d| _ d | _d S )Nr )ri �sock)�selfr r r �__init__� s zHostwatch.__init__N)�__name__�
__module__� __qualname__rn r r r r rk � s rk c @ s0 e Zd Zdd� Zedd� �Zdd� Zdd� Zd S )
�DnsProxyc C s� t j| g � tj� d | _|| _|| _d| _|| _i | _d | _ d | _
|d krTd | _ n$|j d�\| _ | _
| j
| j | j
�| _ | j� d S )N� r �@)r rn �time�timeout�mux�chan�tries�request�peers�
to_ns_peer�
to_ns_port�
to_nameserverr � _addrinfo�try_send)rm rw rx rz r~ r r r rn � s
zDnsProxy.__init__c C s2 t |�dkrd}tj| |�d \}}}}}||fS )Nr �5 )r* r. Z getaddrinfo)�peer�portrU �_�sockaddrr r r r � s zDnsProxy._addrinfoc C s. | j dkrd S | j d7 _ | jd kr6t� \}}d}n| j}t| j�}| j||�\}}tj|tj�}|j tj
tj d� |j |� || j
|< td||| j f � y |j| j� | jj|� W nn tjk
�r( tj� d d� \}}|jd tjk�rtd||f � | j� d S td||f � d S Y nX d S ) Nr r" r� �? z DNS: sending to %r:%d (try %d)
r$ r zDNS send to %r: %s
)ry r~ r r| r* r} r r. �
SOCK_DGRAM�
setsockopt�SOL_IP�IP_TTLZconnectr{ r �sendrz �socksrL �errorrX rY �args�ssnet�NET_ERRSr� r )rm r� r� r� rU r� rl �er r r r� � s2
zDnsProxy.try_sendc C s� | j | }y|jd�}W n~ tjk
r� tj� d d� \}}| jj|� | j |= |jd t j
kr~t d||f � | j � d S t
d||f � d S Y nX t dt|� � | jj| jt j|� d| _d S )Ni r$ r zDNS recv from %r: %s
zDNS response: %d bytes
F)r{ �recvr. r� rX rY r� �remover� r� r� r r� r r; rw r� rx ZCMD_DNS_RESPONSE�ok)rm rl r� �datar� r� r r r �callback� s
zDnsProxy.callbackN)ro rp rq rn � staticmethodr r� r� r r r r rr � s $rr c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�UdpProxyc C s^ t j |t j�}tj| |g� tj� d | _|| _|| _|| _|t j krZ| jj
t j t j d� d S )Nrs r� )
r. r� r rn ru rv rw rx rl rM r� r� r� )rm rw rx rU rl r r r rn s
zUdpProxy.__init__c C sh t d| � y| jj||� W nD tjk
rb tj� d d� \}}td|d |d |f � d S X d S )NzUDP: sending to %r port %d
r$ zUDP send to %r port %d: %s
r r" )r rl Zsendtor. r� rX rY r )rm �dstipr� r� r� r r r r� s z
UdpProxy.sendc
C s� y|j d�\}}W nD tjk
rV tj� d d� \}}td|d |d |f � d S X tdt|� � td|d |d f �}| j j
| j t j
|| � d S )Ni r$ z UDP recv from %r port %d: %s
r r" zUDP response: %d bytes
z%s,%r,)Zrecvfromr. r� rX rY r r r; r rw r� rx r� � CMD_UDP_DATA)rm rl r� r� r� r� Zhdrr r r r� s zUdpProxy.callbackN)ro rp rq rn r� r� r r r r r� s
r� c s� t dtj� � tjdkr"dt_ndt_t d| � tjjd� tjj � g �t
tj tj���j �� t dt
|� d � |r�tt� �}t d � x |D ]}t d
| � q�W ng }d }x|D ]}|d | 7 }q�W �jd
tjt|�� t� �td ��_��fdd��� ���fdd�}|�_��fdd�}|�_i �����fdd�} | �_i ���fdd������fdd�}
|
�_�x$�j�r��j�r�t j �jt j �\} } | �r�t d| ��tj ��� | �rƈj!� ��r8t"j"� }
g }xH�j#� D ]<\}}|j$|
k �s |j �r�t%d| � |j |� d |_�q�W x|D ] }�|= �q(W ��rvg }x:�j#� D ].\}}|j�sLt%d | � |j |� d |_�qLW x|D ] }�|= �q�W �qvW d S ) Nz'Starting server with Python version %s
r" z s: zserver: z latency control setting = %r
z SSHUTTLE0001z
auto-nets:�
zavailable routes:
z %d/%s/%d
rW z %d,%s,%d
r c sx � j jd�}|rl� j| jtd��}|d rF|j� � _|jtd�� n
td�� _�jdtj td�j
|�� nt d��d S )Ni r� r" rW r zhostwatch process died���) rl r� �leftoverr r �poprL r� r� Z
CMD_HOST_LISTr r )rl Zcontent�lines)�hwrw r r �hostwatch_readyK s
z main.<locals>.hostwatch_readyc s@ �j s<t| jd�j� j� � �\�_ �_�jt�jg�d�� d S )NrF )r� r� )ri rj rK rJ r rl rL r )r� )rf �handlersr� r� r r � got_host_reqZ s
zmain.<locals>.got_host_reqc s` |j d�jdd�\}}}t|�}|tjkr0tj}t|�}tj|||�}� jt t
�| �|�� d S )NrF �,r$ ) rK r r* r. rM ZAF_INET6r� Z connect_dstrL r r )�channelr� rU r� �dstportZoutwrap)r� rw r r � new_channelb s
zmain.<locals>.new_channelc s0 t d| � t�| |��}�j|� |� | < d S )Nz!Incoming DNS request channel=%d.
)r rr rL )r� r� �h)� dnshandlersr� rw r~ r r �dns_reqr s
zmain.<locals>.dns_reqc s� t d| |f � |tjkrb|jtd�d�\}}}t|�}t d||f � �| }|j||f|� n(|tjkr�t d� �| }d|_� j | = d S )Nz(Incoming UDP request channel=%d, cmd=%d
r� r$ z is incoming UDP data. %r %d.
zis incoming UDP close
F)
r r� r� r r r* r� Z
CMD_UDP_CLOSEr� �channels)r� �cmdr� r� r� r� )rw � udphandlersr r �udp_req{ s
zmain.<locals>.udp_reqc s\ t d� t|�}� �fdd��j� < � �kr:td� ��n t�� |�}�j|� |�� < d S )NzIncoming UDP open.
c s �� | |�S )Nr )r� r� )r� r� r r �<lambda>� s z(main.<locals>.udp_open.<locals>.<lambda>z&UDP connection channel %d already open)r r* r� r r� rL )r� r� rU r� )r� rw r� r� )r� r �udp_open� s
zmain.<locals>.udp_openz+hostwatch exited unexpectedly: code 0x%04x
z expiring dnsreqs channel=%d
Fzexpiring UDP channel=%d
)&r �platformZpython_version�helpers�verboseZ logprefixrX rD �write�flushr �stdinrL �str�listrV r� r� Z
CMD_ROUTESr rk r� r� r� Z got_dns_reqZ got_udp_openr� ri r �waitpid�WNOHANGr ZrunonceZcheck_fullnessru �itemsrv r )Zlatency_controlrf r~ Z auto_netsrQ �rZroutepktr� r� r� r� ZrpidrR Znowr� r� r� r ) rf r� r� r� r� rw r~ r� r� r �main( s�
r� )r r ).r' r, r. rZ ru rX r r� � version_infoZshutilr �F_OK�X_OKZsshuttle.ssnetr� Zsshuttle.helpersr� Zsshuttle.hostwatchrc �
subprocessrG r r r r r r r r r r r r2 r4 r8 r6 r? rA rS rV r\ rj rk rr r� r� r r r r �<module> s<
$
X!