o
j@ @ sh d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
G dd dZ G dd d Z G d
d d Z
G d d
d
ZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG dd dZG d d d ZG d d d ZG d d! d!ZG d"d# d#ZG d$d% d%ZG d&d' d'ZG d(d) d)Z G d*d+ d+Z G d,d- d-Z G d.d/ d/Z G d0d1 d1Z G d2d3 d3Z!G d4d5 d5Z"G d6d7 d7Z#G d8d9 d9Z$dXd;d<Z%d=d> Z&dYd@dAZ'dBdC Z(i dDe$dEedFedGe!dHe"dIe#dJedKedLedMedNe dOe dPedQedRedSedTe e e
eee dUZ)G dVdW dWej*Z+dS )Z N) parseString)GUID)SID)FILETIMEc @ s4 e Zd Zdd Zdd Zedd Zedd Zd S )
Fragmentc C s g | _ g | _d S N)FragmentHeadersvaluesself r /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/aiosmb/dcerpc/v5/common/even6/binxml.py__init__
zFragment.__init__c C s d}| j D ]}|| 7 }q|S )N )r to_xml)r tvr r r
r s
zFragment.to_xmlc C s t t| S r )r from_bufferioBytesIO)datar r r
from_bytes s zFragment.from_bytesc C sN t } t| ttf}t|tr |S |j| t| ttf}|j | qr )
r
parse_nextFragmentHeaderEOF
isinstancer appendElementTemplateInstancer )bufftir r r r
r s
zFragment.from_bufferN)__name__
__module__ __qualname__r r staticmethodr r r r r r
r
s
r c @ e Zd Zdd Zedd ZdS )r c C s d | _ d | _d | _d | _d S r )FragmentHeaderTokenmajorminorflagsr
r r r
r ) s
zFragmentHeader.__init__c C sp t }| d|_tj| dddd|_tj| dddd|_tj| dddd|_|jdkr6td|S )N littleF byteordersignedzFragmentHeader err) r readr' intr r( r) r* Exception)r resr r r
r / s
zFragmentHeader.from_bufferNr" r# r$ r r% r r r r r
r ( s r c @ s( e Zd Zdd Zdd Zedd ZdS )r c C d | _ d | _d | _d S r )TemplateInstanceToken TemplateDefTemplateInstanceDatar
r r r
r <
zTemplateInstance.__init__c C s g }g }d}|t | jjk r| jj| }t|tr[||jj t| jj|d tr:|| | j
|d7 }qt| jj|d t rZ| d ||j | j
dd |d7 }qnt|t
r|jdkrt| j
j|j t}| j
jj|j jtjks}|r| d | d d} t| jj|| tr|d7 }q ||7 }|r|| | j
q|| | j
nAt|tr|| n4t|tr|d| d n$t|trt |dkr|d| d t |dksntd | |d7 }|t | jjk sd
|S ) Nr r+ T)is_empty zzElement not processed! %sz
)lenr7 Elementsr StartElementr NamevalueCloseStartElementr r8 CloseEmptyElementpop SubstitutiontokenValuesSubstitutionIdlist ValueSpecValueSpecEntries ValueTypeNullType
EndElement ValueTextr r2 join)r xml_elements open_tagsiellist_replacementnr r r
r A s^
2zTemplateInstance.to_xmlc C s6 t }| d|_t| |_t| |_| |S Nr+ )r r0 r6 r7 r r8 r )r r! r r r
r { s z TemplateInstance.from_bufferNr" r# r$ r r r% r r r r r
r ; s
:r c @ r& )r7 c C s" d | _ d | _d | _d | _g | _d S r )dunno
TemplateIdTemplateDefByteLengthr r? r
r r r
r
zTemplateDef.__init__c
C s~ t }| d|_t| |_tj| dddd|_t | t
f|_
t | t t t
tttttf}|j| t|tr> |S q#)Nr+ r, Fr- )r7 r0 rZ r r r[ r1 r r\ r r r r@ rC rD rF rO rP r r? r r )r tdr r r r
r s
zTemplateDef.from_bufferNr4 r r r r
r7 s r7 c @ r& )r8 c C d | _ g | _d S r )rK rH r
r r r
r r z TemplateInstanceData.__init__c C sD t }t| |_|jjD ]}t| |j|j}|j | q
|S r )
r8 rK r rL
decode_valr0 ValueByteLengthrM rH r )r tidvpvalr r r
r s z TemplateInstanceData.from_bufferNr4 r r r r
r8 r8 c @ r& )rK c C r` r ) NumValuesrL r
r r r
r r zValueSpec.__init__c C sD t }tj| dddd|_t|jD ] }|jt | q|S Nr^ r, Fr- )
rK r1 r r0 rg rangerL r ValueSpecEntryr )r vs_r r r
r s
zValueSpec.from_bufferNr4 r r r r
rK rf rK c @ r& )rj c C d | _ d | _d S r )rb rM r
r r r
r r zValueSpecEntry.__init__c C s< t }tj| dddd|_t| d|_| d}|S )Nr: r, Fr- r+ )rj r1 r r0 rb rM )r rk zeror r r
r s
zValueSpecEntry.from_bufferNr4 r r r r
rj rf rj c @ r& )r c C r` r )r@ Contentsr
r r r
r r zElement.__init__c C sh t }t| tf|_t| ttf}|j| t|tr |S t| ttf}|j| t|tr3|S q r ) r r r@ rC rD ro r r rO )r rU r r r r
r s
zElement.from_bufferNr4 r r r r
r rf r c @ r& )CharDatac C s d S r r r
r r r
r s zCharData.__init__c C s t | S r )r )r r r r
r s zCharData.from_bufferNr4 r r r r
rp rp c @ s* e Zd Zdd Zd ddZedd ZdS )
r@ c C s" d | _ d | _d | _d | _d | _d S r )OpenStartElementToken DependencyIdElementByteLengthrA
AttributeListr
r r r
r r] zStartElement.__init__NFc C sJ | j d u r d| jj S d| jj }|| j |7 }||du r dnd7 }|S )Nz<%s>z<%s F>z />)ru rA rB r )r subtabler< r r r r
r s
zStartElement.to_xmlc C sh t }| d|_tj| dddd|_tj| dddd|_t| |_|jdkr2t | |_ |S )Nr+ r: r, Fr- r^ A)
r@ r0 rr r1 r rs rt rA r ru )r ser r r
r s
zStartElement.from_buffer)NFrY r r r r
r@ s
r@ c @ r& ) Attributec C r5 r )AttributeTokenrA AttributeCharDatar
r r r
r
r9 zAttribute.__init__c C s. t }| d|_t| |_t| |_|S rX )rz r0 r{ rA r r| r vtr r r
r s
zAttribute.from_bufferNr4 r r r r
rz rz c @ * e Zd Zdd ZdddZedd ZdS ) r| c C rm r )r{ rB r
r r r
r r zAttributeCharData.__init__Nc C s | j |S r )rB r r rw r r r
r s zAttributeCharData.to_xmlc C s t }t| |_|S r )r| r rB r} r r r
r " s
z AttributeCharData.from_bufferr rY r r r r
r| s
r| c @ r ) ru c C r` r )AttributeListByteLength
Attributesr
r r r
r ) r zAttributeList.__init__Nc C s8 g }| j D ]}|d|jj|j|f qd|S )Nz%s="%s" )r r rA rB r| r rQ )r rw r attrr r r
r - s
zAttributeList.to_xmlc C sZ t }tj| dddd|_| |j }| |kr+|jt | | |ks |S rh )
ru r1 r r0 r tellr r rz r )r r~ posr r r
r 3 s zAttributeList.from_bufferr rY r r r r
ru ( s
ru c @ r ) rP c C r5 r )ValueTextToken
StringTyperB r
r r r
r > r9 zValueText.__init__Nc C s | j S r )rB r r r r
r C s zValueText.to_xmlc C s, t }| d|_| d|_t| |_|S rX )rP r0 r r read_LengthPrefixedUnicodeStringrB r} r r r
r F s
zValueText.from_bufferr rY r r r r
rP = s
rP c @ r ) rF c C s d | _ d | _d | _d| _d S )N substitution)rG rI rM rB r
r r r
r O s
zSubstitution.__init__Nc C st | j t jkr|j| j S t|j| j tr2d}|j| j D ]}|dkr)|d7 }|d| 7 }q |S t|j| j S )Nr zz%s)rM
BinXmlTyperH rI r r rJ str)r rw r xr r r
r V s zSubstitution.to_xmlc C s> t }| d|_tj| dddd|_t| d|_|S Nr+ r: r, Fr- )rF r0 rG r1 r rI rM )r sr r r
r f s
zSubstitution.from_bufferr rY r r r r
rF N s
rF c @ r& )CharRefc C rm r ) CharRefTokenrB r
r r r
r p r zCharRef.__init__c C s. t }| d|_tj| dddd|_|S r )r r0 r r1 r rB r err r r
r t s zCharRef.from_bufferNr4 r r r r
r o rf r c @ r& ) EntityRefc C rm r )EntityRefTokenrA r
r r r
r | r zEntityRef.__init__c C " t }| d|_t| |_|S rX )r r0 r rA r r r r r
r zEntityRef.from_bufferNr4 r r r r
r { rf r c @ r& ) CDATASectionc C rm r )CDATASectionTokenrB r
r r r
r r zCDATASection.__init__c C t }| d|_t| |_|S rX )r r0 r r rB )r cdr r r
r
zCDATASection.from_bufferNr4 r r r r
r rf r c @ r& )PIc C rm r )PITargetPIDatar
r r r
r r z PI.__init__c C s" t }t| |_t| |_|S r )r r r rA r )r pir r r
r r zPI.from_bufferNr4 r r r r
r rf r c @ r& )r c C rm r ) PIDataTokenrB r
r r r
r r zPIData.__init__c C r rX )r r0 r r rB )r pidr r r
r r zPIData.from_bufferNr4 r r r r
r rf r c @ r& )r c C rm r )
PITargetTokenrA r
r r r
r r zPITarget.__init__c C r rX )r r0 r rA r )r pitr r r
r r zPITarget.from_bufferNr4 r r r r
r rf r c @ r& )rA c C r5 r )NameHash NameNumCharsrB r
r r r
r r9 z
Name.__init__c C sb t }tj| dddd|_tj| dddd|_| |jd d }|dd d |_|S )Nr: r, Fr- utf-16-ler; )rA r1 r r0 r r decoderB )r rW Name_valr r r
r s zName.from_bufferNr4 r r r r
rA r rA c @ r& )rC c C
d | _ d S r rG r
r r r
r
zCloseStartElement.__init__c C t }| d|_|S rX )rC r0 rG r cr r r
r z CloseStartElement.from_bufferNr4 r r r r
rC rq rC c @ r& )rD c C r r r r
r r r
r r zCloseEmptyElement.__init__c C r rX )rD r0 rG r r r r
r r z CloseEmptyElement.from_bufferNr4 r r r r
rD rq rD c @ r& )rO c C r r r r
r r r
r r zEndElement.__init__c C r rX )rO r0 rG r r r r
r r zEndElement.from_bufferNr4 r r r r
rO rq rO c @ r& )r c C rm r )rG rB r
r r r
r r z EOF.__init__c C r rX )r r0 rG r r r r
r r zEOF.from_bufferNr4 r r r r
r rf r r+ c C s" | }| |}| |d |S )Nr )r r0 seek)r lengthr r r r r
peek_buffer s
r c C s. t j| dddd}| |d }|dS )Nr: r, Fr- r )r1 r r0 r )r NumUnicodeChars StringValuer r r
r s
r r c C sD t t| | }t|dkr|S t||r|S td|t|f )Nr z*parse_next error! %s expected but %s found)token_node_lookupr r r> r r2 type)r expectedr r r r
r s
r c C s |t jkrdS |t jkr| dS |t jkr| S |t jkr'tj| dddS |t jkr4tj| dddS |t j krAtj| dddS |t j
krNtj| dddS |t j kr[tj| dddS |t j krhtj| dddS |t j
krutj| dddS |t jkrtj| dddS |t jkrtd| S |t jkrtd| S |t jkrttj| dddS |t jkr| S |t jkrt| S |t jkrd | d d d
S |t jkrt| j S |t j krt| j S |t j krt | S |t j!krd | d d d
S |t j"kr
d | d d d
S |t j#krt$| S |t j%kr^g }d }d }|t&| k r\|| ||d
7 }|dd dksC|dkrQ|'|dd d
d }|d
7 }|t&| k s+|S t(d| )Nr r r, Tr- Fz r r2 )r data_typesarT r r r r
ra sv
ra rx E F G H I
)
r= c @ s e Zd ZdZdZdZdZdZdZdZ dZ
d Z d
Z d Z
d Zd
ZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Z!d Z"d!Z#d"Z$d#Z%d$Z&d%Z'd&Z(d'Z)d(Z*d)Z+d*Z,d+Z-d,S )-rM r r r r r r r r r r r r r r r= r ! N).r" r# r$ rN r r r r r r r r r r r r r r r r r r r r r r r AnsiStringArrayType
Int8ArrayTypeUInt8ArrayTypeInt16ArrayTypeUInt16ArrayTypeInt32ArrayTypeUInt32ArrayTypeInt64ArrayTypeUInt64ArrayTypeReal32ArrayTypeReal64ArrayType
BoolArrayType
GuidArrayTypeSizeTArrayTypeFileTimeArrayTypeSysTimeArrayType SidArrayTypeHexInt32ArrayTypeHexInt64ArrayTyper r r r
rM ] sX rM )r+ )r ),r enumr xml.dom.minidomr .aiosmb.wintypes.dtyp.constrcuted_security.guidr -aiosmb.wintypes.dtyp.constrcuted_security.sidr (aiosmb.wintypes.dtyp.structures.filetimer r r r r7 r8 rK rj r rp r@ rz r| ru rP rF r r r r r r rA rC rD rO r r r r ra r EnumrM r r r r
s J
! !
A