o
j| @ s d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ d dlmZ d d lmZ d d
lmZmZmZmZmZmZmZmZ d d l m Z e dddZ G dd dZ dS ) )TupleN)logger)
SMBConnection)DCERPC5Connection)tsch)EPM)
SMBDCEFactory)NULL)RPC_C_AUTHN_LEVEL_NONERPC_C_AUTHN_LEVEL_CONNECTRPC_C_AUTHN_LEVEL_CALLRPC_C_AUTHN_LEVEL_PKT RPC_C_AUTHN_LEVEL_PKT_INTEGRITY RPC_C_AUTHN_LEVEL_PKT_PRIVACYDCERPCExceptionRPC_C_AUTHN_GSS_NEGOTIATE)asynccontextmanagerTFc C sR t j| |||dI d H \}}|r|z
|V W | I d H d S | I d H w )N
auth_levelopen
perform_dummy)TSCHRPCfrom_smbconnectionclose)
connectionr r r instanceerr r /root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/aiosmb/dcerpc/v5/interfaces/tschmgr.pytschrpc_from_smb s "r c @ s e Zd Zdd Zdd Zdd Zdd Zed-d ed
e de fddZ
ed-d e d
e de fddZ d e
jee
jfddZdd Zdd Zdd Zd.d d Zd d Zd d! Zd.d"d#Zd/d%d&Zd'd( Zd)d* Zd+d, Z d S )0r c C s d| _ tj| _d | _d S )Nz\atsvc)service_pipenamer MSRPC_UUID_TSCHS service_uuiddceselfr r r __init__# s
zTSCHRPC.__init__c s | S Nr r$ r r r
__aenter__( s zTSCHRPC.__aenter__c s | I d H dS NTN)r )r% exc_typeexc tracebackr r r __aexit__+ s zTSCHRPC.__aexit__c
sd z| j rz | j I d H W W d S Y W d S W dS ty1 } z
d |fW Y d }~S d }~ww r) )r#
disconnect Exception)r% er r r r / s z
TSCHRPC.closeNTFr r r c
s z=t }| |_|j| |d u r|jt |j I d H \}}|d ur(||j|jI d H \}}|d ur:||d fW S tyT } z
d|fW Y d }~S d }~ww )NF)r r# set_auth_levelr connectbindr" r0 )r r r r service_r r1 r r r from_rpcconnection< s$
zTSCHRPC.from_rpcconnectionc
s z)|du rt }t| t jd}tj| |||dI dH \}}|dur&||dfW S ty@ } z
d|fW Y d}~S d}~ww )z
Creates the connection to the service using an established SMBConnection.
This connection will use the given SMBConnection as transport layer.
N)filenamer )r r r r r7 get_dce_rpcr0 )r r r r rpctransportr5 r r1 r r r r R s
zTSCHRPC.from_smbconnectionc sx |d u r t d }|d dkrd| }|d u r t}t| j|||||I d H \}}|d ur4d |fS |d }||fS )N r \ pActualPath)osurandomhexr r hSchRpcRegisterTaskr# )r% template task_nameflagssddl
logon_typeresr r r r
register_taskd s zTSCHRPC.register_taskc
s |d dkr d| }t | j|I d H \}}|d ur d |fS |d d |d d |d d |d d |d d |d d |d d
|d d |d d }||fS )
Nr r< pLastRuntimewYearwMonth
wDayOfWeekwDaywHourwMinutewSecond
wMilliseconds) YearMonth DayOfWeekDayHourMinuteSecond MilisecondsLastReturnCode)r hSchRpcGetLastRunInfor# )r% rC rG r tr r r get_task_last_run_infou s"
z TSCHRPC.get_task_last_run_infoc sF |d dkr d| }t | j|I dH \}}|dur d|fS d|fS )z
Deletes a task
r r< NFT)r
hSchRpcDeleter# r% rC r6 r r r r delete_task s zTSCHRPC.delete_taskc s6 |d dkr d| }t | j|I dH \}}d|fS )z
Starts a task
r r< NT)r
hSchRpcRunr# r_ r r r run_task s
zTSCHRPC.run_taskr< c
C z-t | j|I dH \}}|durd|fV W dS |d D ]
}|d dddfV q W dS tyF } z d|fV W Y d}~dS d}~ww )z5
Lists all available tasks on the remote machine
NpNamesData )r hSchRpcEnumTasksr# replacer0 r% pathrespr namer1 r r r
list_tasks
zTSCHRPC.list_tasksc J |d dkr d| }t | j|I dH \}}|dur d|fS |d dfS )z
Returns the task XML
r r< NpXml)r hSchRpcRetrieveTaskr# r% rC rl r r r r get_task zTSCHRPC.get_taskc rp )z5
Returns the selected task's security descriptor
r r< NrE )r hSchRpcGetSecurityr# rs r r r get_task_sd ru zTSCHRPC.get_task_sdc
C rc )z7
Lists all available folders on the remote machine
Nrd re rf rg )r hSchRpcEnumFoldersr# ri r0 rj r r r list_folders ro zTSCHRPC.list_folders
c s | |}| |I dH \}}|durd|fS | |I dH \}}|dur*d|fS |dk r5dtdfS | |I dH \}}|durGd|fS |d dkrNn
|d8 }tdI dH q+| |I dH \}}|durmd|fS dS )zy
Executes a list of shell commands by scheduling a new task and starts it.
At the end the new task will be deleted
NTr zmaxwait timeout!rR r* )gen_xmlrH rb r0 r] asynciosleepr` )r% commandsmaxwaitrB rC r r6 rG r r r run_commands s.
zTSCHRPC.run_commandsc C s d | |S )Na
2015-07-15T20:35:13.2757294
true
1
S-1-5-18
HighestAvailable
IgnoreNew
false
false
true
false
true
false
true
true
true
false
false
P3D
7
{}
)format gen_commands)r% r r r r r| s 'zTSCHRPC.gen_xmlc C s& d}|D ] }|d | |7 }q|S )Nrg zQ
cmd.exe
/C {}
)r
xml_escape)r% r retcommandr r r r . s zTSCHRPC.gen_commandsc s( dddddd d fdd |D S )
Nz&z"z'z>z<)&"'>A s z%TSCHRPC.xml_escape..)join)r% datar r r r 9 s zTSCHRPC.xml_escapeNTF)r< )rz ) __name__
__module__ __qualname__r& r( r. r staticmethodr boolr7 r r r TASK_CREATEr TASK_LOGON_NONErH r] r` rb rn rt rw ry r r| r r r r r r r " s*
%+ r r ) typingr r> r} r- aiosmbr aiosmb.connectionr aiosmb.dcerpc.v5.connectionr aiosmb.dcerpc.v5r 'aiosmb.dcerpc.v5.interfaces.endpointmgrr 0aiosmb.dcerpc.v5.common.connection.smbdcefactoryr aiosmb.dcerpc.v5.dtypesr aiosmb.dcerpc.v5.rpcrtr
r r r
r r r r
contextlibr r r r r r r s (