Current File : //lib/python3.6/site-packages/pyzor/__pycache__/client.cpython-36.pyc
3
�2�^J* � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl Zddl
Zddl Zej j
j� G dd� de�ZG dd� de�ZG dd� de�ZG d d
� d
e�ZG d d � d e�ZdS )
ay Networked spam-signature detection client.
>>> import pyzor
>>> import pyzor.client
>>> import pyzor.digest
>>> import pyzor.config
To load the accounts file:
>>> accounts = pyzor.config.load_accounts(filename)
To create a client (to then issue commands):
>>> client = pyzor.client.Client(accounts)
To create a client, using the anonymous user:
>>> client = pyzor.client.Client()
To get a digest (of an email.message.Message object, or similar):
>>> digest = pyzor.digest.DataDigester(msg).value
To query a server (where address is a (host, port) pair):
>>> client.ping(address)
>>> client.info(digest, address)
>>> client.report(digest, address)
>>> client.whitelist(digest, address)
>>> client.check(digest, address)
To query the default server (public.pyzor.org):
>>> client.ping()
>>> client.info(digest)
>>> client.report(digest)
>>> client.whitelist(digest)
>>> client.check(digest)
Response will contain, depending on the type of request, some
of the following keys (e.g. client.ping()['Code']):
All responses will have:
- 'Diag' 'OK' or error message
- 'Code' '200' if OK
- 'PV' Protocol Version
- 'Thread'
`info` and `check` responses will also contain:
- '[WL-]Count' Whitelist/Blacklist count
`info` responses will also have:
- '[WL-]Entered' timestamp when message was first whitelisted/blacklisted
- '[WL-]Updated' timestamp when message was last whitelisted/blacklisted
� Nc @ s� e Zd ZdZdZd dd�Zd dd �Zd d
d �Zd"d d
�Zd$dd�Z d&dd�Z
d(dd�Z d)dd�Z d+dd�Z
edd� �Zdd� ZdS ),�Client� i Nc C sX |d kr i }t dd� |j� D ��| _|d kr4tjj}|| _|d k rH|| _tj d�| _
d S )Nc s s& | ] \\}}}|t |�f|fV qd S )N)�int)�.0�hostZport�account� r � /usr/lib/python3.6/client.py� <genexpr>P s z"Client.__init__.<locals>.<genexpr>�pyzor) �dict�items�accountsr �digestZ digest_spec�spec�timeout�logging� getLogger�log)�selfr r r r r r �__init__M s zClient.__init__�public.pyzor.org�y_ c C s& t jj� }| j||�}| j||j� �S )N)r �messageZ PingRequest�send�
read_response�
get_thread)r �address�msg�sockr r r �pingY s
z Client.pingc C s( t jj|�}| j||�}| j||j� �S )N)r r Z PongRequestr r r )r r r r r r r r �pong^ s z Client.pongc C s( t jj|�}| j||�}| j||j� �S )N)r r Z InfoRequestr r r )r r r r r r r r �infoc s z Client.infoc C s, t jj|| j�}| j||�}| j||j� �S )N)r r �
ReportRequestr r r r )r r r r r r r r �reporth s z
Client.reportc C s, t jj|| j�}| j||�}| j||j� �S )N)r r �WhitelistRequestr r r r )r r r r r r r r � whitelistm s zClient.whitelistc C s( t jj|�}| j||�}| j||j� �S )N)r r Z CheckRequestr r r )r r r r r r r r �checkr s z Client.checkc C s, dt jjjt jf jd�}tj|t jjd�S )Nz:Code: %s
Diag: OK
PV: %s
Thread: 1024
Count: 0
WL-Count: 0�ascii)�_class)r r �ResponseZok_codeZ
proto_version�encode�email�message_from_bytes)r Zdigestsr r r r r � _mock_checkw s zClient._mock_checkc C s� |d t |d �f}|j� y| j| }W n tk
rF tjj}Y nX t tj� �}|j|d< t |�|d< tjj
tjj |j |j�||�|d< | j
jd|j� � | j||�S )Nr � ZUserZTimeZSigz sending: %r)r Zinit_for_sendingr �KeyErrorr r ZAnonymousAccount�timeZusername�strZsign_msgZhash_key�keyr �debug� as_string�_send)r r r r Z timestampr r r r } s
z Client.sendc C s� d }x�t j|d |d dt jt j�D ]�}|\}}}}}yt j |||�}W n t jk
rd d }w$Y nX y |j| j� jd�d|� W nN t jk
r� |j � t
j d| ��Y n$ t jk
r� |j � d }w$Y nX P q$W |d kr�t
j d| ��|S )Nr r/ �utf8zSending to %s time-outedzUnable to send to %s:%s)
�socketZ getaddrinfoZ
SOCK_DGRAMZ IPPROTO_UDP�errorZsendtor5 r+ r �closer � TimeoutError� CommError) r Zaddrr �resZafZsocktype�proto�_Zsar r r r6 � s, z Client._sendc C s |j | j� y|j| j�\}}W nn tjk
rV } z|j� tjd��W Y d d }~X n: tjk
r� } z |j� tj d| ��W Y d d }~X nX | j
j d||� t j
|tjjd�}|j� y>|j� }||kr�|j� r�tjd||f ��| j
jd||� W n" tk
�r | j
jd� Y nX |S )NzReading response timed-out.z'Socket error while reading response: %szreceived: %r/%r)r) z.received unexpected thread id %d (expected %d)z)received error thread id %d (expected %d)zno thread id received)Z
settimeoutr Zrecvfrom�max_packet_sizer8 r: r r; r9 r< r r4 r, r- r r* Zensure_completer Z in_ok_rangeZ
ProtocolError�warnr0 )r r Z expected_idZpacketr Zexr Z thread_idr r r r � s2 zClient.read_response)NNN�r r )rB �r r )rC �r r )rD �r r )rE �r r )rF �r r )rG )N�r r )rH )�__name__�
__module__� __qualname__r r@ r r r! r" r$ r& r' r. r � staticmethodr6 r r r r r r I s
r c @ sN e Zd ZdZddd�Zddd �Zdd
d �Zd d
� Zdd� Zdd� Z dd� Z
dS )� BatchClientz=Like the normal Client but with support for batching reports.N�
c C s0 t j| |||d� || _i | _i | _| j� d S )N)r r r )r r �
batch_size�
r_requests�
w_requests�flush)r r r r rO r r r r � s
zBatchClient.__init__�public.pyzor.org�y_ c C s | j ||| j� d S )N)� _add_digestrP )r r r r r r r$ � s zBatchClient.reportc C s | j ||| j� d S )N)rU rQ )r r r r r r r& � s zBatchClient.whitelistc C sL |d t |d �f}|| }|j|� |j| jkrHz | j||�S ||= X d S )Nr r/ )r Z
add_digestZ digest_countrO r )r r r Zrequestsr r r r rU � s
zBatchClient._add_digestc C s< t jtjtjj| jd��| _t jtjtjj | jd��| _
dS )z"Deleting any saved digest reports.)r N) � collections� defaultdict� functools�partialr r r# r rP r% rQ )r r r r rR � s zBatchClient.flushc
C st x6| j j� D ](\}}y| j||� W q w Y q X q W x6| jj� D ](\}}y| j||� W qD wDY qDX qDW dS )z!Force send any remaining reports.N)rP r
r rQ )r r r r r r �force� s zBatchClient.forcec C s | j � d S )N)rZ )r r r r �__del__� s zBatchClient.__del__)NNNrN �rS rT )r\ �rS rT )r] ) rI rJ rK �__doc__r r$ r& rU rR rZ r[ r r r r rM � s
rM c @ s& e Zd Zdd� Zddd�Zdd� ZdS ) � ClientRunnerc C s" t jd�| _|| _d| _g | _d S )Nr T)r r r �routine�all_ok�results)r r` r r r r � s zClientRunner.__init__Nc C s� |d kr i }d| }d }y | j ||�}| j||� W nd tjttfk
r� } z@| jjd||jt |�ff � | j
j d||j j
|� d| _W Y d d }~X nX d S )Nz%s:%s z%s%s
z %s %s: %sF)r` �handle_responser r< r0 �
ValueErrorrb �append�coder2 r r9 � __class__rI ra )r Zserver�args�kwargsr �response�er r r �run� s zClientRunner.runc C s* |j � sd| _| jjd||j� f � dS )z)mesaage is a string we've built up so farFz%s%s
N)�is_okra rb re �
head_tuple)r rj r r r r rc s z ClientRunner.handle_response)N)rI rJ rK r rl rc r r r r r_ � s
r_ c @ s e Zd Zddd�Zdd� ZdS )�CheckClientRunnerr c C s4 t j| |� d| _d| _d| _d| _|| _|| _d S )NFr )r_ r � found_hit� whitelisted� hit_count�whitelist_count�
r_count_found�wl_count_clears)r r` Zr_countZwl_countr r r r s zCheckClientRunner.__init__c C s� |dt |j� � 7 }|j� rtt|d �| _t|d �| _| j| jkrLd| _n| j| jkr^d| _ |d| j| jf 7 }nd| _
| j j |d � d S )Nz%s �CountzWL-CountTz%d %dF�
)
r2 rn rm r rr rs ru rq rt rp ra rb re )r rj r r r r rc s z!CheckClientRunner.handle_responseN)r r )rI rJ rK r rc r r r r ro s
ro c @ s e Zd Zdd� ZdS )�InfoClientRunnerc C s� |dt |j� � 7 }|j� rzxbd
D ]R}||kr"t|| �}d|krLt |�}n|dkrZd }n
tj|�}|d
||f 7 }q"W nd | _| jj|d � d S )Nz%s
rv �Entered�Updated�WL-Count�
WL-Entered�
WL-Updatedr/ ZNeverz %s: %s
Frw )rv ry rz r{ r| r} ���) r2 rn rm r r1 Zctimera rb re )r rj r r3 �valZstringedr r r rc * s
z InfoClientRunner.handle_responseN)rI rJ rK rc r r r r rx ) s rx )r^ r1 r, r8 r rX rV Z pyzor.digestr Z
pyzor.accountZ
pyzor.messageZpyzor.hacks.py26ZhacksZpy26Z
hack_email�objectr rM r_ ro rx r r r r �<module>7 s y3
Mini Shell