fix: remove temp dir if installer fails to download

orbit/changes/25373-tmp-dir-fix

@@ -0,0 +1,2 @@
+- Fixed a bug where fleetd was not properly cleaning up temporary directories during software
+  installation.

orbit/pkg/installer/installer.go

@@ -235,6 +235,18 @@ func (r *Runner) installSoftware(ctx context.Context, installID string) (*fleet.
 		return payload, fmt.Errorf("creating temporary directory: %w", err)
 	}
 
+	// remove tmp directory and installer
+	defer func() {
+		removeAllFn := r.removeAllFn
+		if removeAllFn == nil {
+			removeAllFn = os.RemoveAll
+		}
+		err := removeAllFn(tmpDir)
+		if err != nil {
+			log.Err(err)
+		}
+	}()
+
 	var installerPath string
 	if installer.SoftwareInstallerURL != nil && installer.SoftwareInstallerURL.URL != "" {
 		log.Debug().Str("install_id", installID).Msgf("about to download software installer from URL")
@@ -255,18 +267,6 @@ func (r *Runner) installSoftware(ctx context.Context, installID string) (*fleet.
 		}
 	}
 
-	// remove tmp directory and installer
-	defer func() {
-		removeAllFn := r.removeAllFn
-		if removeAllFn == nil {
-			removeAllFn = os.RemoveAll
-		}
-		err := removeAllFn(tmpDir)
-		if err != nil {
-			log.Err(err)
-		}
-	}()
-
 	scriptExtension := ".sh"
 	if runtime.GOOS == "windows" {
 		scriptExtension = ".ps1"

orbit/pkg/installer/installer_test.go

@@ -421,6 +421,20 @@ func TestInstallerRun(t *testing.T) {
 		assert.Equal(t, 2, numPostInstallMatches)
 	})
 
+	t.Run("failed installer download", func(t *testing.T) {
+		resetAll()
+
+		oc.downloadInstallerFn = func(installerID uint, downloadDir string) (string, error) {
+			return "", errors.New("failed to download installer")
+		}
+
+		err := r.run(context.Background(), &config)
+		require.Error(t, err)
+
+		require.True(t, removeAllFnCalled)
+		require.True(t, tmpDirFnCalled)
+		require.Equal(t, tmpDir, removedDir)
+	})
 	t.Run("failed results upload", func(t *testing.T) {
 		var retries int
 		// set a shorter interval to speed up tests
@@ -669,7 +683,6 @@ func TestInstallerRunWithInstallerFromURL(t *testing.T) {
 		assert.True(t, getInstallerDetailsFnCalled)
 		assert.Equal(t, installDetails.ExecutionID, installIdRequested)
 	})
-
 }
 
 func TestScriptsDisabled(t *testing.T) {