Update output,IO; add output support to WMI

1 files changed, 11 insertions, 12 deletions

diff --git a/cmd/root.go b/cmd/root.go
index 3d696aa..913a44a 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -54,15 +54,17 @@ var (
 				log = log.Level(zerolog.DebugLevel)
 			}
 
-			if outputMethod == "smb" {
-				if exec.Output.RemotePath == "" {
-					exec.Output.RemotePath = util.RandomWindowsTempFile()
-				}
-				exec.Output.Provider = &smb.OutputFileFetcher{
-					Client:           &smbClient,
-					Share:            `C$`,
-					File:             exec.Output.RemotePath,
-					DeleteOutputFile: exec.Output.NoDelete, // TEMP
+			if outputPath != "" {
+				if outputMethod == "smb" {
+					if exec.Output.RemotePath == "" {
+						exec.Output.RemotePath = util.RandomWindowsTempFile()
+					}
+					exec.Output.Provider = &smb.OutputFileFetcher{
+						Client:           &smbClient,
+						Share:            `C$`,
+						File:             exec.Output.RemotePath,
+						DeleteOutputFile: !exec.Output.NoDelete,
+					}
 				}
 			}
 		},
@@ -81,13 +83,10 @@ func init() {
 
 		dcomCmdInit()
 		rootCmd.AddCommand(dcomCmd)
-
 		wmiCmdInit()
 		rootCmd.AddCommand(wmiCmd)
-
 		scmrCmdInit()
 		rootCmd.AddCommand(scmrCmd)
-
 		tschCmdInit()
 		rootCmd.AddCommand(tschCmd)
 	}