a
XC?hXn @ s g d Z ddlZddlZddlZddlZddlmZ ddlZddlZddl Z ddl
Z
ddl m Z ej
ddedZe ej G dd d Zd
Ze d Zd=d
dZdd Ze de jZdd Zdd Z dd Z G dd dZ G dd dZ G d d d e Z d d Z!G d d! d!e"Z#d"d#d$d%d&Z$e% Z&G d'd( d(Z'd)d* Z(e)e*e+eje,fZ-G d+d, d,e%Z.d-d. Z/e0e1e!e e d/e2e1e/e'e.d/iZ3de1d0d1d2Z4de1d0d3d4Z5e0d5d6d7d8d9Z6e0d6d5d:d;d<Z7dS )>)InvalidFileExceptionFMT_XML
FMT_BINARYloaddumploadsdumpsUID N)BytesIO) ParserCreate PlistFormatzFMT_XML FMT_BINARY)modulec @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )r c C s< t |tstd|dkr"td|dk r2td|| _d S )Nzdata must be an int zUIDs cannot be >= 2**64r UIDs must be positive)
isinstanceint TypeError
ValueErrordataselfr r -/opt/alt/python39/lib64/python3.9/plistlib.py__init__E s
z UID.__init__c C s | j S N)r r r r r __index__N s z
UID.__index__c C s d| j jt| jf S )Nz%s(%s)) __class____name__reprr r r r r __repr__Q s z UID.__repr__c C s | j | jffS r )r r r r r r
__reduce__T s zUID.__reduce__c C s t |tstS | j|jkS r )r r NotImplementedr )r otherr r r __eq__W s
z
UID.__eq__c C s
t | jS r )hashr r r r r __hash__\ s z UID.__hash__N) r
__module__ __qualname__r r r r! r$ r& r r r r r D s r s
zv[\x00\x01\x02\x03\x04\x05\x06\x07\x08\x0b\x0c\x0e\x0f\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f]L c C sP |d d }g }t dt| |D ]$}| ||| }|t| q d|S )N r )rangelenappendbinasciiZ
b2a_base64join)s
maxlinelengthZ
maxbinsizepiecesichunkr r r _encode_base64p s r7 c C s( t | trt| dS t| S d S )Nutf-8)r strr0 Z
a2b_base64encode)r2 r r r _decode_base64y s
r; z{(?P\d\d\d\d)(?:-(?P\d\d)(?:-(?P\d\d)(?:T(?P\d\d)(?::(?P\d\d)(?::(?P\d\d))?)?)?)?)?Zc C sL d}t | }g }|D ]&}|| }|d u r2 qB|t| qtj| S )NZyearZmonthZdayZhourZminutesecond) _dateParsermatch groupdictr/ r datetime)r2 orderZgdZlstkeyvalr r r _date_from_string s rE c C s d| j | j| j| j| j| jf S )Nz %04d-%02d-%02dT%02d:%02d:%02dZr< )dr r r _date_to_string s rG c C sZ t | }|d urtd| dd} | dd} | dd} | dd} | d d
} | S ) Nzz>)_controlCharPatsearchr replace)textmr r r _escape s
rR c @ s e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)S )* _PlistParserc C s g | _ d | _d | _|| _d S r )stack current_keyroot
_dict_typer dict_typer r r r s z_PlistParser.__init__c C sB t | _| j| j_| j| j_| j| j_| j| j_ | j
| | j S r ) r parserhandle_begin_elementZStartElementHandlerhandle_end_elementZEndElementHandler handle_dataZCharacterDataHandlerhandle_entity_declZEntityDeclHandlerZ ParseFilerV )r Zfileobjr r r parse s
z_PlistParser.parsec C s t dd S )Nz8XML entity declarations are not supported in plist files)r )r Z entity_nameZis_parameter_entityvaluebaseZ system_idZ public_idZ
notation_namer r r r^ s z _PlistParser.handle_entity_declc C s* g | _ t| d| d }|d ur&|| d S )NZbegin_)r getattr)r elementattrshandlerr r r r[ s z!_PlistParser.handle_begin_elementc C s" t | d| d }|d ur | d S )NZend_)rb )r rc re r r r r\ s z _PlistParser.handle_end_elementc C s | j | d S r )r r/ r r r r r] s z_PlistParser.handle_datac C s | j d urFt| jd ti s.td| jj || jd | j < d | _ nB| jsT|| _n4t| jd tg sxtd| jj | jd | d S )Nz unexpected element at line %d) rU r rT typer rZ CurrentLineNumberrV r/ r r` r r r
add_object s
z_PlistParser.add_objectc C s d | j}g | _|S )N )r1 r r r r r get_data s z_PlistParser.get_datac C s" | }| | | j| d S r )rW rj rT r/ )r rd rF r r r
begin_dict s
z_PlistParser.begin_dictc C s* | j r td| j | jjf | j d S )Nz%missing value for key '%s' at line %d)rU r rZ rh rT popr r r r end_dict s
z_PlistParser.end_dictc C s8 | j st| jd ti s*td| jj | | _ d S )Nrf zunexpected key at line %d)rU r rT rg r rZ rh rl r r r r end_key s
z_PlistParser.end_keyc C s g }| | | j| d S r )rj rT r/ )r rd ar r r begin_array s
z_PlistParser.begin_arrayc C s | j d S r )rT rn r r r r end_array s z_PlistParser.end_arrayc C s | d d S )NTrj r r r r end_true s z_PlistParser.end_truec C s | d d S )NFrt r r r r end_false s z_PlistParser.end_falsec C s@ | }|ds |dr.| t|d n| t| d S )NZ0xZ0X )rl
startswithrj r )r rawr r r end_integer s z_PlistParser.end_integerc C s | t| d S r )rj floatrl r r r r end_real s z_PlistParser.end_realc C s | | d S r )rj rl r r r r
end_string s z_PlistParser.end_stringc C s | t| d S r )rj r; rl r r r r end_data s z_PlistParser.end_datac C s | t| d S r )rj rE rl r r r r end_date s z_PlistParser.end_dateN)r r' r( r r_ r^ r[ r\ r] rj rl rm ro rp rr rs ru rv rz r| r} r~ r r r r r rS s( rS c @ s8 e Zd ZdddZdd Zdd Zdd
d Zd d
Zd S )_DumbXMLWriterr c C s || _ g | _|| _|| _d S r )filerT
_indent_levelindent)r r indent_levelr r r r r s z_DumbXMLWriter.__init__c C s, | j | | d| | jd7 _d S )Nz<%s> )rT r/ writelnr r rc r r r
begin_element s z _DumbXMLWriter.begin_elementc C s | j d8 _ | d| d S )Nr z)r r r r r r end_element s z_DumbXMLWriter.end_elementNc C s8 |d ur&t |}| d|||f n| d| d S )Nz <%s>%sz<%s/>)rR r )r rc r` r r r simple_element$ s z _DumbXMLWriter.simple_elementc C sH |r8t |tr|d}| j| j| j | j| | jd d S )Nr8
)r r9 r: r writer r )r liner r r r , s
z_DumbXMLWriter.writeln)r r )N)r r' r( r r r r r r r r r r s
r c @ s> e Zd ZdddZdd Zd
d Zd d
Zdd Zdd ZdS ) _PlistWriterr r TFc C s. |r| t t| ||| || _|| _d S r )r PLISTHEADERr r
_sort_keys _skipkeys)r r r r Z writeHeader sort_keysskipkeysr r r r 9 s
z_PlistWriter.__init__c C s" | d | | | d d S )Nzz)r write_valueri r r r r C s
z_PlistWriter.writec C s t |tr| d| n|du r,| d n|du r@| d nt |tr~d| kr^dk rtn n| dd | nt|nt |tr| d
t| nzt |tr| | ndt |t t
fr| | nJt |t j r| d t
| n,t |ttfr| | ntd t| d S )
NstringTtrueFZfalsel r Zintegerz%drealdatezunsupported type: %s)r r9 r r
OverflowErrorr{ r dict
write_dictbytes bytearray write_bytesrA rG tuplelist write_arrayr rg ri r r r r H s*
z_PlistWriter.write_valuec C sz | d | jd8 _tddt| jdd| j }t||dD ]}|rJ| | qJ| jd7 _| d d S )Nr r rw r) r s r )
r r maxr. r rO r7 splitr r )r r r3 r r r r r j s
z_PlistWriter.write_bytesc C s |rt| d | jr"t| }n| }|D ]8\}}t|tsP| jrHq.td| d| | | q.|
d n
| d d S )Nr keys must be stringsrC ) r r sorteditemsr r9 r r r r r )r rF r rC r` r r r r w s
z_PlistWriter.write_dictc C s< |r.| d |D ]}| | q| d n
| d d S )Narray)r r r r )r r r` r r r r s
z_PlistWriter.write_arrayN)r r r TF) r r' r( r r r r r r r r r r r 8 s
"
r c C s d}|D ]}| |r dS qtjdftjdftjdffD ]N\}}| |sNq:|D ]4}||d| }| d t| |krR dS qRq:dS )N)s 6xBBQQQ)_fpseekosSEEK_ENDreadr. r structunpack _ref_size
_read_ints_object_offsets
_undefined_objects _read_objectOSError
IndexErrorerrorr r )r fptrailer offset_size num_objects
top_objectoffset_table_offsetr r r r_ s$
z_BinaryPlistParser.parsec C sL |dkrH| j dd d@ }d|> }dt| }t|| j |d S |S )N r r r+ rL )r r _BINARY_FORMATr r )r tokenLrQ r2 fr r r _get_size s z _BinaryPlistParser._get_sizec st | j | tv r2td| t S rFt | krLt t fddtd| D S d S )NrL c 3 s& | ] }t || d V qdS )bigN)r
from_bytes).0r5 r sizer r s z0_BinaryPlistParser._read_ints..r ) r r r r r r. r r r- )r nr r r r r s z _BinaryPlistParser._read_intsc C s | || jS r )r r )r r r r r
_read_refs s z _BinaryPlistParser._read_refsc s
j | }|tur|S j| } j| jdd }|d@ |d@ }}|dkr^d }n|dkrnd}n|dkr~d}n~|dkrd }nn|d
krtj jd|> d |d kd
}nB|dkrt d jdd }n |dkrt d jdd }n|dkrDt d jdd }t
dddt
j |d }n|dkr| |} j|}t
||krt n|dkr |} j|} t
| |krt | d}n>|dkr |d } j|} t
| |krt | d}n|d kr*tt jd| d }n|d krp |} |}
g }| j |< | fd d |
D n|d kr |} |} |}
}| j |< z.t| |
D ] \} }
|
| | < qW n ty t Y n0 nt | j |< |S )!Nr r r r F Tr, rw r r+ Zsigned" z>fr* # z>d3 )Zseconds@ P r ` r utf-16be c 3 s | ]} |V qd S r )r )r xr r r r F r, z2_BinaryPlistParser._read_object.. )r r r r r r r r r r rA Z timedeltar r. r r r r extendrW zipr r )r refresultoffsettokenZtokenHr r r2 r Zobj_refsZkey_refskor r r r s
z _BinaryPlistParser._read_objectN) r r' r( r r_ r r r r r r r r r s
r c C s, | dk r dS | dk rdS | dk r$dS dS d S )N r r r* r r )countr r r _count_to_size_ s r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )_BinaryPlistWriterc C s || _ || _|| _d S r )r r r )r r r r r r r r o s z_BinaryPlistWriter.__init__c
C s g | _ i | _i | _| | t| j }dg| | _t|| _t| j | _ | j
d | j D ]}| | qZ|
|}| j
}t|}dt| | }| j
tj|g| jR d}||| j|||f} | j
tjdg| R d S )Nr bplist00rL z >5xBBBQQQ)_objlist _objtable _objidtable_flattenr. r r r r _ref_formatr r
_write_object
_getrefnumtellr pack)
r r` r objr r r Z
offset_formatZ sort_versionr r r r r t s*
z_BinaryPlistWriter.writec C s" t |tr"t||f| jv r4d S nt|| jv r4d S t| j}| j| t |trh|| jt||f< n|| jt|< t |t rg }g }|
}| j rt |}|D ]6\}}t |t
s| jrqtd|| || qt||D ]}| | qn&t |ttfr |D ]}| | q d S )Nr )r _scalarsrg r idr r. r r/ r r r r r9 r r itertoolschainr r r ) r r` Zrefnumkeysvaluesr r vr r r r r s8
z_BinaryPlistWriter._flattenc C s. t |tr | jt||f S | jt| S d S r )r r r rg r r ri r r r r s
z _BinaryPlistWriter._getrefnumc C s |dk r"| j td||B n|dk rH| j td|dB d| nh|dk rn| j td|dB d| nB|d k r| j td
|dB d | n | j td |dB d
| d S )Nr z>Br z>BBBrw r z>BBH r z>BBL z>BBQ )r r r r )r r r r r r _write_size s z _BinaryPlistWriter._write_sizec st |} j j|< |d u r2 jd n>|du rJ jd n&|du rb jd nt|trj|dk rz jtdd| W n tj y t
|d Y n0 n|d k r҈ jtd
d | n|d k r jtd
d| nt|dk r jtdd| nR|dk r6 jtdd| n0|dk r^ jd|j d ddd nt
|nt|t r jtdd| nt|t
j
r|t
ddd } jtdd | nt|ttfr d t| j| nrt|trlz |d } d t| W n2 tyZ |d } d!t|d" Y n0 j| nt|tr*|jdk rtd#n|jd k r jtd
d$| nt|jd k rֈ jtd
d%| nP|jdk r jtdd&| n,|jdk r jtdd'| nt
|nFt|ttfr fd(d)|D }t|} d*| jtjd+ j| g|R nt|t rhg g }} j rt | } n| } | D ]J\}
} t|
ts j rސqt!d,|" |
|" | qt|} d-| jtjd+ j| g|R jtjd+ j| g|R nt!|d S ).N F Tr r z>Bqr r z>BBrw r z>BHr r z>BLr l z>BQr r r z>Bdr r r r r r r r r r r r c s g | ]} |qS r )r )r r r r r
$ r, z4_BinaryPlistWriter._write_object..r rL r r )#r r r r r r r r r r r to_bytesr{ rA Z
total_secondsr r r r. r9 r: UnicodeEncodeErrorr r r r r r r r r r r r r/ ) r r` r r tZrefsr2 ZkeyRefsZvalRefsZ rootItemsr r r r r r s
&
$&z _BinaryPlistWriter._write_objectN) r r' r( r r r r r r r r r r r n s -*r c C s | d d dkS )Nr r r )r r r r _is_fmt_binaryB s r )detectrZ writerfmtrY c C sj |d u rJ| d}| d t D ] }|d |r$|d } qVq$t n t| d }||d}|| S )Nr r r! rZ )rY )r r _FORMATSr r r_ )r r$ rY r infoPpr r r r X s
r c C s t | }t|||dS )Nr# )r
r )r` r$ rY r r r r r n s r TF)r$ r r c C s: |t vrtd|f t | d |||d}|| d S )NzUnsupported format: %rr" )r r )r% r r )r` r r$ r r r" r r r r v s r r$ r r c C s t }t| ||||d | S )Nr) )r
r getvalue)r` r$ r r r r r r r s r )r) )8__all__r0 r rA enumior
r
r rer Zxml.parsers.expatr Enumr r globalsupdate __members__r r compilerM r7 r; ASCIIr> rE rG rR rS r r r r r r objectr r r r9 r r{ r r r r r r r r% r r r r r r r r 0 sh !
n&^! !
U