Move memory training code into it's own stage (fusee-mtc)

fusee/fusee-mtc/src/main.c

@@ -0,0 +1,57 @@
+/*
+ * Copyright (c) 2018-2019 Atmosphère-NX
+ *
+ * This program is free software; you can redistribute it and/or modify it
+ * under the terms and conditions of the GNU General Public License,
+ * version 2, as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
+ * more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <string.h>
+#include "mtc.h"
+#include "stage2.h"
+#include "display/video_fb.h"
+
+static void *g_framebuffer;
+static __attribute__((__aligned__(0x200))) stage2_mtc_args_t g_mtc_args_store;
+static stage2_mtc_args_t *g_mtc_args;
+
+/* Allow for main(int argc, void **argv) signature. */
+#pragma GCC diagnostic ignored "-Wmain"
+
+int main(int argc, void **argv) {
+    ScreenLogLevel log_level = SCREEN_LOG_LEVEL_NONE;
+    
+    /* Check argc. */
+    if (argc != MTC_ARGC) {
+        return 1;
+    }
+    
+    /* Extract arguments from argv. */
+    g_mtc_args = &g_mtc_args_store;
+    memcpy(g_mtc_args, (stage2_mtc_args_t *)argv[MTC_ARGV_ARGUMENT_STRUCT], sizeof(*g_mtc_args));
+    log_level = g_mtc_args->log_level;
+    
+    /* Override the global logging level. */
+    log_set_log_level(log_level);
+    
+    if (log_level != SCREEN_LOG_LEVEL_NONE) {
+        /* Set framebuffer address. */
+        g_framebuffer = (void *)0xC0000000;
+    
+        /* Zero-fill the framebuffer and register it as printk provider. */
+        video_init(g_framebuffer);
+    }
+    
+    /* Train DRAM. */
+    train_dram();
+
+    return 0;
+}