--- zzzz-none-000/linux-4.4.60/arch/mips/ath79/prom.c	2017-04-08 07:53:53.000000000 +0000
+++ dragonfly-4020-701/linux-4.4.60/arch/mips/ath79/prom.c	2018-11-08 13:36:17.000000000 +0000
@@ -1,7 +1,6 @@
 /*
  *  Atheros AR71XX/AR724X/AR913X specific prom routines
  *
- *  Copyright (C) 2015 Laurent Fasnacht <l@libres.ch>
  *  Copyright (C) 2008-2010 Gabor Juhos <juhosg@openwrt.org>
  *  Copyright (C) 2008 Imre Kaloz <kaloz@openwrt.org>
  *
@@ -14,29 +13,122 @@
 #include <linux/init.h>
 #include <linux/io.h>
 #include <linux/string.h>
-#include <linux/initrd.h>
+
+#include <linux/mm.h>
+#include <linux/bootmem.h>
+#include <linux/env.h>
+
+#include <linux/of_fdt.h>
+#include <linux/avm_kernel_config.h>
 
 #include <asm/bootinfo.h>
 #include <asm/addrspace.h>
-#include <asm/fw/fw.h>
+#include <asm/traps.h>
+#include <asm/cacheflush.h>
+#include <asm/prom.h>
 
+#include "machtypes.h"
 #include "common.h"
 
-void __init prom_init(void)
-{
-	fw_init_cmdline();
+/* --- #define DEBUG_ATH_PROM --- */
 
-#ifdef CONFIG_BLK_DEV_INITRD
-	/* Read the initrd address from the firmware environment */
-	initrd_start = fw_getenvl("initrd_start");
-	if (initrd_start) {
-		initrd_start = KSEG0ADDR(initrd_start);
-		initrd_end = initrd_start + fw_getenvl("initrd_size");
-	}
+#if defined(DEBUG_ATH_PROM)
+#define ath_prom_print(arg...)      printk(arg)
+#else
+#define ath_prom_print(arg...)   
 #endif
+
+void prom_printf (const char *fmt, ...) {
+    va_list args;
+
+    va_start (args, fmt);
+    ath_prom_print (fmt, args);
+    va_end (args);
+}
+
+/*------------------------------------------------------------------------------------------*\
+\*------------------------------------------------------------------------------------------*/
+static __init void ath79_prom_init_cmdline(int argc, char **argv) {
+
+	int i;
+
+    ath_prom_print("\n");     /*--- damit es schön aussieht ---*/
+    ath_prom_print("[%s]:\n", __func__);
+    ath_prom_print("[%s]: argv[0] = '%s'\n", __func__, argv[0]);
+    ath_prom_print("[%s]: argv[1] = '%s'\n", __func__, argv[1]);
+    ath_prom_print("[%s]: argv[2] = '%s'\n", __func__, argv[2]);
+
+#if 0
+	if (!is_valid_ram_addr(argv))
+		return;
+#endif
+
+	for (i = 0; i < argc; i++) {
+#if 0
+		if (is_valid_ram_addr(argv[i])) {
+#endif
+			strlcat(arcs_cmdline, " ", sizeof(arcs_cmdline));
+			strlcat(arcs_cmdline, argv[i], sizeof(arcs_cmdline));
+#if 0
+		}
+#endif
+    }
+
+    ath_prom_print("[%s]:commandline = '%s'\n", __FUNCTION__, arcs_cmdline);
+}
+
+/*------------------------------------------------------------------------------------------*\
+\*------------------------------------------------------------------------------------------*/
+static void __init mips_nmi_setup(void) {
+
+	void *base;
+	extern char except_vec_nmi;
+
+	base = cpu_has_veic ?
+		(void *)(CAC_BASE + 0xa80) :
+		(void *)(CAC_BASE + 0x380);
+	memcpy(base, &except_vec_nmi, 0x80);
+    printk(KERN_ERR "[%s] setup NMI vector to base 0x%p\n", __FUNCTION__, base);
+	flush_icache_range((unsigned long)base, (unsigned long)base + 0x80);
+}
+
+/*------------------------------------------------------------------------------------------*\
+\*------------------------------------------------------------------------------------------*/
+void __init prom_init(void) {
+    /* No Env from Urlader, load from AVM Config space */
+    if(avm_kernel_urlader_env) {
+        /* === Search for kernel_args_tmp in urlader_env === */
+        struct _avm_kernel_urlader_env * env_var = avm_kernel_urlader_env;
+        while(env_var->name[0]){
+            if(strncmp(env_var->name, "kernel_args_tmp", 64) == 0){
+                strlcat(arcs_cmdline, " ", sizeof(arcs_cmdline));
+    			strlcat(arcs_cmdline, env_var->value, sizeof(arcs_cmdline));
+                break;
+            }
+            env_var++;
+        }
+ 
+        /* == Init Env with config content == */
+        ath_prom_print("[%s]: call env_init\n", __func__);
+        env_init((int *)avm_kernel_urlader_env, ENV_LOCATION_AVM_CONF);
+
+        board_nmi_handler_setup = NULL;
+    } else {
+        ath_prom_print("[%s]:\n", __func__);
+        ath79_prom_init_cmdline(fw_arg0, (char **)fw_arg1);
+        ath_prom_print("[%s]: call env_init\n", __func__);
+        env_init((int *)fw_arg2, ENV_LOCATION_FLASH);
+
+        ath_prom_print("[%s]: call set_wlan_dect_config_address\n", __func__);
+        set_wlan_dect_config_address((unsigned int *)fw_arg3);
+
+	    board_nmi_handler_setup = mips_nmi_setup;
+    }
+    ath_prom_print("[%s]: done\n", __func__);
 }
 
 void __init prom_free_prom_memory(void)
 {
 	/* We do not have to prom memory to free */
 }
+