File: //usr/lib/fm-agent/library/__pycache__/forticlient_helper.cpython-39.pyc
a
��i� � @ s@ d dl Z d dlmZ d dlZeeed�dd�ZG dd� d�ZdS )� N)�path)�
ems_serial� environment�returnc C s: t �d�}|�| �� � |�|�� � |�� dd� }|S )zO
Calculate a new customer key based on the ems_serial and environment.
�sha256r � )�hashlib�new�update�encode� hexdigest)r r Zhasher� customer_key� r �//usr/lib/fm-agent/library/forticlient_helper.py�calculate_customer_key s
r c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd S ) �ForticlientHelperz7/Library/Application Support/Fortinet/FortiClient/data/z@/Library/Application Support/Fortinet/FortiClient/conf/vpn.plistc C s ddd�| _ t�� | _d S )NZforticlient_serialr )z fct_uid.confz EMS_sn.conf)�data_files_map�logging� getLogger�log)�selfr r r �__init__ s �zForticlientHelper.__init__c C s� z�dd l m} |�|�}|jdkr0|�d| �}|j}z|�d�}|d|� }W n Y n0 |�� �d�}t|�dk r~W d S |d }|d }|d } t|�d
kr�|d }
n|d }
|d
kr�|dkr�|
dkr�| W S W d S Y d S 0 d S )Nr � zhttp://�:�.� ���������� ������ZcomZ fortisase�edge) � urllib.parse�parse�urlparse�scheme�netloc�index�lower�split�len) r �urlr% Zparsed�host�idxZ
host_partsZ top_level�domainr r" r r r �
parse_address s2
z ForticlientHelper.parse_addressc C s� dd l }t| jd��`}|�|�}|d �� D ]8}|�dd �}|r,| �|�}|r,| W d � S q,W d � n1 sz0 Y td��d S )Nr �rbZProfilesZServerz No environment profile found)�plistlib�open�vpn_data�load�values�getr0 � Exception)r r2 �f�dataZ profileInfo� server_name�envr r r �get_vpn_environmentF s
6z%ForticlientHelper.get_vpn_environmentc C s� t �| j�std��i }d}| j�� D ]d\}}|d7 }tt �| j|�d��0}|�� � � }t
|�dkrl|||< W d � q&1 s�0 Y q&t
|� � �|kr�td��z| � � |d< W n td��Y n0 |S )NzMissing forticlient datar � �rzMissing forticlient keys�forticlient_environmentzMissing VPN data)
r �isdir� data_pathr8 r �itemsr3 �join�readline�stripr+ �keysr= )r �rvZ key_count�fileZhs_keyZdfZdvr r r �get_handshake_dataT s" (z$ForticlientHelper.get_handshake_dataN) �__name__�
__module__� __qualname__rB r4 r r0 r= rJ r r r r r s
'r )r �osr r �strr r r r r r �<module> s