diff --git a/stmhal/main.c b/stmhal/main.c index 86cbe36011ffe4ce6a42757d01d0f46fa8e541e0..bfcee76b636b0ee906f851204a14005a8423719e 100644 --- a/stmhal/main.c +++ b/stmhal/main.c @@ -458,8 +458,7 @@ soft_reset: #endif #if MICROPY_HW_ENABLE_RNG - // RNG - rng_init(); + rng_init0(); #endif #if MICROPY_HW_ENABLE_TIMER diff --git a/stmhal/rng.c b/stmhal/rng.c index f0d14fe97552cd61d4a764eaf1c047514db098c6..19e624f79238df9ba409427461c8407efff3328c 100644 --- a/stmhal/rng.c +++ b/stmhal/rng.c @@ -1,3 +1,5 @@ +#include <string.h> + #include "stm32f4xx_hal.h" #include "misc.h" @@ -6,20 +8,35 @@ #include "obj.h" #include "rng.h" -STATIC RNG_HandleTypeDef RngHandle; +#if MICROPY_HW_ENABLE_RNG + +STATIC RNG_HandleTypeDef RNGHandle = {.Instance = NULL}; + +void rng_init0(void) { + // reset the RNG handle + memset(&RNGHandle, 0, sizeof(RNG_HandleTypeDef)); + RNGHandle.Instance = RNG; +} void rng_init(void) { __RNG_CLK_ENABLE(); - RngHandle.Instance = RNG; - HAL_RNG_Init(&RngHandle); + HAL_RNG_Init(&RNGHandle); } uint32_t rng_get(void) { - return HAL_RNG_GetRandomNumber(&RngHandle); + if (RNGHandle.State == HAL_RNG_STATE_RESET) { + rng_init(); + } + return HAL_RNG_GetRandomNumber(&RNGHandle); } STATIC mp_obj_t pyb_rng_get(void) { - return mp_obj_new_int(HAL_RNG_GetRandomNumber(&RngHandle) >> 2); + if (RNGHandle.State == HAL_RNG_STATE_RESET) { + rng_init(); + } + return mp_obj_new_int(HAL_RNG_GetRandomNumber(&RNGHandle) >> 2); } MP_DEFINE_CONST_FUN_OBJ_0(pyb_rng_get_obj, pyb_rng_get); + +#endif // MICROPY_HW_ENABLE_RNG diff --git a/stmhal/rng.h b/stmhal/rng.h index 5442d453c2a0aacc1a8f387f5a2ce590b3769549..650d69026d3815698d5dabe2669abc1ab7e719e0 100644 --- a/stmhal/rng.h +++ b/stmhal/rng.h @@ -1,4 +1,4 @@ -void rng_init(void); +void rng_init0(void); uint32_t rng_get(void); MP_DECLARE_CONST_FUN_OBJ(pyb_rng_get_obj);