CasperSecurity
o
���\v � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z
dZ d Z e e Z
G d
d � d e�ZG d d
� d
e
j�ZdS )� N)�as_bytes)� as_string)� encodestring)�PY2)�urlparse)� asynchat_25�
�
c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�Listenerc C � d S �N� )�self�url�statusr
r
�8/usr/lib/python3/dist-packages/supervisor/http_client.pyr � zListener.statusc C s t j�d||f � d S )Nz%s %s
)�sys�stderr�write)r r �errorr
r
r r s zListener.errorc C r r r
)r r �name�valuer
r
r �response_header r zListener.response_headerc C r r r
�r r r
r
r �done r z
Listener.donec C s� zt |�}W n
ty d| }Y nw ztj�|� W n ty8 tr,tj�|� n
dtjj }t|��Y nw tj� � d S )NzUndecodable: %rz<Unable to write Unicode to stdout because it has encoding %s)
r �UnicodeDecodeErrorr �stdoutr �UnicodeEncodeErrorr �encoding�
ValueError�flush)r r �data�sdata�sr
r
r �feed s" � ���
z
Listener.feedc C r r r
r r
r
r �close4 r zListener.closeN) �__name__�
__module__� __qualname__r r r r r% r& r
r
r
r r
s r
c @ s� e Zd Z d'dd�Zd(dd�Zdd� Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS ))� HTTPHandler� Nc C st t j�| ||� || _d| _d| _| �t� d| _| j | _
d| _ d| _ d| _
d| _d | _|| _|| _d | _d| _d S )NzSupervisor HTTP Client� r F)�asynchat�
async_chat�__init__�listener�
user_agent�buffer�set_terminator�CRLF� connected� status_line�part�
chunk_size�
chunk_read� length_read�lengthr �username�passwordr �
error_handled)r r0 r< r= �conn�mapr
r
r r/ 8 s
zHTTPHandler.__init__c
C s� | j d ur td��|| | _ t�| j �\}}}}}}|dvr t�|| _d|v r4|�dd�\} }
t|
�}
n|} d}
|| _|
| _|dkrV| } | � t
j t
j � | �
| | jf� d S |dkro|dd � } | � t
jt
j � | �
| � d S d S ) NzAlready doing a get)�http�unix�:� �P rA rB � )r �AssertionErrorr �NotImplementedError�host�split�int�path�port�
create_socket�socket�AF_INET� SOCK_STREAM�connect�AF_UNIX)
r � serverurlrL �schemerI � path_ignored�params�query�fragment�hostnamerM �ip�
socketnamer
r
r �getQ s2
�
�zHTTPHandler.getc C s0 | j �| j� d| _| �� | j�� d| _d S )Nr �CLOSED)r0 r& r r5 � del_channelrO �r r
r
r r& m s
zHTTPHandler.closec C s | � d||f � | � t� d S )Nz%s: %s)�pushr4 �r r r r
r
r �headert s zHTTPHandler.headerc C sV | j rd S t�� \}}}d||f }| j�| j|� | j| _| � � d| _ ~~~d S )NrD z Cannot connect, error: %s (%s)T)
r> r5 r �exc_infor0 r r �ignorer7 r& )r �t�v�tb�msgr
r
r � handle_errorx s zHTTPHandler.handle_errorc C s� d| _ d}d}| �d|| j|f � | �t� | �d| j� | �dd� | �dd � | �d
| j� | jrNd | j| jf }t t
t |���� � }| �d d
| � | �t� | �t� d S )NrD �GETzHTTP/1.1z%s %s %s�HostzAccept-Encoding�chunked�Acceptz*/*z
User-agentz%s:%s�
AuthorizationzBasic %s)
r5 ra rL r4 rc rI r1 r= r< r r r �strip)r �method�version�authr
r
r �handle_connect� s
zHTTPHandler.handle_connectc C s | j �| j|� d S r )r0 r% r )r r" r
r
r r% � s zHTTPHandler.feedc C s2 | j | | _ | j| jkr| �| j � d| _ d S d S �Nr, )r2 r7 �bodyr% )r �bytesr
r
r �collect_incoming_data� s
�z!HTTPHandler.collect_incoming_datac C s | � � d| _d S ru )r7 r2 r` r
r
r �found_terminator� s
z HTTPHandler.found_terminatorc C s
d| _ d S ru )r2 r` r
r
r re � s
zHTTPHandler.ignorec C s� | j }|�d d�\}}}t|�}|�d�st|��| j�| j|� |dkr*| j| _ n| j
| _ d| }| j� | j|� | � � |||fS )N� s HTTP/�� zCannot read, status code %s)
r2 rJ rK �
startswithr r0 r r �headersr7 re r r& )r �linerr r �reasonri r
r
r r6 � s
zHTTPHandler.status_linec C s� | j }|s | jdkr| j| _d S | j| _| �| j� d S |�dd�\}}|rI|rK|�� }|� � }|dkr8|| _n |dkrAt
|�| _| � ||� d S d S d S )Ns chunked� :rD s transfer-encodings content-length) r2 r � chunked_sizer7 rv r3 r; rJ �lowerrp rK r )r r~ r r r
r
r r} � s
�zHTTPHandler.headersc C s | j �| j||� d S r )r0 r r rb r
r
r r � s zHTTPHandler.response_headerc C s | � � | �� d S r )r r& r` r
r
r rv � s zHTTPHandler.bodyc C s | j �| j� d S r )r0 r r r` r
r
r r � s zHTTPHandler.donec C sV | j }|sd S t|�� d d�}|dkr| j| _n | �|� | j| _| j|7 _d S )Nr � )r2 rK rJ �trailerr7 r3 � chunked_bodyr; )r r~ r8 r
r
r r� � s
zHTTPHandler.chunked_sizec C s& | j }| �t� | j| _| �|� d S r )r2 r3 r4 r� r7 r% �r r~ r
r
r r� � s
zHTTPHandler.chunked_bodyc C s& | j }|tkr| �� | �� d S d S r )r2 r4 r r& r� r
r
r r� � s
�zHTTPHandler.trailer)r+ NNN)r+ )r' r( r) r/ r] r& rc rj rt r% rx ry re r6 r} r rv r r� r� r� r
r
r
r r* 7 s.
�
r* )r rO �supervisor.compatr r r r r �supervisor.medusar r- �CR�LFr4 �objectr
r. r* r
r
r
r �<module> s &