From 662c41794b34f59ed357183d2bd57cace090ae9c Mon Sep 17 00:00:00 2001
From: wh1te909 <dcparsi@gmail.com>
Date: Tue, 9 Aug 2022 12:41:07 -0700
Subject: [PATCH] better cleanup

---
 agent/agent.go         | 1 +
 agent/agent_windows.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/agent/agent.go b/agent/agent.go
index e7ae4a9..84f1889 100644
--- a/agent/agent.go
+++ b/agent/agent.go
@@ -444,6 +444,7 @@ func (a *Agent) RunPythonCode(code string, timeout int, args []string) (string,
 		a.Logger.Debugln(err)
 		return "", err
 	}
+	defer os.Remove(tmpfn.Name())
 	if err := tmpfn.Close(); err != nil {
 		a.Logger.Debugln(err)
 		return "", err
diff --git a/agent/agent_windows.go b/agent/agent_windows.go
index 6b3ab87..e99af6b 100644
--- a/agent/agent_windows.go
+++ b/agent/agent_windows.go
@@ -577,6 +577,7 @@ func (a *Agent) UninstallCleanup() {
 	a.PatchMgmnt(false)
 	a.CleanupAgentUpdates()
 	CleanupSchedTasks()
+	os.RemoveAll(winTempDir)
 }
 
 func (a *Agent) AgentUpdate(url, inno, version string) {