5b23081fc6787e6621015936690ac7f1a369db7a
[fw/altos] / src / stm32l0 / ao_arch_funcs.h
1 /*
2  * Copyright © 2012 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License along
15  * with this program; if not, write to the Free Software Foundation, Inc.,
16  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
17  */
18
19 #ifndef _AO_ARCH_FUNCS_H_
20 #define _AO_ARCH_FUNCS_H_
21
22 /* ao_spi_stm.c
23  */
24
25 /* PCLK is set to 16MHz (HCLK 32MHz, APB prescaler 2) */
26
27 #define AO_SPI_SPEED_8MHz       STM_SPI_CR1_BR_PCLK_2
28 #define AO_SPI_SPEED_4MHz       STM_SPI_CR1_BR_PCLK_4
29 #define AO_SPI_SPEED_2MHz       STM_SPI_CR1_BR_PCLK_8
30 #define AO_SPI_SPEED_1MHz       STM_SPI_CR1_BR_PCLK_16
31 #define AO_SPI_SPEED_500kHz     STM_SPI_CR1_BR_PCLK_32
32 #define AO_SPI_SPEED_250kHz     STM_SPI_CR1_BR_PCLK_64
33 #define AO_SPI_SPEED_125kHz     STM_SPI_CR1_BR_PCLK_128
34 #define AO_SPI_SPEED_62500Hz    STM_SPI_CR1_BR_PCLK_256
35
36 #define AO_SPI_SPEED_FAST       AO_SPI_SPEED_8MHz
37
38 /* Companion bus wants something no faster than 200kHz */
39
40 #define AO_SPI_SPEED_200kHz     AO_SPI_SPEED_125kHz
41
42 #define AO_SPI_CPOL_BIT         4
43 #define AO_SPI_CPHA_BIT         5
44
45 #define AO_SPI_CONFIG_1         0x00
46 #define AO_SPI_1_CONFIG_PA5_PA6_PA7     AO_SPI_CONFIG_1
47 #define AO_SPI_2_CONFIG_PB13_PB14_PB15  AO_SPI_CONFIG_1
48
49 #define AO_SPI_CONFIG_2         0x04
50 #define AO_SPI_1_CONFIG_PB3_PB4_PB5     AO_SPI_CONFIG_2
51 #define AO_SPI_2_CONFIG_PD1_PD3_PD4     AO_SPI_CONFIG_2
52
53 #define AO_SPI_CONFIG_3         0x08
54 #define AO_SPI_1_CONFIG_PE13_PE14_PE15  AO_SPI_CONFIG_3
55
56 #define AO_SPI_CONFIG_NONE      0x0c
57
58 #define AO_SPI_INDEX_MASK       0x01
59 #define AO_SPI_CONFIG_MASK      0x0c
60
61 #define AO_SPI_1_PA5_PA6_PA7    (STM_SPI_INDEX(1) | AO_SPI_1_CONFIG_PA5_PA6_PA7)
62 #define AO_SPI_1_PB3_PB4_PB5    (STM_SPI_INDEX(1) | AO_SPI_1_CONFIG_PB3_PB4_PB5)
63 #define AO_SPI_1_PE13_PE14_PE15 (STM_SPI_INDEX(1) | AO_SPI_1_CONFIG_PE13_PE14_PE15)
64
65 #define AO_SPI_2_PB13_PB14_PB15 (STM_SPI_INDEX(2) | AO_SPI_2_CONFIG_PB13_PB14_PB15)
66 #define AO_SPI_2_PD1_PD3_PD4    (STM_SPI_INDEX(2) | AO_SPI_2_CONFIG_PD1_PD3_PD4)
67
68 #define AO_SPI_INDEX(id)        ((id) & AO_SPI_INDEX_MASK)
69 #define AO_SPI_CONFIG(id)       ((id) & AO_SPI_CONFIG_MASK)
70 #define AO_SPI_PIN_CONFIG(id)   ((id) & (AO_SPI_INDEX_MASK | AO_SPI_CONFIG_MASK))
71 #define AO_SPI_CPOL(id)         ((uint32_t) (((id) >> AO_SPI_CPOL_BIT) & 1))
72 #define AO_SPI_CPHA(id)         ((uint32_t) (((id) >> AO_SPI_CPHA_BIT) & 1))
73
74 #define AO_SPI_MAKE_MODE(pol,pha)       (((pol) << AO_SPI_CPOL_BIT) | ((pha) << AO_SPI_CPHA_BIT))
75 #define AO_SPI_MODE_0           AO_SPI_MAKE_MODE(0,0)
76 #define AO_SPI_MODE_1           AO_SPI_MAKE_MODE(0,1)
77 #define AO_SPI_MODE_2           AO_SPI_MAKE_MODE(1,0)
78 #define AO_SPI_MODE_3           AO_SPI_MAKE_MODE(1,1)
79
80 uint8_t
81 ao_spi_try_get(uint8_t spi_index, uint32_t speed, uint8_t task_id);
82
83 void
84 ao_spi_get(uint8_t spi_index, uint32_t speed);
85
86 void
87 ao_spi_put(uint8_t spi_index);
88
89 void
90 ao_spi_send(const void *block, uint16_t len, uint8_t spi_index);
91
92 void
93 ao_spi_send_fixed(uint8_t value, uint16_t len, uint8_t spi_index);
94
95 void
96 ao_spi_send_sync(const void *block, uint16_t len, uint8_t spi_index);
97
98 void
99 ao_spi_start_bytes(uint8_t spi_index);
100
101 void
102 ao_spi_stop_bytes(uint8_t spi_index);
103
104 static inline void
105 ao_spi_send_byte(uint8_t byte, uint8_t spi_index)
106 {
107         struct stm_spi  *stm_spi;
108
109         switch (AO_SPI_INDEX(spi_index)) {
110         case 0:
111                 stm_spi = &stm_spi1;
112                 break;
113         case 1:
114                 stm_spi = &stm_spi2;
115                 break;
116         }
117
118         while (!(stm_spi->sr & (1 << STM_SPI_SR_TXE)))
119                 ;
120         stm_spi->dr = byte;
121         while (!(stm_spi->sr & (1 << STM_SPI_SR_RXNE)))
122                 ;
123         (void) stm_spi->dr;
124 }
125
126 static inline uint8_t
127 ao_spi_recv_byte(uint8_t spi_index)
128 {
129         struct stm_spi  *stm_spi;
130
131         switch (AO_SPI_INDEX(spi_index)) {
132         case 0:
133                 stm_spi = &stm_spi1;
134                 break;
135         case 1:
136                 stm_spi = &stm_spi2;
137                 break;
138         }
139
140         while (!(stm_spi->sr & (1 << STM_SPI_SR_TXE)))
141                 ;
142         stm_spi->dr = 0xff;
143         while (!(stm_spi->sr & (1 << STM_SPI_SR_RXNE)))
144                 ;
145         return stm_spi->dr;
146 }
147
148 void
149 ao_spi_recv(void *block, uint16_t len, uint8_t spi_index);
150
151 void
152 ao_spi_duplex(const void *out, void *in, uint16_t len, uint8_t spi_index);
153
154 extern uint16_t ao_spi_speed[STM_NUM_SPI];
155
156 void
157 ao_spi_init(void);
158
159 #define ao_spi_set_cs(reg,mask) ((reg)->bsrr = ((uint32_t) (mask)) << 16)
160 #define ao_spi_clr_cs(reg,mask) ((reg)->bsrr = (mask))
161
162 #define ao_spi_get_mask(reg,mask,bus, speed) do {               \
163                 ao_spi_get(bus, speed);                         \
164                 ao_spi_set_cs(reg,mask);                        \
165         } while (0)
166
167 static inline uint8_t
168 ao_spi_try_get_mask(struct stm_gpio *reg, uint16_t mask, uint8_t bus, uint32_t speed, uint8_t task_id)
169 {
170         if (!ao_spi_try_get(bus, speed, task_id))
171                 return 0;
172         ao_spi_set_cs(reg, mask);
173         return 1;
174 }
175
176 #define ao_spi_put_mask(reg,mask,bus) do {      \
177                 ao_spi_clr_cs(reg,mask);        \
178                 ao_spi_put(bus);                \
179         } while (0)
180
181 #define ao_spi_get_bit(reg,bit,bus,speed) ao_spi_get_mask(reg,(1<<bit),bus,speed)
182 #define ao_spi_put_bit(reg,bit,bus) ao_spi_put_mask(reg,(1<<bit),bus)
183
184 #define ao_enable_port(port) do {                                       \
185                 if ((port) == &stm_gpioa)                               \
186                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPAEN); \
187                 else if ((port) == &stm_gpiob)                          \
188                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPBEN); \
189                 else if ((port) == &stm_gpioc)                          \
190                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPCEN); \
191                 else if ((port) == &stm_gpiod)                          \
192                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPDEN); \
193                 else if ((port) == &stm_gpioe)                          \
194                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPEEN); \
195                 else if ((port) == &stm_gpioh)                          \
196                         stm_rcc.iopenr |= (1 << STM_RCC_IOPENR_IOPHEN); \
197         } while (0)
198
199 #define ao_disable_port(port) do {                                      \
200                 if ((port) == &stm_gpioa)                               \
201                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPAEN); \
202                 else if ((port) == &stm_gpiob)                          \
203                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPBEN); \
204                 else if ((port) == &stm_gpioc)                          \
205                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPCEN); \
206                 else if ((port) == &stm_gpiod)                          \
207                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPDEN); \
208                 else if ((port) == &stm_gpioe)                          \
209                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPEEN); \
210                 else if ((port) == &stm_gpioh)                          \
211                         stm_rcc.iopenr &= ~(1 << STM_RCC_IOPENR_IOPHEN); \
212         } while (0)
213
214
215 #define ao_gpio_set(port, bit, v) stm_gpio_set(port, bit, v)
216
217 #define ao_gpio_get(port, bit) stm_gpio_get(port, bit)
218
219 #define ao_gpio_set_bits(port, bits) stm_gpio_set_bits(port, bits)
220
221 #define ao_gpio_set_mask(port, bits, mask) stm_gpio_set_mask(port, bits, mask)
222
223 #define ao_gpio_clr_bits(port, bits) stm_gpio_clr_bits(port, bits);
224
225 #define ao_gpio_get_all(port) stm_gpio_get_all(port)
226
227 #define ao_enable_output(port,bit,v) do {                       \
228                 ao_enable_port(port);                           \
229                 ao_gpio_set(port, bit, v);                      \
230                 stm_moder_set(port, bit, STM_MODER_OUTPUT);\
231         } while (0)
232
233 #define ao_enable_output_mask(port,bits,mask) do {              \
234                 ao_enable_port(port);                           \
235                 ao_gpio_set_mask(port, bits, mask);             \
236                 ao_set_output_mask(port, mask);                 \
237         } while (0)
238
239 #define AO_OUTPUT_PUSH_PULL     STM_OTYPER_PUSH_PULL
240 #define AO_OUTPUT_OPEN_DRAIN    STM_OTYPER_OPEN_DRAIN
241
242 #define ao_gpio_set_output_mode(port,bit,mode) \
243         stm_otyper_set(port, pin, mode)
244
245 #define ao_gpio_set_mode(port,bit,mode) do {                            \
246                 if (mode == AO_EXTI_MODE_PULL_UP)                       \
247                         stm_pupdr_set(port, bit, STM_PUPDR_PULL_UP);    \
248                 else if (mode == AO_EXTI_MODE_PULL_DOWN)                \
249                         stm_pupdr_set(port, bit, STM_PUPDR_PULL_DOWN);  \
250                 else                                                    \
251                         stm_pupdr_set(port, bit, STM_PUPDR_NONE);       \
252         } while (0)
253
254 #define ao_gpio_set_mode_mask(port,mask,mode) do {                      \
255                 if (mode == AO_EXTI_MODE_PULL_UP)                       \
256                         stm_pupdr_set_mask(port, mask, STM_PUPDR_PULL_UP); \
257                 else if (mode == AO_EXTI_MODE_PULL_DOWN)                \
258                         stm_pupdr_set_mask(port, mask, STM_PUPDR_PULL_DOWN); \
259                 else                                                    \
260                         stm_pupdr_set_mask(port, mask, STM_PUPDR_NONE); \
261         } while (0)
262
263 #define ao_set_input(port, bit) do {                            \
264                 stm_moder_set(port, bit, STM_MODER_INPUT);      \
265         } while (0)
266
267 #define ao_set_output(port, bit, v) do {                        \
268                 ao_gpio_set(port, bit, v);                      \
269                 stm_moder_set(port, bit, STM_MODER_OUTPUT);     \
270         } while (0)
271
272 #define ao_set_output_mask(port, mask) do {                     \
273                 stm_moder_set_mask(port, mask, STM_MODER_OUTPUT);       \
274         } while (0)
275
276 #define ao_set_input_mask(port, mask) do {                              \
277                 stm_moder_set_mask(port, mask, STM_MODER_INPUT);        \
278         } while (0)
279
280 #define ao_enable_input(port,bit,mode) do {                             \
281                 ao_enable_port(port);                                   \
282                 ao_set_input(port, bit);                                \
283                 ao_gpio_set_mode(port, bit, mode);                      \
284         } while (0)
285
286 #define ao_enable_input_mask(port,mask,mode) do {       \
287                 ao_enable_port(port);                   \
288                 ao_gpio_set_mode_mask(port, mask, mode);        \
289                 ao_set_input_mask(port, mask);          \
290         } while (0)
291
292 #define _ao_enable_cs(port, bit) do {                           \
293                 stm_gpio_set((port), bit, 1);                   \
294                 stm_moder_set((port), bit, STM_MODER_OUTPUT);   \
295         } while (0)
296
297 #define ao_enable_cs(port,bit) do {                             \
298                 ao_enable_port(port);                           \
299                 _ao_enable_cs(port, bit);                       \
300         } while (0)
301
302 #define ao_spi_init_cs(port, mask) do {                         \
303                 ao_enable_port(port);                           \
304                 if ((mask) & 0x0001) _ao_enable_cs(port, 0);    \
305                 if ((mask) & 0x0002) _ao_enable_cs(port, 1);    \
306                 if ((mask) & 0x0004) _ao_enable_cs(port, 2);    \
307                 if ((mask) & 0x0008) _ao_enable_cs(port, 3);    \
308                 if ((mask) & 0x0010) _ao_enable_cs(port, 4);    \
309                 if ((mask) & 0x0020) _ao_enable_cs(port, 5);    \
310                 if ((mask) & 0x0040) _ao_enable_cs(port, 6);    \
311                 if ((mask) & 0x0080) _ao_enable_cs(port, 7);    \
312                 if ((mask) & 0x0100) _ao_enable_cs(port, 8);    \
313                 if ((mask) & 0x0200) _ao_enable_cs(port, 9);    \
314                 if ((mask) & 0x0400) _ao_enable_cs(port, 10);\
315                 if ((mask) & 0x0800) _ao_enable_cs(port, 11);\
316                 if ((mask) & 0x1000) _ao_enable_cs(port, 12);\
317                 if ((mask) & 0x2000) _ao_enable_cs(port, 13);\
318                 if ((mask) & 0x4000) _ao_enable_cs(port, 14);\
319                 if ((mask) & 0x8000) _ao_enable_cs(port, 15);\
320         } while (0)
321
322 /* ao_dma_stm.c
323  */
324
325 extern uint8_t ao_dma_done[STM_NUM_DMA];
326
327 void
328 ao_dma_set_transfer(uint8_t             index,
329                     volatile void       *peripheral,
330                     void                *memory,
331                     uint16_t            count,
332                     uint32_t            ccr);
333
334 void
335 ao_dma_set_isr(uint8_t index, void (*isr)(int index));
336
337 void
338 ao_dma_start(uint8_t index);
339
340 void
341 ao_dma_done_transfer(uint8_t index);
342
343 void
344 ao_dma_alloc(uint8_t index);
345
346 void
347 ao_dma_init(void);
348
349 /* ao_i2c_stm.c */
350
351 void
352 ao_i2c_get(uint8_t i2c_index);
353
354 uint8_t
355 ao_i2c_start(uint8_t i2c_index, uint16_t address);
356
357 void
358 ao_i2c_put(uint8_t i2c_index);
359
360 uint8_t
361 ao_i2c_send(void *block, uint16_t len, uint8_t i2c_index, uint8_t stop);
362
363 uint8_t
364 ao_i2c_recv(void *block, uint16_t len, uint8_t i2c_index, uint8_t stop);
365
366 void
367 ao_i2c_init(void);
368
369 #if USE_SERIAL_1_SW_FLOW || USE_SERIAL_2_SW_FLOW || USE_SERIAL_3_SW_FLOW
370 #define HAS_SERIAL_SW_FLOW 1
371 #else
372 #define HAS_SERIAL_SW_FLOW 0
373 #endif
374
375 #if USE_SERIAL_1_FLOW && !USE_SERIAL_1_SW_FLOW || USE_SERIAL_2_FLOW && !USE_SERIAL_2_SW_FLOW || USE_SERIAL_3_FLOW && !USE_SERIAL_3_SW_FLOW
376 #define HAS_SERIAL_HW_FLOW 1
377 #else
378 #define HAS_SERIAL_HW_FLOW 0
379 #endif
380
381 /* ao_serial_stm.c */
382 struct ao_stm_usart {
383         struct ao_fifo          rx_fifo;
384         struct ao_fifo          tx_fifo;
385         struct stm_usart        *reg;
386         uint8_t                 tx_running;
387         uint8_t                 draining;
388 #if HAS_SERIAL_SW_FLOW
389         /* RTS - 0 if we have FIFO space, 1 if not
390          * CTS - 0 if we can send, 0 if not
391          */
392         struct stm_gpio         *gpio_rts;
393         struct stm_gpio         *gpio_cts;
394         uint8_t                 pin_rts;
395         uint8_t                 pin_cts;
396         uint8_t                 rts;
397 #endif
398 };
399
400 void
401 ao_debug_out(char c);
402
403 #if HAS_SERIAL_1
404 extern struct ao_stm_usart      ao_stm_usart1;
405 #endif
406
407 #if HAS_SERIAL_2
408 extern struct ao_stm_usart      ao_stm_usart2;
409 #endif
410
411 #if HAS_SERIAL_3
412 extern struct ao_stm_usart      ao_stm_usart3;
413 #endif
414
415 #define ARM_PUSH32(stack, val)  (*(--(stack)) = (val))
416
417 typedef uint32_t        ao_arch_irq_t;
418
419 static inline void
420 ao_arch_block_interrupts(void) {
421 #ifdef AO_NONMASK_INTERRUPTS
422         asm("msr basepri,%0" : : "r" (AO_STM_NVIC_BASEPRI_MASK));
423 #else
424         asm("cpsid i");
425 #endif
426 }
427
428 static inline void
429 ao_arch_release_interrupts(void) {
430 #ifdef AO_NONMASK_INTERRUPTS
431         asm("msr basepri,%0" : : "r" (0x0));
432 #else
433         asm("cpsie i");
434 #endif
435 }
436
437 static inline uint32_t
438 ao_arch_irqsave(void) {
439         uint32_t        val;
440 #ifdef AO_NONMASK_INTERRUPTS
441         asm("mrs %0,basepri" : "=r" (val));
442 #else
443         asm("mrs %0,primask" : "=r" (val));
444 #endif
445         ao_arch_block_interrupts();
446         return val;
447 }
448
449 static inline void
450 ao_arch_irqrestore(uint32_t basepri) {
451 #ifdef AO_NONMASK_INTERRUPTS
452         asm("msr basepri,%0" : : "r" (basepri));
453 #else
454         asm("msr primask,%0" : : "r" (basepri));
455 #endif
456 }
457
458 static inline void
459 ao_arch_memory_barrier(void) {
460         asm volatile("" ::: "memory");
461 }
462
463 static inline void
464 ao_arch_irq_check(void) {
465 #ifdef AO_NONMASK_INTERRUPTS
466         uint32_t        basepri;
467         asm("mrs %0,basepri" : "=r" (basepri));
468         if (basepri == 0)
469                 ao_panic(AO_PANIC_IRQ);
470 #else
471         uint32_t        primask;
472         asm("mrs %0,primask" : "=r" (primask));
473         if ((primask & 1) == 0)
474                 ao_panic(AO_PANIC_IRQ);
475 #endif
476 }
477
478 #if HAS_TASK
479 static inline void
480 ao_arch_init_stack(struct ao_task *task, void *start)
481 {
482         uint32_t        *sp = &task->stack32[AO_STACK_SIZE >> 2];
483         uint32_t        a = (uint32_t) start;
484         int             i;
485
486         /* Return address (goes into LR) */
487         ARM_PUSH32(sp, a);
488
489         /* Clear register values r0-r7 */
490         i = 8;
491         while (i--)
492                 ARM_PUSH32(sp, 0);
493
494         /* APSR */
495         ARM_PUSH32(sp, 0);
496
497         /* PRIMASK with interrupts enabled */
498         ARM_PUSH32(sp, 0);
499
500         task->sp32 = sp;
501 }
502
503 static inline void ao_arch_save_regs(void) {
504         /* Save general registers */
505         asm("push {r0-r7,lr}\n");
506
507         /* Save APSR */
508         asm("mrs r0,apsr");
509         asm("push {r0}");
510
511         /* Save PRIMASK */
512         asm("mrs r0,primask");
513         asm("push {r0}");
514 }
515
516 static inline void ao_arch_save_stack(void) {
517         uint32_t        *sp;
518         asm("mov %0,sp" : "=&r" (sp) );
519         ao_cur_task->sp32 = (sp);
520 }
521
522 static inline void ao_arch_restore_stack(void) {
523         /* Switch stacks */
524         asm("mov sp, %0" : : "r" (ao_cur_task->sp32) );
525
526         /* Restore PRIMASK */
527         asm("pop {r0}");
528         asm("msr primask,r0");
529
530         /* Restore APSR */
531         asm("pop {r0}");
532         asm("msr apsr_nczvq,r0");
533
534         /* Restore general registers */
535         asm("pop {r0-r7,pc}\n");
536 }
537
538 #ifndef HAS_SAMPLE_PROFILE
539 #define HAS_SAMPLE_PROFILE 0
540 #endif
541
542 #if DEBUG
543 #define HAS_ARCH_VALIDATE_CUR_STACK     1
544
545 static inline void
546 ao_validate_cur_stack(void)
547 {
548         uint8_t         *psp;
549
550         asm("mrs %0,psp" : "=&r" (psp));
551         if (ao_cur_task &&
552             psp <= ao_cur_task->stack &&
553             psp >= ao_cur_task->stack - 256)
554                 ao_panic(AO_PANIC_STACK);
555 }
556 #endif
557
558 #if !HAS_SAMPLE_PROFILE
559 #define HAS_ARCH_START_SCHEDULER        1
560
561 static inline void ao_arch_start_scheduler(void) {
562         uint32_t        sp;
563         uint32_t        control;
564
565         asm("mrs %0,msp" : "=&r" (sp));
566         asm("msr psp,%0" : : "r" (sp));
567         asm("mrs %0,control" : "=r" (control));
568         control |= (1 << 1);
569         asm("msr control,%0" : : "r" (control));
570         asm("isb");
571 }
572 #endif
573
574 #define ao_arch_isr_stack()
575
576 #endif
577
578 static inline void
579 ao_arch_wait_interrupt(void) {
580 #ifdef AO_NONMASK_INTERRUPTS
581         asm(
582             "dsb\n"                     /* Serialize data */
583             "isb\n"                     /* Serialize instructions */
584             "cpsid i\n"                 /* Block all interrupts */
585             "msr basepri,%0\n"          /* Allow all interrupts through basepri */
586             "wfi\n"                     /* Wait for an interrupt */
587             "cpsie i\n"                 /* Allow all interrupts */
588             "msr basepri,%1\n"          /* Block interrupts through basepri */
589             : : "r" (0), "r" (AO_STM_NVIC_BASEPRI_MASK));
590 #else
591         asm("\twfi\n");
592         ao_arch_release_interrupts();
593         ao_arch_block_interrupts();
594 #endif
595 }
596
597 #define ao_arch_critical(b) do {                        \
598                 uint32_t __mask = ao_arch_irqsave();    \
599                 do { b } while (0);                     \
600                 ao_arch_irqrestore(__mask);             \
601         } while (0)
602
603 void start(void);
604
605 #endif /* _AO_ARCH_FUNCS_H_ */