File: //proc/self/root/lib/fm-agent/library/__pycache__/inspector.cpython-39.pyc
a
��i�? � @ s. d dl m Z d dlZd dlZd dlZd dlZd dlZd dlZz d dlZW n d dlm Z Y n0 z d dl
Z
W n dZ
Y n0 z d dl Z W n dZ Y n0 z d dl Z W n6 e
y� z d dlZ W n e
y� dZ Y n0 Y n0 dd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Zdd� ZG dd� de�ZdS )� )�datetimeNc c s< t jd dkrt}nt}|d�D ]}| d@ V | dL } q d S )Nr � � � � )�sys� version_infoZxrange�range)�n�rng�i� r
�&/usr/lib/fm-agent/library/inspector.py�mac_address_octets% s
r c C s dd� t | �D �}d�t|��S )zw
Expresses a decimal integer in standard MAC address format
ex:
7267271067680 -> '06:9c:0b:1c:48:20'
c S s g | ] }d | �qS )z%02xr
)�.0�ar
r
r �
<listcomp>7 � z&int_to_mac_address.<locals>.<listcomp>�:)r �join�reversed)r
�valuesr
r
r �int_to_mac_address1 s r c C s, t �� } dtjkr(| �� �d�r(t �� } | S )N�darwinz.arpa)�socket�getfqdnr �platform�lower�endswith� gethostname)�hostnamer
r
r �get_fqdn; s
r! c C sN dt jkrd S z,tjddd�} t�d�| ��\}}|�� W S Y d S 0 d S )Nr ZscutilT)�excz{} --get ComputerName)r r �
agent_util�which�execute_command�format�strip)Zsc_util�_�outputr
r
r �get_server_nameH s
r* c
C s� t �d�} tj�� }dtj�� i}d|kr�z\t�d�}t�d�|��\}}d|krt|� � �
d�d � � � d �|d
< n| � d �|�� W n8 t y� } z | �
d �t|��� W Y d }~n
d }~0 0 |S )
N� Inspectorr r Zioregz6{} -d2 -c IOPlatformExpertDevice | grep IOPlatformUUIDr �=� �"�uuidzioreg error: {}zGather UUID failure: {})�logging� getLoggerr r r r# r$ r% r& r' �split�error� Exception�info�str)�logZ my_platform�rvZ ioreg_bin�rcr) �er
r
r �get_platform_uuidsX s$
��"r; c c sJ t rFt �� �� D ]4\} }|D ]&}|jt jkr.q |jdkr:q |jV q qd S )Nz00:00:00:00:00:00)�psutilZ net_if_addrs�items�familyZAF_LINK�address)Ziface�addrs�addrr
r
r �mac_address_iterq s
rB c
C s� dg i} z dd l }t|�� �| d< W n8 ty\ } z t�|� t�d� W Y d }~n
d }~0 0 z&tt� �D ]\}}| d � |� qjW nR t
y� t�d� Y n8 ty� } z t�|� t�d� W Y d }~n
d }~0 0 | S )NZ
macaddressr Z uuid_getnodezCUnable to import uuid module. Skipping MAC address fact collection.z=Unable to import psutil, skipping MAC address fact collectionz+Unknown error during MAC address collection) r/ r Zgetnoder4 r0 �warningr5 � enumeraterB �append� ImportError)�factsr/ �errr Zmac_addrr
r
r �get_mac_addresses| s$
�
rI c @ s� e Zd ZddgZdZdd� Zd!dd�Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd"dd �Zd d � Zd d � ZdS )#r+ z/etc/apt/sources.listz /etc/yum.repos.d/panopta.repoz http://packages.panopta.com/c C s
|| _ d S )N)�agent)�selfrJ r
r
r �__init__� s zInspector.__init__Nc
C s� i }t �d�}z�|�| �� � |�| �� � |�| �� � |�t� � |�| �� � |�| �� � |�| � � � |�| �
� � |�| � � � |�| � |�� |�| �
� � tjdkr�|�| �� � W n4 ty� } z |�d�|�� W Y d }~n
d }~0 0 |S )Nr+ )r r r zget_all facts: {})r0 r1 �update�get_agent_facts�get_python_facts�get_process_factsrI �get_hardware_facts�get_time_facts� get_hostname�get_platform_facts�get_machine_facts�get_wifi_facts�get_dem_network_factsr r �get_cloud_factsr4 r5 r& )rK � wifi_inforG r7 r: r
r
r �
get_all_facts� s&
&zInspector.get_all_factsc C sL i }dt jkrHz,t�d�\}}|�d�d �� }||d< W n Y n0 |S )Nr z sw_vers | grep ProductVersionr r- Z MacOSVersion)r r r# r% r2 r' )rK rG r( r) �pvr
r
r rT � s
z Inspector.get_platform_factsc C s i }t � |d< |S )Nr )r! �rK rG r
r
r rS � s
zInspector.get_hostnamec C sd i }z6t �d�\}}d�|�� �d�dd � �}||d< W n$ t�| jj�}|� d� Y n0 |S )Nzls -l /etc/localtime�/����timezonezUnable to get Time Zone)
r# r% r r' r2 r0 r1 � __class__�__name__� exception)rK rG �retcoder) r_ r7 r
r
r rR � s zInspector.get_time_factsc
C sJ di i}dt j�� kr�zht�d�\}}|�� �d�D ]F}z0|�� �dd�\}}|dv rf|�� |d |< W q4 Y q4Y q40 q4W n$ t�| j j
�}|� d� Y n0 z$t �
d �t �
d
� }||d d < W n$ t�| j j
�}|� d � Y n0 �nTz�t�d
�\}}|�� �d�D ]b}z|�� �dd�\}}W n Y �qY n0 |�� �� �dd��dd�}|�� }||d |< �qW n$ t�| j j
�}|� d� Y n0 z�t�d�\}}|�� �d�D ]b}|�� �� } | d �� �d��r�t| d �|d d < | d �� �d��r�t| d �|d d< �q�W n$ t�| j j
�}|� d � Y n0 |S )z=
Gather CPU and memory specs for the machine
�hardwarer z sysctl -a | grep machdep.cpu�
r r- )zmachdep.cpu.brand_stringzmachdep.cpu.core_countz Unable to get CPU hardware facts�
SC_PHYS_PAGES� SC_PAGE_SIZEZ mem_totalz#Unable to get memory hardware factsZlscpuz(s)� � r( zfree -mr ZmemZswapZ
swap_total)r r r r# r% r' r2 r0 r1 r` ra rb �os�sysconf�replace�
startswith�int)
rK rG rc r) �line�key�valuer7 Z total_mem�fieldsr
r
r rQ � s^ � z Inspector.get_hardware_factsc C s� i }t j�t j�t��|d< t j�| jj�|d< d|d< | jD ]R}t j�|�r>z4t |�}| j
|� � v |d< |� � |d r�W q�W q> Y q>0 q>d}dt
jks�t j�d�r�d}||d < |S )
NZinstalled_timeZ
used_manifestFZinstalled_from_repoZpanoptar z
/etc/fm-agentZ fortimonitorZ agent_brand)rj �path�getctime�abspath�__file__�existsrJ �
manifest_file�SOURCE_LIST_PATHS�open� PANOPTA_REPO�read�closer r )rK rG Zsource_list_pathZ source_list�brandr
r
r rN s(
�
zInspector.get_agent_factsc C s� di i}t � � |d d< t �� |d d< t �� |d d< t �� |d d< zt �� |d d< W n, ty� tr�d�t�� �|d d< Y n0 t j |d d< |S ) N�pythonr � processor�version�uname�distri Z sys_platform)
r r� �python_versionr� r� �AttributeError�distror Zlinux_distributionr r\ r
r
r rO , s zInspector.get_python_factsc C s
i }z`t jddd�}|�dd� |�� }|jdkrbt�|�� �}d|d< |d |d
< ||d < |W S W n Y n0 �zd d
i}t jddd�}|jdd|d� |�� }|jdk�rvt�|�� �}d|d< |d |d
< d|v r�|d= |�di �� � D ] \}}|dv r�|d �
|� q�||d < z*| j � � }|�dd�dk} | |d d< W n, t
�yn t�| jj�}
|
�d� Y n0 |W S W n Y n0 zrddi}t jddd�}|jdd|d� |�� }|jdk�r�t�|�� �}d |d< |d d |d
< ||d < |W S W n Y n0 i S ) Nz169.254.169.254� )�timeout�GETz*/latest/dynamic/instance-identity/document�� ZawsZcloud_providerZ
instanceIdZcloud_instance_idZcloud_metadatazMetadata-FlavorZGooglezmetadata.google.internalz,/computeMetadata/v1/instance/?recursive=true)�headersZgcp�idZserviceAccounts�
attributes)z
/attributez/sshkeysz
/vmdnssettingz/enable-osloginrJ Zenable_gcp_attributes�truezBUnable to parse manifest file to determine gcp attributes actions.ZMetadataz)/metadata/instance?api-version=2017-04-02ZazureZcomputeZvmId)�httplib�HTTPConnection�request� getresponse�status�json�loadsr| �getr= �poprJ � get_manifestr4 r0 r1 r` ra rb ) rK rG �h�r�datar� rp rq �manifestZenabled_gcp_attributesr7 r
r
r rX > sx
�
���
�
zInspector.get_cloud_factsc C sB i }t d u r|S t� }t �� D ]}|�|�� � q t|�|d< |S )N� processes)r< �setZ process_iter�add�name�list)rK rG r� �procr
r
r rP � s zInspector.get_process_factsc C sv i }dt jkr|S d|i}zBd}| �|�}|rZ|| d }|�dd�|d< |�dd�|d < W n |�� Y n0 |S )
Nr Zmachine_descriptionZSPHardwareDataTyper Z
serial_numberrh �serialZ
machine_modelZmodel)r r �query_system_profilerr� �clear)rK �mdrG �tkr� r5 r
r
r rU � s
zInspector.get_machine_factsc C s� i }d|i}zT|� dd�|d< |� dd�|d< |� dd�|d< |� dd�|d < |� d
d�|d < W n |�� Y n0 t�� �� |d < |S )
NZ DEMWifiInfoZSSIDrh ZssidZBSSIDZbssidZchannelz link authZsecurity_algorithmZNSSZnss� timestamp)r� r� r �utcnowr� )rK rY r� rG r
r
r rV � s zInspector.get_wifi_factsc C s� dt jkri S g }d|i}z�d}| �|�}|r�|�|g �}|D ]t}|�di ��dd �}|s\q>|�d�|�d�g |gd�}|�|� |�d d �} | r>| �d
g �D ]}
|d �|
� q�q>W n t�d � |�� Y n0 |S )
Nr Z!DEMNetworkInterfaceConfigurationsZSPNetworkDataTypeZIPv4ZRouter�_namerd )r� � description� dns_serversZgateways�DNSZServerAddressesr� rW )r r r� r� rE r0 rb r� ) rK r� rG r� r5 Z
network_infosZniZrouter�itemZdns_info�sar
r
r rW � s6
�
z Inspector.get_dem_network_factsc C s` zBt �d�}ddlm}m} ||d|g|d�}t�|j�� �}|W S t �
d� Y d S 0 d S )NZsystem_profilerr )�Popen�PIPEz-json)�stdoutr� ) r# r$ �
subprocessr� r� r� r� r� r| r0 rb )rK Z info_type�spr� r� �dr� r
r
r r� � s
z Inspector.query_system_profiler)N)N)ra �
__module__� __qualname__ry r{ rL rZ rT rS rR rQ rN rO rX rP rU rV rW r� r
r
r
r r+ � s
<!P
r+ )r r# r rj r r0 r r� � http.client�clientr< r� r� rF �
simplejsonr r r! r* r; rB rI �objectr+ r
r
r
r �<module> sB