1 files changed, 18 insertions, 1 deletions

diff --git a/firmware/target/arm/imx233/partitions-imx233.h b/firmware/target/arm/imx233/partitions-imx233.h
index 4490aad77f..b7ed251dee 100644
--- a/firmware/target/arm/imx233/partitions-imx233.h
+++ b/firmware/target/arm/imx233/partitions-imx233.h
@@ -24,10 +24,27 @@
 #include "system.h"
 #include "storage.h"
 
+#ifndef IMX233_PARTITIONS
+#error You must define IMX233_PARTITIONS
+#endif
+
+enum imx233_part_t
+{
+    IMX233_PART_USER,
+#if (IMX233_PARTITIONS & IMX233_FREESCALE)
+    IMX233_PART_BOOT,
+#endif
+#if (IMX233_PARTITIONS & IMX233_CREATIVE)
+    IMX233_PART_CFS,
+    IMX233_PART_MINIFS,
+#endif
+};
+
 /* Enable/Disable window computations for internal storage following the
  * Freescale convention */
 void imx233_partitions_enable_window(bool enable);
 bool imx233_partitions_is_window_enabled(void);
-int imx233_partitions_compute_window(uint8_t mbr[512], unsigned *start, unsigned *end);
+int imx233_partitions_compute_window(IF_MD(int drive,) enum imx233_part_t part,
+    unsigned *start, unsigned *end);
 
 #endif /* __PARTITIONS_IMX233__ */
\ No newline at end of file