@@ -95,6 +95,7 @@ def on_admin_login(self, context, connection):
9595 else :
9696 self .logger .fail (f"Failed to execute command: { e } " )
9797
98+
9899class TSCH_EXEC :
99100 def __init__ (self , target , share_name , username , password , domain , user , cmd , file , task , location , doKerberos = False , aesKey = None , remoteHost = None , kdcHost = None , hashes = None , logger = None , tries = None , share = None ):
100101 self .__target = target
@@ -156,7 +157,7 @@ def deleteartifact(self):
156157 self .logger .display (f"Deleting task \\ { tmpName } " )
157158 tsch .hSchRpcDelete (dce , f"\\ { tmpName } " )
158159 dce .disconnect ()
159-
160+
160161 def execute (self , command , output = False ):
161162 self .__retOutput = output
162163 self .execute_handler (command )
@@ -245,7 +246,6 @@ def execute_handler(self, command, fileless=False):
245246 xml = self .gen_xml (command , fileless )
246247
247248 self .logger .info (f"Task XML: { xml } " )
248- taskCreated = False
249249 self .logger .info (f"Creating task \\ { tmpName } " )
250250 try :
251251 # windows server 2003 has no MSRPC_UUID_TSCHS, if it bind, it will return abstract_syntax_not_supported
@@ -270,10 +270,10 @@ def execute_handler(self, command, fileless=False):
270270 self .logger .fail (f"Schtask_as: Create schedule task failed: { e } " )
271271 tsch .hSchRpcDelete (dce , f"\\ { tmpName } " )
272272 return
273- else :
274- taskCreated = True
273+
275274 self .logger .info (f"Running task \\ { tmpName } " )
276- tsch .hSchRpcRun (dce , f"\\ { tmpName } " )
275+ tsch .hSchRpcRun (dce , f"\\ { tmpName } " )
276+
277277 done = False
278278 while not done :
279279 self .logger .debug (f"Calling SchRpcGetLastRunInfo for \\ { tmpName } " )
@@ -285,10 +285,6 @@ def execute_handler(self, command, fileless=False):
285285
286286 self .logger .info (f"Deleting task \\ { tmpName } " )
287287 tsch .hSchRpcDelete (dce , f"\\ { tmpName } " )
288- taskCreated = False
289-
290- if taskCreated is True :
291- tsch .hSchRpcDelete (dce , f"\\ { tmpName } " )
292288
293289 if self .__retOutput :
294290 if fileless :
0 commit comments