prevent missuse of cleanup

benchkram · Mar 31, 2023 · 667167c · 667167c
1 parent efda6ff
commit 667167c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bobtask/clean.go b/bobtask/clean.go
@@ -21,6 +21,11 @@ func (t *Task) Clean(invalidFiles map[string][]target.Reason, verbose ...bool) e
 	if t.target != nil {
 		boblog.Log.V(5).Info(fmt.Sprintf("Cleaning targets for task %s", t.Name()))
 
+		homeDir, err := os.UserHomeDir()
+		if err != nil {
+			return err
+		}
+
 		if vb {
 			fmt.Printf("Cleaning targets of task %s \n", t.name)
 		}
@@ -36,6 +41,10 @@ func (t *Task) Clean(invalidFiles map[string][]target.Reason, verbose ...bool) e
 						fmt.Printf(" %s ", filename)
 					}
 
+					if filename == "/" || filename == homeDir {
+						return fmt.Errorf("Cleanup of %s is not allowed", filename)
+					}
+
 					err := os.RemoveAll(filename)
 					if err != nil {
 						if vb {