o
    f^;                     @   sh  d Z ddlZddlZddlZddlZzddlZdZW n ey%   dZY nw ddgZG dd de	Z
dZd	Zd
ZdZee ZdZG dd dZerVG dd deZed edkrddlZeejd Zee  eejd  eejd  e  e \ZZeded D ]Z e!e \Z"Z#Z$ede   e#D ]Z%ede%  qed qe&  dS dS )z@A POP3 client class.

Based on the J. Myers POP3 draft, Jan. 96
    NTFPOP3error_protoc                   @   s   e Zd ZdS )r   N)__name__
__module____qualname__ r   r   /usr/lib/python3.10/poplib.pyr      s    n   i        
i   c                   @   s  e Zd ZdZdZeej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d Zdd Zdd Zdd Zdd Zdd Zd=d d!Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zed0Z d1d2 Z!d3d4 Z"d=d5d6Z#d7d8 Z$d9d: Z%d=d;d<Z&dS )>r   aP  This class supports both the minimal and optional command sets.
    Arguments can be strings or integers (where appropriate)
    (e.g.: retr(1) and retr('1') both work equally well.

    Minimal Command Set:
            USER name               user(name)
            PASS string             pass_(string)
            STAT                    stat()
            LIST [msg]              list(msg = None)
            RETR msg                retr(msg)
            DELE msg                dele(msg)
            NOOP                    noop()
            RSET                    rset()
            QUIT                    quit()

    Optional Commands (some servers support these):
            RPOP name               rpop(name)
            APOP name digest        apop(name, digest)
            TOP msg n               top(msg, n)
            UIDL [msg]              uidl(msg = None)
            CAPA                    capa()
            STLS                    stls()
            UTF8                    utf8()

    Raises one exception: 'error_proto'.

    Instantiate with:
            POP3(hostname, port=110)

    NB:     the POP protocol locks the mailbox from user
            authorization until QUIT, so be sure to get in, suck
            the messages, and quit, each time you access the
            mailbox.

            POP is a line-based protocol, which means large mail
            messages consume lots of python cycles reading them
            line-by-line.

            If it's available on your mail server, use IMAP4
            instead, it doesn't suffer from the two problems
            above.
    zUTF-8c                 C   sP   || _ || _d| _td| || | || _| jd| _d| _	| 
 | _d S )NFzpoplib.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(   |d ur
|s
t dt| j| jf|S )Nz0Non-blocking socket (timeout=0) is not supported)
ValueErrorsocketcreate_connectionr   r   )r   r   r   r   r   r   m   s   zPOP3._create_socketc                 C   s:   | j dkrtdt| td| | | j|t  d S )N   z*put*zpoplib.putline)r   printreprr   r   r   sendallCRLFr   liner   r   r   _putliner   s   zPOP3._putlinec                 C   s.   | j r
tdt| t|| j}| | d S )Nz*cmd*)r   r    r!   bytesencodingr&   r$   r   r   r   _putcmdz   s   zPOP3._putcmdc                 C   s   | j td }t|tkrtd| jdkrtdt| |s$tdt|}|dd  tkr8|d d |fS |d d t	krH|dd |fS |d d |fS )Nr   zline too longz*get*z-ERR EOF)
r   readline_MAXLINElenr   r   r    r!   r#   CR)r   r%   octetsr   r   r   _getline   s   zPOP3._getlinec                 C   s:   |   \}}| jdkrtdt| |dst||S )Nr   z*resp*   +)r1   r   r    r!   
startswithr   )r   respor   r   r   r      s
   
zPOP3._getrespc                 C   sr   |   }g }d}|  \}}|dkr4|dr!|d }|dd  }|| }|| |  \}}|dks|||fS )Nr      .s   ..r   )r   r1   r3   append)r   r4   listr0   r%   r5   r   r   r   _getlongresp   s   


zPOP3._getlongrespc                 C      |  | |  S N)r)   r   r$   r   r   r   	_shortcmd      
zPOP3._shortcmdc                 C   r:   r;   )r)   r9   r$   r   r   r   _longcmd   r=   zPOP3._longcmdc                 C   s   | j S r;   )r   r   r   r   r   
getwelcome   s   zPOP3.getwelcomec                 C   s
   || _ d S r;   )r   )r   levelr   r   r   set_debuglevel   s   
zPOP3.set_debuglevelc                 C      |  d| S )zVSend user name, return response

        (should indicate password required).
        zUSER %sr<   r   userr   r   r   rF         z	POP3.userc                 C   rC   )zSend password, return response

        (response includes message count, mailbox size).

        NB: mailbox is locked by server from here to 'quit()'
        zPASS %srD   )r   pswdr   r   r   pass_   s   z
POP3.pass_c                 C   sF   |  d}| }| jrtdt| t|d }t|d }||fS )z]Get mailbox status.

        Result is tuple of 2 ints (message count, mailbox size)
        STATz*stat*r      )r<   splitr   r    r!   int)r   retvalretsnumMessagessizeMessagesr   r   r   stat   s   
z	POP3.statNc                 C       |dur|  d| S | dS )a  Request listing, return result.

        Result without a message number argument is in form
        ['response', ['mesg_num octets', ...], octets].

        Result when a message number argument is given is a
        single response: the "scan listing" for that message.
        NzLIST %sLISTr<   r>   r   whichr   r   r   r8      s   	
z	POP3.listc                 C   rC   )zoRetrieve whole message number 'which'.

        Result is in form ['response', ['line', ...], octets].
        zRETR %sr>   rV   r   r   r   retr   rG   z	POP3.retrc                 C   rC   )zFDelete message number 'which'.

        Result is 'response'.
        zDELE %srD   rV   r   r   r   dele   rG   z	POP3.delec                 C   
   |  dS )zXDoes nothing.

        One supposes the response indicates the server is alive.
        NOOPrD   r?   r   r   r   noop  s   
z	POP3.noopc                 C   r[   )z(Unmark all messages marked for deletion.RSETrD   r?   r   r   r   rset  s   
z	POP3.rsetc                 C   s   |  d}|   |S )zDSignoff: commit changes on server, unlock mailbox, close connection.QUIT)r<   close)r   r4   r   r   r   quit  s   
z	POP3.quitc                 C   sB  zY| j }d| _ |dur|  W | j}d| _|durXz8z|tj W n! tyD } z|jtjkr:t	|dddkr: W Y d}~nd}~ww W |  dS W |  dS |  w dS | j}d| _|durz6z|tj W n! ty } z|jtjkrt	|dddkr W Y d}~nd}~ww W |  w W |  w |  w w )z8Close the connection without assuming anything about it.Nwinerrorr   i&'  )
r   ra   r   shutdownr   	SHUT_RDWROSErrorerrnoENOTCONNgetattr)r   r   r   excr   r   r   ra     sL   	 
	 z
POP3.closec                 C   rC   )zNot sure what this does.zRPOP %srD   rE   r   r   r   rpop7  s   z	POP3.rpops   \+OK.[^<]*(<.*>)c                 C   s\   t || j}| j| j}|stdddl}|d| }||	 }| 
d||f S )a  Authorisation

        - only possible if server has supplied a timestamp in initial greeting.

        Args:
                user     - mailbox user;
                password - mailbox password.

        NB: mailbox is locked by server from here to 'quit()'
        z!-ERR APOP not supported by serverr   Nr   z
APOP %s %s)r'   r(   	timestampmatchr   r   hashlibgroupmd5	hexdigestr<   )r   rF   passwordsecretmrn   digestr   r   r   apop>  s   z	POP3.apopc                 C   s   |  d||f S )zRetrieve message header of message number 'which'
        and first 'howmuch' lines of message body.

        Result is in form ['response', ['line', ...], octets].
        z	TOP %s %srX   )r   rW   howmuchr   r   r   topS  s   zPOP3.topc                 C   rS   )zReturn message digest (unique id) list.

        If 'which', result contains unique id for that message
        in the form 'response mesgnum uid', otherwise result is
        the list ['response', ['mesgnum uid', ...], octets]
        NzUIDL %sUIDLrU   rV   r   r   r   uidl\  s   
z	POP3.uidlc                 C   r[   )zITry to enter UTF-8 mode (see RFC 6856). Returns server response.
        UTF8rD   r?   r   r   r   utf8h  s   
z	POP3.utf8c                 C   sZ   dd }i }z|  d}|d }|D ]}||\}}|||< qW |S  ty,   tdw )a   Return server capabilities (RFC 2449) as a dictionary
        >>> c=poplib.POP3('localhost')
        >>> c.capa()
        {'IMPLEMENTATION': ['Cyrus', 'POP3', 'server', 'v2.2.12'],
         'TOP': [], 'LOGIN-DELAY': ['0'], 'AUTH-RESP-CODE': [],
         'EXPIRE': ['NEVER'], 'USER': [], 'STLS': [], 'PIPELINING': [],
         'UIDL': [], 'RESP-CODES': []}
        >>>

        Really, according to RFC 2449, the cyrus folks should avoid
        having the implementation split into multiple arguments...
        c                 S   s"   |  d }|d |dd  fS )Nasciir   r   )decoderL   )r%   lstr   r   r   	_parsecap{  s   zPOP3.capa.<locals>._parsecapCAPAr   z!-ERR CAPA not supported by server)r>   r   )r   r   capsr4   rawcapscaplinecapnmcapargsr   r   r   capan  s   

z	POP3.capac                 C   sx   t std| jrtd|  }d|vrtd|du r!t }| d}|j| j| j	d| _| j
d| _d| _|S )	z{Start a TLS session on the active connection as specified in RFC 2595.

                context - a ssl.SSLContext
        z-ERR TLS support missing$-ERR TLS session already establishedSTLSz!-ERR STLS not supported by serverNserver_hostnamer   T)HAVE_SSLr   r   r   ssl_create_stdlib_contextr<   wrap_socketr   r   r   r   )r   contextr   r4   r   r   r   stls  s    
z	POP3.stlsr;   )'r   r   r   __doc__r(   	POP3_PORTr   _GLOBAL_DEFAULT_TIMEOUTr   r   r&   r)   r1   r   r9   r<   r>   r@   rB   rF   rI   rR   r8   rY   rZ   r]   r_   rb   ra   rk   recompilerl   rv   rx   rz   r|   r   r   r   r   r   r   r   3   sD    +






	c                   @   s8   e Zd ZdZeddejdfddZdd Zd	ddZ	dS )
POP3_SSLa  POP3 client class over SSL connection

        Instantiate with: POP3_SSL(hostname, port=995, keyfile=None, certfile=None,
                                   context=None)

               hostname - the hostname of the pop3 over ssl server
               port - port number
               keyfile - PEM formatted file that contains your private key
               certfile - PEM formatted certificate chain file
               context - a ssl.SSLContext

        See the methods of the parent class POP3 for more documentation.
        Nc                 C   s   |d ur|d urt d|d ur|d urt d|d us |d ur+dd l}|dtd || _|| _|d u r<tj||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 insteadrK   )certfilekeyfile)r   warningswarnDeprecationWarningr   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   r   r   r   r     s"   zPOP3_SSL.__init__c                 C   s"   t | |}| jj|| jd}|S )Nr   )r   r   r   r   r   )r   r   r   r   r   r   r     s
   zPOP3_SSL._create_socketc                 C   s   t d)zThe method unconditionally raises an exception since the
            STLS command doesn't make any sense on an already established
            SSL/TLS session.
            r   )r   )r   r   r   r   r   r   r   r     s   zPOP3_SSL.stls)NNN)
r   r   r   r   POP3_SSL_PORTr   r   r   r   r   r   r   r   r   r     s    
r   __main__r   rK      zMessage %d:z   z-----------------------)'r   rg   r   r   r   r   r   ImportError__all__	Exceptionr   r   r   r/   LFr#   r-   r   r   r7   r   argvar    r@   rF   rI   r8   rR   numMsgs	totalSizerangeirY   headermsgr0   r%   rb   r   r   r   r   <module>   sR      p
0
