From 158e436c25224fe6faf47854795b8c89913ae0cf Mon Sep 17 00:00:00 2001
From: Dominik Ermel <dominik.ermel@nordicsemi.no>
Date: Mon, 4 Sep 2023 16:26:35 +0000
Subject: [PATCH] zephyr: Provide third image cases for direct image upload

The commit adds missing support for direct upload of third
image slots.

Signed-off-by: Dominik Ermel <dominik.ermel@nordicsemi.no>
---
 boot/zephyr/flash_map_extended.c | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/boot/zephyr/flash_map_extended.c b/boot/zephyr/flash_map_extended.c
index be90a8e1d..64e80085c 100644
--- a/boot/zephyr/flash_map_extended.c
+++ b/boot/zephyr/flash_map_extended.c
@@ -109,6 +109,14 @@ int flash_area_id_from_direct_image(int image_id)
 #if FIXED_PARTITION_EXISTS(slot3_partition)
     case 4:
         return FIXED_PARTITION_ID(slot3_partition);
+#endif
+#if FIXED_PARTITION_EXISTS(slot4_partition)
+    case 5:
+        return FIXED_PARTITION_ID(slot4_partition);
+#endif
+#if FIXED_PARTITION_EXISTS(slot5_partition)
+    case 6:
+        return FIXED_PARTITION_ID(slot5_partition);
 #endif
     }
     return -EINVAL;