Your IP : 216.73.216.86
3
q��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 krd| _ || _| jd kr.td��tjd�| _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 s dd l }|rdj||�}|d7 }y�|j |j|j���}|jd� |jtjj| j | j
�� |j d�}|j |� t
� }xH|jd�}| jjtj�r�| jjdjt|��� dt|�kr�P ||7 }qrW |jd�}d|kr�d S |S Q R X W n4 tk
�r } z| jjd j| �� d S d } ~ X nX 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 s2
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| �j||� tjd�| _d S )N�
dem_client)�superr+ r
r r r )r
�dem_portr )� __class__r r r
1 s zDEMClient.__init__c C sV | j d�}y|std��tj|�S tk
rP } ztjd|� �� d S d }~X nX 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 }| j d�}|d kr|S tj|�S )N�collect)r' r1 r2 )r
�rvr4 r r r �get_dem_results? s
zDEMClient.get_dem_resultsc C sV y| j dtj|�d� W n8 tk
rP } z | jjdjt|��� W Y d d }~X nX d S )N�
initSchedules)r# zinit_schedules error: {})r' r1 �dumpsr r r! r �str)r
� schedules�exr r r �init_dem_schedulesG s
(z DEMClient.init_dem_schedulesc C s | j dtj|�d�}d S )Nzupdate-schedules)r# )r' r1 r; )r
� newSchedules�_r r r �update_schedulesV s zDEMClient.update_schedules)N) r( r) r* r
r6 r9 r? rB �
__classcell__r r )r/ r r+ 0 s
r+ )r1 r r r r+ r r r r �<module> s *