✏️ 正在编辑: aggregator.cpython-39.pyc
路径:
/usr/lib/fm-agent/library/__pycache__/aggregator.cpython-39.pyc
提示:
您可以编辑任何文件（包括二进制文件），但请注意不当修改可能导致文件损坏。
a <+j? � @ sD d dl m Z z d dlZW n ey6 d dlmZ Y n0 d dlZd dlZd dlZd dlZd dl Z d dl Z z d dl Z W n$ d dl m Z d dlmZ Y n0 z d dlZW n6 ey� z d dlZW n ey� dZY n0 Y n0 z d dlZW n dZY n0 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 )�ReinstallResponseNr 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 �| jj�| _|| _|| _|| _|| _d| _ d | _ d}|d urz|� dd�}t |�t d�krv|� � }|dv rrd}nd}|}|r�|r�| �|�| _ d S )N� T�enabled� )�0�falseF)�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 s~ d }|s |S z2dD ]&}|� |�}|s&qt�|�}| W S W d S tyx t�� d }t|�}| j�d�|�� Y d S 0 d S )N)�https�http� zparse_proxy_info error: {}) 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 }|dv r�dt jv r,tj||t�� d�}q�| j�t j� trvt td�rv| j�dt td�� tj||| j t�� d�}q�t jdkr�tj||| j d�}q�t�||�}n4t jdkr�dt jvr�tj ||| j d�}n t� ||�}|S ) N)� i� zhp-ux)�context�_create_unverified_contextz0ssl has _create_unverified_context attribute: %s��timeoutr3 )� � r )r6 ) r* �platform�httplib�HTTPSConnection�sslr4 r �debug� version_info�hasattrr �HTTPConnection)r" �url�port� connectionr r r �create_connectiono s: ��� �� zClient.create_connection�POSTc C s� dd� | j �d�D �}t�|� d }|D �]�}zTt�|�}|d dksR|d dkr`t�d| �}|d dvsx|d dkr�td ��W n ty� td ��Y n0 |d }d |v r�|�d �\}} t| p�d �} n|}d } d |v r�|d |�d �� }| j � d||| � d } d| j |f } |dk�r\| j |d< d|d< t � t�� �� �|d< | j �d| � t�|�} ddd�} | j�rx| j| d< d } | j�r(tj| jj| jj| jt� � d�} | j � d� | j | j�� | jj }| jj }|�r|�rd� ||�}t!�"t#|d ���$d �}d � |�}| j%|| d |id � n | �%|| � n | �&|| �} �z,| j �d || | | f � | �'|| | | � | �(� }t|j)�d!k�r�t�*|�+� �$d ��}|�,d"i ��,d#��r�|d" d# }t-|��t|j.��|dk�r�t|j)�d$k�r�t|j.��n$|dk�rt|j)�d%k�rt|j.��t�*|�+� �$d ��}t/|�t0k�rZ|�,d&�d'u �rZ|�,d(�d)k�rZt1d*| � d+���|W S t-�yx � Y q( t1�y� � Y q( t�y� t2�3� d }t4|�}| j �5d,| � Y q(Y q(0 q(t|��d S )-Nc S s g | ]}|� � r|� � �qS r )�strip)�.0rA r r r � <listcomp>� � z Client.call.<locals>.<listcomp>�,r r r' �http://)r& r% zInvalid aggregator URLz Invalid aggregator URL format.�:r2 z/vzaggregator call %r to %r:%dz/v%s/%srE Z agent_versionZv3Zmetadata_versionZ agent_timezData: %rzapplication/json)z Content-type�Accept� Authorizationr5 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)6r �split�random�shuffler( r) � ValueError�int�indexr r- r �time�mktimer �now� timetupler= �json�dumpsr r r: r; �hostnamerB r r< r4 r. �host�username�password�base64� b64encode�bytes�decode� set_tunnelrD �request� getresponse�status�loads�readr r rU r �dictr r* r+ r, r1 )r" �action�data�methodZagg_urlsr1 r Z url_partsrA rB �params�urirO rC �unameZpswd�credsZ auth_hashZ auth_value�respZ pause_timeZ resp_datar0 r r r �call� s� ��� � � � � � z Client.callc C s� t � t jt j���}|�d� | j�d�D ]�}|�d��d�}d|v rT|�d�\}}n|}| j�d�rjd}nd}| j� d � ||�� |� |t |�f� |� � d W d � S W d � n1 s�0 Y d S ) Ng @rJ zhttps://rK rL r% r2 �P z*get_local_ip: creating connection to {} {}r )�socket�AF_INET� SOCK_DGRAM� settimeoutr rV � removeprefix� startswithr r- r. �connectrZ � getsockname)r" �clientr rc rB r r r � get_local_ip s �zClient.get_local_ipFc C st t ||||||||||| | | | | |d�}|r>t� }|r>||d< |r\|�� D ]}|| ||< qJ|rh||d< | �d|�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)rp r �keysry )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� s8 � z Client.syncc C sT d}| j �d� z| jdd|id� W n | j �d� Y n0 | j �d� d}|S ) NFz beginning uninstall notification� uninstall�remove_instance)rr zerror connectingznotified successfullyT)r r- ry r1 )r" r� �successr r r �notify_of_uninstallH s zClient.notify_of_uninstallc C s | j �d� t� }| j||t� d�}t� }|r6||d< |�|� |�| �||�� | j �d� | j �tj |dd�� d }d}d} d } d } zL| � d|�}| j � |� |d }|d }d |v r�d |d v r�|d d } W n4 t �y t �� d } | j �d | � | } Y n0 d} | j �d|� |�d ��s<|�d ��s<d} d} d|v �r�|�d��r�d} |d dk�rjd} n~|d dk�r~d} nj|d dk�r�d} nV|d dk�r�d} nB|d dk�r�d} n.|d �d ��r�d |d �d d � } n|d } | j �d | ||| � | ||| | fS )!NzBeginning handshake)r rb r� Z mac_addressesr� z Handshaking with aggregator:r' )�indentF� handshaker rT rP � log_levelzError connecting: %sTz/Connected successfully! Received server key: %rzIError performing handshake with the aggregator, please check connectivityr1 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` ra ry r= r) r* r+ r1 r r� �replace) r" r� � properties� attributesrb rr r� r rT r� r1 r� r0 r r r r� V sx � ��zClient.handshakec C s ||r |p g d�}| � 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�ry )r" r� r� r� r r r r� � s �zClient.maintenancec C s | � d�S )z� Request to the aggregator that it ends all active maintenances for our server. POST call with no parameters. �end_maintenancer� )r" r r r r� � s zClient.end_maintenancec C s d|v r|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! rD ry r� r� r� r� r� r� r� r r r r r 3 s$ � $)h"� 5 I r ) r r: � ImportError� http.clientr� rf r rW r{ r* r\ r( �urllib.requestrk � urlrequest� urllib.parse�parser` � simplejsonr< � inspectorr r r r) r r �objectr r r r r �<module> s<
💾 保存文件
← 返回文件管理器