File: //usr/lib/python3.6/site-packages/glances/plugins/__pycache__/glances_ip.cpython-36.opt-1.pyc
3
Eo�c�* � @ s� d Z ddlZddlmZ ddlmZmZmZ ddlm Z ddl
m Z m Z ddl
mZ y ddlZW n6 ek
r� Z zdZe jd je�� W Y ddZ[X nX d
ZdddgZG dd� de�ZG dd� de�ZG dd� de�ZdS )z
IP plugin.� N)�loads)�queue�urlopen� urlopen_auth)�logger)�Timer�getTimeSinceLastUpdate)�
GlancesPluginTz.Missing Python Lib ({}), IP plugin is disabledF�https://httpbin.org/ip�origin�"https://api.ipify.org/?format=json�ip�https://ipv4.jsonip.comc s^ e Zd ZdZdZdgZd� fdd� Zejej dd� ��Z
dd d
�Z d d � Z e
d
d� �Z� ZS )�Pluginz,Glances IP Plugin.
stats is a dict
i, �FalseNc s� t t| �j||d� d| _d| _| jd| jd�| _| jd| jd�}|dgkrPdnd| _ d| _
| jd d
gd�d | _ | jd d
gd�d | _ | jd
d
gd�d | _
| jdd
gd�| _| j d
kp�| j d
kp�| j
d
kp�| jd
k| _d
S )zInit the plugin.)�args�configT� Zpublic_refresh_interval)�default�public_ip_disabled�TrueF�
censys_urlNr �censys_username�censys_password�
censys_fields)�superr �__init__Z
display_curse�public_addressZget_conf_value� _default_public_refresh_interval� public_address_refresh_interval�_default_public_ip_disabledr � public_infor r r r �public_info_disabled)�selfr r r )� __class__� � /usr/lib/python3.6/glances_ip.pyr 2 s
zPlugin.__init__c , C s� | j � }| jdkot �r�ytj� d tj }W n4 ttfk
rd } ztj dj
|�� i S d}~X nX |d |d< y<tj |d �tj d d }tj |d �tj d d }W n4 ttfk
r� } ztj d
j
|�� i S d}~X n$X ||d < ||d < | j |d �|d
< t
d�}y\| j �rh| jjd �|k�s8|| jk�rht� j� | _| j�sht| j| j| j| j�j� | _W n8 ttfk
�r� } ztj dj
|�� W Y dd}~X n X | j|d< | j| j�|d< n | jdk�r�|| _| jS )zQUpdate IP stats using the input method.
:return: the stats dict
Zlocalr z+Cannot grab default gateway IP address ({})Nr Zgateway� ZaddrZnetmaskz#Cannot grab private IP address ({})�address�mask� mask_cidrz public-ipz&Cannot grab public IP information ({})r �public_info_humanZsnmp)Zget_init_valueZ input_method�import_error_tag� netifacesZgatewaysZAF_INET�KeyError�AttributeErrorr �debug�formatZ ifaddresses�
ip_to_cidrr r �stats�getr �PublicIpAddressr r" � PublicIpInfor r r r! �public_info_for_human)r# r3 Z
default_gw�er( r) Ztime_since_updater% r% r& �updateO s@
"
z
Plugin.updatec C sB g }| j s| j� str |S d}|j| j|dd�� d}|j| j|ddd�� d| j kr|dj| j d �}|j| j|dd�� d| j kr�d j| j d �}|j| j|dd�� ydj| j d
�}W n ttfk
r� Y nhX | j d
�rd }|j| j|ddd�� |j| j|dd�� | j d �r>|j| jd
j| j d �dd�� |S )z2Return the dict to display in the curse interface.z - T)ZoptionalzIP ZTITLEr( z{}r* z/{}r z Pub r+ z {})r3 Z is_disabledr, �appendZcurse_add_liner1 �UnicodeEncodeErrorr. )r# r Z max_width�ret�msgZmsg_pubr% r% r& � msg_curse� s0
zPlugin.msg_cursec C s� |sdS g }x�| j D ]�}|jd�}t|�dkrT|d |krT|jdj||d �� qt|�dkr|d |kr|d ||d kr|jdj||d |d �� qW dj|�S )z&Return the data to pack to the client.r �:r' r z{}� �/)r �split�lenr: r1 �join)r# r! Z field_result�fZfieldr% r% r& r7 � s
,$z Plugin.public_info_for_humanc C s$ | dkr dS t dd� | jd�D ��S )zUConvert IP address to CIDR.
Example: '255.255.255.0' will return 24
Nr c s s | ]}t t|��jd �V qdS )�1N)�bin�int�count)�.0�xr% r% r& � <genexpr>� s z$Plugin.ip_to_cidr.<locals>.<genexpr>�.)�sumrB )r
r% r% r& r2 � s zPlugin.ip_to_cidr)NN)NN)�__name__�
__module__� __qualname__�__doc__r r r r Z_check_decoratorZ_log_result_decoratorr9 r> r7 � staticmethodr2 �
__classcell__r% r% )r$ r& r ) s >
)r c @ s, e Zd ZdZd dd�Zdd� Zd d d
�ZdS )
r5 z+Get public IP address from online services.r@ c C s
|| _ dS )zInit the class.N)�timeout)r# rU r% r% r& r � s zPublicIpAddress.__init__c C s� t j� }x8tD ]0\}}}tj| j||||fd�}d|_|j� qW t| j �}d}x*|j
� rz|dkrz|j � dkrR|j � }qRW |dkr�dS dj
tdd� |jd�D ���S ) zGGet the first public IP address returned by one of the online services.)�targetr TNr z, c S s g | ] }|j � �qS r% )�strip)rJ rK r% r% r& �
<listcomp>� s z'PublicIpAddress.get.<locals>.<listcomp>�,)r �Queue�urls� threading�Thread�_get_ip_public�daemon�startr rU �finished�qsizer4 rD �setrB )r# �q�u�j�k�t�timerr
r% r% r& r4 � s
zPublicIpAddress.getFNc C s� y t || jd�j� jd�}W n> tk
rZ } z"tjdj||�� |jd� W Y dd}~X nHX y&|sn|j|� n|jt |�| � W n t
k
r� |jd� Y nX dS )z?Request the url service and put the result in the queue_target.)rU zutf-8z#IP plugin - Cannot open URL {} ({})N) r rU �read�decode� Exceptionr r0 r1 �putr �
ValueError)r# � queue_target�urlZjson�key�responser8 r% r% r& r^ � s z PublicIpAddress._get_ip_public)r@ )FN)rO rP rQ rR r r4 r^ r% r% r% r& r5 � s
r5 c @ s* e Zd ZdZd
dd�Zdd� Zdd� Zd S ) r6 z5Get public IP information from Censys online service.r@ c C s" || _ || _|| _|| _|| _dS )zInit the class.N)r
rp �username�passwordrU )r# r
rp rs rt rU r% r% r& r s
zPublicIpInfo.__init__c C s� t j� }tj| j|| j| j| j| jfd�}d|_ |j
� t | j �}d}x*|j
� rp|dkrp|j� dkrH|j� }qHW |dkr~dS |S )zGReturn the public IP information returned by one of the online service.)rV r TNr )r rZ r\ r] �_get_ip_public_infor
rp rs rt r_ r` r rU ra rb r4 )r# rd rh ri �infor% r% r& r4 s "
zPublicIpInfo.getc C s� dj ||�}yt|||�j� }W n> tk
r^ } z"tjdj ||�� |jd� W Y dd}~X n\X y|jt|�d � W nB tt fk
r� } z"tjdj ||�� |jd� W Y dd}~X nX dS )z?Request the url service and put the result in the queue_target.z{}/v2/hosts/{}z#IP plugin - Cannot open URL {} ({})N�resultz0IP plugin - Cannot get result field from {} ({}))
r1 r rj rl r r0 rm r rn r. ) r# ro r
rp rs rt Z request_urlrr r8 r% r% r& ru s z PublicIpInfo._get_ip_public_infoN)r@ )rO rP rQ rR r r4 ru r% r% r% r& r6 s
r6 )r
Tr )r Tr
)r Tr
)rR r\ Zujsonr Zglances.compatr r r Zglances.loggerr Z
glances.timerr r Z glances.plugins.glances_pluginr r- � ImportErrorr8 r, Zwarningr1 r[ r �objectr5 r6 r% r% r% r& �<module>
s& " +-