altos/stm: Figure out available flash space based on chip id registers

[fw/altos] / src / stm / ao_interrupt.c

diff --git a/src/stm/ao_interrupt.c b/src/stm/ao_interrupt.c
index a423d8b13b1f76ad4ab2524602da956e8a74eadb..56cce0c02ff90af23fac42ee05af44783b59626c 100644 (file)
--- a/src/stm/ao_interrupt.c
+++ b/src/stm/ao_interrupt.c
@@ -18,6 +18,7 @@
 #include <ao.h>
 #include "stm32l.h"
 #include <string.h>
 #include <ao.h>
 #include "stm32l.h"
 #include <string.h>

+#include <ao_boot.h>

 
 extern void main(void);
 extern char __stack__;
 
 extern void main(void);
 extern char __stack__;


@@ -36,7 +37,51 @@ void stm_ignore_isr(void)
 {
 }
 
 {
 }
 

-void start(void) {
+const void *stm_interrupt_vector[];
+
+uint32_t
+stm_flash_size(void) {
+       uint16_t        dev_id = stm_dev_id();
+       uint16_t        kbytes = 0;
+
+       switch (dev_id) {
+       case 0x416:     /* cat 1 */
+               kbytes = stm_flash_size_medium.f_size;
+               break;
+       case 0x429:     /* cat 2 */
+               kbytes = stm_flash_size_medium.f_size & 0xff;
+               break;
+       case 0x427:     /* cat 3 */
+               kbytes = stm_flash_size_large.f_size;
+               break;
+       case 0x436:     /* cat 4 */
+               switch (stm_flash_size_large.f_size) {
+               case 0:
+                       kbytes = 256;
+                       break;
+               case 1:
+                       kbytes = 384;
+                       break;
+               }
+               break;
+       case 0x437:     /* cat 5 */
+               kbytes = stm_flash_size_large.f_size;
+               break;
+       }
+       return (uint32_t) kbytes * 1024;
+}
+
+void start(void)
+{
+#ifdef AO_BOOT_CHAIN
+       if (ao_boot_check_chain()) {
+#ifdef AO_BOOT_PIN
+               ao_boot_check_pin();
+#endif
+       }
+#endif
+       /* Set interrupt vector table offset */
+       stm_nvic.vto = (uint32_t) &stm_interrupt_vector;

        memcpy(&__data_start__, &__text_end__, &__data_end__ - &__data_start__);
        memset(&__bss_start__, '\0', &__bss_end__ - &__bss_start__);
        main();
        memcpy(&__data_start__, &__text_end__, &__data_end__ - &__data_start__);
        memset(&__bss_start__, '\0', &__bss_end__ - &__bss_start__);
        main();