Your IP : 216.73.216.86
3
q��i�$ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z y ddl
Z
W n e k
rt ddl Z
Y nX y ddl
Z
W n ddljZ
Y nX ejd dkr�ddlmZ n ddlmZ dejkr�dejd< ejej� dd � ZeZi Zd
Zd d � Zd
d� Z dd� Z dZ dZ dZ ddddddgZ!d-dd�Z"d.d d �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 )0zEThis file is a library of some essential classes for our agent's use.� N�3)�BytesIO)�StringIO�LC_ALL�Cc C sJ yt jt| ��S tk
rD tt�ti �kr:td | �S tj| �S X 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 x| D ] }|rdS qW dS )NTFr )�iterable�elementr r r �any5 s
r c C s x| D ] }|sdS qW dS )NFTr )r r r r r �all< s
r c C s" | j | j| jd d d d S )N� i �
� i@B 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 jj| �\}}|r*|| �rv| S nLdt jkr8d S x<tjt jd jt j�t�D ] }t jj|| �}||�rT|S qTW |r�t d| ��nd S d S )Nc S s t jj| �ot j| 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�}|jjd�S )NFT)�shell�timeout�checkZcapture_outputzutf-8)�
subprocess�run�stdout�decode)Z command_argsr3 �pr r r � run_commands s r: T� c s� |dkr| j � } tjd�}|rdtj| �} | rdtj� | jd� |k rd|jd| � | jd�| jd�fS d}
dtjj � k�r,t
j | ||t
j t
j |d �}
|s�dS y|
j
|d
�\} } W nx t
jk
�r& }
zX|
jg}|r�|jt|
j�� x6|D ].}ytj|tj� W q� tk
�r Y q�X q�W dS d}
~
X nX �n
G d
d� dt�� � fdd�}t
j | ||t
j t
j |d �}
|�sldS |dk�r�|dk�r�tjtj|� tj|� y.|
j
� \} } |dk�r�|dk�r�tjd� W nn � k
�r4 |
jg}|�r�|jt|
j�� x8|D ]0}ytj|tj� W n tk
�r( Y nX �q�W d S X |
j}| �sF| } | jd�}|�rx|jd| � ||tj� d�t| < |jd| ||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 �cwdr7 �stderr�env)r3 � � Timeout exceeded, process killedc @ s e Zd ZdS )z execute_command.<locals>.AlarmN)�__name__�
__module__� __qualname__r r r r �Alarm� s rI c s � �d S )Nr )Zsignum�frame)rI r r �
alarm_handler� s z&execute_command.<locals>.alarm_handlerr r �utf8zCreated cache for cmd %s)r> r? r= z %s: %s %s���)rM rE ���rN ���)rO rE ) r' �logging� getLogger�LOCAL_CACHE_RESULTS�get�time�debug�sys�platform�lowerr5 �Popen�PIPE� communicate�TimeoutExpired�pid�extend�get_process_childrenr �kill�signal�SIGKILL�OSErrorr
�SIGALRM�alarm�
returncoder8 )�cmdrA r2 Z kill_treer3 rC �blockZ
cache_timeout�logZ
cached_resultr9 r7 rB Zte�pidsr] rK r> r? r )rI r r< z s�
r< c C s8 t jd| dt jt jd�}|j� \}}dd� |j� D �S )z?Helper method for killing off child processes when they timeoutz ps --no-headers -o pid --ppid %dT)r2 r7 rB c S s g | ] }t |��qS r )�int)�.0r9 r r r �
<listcomp>� s z(get_process_children.<locals>.<listcomp>)r5 rY rZ r[ r' )r] r9 r7 rB r r r r_ � s
r_ c C s` y$| d d }t |j� �d }|d S tk
rZ tj� d d� \}}tj|� |�Y nX d S )NZNetworkSettingsZNetworksr Z IPAddressr )�list�valuesr
rV �exc_inforP � exception)� containerZnetworksZnetwork�t�er r r �get_container_ip� s
ru c K s6 yt j| f|�S tk
r0 t j| j� f|�S X d S )N)�json�loads� TypeErrorr8 )�val�kwargsr r r �
json_loads s r{ c @ s e Zd Zddd�Zdd� ZdS )�UnixHTTPConnection� localhostc K s t jj| |f|� || _d S )N)�httplib�HTTPConnection�__init__r )�selfr �hostrz r r r r� s zUnixHTTPConnection.__init__c C s& t j t jt j�}|j| j� || _d S )N)�socketZAF_UNIXZ SOCK_STREAM�connectr �sock)r� r� r r r r� s zUnixHTTPConnection.connectN)r} )rF rG rH 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 || _ tjd| j �| _d S )Nz plugin %r)�schedulerP rQ �textkeyri )r� r� r r r r� $ s zPlugin.__init__Fc C sb t j � }| jj}|j|i �}|r(g ||< |j|g �}|j||f� xt|�dkr\|jd� qDW d S )Ni� r )rT 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 sf t j � }| jj}|j|i �}|j|g �}g }x,|| d � D ]\} }
|j|| |
f� q<W |j� |S )N)rT 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 )Nr )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 ) rF rG rH r� � descriptionrP ri r� r� r� r� r4 r r r r r� s
r� )F)r1 )NTTr; NTN),�__doc__rT rP r r) r5 rV ra r r� rv � ImportError�
simplejsonr~ � http.client�client�version�ior r r( � setlocale�
LC_NUMERICr r rR ZDEFAULT_CACHE_TIMEOUTr r r Z SUPPORTEDZ UNSUPPORTEDZ
MISCONFIGUREDr, r0 r: r< r_ ru r{ r r| �objectr� r r r r �<module> sj
j