From 6ca17c19225c228c39dcc1f8be4ec269c702a165 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Tue, 29 Mar 2016 10:29:57 +0300
Subject: [PATCH] esp8266: Implement os.urandom function.
Uses what is suspected to be a hardware random number generator.
---
esp8266/etshal.h | 3 +++
esp8266/moduos.c | 13 +++++++++++++
2 files changed, 16 insertions(+)
diff --git a/esp8266/etshal.h b/esp8266/etshal.h
index 115ff7a88..d8a57e8c7 100644
--- a/esp8266/etshal.h
+++ b/esp8266/etshal.h
@@ -3,6 +3,9 @@
#include <os_type.h>
+// see http://esp8266-re.foogod.com/wiki/Random_Number_Generator
+#define WDEV_HWRNG ((volatile uint32_t*)0x3ff20e44)
+
void ets_delay_us();
void ets_intr_lock(void);
void ets_intr_unlock(void);
diff --git a/esp8266/moduos.c b/esp8266/moduos.c
index bef7cddc3..a9f19d701 100644
--- a/esp8266/moduos.c
+++ b/esp8266/moduos.c
@@ -34,6 +34,7 @@
#include "py/objstr.h"
#include "py/runtime.h"
#include "genhdr/mpversion.h"
+#include "etshal.h"
#include "user_interface.h"
extern const mp_obj_type_t mp_fat_vfs_type;
@@ -92,9 +93,21 @@ STATIC mp_obj_t os_remove(mp_obj_t path_in) {
STATIC MP_DEFINE_CONST_FUN_OBJ_1(os_remove_obj, os_remove);
#endif
+STATIC mp_obj_t os_urandom(mp_obj_t num) {
+ mp_int_t n = mp_obj_get_int(num);
+ vstr_t vstr;
+ vstr_init_len(&vstr, n);
+ for (int i = 0; i < n; i++) {
+ vstr.buf[i] = *WDEV_HWRNG;
+ }
+ return mp_obj_new_str_from_vstr(&mp_type_bytes, &vstr);
+}
+STATIC MP_DEFINE_CONST_FUN_OBJ_1(os_urandom_obj, os_urandom);
+
STATIC const mp_rom_map_elem_t os_module_globals_table[] = {
{ MP_ROM_QSTR(MP_QSTR___name__), MP_ROM_QSTR(MP_QSTR_uos) },
{ MP_ROM_QSTR(MP_QSTR_uname), MP_ROM_PTR(&os_uname_obj) },
+ { MP_ROM_QSTR(MP_QSTR_urandom), MP_ROM_PTR(&os_urandom_obj) },
#if MICROPY_VFS_FAT
{ MP_ROM_QSTR(MP_QSTR_VfsFat), MP_ROM_PTR(&mp_fat_vfs_type) },
{ MP_ROM_QSTR(MP_QSTR_listdir), MP_ROM_PTR(&os_listdir_obj) },
--
GitLab