File: //usr/local/lib/python3.7/test/__pycache__/ssl_servers.cpython-37.opt-2.pyc
B
��g+ � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ m
Z
m Z d dl m
Z
e j�e�Ze
jZe j�ed�ZG dd� de �ZG dd� de
�ZG d d
� d
e �ZG d d � d ej�Zdeedd
�dd�Zedk�rd dl Z e j dd�Z e j dde ddd� e j ddddd 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,d-� e j d.d/e!d'd0d)� e �"� Z#e#j$e
_$e#j%�r|eZ&neZ&e �'� e&_(e�)ej*j+�Z,e,�-e� e#j.�r�e,�/e#j.� e#j0�r�e,�1e#j0� e#j2�r�e,�3e#j2� ed1e#j4fe&e,�Z5e#j$�re6d2�7e#�� e5�8d3� dS )4� N)�
HTTPServer�SimpleHTTPRequestHandler�BaseHTTPRequestHandler)�supportz keycert.pemc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� HTTPSServerc C s t �| ||� || _d S )N)� _HTTPServer�__init__�context)�self�server_address�
handler_classr � r
�,/usr/local/lib/python3.7/test/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 sf y"| j �� \}}| jj|dd�}W n: tk
r\ } z tjrJtj� d| � � W d d }~X Y nX ||fS )NT)Z server_sidezGot an error:
%s
)
ZsocketZacceptr Z wrap_socket�OSErrorr �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 j�|�d }tj�t j�|��}|�d�}td |�}| j }x:|D ]2}tj�
|�\}}tj�|�\}}tj� ||�}qDW |S )N� �/) �urllib�parseZurlparse�os�path�normpathZunquote�split�filter�root�
splitdrive�join)r
r$ ZwordsZwordZdrive�headr
r
r �translate_path6 s
z'RootedHTTPRequestHandler.translate_pathc G s: t jr6tj�d| 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_messageJ s z$RootedHTTPRequestHandler.log_messageN) r r r �server_version�herer( Ztimeoutr, r2 r
r
r
r r , s
r c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) �StatsRequestHandlerzStatsHTTPS/1.0Tc C s� | j jj}|j}|�� |�� |�� d�}t�|�}|� d�}| �
d� | � dd� | � dt t
|��� | �� |r~| j�|� d S )N)Z
session_cacher/ � compressionzutf-8�� z Content-typeztext/plain; charset=utf-8zContent-Length)Zrfile�rawZ_sockr Z
session_statsr/ r6 �pprintZpformat�encodeZ
send_responseZ send_header�str�lenZ end_headersZwfiler )r
� send_bodyr r ZstatsZbodyr
r
r �do_GET\ s
zStatsRequestHandler.do_GETc C s | j dd� d S )NF)r= )r> )r
r
r
r �do_HEADn s zStatsRequestHandler.do_HEADc G s t jrtj| |f|�� d S )N)r r r � log_request)r
r0 r1 r
r
r r@ r s z StatsRequestHandler.log_requestN)T)r r r r3 r> r? r@ r
r
r
r r5 U s
r5 c @ s< e Zd Zedfdd�Zdd� Zd dd�Zdd � Zd
d � ZdS )
�HTTPSServerThreadNc C s< d | _ t|df|pt|�| _| jj| _tj�| � d| _ d S )Nr T)
�flagr r r. r �port� threading�Threadr Zdaemon)r
r �hostr r
r
r r y s
zHTTPSServerThread.__init__c C s d| j j| jf S )Nz<%s %s>)r r r. )r
r
r
r r � s zHTTPSServerThread.__str__c C s || _ tj�| � d S )N)rB rD rE �start)r
rB r
r
r rG � s zHTTPSServerThread.startc C s2 | j r| j �� z| j�d� W d | j�� X d S )Ng�������?)rB �setr. �
serve_foreverZ server_close)r
r
r
r �run� s
zHTTPSServerThread.runc C s | j �� d S )N)r. Zshutdown)r
r
r
r �stop� s zHTTPSServerThread.stop)N) r r r �HOSTr r rG rJ rK r
r
r
r rA w s
rA )r �certfilerF r c s` |d krt �t jj�}|�|� t|||�� t�� }� �|� |� � � fdd�}| �
|� � S )Nc s8 t jrtj�d� � �� t jr,tj�d� � �� d S )Nzstopping HTTPS server
zjoining HTTPS thread
)r r r r- r rK r* r
)r. r
r �cleanup� s z"make_https_server.<locals>.cleanup) �ssl�create_default_context�Purpose� CLIENT_AUTH�load_cert_chainrA rD ZEventrG �waitZ
addCleanup)Zcaser rM rF r rB rN r
)r. r �make_https_server� s
rU �__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 T� store_falsezbe less verbose)�destrY �actionrZ z-sz--stats�use_stats_handlerF�
store_truezalways return stats pagez --curve-name�
curve_name�storez&curve name for EC-based Diffie-Hellman)r\ rX r] rZ z --ciphers�cipherszallowed cipher list)r\ rX rZ z--dh�dh_filez!PEM file containing DH parameters� z'Listening on https://localhost:{0.port}g�������?)9r# r rO r9 rD Z urllib.parser! Z http.serverr r r r Ztestr r$ �dirname�__file__r4 rL r* ZCERTFILEr r r5 rE rA rU r �argparse�ArgumentParser�parser� add_argument�intr; �
parse_argsr1 r r^ r �getcwdr( rP rQ rR r rS r` Zset_ecdh_curverc Zload_dh_paramsrb Z set_ciphersrC r. �printr0 rI r
r
r
r �<module> sf )"