U
    @6^>,                     @   s\   d dl Z d dlmZmZ d dlmZmZmZ ddddgdd	d
gg diZG dd deZ	dS )    N)	IPAddress	IPNetwork)
SpiderFootSpiderFootPluginSpiderFootEventz&Watchguard Reputation Authority Lookup_watchguardqueryipz,http://reputationauthority.org/lookup?ip={0}z.*>[6-9][0-9]/100 </td>.*z.*>100/100 </td>.*)idtypechecksurlbadregex	goodregexc                   @   sl   e Zd ZdZdddZddiZdZe fddZd	d
 Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )sfp_watchguardzWatchguard:Investigate,Passive:Reputation Systems::Check if an IP is malicious according to Watchguard's reputationauthority.org.T)r   checkaffiliatesr   zApply checks to affiliates?Nc                 C   s4   || _ |  | _t| D ]}|| | j|< qd S )N)sftempStorageresultslistkeysopts)selfsfcuserOptsopt r   9/var/www/spiderfoot.crq.systems/modules/sfp_watchguard.pysetup1   s    
zsfp_watchguard.setupc                 C   s   ddgS )N
IP_ADDRESSAFFILIATE_IPADDRr   r   r   r   r   watchedEvents=   s    zsfp_watchguard.watchedEventsc                 C   s   dgS )NZ*MALICIOUS_IPADDRMALICIOUS_AFFILIATE_IPADDRr   r!   r   r   r   producedEventsC   s    zsfp_watchguard.producedEventsc                 C   s   t |dkrB|D ]0}t||tjtjB r| jd|   dS qt |dkr|D ]0}t||tjtjB rR| jd|   dS qR| jd d S )Nr   z#Found to be bad against bad regex: Tz$Found to be good againt good regex: FzNeither good nor bad, unknown.)lenrematch
IGNORECASEDOTALLr   debug)r   contentr   r   rxr   r   r   contentMaliciousG   s    zsfp_watchguard.contentMaliciousc                 C   s   | j d| d |  tt D ]}t| d }||kr$t| d dkr$tt| d }| j j||| jd | jd d	}|d
 d kr| j 	d|| d  d S | 
|d
 t| d t| d r$||  S q$d S )Nz	Querying z for maliciousness of r
   r   r   r   _fetchtimeout
_useragenttimeout	useragentr*   Unable to fetch Fr   r   )r   r)   r   	malchecksr   strfetchUrlformatr   errorr,   )r   r
   target
targetTypecheckcidr   resr   r   r   resourceQuery[   s    $


zsfp_watchguard.resourceQueryc                 C   s  d}|dkr | j || jd }tt D ]t}t| d }||kr,t| d dkr,t }t| d }| j d| | jd	d
|d< |d d kr| j j	|| jd | jd d}|d d kr| j 
d| d  d S | j d| |d  |dkrNt }	dt| krt| d dd}
t|
tj}| j d| d |
  |d dD ].}t||}t|d
krR|	|d
  qRn|d d}	|	D ]}t|dk s|drq| }z@t|t|kr| j |d | d |  |W     S W nB tk
rB } z"| j dt|  W Y qW 5 d }~X Y nX q d S dt| kr|d dD ]P}||ks|dkrj||krj| j |d | d | d  |    S qjq,ztt| d |}tt| d |}|d dD ]b}|dkrt||tjs,t||tjr| j |d | d | d  |  W   S qW q, tk
r } z | j d t|  W Y q,W 5 d }~X Y q,X q,d S )!N domain_internettldsr
   r   r   r   sfmal_cacheperiodr   r*   r-   r.   r/   r2   Fnetblockregexz{0}z(\d+\.\d+\.\d+\.\d+)zNew regex for z: 
   #z found within netblock/subnet z in zError encountered parsing: /z
 found in z list.zError encountered parsing 2: )r   
hostDomainr   r   r3   r   dictcacheGetgetr5   r7   cachePutreplacer%   compiler'   r)   splitfindallr$   append
startswithstripr   r   	Exceptionr4   r6   r&   BaseException)r   r
   r8   r9   	targetDomr:   r;   datar   iplistr+   patlinegrpr	   erxDomrxTgtr   r   r   resourceListm   s     
   zsfp_watchguard.resourceListc                 C   s   t t D ]}t| d }||kr|t| d kr| jd| d | d |  t| d dkrv| |||  S t| d dkr| |||  S qd S )	Nr
   r   zChecking maliciousness of z (z) with: r   r   r   )r   r3   r   r   r)   r=   r`   )r   
resourceIditemTyper8   r:   r;   r   r   r   
lookupItem   s    zsfp_watchguard.lookupItemc                 C   s  |j }|j}|j}| jd| d |  || jkrL| jd| d  d S d| j|< |dkrp| jddspd S |d	kr| jd
dsd S |dkr| jddsd S |dkr| jddsd S tt	
 D ]}t	| d }| j| r|dkrd}|dkrd}nd}|dkrd}d}|dkrRd}|dkr6d}|dkrDd}|dkrRd}|dkrdd}d }|dkrvd}d!}| |||}	|  r d S |	d k	r|d" | d# d$ |	 d% }
t||
| j|}| | qd S )&NzReceived event, z, from z	Skipping z, already checked.TCO_HOSTED_SITEcheckcohostsFr    r   NETBLOCK_OWNERchecknetblocksNETBLOCK_MEMBERchecksubnetsr
   )r   r    r	   r   MALICIOUS_IPADDRMALICIOUS_AFFILIATE_IPADDR)BGP_AS_OWNERBGP_AS_MEMBERasnMALICIOUS_ASN)INTERNET_NAMErd   AFFILIATE_INTERNET_NAMEr?   rp   MALICIOUS_INTERNET_NAMErq   !MALICIOUS_AFFILIATE_INTERNET_NAMEMALICIOUS_COHOSTrC   MALICIOUS_NETBLOCKMALICIOUS_SUBNETz [z]
z<SFURL>z</SFURL>)	eventTypemodulerX   r   r)   r   r   rL   r   r3   r   rc   checkForStopr   __name__notifyListeners)r   event	eventNamesrcModuleName	eventDatar:   r;   typeIdevtTyper   textevtr   r   r   handleEvent   sf    












zsfp_watchguard.handleEvent)rz   
__module____qualname____doc__r   optdescsr   rJ   r   r"   r#   r,   r=   r`   rc   r   r   r   r   r   r      s     Mr   )
r%   netaddrr   r   sflibr   r   r   r3   r   r   r   r   r   <module>   s   