CasperSecurity
o
O0�^�^ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dlm
Z
d dlm Z d dlm Z d dl
mZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl m Z d dl!m"Z" G dd� d�Z#G dd� d�Z$G dd� d�Z%G d d � d e%�Z&G d d � d e%�Z'G d d!� d!�Z(d"e'd#�d$e&d#�de(d#�d%�Z)G d&d'� d'�Z*dS )(� N)�
templating)�urllib)�urlparse)� as_string)�PY2)�unicode)� producers)� http_date)�
get_header)� collector)�
ProcessStates�� NOT_DONE_YET)�VERSION)�
make_namespec)�split_namespec)�SystemNamespaceRPCInterface)�RootRPCInterface)�Faults)�RPCError)� SupervisorNamespaceRPCInterfacec @ s6 e Zd ZdZe�dej�Zdd� Zdd� Z dd� Z
d S )
�DeferredWebProducerz� A medusa producer that implements a deferred callback; requires
a subclass of asynchat.async_chat that handles NOT_DONE_YET sentinel zConnection: (.*)c C s" || _ || _d| _t|j�| _d S )NF)�callback�request�finished�float�delay)�selfr r � r �0/usr/lib/python3/dist-packages/supervisor/web.py�__init__' s z DeferredWebProducer.__init__c C sn | j rdS z| �� }|tu rtW S d| _ | �|�W S t�� }| jjjj �
d|� d| _ | j� d� Y d S )N� TzWeb interface errori� ) r r r � sendresponse� traceback�
format_excr �channel�server�logger�log�error)r �response�tbr r r �more- s zDeferredWebProducer.morec
C s& |� di �}|D ] }|| | j|< qd| jvr d| jd< |� d�r.d| jd< | j�d� d S |� dd �}t|�| jd< | j�|� t| j| jj�}d}d}| jjd
kri|d krf| j� d�s`d }nAd
| jd< n;d }n8| jjdkr�|dkrvd }n+d| jvr�d| jv r�| jd dks�d }n| jj
r�d| jd< d }n d }n| jjd u r�d }t � | j�
� �}|r�d| jd< |r�t �t �| jj��} t �|| g�} n-tr�t| jj�dkr�| jjd }t|t�r�t � |�| jjd< | jj�d|� t �| jj�} | jj�t �t �| | jj��� d | jj_|�r| jj�� d S d S )N�headers� Content-Typez
text/plain�Locationr zContent-Lengthi- �bodyr! z1.0z
keep-alive� z
Keep-Alive�
Connectionz1.1�closezTransfer-Encoding�chunked) �getr r) �len�pushr
�
CONNECTION�header�version�has_key� use_chunkedr �simple_producer�build_reply_header�chunked_producer�composite_producer�outgoingr �
isinstancer �insertr% �push_with_producer�globbing_producer�hooked_producerr( �current_request�close_when_done)
r r* r- r9 r0 �
connection�close_it�wrap_in_chunking�outgoing_header�outgoing_producerr r r r" ? s�
�
� �
��
����
�z DeferredWebProducer.sendresponseN) �__name__�
__module__� __qualname__�__doc__�re�compile�
IGNORECASEr8 r r, r" r r r r r " s r c @ � e Zd Zdd� ZdS )� ViewContextc K s | j �|� d S �N)�__dict__�update)r �kwr r r r � s zViewContext.__init__N)rN rO rP r r r r r rV � � rV c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �MeldViewztext/html;charset=utf-8g �?c C sR || _ | j j}tj�|�s tj�tj�t��}tj�||�}t �
|�| _ d | _ d S rW )
�context�template�os�path�isabs�abspath�dirname�__file__�joinr � parse_xml�rootr )r r] r^ �herer r r r � s
zMeldView.__init__c C s\ | � � }|tu r
tS | jj}|d }| j|d< d|d< d|d< t�d�|d< t|�|d< |S ) Nr- r. zno-cache�Pragmaz
Cache-Controlr �Expiresr0 )�renderr r] r* � content_typer �build_http_dater )r r0 r* r- r r r �__call__� s
zMeldView.__call__c C s d S rW r �r r r r rk � � zMeldView.renderc C s
| j �� S rW )rg �clonero r r r rq � �
zMeldView.cloneN) rN rO rP rl r r rn rk rq r r r r r\ � s r\ c @ rU )�TailViewc
C sL | j j}| j j}d|vrd}d }nS|d }d}|�dd�}td|�� r)t|�d nd�}|s1d}n3t|�}z |�|||�}W n% t yc } z|j
t j krQd| }nd |j
|j
f }W Y d }~nd }~ww | �� } | �d
�}
|
�d | � | �d �} | �|� | �d
�} |d ur�| jdt�|�t�tt|���f d� n| �� t| �� �S )N� processnamezNo process name foundr �limit�1024i ������zNo file for %sz#ERROR: unexpected rpc fault [%d] %s�titlez Supervisor tail of process %s�tailbody�refresh_anchorz!tail.html?processname=%s&limit=%s��href)r] � supervisord�formr5 �min�isdigit�intr �readProcessStdoutLogr �coder �NO_FILE�textrq �findmeld�content�
attributesr �quote�str�abs�deparentr �write_xhtmlstring)
r r} r~ �tailrt �offsetru � rpcinterface�erg rx ry rz r r r rk � sL �
���
�� zTailView.renderN)rN rO rP rk r r r r rs � r[ rs c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
StatusViewc C s� |� � }t�t|jjj|jj��}dd| d d�}dd| d d�}dd| d d�}dd | d d�}d
d | d d�}d
d| d d�} |tjkrO||||| g}
|
S |tj tj
tj fv rb|d ||| g}
|
S d d ||| g}
|
S )N�Startz*index.html?processname=%s&action=start)�namer| �target�Restartz,index.html?processname=%s&action=restart�Stopz)index.html?processname=%s&action=stopz Clear Logz-index.html?processname=%s&action=clearlogzTail -f Stdoutz
logtail/%s�_blankzTail -f Stderrzlogtail/%s/stderr) � get_stater r� r �group�configr� r �RUNNING�STOPPED�EXITED�FATAL) r �process�statert �start�restart�stop�clearlog� tailf_stdout� tailf_stderr�actionsr r r �actions_for_process� sL �������
���z StatusView.actions_for_processc C s& |t jkrdS |t jt jfv rdS dS )N�
statusrunning� statuserror�
statusnominal)r r� r� �BACKOFF)r r� r r r �css_class_for_state s
z StatusView.css_class_for_statec
s� | j j}dt|�f}dt|g�f}t||g�}|�r�|dkr'dd� }d|_|S |dkr;|j�� ��fdd �}d|_|S |d
krV|j� d d id d
ig���fdd�} d| _| S ��r��fdd�}
d|
_t
��\} } |j � | �}
|
d u rt|
S |
j
� | �}|d u r�|
S |dkr�z|j���� W nS ty� } zG|jtjkr�d�n,|jtjkr�d�n#|jtjkr�d�n|jtjkr�d�n|jtjkr�d�nd|j|jf ���fdd�}d|_|W Y d }~S d }~ww t� �r� �fdd �}d|_|S �fd d �}d|_|S |d k�rKz|j���� W n& t�y. } zd|j|jf ��fd d!�}d|_|W Y d }~S d }~ww t� ��r@� �fd"d#�}d|_|S �fd$d%�}d|_|S |d&k�r~|j� d'�gd(�d)�gd(�g�}t|��rs|���fd*d+�}d|_|S �fd,d-�}d|_|S |d.k�r�z|j����W n& t�y� } zd|j|jf ��fd/d0�}d|_|W Y d }~S d }~ww �fd1d2�}d|_|S t|��)3N�
supervisor�system�refreshc S s dt �� } | S )NzPage refreshed at %s)�time�ctime)�messager r r � donothing. s z+StatusView.make_callback.<locals>.donothingg�������?�stopallc s � � t u rt S dt�� S )NzAll stopped at %s�r r� r� r �r r r r� 6 s
z)StatusView.make_callback.<locals>.stopall�
restartall�
methodNamezsupervisor.stopAllProcessesz supervisor.startAllProcessesc s � � } | t u r t S dt�� S )NzAll restarted at %sr� )�resultr� r r r� B s z,StatusView.make_callback.<locals>.restartallc � d� S )NzNo such process named %sr r ��namespecr r �wrongK � z'StatusView.make_callback.<locals>.wrongr� z no such filezfile not executablezalready started� spawn error�abnormal termination� unexpected rpc fault [%d] %sc s d�� f S )N�ERROR: Process %s: %sr r )�msgr� r r �starterri s z*StatusView.make_callback.<locals>.starterrc
s� z� � } W n2 t y7 } z&|jtjkrd}n|jtjkr d}nd|j|jf }d�|f W Y d }~S d }~ww | tu r>tS d� S )Nr� r� r� r� �Process %s started)r r� r � SPAWN_ERROR�ABNORMAL_TERMINATIONr� r )r� r� r� ��bool_or_callbackr� r r � startprocesso s
���
z.StatusView.make_callback.<locals>.startprocessc r� )Nr� r r r� r r � startdone� r� z+StatusView.make_callback.<locals>.startdoner� c � � S rW r r �r� r r �stoperr� rp z)StatusView.make_callback.<locals>.stoperrc
sT z� � } W n t y } zd|j|jf W Y d }~S d }~ww | tu r&tS d� S )Nr� �Process %s stopped)r r� r� r )r� r� r� r r � stopprocess� s
���z-StatusView.make_callback.<locals>.stopprocessc r� )Nr� r r r� r r �stopdone� r� z*StatusView.make_callback.<locals>.stopdoner� zsupervisor.stopProcess)r� �paramszsupervisor.startProcessc s � � } | t u r t S d� S �NzProcess %s restartedr
)�results)r r� r r �restartprocess� s z0StatusView.make_callback.<locals>.restartprocessc r� r� r r r� r r � restartdone� r� z-StatusView.make_callback.<locals>.restartdoner� c r� rW r r r� r r �clearerr� rp z*StatusView.make_callback.<locals>.clearerrc r� )NzLog for %s clearedr r r� r r r� � r� z*StatusView.make_callback.<locals>.clearlog) r] r} r r r r r� �stopAllProcessesr� � multicallr �process_groupsr5 � processes� startProcessr r� r r� �NOT_EXECUTABLE�ALREADY_STARTEDr� r� r� �callable� stopProcess�clearProcessLogs�
ValueError)r r� �actionr} �mainr� r� r� r� r� r� �
group_name� process_namer� r� r� r� r� r� r� r� r� �results_or_callbackr� r� r� r� r )r� r r� r� r �
make_callback! s�
��
� ���
� ��
����
� �� zStatusView.make_callbackc
C s | j j}| j j}|�d�}|�d�}|�d�}|rH| js%| �||�| _tS | �� }|tu r/tS |d urH|d }|d dt�|� }||d d< | j j }t
d t |�fg�} g }
|j �
� D ]} | j�� D ] } |
�| jj| f� qcq\|
�� g }
|
D ],\}}| �|j | j| �}t||�}| j�|�}|
�|d
||||d |d d
�� qx| �� }|d ur�|�d�}d|jd< |�|� |
�rb|�d�� |
�}d}|D ]�\}}|�d�}|�|d � � � | � |d �|jd< |�d�}|�|d � |�d�}t|d |d �}|j dt�|� d� |�|� |d }|�d �}|� |�D ]5\} } | �d �}| d u �r4d |jd< �q |j | d | d d � |�| d � | d! �rS|j | d! d"� �q |�r]d#|jd< | }q�n
|�d$�} | � d%� |�d&��t!� t"t#j$�%� j&�} |�d'��| � t'|�(� �S )(Nrt r� r� �
SERVER_URL�/z ?message=%sr- r/ r� � statenamer� � description)�statusr� r� r� r� r� �
statusmessage�
status_msg�class�trF� status_textr� � info_text� name_anchorr� r� ztail.html?processname=%sr{ r� �
actionitem_td�actionitem_anchor�hiddenr| )r| r� r� )r� �shade� statustablezNo programs to manage�supervisor_version�copyright_date))r] r~ r* r5 r r� r r r� r} r r r� �valuesr� �keys�appendr� r� �sortr� r r� �getProcessInforq r� �attribr� �repeat�lowerr� r� �replacer r� �datetime�date�today�yearr r� ) r r~ r* rt r� r� �
server_url�locationr} r� � processnamesr� � gprocname�data� groupnamer� � sent_name�inforg �
statusarea�iterator� shaded_tr�
tr_element�itemr� r� �anchorr� �
li_element�
actionitem�table�copyright_yearr r r rk � s�
� ��� �
�
�
�
�
�
�
zStatusView.renderN)rN rO rP r� r� r� rk r r r r r� � s + .r� c @ s e Zd ZdZdd� Zdd� ZdS )�OKViewr c C �
|| _ d S rW )r] )r r] r r r r 6 rr zOKView.__init__c C s ddiS )Nr0 �OKr ro r r r rn 9 r� zOKView.__call__N)rN rO rP r r rn r r r r r 4 s r zui/status.html)r^ �viewz ui/tail.html)�
index.htmlz tail.htmlzok.htmlc @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �supervisor_ui_handlerz&Supervisor Web UI HTTP Request Handlerc C r rW )r} )r r} r r r r O rr z supervisor_ui_handler.__init__c C sf |j dvrdS |�� \}}}}|�d�r |dd � }|�d�s|s#d}t�� D ] }||kr0 dS q'd S )N)�POST�GETFr� r1 r! T)�command� split_uri�
startswith�VIEWSr� )r r r` r� �query�fragment�viewnamer r r �matchR s
� ��zsupervisor_ui_handler.matchc C s* |j dkr
t| |�|_d S | �d|� d S )Nr# r! )r% r �continue_request)r r r r r �handle_requestb s
z$supervisor_ui_handler.handle_requestc C s" i }|� � }|�|� d|vrd|d< |d }t�|�}t�|�}|�� D ]
\}} | d ||< q%|D ]\}} | ||< q2|�� |d< |d }
|
r[|
d dkr[|
dd � }
|
r[|
d dksM|
s_d}
t�|
�} | d u rjd S d i i} | d
}
| d }t |||| | j
d �}|
|�}|j j }|t
||�� d S )
N� QUERY_STRINGr! r r� � PATH_INFOr� r1 r! r- r r^ )r^ r r~ r* r} )�cgi_environmentrY r � parse_qsl�parse_qs�items�get_server_urlr( r5 rV r} r% rD r )r r r r~ �cgi_envr) �form_urlencoded�
query_data�k�vr` �viewinfor* � viewclass� viewtemplater] r � pushproducerr r r r- h sD
�
�z&supervisor_ui_handler.continue_requestN)rN rO rP �IDENTr r, r. r- r r r r r" L s r" )+r_ rR r� r# r r� r �supervisor.compatr r r r r �supervisor.medusar � supervisor.medusa.http_serverr r
� supervisor.medusa.xmlrpc_handlerr �supervisor.processr �supervisor.httpr �supervisor.optionsr r r �supervisor.xmlrpcr r r r �supervisor.rpcinterfacer r rV r\ rs r� r r( r" r r r r �<module> sX w". I
����