3
_Vj 2 @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
m
Z
m Z m Z ddl
mZ yddlZdZW n ek
r dZY nX ddd d
d d gZd
ZdZdZdZdgZdZdhdidjdkdldmdndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddd)ZejdZejdZ ejdZ ejdej Z ejdZ ejd Z!ejd Z"ejd ej Z#dZ$d Z%ejd Z&G d d dZ'erG d!d" d"e'Z(ej)d" G d#d de'Z*G d$d% d%Z+d&j,d'Z-d(d) e.e-d*d D Z/d+d Z0d,d
Z1d-d Z2d.d Z3e4d/krddl5Z5ddl6Z6y e5j5ej7d*d d0\Z8Z9W n0 e5j:k
rx Z; zf f Z8Z9W Y ddZ;[;X nX dZe;Zne=d2kre;Z.*))?s .*FLAGS \((?P[^\)]*)\)s .*INTERNALDATE "(?P[ 0123][0-9])-(?P[A-Z][a-z][a-z])-(?P[0-9][0-9][0-9][0-9]) (?P[0-9][0-9]):(?P[0-9][0-9]):(?P[0-9][0-9]) (?P[-+])(?P[0-9][0-9])(?P[0-9][0-9])"s .*{(?P\d+)}$s
\r\n|\r|\ns% \[(?P[A-Z-]+)( (?P.*))?\]s$ \* (?P[A-Z-]+)( (?P.*))?s3 \* (?P\d+) (?P[A-Z-]+)( (?P.*))?s [ - ]c @ s e Zd ZG dd deZG dd deZG dd deZdefdd Zd
d Z d d
Z
dd Z dd Z dd Z
dd Zdd ZdefddZdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4d5 Z d6d7 Z d8d9 Z d:d; Z!dd? Z#d@dA Z$dBdC Z%dDdE Z&dFdG Z'ddJdKZ(dLdM Z)dNdO Z*dPdQ Z+dRdS Z,ddTdUZ-dVdW Z.dXdY Z/dZd[ Z0d\d] Z1d^d_ Z2d`da Z3dbdc Z4ddfdgZ5dhdi Z6djdk Z7dldm Z8dndo Z9ddqdrZ:dsdt Z;dudv Zd{d| Z?d}d~ Z@dd ZAdd ZBdd ZCdd ZDdd ZEdd ZFdd ZGdd ZHdd ZIdd ZJdd ZKdd ZLdd ZMdd ZNdpS )r c @ s e Zd ZdS )z IMAP4.errorN)__name__
__module__ __qualname__ r< r< /usr/lib64/python3.6/imaplib.pyerror s r> c @ s e Zd ZdS )z IMAP4.abortN)r9 r: r; r< r< r< r= abort s r? c @ s e Zd ZdS )zIMAP4.readonlyN)r9 r: r; r< r< r< r= readonly s r@ c C s t | _d| _d | _i | _i | _d| _d| _d| _d| _ | j
| j || y | j W n8 t
k
r y | j W n tk
r Y nX Y nX d S )Nr rA Fr )Debugdebugstateliteraltagged_commandsuntagged_responsescontinuation_response is_readonlytagnum_tls_established _mode_asciiopen_connect ExceptionshutdownOSError)selfhostportr< r< r= __init__ s& zIMAP4.__init__c C s0 d| _ d| _tjttj| _tjttj| _d S )NFascii) utf8_enabled _encodingrecompile_LiteralASCIILiteral_Untagged_statusUntagged_status)rR r< r< r= rL s zIMAP4._mode_asciic C s( d| _ d| _tjt| _tjt| _d S )NTzutf-8)rW rX rY rZ r[ r] r^ r_ )rR r< r< r=
_mode_utf8 s zIMAP4._mode_utf8c C s t tjdd| _tjd| j d tj| _| j | _ d| j
krHd| _ n d| j
krZd| _ n | j | j | j
x tD ]}|| jkrqt|| _d S W | j d d S )
Ni i s (?Ps" \d+) (?P[A-Z]+) (?P.*)ZPREAUTHr OKr zserver not IMAP4 compliant)r randomZrandinttagprerY rZ r\ tagre
_get_responseZwelcomerG rD r> _get_capabilitiesAllowedVersions capabilitiesPROTOCOL_VERSION)rR versionr< r< r= rN s
zIMAP4._connectc C s& |t krt| |j S td| d S )NzUnknown IMAP4 command: '%s')CommandsgetattrlowerAttributeError)rR attrr< r< r= __getattr__ s zIMAP4.__getattr__c C s | S )Nr< )rR r< r< r= __enter__ s zIMAP4.__enter__c G s& y | j W n tk
r Y nX d S )N)logoutrQ )rR argsr< r< r= __exit__ s zIMAP4.__exit__c C s | j s
d n| j }tj|| jfS )N)rS socketZcreate_connectionrT )rR rS r< r< r= _create_socket s zIMAP4._create_socketc C s( || _ || _| j | _| jjd| _d S )Nrb)rS rT rv sockmakefilefile)rR rS rT r< r< r= rM # s
z
IMAP4.openc C s | j j|S )N)rz read)rR sizer< r< r= r{ / s z
IMAP4.readc C s. | j jtd }t|tkr*| jdt |S )N zgot more than %d bytes)rz readline_MAXLINElenr> )rR liner< r< r= r~ 4 s zIMAP4.readlinec C s | j j| d S )N)rx Zsendall)rR datar< r< r= send< s z
IMAP4.sendc C st | j j zXy| jjtj W n@ tk
r^ } z$|jtjkrNt |dddkrN W Y d d }~X nX W d | jj X d S )NZwinerrorr i&' )
rz closerx rP ru Z SHUT_RDWRrQ errnoZENOTCONNrl )rR excr< r< r= rP A s
zIMAP4.shutdownc C s | j S )N)rx )rR r< r< r= ru Q s z IMAP4.socketc C sB d}| j dd g|\}}|d r(||fS | j \}}| j |||S )NZRECENTra r} )_untagged_responsenoop)rR nametypdatr< r< r= recent] s z IMAP4.recentc C s | j |d g|j S )N)r upper)rR coder< r< r= responsen s zIMAP4.responsec C sx d}|s d}|r.|d |d
fd kr2d| }nd }|r@t |}nd }tjt|}| jrbd| d }|| _| j||||S ) Nr INBOXr r} ()z(%s)s UTF8 ( )r )r r )r MapCRLFsubCRLFrW rE _simple_command)rR mailboxflags date_timemessager rE r< r< r= append| s
z IMAP4.appendc C sP |j }t|j| _| jd|\}}|dkrB| j|d jddd| _||fS )Nr ra r} zutf-8replacer r )r _AuthenticatorprocessrE r r> decoderD )rR Z mechanismZ
authobjectmechr r r< r< r= authenticate s zIMAP4.authenticatec C s d}| j |\}}| j|||S )Nr )r r )rR r r r r< r< r=
capability s zIMAP4.capabilityc C s
| j dS )Nr )r )rR r< r< r= check s z IMAP4.checkc C s$ z| j d\}}W d d| _X ||fS )Nr r )r rD )rR r r r< r< r= r s z IMAP4.closec C s | j d||S )Nr )r )rR message_setZ new_mailboxr< r< r= copy s z
IMAP4.copyc C s | j d|S )Nr )r )rR r r< r< r= create s z IMAP4.createc C s | j d|S )Nr )r )rR r r< r< r= delete s z IMAP4.deletec C s | j d||S )Nr )r )rR r whor< r< r= deleteacl s zIMAP4.deleteaclc C sH d| j krtjd| jd|\}}|dkr@d|j kr@| j ||fS )Nr z Server does not support ENABLEra z UTF8=ACCEPT)rh r r> r r r` )rR r r r r< r< r= enable s
z IMAP4.enablec C s d}| j |\}}| j|||S )Nr )r r )rR r r r r< r< r= expunge s z
IMAP4.expungec C s$ d}| j |||\}}| j|||S )Nr )r r )rR r Z
message_partsr r r r< r< r= fetch s
z IMAP4.fetchc C s | j d|\}}| j||dS )Nr ZACL)r r )rR r r r r< r< r= getacl s z IMAP4.getaclc C s" | j d|||\}}| j||dS )Nr
ANNOTATION)r r )rR r entryZ attributer r r< r< r=
getannotation# s zIMAP4.getannotationc C s | j d|\}}| j||dS )Nr! QUOTA)r r )rR rootr r r< r< r= getquota+ s zIMAP4.getquotac C s@ | j d|\}}| j||d\}}| j||d\}}|||gfS )Nr" r Z QUOTAROOT)r r )rR r r r ZquotaZ quotarootr< r< r= getquotaroot6 s zIMAP4.getquotaroot""*c C s$ d}| j |||\}}| j|||S )Nr$ )r r )rR directorypatternr r r r< r< r= listA s z
IMAP4.listc C s< | j d|| j|\}}|dkr.| j|d d| _||fS )Nr% ra r} r r )r _quoter> rD )rR userpasswordr r r< r< r= loginM s
z IMAP4.loginc C s || | _ | _| jd| jS )NzCRAM-MD5)r r r _CRAM_MD5_AUTH)rR r r r< r< r= login_cram_md5[ s zIMAP4.login_cram_md5c C sB dd l }t| jtr | jjdn| j}| jd |j||dj S )Nr zutf-8 Zmd5)hmac
isinstancer strencoder ZHMACZ hexdigest)rR Z challenger pwdr< r< r= r d s zIMAP4._CRAM_MD5_AUTHc C sj d| _ y| jd\}}W n( ddtj d d g }}Y nX | j d| jkrbd| jd fS ||fS )Nr NOz%s: %s BYE)rD r sysexc_inforP rG )rR r r r< r< r= rr l s "
z IMAP4.logoutc C s$ d}| j |||\}}| j|||S )Nr& )r r )rR r r r r r r< r< r= lsub| s z
IMAP4.lsubc C s | j d|\}}| j||dS )Nr# )r r )rR r r r r< r< r= myrights s zIMAP4.myrightsc C s d}| j |\}}| j|||S )Nr' )r r )rR r r r r< r< r= namespace s zIMAP4.namespacec C s
| j dS )Nr( )r )rR r< r< r= r s z
IMAP4.noopc C s( d}| j |||||\}}| j||dS )Nr) r )r r )rR Z message_numZ message_partstartZlengthr r r r< r< r= partial s z
IMAP4.partialc C s d}| j d|S )Nr* )r )rR r r r< r< r= proxyauth s zIMAP4.proxyauthc C s | j d||S )Nr+ )r )rR Z
oldmailboxZ
newmailboxr< r< r= rename s z IMAP4.renamec G sT d}|r2| j rtjd| j|d|f| \}}n| j|f| \}}| j|||S )Nr, z'Non-None charset not valid in UTF8 modeCHARSET)rW r r> r r )rR charsetZcriteriar r r r< r< r= search s
z IMAP4.searchr Fc C sx i | _ || _|rd}nd}| j||\}}|dkr@d| _||fS d| _d| j krd| rd| jd| || j jdd gfS ) Nr r- ra r r z READ-ONLYz%s is not writableEXISTS)rG rI r rD r@ get)rR r r@ r r r r< r< r= select s
z IMAP4.selectc C s | j d|||S )Nr. )r )rR r r Zwhatr< r< r= setacl s z IMAP4.setaclc G s | j d| \}}| j||dS )Nr/ r )r/ )r r )rR rs r r r< r< r=
setannotation s zIMAP4.setannotationc C s | j d||\}}| j||dS )Nr0 r )r r )rR r Zlimitsr r r< r< r= setquota s zIMAP4.setquotac G sF d}|d |d fdkr d| }| j |||f| \}}| j|||S ) Nr1 r r} r r z(%s)r )r r )r r )rR Z
sort_criteriar search_criteriar r r r< r< r= sort s
z
IMAP4.sortNc C s d}t s| jd| jr"| jd|| jkr6| jd|d krFtj }| j|\}}|dkr|j| j | j
d| _ | j j d| _ d| _| j
n
| jd | j|||S )
Nr2 zSSL support missingz TLS session already establishedzTLS not supported by serverra )server_hostnamerw Tz Couldn't establish TLS session)HAVE_SSLr> rK r? rh ssl_create_stdlib_contextr wrap_socketrx rS ry rz rf r )rR ssl_contextr r r r< r< r= starttls s$
zIMAP4.starttlsc C s$ d}| j |||\}}| j|||S )Nr3 )r r )rR r namesr r r r< r< r= status5 s z IMAP4.statusc C s> |d |d fd kr d| }| j d|||\}}| j||dS )
Nr r} r r z(%s)r4 r r )r r )r r )rR r commandr r r r< r< r= storeA s z IMAP4.storec C s | j d|S )Nr5 )r )rR r r< r< r= subscribeL s zIMAP4.subscribec G s* d}| j |||f| \}}| j|||S )Nr6 )r r )rR Zthreading_algorithmr r r r r r< r< r= threadT s z IMAP4.threadc G s |j }|tkr | jd| | jt| krL| jd|| jdjt| f d}| j||f| \}}|d krt|}nd}| j|||S )
Nz Unknown IMAP4 UID command: %sz9command %s illegal in state %s, only allowed in states %sz, r7 r, r1 r6 r )r, r1 r6 )r rk r> rD joinr r )rR r rs r r r r< r< r= uid^ s z IMAP4.uidc C s | j d|S )Nr8 )r )rR r r< r< r= unsubscribew s zIMAP4.unsubscribec G s, |j }|tkr | jft|< | j|f| S )N)r rk rD r )rR r rs r< r< r= xatom s
z IMAP4.xatomc C s8 |d kr d}| j }||kr*|| j| n
|g||< d S )N )rG r )rR r r Zurr< r< r= _append_untagged s zIMAP4._append_untaggedc C s, | j jd}|r(| j|d j| jdd S )Nr r} r r )rG r r? r rX )rR Zbyer< r< r=
_check_bye s zIMAP4._check_byec
G s | j t| kr4d | _| jd|| j djt| f x d D ]}|| jkr:| j|= q:W d| jkrp| j rp| jd| j }t || j
}|d | }xJ|D ]B}|d krqt |t rt || j
}t
j|rtd |d | }qW | j}|d k r(d | _t|t| jkr |}n d }|t d
t| | j
}y| j|t W n2 tk
rl } z| jd | W Y d d } ~ X nX |d kr||S xx | j r| j| r|S qW |r|| j}y| j| | jt W n2 tk
r } z| jd | W Y d d } ~ X nX |s~P q~W |S )
Nz9command %s illegal in state %s, only allowed in states %sz, ra r BADz READ-ONLYz#mailbox status changed to READ-ONLY z*Control characters not allowed in commandsz {%s}zsocket error: %s)ra r r )rD rk rE r> r rG rI r@ _new_tagbytesrX r r _control_charsr
ValueErrortype_commandr r r rQ r? re rF rH )
rR r rs r tagr argrE Z literatorvalr< r< r= r sb
zIMAP4._commandc C s |dkr| j y| j|\}}W nj | jk
rX } z| jd||f W Y d d }~X n6 | jk
r } z| jd||f W Y d d }~X nX |dkr| j |dkr| jd|||f ||fS )Nr zcommand: %s => %sr z%s command error: %s %s)r _get_tagged_responser? r> )rR r r r r r r< r< r= _command_complete s "$zIMAP4._command_completec C sJ | j \}}|d gkr | jdt|d | j}|j }t|j | _d S )Nz"no CAPABILITY response from serverr} r )r r> r rX r tuplesplitrh )rR r r r< r< r= rf s
zIMAP4._get_capabilitiesc C s | j }| j| j|rp| jjd}|| jkr:| jd| | jjd}t|| j}| jjd}||gf| j|< nd }| jt |s| j| j
|r| jjd}| jd kr| jt |r| jjd| _ d S | jd| | jjd}t|| j}| jjd}|d krd}|r|d | }xH| j| j
|r\t| jjd }| j|}| j|||f | j }qW | j|| |d
kr| jt|r| jjd}t|| j}| j|| jjd |S )Nr z unexpected tagged response: %rr r Zdata2zunexpected response: %rr r r| ra r r )ra r r ) _get_line_matchrd mogrouprF r? r rX Untagged_responser_ ContinuationrH r] intr{ r
Response_code)rR respr r r Zdat2r| r r< r< r= re sH
zIMAP4._get_responsec C sd x^| j | }|d k r | j |= |S | j y | j W q | jk
rZ } z W Y d d }~X qX qW d S )N)rF r re r? )rR r resultr r< r< r= r b s
zIMAP4._get_tagged_responsec C s> | j }|s| jd|jds.| jd| |d d }|S )Nzsocket error: EOFs
z#socket error: unterminated line: %rr )r~ r? endswith)rR r r< r< r= r } s
zIMAP4._get_linec C s |j || _| jd k S )N)matchr )rR Zcresr< r< r= r s z IMAP4._matchc C s2 | j tt| j| j }| jd | _d | j|< |S )Nr} )rc r r rJ rX rF )rR r r< r< r= r s
zIMAP4._new_tagc C s$ |j dd}|j dd}d| d S )N\z\\"z\")r )rR r r< r< r= r s z IMAP4._quotec G s | j || j|f| S )N)r r )rR r rs r< r< r= r s zIMAP4._simple_commandc C s8 |dkr||fS || j kr$|d gfS | j j|}||fS )Nr )rG pop)rR r r r r r< r< r= r s
zIMAP4._untagged_response)r r )r r )r F)N)Or9 r: r; rO r> r? r@
IMAP4_PORTrU rL r` rN rp rq rt rv rM r{ r~ r rP ru r r r r r r r r r r r r r r r r r r r r r r rr r r r r r r r r r r r r r r r r r r r r r r r r r rf re r r r r r r r r< r< r< r= r s .*
O P c @ s6 e Zd ZdedddfddZdd ZdefddZdS ) IMAP4_SSLrA Nc C s |d k r|d k rt d|d k r0|d k r0t d|d k s@|d k rVdd l}|jdtd || _|| _|d krxtj||d}|| _t j
| || d S )Nz8ssl_context and keyfile arguments are mutually exclusivez9ssl_context and certfile arguments are mutually exclusiver zEkeyfile and certfile are deprecated, use a custom ssl_context insteadr )certfilekeyfile) r warningswarnDeprecationWarningr r r r r r rU )rR rS rT r r r r r< r< r= rU s zIMAP4_SSL.__init__c C s t j| }| jj|| jdS )N)r )r rv r r rS )rR rx r< r< r= rv s
zIMAP4_SSL._create_socketc C s t j| || d S )N)r rM )rR rS rT r< r< r= rM s zIMAP4_SSL.open)r9 r: r; IMAP4_SSL_PORTrU rv rM r< r< r< r= r s r c @ s> e Zd Zdd ZdddZdd Zdd Zd
d Zd d
ZdS )r c C s || _ tj| d S )N)r r rU )rR r r< r< r= rU ( s zIMAP4_stream.__init__Nc C sN d | _ d | _d | _d | _tj| jttjtjddd| _ | j j
| _ | j j | _
d S )NT)bufsizestdinstdoutshellZ close_fds)rS rT rx rz
subprocessPopenr r PIPEr r writefiler readfile)rR rS rT r< r< r= rM - s
zIMAP4_stream.openc C s | j j|S )N)r# r{ )rR r| r< r< r= r{ = s zIMAP4_stream.readc C s
| j j S )N)r# r~ )rR r< r< r= r~ B s zIMAP4_stream.readlinec C s | j j| | j j d S )N)r" writeflush)rR r r< r< r= r G s zIMAP4_stream.sendc C s" | j j | jj | jj d S )N)r# r r" r wait)rR r< r< r= rP M s
zIMAP4_stream.shutdown)NN) r9 r: r; rU rM r{ r~ r rP r< r< r< r= r s
c @ s, e Zd Zdd Zdd Zdd Zdd Zd S )
r c C s
|| _ d S )N)r )rR Zmechinstr< r< r= rU [ s z_Authenticator.__init__c C s& | j | j|}|d kr dS | j|S )N *)r r r )rR r Zretr< r< r= r ^ s z_Authenticator.processc C sr d}t |tr|jd}xT|rlt|dkrD|d d }|dd }n|}d}tj|}|r||d d }qW |S )Nr zutf-80 r} r )r r r r binasciiZ
b2a_base64)rR inpZoupter< r< r= r d s
z_Authenticator.encodec C s |sdS t j|S )Nr )r) Z
a2b_base64)rR r* r< r< r= r | s z_Authenticator.decodeN)r9 r: r; rU r r r r< r< r< r= r U s r z0 Jan Feb Mar Apr May Jun Jul Aug Sep Oct Nov Decr c C s i | ]\}}|d |j qS )r} )r ).0nr r< r< r=
s r/ r} c C s t j| }|sd S t|jd }|jd}t|jd}t|jd}t|jd}t|jd}t|jd}t|jd} t|jd }
| d
|
d
} |d kr| } ||||||d
ddf } tj| | }
tj|
S )Nmonzonendayyearhourminseczonehzonem< -r} r r r ) InternalDater Mon2numr r calendarZtimegmtime localtime)r
r r0 r1 r2 r3 r4 r5 r6 r7 r8 ZzoneZttutcr< r< r= r s$
c C sD d}d}t t| } x*| r>t| d\} }|||d | }qW |S )Nr s ABCDEFGHIJKLMNOP r} )r absdivmod)Znumr ZAPmodr< r< r= r s c C s$ t j| }|sf S t|jdj S )Nr )Flagsr r r r )r
r r< r< r= r
s
c
C s t | ttfr"tj| tjj }nt | try
| j }W nZ t
k
r t j r| d }|d krpt j
t j| d }t jt jf| }nt j }Y nX t|d}t| d d dt|i}nNt | tr| jd krtd| }n,t | to| d | d fd
k r| S td d
jt|j }|j|S )N r} )Zseconds tzinfozdate_time must be awarer r z date_time not of a known typez"%d-{}-%Y %H:%M:%S %z"r r )r r )r r floatr Z
fromtimestampr r@ Z
astimezoner Z tm_gmtoffrn r> Zdaylightr? ZmktimeZaltzoner rH r r formatMonthsZmonthZstrftime)r ZdtZgmtoffdstZdeltaZfmtr< r< r= r s0
__main__zd:s:z-dz-srA z IMAP password for %s on %s: Z localhostzJFrom: %(user)s@localhost%(lf)sSubject: IMAP4 test%(lf)s%(lf)sdata...%(lf)s
)r Zlfr r
/tmp/xxx 1r /tmp/yyyr
/tmp/yyz 2r r /tmpyy*r r SUBJECTtestr 1(FLAGS INTERNALDATE RFC822)r FLAGS
(\Deleted)r r r r r UIDVALIDITYr r, ALLr rr c C sL t jd| |f tt | | \}}t jd| ||f |dkrH|d |S )Nz%s %sz %s => %s %sr r )M_mesgrl )cmdrs r r r< r< r= run s r_ zPROTOCOL_VERSION = %szCAPABILITIES = %r/tmp/yy%z .*"([^"]+)"$r r z%sz:(FLAGS INTERNALDATE RFC822.SIZE RFC822.HEADER RFC822.TEXT)z
All tests OK.z
Tests failed.z8
If you would like to see debugging output,
try: %s -d5
)r
r )r r )r )r r r r )r )r )r )r r )r r )r r )r )r r )r )r )r r )r r )r r )r r )r r )r r )r )r r r r )r r )r )r r )r r r r )r )r )r r )r )r r )r r )r r )r r )r )r )r r )r )r r )r )r )r r )rA rO )r rb rO rP )r rc rQ )r rd rR rS )r re rQ )r rf NrT rU )r rg rV rW )r rh rV rX rY )r ri rZ )r rj r, r[ )r rk r )r rl )r` ra r r, r[ )r rm r r )U __version__r) r rb rY ru r r r> r= r r r ior r r ImportError__all__r rB r r rg r rk rZ r rE r; r\ r] r r r r_ r[ r^ r r r r r r r rK enumerater< r r r
r r9 ZgetoptZgetpassargvZoptlistrs r> r Zstream_commandoptr rS ZgetuserZUSERZPASSWDZ test_mesgZ test_seq1Z test_seq2r_ r\ rD r] ri rh r^ Zmlr r r pathr r printr< r< r< r= sF H
e4
9,
#
)