o va1 @sdZddlZddlZddlZddlZddlZddlZddlmZm Z m Z ddl m Z ddl mZddlmZmZddlmZddlmZdd lmZzddlZd ZWneynZzd Zed eWYdZ[ndZ[wwGd ddeZGdddejZ dS)zPorts scanner plugin.N)WINDOWSMACOSBSD)GlancesPortsList)GlancesWebList)TimerCounter) bool_type)logger) GlancesPluginTFzAMissing Python Lib ({}), Ports plugin is limited to port scanningcsheZdZdZdfdd ZfddZejejddZ d d Z dd dZ dddZ dddZ ZS)PluginzGlances ports scanner plugin.NcsTtt|j||gd||_||_d|_t||dt||d |_ d|_ dS)zInit the plugin.)argsconfigZstats_init_valueT)rr N) superr __init__r rZ display_curserZget_ports_listrZ get_web_liststats_thread)selfr r __class__?/usr/lib/python3/dist-packages/glances/plugins/glances_ports.pyr0s  zPlugin.__init__cs&|jdur |jtt|dS)z+Overwrite the exit method to close threads.N)rstoprr exitrrrrrBs  z Plugin.exitcCsL|jdkr"|jdur d}n|j}|st|j|_|j|jS |jS)zUpdate the ports list.ZlocalNF)Z input_methodrZis_alive ThreadScannerrstart)rZthread_is_runningrrrupdateIs      z Plugin.updatecCsdS)zReturn the key of the list.indicerrrrrget_keydszPlugin.get_keyFcCsd}|ddur d}n"|ddkrd}nt|dttfr-|ddur-|d|dkr-d}|j|d }||||||||||S) z?Return the alert status relative to the port scan return value.OKstatusNCAREFULrCRITICAL rtt_warningWARNINGheader) isinstancefloatint get_stat_namemanage_threshold manage_actionlowerr)rportr(logret stat_namerrrget_ports_alerths&      zPlugin.get_ports_alertcCsd}|ddur d}n|ddvrd}n|ddur$|d|dkr$d }|j|d }||||||||||S) zBReturn the alert status relative to the web/url scan return value.r!r"Nr#)i-i.r$r%elapsedr&r')r,r-r.r/r)rwebr(r1r2r3rrr get_web_alerts     zPlugin.get_web_alertc Csg}|jr|jr |S|d}|jD]}d|vr|ddur d}n+|ddur)d}n"t|dtr9|ddur9d}n|dd krBd }n d |dd }d j|dd ||d}|||d|}||||j||ddd||qd|vrd j|dd ||d}|||t|dt j rd|d}n |ddurd}n|d}d|}||||j ||ddd||qz| W|St yY|Sw)z2Return the dict to display in the curse interface.hostNNoner"ZScanningTZOpenrZTimeoutz {0:.0f}msg@@z {:{width}} description)widthz{:>9}rZ_rttr'urlzCode {})rZ disable_portsr)r formatappendZcurse_add_liner4Zcurse_new_linenumbersNumberr8pop IndexError)rr Z max_widthr2Zname_max_widthpr"msgrrr msg_cursesf              zPlugin.msg_curse)NN)r F)__name__ __module__ __qualname____doc__rrr Z_check_decoratorZ_log_result_decoratorrrr4r8rG __classcell__rrrrr -s    r cs|eZdZdZfddZddZeddZejddZdd d Z d d Z ddZ ddZ ddZ ddZddZZS)rzP Specific thread for the port/web scanner. stats is a list of dict cs8td|tt|t|_||_ d|_ dS)zInit the class.z-ports plugin - Create thread for scan list {}ZportsN) r debugr?rrr threadingZEvent_stopper_stats plugin_name)rrrrrrs   zThreadScanner.__init__cCsR|jD]#}|r dSd|vr||tdqd|vr&tr&||qdS)z`Grab the stats. Infinite loop, should be stopped by calling the stop() method. r0r>N)rPstopped _port_scantimesleep requests_tag _web_scan)rrErrrruns     zThreadScanner.runcCs|jS)z Stats getter.rPrrrrrszThreadScanner.statscCs ||_dS)z Stats setter.NrZ)rvaluerrrrs NcCs td|j|jdS)zStop the thread.z,ports plugin - Close thread for scan list {}N)r rMr?rPrOset)rtimeoutrrrrszThreadScanner.stopcCs |jS)z%Return True is the thread is stopped.)rOZisSetrrrrrS s zThreadScanner.stoppedc Csztj|dd|d|d|dd}Wn ty4}zt|d|d<d |d <WYd }~|Sd }~ww|j|d<|j|d <|S) z3Scan the Web/URL (dict) and update the status key.r>TZ ssl_verifyproxiesr])Zallow_redirectsZverifyr^r]Errorr"rr6N)requestshead Exceptionr rMZ status_coder6Z total_seconds)rr7ZreqerrrrXs"    zThreadScanner._web_scancCs$t|ddkr ||S||S)z9Scan the port structure (dict) and update the status key.r0r)r+_port_scan_icmp_port_scan_tcp)rr0rrrrTs  zThreadScanner._port_scanc CsR|}zt|}W|Sty(}ztd|j||WYd}~|Sd}~ww)zConvert hostname to IP address.z({}: Cannot convert {} to IP address ({})N)socketZ gethostbynamerbr rMr?rQ)rZhostnameiprcrrr _resolv_name&s  zThreadScanner._resolv_namec Csd}tr d}d}n ts trd}d}nd}d}d|d|t||d ||d g}ttjd }z t}t j |||d d }|dkrJ| |d<W|Sd|d<W|St j y_d|d<Y|St y~}ztd|j|d |WYd}~|Sd}~ww)z@Scan the (ICMP) port structure (dict) and update the status key.Nz-wz-nz-tz-cz-WZping1r]r:wT)stdoutstderrZ close_fdsrr"Fz${}: Error while pinging host {} ({}))rrrstrrhopenosdevnullr subprocessZ check_callgetZCalledProcessErrorrbr rMr?rQ) rr0r2Z timeout_optZ count_optcmdZfnullcounterrcrrrrd/s>    $zThreadScanner._port_scan_icmpc Csd}zt|dttjtj}Wnty/}ztd|j|WYd}~nd}~ww| |d}t }zOz | |t |df}Wntye}ztd|j||WYd}~nd}~ww|dkrq| |d<nd |d<W||SW||SW||S|w) z?Scan the (TCP) port structure (dict) and update the status key.Nr]z-{}: Error while creating scanning socket ({})r:r0z%{}: Error while scanning port {} ({})rr"F)rfZsetdefaulttimeoutZAF_INETZ SOCK_STREAMrbr rMr?rQrhrZ connect_exr+rrclose)rr0r2Z_socketrcrgrtrrrreWs8 "  zThreadScanner._port_scan_tcp)N)rHrIrJrKrrYpropertyrsetterrrSrXrTrhrdrerLrrrrrs      (r)!rKrorqrNrfrUrAZglances.globalsrrrZglances.ports_listrZglances.web_listrZ glances.timerrrZglances.compatr Zglances.loggerr Zglances.plugins.glances_pluginr r`rW ImportErrorrcZwarningr?r ZThreadrrrrrs2     *