From 7574b7370be083ff563fa8ad6d01d5ac776d7e4d Mon Sep 17 00:00:00 2001
From: Bryan McNulty <bryanmcnulty@protonmail.com>
Date: Sat, 8 Mar 2025 06:07:29 -0600
Subject: Add a bunch of DCE related options to TSCH module

---
 cmd/wmi.go | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'cmd/wmi.go')

diff --git a/cmd/wmi.go b/cmd/wmi.go
index 1808226..cba2473 100644
--- a/cmd/wmi.go
+++ b/cmd/wmi.go
@@ -59,7 +59,7 @@ References:
   https://learn.microsoft.com/en-us/windows/win32/wmisdk/wmi-classes
   `,
     Args: func(cmd *cobra.Command, args []string) (err error) {
-      if err = needsTarget(cmd, args); err == nil {
+      if err = needsTarget("cifs")(cmd, args); err == nil {
         if wmiArgMethod != "" && !methodRegex.MatchString(wmiArgMethod) {
           return fmt.Errorf("invalid CLASS.METHOD syntax: %s", wmiArgMethod)
         }
@@ -112,10 +112,11 @@ References:
 References:
   https://learn.microsoft.com/en-us/windows/win32/cimwin32prov/create-method-in-class-win32-process
 `,
-    Args: needsTarget,
+    Args: needsTarget("cifs"),
     Run: func(cmd *cobra.Command, args []string) {
-      module := wmiexec.Module{}
+      log = log.With().Str("module", "wmi").Logger()
 
+      module := wmiexec.Module{}
       connCfg := &exec.ConnectionConfig{}
       cleanCfg := &exec.CleanupConfig{}
 
@@ -123,6 +124,7 @@ References:
         ExecutableName:  executable,
         ExecutableArgs:  executableArgs,
         ExecutionMethod: wmiexec.MethodProcess,
+
         ExecutionMethodConfig: wmiexec.MethodProcessConfig{
           Command:          command,
           WorkingDirectory: workingDirectory,
@@ -130,10 +132,8 @@ References:
       }
       if err := module.Connect(log.WithContext(ctx), creds, target, connCfg); err != nil {
         log.Fatal().Err(err).Msg("Connection failed")
-
       } else if err := module.Exec(log.WithContext(ctx), execCfg); err != nil {
         log.Fatal().Err(err).Msg("Execution failed")
-
       } else if err := module.Cleanup(log.WithContext(ctx), cleanCfg); err != nil {
         log.Error().Err(err).Msg("Cleanup failed")
       }
-- 
cgit v1.2.3