3
\�i � @ s d Z ddlZddlZddlZddlZddlZy ddlZW n ek
rT ddlZY nX ddl m
Z
ddl m Z ddddd d
d d d
g Z
eed�r�e
jdddg� eed�r�e
jddddg� eed�r�ejZnejZG dd� d�ZG dd� de�ZG dd� de�Zeed��rG dd� d�ZG d d
� d
�Zeed��rXG d d� dee�ZG d d� dee�ZG d d� dee�ZG d d � d ee�Zeed��r�G d!d� de�Z G d"d� de�Z G d#d� dee �Z G d$d� dee �Z G d%d
� d
�Z G d&d � d e �Z!G d'd(� d(e
�Z"G d)d � d e �Z#dS )*z0.4� N)�BufferedIOBase)� monotonic�
BaseServer� TCPServer� UDPServer�ThreadingUDPServer�ThreadingTCPServer�BaseRequestHandler�StreamRequestHandler�DatagramRequestHandler�ThreadingMixIn�fork�ForkingUDPServer�ForkingTCPServer� ForkingMixIn�AF_UNIX�UnixStreamServer�UnixDatagramServer�ThreadingUnixStreamServer�ThreadingUnixDatagramServer� PollSelectorc @ s� e Zd ZdZdd� Zdd� Zd%dd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� ZdS )&r Nc C s || _ || _tj� | _d| _d S )NF)�server_address�RequestHandlerClass� threadingZEvent�_BaseServer__is_shut_down� _BaseServer__shutdown_request)�selfr r � r �$/usr/lib64/python3.6/socketserver.py�__init__� s
zBaseServer.__init__c C s d S )Nr )r r r r �server_activate� s zBaseServer.server_activate� �?c C sx | j j� zVt� �F}|j| tj� x0| jsR|j|�}| jr<P |rH| j� | j � q$W W d Q R X W d d| _| j j
� X d S )NF) r �clear�_ServerSelector�register� selectors�
EVENT_READr �select�_handle_request_noblock�service_actions�set)r Z
poll_interval�selector�readyr r r �
serve_forever� s
zBaseServer.serve_foreverc C s d| _ | jj� d S )NT)r r �wait)r r r r �shutdown� s zBaseServer.shutdownc C s d S )Nr )r r r r r) s zBaseServer.service_actionsc
C s� | j j� }|d kr| j}n| jd k r0t|| j�}|d k rBt� | }t� �R}|j| tj� x<|j |�}|rp| j
� S |d k rZ|t� }|dk rZ| j � S qZW W d Q R X d S )Nr ) �socketZ
gettimeout�timeout�min�timer# r$ r% r& r' r( �handle_timeout)r r1 Zdeadliner+ r, r r r �handle_request s"
zBaseServer.handle_requestc C s� y| j � \}}W n tk
r$ d S X | j||�r�y| j||� W q� tk
rl | j||� | j|� Y q� | j|� � Y q�X n
| j|� d S )N)� get_request�OSError�verify_request�process_request� Exception� handle_error�shutdown_request)r �request�client_addressr r r r( 3 s
z"BaseServer._handle_request_noblockc C s d S )Nr )r r r r r4 J s zBaseServer.handle_timeoutc C s dS )NTr )r r= r> r r r r8 Q s zBaseServer.verify_requestc C s | j ||� | j|� d S )N)�finish_requestr< )r r= r> r r r r9 Y s zBaseServer.process_requestc C s d S )Nr )r r r r � server_closeb s zBaseServer.server_closec C s | j ||| � d S )N)r )r r= r> r r r r? j s zBaseServer.finish_requestc C s | j |� d S )N)�
close_request)r r= r r r r< n s zBaseServer.shutdown_requestc C s d S )Nr )r r= r r r rA r s zBaseServer.close_requestc C sH t dd tjd� t d|tjd� dd l}|j� t dd tjd� d S )N�-�( )�filez4Exception happened during processing of request fromr )�print�sys�stderr� traceback� print_exc)r r= r> rH r r r r; v s zBaseServer.handle_errorc C s | S )Nr )r r r r � __enter__� s zBaseServer.__enter__c G s | j � d S )N)r@ )r �argsr r r �__exit__� s zBaseServer.__exit__)r! )�__name__�
__module__� __qualname__r1 r r r- r/ r) r5 r( r4 r8 r9 r@ r? r< rA r; rJ rL r r r r r � s$ -
c @ sb e Zd ZejZejZdZdZ ddd�Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� ZdS )r � FTc C sT t j| ||� tj| j| j�| _|rPy| j� | j� W n | j� � Y nX d S )N)r r r0 �address_family� socket_type� server_bindr r@ )r r r Zbind_and_activater r r r � s
zTCPServer.__init__c C s8 | j r| jjtjtjd� | jj| j� | jj� | _d S )N� )�allow_reuse_addressr0 �
setsockoptZ
SOL_SOCKETZ SO_REUSEADDRZbindr Z getsockname)r r r r rS � s zTCPServer.server_bindc C s | j j| j� d S )N)r0 Zlisten�request_queue_size)r r r r r � s zTCPServer.server_activatec C s | j j� d S )N)r0 �close)r r r r r@ � s zTCPServer.server_closec C s
| j j� S )N)r0 �fileno)r r r r rY � s zTCPServer.filenoc C s
| j j� S )N)r0 Zaccept)r r r r r6 � s zTCPServer.get_requestc C s4 y|j tj� W n tk
r$ Y nX | j|� d S )N)r/ r0 ZSHUT_WRr7 rA )r r= r r r r< � s
zTCPServer.shutdown_requestc C s |j � d S )N)rX )r r= r r r rA s zTCPServer.close_requestN)T)rM rN rO r0 ZAF_INETrQ Z SOCK_STREAMrR rW rU r rS r r@ rY r6 r< rA r r r r r � s /
c @ s: e Zd ZdZejZdZdd� Zdd� Z dd� Z
d d
� Z d S ) r Fi c C s | j j| j�\}}|| j f|fS )N)r0 Zrecvfrom�max_packet_size)r �dataZ client_addrr r r r6 s zUDPServer.get_requestc C s d S )Nr )r r r r r s zUDPServer.server_activatec C s | j |� d S )N)rA )r r= r r r r< s zUDPServer.shutdown_requestc C s d S )Nr )r r= r r r rA s zUDPServer.close_requestN) rM rN rO rU r0 Z
SOCK_DGRAMrR rZ r6 r r< rA r r r r r s c sR e Zd ZdZdZdZdZdd�dd�Zdd � Zd
d � Z d d
� Z
� fdd�Z � Z S )r i, NrC F)�blockingc C s� | j d krd S xht| j �| jkrvy tjdd�\}}| j j|� W q tk
r^ | j j� Y q tk
rr P Y qX qW xt| j j � D ]f}y.|r�dntj
}tj||�\}}| j j|� W q� tk
r� | j j|� Y q� tk
r� Y q�X q�W d S )NrT r ���) �active_children�len� max_children�os�waitpid�discard�ChildProcessErrorr" r7 �copy�WNOHANG)r r\ �pid�_�flagsr r r �collect_children, s&
z ForkingMixIn.collect_childrenc C s | j � d S )N)rj )r r r r r4 O s zForkingMixIn.handle_timeoutc C s | j � d S )N)rj )r r r r r) V s z ForkingMixIn.service_actionsc C s� t j� }|r8| jd kr t� | _| jj|� | j|� d S d}z:y| j||� d}W n tk
rr | j||� Y nX W d z| j |� W d t j
|� X X d S )NrT r ) ra r
r^ r* �addrA r? r: r; r< �_exit)r r= r> rg Zstatusr r r r9 ] s
z ForkingMixIn.process_requestc s t � j� | j| jd� d S )N)r\ )�superr@ rj �_block_on_close)r )� __class__r r r@ v s
zForkingMixIn.server_close)
rM rN rO r1 r^ r` rn rj r4 r) r9 r@ �
__classcell__r r )ro r r # s #c s8 e Zd ZdZdZdZdd� Zdd� Z� fdd�Z� Z S ) r FNc C sH z6y| j ||� W n tk
r2 | j||� Y nX W d | j|� X d S )N)r? r: r; r< )r r= r> r r r �process_request_thread� s
z%ThreadingMixIn.process_request_threadc C sR t j| j||fd�}| j|_|j rF| jrF| jd kr:g | _| jj|� |j� d S )N)�targetrK ) r ZThreadrq �daemon_threadsZdaemonrn �_threads�append�start)r r= r> �tr r r r9 � s
z ThreadingMixIn.process_requestc s: t � j� | jr6| j}d | _|r6x|D ] }|j� q&W d S )N)rm r@ rn rt �join)r ZthreadsZthread)ro r r r@ � s
zThreadingMixIn.server_close)
rM rN rO rs rn rt rq r9 r@ rp r r )ro r r { s
c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s e Zd ZejZdS )r N)rM rN rO r0 r rQ r r r r r � s c @ s e Zd ZejZdS )r N)rM rN rO r0 r rQ r r r r r � s c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s e Zd ZdS )r N)rM rN rO r r r r r � s c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r c
C s6 || _ || _|| _| j� z | j� W d | j� X d S )N)r= r> �server�setup�handle�finish)r r= r> ry r r r r � s zBaseRequestHandler.__init__c C s d S )Nr )r r r r rz � s zBaseRequestHandler.setupc C s d S )Nr )r r r r r{ � s zBaseRequestHandler.handlec C s d S )Nr )r r r r r| � s zBaseRequestHandler.finishN)rM rN rO r rz r{ r| r r r r r � s
c @ s, e Zd Zd ZdZdZdZdd� Zdd� ZdS )
r
rT r NFc C sz | j | _| jd k r | jj| j� | jr:| jjtjtjd� | jj d| j
�| _ | j dkrdt
| j�| _n| jj d| j �| _d S )NT�rbr �wb)r= Z
connectionr1 Z
settimeout�disable_nagle_algorithmrV r0 Z IPPROTO_TCPZ TCP_NODELAY�makefile�rbufsize�rfile�wbufsize�
_SocketWriter�wfile)r r r r rz � s
zStreamRequestHandler.setupc C sF | j js.y| j j� W n tjk
r, Y nX | j j� | jj� d S )N)r� �closed�flushr0 �errorrX r� )r r r r r| s
zStreamRequestHandler.finishr] ) rM rN rO r� r� r1 r rz r| r r r r r
� s
c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
r� c C s
|| _ d S )N)�_sock)r Zsockr r r r s z_SocketWriter.__init__c C s dS )NTr )r r r r �writable s z_SocketWriter.writablec C s& | j j|� t|��}|jS Q R X d S )N)r� Zsendall�
memoryview�nbytes)r �bZviewr r r �write" s
z_SocketWriter.writec C s
| j j� S )N)r� rY )r r r r rY ' s z_SocketWriter.filenoN)rM rN rO r r� r� rY r r r r r� s r� c @ s e Zd Zdd� Zdd� ZdS )r c C s2 ddl m} | j\| _| _|| j�| _|� | _d S )Nr )�BytesIO)�ior� r= Zpacketr0 r� r� )r r� r r r rz . s z DatagramRequestHandler.setupc C s | j j| jj� | j� d S )N)r0 Zsendtor� �getvaluer> )r r r r r| 4 s z DatagramRequestHandler.finishN)rM rN rO rz r| r r r r r * s )$� __version__r0 r% ra �errnorF r � ImportErrorZdummy_threadingr� r r3 r �__all__�hasattr�extendr r# ZSelectSelectorr r r r r r r r r r r r r r r
r� r r r r r �<module>{ sZ
n~ X. .-