Current File : //lib/python3.6/site-packages/pyzor/__pycache__/server.cpython-36.pyc
3
�2�^�= �
@ s" d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z y ddl Z W n e
k
r| ddlZ Y nX ddlZddlZddlZddlZejjj� dd� ZG dd� de j�ZG dd� de�ZG d d
� d
e je�Z G d d � d e �Z G d
d� de j e�Z G dd� de j �Z!dS )a� Networked spam-signature detection server.
The server receives the request in the form of a RFC5321 message, and
responds with another RFC5321 message. Neither of these messages has a
body - all of the data is encapsulated in the headers.
The response headers will always include a "Code" header, which is a
HTTP-style response code, and a "Diag" header, which is a human-readable
message explaining the response code (typically this will be "OK").
Both the request and response headers always include a "PV" header, which
indicates the protocol version that is being used (in a major.minor format).
Both the requestion and response headers also always include a "Thread",
which uniquely identifies the request (this is a requirement of using UDP).
Responses to requests may arrive in any order, but the "Thread" header of
a response will always match the "Thread" header of the appropriate request.
Authenticated requests must also have "User", "Time" (timestamp), and "Sig"
(signature) headers.
� Nc G sH xBy| |� S t k
r> } z|jd tjkr.� W Y dd}~X qX qW dS )z*restart a system call interrupted by EINTRr N)�OSError�args�errnoZEINTR)�funcr �e� r � /usr/lib/python3.6/server.py� _eintr_retry/ s r c @ sB e Zd ZdZdZdZddd�Zdd� Zd d
� Zd d � Z d
d� Z
dS )�Serverz_The pyzord server. Handles incoming UDP connections in a single
thread and single process.i � Nc C s d|d krt jt_nt jt_tjd�| _tjd�| _|| _ t
| j dd�| _ || _ || _
i | _i | _| j� || _| jjd|� tjj| |tdd� y| j jt jt jd� W n6 tt jfk
r� } z| jjd |� W Y d d }~X nX | j � | j � t j t j | j � t j t j!| j"� d S )
N�:r Zpyzordz pyzord-usageZhandles_one_stepFzListening on %s)Zbind_and_activatez%Unable to set IPV6_V6ONLY to false %s)#�socketZAF_INET6r
Zaddress_familyZAF_INET�loggingZ getLogger�log� usage_log�database�getattr�one_step� passwd_fn� access_fn�accounts�acl� load_config� forwarder�debug� SocketServer� UDPServer�__init__�RequestHandlerZ
setsockoptZ IPPROTO_IPV6Z IPV6_V6ONLY�AttributeError�errorZ server_bindZserver_activate�signal�SIGUSR1�reload_handler�SIGTERM�shutdown_handler)�self�addressr r r r r r r r r ? s0
zServer.__init__c C s( t jj| j�| _t jj| j| j�| _dS )z>Reads the configuration files and loads the accounts and ACLs.N)�pyzor�configZload_passwd_filer r Zload_access_filer r )r& r r r r a s zServer.load_configc O s& | j jd� tj| jd�}|j� dS )zsHandler for the SIGTERM signal. This should be used to kill the
daemon and ensure proper clean-up.
z SIGTERM received. Shutting down.)�targetN)r �info� threading�Thread�shutdown�start)r& r �kwargs�tr r r r% f s zServer.shutdown_handlerc O s& | j jd� tj| jd�}|j� dS )zgHandler for the SIGUSR1 signal. This should be used to reload
the configuration files.
z*SIGUSR1 received. Reloading configuration.)r* N)r r+ r, r- r r/ )r& r r0 r1 r r r r# n s zServer.reload_handlerc C s | j jd|dd� d S )Nz'Error while processing request from: %sT)�exc_info)r r )r& �request�client_addressr r r � handle_errorv s zServer.handle_error)N) �__name__�
__module__� __qualname__�__doc__Zmax_packet_sizeZtime_diff_allowancer r r% r# r5 r r r r r
9 s
!r
c @ s4 e Zd ZdZd
dd�Zddd�Zdd � Zd
d � Zd S )�
PreForkServerz�The same as Server, but prefork itself when starting the self, by
forking a number of child-processes.
The parent process will then wait for all his child process to complete.
� c C s" d| _ tj| ||||� || _dS )zvThe same as Server.__init__ but requires a list of databases
instead of a single database connection.
N)�pidsr
r �_prefork)r& r' r r r Zpreforkr r r r � s zPreForkServer.__init__� �?c C s� g }xTt | j�D ]F}t| j�}tj� }|sL|� | _tj| |d� tjd� q|j |� qW || _
x | j
D ]}t tj |d� qhW dS )z=Fork the current process and wait for all children to finish.)�
poll_intervalr N)
�ranger= �nextr �os�forkr
�
serve_forever�_exit�appendr< r �waitpid)r& r? r< Zdummyr �pidr r r rD � s
zPreForkServer.serve_foreverc C s: x | j p
f D ]}tj|tj� q W | j dkr6tj| � dS )zpIf this is the parent process send the TERM signal to all children,
else call the super method.
N)r< rB �killr! r$ r
r. )r& rH r r r r. � s
zPreForkServer.shutdownc C s: x | j p
f D ]}tj|tj� q W | j dkr6tj| � dS )zpIf this is the parent process send the USR1 signal to all children,
else call the super method.
N)r< rB rI r! r" r
r )r& rH r r r r � s
zPreForkServer.load_configN)r; )r> )r6 r7 r8 r9 r rD r. r r r r r r: { s
r: c @ s e Zd ZdZdS )�ThreadingServerz�A threaded version of the pyzord server. Each connection is served
in a new thread. This may not be suitable for all database types.N)r6 r7 r8 r9 r r r r rJ � s rJ c @ s* e Zd ZdZd dd�Zdd� Zdd� ZdS )
�BoundedThreadingServerz^Same as ThreadingServer but this also accepts a limited number of
concurrent threads.
Nc C s& t j| |||||d� tj|�| _d S )N)r )rJ r r, Z Semaphore� semaphore)r& r' r r r Z max_threads�forwarding_serverr r r r � s z BoundedThreadingServer.__init__c C s | j j� tj| ||� d S )N)rL �acquirerJ �process_request)r& r3 r4 r r r rO � s
z&BoundedThreadingServer.process_requestc C s t j| ||� | jj� d S )N)rJ �process_request_threadrL �release)r& r3 r4 r r r rP � s z-BoundedThreadingServer.process_request_thread)N)r6 r7 r8 r9 r rO rP r r r r rK � s
rK c @ s e Zd ZdZddd�ZdS )�
ProcessServerz�A multi-processing version of the pyzord server. Each connection is
served in a new process. This may not be suitable for all database types.
�( Nc C s |t _tj| |||||d� d S )N)r )rR � max_childrenr
r )r& r' r r r rT rM r r r r � s zProcessServer.__init__)rS N)r6 r7 r8 r9 r r r r r rR � s rR c @ sj e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dee e e
e d�Z
dS )r z Handle a single pyzord request.c O s$ t jj� | _tjj| f|�|� d S )N)�email�messageZMessage�responser �DatagramRequestHandlerr )r& r r0 r r r r � s zRequestHandler.__init__c C s� d| j d< d| j d< dtj | j d< y| j� W �nB tk
rf } z| jdd| � W Y d d }~X �n tjk
r� } z| jd
d | � W Y d d }~X n� tjk
r� } z| jd d
| � W Y d d }~X n� tjk
r� } z| jdd| � W Y d d }~X nz tj k
�r0 } z| jdd| � W Y d d }~X nF t
k
�rt } z(| jdd| � | j j j
tj� � W Y d d }~X nX | j j jd| j j� � | jj| j j� jd�� d S )z7Handle a pyzord operation, cleanly handling any errors.Z200�CodeZOK�Diagz%s�PVi� zNot implemented: %sNi� zVersion Not Supported: %si� zBad request: %si� z!Unauthorized: Signature Error: %si� z
Forbidden: %si� zInternal Server Error: %sz Sending: %r�utf8)rW r( �
proto_version�_really_handle�NotImplementedErrorr5 �UnsupportedVersionError�
ProtocolError�SignatureError�AuthorizationError� Exception�serverr r � traceback�
format_excr Z as_stringZwfile�write�encode)r& r r r r �handle� s(
" $zRequestHandler.handlec C s� | j jjd| j� tj| jj� jdd�d �}|d | j d< |d pHt
j }|t
j kr�yt
j j
|| j j| � W n tk
r� t
jd��Y nX d|kr�t
jd��y&tt|d ��tt
j�kr�t
j� �W n2 tk
r� | j jjd |d � t
jd
��Y nX |d }|| j j| k�r t
jd ��| j jjd
|| jd � y| j| }W n tk
�rb t d��Y nX |j d�}|�r�|�r�|| |� | j j j d|| jd ||| j d � dS )z(handle() without the exception handling.z Received: %rs
�
r- ZUserz
Unknown user.r[ z)Protocol Version not specified in requestzInvalid PV: %szInvalid Protocol VersionZOpz0User is not authorized to request the operation.zGot a %s command from %sr z'Requested operation is not implemented.z Op-Digestz%s,%s,%s,%r,%srY N) re r r ZpacketrU Zmessage_from_bytesZrfile�read�replacerW r( Zanonymous_userZaccountZverify_signaturer �KeyErrorrb ra �int�floatr] r` �
ValueError�warnr rc r4 �
dispatchesr_ Zget_allr r+ )r& r3 �userZopcode�dispatch�digestsr r r r^ � sF
z RequestHandler._really_handlec C s6 | j jjd||� | jjdd| � | jjd|� dS )z,Create an appropriate response for an error.z%s: %srY z%drZ N)re r r rW Zreplace_header)r& �coderV r r r r5 / s zRequestHandler.handle_errorc C s6 | j jjd|d � dtj | jd< dd | jd< dS )zkHandle the 'pong' command.
This command returns maxint for report counts and 0 whitelist.
zRequest pong for %sr z%d�CountzWL-CountN)re r r �sys�maxsizerW )r& rv r r r � handle_pong5 s zRequestHandler.handle_pongc C sn |d }y| j j| }W n tk
r8 tjjj� }Y nX | j jjd|� d|j | j
d< d|j | j
d< dS )zpHandle the 'check' command.
This command returns the spam/ham counts for the specified digest.
r zRequest to check digest %sz%drx zWL-CountN) re r rn r( �engines�common�Recordr r �r_countrW �wl_count)r& rv �digest�recordr r r � handle_check> s zRequestHandler.handle_checkc C s� | j jjd|� | j jr(| j jj|� nTxR|D ]J}y| j j| }W n tk
rb tjj j
� }Y nX |j � || j j|< q.W | j j r�x|D ]}| j j j
|� q�W dS )zwHandle the 'report' command in a single step.
This command increases the spam count for the specified digests.z Request to report digests %sN)re r r r r �reportrn r( r| r} r~ Z r_incrementr �queue_forward_request)r& rv r� r� r r r �
handle_reportL s
z RequestHandler.handle_reportc C s� | j jjd|� | j jr(| j jj|� nTxR|D ]J}y| j j| }W n tk
rb tjj j
� }Y nX |j � || j j|< q.W | j j r�x |D ]}| j j j
|d� q�W dS )zyHandle the 'whitelist' command in a single step.
This command increases the ham count for the specified digests.z Request to whitelist digests %sTN)re r r r r � whitelistrn r( r| r} r~ Z wl_incrementr r� )r& rv r� r� r r r �handle_whitelist_ s
z RequestHandler.handle_whitelistc C s� |d }y| j j| }W n tk
r8 tjjj� }Y nX | j jjd|� dd� }d||j � | j
d< d||j � | j
d< d||j � | j
d< d||j
� | j
d < d|j | j
d
< d|j | j
d < d S )
z�Handle the 'info' command.
This command returns diagnostic data about a digest (timestamps for
when the digest was first/last seen as spam/ham, and spam/ham
counts).
r z'Request for information about digest %sc S s | sdS t j| j� �S )zoConvert a datetime object to a POSIX timestamp.
If the object is None, then return 0.
r )�timeZmktimeZ timetuple)Ztime_objr r r � time_output� s z/RequestHandler.handle_info.<locals>.time_outputz%dZEnteredZUpdatedz
WL-Enteredz
WL-Updatedrx zWL-CountN)re r rn r( r| r} r~ r r Z r_enteredrW Z r_updatedZ
wl_enteredZ
wl_updatedr r� )r& rv r� r� r� r r r � handle_infor s
zRequestHandler.handle_infoN)ZpingZpongr+ Zcheckr� r� )r6 r7 r8 r9 r rj r^ r5 r{ r� r� r� r� rs r r r r r � s > r )"r9 rB ry r� r r
r! r r, rf Z
email.messagerU r � ImportErrorZ socketserverZ pyzor.configr( Z
pyzor.accountZpyzor.engines.commonZpyzor.hacks.py26ZhacksZpy26Zhack_allr r r
r: ZThreadingMixInrJ rK Z ForkingMixInrR rX r r r r r �<module> s4
B2
Mini Shell