U
    @†6^  ã                   @   s0   d dl Zd dlmZmZmZ G dd„ deƒZdS )é    N)Ú
SpiderFootÚSpiderFootPluginÚSpiderFootEventc                   @   sJ   e Zd ZdZi Zi ZdZeƒ fdd„Zdd„ Z	dd„ Z
d	d
„ Zdd„ ZdS )Ú	sfp_quad9zWQuad9:Investigate,Passive:Reputation Systems::Check if a host would be blocked by Quad9Nc                 C   s4   || _ |  ¡ | _t| ¡ ƒD ]}|| | j|< qd S )N)ÚsfÚtempStorageÚresultsÚlistÚkeysÚopts)ÚselfÚsfcÚuserOptsÚopt© r   ú4/var/www/spiderfoot.crq.systems/modules/sfp_quad9.pyÚsetup   s    
zsfp_quad9.setupc                 C   s
   dddgS )NÚINTERNET_NAMEÚAFFILIATE_INTERNET_NAMEÚCO_HOSTED_SITEr   ©r   r   r   r   ÚwatchedEvents'   s    zsfp_quad9.watchedEventsc                 C   s
   dddgS )NÚMALICIOUS_INTERNET_NAMEÚ!MALICIOUS_AFFILIATE_INTERNET_NAMEÚMALICIOUS_COHOSTr   r   r   r   r   ÚproducedEvents-   s    ÿzsfp_quad9.producedEventsc              
   C   sz   t j ¡ }dg|_z"| |¡}| j dt|ƒ ¡ W n8 tk
rl } z| j d| ¡ W Y ¢dS d }~X Y nX |rvdS dS )Nz9.9.9.9zAddresses returned: zUnable to resolve FT)	ÚdnsÚresolverÚResolverÚnameserversÚqueryr   ÚdebugÚstrÚBaseException)r   ZqaddrÚresÚaddrsÚer   r   r   Ú	queryAddr1   s    

zsfp_quad9.queryAddrc              
   C   sà   |j }|j}|j}|}d}| j d| d | ¡ || jkr@d S d| j|< z| j |¡r\d}W n( tk
r† } z
W Y ¢d S d }~X Y nX |rÜ|  |¡}d| }	|dkrªd}	|sÜt	|	d| d	 d
 | d | j
|ƒ}
|  |
¡ d S )NFzReceived event, z, from TZ
MALICIOUS_r   r   zBlocked by Quad9 [z]
z%<SFURL>https://quad9.net/result/?url=z</SFURL>)Ú	eventTypeÚmoduleÚdatar   r!   r   ÚresolveHostr#   r'   r   Ú__name__ÚnotifyListeners)r   ÚeventÚ	eventNameÚsrcModuleNameÚ	eventDataZparentEventÚresolvedr&   ÚfoundÚtypÚevtr   r   r   ÚhandleEventA   s>    


ÿÿÿ þzsfp_quad9.handleEvent)r,   Ú
__module__Ú__qualname__Ú__doc__r   Úoptdescsr   Údictr   r   r   r'   r6   r   r   r   r   r      s   r   )Údns.resolverr   Úsflibr   r   r   r   r   r   r   r   Ú<module>   s   