File: //lib/python3/dist-packages/future/backports/test/__pycache__/ssl_servers.cpython-310.pyc
o
,�]) � @ 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 e�d�Ze
j� e �Z ej Z e
j� 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+�.� Z/e/j0e_0e/j1r�e$Z2ne#Z2e j3r�e
�4� e2_5ne
�6� e2_5e �7e j8�Z9e9�:e!� e/j;r�e9�<e/j;� e/j=�re9�>e/j=� e"d1e/j?fe2e9�Z@e/j0�reAd2�Be/�� e@�Cd3� dS 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 �| ||� || _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
zHTTPSServer.__init__c C s d| j j| j| jf S )Nz
<%s %s:%s>)� __class__�__name__� server_name� server_port�r r r r �__str__ s ��zHTTPSServer.__str__c
C s\ z| j �� \}}| jj|dd�}W ||fS t jy- } z tjr(tj� d| � � d }~ww )NT)� server_sidezGot an error:
%s
)
�socket�acceptr � wrap_socket�errorr
�verbose�sys�stderr�write)r �sock�addr�sslconn�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 sz t j�|�d }tj�t j�|��}|�d�}td|�}| j }|D ]}tj�
|�\}}tj�|�\}}tj� ||�}q!|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) �urllibr �urlparse�os�path�normpath�unquote�splitr �root�
splitdrive�join)r r6 �words�word�drive�headr r r �translate_path8 s
z'RootedHTTPRequestHandler.translate_pathc G s> t jr tj�d| jj| jj| j� � | �
� || f � d S d S )Nz server (%s:%d %s):
[%s] %s
) r
r$ r% �stdoutr' �serverr r �request�cipher�log_date_time_string�r �format�argsr r r � log_messageL s �
��z$RootedHTTPRequestHandler.log_messageN) r r- r. �server_version�herer: �timeoutrA rJ 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}|�� |�� |�� d�}t�|�}|� d�}| �
d� | � dd� | � dt t
|��� | �� |rA| j�|� dS dS )zServe a GET request.)�
session_cacherE � compressionzutf-8�� z Content-typeztext/plain; charset=utf-8zContent-LengthN)�rfile�raw�_sockr �
session_statsrE rP �pprint�pformat�encode�
send_response� send_headerr �len� end_headers�wfiler' )r � send_bodyr( r �stats�bodyr r r �do_GET^ s
�
�zStatsRequestHandler.do_GETc C s | j dd� dS )zServe a HEAD request.F)r^ N)ra r r r r �do_HEADp s zStatsRequestHandler.do_HEADc G s" t jrtj| |g|�R � d S d S r )r
r$ r � log_requestrG r r r rc t s �z StatsRequestHandler.log_requestN)T)r r- r. �__doc__rK ra rb rc r r r r rN W s
rN c @ s< e Zd Zedfdd�Zdd� Zd dd�Zdd � Zd
d � ZdS )
�HTTPSServerThreadNc C s< d | _ t|df|p
t|�| _| jj| _tj�| � d| _ d S )Nr T)
�flagr r/ rC r �portr �Threadr �daemon)r r �hostr r r r r { s �
zHTTPSServerThread.__init__c C s d| j j| jf S )Nz<%s %s>)r r rC r r r r r � s zHTTPSServerThread.__str__c C s || _ tj�| � d S r )rf r rh �start)r rf r r r rk � s zHTTPSServerThread.startc C s: | j r| j �� z| j�d� W | j�� d S | j�� w )Ng�������?)rf �setrC �
serve_forever� server_closer r r r �run� s
zHTTPSServerThread.runc C s | j �� d S r )rC �shutdownr r r r �stop� s zHTTPSServerThread.stopr ) r r- r. �HOSTr r rk ro rq r r r r re y s
re c sV t �t j�}|�|� t|||�� t�� }� �|� |�� � fdd�}| � |� � S )Nc s8 t jr tj�d� � �� t jrtj�d� � �� d S )Nzstopping HTTPS server
zjoining HTTPS thread
)r
r$ r% rB r' rq r< r �rC r r �cleanup� s z"make_https_server.<locals>.cleanup)
�ssl�
SSLContext�PROTOCOL_SSLv23�load_cert_chainre r �Eventrk �wait�
addCleanup)�case�certfilerj r r rf rt r rs r �make_https_server� s
r~ �__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)�destr� �actionr� z-sz--stats�use_stats_handlerF�
store_truezalways return stats pagez --curve-name�
curve_name�storez&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�������?)D�
__future__r r r r �future.builtinsr r �futurer r5 r% ru rV r �future.backports.urllibr � urllib_parse� future.backports.http.serverr
r r r �future.backports.testr
�
import_moduler r6 �dirname�__file__rL rr r< �CERTFILEr r/ rN rh re r~ r �argparse�ArgumentParser�parser� add_argument�int�
parse_argsrI r$ r� r �PY2�getcwdur: �getcwdrv �PROTOCOL_TLSv1r rx r� �set_ecdh_curver� �load_dh_paramsrg rC �printrH rm r r r r �<module> sx
)"
� � � �
� �
�