From 2307e756d32f19bb30b3d49a0abb27bcf6abca8b Mon Sep 17 00:00:00 2001
From: Sven Peter <sven@svenpeter.dev>
Date: Mon, 10 Jan 2022 18:16:58 +0100
Subject: [PATCH] nvme: Add stub ANS NVMe driver

Right now it only initializes the co-processor and makes sure
the firmware has booted successfully.

Signed-off-by: Sven Peter <sven@svenpeter.dev>
---
 Makefile   |  1 +
 src/nvme.c | 92 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/nvme.h | 11 +++++++
 3 files changed, 104 insertions(+)
 create mode 100644 src/nvme.c
 create mode 100644 src/nvme.h

diff --git a/Makefile b/Makefile
index deb03c93..8c0a47fa 100644
--- a/Makefile
+++ b/Makefile
@@ -71,6 +71,7 @@ OBJECTS := \
 	main.o \
 	mcc.o \
 	memory.o memory_asm.o \
+	nvme.o \
 	payload.o \
 	pcie.o \
 	pmgr.o \
diff --git a/src/nvme.c b/src/nvme.c
new file mode 100644
index 00000000..8e1738b9
--- /dev/null
+++ b/src/nvme.c
@@ -0,0 +1,92 @@
+/* SPDX-License-Identifier: MIT */
+
+#include "adt.h"
+#include "nvme.h"
+#include "pmgr.h"
+#include "rtkit.h"
+#include "sart.h"
+#include "utils.h"
+
+#define NVME_BOOT_STATUS    0x1300
+#define NVME_BOOT_STATUS_OK 0xde71ce55
+
+static bool nvme_initialized = false;
+
+static asc_dev_t *nvme_asc = NULL;
+static rtkit_dev_t *nvme_rtkit = NULL;
+static sart_dev_t *nvme_sart = NULL;
+
+static u64 nvme_base;
+
+bool nvme_init(void)
+{
+    if (nvme_initialized) {
+        printf("nvme: already initialized\n");
+        return true;
+    }
+
+    int adt_path[8];
+    int node = adt_path_offset_trace(adt, "/arm-io/ans", adt_path);
+    if (node < 0) {
+        printf("nvme: Error getting NVMe node /arm-io/ans\n");
+        return NULL;
+    }
+
+    if (adt_get_reg(adt, adt_path, "reg", 3, &nvme_base, NULL) < 0) {
+        printf("nvme: Error getting NVMe base address.\n");
+        return NULL;
+    }
+
+    nvme_asc = asc_init("/arm-io/ans");
+    if (!nvme_asc)
+        return false;
+    asc_cpu_start(nvme_asc);
+
+    nvme_sart = sart_init("/arm-io/sart-ans");
+    if (!nvme_sart)
+        goto out_asc;
+
+    nvme_rtkit = rtkit_init("nvme", nvme_asc, NULL, NULL, nvme_sart);
+    if (!nvme_rtkit)
+        goto out_sart;
+
+    if (!rtkit_boot(nvme_rtkit))
+        goto out_rtkit;
+
+    if (poll32(nvme_base + NVME_BOOT_STATUS, 0xffffffff, NVME_BOOT_STATUS_OK, USEC_PER_SEC) < 0) {
+        printf("nvme: ANS did not boot correctly.\n");
+        goto out_shutdown;
+    }
+
+    nvme_initialized = true;
+    printf("nvme: initialized at 0x%lx\n", nvme_base);
+    return true;
+
+out_shutdown:
+    rtkit_shutdown(nvme_rtkit);
+    pmgr_reset("ANS2");
+out_rtkit:
+    rtkit_free(nvme_rtkit);
+out_sart:
+    sart_free(nvme_sart);
+out_asc:
+    asc_free(nvme_asc);
+    return false;
+}
+
+void nvme_shutdown(void)
+{
+    if (!nvme_initialized) {
+        printf("nvme: trying to shut down but not initialized\n");
+        return;
+    }
+
+    rtkit_shutdown(nvme_rtkit);
+    pmgr_reset("ANS2");
+    rtkit_free(nvme_rtkit);
+    sart_free(nvme_sart);
+    asc_free(nvme_asc);
+    nvme_initialized = false;
+
+    printf("nvme: shutdown done\n");
+}
diff --git a/src/nvme.h b/src/nvme.h
new file mode 100644
index 00000000..99ecd8b8
--- /dev/null
+++ b/src/nvme.h
@@ -0,0 +1,11 @@
+/* SPDX-License-Identifier: MIT */
+
+#ifndef NVME_H
+#define NVME_H
+
+#include "types.h"
+
+bool nvme_init(void);
+void nvme_shutdown(void);
+
+#endif