o ecR@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__rr?/var/www/html/venv/lib/python3.10/site-packages/xlrd/compdoc.pyr sr c@s&eZdZdejfddZdddZdS) DirNodercCs||_||_td|dd\}|_|_|_|_|_td|dd\|_|_ |dkr1t d|_ n t |d|d d |_ g|_ d |_td |d d|_|rV||dSdS)Nztd ||f|d ||_td|dd \}}|d krZtd||d d}||krhtd||d d}d|>|_} 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}|dkrC|ttdfvrCn|}|tttfvr|dkr\td%||f|d ||krwd&||f}|dkrst||d nYt||dkrtd'|||rtd(|||fd||<|d7}|r||krtd)||||||d d| |}|t||||| |}|tttfvsM|r||krtd*||||||d |rtd+t ||d t|d,|g|_d}d}tt |D]~} || }!|!ttfvr q|!|kr/|s'td-|d td.|!|f|d d}t|| <d}q|!d/kr:td0|!||!rItd1|!||!fd$||!<|d7}|rf||krftd2|||||||!|d d| |!}|jt||||| q|rtd3t |j|d t|jd,|t|d |r|rtd+t ||d t|d,|t|t |jD]}"t|j|"<qtd3t |j|d t|jd,||j|jd|j|j|j d4d5d6}#g}$d7}%tdt |#d8D]}&|%d7}%|$t|%|#|&|&d8d|q|$|_ t!|$d|$dj"|r|$D]}'|'#|q|j d}(|(j$d9ks(J|(j%dks4|(j&dkr8d:|_'n|j|jd|j| |(j%|(j&d;d d6|_'g|_(| dkr^|(j&dkr^td<|d |(j&dkr| }| })|dkr|)dkr||rtd=|||fd9||<|)d8})d|| }*tt|||*|*| }+|j(|+|j|}|dkr|)dksr|rtd>||)f|d |)dkr|tksJ|rtd?|d t|j(d,||rtd@|d t|d |dSdS)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 rr2memsec_sizeshort_sec_sizedir_first_sec_sidmin_size_std_streamlendivmod mem_data_secs mem_data_lenarrayrRlistmaxEOCSIDFREESIDMSATSIDextendpop dump_listSATxrangeEVILSID _get_streamr6rr7r5r r)rr!r"rPrQ),r*rUrr,rSrevisionversionsszssszrV SAT_tot_secs_unusedSSAT_first_sec_sid SSAT_tot_secsMSATX_first_sec_sidMSATX_tot_secsr]r\ left_overrRnentfmt trunc_warnedMSATSAT_sectors_reqdexpected_MSATX_sectorsactual_MSATX_sectorssidmsgoffsetactual_SAT_sectors dump_againmsidxmsidsatxdbytesr7didposdsscs_dirnsecs start_posnewsrrrr.Tsh                     "   $          zCompDoc.__init__Nrc Csg} |} |durX| dkrQ|dur&|j| r!td|| |j| f||j| <|| |} | || | |z|| } WntyLtd|| fw| dks | tksWJnl|} | dkr|durx|j| rstd|| |j| f||j| <|| |} |} | | kr| } | | 8} | || | | z|| } Wntytd|| fw| dks^| tksJ| 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 )rRr r6 IndexErrorrar1rjoin)r*rUbasesatrV start_sidsizer$rMsectorssrtodograbrrrrjsd             zCompDoc._get_streamcCs|d}|dd}|j}||jD]9}||j|krK||j}|dkr.||S|dkr@|s8td|||S||dtdqdS)Nrrrz"Requested component is a 'storage'z'Requested stream is not a 'user stream')r7r&r$lowerrr _dir_searchr))r*path storage_DIDheadtaildlchildetrrrrNs     zCompDoc._dir_searchc Csz||d}|durdS|j|jkr)|j|jd|j|j|j|j||j ddS|j|j d|j |j |j|j|dddS)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. /NrCrBrLr (from SSCS)) rsplitr"rYrjrUrgrVr!rrPrQrW)r*qnamerrrrget_named_streamas   zCompDoc.get_named_streamc Cs||d}|durdS|j|jkrtd||j|jf|j|jkrM||jd|j|j |j |j||j d}|j rKt d|jdt|jd |j|S||jd |j|j|j |j|d dd |jfS) 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)rCrBz seenr/r@rr)rrr"r]r rY_locate_streamrUrgrVr!rr,r2rrfrRrjrPrQrW)r*rrresultrrrlocate_named_streamvs.    zCompDoc.locate_named_streamc sb|} | dkr td|d} d} d} g} d}||d|}| dkr|j| rG|jsGtd||jdt|jd |jtd || |j| f||j| <|d7}||kr^td |||f| | dkri| |7} n| dkrt| | | f|| |} | |} | } || } | dks"| tksJ||ksJ| s| |fS| | | fd fd d| Dd|fS)Nrz%_locate_stream: start_sid (%d) is -veiiiHrz_locate_stream(%s): seenr/r@rz,%s: size exceeds expected %d bytes; corrupt?rc3s |] \}}||VqdS)Nr).0rend_posrUrr sz)CompDoc._locate_stream..) r rRrSr2rrfr6rar)r*rUrrrVrexpected_stream_sizerrMrprrslices tot_found found_limitrrrrsJ  "        zCompDoc._locate_stream)NrNr) r r r__doc__r3r4r.rjrrrrrrrrr:Js  J 1 (r:cCsPtd|d|fd|d||||D] }tt|d|dqt|ddSNz%5d%sz = )endr0r/r2str)aliststridefdposequalvaluerrr x_dump_linercsdfdd }d}d}tdtD]1}|dur"|||}q||||krF||kr@||dd|||}q|dur[|dur]||kr_||dddSdSdSdS)NrcsPtd|d|fdd||D] }tt|ddqtddSrr)rrrrrrrr _dump_linerzdump_list.._dump_liner)rr)rhrZ)rrrrroldposrrrrfs    rfr)r __future__rr^r3structr timemachinerTrarbSATSIDrcri Exceptionr objectrr5r:rr4rfrrrrs*