random numbers for fe310

4 files changed, 64 insertions, 0 deletions

diff --git a/fw/esp32/components/eos/include/rng.h b/fw/esp32/components/eos/include/rng.h
new file mode 100644
index 0000000..0690973
--- /dev/null
+++ b/fw/esp32/components/eos/include/rng.h
@@ -0,0 +1 @@
+void eos_rng_init(void);
diff --git a/fw/esp32/components/eos/rng.c b/fw/esp32/components/eos/rng.c
new file mode 100644
index 0000000..79eca21
--- /dev/null
+++ b/fw/esp32/components/eos/rng.c
@@ -0,0 +1,31 @@
+#include <esp_system.h>
+#include <esp_log.h>
+#include <esp_err.h>
+
+#include "net.h"
+#include "rng.h"
+
+static const char *TAG = "EOS RNG";
+
+static void rng_handler(unsigned char type, unsigned char *buffer, uint16_t buf_len) {
+    uint16_t rng_len = 0;
+
+    if (buf_len < sizeof(uint16_t)) goto rng_handler_fin;
+
+    rng_len  = (uint16_t)buffer[0] << 8;
+    rng_len |= (uint16_t)buffer[1];
+    if (rng_len > EOS_NET_MTU) {
+        rng_len = 0;
+        goto rng_handler_fin;
+    }
+    esp_fill_random(buffer, rng_len);
+
+rng_handler_fin:
+    eos_net_reply(EOS_NET_MTYPE_RNG, buffer, rng_len);
+}
+
+void eos_rng_init(void) {
+    eos_net_set_handler(EOS_NET_MTYPE_RNG, rng_handler);
+    ESP_LOGI(TAG, "INIT");
+}
+
diff --git a/fw/fe310/eos/rng.c b/fw/fe310/eos/rng.c
new file mode 100644
index 0000000..317941d
--- /dev/null
+++ b/fw/fe310/eos/rng.c
@@ -0,0 +1,27 @@
+#include <stdlib.h>
+#include <stdint.h>
+#include <string.h>
+
+#include "net.h"
+
+int getentropy(unsigned char *b, size_t sz) {
+    unsigned char type;
+    unsigned char *buffer;
+    uint16_t len;
+    int rv;
+
+    buffer = eos_net_alloc();
+
+    type = EOS_NET_MTYPE_RNG;
+    len = sizeof(uint16_t);
+    buffer[0] = sz >> 8;
+    buffer[1] = sz;
+
+    rv = eos_net_xchg(&type, buffer, &len);
+    if (rv || (len != sz)) rv = -1;
+
+    if (!rv) memcpy(b, buffer, sz);
+    eos_net_free(buffer, 1);
+
+    return rv;
+}
diff --git a/fw/fe310/eos/rng.h b/fw/fe310/eos/rng.h
new file mode 100644
index 0000000..27bbf74
--- /dev/null
+++ b/fw/fe310/eos/rng.h
@@ -0,0 +1,5 @@
+#include <stdlib.h>
+
+int getentropy(unsigned char *b, size_t sz);
+int arc4random_alloc(void **rsp, size_t rsp_size, void **rsxp, size_t rsxp_size);
+void arc4random_close(void **rsp, void **rsxp);
\ No newline at end of file