gbc @sdZddlZddlZddlZddlZddlZddlZddlZyddlZej Z Wn e k re Zej Z nXddddddd d d g Z de fd YZd e fd YZeZdd>dZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!dZ"dZ#d Z$d!Z%dZ&d"Z'd#Z(d$Z)d%Z*d&Z+d'Z,d(Z-d)Z.d*Z/d+Z0dZ1dZ2dZ3dZ4dZ5dZ6d Z7d!Z8dZ9d"Z:d#Z;d$Z<d,Z=d-Z>d.Z?de@fd/YZAd0fd1YZBd2fd3YZCd fd4YZDd eDfd5YZEe d6ZFeGd7kreFndS(8s Read and write ZIP files. iNt BadZipfileterrort ZIP_STOREDt ZIP_DEFLATEDt is_zipfiletZipInfotZipFilet PyZipFilet LargeZipFilecBseZRS((t__name__t __module__(((s/lib/python2.7/site-packages/obitools/zipfile.pyRscBseZdZRS(su Raised when writing a zipfile, the zipfile requires ZIP64 extensions and those extensions are disabled. (R R t__doc__(((s/lib/python2.7/site-packages/obitools/zipfile.pyRsiiiis<4s4H2LHsPKs<4s4B4HLLL5HLLsPKs <4s2B4HLLL2HsPKs<4sLQLsPKs <4sQHHLLQQQQsPKiiiiiii i i i i iiiiicCsKy3t|d}t|}|j|r2tSWntk rFnXtS(s?Quickly see if file is a ZIP file by checking the magic number.trb(topent _EndRecDatatclosetTruetIOErrortFalse(tfilenametfpintendrec((s/lib/python2.7/site-packages/obitools/zipfile.pyRTs   c Cs=tjt}|j||d|j|}tjt|\}}}}|tkr`|S|dksx|dkrtdntjt} |j||| d|j| }tjt|\ }} } } } }}}}}|t kr|S| |d<||d<||d<||d<||d<||d<|S( sM Read the ZIP64 end-of-archive records and use that to update endrec iiis3zipfiles that span multiple disks are not supportediiii( tstructtcalcsizetstructEndArchive64LocatortseektreadtunpacktstringEndArchive64LocatorRtstructEndArchive64tstringEndArchive64(RtoffsetRt locatorSizetdatatsigtdisknotrelofftdiskstendArchiveSizetsztcreate_versiont read_versiontdisk_numtdisk_dirtdircountt dircount2tdirsizet diroffset((s/lib/python2.7/site-packages/obitools/zipfile.pyt _EndRecData64`s* 0       cCs|jdd|jd}|j}|dd!tkr|ddkrtjt|}t|}|jd|j|d|d d krt |d|S|St |d}|j||d|j}|j t}|dkrtjt|||d!}t|}||d}|d t |kr|j||j||||d d krt || ||S|Snd S(sReturn data from the "End of Central Directory" record, or None. The data is a list of the nine items in the ZIP "End of central dir" record followed by a tenth item, the file seek offset of this record.iiiiiistiIiiNi( RttellRtstringEndArchiveRRtstructEndArchivetlisttappendR0tmintrfindtlen(RtfilesizeR!Rt END_BLOCKtstarttcomment((s/lib/python2.7/site-packages/obitools/zipfile.pyRs4 #      cBs5eZdZdZdddZdZdZRS(s>Class with attributes describing each file in the ZIP archive.t orig_filenameRt date_timet compress_typeR=textrat create_systemR(textract_versiontreservedt flag_bitstvolumet internal_attrt external_attrt header_offsettCRCt compress_sizet file_sizet _raw_timetNoNameiiicCs||_|jtd}|dkr:|d|!}ntjdkrptj|krp|jtjd}n||_||_t|_ d|_ d|_ t j dkrd|_n d|_d|_d|_d|_d|_d|_d|_d|_dS(Nit/R1twin32ii(R>tfindtchrtostseptreplaceRR?RR@R=RAtsystplatformRBR(RCRDRERFRGRH(tselfRR?t null_byte((s/lib/python2.7/site-packages/obitools/zipfile.pyt__init__s*               c Csw|j}|ddd>|dd>B|dB}|dd>|d d>B|ddB}|jd @rsd}}}n|j}|j}|j}|j}|tks|tkrd }|tj|dtj |d ||}d }d }t d |j |_ t d |j |_ ntjt t|j |j|j|j|||||t|jt| } | |j|S(s'Return the per-file header as a string.iii iiiii iisRR?R@R=RARBR(RCRDRERFRGRHRIRJRKRLRM(iiiiii(R R R t __slots__RZReRl(((s/lib/python2.7/site-packages/obitools/zipfile.pyRs.& !t _ZipDecryptercBsDeZdZdZeZdZdZdZdZRS(s Class to handle decryption of files stored within a ZIP archive. ZIP supports a password-based form of encryption. Even though known plaintext attacks have been found against it, it is still useful to be able to get data out of such a file. Usage :: zd = _ZipDecrypter(mypwd) plain_char = zd(cypher_char) plain_text = map(zd, cypher_text) cCsd}dgd}xhtdD]Z}|}xAtdD]3}|d@r^|d?d@|A}q9|d?d@}q9W|||Open the ZIP file with mode read "r", write "w" or append "a".Rtwtas(ZipFile() requires mode "r", "w", or "a"s.Compression requires the (missing) zlib modules(That compression method is not supporteditbR1R twbsr+biRisMode must be "r", "w" or "a"N(RRR(RfRRRt _allowZip64Rt _didModifytdebugt NameToInfotfilelistt compressionRURRR~t isinstancet basestringt _filePassedRR tfpRtgetattrt _GetContentst_RealGetContentsRt start_dirRR(RXtfileRRt allowZip64tkeytmodeDict((s/lib/python2.7/site-packages/obitools/zipfile.pyRZ>sZ                             cCsKy|jWn6tk rF|js@|jjd|_nnXdS(sORead the directory, making sure we close the file if the format is bad.N(RRRRRR(RX((s/lib/python2.7/site-packages/obitools/zipfile.pyRxs    c Cs|j}t|}|s'tdn|jdkr>|GHn|d}|d}|d|_|dtkr|d|dd}n|d|}||}|jd krd G|G|G|GHn|||_|j|jd |j|}t j |}d }x||kr|jd } |d }| d d !t krNtdnt j t| } |jd krw| GHn|j| t} t| }|j| t|_|j| t|_|| t| t| t}| t|_| dd!\ |_|_|_|_|_|_} } |_|_|_| dd!\|_|_ |_!| |_"| d?d| d?d@| d@| d?| d?d@| d@d f|_#|j$|j||_|j%j&|||j'|j(<|jd kr dG|GHq q WdS(s/Read in the table of contents for the ZIP file.sFile is not a zip fileiiiii i8iisgiven, inferred, offsetii.is&Bad magic number for central directoryi iiiii i?ttotalN()RRRRR=R[RRRt cStringIOtStringIOtstringCentralDirRRtstructCentralDirt_CD_FILENAME_LENGTHRt_CD_EXTRA_FIELD_LENGTHRAt_CD_COMMENT_LENGTHt_CD_LOCAL_HEADER_OFFSETRIR(RBRCRDRER@RJRKRLRFRGRHRMR?RlRR6RR( RXRRtsize_cdt offset_cdtxtconcatR!RtcentdirRtttd((s/lib/python2.7/site-packages/obitools/zipfile.pyRsZ            I & cCs.g}x!|jD]}|j|jqW|S(s+Return a list of file names in the archive.(RR6R(RXtlR!((s/lib/python2.7/site-packages/obitools/zipfile.pytnamelistscCs|jS(sJReturn a list of class ZipInfo instances for files in the archive.(R(RX((s/lib/python2.7/site-packages/obitools/zipfile.pytinfolistscCsJdd GHx:|jD]/}d|jd }d|j||jfGHqWdS( s+Print a table of contents for the zip file.s%-46s %19s %12ss File Names Modified tSizes%d-%02d-%02d %02d:%02d:%02dis %-46s %s %12dN(s File Names Modified R(RR?RRL(RXtzinfotdate((s/lib/python2.7/site-packages/obitools/zipfile.pytprintdirs cCsDx=|jD]2}y|j|jWq tk r;|jSXq WdS(s%Read all the files and check the CRC.N(RRRR(RXR((s/lib/python2.7/site-packages/obitools/zipfile.pyttestzips  cCs5|jj|}|dkr1td|n|S(s,Return the instance of ZipInfo given 'name'.s(There is no item named %r in the archiveN(RtgetRtKeyError(RXRtinfo((s/lib/python2.7/site-packages/obitools/zipfile.pytgetinfos  cCs ||_dS(s)Set default password for encrypted files.N(R~(RXR~((s/lib/python2.7/site-packages/obitools/zipfile.pyt setpasswordscCs|j|d|jS(s)Return file bytes (as a string) for name.R(R R(RXRR~((s/lib/python2.7/site-packages/obitools/zipfile.pyRscCs:|dkrtdn|js-tdn|jrB|j}nt|jd}|j|}|j}|j|jd|j d}|dd !t krt d nt j t|}|j |t}|tr|j |tn||jkrt d |j|fn|jd @} d} | r|sF|j}n|s\td |nt|} |j d} t| | dd!} |jd@r|jd?d@} n|jd?d@} t| d| krtd|qn| dkrt||}nt||| }d|kr6|jtn|S(s#Return file-like object for 'name'.RtUtrUs&open() requires mode "r", "U", or "rU"s3Attempt to read ZIP archive that was already closedR iiis Bad magic number for file headers3File name in directory "%s" and header "%s" differ.is6File %s is encrypted, password required for extractioni iiii sBad password for file(RRRN(RfRRR RRR2RRIRR_RRRR^t_FH_FILENAME_LENGTHt_FH_EXTRA_FIELD_LENGTHR>RERR~RnRRMRJRwRRR(RXRRR~tzef_fileRtfilepostfheadertfnamet is_encryptedtzdRtht check_bytetzef((s/lib/python2.7/site-packages/obitools/zipfile.pyR sX                cCsOt|ts!|j|}n|dkr<tj}n|j|||S(s#Extract a member from the archive to the current working directory, using its full name. Its file information is extracted as accurately as possible. `member' may be a filename or a ZipInfo object. You can specify a different directory using `path'. N(RRRRRStgetcwdt_extract_member(RXtmembertpathR~((s/lib/python2.7/site-packages/obitools/zipfile.pytextract7s  cCsC|dkr|j}nx!|D]}|j|||q"WdS(sExtract all members from the archive to the current working directory. `path' specifies a different directory to extract to. `members' is optional and must be a subset of the list returned by namelist(). N(RRR(RXRtmembersR~R((s/lib/python2.7/site-packages/obitools/zipfile.pyt extractallEs  cCs|ddkr|d }ntjj|jrQtjj||jd}ntjj||j}tjj|}tjj|}|rtjj| rtj|n|j |jd|}t |d}t j |||j |j |S(sbExtract the ZipInfo object 'member' to a physical file on the path targetpath. iROiR~R(RSRtisabsRRtnormpathtdirnametexiststmakedirsR Rtshutilt copyfileobjR(RXRt targetpathR~t upperdirstsourcettarget((s/lib/python2.7/site-packages/obitools/zipfile.pyRQs   cCs|j|jkr-|jr-dG|jGHq-n|jd krHtdn|js]tdn|jtkrt rtdn|jt tfkrtdn|j t kr|j st dqn|jt kr|j st d qnd S( s6Check for errors before writing a file to the archive.sDuplicate name:RRs write() requires mode "w" or "a"s4Attempt to write ZIP archive that was already closeds.Compression requires the (missing) zlib modules(That compression method is not supporteds'Filesize would require ZIP64 extensionss+Zipfile size would require ZIP64 extensionsN(RR(RRRRRfRR@RRRRLR[RRRI(RXR((s/lib/python2.7/site-packages/obitools/zipfile.pyt _writecheckos(        cCs|jstdntj|}tj|j}|dd!}|dkr[|}ntjj tjj |d}x*|dtj tj fkr|d}qWt ||}|dd@d>|_|dkr|j|_n ||_|j|_d|_|jj|_|j|t|_t|d}d|_} d|_} d|_} |jj|j|jtkrtj tj!tj"d} nd} xx|j#d} | sPn| t$| } t%| | d @} | r| j&| } | t$| } n|jj| qW|j'| rq| j(} | t$| } |jj| | |_n | |_| |_| |_|jj}|jj)|jd d|jjt*j+d |j|j|j|jj)|d|j,j-|||j.|j/|dd>B|d B}|d d >|d d>B|dd B}g}|jtks|j tkr|j |j|j |j d }d } n|j}|j } |j tkr<|j |j d} n |j } |j } |rt jddt|ddt||| } td|j} td|j} n|j} |j} yyt jtt| |j| |j|j|j|||j| |t|jt| t|jd|j|j| }Wntk rtj tt| |j| |j|j|j|||j| |t|jt| t|jd|j|j| fIJnX|jj!||jj!|j|jj!| |jj!|jqJW|jj}|tkrt jt"t#ddddd||||| }|jj!|t jt$t%d|d}|jj!|t jt&t'dd||||dd }|jj!|n;t jt&t'dd|||||d }|jj!||jj(n|j)s:|jj*nd|_dS(sJClose the file, and for mode "w" and "a" write the ending records.NRRiiii iiii iIlss<     " '<*c I