U dcR@sdZddlmZddlZddlZddlmZddlTdZdZ d Z d Z d Z d Z Gd ddeZGdddeZddZGdddeZdddZejfddZdS)z Implements the minimal functionality required to extract a "Workbook" or "Book" stream (as one big string) from an OLE2 Compound Document file. )print_functionN)unpack)*sࡱc@s eZdZdS) CompDocErrorN)__name__ __module__ __qualname__rr0/tmp/pip-unpacked-wheel-zwq1az3t/xlrd/compdoc.pyr sr c@s&eZdZdejfddZdddZdS) DirNodercCs||_||_td|dd\}|_|_|_|_|_td|dd\|_|_ |dkrbt d|_ nt |d|d d |_ g|_ d |_td |d d|_|r||dS)Nz|_} d|>|_ |jdks|j dkrtd|j|j f|d td|dd\} |_ } |_ } } }}t |d}t || \}}|r||d7}tdt || f|d ||_||_tddg|}|_|rtd||| |j |d td||f|d td| |j |j f|d td| | f|d td||f|d | d }d!|}d}ttd"|dd}||d|}td|d#|d$|d}d}|dkr|ttdfkrn |}|tttfkr|dkrtd%||f|d ||krd&||f}|dkrt||d qt|n|dkrtd'|||r,td(|||fd||<|d7}|rb||krbtd)||||||d d| |}|t||||| |}q|r||krtd*||||||d |rtd+t ||d t|d,|g|_d}d}tt |D]} || }!|!ttfkrq|!|kr^|sLtd-|d td.|!|f|d d}t|| <d}qn|!d/krttd0|!||!rtd1|!||!fd$||!<|d7}|r||krtd2|||||||!|d d| |!}|jt||||| q|r,td3t |j|d t|jd,|t|d |r|rtd+t ||d t|d,|t|t |jD]}"t|j|"<qftd3t |j|d t|jd,||j|jd|j|j|j d4d5d6}#g}$d7}%tdt |#d8D].}&|%d7}%|$t|%|#|&|&d8d|q|$|_ t!|$d|$dj"|r6|$D]}'|'#|q$|j d}(|(j$d9ksPt%|(j&dksh|(j'dkrpd:|_(n$|j|jd|j| |(j&|(j'd;d d6|_(g|_)| dkr|(j'dkrtd<|d |(j'dkr| }| })|dkrR|)dkrR||rtd=|||fd9||<|)d8})d|| }*tt|||*|*| }+|j)|+|j|}q|rltd>||)f|d |)dkr|tkst%|rtd?|d t|j)d,||rtd@|d t|d |dS)ANrzNot an OLE2 compound documentsz)Expected "little-endian" marker, found %rz>>z [2]===>>>z MSAT: len = z8WARNING *** File is truncated, or OLE2 MSAT is corrupt!!z6INFO: Trying to access sector %d but only %d availablerzMSAT: invalid sector id: %dz)MSAT extension corruption: seen[%d] == %dz [3]===>>>z SAT: len = directoryr"seen_idrrSSCSzHWARNING *** OLE2 inconsistency: SSCS size is 0 but SSAT size is non-zerozSSAT corruption: seen[%d] == %dz&SSAT last sid %d; remaining sectors %dSSATseen)*rignore_workbook_corruptionr) SIGNATUREr rr.memsec_sizeshort_sec_sizeZdir_first_sec_sidmin_size_std_streamlendivmod mem_data_secs mem_data_lenarrayrLlistmaxEOCSIDFREESIDMSATSIDextendpop dump_listSATxrangeEVILSID _get_streamr2rr3r1rr'rAssertionErrorr r!rJrK),r(rOrr)rMrevisionversionZsszZssszrPZ SAT_tot_secsZ_unusedZSSAT_first_sec_sidZ SSAT_tot_secsZMSATX_first_sec_sidZMSATX_tot_secsrVrUZ left_overrLZnentfmtZ trunc_warnedZMSATZSAT_sectors_reqdZexpected_MSATX_sectorsZactual_MSATX_sectorsZsidmsgoffsetZactual_SAT_sectorsZ dump_againZmsidxZmsidZsatxZdbytesr3ZdidposdZsscs_dirZnsecs start_posnewsrrrr*Ts                      "   $        zCompDoc.__init__Nrc Csg} |} |dkr| dkr|dk rL|j| rBtd|| |j| f||j| <|| |} | || | |z || } Wqtk rtd|| fYqXq| tkstn|} | dkrd|dk r|j| rtd|| |j| f||j| <|| |} |} | | kr| } | | 8} | || | | z || } Wqtk r`td|| fYqXq| tksrt| dkrt|jd|||| d| S)Nr%s corruption: seen[%d] == %dz:OLE2 stream %r: sector allocation table invalid entry (%d)z=WARNING *** OLE2 stream %r: expected size %d, actual size %d ) rLr r2 IndexErrorrZrdr-rjoin)r(rObasesatrP start_sidsizer"rGZsectorssrltodoZgrabrrrrcs`             zCompDoc._get_streamcCs|d}|dd}|j}||jD]r}||j|kr$||j}|dkr\||S|dkr|sptd|||S||dtdq$dS)Nrrrz"Requested component is a 'storage'z'Requested stream is not a 'user stream')r3r$r"lowerrr _dir_searchr')r(pathZ storage_DIDheadtaildlchildetrrrryNs    zCompDoc._dir_searchc Cs~||d}|dkrdS|j|jkrR|j|jd|j|j|j|j||j ddS|j|j d|j |j |j|j|dddSdS)a Interrogate the compound document's directory; return the stream as a string if found, otherwise return ``None``. :param qname: Name of the desired stream e.g. ``'Workbook'``. Should be in Unicode or convertible thereto. /Nr=r<rFr (from SSCS)) rysplitr!rRrcrOr`rPr rrJrKrQ)r(qnamerkrrrget_named_streamas0  zCompDoc.get_named_streamc Cs||d}|dkrdS|j|jkr>td||j|jf|j|jkr||jd|j|j |j |j||j d}|j rt d|jdt|jd |j|S||jd |j|j|j |j|d dd |jfSdS) aq Interrogate the compound document's directory. If the named stream is not found, ``(None, 0, 0)`` will be returned. If the named stream is found and is contiguous within the original byte sequence (``mem``) used when the document was opened, then ``(mem, offset_to_start_of_stream, length_of_stream)`` is returned. Otherwise a new string is built from the fragments and ``(new_string, 0, length_of_stream)`` is returned. :param qname: Name of the desired stream e.g. ``'Workbook'``. Should be in Unicode or convertible thereto. rN)Nrrz7%r stream length (%d bytes) > file data size (%d bytes)r=r<z seenr+r:rr)ryrr!rVr rR_locate_streamrOr`rPr rr)r.rr_rLrcrJrKrQ)r(rrkresultrrrlocate_named_streamvsF   zCompDoc.locate_named_streamc sd|} | dkrtd|d} d} d} g} d}||d|}| dkr |j| r|jstd||jdt|jd |jtd || |j| f||j| <|d7}||krtd |||f| | dkr| |7} n*| dkr| | | f|| |} | |} | } || } q<| tkst||ks(t| s8| |fS| | | fd fd d| Dd|fS)Nrz%_locate_stream: start_sid (%d) is -veiiiHrz_locate_stream(%s): seenr+r:rnz,%s: size exceeds expected %d bytes; corrupt?roc3s|]\}}||VqdS)Nr).0rlend_posrOrr sz)CompDoc._locate_stream..) r rLrMr.rr_r2rZrdrq)r(rOrrrsrPrtZexpected_stream_sizerrGrvprlrZslicesZ tot_foundZ found_limitrrrrsJ          zCompDoc._locate_stream)NrN)r) r r r__doc__r/r0r*rcryrrrrrrrr4Js J 1 (r4cCsPtd|d|fd|d||||D]}tt|d|dq*t|ddSNz%5d%sz = )endr,r+r.str)aliststridefdposequalvaluerrr x_dump_linesrcsdfdd }d}d}tdtD]b}|dkrD|||}q*||||kr*||kr||dd|||}q*|dk r|dk r||kr||dddS)NrcsPtd|d|fdd||D]}tt|ddq*tddSrr)rrrrrrrr _dump_lineszdump_list.._dump_liner)r)r)rarS)rrrrrjoldposrrrr_s  r_)r)r __future__rrWr/structrZ timemachinerNrZr[ZSATSIDr\rb Exceptionr objectrr1r4rr0r_rrrrs(