CasperSecurity
o
;s*b� � @ sN d Z dZdZddlmZmZmZ ddlmZ ddl Z ddl
Z
G dd � d �Z dS )
z
Cyril Jaquierz Copyright (c) 2004 Cyril Jaquier�GPL� )�dumps�loads�HIGHEST_PROTOCOL� )�CSPROTONc @ sR e Zd Zddd�Zdd� Zdd d
�Zd d � Zd
d� Zedd� �Z eddd��Z
dS )�CSocket� /var/run/fail2ban/fail2ban.sock���c C s@ t � t jt j�| _| j�� | _|dkr| �|� | j�|� d S �Nr
)�socket�AF_UNIX� SOCK_STREAM�_CSocket__csock�
gettimeout�_CSocket__deftout�
settimeout�connect)�self�sock�timeout� r �9/usr/lib/python3/dist-packages/fail2ban/client/csocket.py�__init__# s
zCSocket.__init__c C s | � � d S �N)�close�r r r r �__del__- s zCSocket.__del__FNc C s@ t tttj|��t�}| j�|� | j�tj � | �
| j||�S r ) r �list�mapr �convertr r �sendr �END�receive)r �msg� nonblockingr �objr r r r! 0 s z CSocket.sendc C s$ | j �|dkr |� d S | j� d S r )r r r )r r r r r r 7 s $zCSocket.settimeoutc C st | j sd S z| j �tjtj � | j �tj� W n
tjy" Y nw z| j � � W n
tjy4 Y nw d | _ d S r )
r �sendallr �CLOSEr" �shutdownr � SHUT_RDWR�errorr r r r r r : s ��
z
CSocket.closec C s$ t | tttttttf�r| S t| �S )z6Convert every "unexpected" member of message to string)�
isinstance�str�bool�int�floatr �dict�set)�mr r r r H s zCSocket.convertc C s� t j}|r
| �d� |r| �|� d}|�t jd�dkrI| �|�}t|�s+t� dd��|t jkr4t
|�S || }|dk r@|dK }|�t jd�dks t
|�S ) Nr i i��r
�h zConnection reset by peeri � � ) r �EMPTY� setblockingr �rfindr" �recv�lenr r+ r )r r% r r$ �bufsize�chunkr r r r# P s
��zCSocket.receive)r r
)FN) �__name__�
__module__� __qualname__r r r! r r � staticmethodr r# r r r r r ! s
r ) �
__author__�
__copyright__� __license__�pickler r r �protocolr r �sysr r r r r �<module> s