File: //opt/alt/python37/lib/python3.7/site-packages/pydnsbl/__pycache__/checker.cpython-37.pyc
B
��f� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m Z m Z m
Z
ejdkr�ejdkr�e�e�� � G dd� d�ZG d d
� d
�ZG d d � d ej�ZG d
d� de�ZG dd� de�ZG dd� de�ZdS )z�
DNSBL async checker
Basic usage:
checker = DNSBLChecker()
result = cheker.check_ip('...')
print(result.blacklisted)
print(result.categories)
print(result.detected_by)
� N� )�Provider�BASE_PROVIDERS�BASE_DOMAIN_PROVIDERS�win32)� � c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
� DNSBLResultaA
DNSBL Result class to keep all info about ip request results.
Attributes:
* addr - checked ip
* providers - dnsbl that was asked for response while checking
* failed_provider - dnsbl that was unable to provide result due
to connection issues (connection timeout etc...)
* detected_by - dnsbl that have ip listed and categories detected by
this dnsbls. dict: {'dnsbl_list_name': list(categories_from_this_dnsbl)}
* categories - set of dnsbl categories from all providers (subset of DNSBL_CATEGORIES)
Nc C s8 || _ || _d| _g | _g | _i | _t� | _| �� d S )NF) �addr�_results� blacklisted� providers�failed_providers� detected_by�set�
categories�process_results)�selfr
�results� r �@/opt/alt/python37/lib/python3.7/site-packages/pydnsbl/checker.py�__init__( s zDNSBLResult.__init__c C sr xl| j D ]b}|j}| j�|� |jr2| j�|� q|js:qd| _|�|j�}| j �
|�| _ t |�| j |j
< qW dS )z Process results by providers TN)r �providerr
�append�errorr �responser �process_responser �union�listr �host)r �resultr Zprovider_categoriesr r r r 2 s zDNSBLResult.process_resultsc C s, | j r
dnd}d| j|t| j�t| j�f S )Nz
[BLACKLISTED]� z <DNSBLResult: %s %s (%d/%d)>)r r
�lenr r
)r r r r r �__repr__B s zDNSBLResult.__repr__)NN)�__name__�
__module__� __qualname__�__doc__r r r# r r r r r s
r c @ s e Zd ZdZddd�ZdS )�
DNSBLResponsez
DNSBL Response object
Nc C s || _ || _|| _|| _d S )N)r
r r r )r r
r r r r r r r K s zDNSBLResponse.__init__)NNNN)r$ r% r& r' r r r r r r( G s r( c @ sR e Zd ZdZeddddfdd�Zdd � Zejd
d � �Z d d
� Z
dd� Z dd� Z dS )�BaseDNSBLCheckerz� BASE Checker for DNSBL lists
Arguments:
* providers(list) - list of providers (Provider instance or str)
* timeout(int) - timeout of dns requests will be passed to resolver
* tries(int) - retry times
� � �� Nc C s� g | _ x*|D ]"}t|t�s"td��| j �|� q W |sjt�� t�� krRt� � | _
qpt� � | _
t� | j
� n|| _
t
j||| j
d�| _t�|�| _d S )Nz0providers should contain only Provider instances)�timeout�tries�loop)r
�
isinstancer �
ValueErrorr � threading�current_thread� main_thread�asyncioZget_event_loop�_loopZnew_event_loopZset_event_loop�aiodnsZ DNSResolver� _resolver� Semaphore�
_semaphore)r r
r- r. � concurrencyr/ r r r r r Y s
zBaseDNSBLChecker.__init__c
� s� d}d}d| � |�|jf }y8| j4 I dH � | j�|d�I dH }W dQ I dH R X W n8 tjjk
r� } z|jd dkr||}W dd}~X Y nX t ||||d�S )a
Make lookup to dnsbl provider for ip
Parameters:
* addr (string) - ip address to check
* provider (string) - dnsbl provider
Returns:
* DNSBLResponse object
Raises:
* ValueError
Nz%s.%s�Ar � )r
r r r )
�
prepare_queryr r: r8 �queryr7 r ZDNSError�argsr( )r �requestr r r Z dnsbl_query�excr r r �
dnsbl_requestl s
(z BaseDNSBLChecker.dnsbl_requestc C s t S )z(
Prepare query to dnsbl
)�NotImplemented)r rA r r r r> � s z BaseDNSBLChecker.prepare_queryc � sB g }x | j D ]}|�| �||�� q W tj|� I d H }t||d�S )N)r
r )r
r rC r5 �gatherr )r rA �tasksr r r r r � check_async� s
z BaseDNSBLChecker.check_asyncc C s | j �| �|��S )N)r6 �run_until_completerG )r rA r r r �check� s zBaseDNSBLChecker.checkc C s4 g }x |D ]}|� | �|�� q
W | j�tj|� �S )N)r rG r6 rH r5 rE )r �requestsrF rA r r r �
bulk_check� s
zBaseDNSBLChecker.bulk_check)
r$ r% r& r' r r rC �abc�abstractmethodr> rG rI rK r r r r r) Q s r) c @ s e Zd ZdZdd� ZdS )�DNSBLIpCheckerz
Checker for ips
c C sd t �|�}|jdkr(d�t|�d���S |jdkrX|j�dd�}d�tdd� |D ���S td��d S ) Nr= �.� �:r! c S s g | ]}|�qS r r )�.0�xr r r �
<listcomp>� s z0DNSBLIpChecker.prepare_query.<locals>.<listcomp>zunknown ip version) � ipaddress�
ip_address�version�join�reversed�split�exploded�replacer1 )r rA �addressZrequest_strippedr r r r> � s
z DNSBLIpChecker.prepare_queryN)r$ r% r& r' r> r r r r rN � s rN c s> e Zd ZdZe�d�Zeddddf� fdd� Zd d
� Z � Z
S ) �DNSBLDomainCheckerz
Checker for domains
zq^(((?!-))(xn--|_{1,1})?[a-z0-9-]{0,61}[a-z0-9]{1,1}\.)*(xn--[a-z0-9][a-z0-9\-]{0,60}|[a-z0-9-]{1,30}\.[a-z]{2,})$r* r+ r, Nc s t � j|||||d� d S )N)r
r- r. r; r/ )�superr )r r
r- r. r; r/ )� __class__r r r � s
zDNSBLDomainChecker.__init__c C s2 |� � }t�|��� }| j�|�s.td| ��|S )Nz should be valid domain, got %s)�lower�idna�encode�decode� DOMAIN_REGEX�matchr1 )r rA Z domain_idnar r r r> � s
z DNSBLDomainChecker.prepare_query) r$ r% r& r' �re�compilere r r r> �
__classcell__r r )r` r r^ � s
r^ c s0 e Zd ZdZ� fdd�Zdd� Zdd� Z� ZS )� DNSBLCheckerz0
Will be deprecated, use DNSBLIpChecker
c s t �dt� t� j||� d S )Nz deprecated, use DNSBLIpChecker)�warnings�warn�DeprecationWarningr_ r )r r@ �kwargs)r` r r r � s zDNSBLChecker.__init__c C s t �dt� | �|�S )Nz$deprecated, use check method instead)rk rl rm rI )r r
r r r �check_ip� s zDNSBLChecker.check_ipc C s t �dt� | �|�S )Nz)deprecated, use bulk check method instead)rk rl rm rK )r �addrsr r r � check_ips� s zDNSBLChecker.check_ips)r$ r% r& r' r ro rq ri r r )r` r rj � s rj )r' rL r5 rb rU rg �sysr2 rk r7 r
r r r �platform� version_infoZset_event_loop_policyZ WindowsSelectorEventLoopPolicyr r( �ABCr) rN r^ rj r r r r �<module> s$ ,
L