EVOLUTION-NINJA
Edit File: aggregator.cpython-36.pyc
3 q��i? � & @ sD d dl m Z y d dlZW n ek r6 d dljZY nX d dlZd dlZd dlZd dlZd dl Z d dl Z y d dl Z W n d dl j Zd dljZ Y nX y d dlZW n: ek r� y d dlZW n ek r� dZY nX Y nX y d dlZW n dZY nX d dlmZmZmZ G dd� de�ZG dd� de�ZG dd � d e�Z dS ) � )�datetimeN)�get_fqdn�get_platform_uuids�get_mac_addressesc @ s e Zd ZdS )�UnauthorizedResponseN)�__name__� __module__� __qualname__� r r �'/usr/lib/fm-agent/library/aggregator.pyr + s r c @ s e Zd ZdS )�ReinstallResponseN)r r r r r r r r / s r c @ s� e Zd ZdZddd�Zdd� Zdd� Zi d fd d �Zd d � Zdi di i fdd�Z d dd�Z d dd�Z d dd�Z dd� Z dd� ZdS ) �Clientzy This is a client used for communicating with an aggregator, either *the* Aggregator or an Onsight instance. Nc C s� t j| jj�| _|| _|| _|| _|| _d| _ d | _ d}|d k rz|j dd�}t |�t d�krv|j � }|dkrrd}nd}|}|r�|r�| j|�| _ d S ) N� T�enabled� �0�falseF)r r )�logging� getLogger� __class__r �log� customer_key� server_key�agg_url�version� call_timeout�proxy�get�type�lower�parse_proxy_info)�selfr r r r � proxy_configZproxies_enabledZpevr r r �__init__9 s$ zClient.__init__c C sz d }|s |S y.x(dD ] }|j |�}|s(qtj|�}|S W d S tk rt tj� d }t|�}| jjdj|�� d S X d S )N�https�http� zparse_proxy_info error: {})r$ r% ) r �urlparse� Exception�sys�exc_info�strr �info�format)r! r" Z proxy_val�schemeZ parsed_uri�err�errorr r r r ] s zClient.parse_proxy_infoc C s� d }|d kr�dt jkr,tj||tj� d�}q�| jjt j� trvt td�rv| jjdt td�� tj||| j tj� d�}q�t jd kr�tj||| j d �}q�tj||�}n4t jdkr�dt jkr�tj ||| j d �}n tj ||�}|S )N� � zhp-ux)�context�_create_unverified_contextz0ssl has _create_unverified_context attribute: %s)�timeoutr3 � � r )r5 )r1 r2 )r6 r7 r )r6 r7 r ) r) �platform�httplib�HTTPSConnection�sslr4 r �debug� version_info�hasattrr �HTTPConnection)r! �url�port� connectionr r r �create_connectiono s0 zClient.create_connection�POSTc C s� dd� | j jd�D �}tj|� d }�x�|D �]�}yTtj|�}|d dksV|d dkrdtjd| �}|d d.ks||d dkr�td ��W n tk r� td ��Y nX |d }d |kr�|jd �\}} t| p�d �} n|}d } d|kr�|d |jd�� }| j j d||| � d } d| j |f } |dk�rb| j |d< d|d< t j tj� j� �|d< | j jd| � tj|�} ddd�} | j�r~| j| d< d } | j�r.tj| jj| jj| jtj � d�} | j j dj | j | j�� | jj }| jj }|�r |�r d j ||�}t!j"t#|d ��j$d �}d j |�}| j%|| d |id � n | j%|| � n | j&|| �} �y&| j jd!|| | | f � | j'|| | | � | j(� }t|j)�d"k�r�tj*|j+� j$d ��}|j,d#i �j,d$��r�|d# d$ }t-|��t|j.��|dk�r�t|j)�d%k�r�t|j.��n$|dk�rt|j)�d&k�rt|j.��tj*|j+� j$d ��}t/|�t0k�r`|j,d'�d(k�r`|j,d)�d*k�r`t1d+| � d,���|S t-k �rz � Y q, t1k �r� � Y q, tk �r� t2j3� d }t4|�}| j j5d-| � w,Y q,X q,W t|��d S )/Nc S s g | ]}|j � r|j � �qS r )�strip)�.0r@ r r r � <listcomp>� s z Client.call.<locals>.<listcomp>�,r r r& zhttp://r% r$ zInvalid aggregator URLz Invalid aggregator URL format.�:i� z/vzaggregator call %r to %r:%dz/v%s/%srD Z agent_versionZv3Zmetadata_versionZ agent_timezData: %rzapplication/json)z Content-type�Accept� Authorization)r5 r3 zConnection via proxy {}:{}z{}:{}zutf-8zBasic {}zProxy-Authorization)�headersz%s - %s - %s - %si� �commands�pause�� �� � found_serverF�reasonzFortiClient serial mismatchzSync z+ failed due to FortiClient serial conflict.zUnable to connect: %s)r% r$ )6r �split�random�shuffler' r( � ValueError�int�indexr r, r �time�mktimer �now� timetupler< �json�dumpsr r r9 r: �hostnamerA r r; r4 r- �host�username�password�base64� b64encode�bytes�decode� set_tunnelrC �request� getresponse�status�loads�readr r rR r �dictr r) r* r+ r0 )r! �action�data�methodZagg_urlsr0 r Z url_partsr@ rA �params�urirL rB �unameZpswd�credsZ auth_hashZ auth_value�respZ pause_timeZ resp_datar/ r r r �call� s� z Client.callc C s� t j t jt j���}|jd� x�| jjd�D ]v}|jd�jd�}d|krV|jd�\}}n|}| jjd�rld}nd}| jj d j ||�� |j |t |�f� |j � d S W W d Q R X d S ) Ng @rH zhttps://zhttp://rI r$ i� �P z*get_local_ip: creating connection to {} {}r )�socket�AF_INET� SOCK_DGRAM� settimeoutr rS Z removeprefix� startswithr r, r- �connectrW � getsockname)r! �clientr r` rA r r r � get_local_ip s zClient.get_local_ipFc C sx t ||||||||||| | | | | |d�}|r>t� }|r>||d< |r`x |j� D ]}|| ||< qLW |rl||d< | jd|�S )N)�results� outage_events�metadata�countermeasures_metadata�facts�discovered_containers�deleted_containers�force_send_schedules�command_results�register�register_custom_metrics� incidents�config�ips�auto_topo_scans� dem_enabled�fm_server_identifiers�fortisase_attributes�sync)rm r �keysrv )r! r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �dem_service_resultsr� �payload�platform_uuidsZdsr r r r� s6 z Client.syncc C sT d}| j jd� y| jdd|id� W n | j jd� Y nX | j jd� d}|S ) NFz beginning uninstall notification� uninstall�remove_instance)ro zerror connectingznotified successfullyT)r r, rv r0 )r! r� �successr r r �notify_of_uninstallH s zClient.notify_of_uninstallc C s | j jd� t� }| j||t� d�}t� }|r6||d< |j|� |j| j||�� | j jd� | j jtj |dd�� d }d}d} d } d } yL| j d|�}| j j |� |d }|d }d |kr�d |d kr�|d d } W n6 t k �r t j� d } | j jd | � | } Y nX d} | j jd|� |jd ��p2|jd ��s>d} d} d|k�r�|jd��r�d} |d dk�rld} n~|d dk�r�d} nj|d dk�r�d} nV|d dk�r�d} nB|d dk�r�d} n.|d jd ��r�d |d jd d � } n|d } | j jd | ||| � | ||| | fS )!NzBeginning handshake)r r_ r� Z mac_addressesr� z Handshaking with aggregator:r& )�indentF� handshaker rQ rM � log_levelzError connecting: %sTz/Connected successfully! Received server key: %rzIError performing handshake with the aggregator, please check connectivityr0 Zinvalid_credentialszIncorrect username/passwordZinvalid_customer_keyz Unknown/invalid customer keyZinvalid_server_groupzUnknown server groupZinactive_customerzCustomer inactiveZno_user_found_for_customerz%No active user found for the customerZinvalid_template_idzUnknown template id %szinvalid_template_id:r z8Success: %s, server key: %s, found server: %s, error: %s)r r, r r r r �update�parse_attributesr] r^ rv r< r( r) r* r0 r r| �replace) r! r� � properties� attributesr_ ro r� r rQ r� r0 r� r/ r r r r� V sr zClient.handshakec C s ||r |p g d�}| j d|�S )a Request to the aggregator that it starts a maintenance for our server. It requires the duration of the maintenance in minutes, and accepts an optional list of tags to mark metrics that are going to be affected by it. )�duration� metric_tags� maintenance)rv )r! r� r� r� r r r r� � s zClient.maintenancec C s | j d�S )z� Request to the aggregator that it ends all active maintenances for our server. POST call with no parameters. �end_maintenance)rv )r! r r r r� � s zClient.end_maintenancec C s d|kr|rd|ipi S i S dS )z� Receive a dictionary containing the attributes stored in the manifest file. Parse them accordingly and store them in an attribute dictionary. Attributes get parsed only if the customer key is present in the properties. r r� Nr )r! r� r� r r r r� � s zClient.parse_attributes)NNN)F)N)N)r r r �__doc__r# r rC rv r� r� r� r� r� r� r� r r r r r 3 s #)h" " I r ) r r9 � ImportError� http.clientr rc r rT rx r) rY r' �urllib.requestrh � urlrequest� urllib.parse�parser] � simplejsonr; � inspectorr r r r( r r �objectr r r r r �<module> s<