esp32: Add scripts to init and mount filesystem.

dpgeorge · dpgeorge · commit 28ecbe04e538 · 2016-12-19T13:06:45.000+11:00
diff --git a/esp32/main.c b/esp32/main.c
@@ -66,6 +66,13 @@ void mp_task(void *pvParameter) {
     mp_obj_list_init(mp_sys_argv, 0);
     readline_init0();
 
+    // run boot-up scripts
+    pyexec_frozen_module("_boot.py");
+    pyexec_file("boot.py");
+    if (pyexec_mode_kind == PYEXEC_MODE_FRIENDLY_REPL) {
+        pyexec_file("main.py");
+    }
+
     for (;;) {
         if (pyexec_mode_kind == PYEXEC_MODE_RAW_REPL) {
             if (pyexec_raw_repl() != 0) {
diff --git a/esp32/modules/_boot.py b/esp32/modules/_boot.py
@@ -0,0 +1,12 @@
+import gc
+import uos
+from flashbdev import bdev
+
+try:
+    if bdev:
+        vfs = uos.VfsFat(bdev, "")
+except OSError:
+    import inisetup
+    vfs = inisetup.setup()
+
+gc.collect()
diff --git a/esp32/modules/flashbdev.py b/esp32/modules/flashbdev.py
@@ -0,0 +1,34 @@
+import esp
+
+class FlashBdev:
+
+    SEC_SIZE = 4096
+    START_SEC = esp.flash_user_start() // SEC_SIZE
+
+    def __init__(self, blocks):
+        self.blocks = blocks
+
+    def readblocks(self, n, buf):
+        #print("readblocks(%s, %x(%d))" % (n, id(buf), len(buf)))
+        esp.flash_read((n + self.START_SEC) * self.SEC_SIZE, buf)
+
+    def writeblocks(self, n, buf):
+        #print("writeblocks(%s, %x(%d))" % (n, id(buf), len(buf)))
+        #assert len(buf) <= self.SEC_SIZE, len(buf)
+        esp.flash_erase(n + self.START_SEC)
+        esp.flash_write((n + self.START_SEC) * self.SEC_SIZE, buf)
+
+    def ioctl(self, op, arg):
+        #print("ioctl(%d, %r)" % (op, arg))
+        if op == 4:  # BP_IOCTL_SEC_COUNT
+            return self.blocks
+        if op == 5:  # BP_IOCTL_SEC_SIZE
+            return self.SEC_SIZE
+
+size = esp.flash_size()
+if size < 1024*1024:
+    # flash too small for a filesystem
+    bdev = None
+else:
+    # for now we use a fixed size for the filesystem
+    bdev = FlashBdev(256 * 1024 // FlashBdev.SEC_SIZE)
diff --git a/esp32/modules/inisetup.py b/esp32/modules/inisetup.py
@@ -0,0 +1,36 @@
+import uos
+from flashbdev import bdev
+
+def check_bootsec():
+    buf = bytearray(bdev.SEC_SIZE)
+    bdev.readblocks(0, buf)
+    empty = True
+    for b in buf:
+        if b != 0xff:
+            empty = False
+            break
+    if empty:
+        return True
+    fs_corrupted()
+
+def fs_corrupted():
+    import time
+    while 1:
+        print("""\
+FAT filesystem appears to be corrupted. If you had important data there, you
+may want to make a flash snapshot to try to recover it. Otherwise, perform
+factory reprogramming of MicroPython firmware (completely erase flash, followed
+by firmware programming).
+""")
+        time.sleep(3)
+
+def setup():
+    check_bootsec()
+    print("Performing initial setup")
+    uos.VfsFat.mkfs(bdev)
+    vfs = uos.VfsFat(bdev, "")
+    with open("/boot.py", "w") as f:
+        f.write("""\
+# This file is executed on every boot (including wake-boot from deepsleep)
+""")
+    return vfs
