o CD©iºÑã@sddlZddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl Tddl m Z dZ dZgZedgƒD]\Z\ZZe e¡eeƒe<q=gZed gƒD]\Z\ZZe e¡eeeƒeƒe<qUGd d „d eƒZGd d „d eƒZGdd„deƒZGdd„deƒZGdd„dƒZGdd„dƒZGdd„dƒZGdd„deƒZ Gdd„deƒZ!Gdd„deƒZ"Gdd„deƒZ#Gd d!„d!eƒZ$Gd"d#„d#eƒZ%Gd$d%„d%eƒZ&Gd&d'„d'eƒZ'Gd(d)„d)eƒZ(Gd*d+„d+eƒZ)Gd,d-„d-eƒZ*Gd.d/„d/eƒZ+Gd0d1„d1eƒZ,Gd2d3„d3eƒZ-Gd4d5„d5eƒZ.Gd6d7„d7eƒZ/Gd8d9„d9eƒZ0Gd:d;„d;eƒZ1Gdd?„d?eƒZ3Gd@dA„dAeƒZ4GdBdC„dCeƒZ5GdDdE„dEeƒZ6GdFdG„dGeƒZ7GdHdI„dIeƒZ8GdJdK„dKeƒZ9GdLdM„dMeƒZ:GdNdO„dOeƒZ;dS)PéN©Úcopy)ÚStringIOé)Ú*)ÚDataFaultToleranceStrategyT)ÚNT_LAN_MANAGER_DIALECTs NT LM 0.12)Ú SMB2_DIALECTs SMB 2.002c@seZdZdZdS)ÚUnsupportedFeaturezz Raised when an supported feature is present/required in the protocol but is not currently supported by pysmb N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©rrúH/var/www/Datamplify/venv/lib/python3.10/site-packages/smb/smb_structs.pyr sr c@seZdZddd„Zdd„ZdS)Ú ProtocolErrorNcCó||_||_||_dS©N)ÚmessageÚdata_bufÚ smb_message)ÚselfrrrrrrÚ__init__$ó zProtocolError.__init__cCs~tƒ}| |jtj¡|jr| dtj¡| t|jƒ¡|jr;| dtj¡| tt  |j¡ƒ¡| tj¡|  ¡S)Nz5==================== SMB Message ====================z?==================== SMB Data Packet (hex) ====================) rÚwriterÚosÚlineseprÚstrrÚbinasciiÚhexlifyÚgetvalue©rÚbrrrÚ__str__)s zProtocolError.__str__)NN©r r r rr#rrrrr"s  rc@seZdZdd„ZdS)ÚSMB2ProtocolHeaderErrorcCst |d¡dS)NzPacket header belongs to SMB2)rr©rrrrr9sz SMB2ProtocolHeaderError.__init__N)r r r rrrrrr%7s r%c@seZdZdd„Zdd„ZdS)ÚOperationFailurecCs&||_dd„|Dƒ|_||jf|_dS)NcSsg|]}t|ƒ‘qSrr)Ú.0ÚmsgrrrÚ @sz-OperationFailure.__init__..)rÚ smb_messagesÚargs)rrr+rrrr>szOperationFailure.__init__cCsÀtƒ}| |jtj¡t|jƒD]J\}}| dd|dtj¡| dtj¡| dtj¡| t|ƒ¡| dtj¡| dtj¡| tt  |j ¡ƒ¡| tj¡q|  ¡S)Nz====================z SMB Message %d z SMB Header:z -----------zSMB Data Packet (hex):z----------------------) rrrrrÚ enumerater+rrrÚraw_datar )rr"ÚidxÚmrrrr#CszOperationFailure.__str__Nr$rrrrr'<s r'c@s0eZdZdd„Zdd„Zdd„Zedd„ƒZd S) ÚSMBErrorcCs | ¡dSr)Úresetr&rrrrVs zSMBError.__init__cCsd|_d|_dS)NrT)Úinternal_valueÚ is_ntstatusr&rrrr2Yó zSMBError.resetcCs(|jrd|jSd|jd?|jd@fS)NzNTSTATUS=0x%08Xz"ErrorClass=0x%02X ErrorCode=0x%04Xééÿÿ)r4r3r&rrrr#]s zSMBError.__str__cCs |jdkS)Nr)r3r&rrrÚhasErrorcs zSMBError.hasErrorN)r r r rr2r#Úpropertyr8rrrrr1Ts r1c@s~eZdZdZe e¡Ze d¡Z dZ ddd„Z dd„Z d d „Z ed d „ƒZed d„ƒZedd„ƒZdd„Zdd„Zdd„ZdS)Ú SMBMessagez<4sBIBHHQxxHHHHBzSMB.SMBMessagerNcCs&| ¡|r||_|j |¡dSdSr)r2ÚpayloadÚ initMessage)rr;rrrrps þzSMBMessage.__init__c Cs<tƒ}| d|jt |jd¡tjf¡| dt|jƒtjf¡| d|j tjf¡| d|j tjf¡| d|j tjf¡| d|j tjf¡| d|j tjf¡| d |jtjf¡| d |jtjf¡| d t|jƒtjtt |j¡ƒtjf¡| d t|jƒtjtt |j¡ƒtjf¡| ¡S) NzCommand: 0x%02X (%s) %sz z Status: %s %szFlags: 0x%02X %szFlags2: 0x%04X %sz PID: %d %sz UID: %d %sz MID: %d %sz TID: %d %szSecurity: 0x%016X %szParameters: %d bytes %s%s %szData: %d bytes %s%s %s)rrÚcommandÚSMB_COMMAND_NAMESÚgetrrrÚstatusÚflagsÚflags2ÚpidÚuidÚmidÚtidÚsecurityÚlenÚparameters_datarrÚdatar r!rrrr#vs",,zSMBMessage.__str__cCsTd|_d|_tƒ|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ dS)Nór)r.r=r1r@rArBrCrFrDrErGrIrJr;r&rrrr2…s zSMBMessage.resetcCót|jt@ƒSr)ÚboolrAÚSMB2_FLAGS_ASYNC_COMMANDr&rrrÚisAsync”ózSMBMessage.isAsynccCrLr)rMrAÚSMB_FLAGS_REPLYr&rrrÚisReply˜rPzSMBMessage.isReplycCrLr)rMrBÚSMB_FLAGS2_EXTENDED_SECURITYr&rrrÚhasExtendedSecurityœrPzSMBMessage.hasExtendedSecuritycCsª|jsJ‚t ¡|_|j |¡t|jƒ}|ddksJ‚t |j d|j |j j |j |j|jd?d@|j|j|jd@|j|jt|dƒ¡ }||jt dt|jƒ¡|jS)a! Encode this SMB message into a series of bytes suitable to be embedded with a NetBIOS session message. AssertionError will be raised if this SMB message has not been initialized with a Payload instance @return: a string containing the encoded SMB message éróÿSMBér7ú|B|_||j_t|j t@ƒ|j_|j}||ddkretd|ƒ‚||d} t d || | d…¡d } | d krŠ|| d| krŠtd |ƒ‚||| …|_| d kr¢|| d| d| …|_||_| ¡|j|dd| S) a  Decodes the SMB message in buf. All fields of the SMBMessage object will be reset to default values before decoding. On errors, do not assume that the fields will be reinstated back to what they are before this method is invoked. @param buf: data containing one complete SMB message @type buf: string @return: a positive integer indicating the number of bytes used in buf to decode this SMB message @raise ProtocolError: raised when decoding fails z$Not enough data to decode SMB headerNsþSMBrVzInvalid 4-byte protocol fieldrWrUz0Not enough data. Parameters list decoding failedrXrz%Not enough data. Body decoding failed)rHÚHEADER_STRUCT_SIZErr2r[Úunpackr]r=rArBrGrFrDrEr%rCr@r3rMÚSMB_FLAGS2_NT_STATUSr4rIrJr.Ú_decodePayload) rÚbufÚbuf_lenÚprotocolr@Úpid_highÚpid_lowÚ params_countÚoffsetÚdatalen_offsetÚbody_lenrrrÚdecodeµs8  þ     zSMBMessage.decodecCsà|jtkr tƒ|_nY|jtkrtƒ|_nO|jtkrtƒ|_nE|jtkr(t ƒ|_n;|jt kr2t ƒ|_n1|jt kr.cSr›rœrrŸrrrr*-r¡)r;rIÚ SUPPORT_SMB2ÚjoinÚDIALECTSÚ DIALECTS2rJr‘rrrrZ's zComNegotiateRequest.prepareN)r r r rr<rZrrrrr˜s r˜c@s2eZdZdZdZe e¡Zdd„Ze dd„ƒZ dS)rƒaÏ Contains information on the SMB_COM_NEGOTIATE response from server After calling the decode method, each instance will contain the following attributes, - security_mode (integer) - max_mpx_count (integer) - max_number_vcs (integer) - max_buffer_size (long) - max_raw_size (long) - session_key (long) - capabilities (long) - system_time (long) - server_time_zone (integer) - challenge_length (integer) If the underlying SMB message's flag2 does not have SMB_FLAGS2_EXTENDED_SECURITY bit enabled, then the instance will have the following additional attributes, - challenge (string) - domain (unicode) If the underlying SMB message's flags2 has SMB_FLAGS2_EXTENDED_SECURITY bit enabled, then the instance will have the following additional attributes, - server_guid (string) - security_blob (string) References: =========== - [MS-SMB]: 2.2.4.5.2.1 - [MS-CIFS]: 2.2.4.52.2 z Not enough data to decode SMB_COM_TREE_CONNECT_ANDX parameters) r=r|r@r8rHrIr¶rr.r[rcrµÚoptional_support©rrr½rrrroPs (üz!ComTreeConnectAndxResponse.decodeN© r r r rrµr[r„r¶rorrrrr}Aó   r}c@s<eZdZdZdZe e¡Z  d dd„Zdd„Z dd „Z d S) ÚComNTCreateAndxRequestz[ References: =========== - [MS-CIFS]: 2.2.4.64.1 - [MS-SMB]: 2.2.4.9.1 z eZdZdZdZe e¡Z   d dd„Zdd „Z d d „Z d S)ÚComTransactionRequestzA References: =========== - [MS-CIFS]: 2.2.4.33.1 zeZdZdZdZe e¡Z   d dd„Zdd„Z d d „Z d S) ÚComTransaction2RequestzA References: =========== - [MS-CIFS]: 2.2.4.46.1 ÚHHHHBBHIHHHHHHrrKc CsP|pt|ƒ|_|p t|ƒ|_||_||_||_| |_| |_||_||_ ||_ dSr) rHrùrúrûrürýrArþrÿrr) rrûrürýrùrúrÿrrrArþrrrr@s zComTransaction2Request.__init__cCr™r)r‰r<rvr=r‘rrrr<Oršz"ComTransaction2Request.initMessagec Cs.t|jƒ}t|jƒ}t|jƒ}d}d}|j|j|d}|ddkr)d}|d7}|d7}d}|ddkr=dd|d}|dkrH|} ||7}nd} d} |ddkrZdd|d} |dkra|} nd} t |j|j |j |j |j |j d|j|jd|| || t|dƒ¡|j|_||||j| |j|_dS)Nr©rKrUrržrrŠ)rHrrÿrrbr¶r[r\rµrùrúrûrürýrArþr^rIrJ) rrrrrrrrlrr r r rrrrZSsV        òñÿ"zComTransaction2Request.prepareN)rrrKrKrKrrrãrrrrr6s  ý rc@rä)rwal Contains information about a SMB_COM_TRANSACTION2 response from the server After decoding, each instance contains the following attributes: - total_params_count (integer) - total_data_count (integer) - setup_bytes (string) - data_bytes (string) - params_bytes (string) References: =========== - [MS-CIFS]: 2.2.4.46.2 z B|_ dSdS)Nz7Not enough data to decode SMB_COM_WRITE_ANDX parametersr7rW)r=rrr@r8rHrIr¶rr.r[rcrµÚ availableÚcount)rrr½r)Ú high_countrrrro`s*ûzComWriteAndxResponse.decodeNrçrrrrrsUó   rsc@rÜ) ÚComReadAndxRequestz[ References: =========== - [MS-CIFS]: 2.2.4.42.1 - [MS-SMB]: 2.2.4.2.1 zsf8      _#81&7V6R6/!0* H