From 1fca7b9a91f6621b848d474b8edb94335b693500 Mon Sep 17 00:00:00 2001
From: Florent Kermarrec <florent@enjoy-digital.fr>
Date: Wed, 28 Oct 2020 18:59:36 +0100
Subject: [PATCH] software/liblitesata/sata_read: handle errors.

---
 litex/soc/software/liblitesata/sata.c | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

diff --git a/litex/soc/software/liblitesata/sata.c b/litex/soc/software/liblitesata/sata.c
index cbc2cd3a8..cef80e8b9 100644
--- a/litex/soc/software/liblitesata/sata.c
+++ b/litex/soc/software/liblitesata/sata.c
@@ -26,11 +26,19 @@ int sata_init(void) {
 void sata_read(uint32_t block, uint32_t count, uint8_t* buf)
 {
 	uint32_t i;
+
+	/* Read blocks */
 	for (i=0; i<count; i++) {
-		sata_block2mem_base_write(((uint32_t) buf) + i*512);
-		sata_block2mem_sector_write(block + i);
-		sata_block2mem_start_write(1);
-		while ((sata_block2mem_done_read() & 0x1) == 0);
+		uint8_t done = 0;
+		while (done == 0) {
+			sata_block2mem_base_write((uint64_t) buf);
+			sata_block2mem_sector_write(block + i);
+			sata_block2mem_start_write(1);
+			while ((sata_block2mem_done_read() & 0x1) == 0);
+			done = ((sata_block2mem_error_read() & 0x1) == 0);
+			busy_wait_us(10);
+		}
+		buf += 512;
 	}
 
 #ifndef CONFIG_CPU_HAS_DMA_BUS