Clean up temp file when layout.WriteBlob errors

google · Jan 6, 2022 · 5965a6c · 5965a6c
1 parent 4ec656f
commit 5965a6c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/v1/layout/write.go b/pkg/v1/layout/write.go
@@ -253,6 +253,10 @@ func (l Path) writeBlob(hash v1.Hash, size int64, r io.Reader, renamer func() (v
 	if err != nil {
 		return err
 	}
+	if renamer != nil {
+		// Delete temp file if an error is encountered before renaming
+		defer os.Remove(w.Name())
+	}
 	defer w.Close()
 
 	// Write to file and exit if not renaming