Current File : //usr/lib/python2.7/site-packages/salt/utils/smb.pyc
�
���^c @@ s� d Z d d l m Z m Z m Z d d l Z d d l Z d d l Z d d l Z d d l
Z d d l Z e j e
� Z d d l m Z y6 d d l Z d d l m Z d d l m Z e Z Wn e k
r� e Z n Xy� d d l m Z d d l m Z d d l m Z d d l! m" Z" m# Z# m$ Z$ m% Z% m& Z& m' Z' m( Z( m) Z) m* Z* d d
l+ m, Z, m- Z- m. Z. d d l/ m0 Z0 m1 Z1 m2 Z2 m3 Z3 m4 Z4 m5 Z5 e j d � j6 e j7 � e Z8 Wn e k
r�e Z8 n Xd
e9 f d � � YZ: d e9 f d � � YZ; d d d d d d � Z= d d d d d d � Z> d d d d d � Z? d d d d d d � Z@ d d d d d d � ZA d d d d d d � ZB d d d d d d � ZC d d d d d d � ZD d d d d d d � ZE d d d d d d � ZF d d d d d d' d � ZG d d d d d d � ZH d d d d d d! � ZI d d d d d d" � ZJ d d d d d d# � ZK d d d d d d$ � ZL d d d d d d% � ZM d d d d d d& � ZN d S(( u;
Utility functions for SMB connections
:depends: impacket
i ( t absolute_importt print_functiont unicode_literalsN( t
MissingSmb( t SessionError( t
Connection( t Session( t TreeConnect( t Opent ImpersonationLevelt FilePipePrinterAccessMaskt FileAttributest CreateDispositiont
CreateOptionst ShareAccesst DirectoryAccessMaskt FileInformationClass( t CreateContextNamet SMB2CreateContextRequestt# SMB2CreateQueryMaximalAccessRequest( t AccessAllowedAcet
AccessMaskt AclPackett SDControlt SIDPackett SMB2CreateSDBufferu smbprotocolt SMBProtoc B@ sn e Z d d � Z d � Z d � Z e d � � Z d � Z e d � � Z e
d � � Z e e d � � Z
RS( i� c C@ s[ t j � } t j | � } | | _ t | | | d t �} t | | | d t �| _ d S( Nt require_signingt require_encryption(
t uuidt uuid4t sockett
gethostbynamet serverR t TrueR t Falset session( t selfR! t usernamet passwordt portt
connection_idt addrt
connection( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt __init__6 s
c C@ s | j j � | j j � d S( N( R+ t connectR$ ( R% ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyR- = s
c C@ s | j j j t � d S( N( R$ R+ t
disconnectR" ( R% ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt closeA s c C@ s
| j j S( N( R$ R+ ( R% ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyR+ D s c C@ sG | j d � r' d j | j | � } n t | j | � } | j � | S( Nu $u \\{}\{}( t endswitht formatR! R R$ R- ( R% t sharet tree( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt tree_connectH s
c C@ s
| j d � S( Nu \( t lstrip( t file( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt normalize_filenameO s c C@ s% | j | � } t � } t j | d <t � | d <t � } | j d � t � } t j | d <| | d <t
� } | g | d <t � } | d j t
j � | j | � t � } t j | d <| | d <| | g } t | | � }
|
j t j t j t j Bt j t j t j Bt j t j � } |
S( Nu buffer_nameu buffer_datau S-1-1-0u masku sidu acesu control(! R7 R R t( SMB2_CREATE_QUERY_MAXIMAL_ACCESS_REQUESTR R t from_stringR R t GENERIC_ALLR R t set_flagR t
SELF_RELATIVEt set_daclt SMB2_CREATE_SD_BUFFERR t createR t
ImpersonationR
t GENERIC_READt
GENERIC_WRITER t FILE_ATTRIBUTE_NORMALR t FILE_SHARE_READt FILE_SHARE_WRITER t FILE_OVERWRITE_IFR
t FILE_NON_DIRECTORY_FILE( t clsR3 R6 t max_reqt everyone_sidt acet aclt sec_desct sd_buffert create_contextst file_opent open_info( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt open_fileS s:
c C@ sX t | | � } | rT | j t j t j t j Bt j t j
t j Bt j
t j � n | S( N( R R? R R@ R RA RB R t FILE_ATTRIBUTE_DIRECTORYR RD RE R t FILE_OPEN_IFR
t FILE_DIRECTORY_FILE( R3 t nameR? t dir_open( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt open_directoryz s
( t __name__t
__module__R, R- R/ t propertyR+ R4 t staticmethodR7 t classmethodRR R# RX ( ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyR 4 s 't StrHandlec B@ s# e Z d Z d � Z d d � Z RS( u�
Fakes a file handle, so that raw strings may be uploaded instead of having
to write files first. Used by put_str()
c C@ s | | _ t | _ d S( u
Init
N( t contentR# t finished( R% R_ ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyR, � s c C@ s | j s t | _ | j Sd S( u*
Looks like a file handle
u ( R` R" R_ ( R% t writesize( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt string� s N( RY RZ t __doc__R, t NoneRb ( ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyR^ � s i� c C@ s; t j j d | d | d | � } | j d | d | � | S( Nt
remoteNamet
remoteHostt myNamet userR' ( t impackett
smbconnectiont
SMBConnectiont login( t hostR& R' t client_nameR( t conn( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _get_conn_impacket� s u c C@ s# t | | | | � } | j � | S( N( R R- ( Rm R& R' Rn R( Ro ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _get_conn_smbprotocol� s
c C@ s| t r& t r& t j j j d d � n t rO t j d � t | | | d | �St rx t j d � t | | | d | �St
S( u
Get an SMB connection
u Sodiumuj Support of impacket has been depricated and will be removed in Sodium. Please install smbprotocol instead.u Get connection smbprotocolR( u Get connection impacket( t HAS_IMPACKETt HAS_SMBPROTOCOLt saltt utilst versionst
warn_untilt logt infoRq Rp R# ( Rm R& R' R( ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt get_conn� s
u C$c
C@ s� | d k r! t | | | � } n | t k r1 t S| j d � } d } xy | D]q } d j | d | !� } y | j | | � Wn4 t t f k
r� t j d � | j
| | � n X| d 7} qM Wd S( u�
Recursively create a directory structure on an SMB share
Paths should be passed in with forward-slash delimiters, and should not
start with a forward-slash.
u /i u \i u' Encountered error running conn.listPathN( Rd Rz R# t splitt joint listPatht smbSessionErrort smb3SessionErrorRx t exceptiont createDirectory(
t pathR2 Ro Rm R& R' t compst post compt cwd( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _mkdirs_impacket� s
c C@ s> | d k r! t | | | � } n | t k r1 t S| j | � } | j d � } d } x� | D]� } d j | d | !� }
| j | |
d t �} | j d t j
d t �| j t d t �g } | j j
j g | D] }
|
d ^ q� | j j | j � } x. t | � D] \ } } | | d | � } qW| d 7} q\ Wd S( Nu /i u \i R? u *t send( Rd Rz R# R4 R{ R| RX R" t query_directoryR t FILE_NAMES_INFORMATIONR/ R$ R+ t
send_compoundt
session_idt tree_connect_idt enumerate( R� R2 Ro Rm R& R' R3 R� R� R� R� RW t compound_messagest xt requestst it requestt response( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _mkdirs_smbprotocol� s(
& c C@ sf t r+ t | | d | d | d | d | �St rV t | | d | d | d | d | �St d � � d S( NRo Rm R& R' u. SMB library required (impacket or smbprotocol)( Rs R� Rr R� R ( R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt mkdirs� s
%%c C@ sW | d k r! t | | | � } n | t k r1 t St | � } | j | | | j � d S( N( Rd Rz R# R^ t putFileRb ( R_ R� R2 Ro Rm R& R' t fh_( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _put_str_impacket� s c C@ s� | d k r! t | | | � } n | t k r1 t S| j | � } z5 | j | | � } | j t j j j | � d � Wd | j
� Xd S( Ni ( Rd Rz R# R4 RR t writeRt Ru t stringutilst to_bytesR/ ( R_ R� R2 Ro Rm R& R' R3 RP ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _put_str_smbprotocol
s #c C@ sl t r. t | | | d | d | d | d | �St r\ t | | | d | d | d | d | �St d � � d S( u�
Wrapper around impacket.smbconnection.putFile() that allows a string to be
uploaded, without first writing it as a local file
Ro Rm R& R' u. SMB library required (impacket or smbprotocol)N( Rs R� Rr R� R ( R_ R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt put_str s
%c C@ s� | d k r! t | | | � } n | t k r1 t St | d � rW | j | | | � d St j j j | d � � } | j | | | j � Wd QXd S( u;
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)
u readNu rb(
Rd Rz R# t hasattrR� Rt Ru t filest fopent read( t
local_pathR� R2 Ro Rm R& R' R� ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _put_file_impacket* s i c
C@ s� | d k r! t | | | � } n | t k r1 t S| j | � } | j | | � } t j j j | d � �h }
zP d } xC t r� |
j
| � } | s� Pn | j | | � | t | � 7} qy WWd | j
t � XWd QXd S( Nu rbi ( Rd Rz R# R4 RR Rt Ru R� R� R" R� R� t lenR/ (
R� R� R2 Ro Rm R& R' t
chunk_sizeR3 RP R� t positiont chunk( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _put_file_smbprotocolB s c C@ sl t r. t | | | d | d | d | d | �St r\ t | | | d | d | d | d | �St d � � d S( u;
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)
Ro Rm R& R' u. SMB library required (impacket or smbprotocol)N( Rs R� Rr R� R ( R� R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt put_fileY s c C@ sE | d k r! t | | | � } n | t k r1 t S| j | | � d S( N( Rd Rz R# t
deleteFile( R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _delete_file_impacketq s
c C@ s5 | d k r! t | | | � } n | t k r1 t S| j | � } t | | � } | j t j t j t j
Bt j t
j t
j Bt j t j t j Bd t �| j t d t �g } | j j g | D] } | d ^ q� | j j | j d t �}
g } x; t |
� D]- \ } }
| | d |
� } | j | � q Wd S( NR� i t relatedi ( Rd Rz R# R4 R R? R R@ R
RA t DELETER RC R RD RE R t FILE_OPENR
RG t FILE_DELETE_ON_CLOSER/ R+ R� R$ R� R� R" R� t append( R� R2 Ro Rm R& R' R3 RP t delete_msgsR� R� t responsesR� R� R� ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _delete_file_smbprotocoly s0
# c C@ sf t r+ t | | d | d | d | d | �St rV t | | d | d | d | d | �St d � � d S( NRo Rm R& R' u. SMB library required (impacket or smbprotocol)( Rs R� Rr R� R ( R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt delete_file� s
%%c C@ sE | d k r! t | | | � } n | t k r1 t S| j | | � d S( N( Rd Rz R# t deleteDirectory( R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _delete_directory_impacket� s
c C@ s* | d k r! t | | | � } n | t k r1 t St j d | | � | j | � } t | | � } | j t j t
j t j
d t j t j t j Bd t �| j t d t �g } | j j g | D] } | d ^ q� d | j j d | j d t �}
x. t |
� D] \ } } | | d | � }
qWd S( Nu3 _delete_directory_smbprotocol - share: %s, path: %si R� t sidt tidR� i ( Rd Rz R# Rx t debugR4 R R? R R@ R R� R RS R R� R
RU R� R/ R+ R� R$ R� R� R" R� ( R� R2 Ro Rm R& R' R3 RW R� R� t delete_reqsR� R� R� ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt _delete_directory_smbprotocol� s.
& c C@ sf t r+ t | | d | d | d | d | �St rV t | | d | d | d | d | �St d � � d S( NRo Rm R& R' u. SMB library required (impacket or smbprotocol)( Rs R� Rr R� R ( R� R2 Ro Rm R& R' ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt delete_directory� s ""i (O Rc t
__future__R R R R R t salt.utils.filesRt t salt.utils.stringutilst salt.utils.versionst loggingt getLoggerRY Rx t salt.exceptionsR t impacket.smbconnectionRi R R~ t
impacket.smb3R R" Rr t ImportErrorR# t smbprotocol.connectionR t smbprotocol.sessionR t smbprotocol.treeR t smbprotocol.openR R R
R R R
R R R t smbprotocol.create_contextsR R R t smbprotocol.security_descriptorR R R R R R t setLevelt WARNINGRs t objectR R^ Rd Rp Rq Rz R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� ( ( ( s2 /usr/lib/python2.7/site-packages/salt/utils/smb.pyt <module> sb
@ .
V
Mini Shell