a
    á `+  ã                   @   s  d 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 ddlm	Z	 ddl
mZ ddl
mZ ddlmZ e e¡ZG d	d
„ d
e	jƒZG dd„ dƒZG dd„ deƒZG dd„ deeƒZG dd„ dejƒZG dd„ deeƒZG dd„ deƒZG dd„ dejƒZG dd„ de	jƒZdS )z1Support for standalone client challenge solvers. é    N)ÚBaseHTTPServer)Úhttp_client)Úsocketserver)Ú
challenges)Úcrypto_util)ÚListc                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )Ú	TLSServerzGeneric TLS Server.c                 O   sp   |  dd¡| _| jrtj| _ntj| _|  di ¡| _|  dtj¡| _	|  dd¡| _
tjj| g|¢R i |¤Ž d S )NÚipv6FÚcertsÚmethodÚallow_reuse_addressT)Úpopr	   ÚsocketÚAF_INET6Úaddress_familyÚAF_INETr
   r   Z_DEFAULT_SSL_METHODr   r   r   Ú	TCPServerÚ__init__©ÚselfÚargsÚkwargs© r   ú1/usr/lib/python3/dist-packages/acme/standalone.pyr      s    
ÿzTLSServer.__init__c                 C   s&   t j| j| jt| dd ƒ| jd| _d S )NÚ_alpn_selection)Zcert_selectionZalpn_selectionr   )r   Z	SSLSocketr   Ú_cert_selectionÚgetattrr   ©r   r   r   r   Ú
_wrap_sock"   s
    
ýzTLSServer._wrap_sockc                 C   s   |  ¡ }| j |d¡S )z.Callback selecting certificate for connection.N)Úget_servernamer
   Úget©r   Z
connectionZserver_namer   r   r   r   (   s    zTLSServer._cert_selectionc                 C   s   |   ¡  tj | ¡S ©N)r   r   r   Úserver_bindr   r   r   r   r#   -   s    zTLSServer.server_bindN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r#   r   r   r   r   r      s
   r   c                   @   s   e Zd ZdZdZdZdS )ÚACMEServerMixinz"ACME server common settings mixin.z'ACME client standalone challenge solverTN)r$   r%   r&   r'   Úserver_versionr   r   r   r   r   r(   2   s   r(   c                   @   s0   e Zd ZdZdd„ Zdd„ Zdd„ Zdd	„ Zd
S )ÚBaseDualNetworkedServersa  Base class for a pair of IPv6 and IPv4 servers that tries to do everything
       it's asked for both servers, but where failures in one server don't
       affect the other.

       If two servers are instantiated, they will serve on the same port.
       c           
   
   O   s  |d }g | _ g | _dD ]Ü}z`||d< |d f|f |dd …  }|f| }||i |¤Ž}	t d|d |d |rtdnd¡ W n\ tjyØ   | jr´t d	|d |d |r¬dnd¡ n t d
|d |d |rÎdnd¡ Y q0 | j |	¡ |	j ¡ d }q| jst d¡‚d S )Né   )TFr	   r   é   z$Successfully bound to %s:%s using %sZIPv6ZIPv4z‚Certbot wasn't able to bind to %s:%s using %s, this is often expected due to the dual stack nature of IPv6 socket implementations.z Failed to bind to %s:%s using %szCould not bind to IPv4 or IPv6.)ÚthreadsÚserversÚloggerÚdebugr   ÚerrorÚappendÚgetsockname)
r   ZServerClassÚserver_addressZremaining_argsr   ZportZ
ip_versionZnew_addressÚnew_argsÚserverr   r   r   r   A   s8    	
þ
ûþ
z!BaseDualNetworkedServers.__init__c                 C   s2   | j D ]&}tj|jd}| ¡  | j |¡ qdS )z*Wraps socketserver.TCPServer.serve_forever)ÚtargetN)r.   Ú	threadingZThreadÚserve_foreverÚstartr-   r2   ©r   r6   Úthreadr   r   r   r9   k   s    
ÿz&BaseDualNetworkedServers.serve_foreverc                 C   s   dd„ | j D ƒS )z/Wraps socketserver.TCPServer.socket.getsocknamec                 S   s   g | ]}|j  ¡ ‘qS r   )r   r3   )Ú.0r6   r   r   r   Ú
<listcomp>u   ó    z9BaseDualNetworkedServers.getsocknames.<locals>.<listcomp>)r.   r   r   r   r   Úgetsocknamess   s    z%BaseDualNetworkedServers.getsocknamesc                 C   s:   | j D ]}| ¡  | ¡  q| jD ]}| ¡  q"g | _dS )zpWraps socketserver.TCPServer.shutdown, socketserver.TCPServer.server_close, and
           threading.Thread.joinN)r.   ZshutdownZserver_closer-   Újoinr;   r   r   r   Úshutdown_and_server_closew   s    



z2BaseDualNetworkedServers.shutdown_and_server_closeN)r$   r%   r&   r'   r   r9   r@   rB   r   r   r   r   r*   9   s
   *r*   c                   @   s.   e Zd ZdZdZddd„Zdd„ Zdd	„ Zd
S )ÚTLSALPN01ServerzTLSALPN01 Server.s
   acme-tls/1Fc                 C   s   t j| |t||d || _d S )N)r
   r	   )r   r   Ú_BaseRequestHandlerWithLoggingÚchallenge_certs)r   r4   r
   rE   r	   r   r   r   r   ‡   s
    þzTLSALPN01Server.__init__c                 C   s"   |  ¡ }t d|¡ | j |d ¡S )Nz)Serving challenge cert for server name %s)r   r/   r0   rE   r    r!   r   r   r   r      s    zTLSALPN01Server._cert_selectionc                 C   sB   t |ƒdkr.|d | jkr.t d| j¡ | jS t dt|ƒ¡ dS )z!Callback to select alpn protocol.r+   r   zAgreed on %s ALPNz#Cannot agree on ALPN proto. Got: %sr?   )ÚlenÚACME_TLS_1_PROTOCOLr/   r0   Ústr)r   Z_connectionZalpn_protosr   r   r   r   ™   s
    zTLSALPN01Server._alpn_selectionN)F)r$   r%   r&   r'   rG   r   r   r   r   r   r   r   rC   ‚   s
   
rC   c                   @   s   e Zd ZdZdd„ ZdS )Ú
HTTPServerzGeneric HTTP Server.c                 O   sD   |  dd¡| _| jrtj| _ntj| _tjj| g|¢R i |¤Ž d S )Nr	   F)	r   r	   r   r   r   r   r   rI   r   r   r   r   r   r   §   s
    
zHTTPServer.__init__N©r$   r%   r&   r'   r   r   r   r   r   rI   ¤   s   rI   c                   @   s   e Zd ZdZddd„ZdS )ÚHTTP01ServerzHTTP01 Server.Fé   c                 C   s    t j| |tj||d|d d S )N©Úsimple_http_resourcesÚtimeout)r	   )rI   r   ÚHTTP01RequestHandlerÚpartial_init)r   r4   Z	resourcesr	   rO   r   r   r   r   ³   s    ÿþzHTTP01Server.__init__N)FrL   rJ   r   r   r   r   rK   °   s   rK   c                   @   s   e Zd ZdZdd„ ZdS )ÚHTTP01DualNetworkedServersz`HTTP01Server Wrapper. Tries everything for both. Failures for one don't
       affect the other.c                 O   s   t j| tg|¢R i |¤Ž d S r"   )r*   r   rK   r   r   r   r   r   ½   s    z#HTTP01DualNetworkedServers.__init__NrJ   r   r   r   r   rR   ¹   s   rR   c                   @   s`   e Zd ZdZe dd¡Zdd„ Zdd„ Zdd	„ Z	d
d„ Z
dd„ Zdd„ Zdd„ Zedd„ ƒZdS )rP   zÍHTTP01 challenge handler.

    Adheres to the stdlib's `socketserver.BaseRequestHandler` interface.

    :ivar set simple_http_resources: A set of `HTTP01Resource`
        objects. TODO: better name?

    ÚHTTP01Resourcezchall response validationc                 O   s<   |  dtƒ ¡| _|  dd¡| _tjj| g|¢R i |¤Ž d S )NrN   rO   rL   )r   ÚsetrN   rO   r   ÚBaseHTTPRequestHandlerr   r   r   r   r   r   Í   s    zHTTP01RequestHandler.__init__c                 G   s   t  d| jd || ¡ dS ©zLog arbitrary message.z	%s - - %sr   N©r/   r0   Zclient_address©r   Úformatr   r   r   r   Úlog_messageÒ   s    z HTTP01RequestHandler.log_messagec                 C   s   |   d¡ tj | ¡ dS ©zHandle request.zIncoming requestN)rZ   r   rU   Úhandler   r   r   r   r\   Ö   s    
zHTTP01RequestHandler.handlec                 C   s>   | j dkr|  ¡  n&| j  dtjj ¡r2|  ¡  n|  ¡  d S )Nú/)ÚpathÚhandle_indexÚ
startswithr   ZHTTP01ZURI_ROOT_PATHÚhandle_simple_http_resourceÚ
handle_404r   r   r   r   Údo_GETÛ   s
    


zHTTP01RequestHandler.do_GETc                 C   s6   |   d¡ |  dd¡ |  ¡  | j | jj ¡ ¡ dS )zHandle index page.éÈ   zContent-Typeú	text/htmlN)Úsend_responseÚsend_headerÚend_headersÚwfileÚwriter6   r)   Úencoder   r   r   r   r_   ã   s    
z!HTTP01RequestHandler.handle_indexc                 C   s4   | j tjdd |  dd¡ |  ¡  | j d¡ dS )zHandler 404 Not Found errors.z	Not Found)ÚmessagezContent-typere   s   404N)rf   r   Z	NOT_FOUNDrg   rh   ri   rj   r   r   r   r   rb   ê   s    zHTTP01RequestHandler.handle_404c                 C   sv   | j D ]R}|jj| jkr|  d|j d¡¡ |  tj¡ |  ¡  | j	 
|j ¡ ¡  dS q|  d¡ |  d| j¡ dS )z$Handle HTTP01 provisioned resources.zServing HTTP01 with token %rÚtokenNzNo resources to servez0%s does not correspond to any resource. ignoring)rN   Zchallr^   rZ   rk   rf   r   ZOKrh   ri   rj   Z
validation)r   Úresourcer   r   r   ra   ñ   s    

ÿ
ÿz0HTTP01RequestHandler.handle_simple_http_resourcec                 C   s   t j| ||dS )z¿Partially initialize this handler.

        This is useful because `socketserver.BaseServer` takes
        uninitialized handler and initializes it with the current
        request.

        rM   )Ú	functoolsÚpartial)ÚclsrN   rO   r   r   r   rQ      s    	þz!HTTP01RequestHandler.partial_initN)r$   r%   r&   r'   ÚcollectionsÚ
namedtuplerS   r   rZ   r\   rc   r_   rb   ra   ÚclassmethodrQ   r   r   r   r   rP   Á   s   ÿrP   c                   @   s    e Zd ZdZdd„ Zdd„ ZdS )rD   z BaseRequestHandler with logging.c                 G   s   t  d| jd || ¡ dS rV   rW   rX   r   r   r   rZ     s    z*_BaseRequestHandlerWithLogging.log_messagec                 C   s   |   d¡ tj | ¡ dS r[   )rZ   r   ÚBaseRequestHandlerr\   r   r   r   r   r\     s    
z%_BaseRequestHandlerWithLogging.handleN)r$   r%   r&   r'   rZ   r\   r   r   r   r   rD     s   rD   )r'   rr   ro   Zloggingr   r8   Z	six.movesr   r   r   Zacmer   r   Zacme.magic_typingr   Z	getLoggerr$   r/   r   r   r(   Úobjectr*   rC   rI   rK   rR   rU   rP   ru   rD   r   r   r   r   Ú<module>   s*   
I"	M