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dgd	d
iZG dd deZ	dS )    )	IPAddress	IPNetworkN)
SpiderFootSpiderFootPluginSpiderFootEventztalosintelligence.com List_talosintellistipnetblockz4https://talosintelligence.com/documents/ip-blacklist)idtypechecksurlc                   @   sx   e Zd ZdZddddddZd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_talosintelzTalos Intelligence:Investigate,Passive:Reputation Systems::Check if a netblock or IP is malicious according to talosintelligence.com.T   )r   checkaffiliatescacheperiodchecknetblockschecksubnetszApply checks to affiliates?z,Hours to cache list data before re-fetching.z=Report if any malicious IPs are found within owned netblocks?zJCheck if any malicious IPs are found within the same subnet of the target?)r   r   r   r   Nc                 C   s4   || _ |  | _t| D ]}|| | j|< qd S )N)sftempStorageresultsr   keysopts)selfsfcuserOptsopt r   9/var/www/spiderfoot.crq.systems/modules/sfp_talosintel.pysetup6   s    
zsfp_talosintel.setupc                 C   s   ddddgS )N
IP_ADDRESSNETBLOCK_MEMBERAFFILIATE_IPADDRNETBLOCK_OWNERr   r   r   r   r   watchedEventsB   s    zsfp_talosintel.watchedEventsc                 C   s   ddddgS )NMALICIOUS_IPADDRMALICIOUS_AFFILIATE_IPADDRMALICIOUS_SUBNETMALICIOUS_NETBLOCKr   r%   r   r   r   producedEventsI   s     zsfp_talosintel.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   contentMaliciousN   s    zsfp_talosintel.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	useragentr2   Unable to fetch Fr3   r4   )r   r1   r   	malchecksr   strfetchUrlformatr   errorr6   )r   r   target
targetTypecheckcidr   resr   r   r   resourceQueryb   s    $


zsfp_talosintel.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_r   r   r2   r8   r9   r:   r=   Fr
   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   r>   r   dictcacheGetgetr@   rB   cachePutreplacer-   compiler/   r1   splitfindallr,   append
startswithstripr   r   	Exceptionr?   rA   r.   BaseException)r   r   rC   rD   	targetDomrE   rF   datar   iplistr5   patlinegrpr	   erxDomrxTgtr   r   r   resourceListt   s     
   zsfp_talosintel.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   r7   r   )r   r>   r   r   r1   rH   ri   )r   
resourceIditemTyperC   rE   rF   r   r   r   
lookupItem   s    zsfp_talosintel.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   r$   r   r"   r   r   )r!   r#   r	   r!   r'   r(   )BGP_AS_OWNERBGP_AS_MEMBERasnMALICIOUS_ASN)INTERNET_NAMErm   AFFILIATE_INTERNET_NAMErJ   rs   MALICIOUS_INTERNET_NAMErt   !MALICIOUS_AFFILIATE_INTERNET_NAMEMALICIOUS_COHOSTr
   r*   r)   z [z]
z<SFURL>z</SFURL>)	eventTypemodulera   r   r1   r   r   rU   r   r>   r   rl   checkForStopr   __name__notifyListeners)r   event	eventNamesrcModuleName	eventDatarE   rF   typeIdevtTyper   textevtr   r   r   handleEvent   sf    












zsfp_talosintel.handleEvent)r{   
__module____qualname____doc__r   optdescsr   rS   r    r&   r+   r6   rH   ri   rl   r   r   r   r   r   r      s*   

Mr   )
netaddrr   r   r-   sflibr   r   r   r>   r   r   r   r   r   <module>   s   
