Add a bunch of DCE related options to TSCH module

1 files changed, 27 insertions, 29 deletions

diff --git a/internal/exec/tsch/module.go b/internal/exec/tsch/module.go
index dbd9ade..8e31840 100644
--- a/internal/exec/tsch/module.go
+++ b/internal/exec/tsch/module.go
@@ -1,48 +1,46 @@
 package tschexec
 
 import (
-	"github.com/FalconOpsLLC/goexec/internal/client/dcerpc"
-	"github.com/RedTeamPentesting/adauth"
-	"github.com/oiweiwei/go-msrpc/msrpc/tsch/itaskschedulerservice/v1"
-	"github.com/rs/zerolog"
-	"time"
+  "github.com/oiweiwei/go-msrpc/dcerpc"
+  "github.com/oiweiwei/go-msrpc/msrpc/tsch/itaskschedulerservice/v1"
+  "time"
 )
 
 type Module struct {
-	creds  *adauth.Credential
-	target *adauth.Target
-
-	log  zerolog.Logger
-	dce  *dcerpc.DCEClient
-	tsch itaskschedulerservice.TaskSchedulerServiceClient
+  // dce holds the working DCE connection interface
+  dce dcerpc.Conn
+  // tsch holds the ITaskSchedulerService client
+  tsch itaskschedulerservice.TaskSchedulerServiceClient
+  // createdTasks holds any tasks that are created - for cleanup
+  createdTasks []string
 }
 
 type MethodRegisterConfig struct {
-	NoDelete    bool
-	CallDelete  bool
-	TaskName    string
-	TaskPath    string
-	StartDelay  time.Duration
-	StopDelay   time.Duration
-	DeleteDelay time.Duration
+  NoDelete   bool
+  CallDelete bool
+  //TaskName    string
+  TaskPath    string
+  StartDelay  time.Duration
+  StopDelay   time.Duration
+  DeleteDelay time.Duration
 }
 
 type MethodDemandConfig struct {
-	NoDelete    bool
-	CallDelete  bool
-	TaskName    string
-	TaskPath    string
-	StopDelay   time.Duration
-	DeleteDelay time.Duration
+  NoDelete    bool
+  CallDelete  bool
+  TaskName    string
+  TaskPath    string
+  StopDelay   time.Duration
+  DeleteDelay time.Duration
 }
 
 type MethodDeleteConfig struct {
-	TaskPath string
+  TaskPath string
 }
 
 const (
-	MethodRegister string = "register"
-	MethodDemand   string = "demand"
-	MethodDelete   string = "delete"
-	MethodChange   string = "update"
+  MethodRegister string = "register"
+  MethodDemand   string = "demand"
+  MethodDelete   string = "delete"
+  MethodChange   string = "update"
 )