1 files changed, 50 insertions, 0 deletions
diff --git a/src/routines/check_magic.c b/src/routines/check_magic.c
new file mode 100644
index 0000000..bb49cff
--- /dev/null
+++ b/src/routines/check_magic.c
@@ -0,0 +1,50 @@
+/*
+ * nanowasm, a tiny WebAssembly/Wasm interpreter
+ * Copyright (C) 2023-2024  Xavier Del Campo Romero
+ *
+ * This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at https://mozilla.org/MPL/2.0/.
+ */
+
+#include <nanowasm/nw.h>
+#include <nw/io.h>
+#include <nw/log.h>
+#include <nw/routines.h>
+#include <string.h>
+
+static enum nw_state run(struct nw_mod *const m)
+{
+    struct nw_sm_cm *const cm = &m->sm.check_magic;
+    const enum nw_state n = nwp_io_read(&m->cfg.io, &cm->io);
+
+    if (n)
+        return n;
+
+    static const uint8_t magic[] = {'\0', 'a', 's', 'm'};
+
+    if (memcmp(cm->buf, magic, sizeof cm->buf))
+    {
+        LOG("%s: wrong magic bytes\n", __func__);
+        return NW_FATAL;
+    }
+
+    nwp_check_version(m);
+    return NW_AGAIN;
+}
+
+void nwp_check_magic(struct nw_mod *const m)
+{
+    struct nw_sm_cm *const cm = &m->sm.check_magic;
+
+    *cm = (const struct nw_sm_cm)
+    {
+        .io =
+        {
+            .buf = &cm->buf,
+            .n = sizeof cm->buf
+        }
+    };
+
+    m->next = run;
+}