U
e5d�: � @ sh d dl Z d dlZd dlZd dlZzd dlZdZW n ek
rH dZY nX ddgZG dd� de�Z dZ
dZ d Z d
Z
e e
Zd ZG d d� d�Zer�G d
d� de�Ze�d� edk�rdd dlZeejd �Zee�� � e�ejd � e�ejd � e�� e�� \Z Z e de d �D ]BZ e� e �\Z!Z"Z#ede � e"D ]Z$ede$ � �qd:d;�Z%d S )?r zUTF-8c C sP || _ || _d| _t�d| ||� | �|�| _| j�d�| _d| _ | �
� | _ d S )NFzpoplib.connect�rbr ) �host�port�_tls_established�sys�audit�_create_socket�sock�makefile�file�
_debugging�_getresp�welcome)�selfr
r �timeoutr r r �__init__b s z
POP3.__init__c C s t �| j| jf|�S �N)�socketZcreate_connectionr
r )r r r r r r m s zPOP3._create_socketc C s: | j dkrtdt|�� t�d| |� | j�|t � d S )N� z*put*zpoplib.putline)r �print�reprr r r Zsendall�CRLF�r �liner r r �_putlinep s
z
POP3._putlinec C s. | j rtdt|�� t|| j�}| �|� d S )Nz*cmd*)r r r �bytes�encodingr$ r" r r r �_putcmdx s z POP3._putcmdc C s� | j �td �}t|�tkr$td��| jdkr )r Zpswdr r r �pass_� s z
POP3.pass_c C sF | � d�}|�� }| jr&tdt|�� t|d �}t|d �}||fS )NZSTATz*stat*r � )r8 �splitr r r �int)r ZretvalZretsZ numMessagesZ sizeMessagesr r r �stat� s
z POP3.statNc C s |d k r| � d| �S | �d�S )NzLIST %sZLIST�r8 r9 �r �whichr r r r6 � s z POP3.listc C s | � d| �S )NzRETR %s�r9 rG r r r �retr� s z POP3.retrc C s | � d| �S )NzDELE %sr> rG r r r �dele� s z POP3.delec C s
| � d�S )NZNOOPr> r: r r r �noop s z POP3.noopc C s
| � d�S )NZRSETr> r: r r r �rset s z POP3.rsetc C s | � d�}| �� |S )NZQUIT)r8 �close)r r2 r r r �quit s
z POP3.quitc C s� z | j }d | _ |d k r |�� W 5 | j }d | _ |d k r�zVz|�tj� W n@ tk
r� } z"|jtjkrxt|dd�dkrx� W 5 d }~X Y nX W 5 |�� X X d S )NZwinerrorr i&' )
r rN Zshutdownr Z SHUT_RDWR�OSError�errnoZENOTCONN�getattrr )r r �excr r r r rN s �z
POP3.closec C s | � d| �S )NzRPOP %sr> r? r r r �rpop5 s z POP3.rpops \+OK.[^<]*(<.*>)c C s\ t || j�}| j�| j�}|s&td��dd l}|�d�| }|�|�� � }| �
d||f �S )Nz!-ERR APOP not supported by serverr r z
APOP %s %s) r% r& � timestamp�matchr r �hashlib�groupZmd5Z hexdigestr8 )r r@ ZpasswordZsecret�mrW Zdigestr r r �apop< s z POP3.apopc C s | � d||f �S )Nz TOP %s %srI )r rH Zhowmuchr r r �topQ s zPOP3.topc C s |d k r| � d| �S | �d�S )NzUIDL %sZUIDLrF rG r r r �uidlZ s z POP3.uidlc C s
| � d�S )NZUTF8r> r: r r r �utf8f s z POP3.utf8c
C sp dd� }i }z4| � d�}|d }|D ]}||�\}}|||< q$W n* tk
rj } z td��W 5 d }~X Y nX |S )Nc S s" | � d��� }|d |dd � fS )N�asciir r )�decoderC )r# Zlstr r r � _parsecapy s z POP3.capa.._parsecapZCAPAr z!-ERR CAPA not supported by server)r9 r ) r r` �capsr2 ZrawcapsZcaplineZcapnmZcapargsZ_errr r r �capal s
z POP3.capac C sx t s td��| jrtd��| �� }d|kr2td��|d krBt�� }| �d�}|j| j| j d�| _| j�
d�| _ d| _|S )Nz-ERR TLS support missing�$-ERR TLS session already establishedZSTLSz!-ERR STLS not supported by server�Zserver_hostnamer T) �HAVE_SSLr r rb �ssl�_create_stdlib_contextr8 � wrap_socketr r
r r )r �contextra r2 r r r �stls� s
�z POP3.stls)N)N)N)&r r r r& � POP3_PORTr �_GLOBAL_DEFAULT_TIMEOUTr r r$ r' r/ r r7 r8 r9 r; r= r@ rA rE r6 rJ rK rL rM rO rN rT �re�compilerU rZ r[ r\ r] rb rj r r r r r 3 s@ -�
c @ s4 e Zd Zeddejdfdd�Zdd� Zddd�ZdS ) �POP3_SSLNc C s� |d k r|d k rt d��|d k r0|d k r0t d��|d k s@|d k rVdd l}|�dtd� || _|| _|d krxtj||d�}|| _t �
| |||� d S )Nz4context and keyfile arguments are mutually exclusivez5context and certfile arguments are mutually exclusiver zAkeyfile and certfile are deprecated, use a custom context insteadrB )�certfile�keyfile) �
ValueError�warnings�warn�DeprecationWarningrq rp rf rg ri r r )r r
r rq rp r ri rs r r r r � s$ ��zPOP3_SSL.__init__c C s" t �| |�}| jj|| jd�}|S )Nrd )r r ri rh r
)r r r r r r r � s
�zPOP3_SSL._create_socketc C s t d��d S )Nrc )r )r rq rp ri r r r rj � s z
POP3_SSL.stls)NNN) r r r �
POP3_SSL_PORTr rl r r rj r r r r ro � s �
ro �__main__r rB � z Message %d:z z-----------------------)&rQ rm r r rf re � ImportError�__all__� Exceptionr rk rv r- ZLFr! r+ r ro r5 r �argv�ar r; r@ rA r6 rE ZnumMsgsZ totalSize�range�irJ �header�msgr. r# rO r r r r � sJ
n0