3
|��\� � @ s" d Z ddlmZmZ ddlZddlmZmZmZm Z e rzdZ
ddl m Z ddl
mZ dd l
mZ dd
l mZ dd l mZ n4d Z
ddlm Z ddlZdZdd
lmZ dd lmZ y ddlZW n ek
r� dZY nX dZddd�ZG dd� de� r�ejne �Z G dd� de �Z dd� Z dd� Z dS ) z*Defused xmlrpclib
Also defuses gzip bomb
� )�print_function�absolute_importN� )� DTDForbidden�EntitiesForbidden�ExternalReferenceForbidden�PY3z
xmlrpc.client)� ExpatParser)�client)�server)� gzip_decode)�GzipDecodedResponse� xmlrpclib� i c C s� t st�|dkrt}tj| �}t jd|d�}y$|dk r@|j� }n|j|d �}W n tk
rl td��Y nX |j � |j � |dkr�t
|�|kr�td��|S )zrgzip encoded data -> unencoded data
Decode data using the gzip content encoding as described in RFC 1952
N�rb)�mode�fileobjr r z invalid dataz#max gzipped payload length exceeded) �gzip�NotImplementedError�MAX_DATA�io�BytesIO�GzipFile�read�IOError�
ValueError�close�len)�data�limit�fZgzfZdecoded� r! � /usr/lib/python3.6/xmlrpc.py�defused_gzip_decode- s"
r# c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�DefusedGzipDecodedResponsezha file-like object to decode a response encoded with the gzip
method, as described in RFC 1952.
Nc C s� t st�|d k r|nt | _}|dk r6|j� }d | _n|j|d �}d| _|dkrft|�|krftd��tj |�| _
t j j | d| j
d� d S )Nr r zmax payload length exceededr )r r )
r r r r r �
readlengthr r r r �stringior �__init__)�selfZresponser r r! r! r" r' K s z#DefusedGzipDecodedResponse.__init__c C sn | j dkr\| j | j }t||d �}tjj| |�}| jt|�7 _| j| j krXtd��|S tjj| |�S d S )Nr r zmax payload length exceeded)r r% �minr r r r r )r( �n�leftr r! r! r" r \ s
z DefusedGzipDecodedResponse.readc C s t jj| � | jj� d S )N)r r r r& )r( r! r! r" r h s z DefusedGzipDecodedResponse.close)N)�__name__�
__module__� __qualname__�__doc__r' r r r! r! r! r" r$ F s
r$ c @ s6 e Zd Zddd�Zdd� Zdd� Zd d
� Zd d � Zd
S )�DefusedExpatParserFTc C sZ t j| |� || _|| _|| _| j}| jr2| j|_| jrH| j|_ | j
|_ | jrV| j |_
d S )N)r r' �
forbid_dtd�forbid_entities�forbid_externalZ_parser�defused_start_doctype_declZStartDoctypeDeclHandler�defused_entity_declZEntityDeclHandler� defused_unparsed_entity_declZUnparsedEntityDeclHandler�#defused_external_entity_ref_handlerZExternalEntityRefHandler)r( �targetr1 r2 r3 �parserr! r! r" r' n s zDefusedExpatParser.__init__c C s t |||��d S )N)r )r( �name�sysid�pubidZhas_internal_subsetr! r! r" r4 | s z-DefusedExpatParser.defused_start_doctype_declc C s t ||||||��d S )N)r )r( r: Zis_parameter_entity�value�baser; r< �
notation_namer! r! r" r5 s z&DefusedExpatParser.defused_entity_declc C s t |d ||||��d S )N)r )r( r: r> r; r< r? r! r! r" r6 � s z/DefusedExpatParser.defused_unparsed_entity_declc C s t ||||��d S )N)r )r( �contextr> r; r< r! r! r" r7 � s z6DefusedExpatParser.defused_external_entity_ref_handlerN)FTT)r, r- r. r' r4 r5 r6 r7 r! r! r! r" r0 m s
r0 c C s t t_tt_tt_tr tt_d S )N)r0 �
xmlrpc_client�
FastParserr$ r
r# r �
xmlrpc_serverr! r! r! r" � monkey_patch� s
rD c C s d t _tt _tt _tr tt_d S )N)rA rB �_OrigGzipDecodedResponser
�_orig_gzip_decoder rC r! r! r! r" �unmonkey_patch� s
rG i x i �)N)!r/ Z
__future__r r r �commonr r r r Z
__origin__Z
xmlrpc.clientr Zxmlrpcr
rA r rC r rF r
rE r r � ImportErrorr r# r �objectr$ r0 rD rG r! r! r! r" �<module> s4
'