File: //lib/python3.6/site-packages/glances/plugins/__pycache__/glances_ports.cpython-36.pyc
3
Eo�cS- � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZ ddlmZ ddlmZ dd lmZ yddlZd
ZW n6 ek
r� Z zd Zejd j e�� W Y ddZ[X nX G d
d� de�Z G dd� dej �Z dS )zPorts scanner plugin.� N)�WINDOWS�MACOS�BSD)�GlancesPortsList)�GlancesWebList)�Counter)� bool_type)�logger)�
GlancesPluginTFzAMissing Python Lib ({}), Ports plugin is limited to port scanningc sh e Zd ZdZd� fdd� Z� fdd�Zejejdd� ��Z d d
� Z
dd
d�Z ddd�Z ddd�Z
� ZS )�Pluginz Glances ports scanner plugin.Nc sT t t| �j||g d� || _|| _d| _t||d�j� t||d�j � | _
d| _ dS )zInit the plugin.)�args�configZstats_init_valueT)r
r N) �superr �__init__r r
Z
display_curser Zget_ports_listr Z get_web_list�stats�_thread)�selfr r
)� __class__� �#/usr/lib/python3.6/glances_ports.pyr ' s zPlugin.__init__c s& | j dk r| j j� tt| �j� dS )z+Overwrite the exit method to close threads.N)r �stopr r �exit)r )r r r r 9 s
z Plugin.exitc C sF | j dkr@| jdkrd}n
| jj� }|s@t| j�| _| jj� n | jS )zUpdate the ports list.ZlocalNF)Z input_methodr Zis_alive�
ThreadScannerr �start)r Zthread_is_runningr r r �update@ s
z
Plugin.updatec C s dS )zReturn the key of the list.�indicer )r r r r �get_key[ s zPlugin.get_key� Fc C s� d}|d dkrd}nD|d dkr(d}n2t |d ttf�rZ|d dk rZ|d |d krZd}| j|d �}| j||� | j||j� ||| j� � |S )
z?Return the alert status relative to the port scan return value.�OK�statusN�CAREFULr �CRITICAL� rtt_warning�WARNING)�header)�
isinstance�float�int�
get_stat_name�manage_threshold�
manage_action�lowerr )r �portr$ �log�ret� stat_namer r r �get_ports_alert_ s zPlugin.get_ports_alertc C s� d}|d dkrd}n2|d d
kr(d}n |d dk rH|d
|d krHd }| j |d �}| j||� | j||j� ||| j� � |S )zBReturn the alert status relative to the web/url scan return value.r r Nr �� �- �. r! r" �elapsedr# )r$ )r1 r2 r3 )r( r) r* r+ r )r �webr$ r- r. r/ r r r �
get_web_alertx s zPlugin.get_web_alertc C s� g }| j s|jr|S |d }�x�| j D �]�}d|k�r|d dkrJd}nV|d dkr\d}nDt|d t�r||d dkr|d}n$|d d kr�d
}nd j|d d �}d
j|d d |� |d�}|j| j|�� dj|�}|j| j|| j||d d d��� |j| j� � q(d|kr(d
j|d d |� |d�}|j| j|�� t|d t j
��r\dj|d �}n |d dk�rpd}n|d }dj|�}|j| j|| j ||d d d��� |j| j� � q(W y |j � W n t
k
�r� Y nX |S )z2Return the dict to display in the curse interface.� �hostN�Noner ZScanningTZOpenr ZTimeoutz {0:.0f}msg @�@z
{:{width}}� description)�widthz{:>9}r Z_rtt)r$ �urlzCode {})r Z
disable_portsr% r �format�appendZcurse_add_liner0 Zcurse_new_line�numbers�Numberr6 �pop�
IndexError)r r Z max_widthr. Zname_max_width�pr �msgr r r � msg_curse� sH
$
$ zPlugin.msg_curse)NN)r F)r F)NN)�__name__�
__module__� __qualname__�__doc__r r r
Z_check_decoratorZ_log_result_decoratorr r r0 r6 rE �
__classcell__r r )r r r $ s
r c s| e Zd ZdZ� fdd�Zdd� Zedd� �Zejdd� �Zdd
d �Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )r zP
Specific thread for the port/web scanner.
stats is a list of dict
c s8 t jdj|�� tt| �j� tj� | _|| _ d| _
dS )zInit the class.z-ports plugin - Create thread for scan list {}ZportsN) r �debugr= r r r � threadingZEvent�_stopper�_stats� plugin_name)r r )r r r r � s
zThreadScanner.__init__c C sR xL| j D ]B}| j� rP d|kr4| j|� tjd� qd|krtr| j|� qW dS )z`Grab the stats.
Infinite loop, should be stopped by calling the stop() method.
r, � r< N)rN �stopped�
_port_scan�timeZsleep� requests_tag� _web_scan)r rC r r r �run� s
zThreadScanner.runc C s | j S )z
Stats getter.)rN )r r r r r � s zThreadScanner.statsc C s
|| _ dS )z
Stats setter.N)rN )r �valuer r r r � s Nc C s t jdj| j�� | jj� dS )zStop the thread.z,ports plugin - Close thread for scan list {}N)r rK r= rN rM �set)r �timeoutr r r r � s zThreadScanner.stopc C s
| j j� S )z%Return True is the thread is stopped.)rM Zis_set)r r r r rQ � s zThreadScanner.stoppedc C s� y(t j|d d|d |d |d d�}W n< tk
rd } z tj|� d|d< d |d
< W Y d d }~X nX |j|d< |jj� |d
< |S ) z3Scan the Web/URL (dict) and update the status key.r< TZ
ssl_verify�proxiesrY )Zallow_redirectsZverifyrZ rY �Errorr r r4 N)�requests�head� Exceptionr rK Z status_coder4 Z
total_seconds)r r5 Zreq�er r r rU � s
zThreadScanner._web_scanc C s( t |d �dkr| j|�S | j|�S dS )z9Scan the port structure (dict) and update the status key.r, r N)r' �_port_scan_icmp�_port_scan_tcp)r r, r r r rR s
zThreadScanner._port_scanc C sP |}yt j|�}W n8 tk
rJ } z tjdj| j||�� W Y dd}~X nX |S )z Convert hostname to IP address.z({}: Cannot convert {} to IP address ({})N)�socketZ
gethostbynamer^ r rK r= rO )r Zhostname�ipr_ r r r � _resolv_name s (zThreadScanner._resolv_namec C s d}t rd}d}ntstr$d}d}nd}d}d|d|t| j|d ��| j|d
�g}ttjd �}y:t� }t j
|||d d
�}|dkr�|j � |d< nd|d< W nX t j k
r� d|d< Y n< t
k
r� } z tjdj| j|d
|�� W Y dd}~X nX |j� |S )z@Scan the (ICMP) port structure (dict) and update the status key.Nz-wz-nz-tz-cz-WZping�1rY r8 �wT)�stdout�stderrZ close_fdsr r Fz${}: Error while pinging host {} ({}))r r r �strrd �open�os�devnullr �
subprocessZ
check_call�getZCalledProcessErrorr^ r rK r= rO �close) r r, r. Z timeout_optZ count_opt�cmdZfnull�counterr_ r r r r` s8 ,z ThreadScanner._port_scan_icmpc % C s� d}y"t j|d � t j t jt j�}W n6 tk
r\ } ztjdj| j|�� W Y dd}~X nX | j |d �}t
� }zvy|j |t |d �f�}W n8 tk
r� } z tjdj| j||�� W Y dd}~X n X |dkr�|j
� |d< nd |d< W d|j� X |S )
z?Scan the (TCP) port structure (dict) and update the status key.NrY z-{}: Error while creating scanning socket ({})r8 r, z%{}: Error while scanning port {} ({})r r F)rb ZsetdefaulttimeoutZAF_INETZ SOCK_STREAMr^ r rK r= rO rd r Z
connect_exr' rn ro )r r, r. Z_socketr_ rc rq r r r ra L s" &(
z ThreadScanner._port_scan_tcp)N)rF rG rH rI r rV �propertyr �setterr rQ rU rR rd r` ra rJ r r )r r r � s
.r ) rI rk rm rL rb rS r? Zglances.globalsr r r Zglances.ports_listr Zglances.web_listr Z
glances.timerr Zglances.compatr Zglances.loggerr Z glances.plugins.glances_pluginr
r\ rT � ImportErrorr_ Zwarningr= r ZThreadr r r r r �<module>
s, "