B \Q@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+lib/python3.7/site-packages/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|jdkrtd|j|jf|d td|dd\} |_ } |_ } } } }t |d}t ||\}}|rv|d7}tdt ||f|d ||_ ||_tddg|}|_|r td||||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| }x|tttfkr|dkrtd%||f|d ||krd&||f}|dkrt||d Pt|n|dkr td'|||r(td(|||fd||<|d7}|r^||kr^td)||||||d d||}|t||||||}qW|r||krtd*||||||d |rtd+t ||d t|d,|g|_d}d}xtt |D]}||} | ttfkrq| |kr`|sNtd-|d td.| |f|d d}t||<d}qn| d/krvtd0| || rtd1| || fd$|| <|d7}|r||krtd2||||||| |d d|| }|jt|||||qW|r0td3t |j|d t|jd,|t|d |r|rtd+t ||d t|d,|x$t|t |jD]}!t|j|!<qlWtd3t |j|d t|jd,||j|jd|j|j|j d4d5d6}"g}#d7}$xBtdt |"d8D].}%|$d7}$|#t|$|"|%|%d8d|qW|#|_t |#d|#dj!|rFx|#D]}&|&"|q2W|jd}'|'j#d9ks`t$|'j%dksx|'j&dkrd:|_'n$|j|jd|j||'j%|'j&d;d d6|_'g|_(| dkr|'j&dkrtd<|d |'j&dkr| }| }(x|dkrd|(dkrd||rtd=|||fd9||<|(d8}(d||})tt|||)|)|}*|j(|*|j|}qW|rtd>||(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 =Z directory)r"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))rr( SIGNATUREr rr,memsec_sizeshort_sec_sizeZdir_first_sec_sidmin_size_std_streamlendivmod mem_data_secs mem_data_lenarrayrGlistmaxEOCSIDFREESIDMSATSIDextendpop dump_listSATxrangeEVILSID _get_streamr0rr1r/rr&rAssertionErrorr r!rErF)+r'rIrr(ZrevisionversionZsszZssszrJZ SAT_tot_secsZ_unusedZSSAT_first_sec_sidZ SSAT_tot_secsZMSATX_first_sec_sidZMSATX_tot_secsrPrOZ left_overrGZnentZfmtZ trunc_warnedZMSATZSAT_sectors_reqdZexpected_MSATX_sectorsZactual_MSATX_sectorsZsidmsgoffsetZactual_SAT_sectorsZ dump_againZmsidxZmsidZsatxZdbytesr1ZdidposdZsscs_dirZnsecs start_posZnewsrrrr)Ts8   &                  $    &         zCompDoc.__init__Nrc Csg} |} |dkrx| dkr|dk rN|j| rDtd|| |j| f||j| <|| |} | || | |y || } Wqtk rtd|| fYqXqW| tkstn|} x| dkrj|dk r|j| rtd|| |j| f||j| <|| |} |} | | kr| } | | 8} | || | | y || } Wqtk rftd|| fYqXqW| tkszt| dkrt|jd|||| d| S)Nrz%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 ) rGr r0 IndexErrorrTr^r+rjoin)r'rIbasesatrJ start_sidsizer"rBZsectorssrdZtodoZgrabrrrr]sR            zCompDoc._get_streamcCs|d}|dd}|j}xx||jD]j}||j|kr&||j}|dkrZ||S|dkrz|sntd|||S||dtdq&WdS)Nrrrz"Requested component is a 'storage'z'Requested stream is not a 'user stream')r1r#r"lowerrr _dir_searchr&)r'pathZ storage_DIDheadtailZdlZchildZetrrrrnMs    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. /Nir:)r"rBrz (from SSCS)) rnsplitr!rLr]rIrZrJr rrErFrK)r'qnamercrrrget_named_stream`s  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. rrN)Nrrz7%r stream length (%d bytes) > file data size (%d bytes)ir:z seen)r*r8rz (from SSCS))rnrsr!rPr rL_locate_streamrIrZrJr rr(r,rrYrGr]rErFrK)r'rtrcresultrrrlocate_named_streamus&  zCompDoc.locate_named_streamc sb|} | dkrtd|d} d} d} g} d}||d|}x| dkr|j| rtd||jdt|jd |jtd || |j| f||j| <|d7}||krtd |||f| | dkr| |7} n*| dkr| | | f|| |} | |} | } || } q>W| tkst||ks&t| s6| |fS| | | fd fd d| Dd|fS)Nrz%_locate_stream: start_sid (%d) is -veiiiHrz_locate_stream(%s): seen)r*r8z%s corruption: seen[%d] == %dz,%s: size exceeds expected %d bytes; corrupt?rec3s|]\}}||VqdS)Nr).0rdend_pos)rIrr sz)CompDoc._locate_stream..) r rGr,rrYr0rTr^rg)r'rIrhrirJrjZexpected_stream_sizertrBrlprdrzZslicesZ tot_foundZ found_limitr)rIrrvsD         zCompDoc._locate_stream)NrN)r) r r r__doc__r-r.r)r]rnrurxrvrrrrr2JsI 1 (r2cCsTtd|d|fd|dx*||||D]}tt|d|dq,Wt|ddS)Nz%5d%sz = )endr*)r*)r,str)aliststridefdposequalvaluerrr x_dump_linesrcsdfdd }d}d}xvtdtD]b}|dkrF|||}q,||||kr,||kr||dd|||}q,W|dk r|dk r||kr||dddS)NrcsTtd|d|fddx*||D]}tt|ddq,WtddS)Nz%5d%sz =r~)rr*)r*)r,r)rrr)rrrrr _dump_lineszdump_list.._dump_liner)r)r)r[rM)rrrrrbZoldposr)rrrrrYs  rY)r)r}Z __future__rrQr-ZstructrZ timemachinerHrTrUZSATSIDrVr\ Exceptionr objectrr/r2rr.rYrrrr s(