EVOLUTION-NINJA
Edit File: agent_util.cpython-36.pyc
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