File: //usr/lib/fm-agent/library/__pycache__/agent_util.cpython-39.pyc
a
��i�$ � @ s~ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z z ddl
Z
W n e yr ddl Z
Y n0 z ddl
Z
W n ddlmZ
Y n0 ejd dkr�ddlmZ n ddlmZ dejvr�dejd< e�ej� dd � ZeZi Zd
Zd d � Zd
d� Z dd� Z dZ dZ dZ g d�Z!d(dd�Z"d)dd�Z#d*d d �Z$d d � Z%d d!� Z&d"d#� Z'G d$d%� d%e
j(�Z)G d&d'� d'e*�Z+dS )+zEThis file is a library of some essential classes for our agent's use.� N�3)�BytesIO)�StringIO�LC_ALL�Cc C sR zt �t| ��W S tyL tt�ti �kr>td | � Y S t�| � Y S 0 d S )N�float)�localeZatof�str� Exception�type� __builtins__r )�value� r �'/usr/lib/fm-agent/library/agent_util.py� custom_float& s r �< c C s | D ]}|r dS qdS )NTFr ��iterable�elementr r r �any5 s r c C s | D ]}|s dS qdS )NFTr r r r r �all< s r c C s" | j | j| jd d d d S )N� i i@B )Z microseconds�seconds�days)� timedeltar r r �
total_secondsC s
��r � � )z/usr/local/sbinz/usr/local/binz /usr/sbinz/usr/binz/sbinz/binFc C s� dd� }t j�| �\}}|r*|| �rv| S nLdt jvr8d S t�t jd �t j�t�D ]"}t j�|| �}||�rR| S qR|r�t d| ��nd S d S )Nc S s t j�| �ot �| t j�S �N)�os�path�exists�access�X_OK)�fpathr r r �is_exe\ s zwhich.<locals>.is_exe�PATHz %r not found)
r r �split�environ� itertools�chain�pathsep�ADDITIONAL_SEARCH_PATHS�joinr
)Zprogram�excr% r$ �fnamer Zexe_filer r r �which[ s
�
r0 � c C s t j| d|ddd�}|j�d�S )NFT)�shell�timeout�check�capture_outputzutf-8)�
subprocess�run�stdout�decode)Z command_argsr3 �pr r r � run_commands s
�r; T� c s� |du r| � � } t�d�}|rdt�| �} | rdt�� | �d� |k rd|�d| � | �d�| �d�fS d}
dtj� � k�r,t
j | ||t
j t
j |d �}
|s�dS z|
j
|d
�\} } W nx t
j�y& }
z\|
jg}|r�|�t|
j�� |D ],}zt�|tj� W q� t�y Y q�0 q�W Y d}
~
d S d}
~
0 0 �nG d d
� d
t�� � fdd�}t
j | ||t
j t
j |d �}
|�sldS |dk�r�|dk�r�t�tj|� t�|� z.|
�
� \} } |dk�r�|dk�r�t�d� W nh � �y. |
jg}|�r�|�t|
j�� |D ].}zt�|tj� W n t�y" Y n0 �q�Y d S 0 |
j}| �s@| } | �d�}|�rr|�d| � ||t�� d�t| < |�d| ||f � ||fS )zNRun a command line call with a timeout after which it will be forcibly killed.F�execute_command� timestampz9Retrieved information from the local cache timeout for %s�retcode�outputN�darwin)r2 �cwdr8 �stderr�env)r3 )i����z Timeout exceeded, process killedc @ s e Zd ZdS )z execute_command.<locals>.AlarmN)�__name__�
__module__� __qualname__r r r r �Alarm� s rH c s � �d S r r )�signum�frame�rH r r �
alarm_handler� s z&execute_command.<locals>.alarm_handler���r �utf8zCreated cache for cmd %s)r? r@ r> z %s: %s %s) r' �logging� getLogger�LOCAL_CACHE_RESULTS�get�time�debug�sys�platform�lowerr6 �Popen�PIPE� communicate�TimeoutExpired�pid�extend�get_process_childrenr �kill�signal�SIGKILL�OSErrorr
�SIGALRM�alarm�
returncoder9 )�cmdrB r2 Z kill_treer3 rD �blockZ
cache_timeout�logZ
cached_resultr: r8 rC Zte�pidsr\ rL r? r@ r rK r r= z s�
��� �
�
r= c C s8 t jd| dt jt jd�}|�� \}}dd� |�� D �S )z?Helper method for killing off child processes when they timeoutz ps --no-headers -o pid --ppid %dT)r2 r8 rC c S s g | ] }t |��qS r )�int)�.0r: r r r �
<listcomp>� � z(get_process_children.<locals>.<listcomp>)r6 rX rY rZ r' )r\ r: r8 rC r r r r^ � s � r^ c C s` z&| d d }t |�� �d }|d W S tyZ t�� d d� \}}t�|� |�Y n0 d S )NZNetworkSettingsZNetworksr Z IPAddressr )�list�valuesr
rU �exc_inforO � exception)� containerZnetworksZnetwork�t�er r r �get_container_ip� s
ru c K sB zt j| fi |��W S ty< t j| �� fi |�� Y S 0 d S r )�json�loads� TypeErrorr9 )�val�kwargsr r r �
json_loads s r{ c @ s e Zd Zddd�Zdd� ZdS )�UnixHTTPConnection� localhostc K s t jj| |fi |�� || _d S r )�httplib�HTTPConnection�__init__r )�selfr �hostrz r r r r� s zUnixHTTPConnection.__init__c C s& t � t jt j�}|�| j� || _d S r )�socketZAF_UNIXZ SOCK_STREAM�connectr �sock)r� r� r r r r� s zUnixHTTPConnection.connectN)r} )rE rF rG r� r� r r r r r|
s
r| c @ sD e Zd ZdZdZeZdd� Zddd�Zdd d
�Z d d � Z
d
d� Z dS )�PluginZ undefined� c C s || _ t�d| j �| _d S )Nz plugin %r)�schedulerO rP �textkeyrh )r� r� r r r r� $ s zPlugin.__init__Fc C s^ t � � }| jj}|�|i �}|r(g ||< |�|g �}|�||f� t|�dkrZ|�d� qBd S )Ni� r )rS r� �cached_results�
setdefault�append�len�pop) r� r� �optionr
�replace�now�cache�tcache�resultsr r r � cache_result* s zPlugin.cache_resultr c C sb t � � }| jj}|�|i �}|�|g �}g }|| d � D ]\} }
|�|| |
f� q:|�� |S r )rS r� r� r� r� �reverse) r� r� r� Znumr� r� r� r� ZretZstored�resultr r r �get_cache_results? s zPlugin.get_cache_resultsc C s i S r r )r� �configr r r � get_metadataL s zPlugin.get_metadatac C s dS )Nr r )r� r� �datar� r r r r4 O s z Plugin.checkN)F)r ) rE rF rG r� � descriptionrO rh r� r� r� r� r4 r r r r r� s
r� )F)r1 )NTTr< NTN),�__doc__rS rO r r) r6 rU r` r r� rv � ImportError�
simplejsonr~ � http.client�client�version�ior r r( � setlocale�
LC_NUMERICr r rQ ZDEFAULT_CACHE_TIMEOUTr r r Z SUPPORTEDZ UNSUPPORTEDZ
MISCONFIGUREDr, r0 r; r= r^ ru r{ r r| �objectr� r r r r �<module> sb
�
r