fix(oci): Add missing writer file closes in directory handler (#628)

Reviewed-by: Alexander Jung <[email protected]> Approved-by: Alexander Jung <[email protected]>
unikraft · Jul 13, 2023 · 54d9504 · 54d9504
2 parents ececfc4 + 87726ee
commit 54d9504
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/oci/handler/directory.go b/oci/handler/directory.go
@@ -307,6 +307,7 @@ func (handle *DirectoryHandler) FetchImage(ctx context.Context, fullref, platfor
 	if err != nil {
 		return err
 	}
+	defer writer.Close()
 
 	if _, err := writer.Write(manifest); err != nil {
 		return err
@@ -343,6 +344,7 @@ func (handle *DirectoryHandler) FetchImage(ctx context.Context, fullref, platfor
 	if err != nil {
 		return err
 	}
+	defer writer.Close()
 
 	// Write the config
 	if _, err = writer.Write(config); err != nil {
@@ -382,11 +384,13 @@ func (handle *DirectoryHandler) FetchImage(ctx context.Context, fullref, platfor
 		if err != nil {
 			return err
 		}
+		defer writer.Close()
 
 		reader, err := layer.Compressed()
 		if err != nil {
 			return err
 		}
+		defer reader.Close()
 
 		if _, err = io.Copy(writer, reader); err != nil {
 			return err