o juR@sddlmZmZmZddlmZddlTddlmZddl m Z ddl m Z ddl Z ddlZddlZddlmZdd lmZdd lmZGd d d ZeddedefddZddZedkrkddlZeedSdS))PureWindowsPathPurePathPath)FileAccessMask)*) FileInfoClass) SecurityInfo)FileAttributesN)asynccontextmanager) SMBConnection)SECURITY_DESCRIPTORc@seZdZddZddZddZedJded efd d Zed efd d Z ede defddZ edefddZ ede defddZ ede defddZede defddZddZde fddZd ed!efd"d#ZdKd%ed&efd'd(ZdLde d+efd,d-ZejejBfde d.efd/d0Zde d+efd1d2ZdKd!ed3efd4d5ZdMd efd7d8ZdNd ed9efd:d;Zd%efde j!fd?d@Z"dAdBZ#dCdDZ$dEefdFdGZ%dHdIZ&d*S)OSMBFilecCs|d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ d|_ d|_ d|_ d|_ d|_d|_d|_d|_d|_d|_dS)NrF)tree_id parent_dirfullpathunc_path share_pathnamesize creation_timelast_access_timelast_write_time change_timeallocation_size attributesfile_idsecurity_descriptor_SMBFile__connection_SMBFile__positionis_pipe maxreadsizemodeselfr%/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/aiosmb/commons/interfaces/file.py__init__s( zSMBFile.__init__cs|SNr%r#r%r%r& __aenter__(szSMBFile.__aenter__cs|IdHdSr()close)r$exc_typeexc_valexc_tbr%r%r& __aexit__+szSMBFile.__aexit__TbasedirrcCs|d}td|}|jddd}|jd}|ddkr(|ddd}|d ur:||gt|jd d}n ||gt|jd d}t|j|}|S) N\z\\\\._$T) lstriprpartsreplacefindlistrjoinpathresolve)r/rpath_with_fileunchostsharedirspath_to_creater%r%r&prepare_mirror_path.s   zSMBFile.prepare_mirror_pathcCs8t|}t}|j|_|j|_d|jdd|_|S)zh Creates SMBFile object from the UNC path supplied. Example uncpath: \\127.0.0.1\C$\temp\test.exe r0r1N)rr driverrjoinr9r)rr@fr%r%r& from_uncpath=s zSMBFile.from_uncpath connection remotepathcCs0d}|ddkr d}||j|f}t|S)zv Creates SMBFile object from the connection and the remote path supplied. Example remotepath: \C$\temp\test.exe z\\%s\%srr0\\%s%s)targetget_hostname_or_ipr rI)rJrKtempr@r%r%r&from_remotepathKs   zSMBFile.from_remotepathrMcCsP|jdurdS|jdd}|ddurd|}d}|||f}t|S)z3 Creates SMBFile object from the SMBUrl object N/r0FrL)pathr: startswithrNr rI)rMfpathrOr@r%r%r&from_smbtargetWs  zSMBFile.from_smbtargetpipenamecCstd|j|fS)Nz \\%s\IPC$\%s)r rIrMrN)rJrVr%r%r& from_pipenamefszSMBFile.from_pipenamec sz^t|}||jIdH\}}|dur||j}tjtjB}tj }t j t j B}t j} d} |j||j|||| | ddIdH\} }|durH|| durU||| IdH||IdHWdStyu} z d| fWYd} ~ Sd} ~ wwNrF return_replyTN)r rI tree_connectrrrDELETEFILE_READ_ATTRIBUTES ShareAccessFILE_SHARE_DELETE CreateOptionsFILE_NON_DIRECTORY_FILEFILE_DELETE_ON_CLOSECreateDisposition FILE_OPENcreaterr*tree_disconnect Exception rJrKremfile tree_entryerrrdesired_access share_modecreate_optionscreate_disposition file_attrsrer%r%r& delete_uncjs.   &zSMBFile.delete_uncc sz_t||}||jIdH\}}|dur||j}tjtjB}tj }t j t j B}t j} d} |j||j|||| | ddIdH\} }|durI|| durV||| IdH||IdHWdStyv} z d| fWYd} ~ Sd} ~ wwrX)r rPr\rrrr]r^r_r`rarbrcrdrerfrr*rgrhrir%r%r&delete_rempaths.   &zSMBFile.delete_rempathc szd|IdH|j|jIdH\}}|dur||j}tjtjB}tj }t j t j B}t j}d}|jj||j|||||ddIdH\} }|durL|| durZ|j|| IdH|j|IdHWdSty{} z d| fWYd} ~ Sd} ~ ww)zcDeletes the file. If the file is open, a flush will be performed followed by a close then a delete.NrFrYr[)r*rr\rrrr]r^r_r`rarbrcrdrerfrrgrh) r$rkrlrrmrnrorprqrrrr%r%r&deletes@   zSMBFile.deletec s|jdurd}zzi|j}|jdur&||jIdH\}}|dur#||j}tj}tj}tj tj B}d} t j } | ||j|||| | IdH\}}|durN||j||tjtjtjtjBtjBtjBddIdH\|_}|durq|Wn7ty} z+d| fWYd} ~ W|dur|||IdH|dur|jdur||IdHSSSd} ~ wwW|dur|||IdH|dur|jdur||IdHn!|dur|||IdH|dur|jdur||IdHwww|jdfS)z,Fetches the security descriptor of the file.Nr) info_typeinformation_classadditional_informationflags)rrr\rr READ_CONTROLr_FILE_SHARE_READrarbFILE_SYNCHRONOUS_IO_NONALERTrdrerfr query_info QueryInfoTypeSECURITYrNONErATTRIBUTE_SECURITY_INFORMATIONDACL_SECURITY_INFORMATIONOWNER_SECURITY_INFORMATIONGROUP_SECURITY_INFORMATIONrhr*rg) r$rJrrrkrlrmrnrorqrprrr%r%r&get_security_descriptorsd   " zSMBFile.get_security_descriptorroffsetc s|jdkr|jj|j|j||dIdH\}}}||fSd}||jjkrc||jj}t|dD]*}|jj|j|j||jjdIdH\}}}|durPd|fS|t|7}||7}q0|d||fS|jj|j|j||jjdIdH\}}}|durd|fS||7}|d||fS)a! This is the main function for reading. It does not do buffering, so if more data is returned it will just discard it If less data is returned than requested it will do more reads until the requested size is reached. Do not call this directly as it could go in an infinite loop T)rlengthNr1)r rreadrr MaxReadSizerangelen) r$rrdata remainingrlbufferir3r%r%r&__reads& $  (   (zSMBFile.__readrrposition_in_filec sz@t|}d}d}|dkr=|jj|j|j||t|||dIdH\}}|dur-|||7}||8}||7}|dks|dfWStyW}z d|fWYd}~Sd}~ww)z Data must be bytes r)rN)rrwriterrrh) r$rrrtotal_bytes_writtenr bytes_writtenrlrrr%r%r&__writes$2 zSMBFile.__writerNr"c sz||_|j|_||_d|jvrd|_d|_|js/||jIdH\}}|dur+||j|_d|vr;d|vr;t dd|vr|t j t j B}|durLt jn|}tjtjB}d}tj} |j|j|j|||| |ddIdH\|_} }|duru|| j|_Wd Sd|vrt jt jB}|durt jt jBn|}tjtjB}d}tj} |j|j|j|||| |ddIdH\|_} }|dur|| j|_Wd Stdty} z d | fWYd} ~ Sd} ~ ww) NpTrrwz(must have exactly one of read/write moderYz/ONLY read and write is supported at the moment!r[F)rrr!r"r rrr\r ValueErrorrFILE_READ_DATAr^r_r{rarbr|rdrerfrr EndofFile GENERIC_READ GENERIC_WRITEFILE_SHARE_WRITE FILE_OPEN_IFrh) r$rJr"rnrkrlrmrorqrp smb_replyrrr%r%r&open#sT   ,  ,z SMBFile.open local_pathc szz|jdur dnd}|jrtd|jdurtd|j|d|dIdH\}}|dur/|t|}|r=||j}t|d%}| 2z3dHW\}}|durU||sYn| |qG6Wdn1sjwYt | dfWW| IdH|jdur|dur||jIdHSSSty} z&d| fWYd} ~ W| IdH|jdur|dur||jIdHSSSd} ~ ww| IdH|jdur|dur||jIdHwww) zntd|j|7_WdStd|dkrfd|j|krT|jkrbntd|j||_WdStdWdSty~}z d|fWYd}~Sd}~ww)zASets the current position of the file buffer to the given offset.rz#Offset must be > 0 when whence is 0zSeeking outside of file size!r1r4r[N)rhrr)r$rrrrr%r%r&seeks4   z SMBFile.seekr6c sdz|jdur|dkr|jj}||dIdH\}}||fWS||jkr&d}|dkr-WdS|dkr^|j|jkr:WdS||j|j|jIdH\}}|durP||jt|7_||fWS|dkr|j|jkrkWdS||j|jkry|j|j}|||jIdH\}}|dur||jt|7_||fWSWdSty}z d|fWYd}~Sd}~ww)NTr6r)rN)r rr_SMBFile__readrrrrh)r$rrrlrrr%r%r&rsD          z SMBFile.read chunksizecCs\|jdur td|dkr|jj}|dkrdVdS|dkrX |}|j|j|kr9|j|j}|dkr9dVt|||jIdH\}}|durR|jt|7_||fVq |dkr|j|jkrgdVt||j|jkru|j|j} |}||j|kr||j}|dkrdVt|||jIdH\}}|dur|jt|7_||fVqvdS)aH Much like read, but yields chuks of chunksize untill the full size is read Use this when reading large files as a whole, as this method doesn't caches the whole data read in memory, only chunskize chunksize -1 will use the maximum chunk allowed by the underlying connection layer, it is advised to not set manually! TzCant stream pipes!r6r)NNN) r rhrrrr StopIterationrr)r$rrreq_sizerrlr%r%r&rsN        zSMBFile.read_chunkedc szht||jjkr-|||jIdH\}}|dur||jdkr(|j|7_|dfWSd}|t|kre|||||jj|jIdH\}}|durO|||7}|jdkr_|j|7_|t|ks5|dfWSty}z d|fWYd}~Sd}~ww)z=Writes the given bytes to the file from the current position.NFr)rr MaxWriteSize_SMBFile__writerr rh)r$rcountrl total_writenrrr%r%r&rs.   (  z SMBFile.writerc szD|jdkr tdd} tdIdH||jj}t|dkr(|dfWS|||j IdH\}}|dur:||j |7_ ||7}qty[}z d|fWYd}~Sd}~ww)zVWrites the contents of the buffer(or file handle) to the file. Doesnt work with pipes!TzDoesnt work with pipes!rN) r rhasynciosleeprrrrrr)r$rrchunkrrlrrr%r%r& write_buffer"s(    zSMBFile.write_buffercs,d|jvrdS|j|j|jIdHdS)z*Issues a flush command on the SMB protocolrN)r"rflushrrr#r%r%r&r8s z SMBFile.flushcsV|jdurdS|jdurdSz |IdHWnY|j|j|jIdHdS)z$Closes the file including the fileIdN)rrrrr*r#r%r%r&r*>s  z SMBFile.closereturncCs|jS)z Returns current position in file)rr#r%r%r&tellKsz SMBFile.tellcCs6d}|jD]}|dr q|d||j|f7}q|S)Nz===== FILE ===== parent_z %s : %s )__dict__rS)r$tkr%r%r&__str__Os   zSMBFile.__str__)T)rrN)r6)r6r6)'__name__ __module__ __qualname__r'r)r. staticmethodrrErIr rPrUrWrsrtrurintrbytesrrr_r{rrrrrrrioBytesIOrrr*rrr%r%r%r&r sD  &*3"(2 r rrRrc Csz|drddlm}||}|durp|}|4IdHB|IdH\}}|dur2||}|||IdH\}}|durG||V WdIdHW|durb| IdHdSdS1IdHsjwYnR|}|||IdH\}}|dur||VW|dur| IdHdSdSt ||}|||IdH\}}|dur||VW|dur| IdHdSdSW|dur| IdHdSdS|dur| IdHww)zj Opens a file on the remote server path can be a smb:// url or a path like \\server\share\path\to\file smbrSMBConnectionFactoryN) lowerrS!aiosmb.commons.connection.factoryrfrom_urlget_connectionloginget_filerr*r rP)rRr"rJrfactoryr3rlfiler%r%r&smb_openXs\      rc s"d}t|d4IdH}|dIdH\}}t|tt|WdIdHn 1IdHs1wYddlm}||}|IdH\}}|durR|tdd|d4IdH&} |dIdH\}}t|tt||d krxnq_WdIdHn 1IdHswYtd d |d4IdH}| d IdH\}}|durt|WdIdHn 1IdHswYtd d|d4IdH3}|dIdH\}}|durt| WdIdHdSt|tt|WdIdHn 1IdHswYtd dS)NzSsmb2+ntlm-password://TEST\Administrator:Passw0rd!1@10.10.10.2/C$/temp/repodata.jsonrbirrzC$/temp/repodata.json)rJTrzC$/temp/repodata1.jsonrsHELLOzDONE!) rrprintrrrrrrr)urlrHrrlrrJr3r%r%r&amainsJ(  ((* r__main__r)pathlibrrraiosmb.wintypes.access_maskraiosmb.protocol.smb2.commands-aiosmb.wintypes.fscc.structures.fileinfoclassr(aiosmb.protocol.smb2.commands.query_infor#aiosmb.wintypes.fscc.FileAttributesr rdatetimer contextlibr aiosmb.connectionr winacl.dtyp.security_descriptorr r rrrrrunr%r%r%r&s0       M($