File: //usr/lib/fm-agent/library/__pycache__/ipc_client.cpython-39.pyc
a
��i� � @ s: d dl Z d dlZd dlZG dd� d�ZG dd� de�ZdS )� Nc @ s e Zd Zddd�Zddd�ZdS )� IPCClientNc C s> || _ | j d u rd| _ || _| jd u r.td��t�d�| _d S )Nz /tmp/com.fortinet.fortimonitorz Missing port�
ipc_client)�ipc_path�port� Exception�logging� getLogger�log)�selfr r � r �'/usr/lib/fm-agent/library/ipc_client.py�__init__ s
zIPCClient.__init__c
C sB dd l }|rd�||�}|d7 }z�|� |j|j���}|�d� |�tj�| j | j
�� |� d�}|� |� t
� }|�d�}| j�tj�r�| j�d�t|��� dt|�kr�q�||7 }qp|�d�}d|kr�W d � W d S |W d � W S 1 � s�0 Y W n: t�y< } z | j�d �| �� W Y d } ~ d S d } ~ 0 0 d S )
Nr z{}:{}�
g $@zutf-8i z received {} bytesZnackzSend/recv failure: {})�socket�format�AF_UNIX� SOCK_STREAM�
settimeout�connect�os�path�joinr r �encode�sendall�bytes�recvr � isEnabledForr �DEBUG�info�len�decoder �error)
r
�command�payloadr �sockZtoSendZ
receivedBytes�rZreceived�er r r � send_receive s0
*zIPCClient.send_receive)N)N)�__name__�
__module__� __qualname__r
r' r r r r r s
r c s> e Zd Zd � fdd� Zdd� Zdd� Zdd � Zd
d � Z� ZS )
� DEMClientNc s" t t| ��||� t�d�| _d S )N�
dem_client)�superr+ r
r r r )r
�dem_portr �� __class__r r r
1 s zDEMClient.__init__c
C s^ | � d�}z|std��t�|�W S tyX } z t�d|� �� W Y d }~d S d }~0 0 d S )Nz wifi-infozNo payload recordedz#Unable to communicate with service )r' �
ValueError�json�loadsr r �warning)r
�response�errr r r �get_dem_wifi_info5 s
zDEMClient.get_dem_wifi_infoc C s$ i }| � d�}|d u r|S t�|�S )N�collect)r' r2 r3 )r
�rvr5 r r r �get_dem_results? s
zDEMClient.get_dem_resultsc
C sX z| j dt�|�d� W n: tyR } z"| j�d�t|��� W Y d }~n
d }~0 0 d S )N�
initSchedules�r# zinit_schedules error: {})r' r2 �dumpsr r r! r �str)r
� schedules�exr r r �init_dem_schedulesG s
,z DEMClient.init_dem_schedulesc C s | j dt�|�d�}d S )Nzupdate-schedulesr< )r' r2 r= )r
� newSchedules�_r r r �update_schedulesV s zDEMClient.update_schedules)N) r( r) r* r
r7 r: rA rD �
__classcell__r r r/ r r+ 0 s
r+ )r2 r r r r+ r r r r �<module> s *