CasperSecurity
o
VƧ\�J � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm
Z
d dl m Z d d l m
Z
d d
lmZ d d � ZG d
d� d�ZG dd� de�ZG dd� de�ZG dd� de�ZdZdZdd� ZG dd� de�Zdd � ZdS ) � N)�compact_traceback)� as_string)�notify)�EventRejectedEvent)�ProcessLogStderrEvent)�ProcessLogStdoutEvent)�EventListenerStates)� getEventListenerStateDescription)�loggersc C sD t |�d }|r | �|d |� �s |d8 }|r | �|d |� �r|S )N� )�len�endswith)�haystack�needle�l� r �8/usr/lib/python3/dist-packages/supervisor/dispatchers.py�find_prefix_at_end s
�r c @ s\ e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dS )� PDispatcherz{ Asyncore dispatcher for mainloop, representing a process channel
(stdin, stdout, or stderr). This class is abstract. Fc C s || _ || _|| _d| _d S �NF)�process�channel�fd�closed��selfr r r r r r �__init__ s
zPDispatcher.__init__c C s d| j jt| �| j| jf S )Nz<%s at %s for %s (%s)>)� __class__�__name__�idr r �r r r r �__repr__ s
�zPDispatcher.__repr__c C � t ��N��NotImplementedErrorr r r r �readable& � zPDispatcher.readablec C r" r# r$ r r r r �writable) r' zPDispatcher.writablec C r" r# r$ r r r r �handle_read_event, r' z PDispatcher.handle_read_eventc C r" r# r$ r r r r �handle_write_event/ r' z PDispatcher.handle_write_eventc C s< t � \}}}}| jjjj�dt| �|||f � | �� d S )Nz:uncaptured python exception, closing channel %s (%s:%s %s))r r �config�options�logger�critical�repr�close)r �nil�t�v�tbinfor r r � handle_error2 s �� zPDispatcher.handle_errorc C s0 | j s| jjjj�d| j| f � d| _ d S d S )Nz#fd %s closed, stopped monitoring %sT)r r r+ r, r- �debugr r r r r r0 ? s �
�zPDispatcher.closec C s d S r# r r r r r �flushE r' zPDispatcher.flushN)r �
__module__� __qualname__�__doc__r r r! r& r( r) r* r5 r0 r7 r r r r r s
r c @ st e Zd ZdZdZdZdZdZdZdd� Z dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zdd� Zdd� ZdS )�POutputDispatcheraP
A Process Output (stdout/stderr) dispatcher. Serves several purposes:
- capture output sent within <!--XSUPERVISOR:BEGIN--> and
<!--XSUPERVISOR:END--> tags and signal a ProcessCommunicationEvent
by calling notify(event).
- route the output to the appropriate log handlers as specified in the
config.
FN� c C s� || _ || _|| _| jj | _}| �|j|� t|jd| �}|r/tj| j jj �
� d|d�| _ | j | _
| jj}| jj}|t|�f| _|t|�f| _tjj| _| j j}|j j| jk| _|j| _|j| _dS )z�
Initialize the dispatcher.
`event_type` should be one of ProcessLogStdoutEvent or
ProcessLogStderrEvent
z%s_capture_maxbytes� %(message)s)�fmt�maxbytesN)r �
event_typer r �_setup_loggingr+ �getattrr
�handle_boundIOr, � getLogger�
capturelog�mainlog�childlog� BEGIN_TOKEN� END_TOKENr �begintoken_data�
endtoken_data� LevelsByName�DEBG�
mainlog_level�loglevel�log_to_mainlog�stdout_events_enabled�stderr_events_enabled) r r r@ r r �capture_maxbytes�
begintoken�endtokenr+ r r r r Y s0 � �
zPOutputDispatcher.__init__c C s� t |d| �}|s dS t |d| �}t |d| �}d}|dkr/t�d| t� d�|j|f�}tj|j� � ||| ||d �| _
t |d
| d �rV|jd }t� | j
|� dS dS )
z�
Configure the main log according to the process' configuration and
channel. Sets `mainlog` on self. Returns nothing.
�
%s_logfileN�%s_logfile_maxbytes�%s_logfile_backupsr= �syslogzFSpecifying 'syslog' for filename is deprecated. Use %s_syslog instead.� )�filenamer> �rotatingr? �backupsz %s_syslogFz %(message)s) rB �warnings�warn�DeprecationWarning�join�namer
� handle_filer, rD rF �
handle_syslog)r r+ r �logfiler? r] r> r r r rA } s2 ���
�z POutputDispatcher._setup_loggingc C s: | j | jfD ]}|d ur|jD ]
}|�� |�� qqd S r# )rF rE �handlers�remove�reopen�r �log�handlerr r r �
removelogs� s
��z POutputDispatcher.removelogsc C s2 | j | jfD ]}|d ur|jD ]}|�� qqd S r# )rF rE rf rh ri r r r �
reopenlogs� s
��z POutputDispatcher.reopenlogsc C s� |ro| j j}|jjrt|�}| jr| j�|� | jrHt|t �s"|}nz|�
d�}W n
t y6 d| }Y nw d}|jj j
| j||j| j|d� | jdkr_| jr]tt| j | j j|�� d S d S | jrqtt| j | j j|�� d S d S d S )Nzutf-8�Undecodable: %rz%%(name)r %(channel)s output:
%(data)s)rb r �data�stdout)r r+ r, �
strip_ansi� stripEscapesrG �inforP �
isinstance�bytes�decode�UnicodeDecodeErrorr- rj rN rb r rQ r r �pidrR r )r ro r+ �text�msgr r r �_log� sF
�
�
������zPOutputDispatcher._logc C s� | j d u r| j}d| _| �|� d S | jr| j\}}n| j\}}t| j�|kr)d S | j}d| _z
|�|d�\}}W n) tyb d }t ||�}|r[| j|| d � | _|d | � }| �|� Y n
w | �|� | �
� || _|rw| � � d S d S )Nr< r ) rE �
output_bufferr{ � capturemoderK rJ r �split�
ValueErrorr �toggle_capturemode�
record_output)r ro �token�tokenlen�before�after�indexr r r r� � s8
�
�z POutputDispatcher.record_outputc C s� | j | _ | jd ur\| j r| j| _d S | jjD ]}|�� q| j�� }| j}| jjj }| �
| j| jj |�}t |� d}| jjj
jj|||d� | jjD ]
}|�� |�� qK| j| _d S d S )Nz-%(procname)r %(channel)s emitted a comm event)�procnamer )r} rE rG rf r7 �getvaluer r r+ rb r@ rx r r, r- r6 rg rh rF )r rk ro r r� �eventrz r r r r� � s*
�
�z$POutputDispatcher.toggle_capturemodec C � dS r r r r r r r( r' zPOutputDispatcher.writablec C � | j rdS dS �NFT�r r r r r r& � zPOutputDispatcher.readablec C s< | j jj�| j�}| j|7 _| �� |s | �� d S d S r# )r r+ r, �readfdr r| r� r0 )r ro r r r r) s �z#POutputDispatcher.handle_read_event)r r8 r9 r: r} rF rE rG r| r rA rl rm r{ r� r� r( r& r) r r r r r; H s"
$ $ r; c @ sx e Zd ZdZdZdZdZdZee�Z ee�Z
dd� Z dd � Z d
d � Z
d d
� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )�PEventListenerDispatcherzN An output dispatcher that monitors and changes a process'
listener_state Nr< s READY
s RESULT c C s� t �| |||� tj| j_d | j_d| _d | _t |j
d| �}|rEt |j
d| �}t |j
d| �}t j |j
j
�� |d| ||d�| _d S d S )Nr< rV rW rX r= )r\ r? r] )r r r � ACKNOWLEDGEDr �listener_stater� �result� resultlenrB r+ r
rc r, rD rG )r r r r re r? r] r r r r s$
��z!PEventListenerDispatcher.__init__c C s0 | j d ur| j jD ] }|�� |�� q d S d S r# )rG rf rg rh �r rk r r r rl 7 s
�z#PEventListenerDispatcher.removelogsc C s( | j d ur| j jD ]}|�� q d S d S r# )rG rf rh r� r r r rm = s
�z#PEventListenerDispatcher.reopenlogsc C r� r r r r r r r( C r' z!PEventListenerDispatcher.writablec C r� r� r� r r r r r& F r� z!PEventListenerDispatcher.readablec C s� | j jj�| j�}|r<| j|7 _| j jj}d|| j|f }| j jjj� |� | j
r;| j jjj r5t |�}| j
�
|� n| �� | �� d S )Nz%r %s output:
%s)r r+ r, r� r � state_bufferrb r r- r6 rG rq rr rs r0 � handle_listener_state_change)r ro r� rz r r r r) K s
� z*PEventListenerDispatcher.handle_read_eventc
C sT | j }|sd S | j}|jj}|j}|tjkrd| _ d S |tjkr[t|�| j k r)d S |�
| j �rD| � tj
� | j }| j |d � | _ d |_n | � tj� d| _ d |_| j rY| �� d S d S |tj
krn| � tj� d| _ d |_d S |tjk�r&| jd u r�|�d�}|dkr�d S | j d |� }| j |d d � | _ || jd � }zt|�| _W nx ty� zt|�}W n
ty� d| }Y nw |jjj�d||f � | � tj� d| _ tt ||j�� d |_Y d S w | jt| j � } | �r | j | j d | � 7 _ | j | d � | _ | jt| j � } | �s | � | j � d | j_d| _ d | _| j �r(| �� d S d S d S )Nr< �
���r rn z%s: bad result line: '%s') r� r r+ rb r� r �UNKNOWNr� r �READY_FOR_EVENTS_LEN�
startswith�READY_FOR_EVENTS_TOKEN�_change_listener_state�READYr� r� �BUSYr� �find�RESULT_TOKEN_START_LEN�intr r rw r, r- r_ r r r� �
handle_result)
r ro r r� �stater� �pos� result_liner� �neededr r r r� _ s�
�
� � �*z5PEventListenerDispatcher.handle_listener_state_changec C s� | j }|jj}|jjj}z| j jj�|j|� |�d| � | � t
j � W d S t yE |�
d| � | � t
j � tt||j�� Y d S |�
d| � | � t
j� tt||j�� Y d S )Nz%s: event was processedz%s: event was rejectedz%s: event caused an error)r r+ rb r, r- �group�result_handlerr� r6 r� r r� � RejectEventr_ r r r� )r r� r r� r- r r r r� � s
z&PEventListenerDispatcher.handle_resultc C sj | j }|jj}|j}d|t|�t|�f }|jjj�|� ||_|tj kr3d| }|jjj�
|� d S d S )Nz %s: %s -> %sz�%s: has entered the UNKNOWN state and will no longer receive events, this usually indicates the process violated the eventlistener protocol) r r+ rb r� r r, r- r6 r r� r_ )r � new_stater r� � old_staterz r r r r� � s �
��z/PEventListenerDispatcher._change_listener_state)r r8 r9 r: rG r� r� �RESULT_TOKEN_STARTr r� r� r rl rm r( r& r) r� r� r� r r r r r� s" W r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
�PInputDispatcherz Input (stdin) dispatcher c C s t �| |||� d| _d S )Nr< )r r � input_bufferr r r r r � s
zPInputDispatcher.__init__c C s | j r| jsdS dS )NTF)r� r r r r r r( � s zPInputDispatcher.writablec C r� r r r r r r r& � r' zPInputDispatcher.readablec C s* | j jj�| j| j�}| j|d � | _d S r# )r r+ r, �writer r� )r �sentr r r r7 � s �zPInputDispatcher.flushc
C sb | j r/z| �� W d S ty. } z|jd tjkr"d| _ | �� n� W Y d }~d S d }~ww d S )Nr r< )r� r7 �OSError�args�errno�EPIPEr0 )r �whyr r r r* � s
����z#PInputDispatcher.handle_write_eventN) r r8 r9 r: r r( r& r7 r* r r r r r� � s r� s [)� H� f� A� B� C� D� R� s� u� J� K� h� l� p� mc C s� d}d}d}t | �}||k rG|dkr | ||d � tv r d}n |r?| �t|�}|dkr3|| |d� S || ||� }|}d}|d7 }||k s|S )z>
Remove all ANSI color escapes from the given string.
r< r r r� N)r �ANSI_TERMINATORSr� �ANSI_ESCAPE_BEGIN)�sr� �show�i�L�nr r r rr � s" � rr c @ s e Zd ZdZdS )r� zY The exception type expected by a dispatcher when a handler wants
to reject an event N)r r8 r9 r: r r r r r� s r� c C s |dkrt |��d S )Ns OK)r� )r� �responser r r �default_handler s �r� )r^ r� � supervisor.medusa.asyncore_25r �supervisor.compatr �supervisor.eventsr r r r �supervisor.statesr r �
supervisorr
r r r; r� r� r� r� rr � Exceptionr� r� r r r r �<module> s. 4 N G