target/riscv: active dm before get nextdm

when Debug Module is inactive, accesses to the nextdm may fail.
Specifically, nextdm might not return correct data.

Author: lz-bro

src/target/riscv/riscv-013.c

@@ -534,16 +534,50 @@ static int dm_write(struct target *target, uint32_t address, uint32_t value)
 	return dmi_write(target, riscv013_get_dmi_address(target, address), value);
 }
 
-static bool check_dbgbase_exists(struct target *target)
+static int wait_for_dm_activation(struct target *target, uint32_t dm_base_addr)
+{
+	uint32_t dmcontrol = 0;
+	const time_t start = time(NULL);
+
+	LOG_TARGET_DEBUG(target, "Waiting for the DM to become active.");
+	do {
+		if (dmi_read(target, &dmcontrol, DM_DMCONTROL + dm_base_addr) != ERROR_OK)
+			return ERROR_FAIL;
+
+		if (time(NULL) - start > riscv_get_command_timeout_sec()) {
+			LOG_TARGET_ERROR(target, "Debug Module (at address dbgbase=0x%" PRIx32 ") did not become active in %d s. "
+					"Increase the timeout with 'riscv set_command_timeout_sec'.",
+					dm_base_addr, riscv_get_command_timeout_sec());
+			return ERROR_TIMEOUT_REACHED;
+		}
+	} while (!get_field32(dmcontrol, DM_DMCONTROL_DMACTIVE));
+	LOG_TARGET_DEBUG(target, "DM has become active.");
+	return ERROR_OK;
+}
+
+static int check_dbgbase_exists(struct target *target)
 {
 	uint32_t next_dm = 0;
 	unsigned int count = 1;
+	uint32_t dmcontrol = 0;
 
 	LOG_TARGET_DEBUG(target, "Searching for DM with DMI base address (dbgbase) = 0x%x", target->dbgbase);
 	while (1) {
 		uint32_t current_dm = next_dm;
 		if (current_dm == target->dbgbase)
-			return true;
+			return ERROR_OK;
+
+		if (dmi_read(target, &dmcontrol, DM_DMCONTROL + current_dm) != ERROR_OK)
+			break;
+		if (!get_field32(dmcontrol, DM_DMCONTROL_DMACTIVE)) {
+			LOG_TARGET_DEBUG(target, "Activating the DM with DMI base address (dbgbase) = 0x%x", current_dm);
+			if (dmi_write(target, DM_DMCONTROL + current_dm, DM_DMCONTROL_DMACTIVE) != ERROR_OK)
+				break;
+
+			if (wait_for_dm_activation(target, current_dm) != ERROR_OK)
+				break;
+		}
+
 		if (dmi_read(target, &next_dm, DM_NEXTDM + current_dm) != ERROR_OK)
 			break;
 		LOG_TARGET_DEBUG(target, "dm @ 0x%x --> nextdm=0x%x", current_dm, next_dm);
@@ -558,7 +592,7 @@ static bool check_dbgbase_exists(struct target *target)
 			break;
 		}
 	}
-	return false;
+	return ERROR_FAIL;
 }
 
 static int dmstatus_read(struct target *target, uint32_t *dmstatus,
@@ -1852,21 +1886,9 @@ static int reset_dm(struct target *target)
 		if (result != ERROR_OK)
 			return result;
 
-		const time_t start = time(NULL);
-		LOG_TARGET_DEBUG(target, "Waiting for the DM to acknowledge reset.");
-		do {
-			result = dm_read(target, &dmcontrol, DM_DMCONTROL);
-			if (result != ERROR_OK)
-				return result;
-
-			if (time(NULL) - start > riscv_get_command_timeout_sec()) {
-				LOG_TARGET_ERROR(target, "DM didn't acknowledge reset in %d s. "
-						"Increase the timeout with 'riscv set_command_timeout_sec'.",
-						riscv_get_command_timeout_sec());
-				return ERROR_TIMEOUT_REACHED;
-			}
-		} while (get_field32(dmcontrol, DM_DMCONTROL_DMACTIVE));
-		LOG_TARGET_DEBUG(target, "DM reset initiated.");
+		result = wait_for_dm_activation(target, dm->base);
+		if (result != ERROR_OK)
+			return result;
 	}
 
 	LOG_TARGET_DEBUG(target, "Activating the DM.");
@@ -2043,7 +2065,7 @@ static int examine(struct target *target)
 			info->abits, RISCV013_DTMCS_ABITS_MIN);
 	}
 
-	if (!check_dbgbase_exists(target)) {
+	if (check_dbgbase_exists(target) != ERROR_OK) {
 		LOG_TARGET_ERROR(target, "Could not find debug module with DMI base address (dbgbase) = 0x%x", target->dbgbase);
 		return ERROR_FAIL;
 	}