U
    @6^-+                     @   sV   d dl mZmZ d dlZd dlmZmZmZ ddddgdd	d
iZG dd deZ	dS )    )	IPAddress	IPNetworkN)
SpiderFootSpiderFootPluginSpiderFootEventZ	PhishTank
_phishtanklistdomainz/http://data.phishtank.com/data/online-valid.csvz?\d+,\w+://(.*\.)?[^a-zA-Z0-9]?{0}.*,http://www.phishtank.com/.*)idtypechecksurlregexc                   @   st   e Zd ZdZdddddZdddd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_phishtankzmPhishTank:Investigate,Passive:Reputation Systems::Check if a host/domain is malicious according to PhishTank.T   )r   checkaffiliatescheckcohostscacheperiodzApply checks to affiliates?z?Apply checks to sites found to be co-hosted on the target's IP?z,Hours to cache list data before re-fetching.)r   r   r   Nc                 C   s4   || _ |  | _t| D ]}|| | j|< qd S )N)sftempStorageresultsr   keysopts)selfsfcuserOptsopt r   8/var/www/spiderfoot.crq.systems/modules/sfp_phishtank.pysetup5   s    
zsfp_phishtank.setupc                 C   s
   dddgS )NINTERNET_NAMEAFFILIATE_INTERNET_NAMECO_HOSTED_SITEr   r   r   r   r   watchedEventsA   s    zsfp_phishtank.watchedEventsc                 C   s
   dddgS )NMALICIOUS_INTERNET_NAME!MALICIOUS_AFFILIATE_INTERNET_NAMEMALICIOUS_COHOSTr   r#   r   r   r   producedEventsG   s    zsfp_phishtank.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   content	goodregexbadregexrxr   r   r   contentMaliciousL   s    zsfp_phishtank.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   queryr   _fetchtimeout
_useragenttimeout	useragentr/   Unable to fetch Fr0   r1   )r   r.   r   	malchecksr   strfetchUrlformatr   errorr3   )r   r
   target
targetTypecheckcidr   resr   r   r   resourceQuery`   s    $


zsfp_phishtank.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 r	   _internettldsr
   r   r   r   sfmal_r   r   r/   r5   r6   r7   r:   Fnetblockr   z{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   r;   r   dictcacheGetgetr=   r?   cachePutreplacer*   compiler,   r.   splitfindallr)   append
startswithstripr   r   	Exceptionr<   r>   r+   BaseException)r   r
   r@   rA   	targetDomrB   rC   datar   iplistr2   patlinegrpiperxDomrxTgtr   r   r   resourceListr   s     
   zsfp_phishtank.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   r4   r   )r   r;   r   r   r.   rE   rf   )r   
resourceIditemTyper@   rB   rC   r   r   r   
lookupItem   s    zsfp_phishtank.lookupItemc                 C   sv  |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 tt	
 D ]}t	| d }| j| r|dkrd}|dkrd}nd}|dkrd}d}|dkrd}|dkrd}|dkrd}|dkrd}| |||}	|  r2 d S |	d k	r|d | d d |	 d }
t||
| j|}| | qd S )NzReceived event, z, from z	Skipping z, already checked.Tr"   r   FAFFILIATE_IPADDRr   r
   )
IP_ADDRESSrj   rb   rk   MALICIOUS_IPADDRMALICIOUS_AFFILIATE_IPADDR)BGP_AS_OWNERBGP_AS_MEMBERasnMALICIOUS_ASN)r    r"   r!   r	   r    r%   r!   r&   r'   z [z]
z<SFURL>z</SFURL>)	eventTypemoduler]   r   r.   r   r   rQ   r   r;   r   ri   checkForStopr   __name__notifyListeners)r   event	eventNamesrcModuleName	eventDatarB   rC   typeIdevtTyper   textevtr   r   r   handleEvent   sR    






zsfp_phishtank.handleEvent)ru   
__module____qualname____doc__r   optdescsr   rO   r   r$   r(   r3   rE   rf   ri   r   r   r   r   r   r      s&   		Mr   )
netaddrr   r   r*   sflibr   r   r   r;   r   r   r   r   r   <module>   s   