File: //lib/python3.6/site-packages/future/backports/test/__pycache__/ssl_servers.cpython-36.opt-1.pyc
3
�c) � @ s> d dl mZmZmZmZ d dlmZmZ d dlm Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlmZ d dlmZmZmZ d dlmZ ejd�Ze
jj e �Z ej Z e
jj e d �Z!G d
d � d e�Z"G d d
� d
e�Z#G dd� de�Z$G dd� dej%�Z&e!e dfdd�Z'e(dk�r:d dl)Z)e)j*dd�Z+e+j,dde-ddd� e+j,d d d d d d!d"� e+j,d#d$d%d&d'd(d"� e+j,d)d*ed+d,d-� e+j,d.d/ed+d0d-� e+j.� Z/e/j0e_0e/j1�r�e$Z2n"e#Z2e j3�r�e
j4� e2_5n
e
j6� e2_5e j7e j8�Z9e9j:e!� e/j;�r�e9j<e/j;� e/j=�re9j>e/j=� e"d1e/j?fe2e9�Z@e/j0�r0eAd2jBe/�� e@jCd3� dS )4� )�absolute_import�division�print_function�unicode_literals)�filter�str)�utilsN)�parse)�
HTTPServer�SimpleHTTPRequestHandler�BaseHTTPRequestHandler)�support� threadingz keycert.pemc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� HTTPSServerc C s t j| ||� || _d S )N)� _HTTPServer�__init__�context)�self�server_address�
handler_classr � r �!/usr/lib/python3.6/ssl_servers.pyr s zHTTPSServer.__init__c C s d| j j| j| jf S )Nz
<%s %s:%s>)� __class__�__name__Z server_name� server_port)r r r r �__str__ s zHTTPSServer.__str__c C sh y"| j j� \}}| jj|dd�}W n< t jk
r^ } z tjrLtjj d| � � W Y d d }~X nX ||fS )NT)Z server_sidezGot an error:
%s
)
�socketZacceptr Z wrap_socket�errorr
�verbose�sys�stderr�write)r �sockZaddrZsslconn�er r r � get_request" s zHTTPSServer.get_requestN)r �
__module__� __qualname__r r r$ r r r r r s r c @ s( e Zd ZdZeZdZdd� Zdd� ZdS )�RootedHTTPRequestHandlerz
TestHTTPS/1.0� c C s~ t jj|�d }tjjt jj|��}|jd�}td|�}| j }x:|D ]2}tjj
|�\}}tjj|�\}}tjj ||�}qDW |S )z�Translate 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) Zurllibr Zurlparse�os�path�normpathZunquote�splitr �root�
splitdrive�join)r r, ZwordsZwordZdrive�headr r r �translate_path8 s
z'RootedHTTPRequestHandler.translate_pathc G s: t jr6tjjd| jj| jj| jj � | j
� || f � d S )Nz server (%s:%d %s):
[%s] %s
) r
r r �stdoutr! �serverr r Zrequest�cipherZlog_date_time_string)r �format�argsr r r � log_messageL s z$RootedHTTPRequestHandler.log_messageN) r r% r& �server_version�herer/ Ztimeoutr3 r9 r r r r r' . s
r' c @ s. e Zd ZdZdZd dd�Zdd� Zdd � Zd
S ) �StatsRequestHandlerzSExample HTTP request handler which returns SSL statistics on GET
requests.
zStatsHTTPS/1.0Tc C s� | j jj}|j}|j� |j� |j� d�}tj|�}|j d�}| j
d� | j dd� | j dt t
|��� | j� |r~| jj|� dS )zServe a GET request.)Z
session_cacher6 � compressionzutf-8�� z Content-typeztext/plain; charset=utf-8zContent-LengthN)Zrfile�rawZ_sockr Z
session_statsr6 r= �pprintZpformat�encodeZ
send_responseZ send_headerr �lenZ end_headersZwfiler! )r � send_bodyr" r ZstatsZbodyr r r �do_GET^ s
zStatsRequestHandler.do_GETc C s | j dd� dS )zServe a HEAD request.F)rC N)rD )r r r r �do_HEADp s zStatsRequestHandler.do_HEADc G s t jrtj| |f|�� d S )N)r
r r � log_request)r r7 r8 r r r rF t s z StatsRequestHandler.log_requestN)T)r r% r& �__doc__r: rD rE rF r r r r r< W s
r< c @ s< e Zd Zedfdd�Zdd� Zd dd�Zdd � Zd
d � ZdS )
�HTTPSServerThreadNc C s< d | _ t|df|pt|�| _| jj| _tjj| � d| _ d S )Nr T)
�flagr r' r5 r �portr �Threadr Zdaemon)r r �hostr r r r r { s
zHTTPSServerThread.__init__c C s d| j j| jf S )Nz<%s %s>)r r r5 )r r r r r � s zHTTPSServerThread.__str__c C s || _ tjj| � d S )N)rI r rK �start)r rI r r r rM � s zHTTPSServerThread.startc
C s2 | j r| j j� z| jjd� W d | jj� X d S )Ng�������?)rI �setr5 �
serve_foreverZ server_close)r r r r �run� s
zHTTPSServerThread.runc C s | j j� d S )N)r5 Zshutdown)r r r r �stop� s zHTTPSServerThread.stop)N) r r% r& �HOSTr r rM rP rQ r r r r rH y s
rH c sV t jt j�}|j|� t|||�� tj� }� j|� |j� � fdd�}| j |� � S )Nc s8 t jrtjjd� � j� t jr,tjjd� � j� d S )Nzstopping HTTPS server
zjoining HTTPS thread
)r
r r r4 r! rQ r1 r )r5 r r �cleanup� s z"make_https_server.<locals>.cleanup)
�ssl�
SSLContextZPROTOCOL_SSLv23�load_cert_chainrH r ZEventrM �waitZ
addCleanup)ZcaseZcertfilerL r r rI rS r )r5 r �make_https_server� s
rX �__main__zERun a test HTTPS server. By default, the current directory is served.)� descriptionz-pz--portiQ z(port to listen on (default: %(default)s))�type�default�helpz-qz--quietr TZ store_falsezbe less verbose)�destr\ �actionr] z-sz--stats�use_stats_handlerF�
store_truezalways return stats pagez --curve-name�
curve_nameZstorez&curve name for EC-based Diffie-Hellman)r^ r[ r_ r] z--dh�dh_filez!PEM file containing DH parameters� z'Listening on https://localhost:{0.port}g�������?)DZ
__future__r r r r Zfuture.builtinsr r Zfuturer r+ r rT r@ r Zfuture.backports.urllibr Z urllib_parseZ future.backports.http.serverr
r r r Zfuture.backports.testr
�
import_moduler r, �dirname�__file__r; rR r1 ZCERTFILEr r' r< rK rH rX r �argparse�ArgumentParser�parser� add_argument�int�
parse_argsr8 r r` r ZPY2Zgetcwdur/ �getcwdrU ZPROTOCOL_TLSv1r rV rb Zset_ecdh_curverc Zload_dh_paramsrJ r5 �printr7 rO r r r r �<module> sh
)"