File: //lib/python3.6/site-packages/glances/plugins/__pycache__/glances_connections.cpython-36.pyc
3
Eo�c� � @ sP d Z ddlmZ ddlmZ ddlmZ ddlmZ ddl Z G dd� de�Z
dS ) zConnections plugin.� )�unicode_literals)�logger)�
GlancesPlugin)� nativestrNc s� e Zd ZdZejejgZejej gZ
ej ej ej
ejejejgZddd�Zd� fdd� Zejejdd � ��Z� fd
d �Zdd d
�Z� ZS )�Pluginz5Glances connections plugin.
stats is a dict
z*/proc/sys/net/netfilter/nf_conntrack_countz(/proc/sys/net/netfilter/nf_conntrack_max)�nf_conntrack_count�nf_conntrack_maxNc s, t t| �j||i d� d| _d| _d| _dS )zInit the plugin.)�args�configZstats_init_valueTN)�superr �__init__Z
display_curse�net_connections_enabled�nf_conntrack_enabled)�selfr r
)� __class__� �)/usr/lib/python3.6/glances_connections.pyr 1 s
zPlugin.__init__c s� | j � }| jdk�r�| j�rytjdd�}W n> tk
rj } z"tjdj|�� d| _|| _ | j S d}~X nX x(| j
D ] � t � fdd�|D ��|� < qtW d }x4| j D ]*� t � fd
d�|D ��|� < ||� 7 }q�W ||d < d }x4| j D ]*� t � fd d�|D ��|� < ||� 7 }q�W ||d
< | j
�r�x�| jD ]~}y6t| j| d�� }t|j� jd��||< W dQ R X W n@ tk
�r� } z"tjdj|�� d| _
|| _ | j S d}~X nX �q*W |d d |d |d< n | jdk�r�|| _ | j S )zRUpdate connections stats using the input method.
Stats is a dict
ZlocalZtcp)Zkindz*Can not get network connections stats ({})FNc s g | ]}|j � kr|�qS r )�status)�.0�c)�sr r �
<listcomp>Z s z!Plugin.update.<locals>.<listcomp>r c s g | ]}|j � kr|�qS r )r )r r )r r r r ] s � initiatedc s g | ]}|j � kr|�qS r )r )r r )r r r r b s �
terminated�r�
z*Can not get network connections track ({})r �d r �nf_conntrack_percentZsnmp)Zget_init_valueZ input_methodr
�psutil�net_connections� Exceptionr �debug�format�stats� status_list�len�initiated_statesr � conntrack�open�float�readline�rstrip�IOError)r r# r �er r �i�fr )r r �updateB sH $ z
Plugin.updatec sH t t| �j� y | jr,| jdd�| jd d< W n tk
rB Y nX dS )zUpdate stats views.r )�header�
decorationN)r r � update_viewsr Z get_alertZviews�KeyError)r )r r r r3 { s zPlugin.update_viewsc C sZ g }| j s| j� r|S | js$| jr@djd�}|j| j|d�� | jr�x�tjdtj dgD ]l}|j| j
� � djt |�j � t
|�d�}|j| j|�� dj| j | |t
|� d d�}|j| j|�� qXW | j�rVd
}|j| j
� � djt |�j � t
|�d�}|j| j|�� djd j| j d | j d
�|t
|� d d�}|j| j|| jddd��� |S )z2Return the dict to display in the curse interface.z{}zTCP CONNECTIONSZTITLEr r z
{:{width}})�widthz {:>{width}}� ZTrackedz{:0.0f}/{:0.0f}r r r r2 )�keyZoption)r# Z is_disabledr
r r" �appendZcurse_add_liner � CONN_LISTEN�CONN_ESTABLISHEDZcurse_new_liner �
capitalizer% Z get_views)r r Z max_width�ret�msgr r r r � msg_curse� s.
zPlugin.msg_curse)NN)NN)�__name__�
__module__� __qualname__�__doc__r r9 r: r$ Z
CONN_SYN_SENTZ
CONN_SYN_RECVr& ZCONN_FIN_WAIT1ZCONN_FIN_WAIT2ZCONN_TIME_WAITZ
CONN_CLOSEZCONN_CLOSE_WAITZ
CONN_LAST_ACKZterminated_statesr' r r Z_check_decoratorZ_log_result_decoratorr0 r3 r> �
__classcell__r r )r r r s 8 r ) rB Z
__future__r Zglances.loggerr Z glances.plugins.glances_pluginr Zglances.compatr r r r r r r �<module>
s