U
    ,])                     @   s>  d dl mZmZmZmZ d dlmZmZ d dlm	Z	 d dl
Z
d dlZd dlZd dlZd dlZd dlmZ d dlmZmZmZ d dlmZ edZe
jeZejZe
j ed	Z!G d
d deZ"G dd deZ#G dd deZ$G dd dej%Z&e!edfddZ'e(dkr:d dl)Z)e)j*ddZ+e+j,dde-ddd e+j,ddddd d!d" e+j,d#d$d%d&d'd(d" e+j,d)d*ed+d,d- e+j,d.d/ed+d0d- e+. Z/e/j0e_0e/j1re$Z2n"e#Z2e	j3re
4 e2_5n
e
6 e2_5e7ej8Z9e9:e! e/j;re9<e/j; e/j=re9>e/j= e"d1e/j?fe2e9Z@e/j0r0eAd2Be/ e@Cd3 dS )4    )absolute_importdivisionprint_functionunicode_literals)filterstr)utilsN)parse)
HTTPServerSimpleHTTPRequestHandlerBaseHTTPRequestHandler)support	threadingzkeycert.pemc                   @   s$   e Zd Zdd Zdd Zdd ZdS )HTTPSServerc                 C   s   t | || || _d S N)_HTTPServer__init__context)selfserver_addresshandler_classr    r   C/usr/lib/python3/dist-packages/future/backports/test/ssl_servers.pyr      s    zHTTPSServer.__init__c                 C   s   d| j j| j| jf S )Nz
<%s %s:%s>)	__class____name__Zserver_nameserver_portr   r   r   r   __str__   s    zHTTPSServer.__str__c              
   C   sh   z"| j  \}}| jj|dd}W n< t jk
r^ } ztjrLtj	d|   W 5 d }~X Y nX ||fS )NT)Zserver_sidezGot an error:
%s
)
socketZacceptr   Zwrap_socketerrorr   verbosesysstderrwrite)r   sockZaddrZsslconner   r   r   get_request"   s    zHTTPSServer.get_requestN)r   
__module____qualname__r   r   r&   r   r   r   r   r      s   r   c                   @   s(   e Zd ZdZeZdZdd Zdd ZdS )RootedHTTPRequestHandlerzTestHTTPS/1.0   c                 C   sz   t j|d }tjt j|}|d}td|}| j	}|D ]2}tj
|\}}tj|\}}tj||}qB|S )zTranslate a /-separated PATH to the local filename syntax.

        Components that mean special things to the local file system
        (e.g. drive or directory names) are ignored.  (XXX They should
        probably be diagnosed.)

           /N)Zurllibr	   ZurlparseospathnormpathZunquotesplitr   root
splitdrivejoin)r   r.   ZwordsZwordZdriveheadr   r   r   translate_path8   s    	

z'RootedHTTPRequestHandler.translate_pathc              	   G   s:   t jr6tjd| jj| jj| j	 | 
 || f  d S )Nz server (%s:%d %s):
   [%s] %s
)r   r    r!   stdoutr#   serverr   r   ZrequestcipherZlog_date_time_stringr   formatargsr   r   r   log_messageL   s    z$RootedHTTPRequestHandler.log_messageN)	r   r'   r(   server_versionherer1   Ztimeoutr5   r<   r   r   r   r   r)   .   s
   r)   c                   @   s.   e Zd ZdZdZdddZdd Zdd	 Zd
S )StatsRequestHandlerzSExample HTTP request handler which returns SSL statistics on GET
    requests.
    zStatsHTTPS/1.0Tc                 C   s   | j jj}|j}| | | d}t|}|	d}| 
d | dd | dtt| |   |r~| j| dS )zServe a GET request.)Zsession_cacher8   compressionzutf-8   zContent-typeztext/plain; charset=utf-8zContent-LengthN)ZrfilerawZ_sockr   Zsession_statsr8   r@   pprintZpformatencodeZsend_responseZsend_headerr   lenZend_headersZwfiler#   )r   	send_bodyr$   r   ZstatsZbodyr   r   r   do_GET^   s    



zStatsRequestHandler.do_GETc                 C   s   | j dd dS )zServe a HEAD request.F)rF   N)rG   r   r   r   r   do_HEADp   s    zStatsRequestHandler.do_HEADc                 G   s   t jrtj| |f|  d S r   )r   r    r   log_requestr9   r   r   r   rI   t   s    zStatsRequestHandler.log_requestN)T)r   r'   r(   __doc__r=   rG   rH   rI   r   r   r   r   r?   W   s
   
r?   c                   @   s<   e Zd ZedfddZdd ZdddZdd	 Zd
d ZdS )HTTPSServerThreadNc                 C   s<   d | _ t|df|pt|| _| jj| _tj|  d| _	d S )Nr   T)
flagr   r)   r7   r   portr   Threadr   Zdaemon)r   r   hostr   r   r   r   r   {   s    
zHTTPSServerThread.__init__c                 C   s   d| j j| jf S )Nz<%s %s>)r   r   r7   r   r   r   r   r      s    zHTTPSServerThread.__str__c                 C   s   || _ tj|  d S r   )rL   r   rN   start)r   rL   r   r   r   rP      s    zHTTPSServerThread.startc                 C   s2   | j r| j   z| jd W 5 | j  X d S )Ng?)rL   setr7   Zserver_closeserve_foreverr   r   r   r   run   s
    
zHTTPSServerThread.runc                 C   s   | j   d S r   )r7   Zshutdownr   r   r   r   stop   s    zHTTPSServerThread.stop)N)	r   r'   r(   HOSTr   r   rP   rS   rT   r   r   r   r   rK   y   s
   	
rK   c                    sV   t t j}|| t||| t } | |   fdd}| 	|  S )Nc                      s8   t jrtjd    t jr,tjd    d S )Nzstopping HTTPS server
zjoining HTTPS thread
)r   r    r!   r6   r#   rT   r3   r   r7   r   r   cleanup   s    z"make_https_server.<locals>.cleanup)
ssl
SSLContextZPROTOCOL_SSLv23load_cert_chainrK   r   ZEventrP   waitZ
addCleanup)ZcaseZcertfilerO   r   r   rL   rW   r   rV   r   make_https_server   s    


r\   __main__zERun a test HTTPS server. By default, the current directory is served.)Zdescriptionz-pz--portiQ  z(port to listen on (default: %(default)s))typedefaulthelpz-qz--quietr    TZstore_falsezbe less verbose)destr_   actionr`   z-sz--statsuse_stats_handlerF
store_truezalways return stats pagez--curve-name
curve_nameZstorez&curve name for EC-based Diffie-Hellman)ra   r^   rb   r`   z--dhdh_filez!PEM file containing DH parameters z'Listening on https://localhost:{0.port}g?)DZ
__future__r   r   r   r   Zfuture.builtinsr   r   Zfuturer   r-   r!   rX   rC   r   Zfuture.backports.urllibr	   Zurllib_parseZfuture.backports.http.serverr
   r   r   r   Zfuture.backports.testr   import_moduler   r.   dirname__file__r>   rU   r3   ZCERTFILEr   r)   r?   rN   rK   r\   r   argparseArgumentParserparseradd_argumentint
parse_argsr;   r    rc   r   ZPY2Zgetcwdur1   getcwdrY   ZPROTOCOL_TLSv1r   rZ   re   Zset_ecdh_curverf   Zload_dh_paramsrM   r7   printr:   rR   r   r   r   r   <module>   sx   
)"
  


