EVOLUTION-NINJA
Edit File: plugin_manager.cpython-36.pyc
3 q��i�. � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZyd dl m Z W n d dl m Z Y nX y d dl Z W n d dl jZ Y nX y d dlZW n d dlZY nX d dd�ZG dd� de�Zd d � ZdS ) � )�PluginN)�sha1)�shac C sJ |d kr i }|j | dd�� t| d�}tt|j� | d�||� |j� d S )N�__main__)�__file__�__name__�rb�exec)�update�openr �compile�read�close)�filepath�globals�locals�f� r �+/usr/lib/fm-agent/library/plugin_manager.py� execfile3 s r c @ sP e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zed d � �Z d d� Z dd� Z dS )� PluginManagerc G s� t j| jj�| _|| _|| _|| _| j|�| _ d | _ d | _ i | _ t jj| j�r�tj� }|j| j� xf|j� D ]Z}y|jdd�}W n d}Y nX d|i| j |< x$|j|�D ]\}}|| j | |< q�W qlW d S )N�agent�debugF)�logging� getLogger� __class__r �log�db� config_file� directories� _load_plugins�plugins� _metadata�_hashed_metadata�config�os�path�exists� configparser� ConfigParserr �sections�get�items)�selfr r r �textkeyr �k�vr r r �__init__) s& zPluginManager.__init__c C s | j jdd �}|r||d< d S )N�dem� wifi_results)r$ r+ )r- r3 Z dem_configr r r �add_dem_wifi_resultsC s z"PluginManager.add_dem_wifi_resultsc C s8 | j d }tjdd�}d}|jd�d }|jd�d }yHtj|�}d|j� krf|j� jd�jd�d }|j|j � � |j � W n( t k r� | j j dtj� � d S X |dk�r0yPtj|j�}x>|j� D ]2} | jd �r�| j jd | � |j| |� |d7 }q�W W n* t k �r, | j j dtj� � Y nX n�|dk�r�yTtj|j�} xB| j� D ]6} | jd ��rR| j jd | � | j| |� |d7 }�qRW W n* t k �r� | j j dtj� � Y nX nR|dk�r�| j jd | � tjd|jtjj ||�f � |d7 }n| j j d| � |�r$| j jd| � tjd|j � d S )zu Grab the remote url and extract the contents, putting them in the appropiate plugin folder. � F)�deleter �.�/zcontent-dispositionzUnable to download URL: %sN�tar�tgzz.pyz Installing plugin %sz"Unable to extract tar contents: %s�zipz"Unable to extract zip contents: %s�pyzcp %s %sz5Unable to install Agent Plugin, unknown extension: %sz Installed %s Agent pluginszrm -f %s���r= r= r= )r9 r: ) r �tempfileZNamedTemporaryFile�split�urllib2�urlopen�info� getheader�writer r � Exceptionr �error� traceback� format_exc�tarfiler �nameZgetnames�endswith�extract�zipfileZZipFileZnamelistr% �systemr& �join) r- �url�base_custom_plugin_dirr Z num_installedZext�fname�r�t�file�zr r r �install_remote_pluginsH s\ z$PluginManager.install_remote_pluginsc C s4 y�| j jdi �jd�}d }t|dd �r�|j|kr�|jjd�d }y| jd | }W n< |jdkrx|jdkrxdd i}n| jj d |� d S Y nX | j |j |�}| j j|ji �}|jd d �}|r�|j ||j|j |�}n|j |j|j |�}|d k r�||9 }W n4 tk �r. | jjd |j|jtj� f � d S X |S )Nr � server_key�:r5 �docker_containersZdockerzstatus.runningZStateZdeletedz#Could not find container with id %s�scaleg �?z%s/%s check failed: %sr= )r$ r+ �getattrrX r? r �plugin_textkey�resource_textkeyr rF r! Z check_docker�option�checkrE � exceptionrG rH ) r- �scheduleZself_key� container� container_id�pluginr$ r[ �valuer r r r` � s@ zPluginManager.checkc s� i �d}t jj�j�rftj� }|j�j� y|jdd�}W n d}Y nX |rftt t j |j d���}� ��fdd�}xJ|� D ]@}t |t�r~t|t�r~t|d�r~|j|kr~|jdkr~|�|j< q~W �S ) N� r �plugin_blacklist�,c 3 s �x� D �]} �j jd| � x�tj| �D ]�}y�|d d � dksF|dkrHw(tjj| |�}i }i }tjd dkrzt|||� n t |||� xt |j � �D ] }|V q�W �j j d|� W q( t k r� tj� d d� \}}�j jd|� �j j |� w(Y q(X q(W �j jd t��� qW d S ) NzLooking in directory %r� z.pyz __init__.pyr z Loaded %r� z#Error importing plugin %s, skippingzLoaded %d plugins���)r rB r% �listdirr& rO �sys� version_infor Zexecfile�list�valuesr rE �exc_info�len)Z plugin_dirrJ Z plugin_file� f_globals�f_locals�objrT �e)r r! r- r r �plugin_file_checker� s* z8PluginManager._load_plugins.<locals>.plugin_file_checkerr. ZTEMPLATE_PLUGIN)r% r& r' r r( r) r r+ rp �map�str�stripr? � isinstance�type� issubclassr �hasattrr. )r- r rh r rx rv r )r r! r- r r � s( ! zPluginManager._load_pluginsc C s� | j d kr�i }x�t| jj� �D ]�}| jj|ji �}yt|j|��}W n | j j d|j� w Y nX | j j dt t|j � ��|j� t|d|j�}||f||j< q W || _ | j S )Nz error getting metadata from %rzgot %d textkeys for %r�label)r" rp r! rq r$ r+ r. �format_metadata� get_metadatar ra rB rs �keysr\ )r- �metadatare r$ �metar� r r r r� � s zPluginManager.metadatac C s� | j d kr�g }xvt| jj� �D ]d\}\}}g }x6t|j� �D ]&\}}tt|��}|jt||f�� q<W |j� |jt|||f�� q W |j� tt|�j � �j � | _ | j S )N) r# rp r� r, �sorted�repr�append�sortr �encode� hexdigest) r- Z flat_datar. r� �plugin_metadataZplugin_optionsZ resource_nameZresource_metadataZ resource_metar r r �hashed_metadata s z PluginManager.hashed_metadatac C s, | j jd| j� | jd � | jd | j� kS )Nz$comparing metadata hashes %r with %r� last_metadata)r rB r� r )r- r r r �is_metadata_stale$ s z PluginManager.is_metadata_staleN) r � __module__� __qualname__r1 r4 rW r` r �propertyr� r� r� r r r r r ( s =5? r c C s� x�| j � D ]�\}}|d }g }|s$q xT|D ]L}tjd dkrXt|�tkrv|jd|i� q*t|�ttfkr*|jd|i� q*W |r�||d< |jd�s ddi|d< q W | S )z� Go over the metadata entry and make sure the option resources are matching to the new style of dictionaries if they are a string. �optionsr rj �resourceZoptions_schema�string)r, rn ro r} rz r� �unicoder+ )r� r� �entryr� Z new_optionsr_ r r r r� - s$ r� )NN)� agent_utilr r r% rn rG r> rI rM �hashlibr r r@ �urllib.request�requestr) r( r �objectr r� r r r r �<module> s0