asrockrack: Rewind file read from beginning in uploadFirmware

bmc-toolbox · Nov 16, 2023 · 32380ce · 32380ce
1 parent cb575b3
commit 32380ce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/providers/asrockrack/helpers.go b/providers/asrockrack/helpers.go
@@ -226,6 +226,11 @@ func (a *ASRockRack) uploadFirmware(ctx context.Context, endpoint string, fwRead
 	fieldName, fileName := "fwimage", "image"
 	contentLength := multipartSize(fieldName, fileName) + fileSize
 
+	// Before reading the file, rewind to the beginning
+	if file, ok := fwReader.(*os.File); ok {
+		file.Seek(0, 0)
+	}
+
 	// setup pipe
 	pipeReader, pipeWriter := io.Pipe()
 	defer pipeReader.Close()