lk2nd: device: 2nd: Backup the partition entry before split

Author: me-cafebabe

lk2nd/device/2nd/partition.c

@@ -13,7 +13,7 @@
 static void partition_split_mmc(const char *base_name, const char *name,
 				uint32_t num_blocks, bool end)
 {
-	struct partition_entry *base, *split;
+	struct partition_entry *backup, *base, *split;
 	int index = partition_get_index(base_name);
 
 	if (index == INVALID_PTN) {
@@ -29,6 +29,15 @@ static void partition_split_mmc(const char *base_name, const char *name,
 		return;
 	}
 
+	backup = partition_allocate();
+	if (backup) {
+		memcpy(backup, base, sizeof(*backup));
+		snprintf((char*)backup->name, sizeof(backup->name), "real_%s", base_name);
+	} else {
+		dprintf(CRITICAL, "Too many partitions, cannot backup %s partition entry\n",
+			base_name);
+	}
+
 	split = partition_allocate();
 	if (split) {
 		memcpy(split, base, sizeof(*split));
@@ -67,6 +76,16 @@ static void partition_split_flash(struct ptable *ptable, const char *base_name,
 		return;
 	}
 
+	if (ptable_size(ptable) < MAX_PTABLE_PARTS) {
+		char backup_name[MAX_PTENTRY_NAME];
+		snprintf(backup_name, sizeof(backup_name), "real_%s", base_name);
+		ptable_add(ptable, backup_name, base->start, base->length,
+			    base->flags, base->type, base->perm);
+	} else {
+		dprintf(CRITICAL, "Too many partitions, cannot backup %s partition entry\n",
+			base_name);
+	}
+
 	if (ptable_size(ptable) < MAX_PTABLE_PARTS) {
 		unsigned start = base->start;
 		if (end)