File manager - Edit - /opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/smb.cpython-310.pyc
Back
o ;j�, � @ s� d Z ddlZddlZddlZddlZddlZddlZddlm Z e� e �Z zOddl mZ ddlmZmZmZ ddlmZmZmZmZmZmZmZmZm Z ddl m Z m Z m Z m!Z!m"Z"m#Z# ddl$m%Z% dd l&m'Z' e� d ��(ej)� d Z*W n e+y d Z*Y nw G d d� d�Z,d+dd�Z-d,dd�Z. d-dd�Z/d-dd�Z0 d-dd�Z1 d-d d �Z2 d.d d �Z3 d-d!d"�Z4 d-d#d$�Z5d-d%d&�Z6 d-d'd(�Z7 d-d)d*�Z8dS )/z; Utility functions for SMB connections :depends: impacket � N)� MissingSmb)� Connection)�CreateContextName�SMB2CreateContextRequest�#SMB2CreateQueryMaximalAccessRequest) �CreateDisposition� CreateOptions�DirectoryAccessMask�FileAttributes�FileInformationClass�FilePipePrinterAccessMask�ImpersonationLevel�Open� ShareAccess)�AccessAllowedAce� AccessMask� AclPacket� SDControl� SIDPacket�SMB2CreateSDBuffer)�Session)� TreeConnectZ smbprotocolTFc @ s` e Zd Zddd�Zdd� Zdd� Zedd � �Zd d � Ze d d � �Z e dd� �Z e ddd��Z dS )�SMBProto� c C sT t �� }t�|d ddtj�d d d }|| _t|||dd�}t|||dd�| _d S )Nr � T)Zrequire_signingF)Zrequire_encryption) �uuidZuuid4�socket� getaddrinfo� IPPROTO_TCP�serverr r �session)�selfr �username�password�portZ connection_id�addr� connection� r' �B/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/smb.py�__init__7 s zSMBProto.__init__c C s | j �� | j�� d S �N)r& �connectr �r! r' r' r( r+ > s zSMBProto.connectc C s | j j�d� d S )NT)r r& Z disconnectr, r' r' r( �closeB s zSMBProto.closec C s | j jS r* )r r& r, r' r' r( r& E s zSMBProto.connectionc C s4 |� d�rd| j� d|� �}t| j|�}|�� |S )N�$z\\�\)�endswithr r r r+ )r! �share�treer' r' r( � tree_connectI s zSMBProto.tree_connectc C s | � d�S )Nr/ )�lstrip)�filer' r' r( �normalize_filenameP s zSMBProto.normalize_filenamec C s� | � |�}t� }tj|d< t� |d< t� }|�d� t� }tj |d< ||d< t � }|g|d< t � }|d � t j� |�|� t� }tj|d< ||d< ||g} t||�} | �tjtjtjB tjtjtj B t j t j �} | S )NZ buffer_nameZ buffer_datazS-1-1-0�mask�sidZacesZcontrol)!r6 r r Z(SMB2_CREATE_QUERY_MAXIMAL_ACCESS_REQUESTr r Z from_stringr r Z GENERIC_ALLr r Zset_flagr Z SELF_RELATIVEZset_daclZSMB2_CREATE_SD_BUFFERr �creater � Impersonationr � GENERIC_READ� GENERIC_WRITEr �FILE_ATTRIBUTE_NORMALr �FILE_SHARE_READ�FILE_SHARE_WRITEr ZFILE_OVERWRITE_IFr �FILE_NON_DIRECTORY_FILE) �clsr2 r5 Zmax_reqZ everyone_sidZaceZaclZsec_descZ sd_bufferZcreate_contexts� file_openZ open_infor' r' r( � open_fileT s> � � � zSMBProto.open_fileFc C s> t | |�}|r |�tjtjtjB tjt j t j B t j tj� |S r* )r r9 r r: r r; r< r �FILE_ATTRIBUTE_DIRECTORYr r> r? r Z FILE_OPEN_IFr �FILE_DIRECTORY_FILE)r2 �namer9 �dir_openr' r' r( �open_directoryz s �zSMBProto.open_directoryN)r )F)�__name__� __module__� __qualname__r) r+ r- �propertyr& r3 � staticmethodr6 � classmethodrC rH r' r' r' r( r 6 s %r � r c C s t | |||�}|�� |S r* )r r+ )�hostr" r# Z client_namer$ �connr' r' r( �_get_conn_smbprotocol� s rR c C s" t rt�d� t| |||d�S dS )z Get an SMB connection zGet connection smbprotocol)r$ F)�HAS_SMBPROTOCOL�log�inforR )rP r" r# r$ r' r' r( �get_conn� s rV �C$c C s� |d u r t |||�}|du rdS |�|�}| �d�}d}|D ]I} d�|d|� �} |j|| dd�} | jdtjdd �| jddd �g} |j j � d d � | D �|j j |j �} t| �D ] \}}| | d |�}qV|d7 }q d S ) NF�/� r/ r T)r9 �*��sendc S � g | ]}|d �qS �r r' ��.0�xr' r' r( � <listcomp>� � z'_mkdirs_smbprotocol.<locals>.<listcomp>)rV r3 �split�joinrH Zquery_directoryr ZFILE_NAMES_INFORMATIONr- r r& � send_compound� session_id�tree_connect_id� enumerate)�pathr1 rQ rP r" r# r2 �comps�pos�comp�cwdrG Zcompound_messages�requests�i�request�responser' r' r( �_mkdirs_smbprotocol� s0 � � � �rs c C � t r t| |||||d�S td���N�rQ rP r" r# �.SMB library required (impacket or smbprotocol))rS rs r �rj r1 rQ rP r" r# r' r' r( �mkdirs� � �ry c C sf |d u r t |||�}|du rdS |�|�}z|�||�}|�tjj�| �d� W |�� d S |�� w )NFr ) rV r3 rC �write�salt�utilsZ stringutils�to_bytesr- ) �contentrj r1 rQ rP r" r# r2 rB r' r' r( �_put_str_smbprotocol� s r� c C �" t r t| ||||||d�S td��)z� Wrapper around impacket.smbconnection.putFile() that allows a string to be uploaded, without first writing it as a local file rv rw )rS r� r )r rj r1 rQ rP r" r# r' r' r( �put_str� s � r� � c C s� |d u r t |||�}|du rdS |�|�}|�||�} tjj�| d��/} z d} | �|�} | s0n | �| | � | t | �7 } q(W | � d� n| � d� w W d � d S 1 sUw Y d S )NF�rbr ) rV r3 rC r| r} �filesZfopen�readr{ �lenr- ) � local_pathrj r1 rQ rP r" r# Z chunk_sizer2 rB Zfh_�position�chunkr' r' r( �_put_file_smbprotocol� s& �"�r� c C r� )a; Wrapper around impacket.smbconnection.putFile() that allows a file to be uploaded Example usage: import salt.utils.smb smb_conn = salt.utils.smb.get_conn('10.0.0.45', 'vagrant', 'vagrant') salt.utils.smb.put_file('/root/test.pdf', 'temp\myfiles\test1.pdf', conn=smb_conn) rv rw )rS r� r )r� rj r1 rQ rP r" r# r' r' r( �put_file s � r� c C s� |d u r t |||�}|du rdS |�|�}t|| �}|jtjtjtjB t j t j t j B tjtjtjB dd�|jddd�g}|jjdd� |D �|jj|jdd�} g } t| �D ]\} } || d | �} | �| � qTd S )NFr[ c S r] r^ r' r_ r' r' r( rb 3 rc z,_delete_file_smbprotocol.<locals>.<listcomp>T)�relatedrY )rV r3 r r9 r r: r r; �DELETEr r= r r> r? r � FILE_OPENr r@ �FILE_DELETE_ON_CLOSEr- r& rf r rg rh ri �append)rj r1 rQ rP r" r# r2 rB � delete_msgsro � responsesrp rq rr r' r' r( �_delete_file_smbprotocol s8 � � � �r� c C rt ru )rS r� r rx r' r' r( � delete_file? rz r� c C s� |d u r t |||�}|du rdS t�d|| � |�|�}t|| �}|jtjtj t j dt j tjtjB dd�|jddd�g}|jjdd� |D �|jj|jdd�} t| �D ] \} } || d | �} qRd S ) NFz3_delete_directory_smbprotocol - share: %s, path: %sr r[ c S r] r^ r' r_ r' r' r( rb _ rc z1_delete_directory_smbprotocol.<locals>.<listcomp>T)r8 �tidr� rY )rV rT �debugr3 r r9 r r: r r� r rD r r� r rE r� r- r& rf r rg rh ri ) rj r1 rQ rP r" r# r2 rG r� Z delete_reqsrp rq rr r' r' r( � _delete_directory_smbprotocolG s6 � � ��r� c C rt ru )rS r� r rx r' r' r( �delete_directoryi s �r� )rO rO rO rO r )rO NNr )rW NNNN)rW NNNNr� )9�__doc__�loggingr r Zsalt.utils.filesr| Zsalt.utils.stringutilsZsalt.utils.versionsZsalt.exceptionsr � getLoggerrI rT Zsmbprotocol.connectionr Zsmbprotocol.create_contextsr r r Zsmbprotocol.openr r r r r r r r r Z smbprotocol.security_descriptorr r r r r r Zsmbprotocol.sessionr Zsmbprotocol.treer �setLevel�WARNINGrS � ImportErrorr rR rV rs ry r� r� r� r� r� r� r� r� r' r' r' r( �<module> s^ , � T � � � � � � " �#�
| ver. 1.4 |
Github
|
.
| PHP 8.2.30 | Generation time: 0 |
proxy
|
phpinfo
|
Settings