[ar71xx] fix handling of invalid arguments passed by some bootloaders (thanks to...
[openwrt.git] / target / linux / ar71xx / files / arch / mips / ar71xx / prom.c
1 /*
2  *  Atheros AR71xx SoC specific prom routines
3  *
4  *  Copyright (C) 2008-2009 Gabor Juhos <juhosg@openwrt.org>
5  *  Copyright (C) 2008 Imre Kaloz <kaloz@openwrt.org>
6  *
7  *  This program is free software; you can redistribute it and/or modify it
8  *  under the terms of the GNU General Public License version 2 as published
9  *  by the Free Software Foundation.
10  */
11
12 #include <linux/kernel.h>
13 #include <linux/init.h>
14 #include <linux/io.h>
15 #include <linux/serial_reg.h>
16
17 #include <asm/bootinfo.h>
18 #include <asm/addrspace.h>
19 #include <asm/fw/myloader/myloader.h>
20
21 #include <asm/mach-ar71xx/ar71xx.h>
22 #include <asm/mach-ar71xx/platform.h>
23
24 struct board_rec {
25         char            *name;
26         unsigned long   mach_type;
27 };
28
29 static int ar71xx_prom_argc __initdata;
30 static char **ar71xx_prom_argv __initdata;
31 static char **ar71xx_prom_envp __initdata;
32
33 static struct board_rec boards[] __initdata = {
34         {
35                 .name           = "411",
36                 .mach_type      = AR71XX_MACH_RB_411,
37         }, {
38                 .name           = "433",
39                 .mach_type      = AR71XX_MACH_RB_433,
40         }, {
41                 .name           = "450",
42                 .mach_type      = AR71XX_MACH_RB_450,
43         }, {
44                 .name           = "493",
45                 .mach_type      = AR71XX_MACH_RB_493,
46         }, {
47                 .name           = "AW-NR580",
48                 .mach_type      = AR71XX_MACH_AW_NR580,
49         }, {
50                 .name           = "AP83",
51                 .mach_type      = AR71XX_MACH_AP83,
52         }, {
53                 .name           = "TEW-632BRP",
54                 .mach_type      = AR71XX_MACH_TEW_632BRP,
55         }, {
56                 .name           = "UBNT-RS",
57                 .mach_type      = AR71XX_MACH_UBNT_RS,
58         }, {
59                 .name           = "UBNT-LSX",
60                 .mach_type      = AR71XX_MACH_UBNT_LSX,
61         }, {
62                 .name           = "WNR2000",
63                 .mach_type      = AR71XX_MACH_WNR2000,
64         }, {
65                 .name           = "PB42",
66                 .mach_type      = AR71XX_MACH_PB42,
67         }, {
68                 .name           = "MZK-W300NH",
69                 .mach_type      = AR71XX_MACH_MZK_W300NH,
70         }, {
71                 .name           = "MZK-W04NU",
72                 .mach_type      = AR71XX_MACH_MZK_W04NU,
73         }
74 };
75
76 static inline int is_valid_ram_addr(void *addr)
77 {
78         if (((u32) addr > KSEG0) &&
79             ((u32) addr < (KSEG0 + AR71XX_MEM_SIZE_MAX)))
80                 return 1;
81
82         if (((u32) addr > KSEG1) &&
83             ((u32) addr < (KSEG1 + AR71XX_MEM_SIZE_MAX)))
84                 return 1;
85
86         return 0;
87 }
88
89 static __init char *ar71xx_prom_getargv(const char *name)
90 {
91         int len = strlen(name);
92         int i;
93
94         if (!is_valid_ram_addr(ar71xx_prom_argv))
95                 return NULL;
96
97         for (i = 0; i < ar71xx_prom_argc; i++) {
98                 char *argv = ar71xx_prom_argv[i];
99
100                 if (!is_valid_ram_addr(argv))
101                         continue;
102
103                 if (strncmp(name, argv, len) == 0 && (argv)[len] == '=')
104                         return argv + len + 1;
105         }
106
107         return NULL;
108 }
109
110 static __init char *ar71xx_prom_getenv(const char *envname)
111 {
112         int len = strlen(envname);
113         char **env;
114
115         if (!is_valid_ram_addr(ar71xx_prom_envp))
116                 return NULL;
117
118         for (env = ar71xx_prom_envp; is_valid_ram_addr(*env); env++)
119                 if (strncmp(envname, *env, len) == 0 && (*env)[len] == '=')
120                         return *env + len + 1;
121
122         return NULL;
123 }
124
125 static __init unsigned long find_board_byname(char *name)
126 {
127         int i;
128
129         for (i = 0; i < ARRAY_SIZE(boards); i++)
130                 if (strcmp(name, boards[i].name) == 0)
131                         return boards[i].mach_type;
132
133         return AR71XX_MACH_GENERIC;
134 }
135
136 static int ar71xx_prom_init_myloader(void)
137 {
138         struct myloader_info *mylo;
139
140         mylo = myloader_get_info();
141         if (!mylo)
142                 return 0;
143
144         switch (mylo->did) {
145         case DEVID_COMPEX_WP543:
146                 ar71xx_mach_type = AR71XX_MACH_WP543;
147                 break;
148         default:
149                 printk(KERN_WARNING "prom: unknown device id: %x\n",
150                                 mylo->did);
151         }
152         ar71xx_set_mac_base(mylo->macs[0]);
153
154         return 1;
155 }
156
157 static void ar71xx_prom_init_generic(void)
158 {
159         char *p;
160
161         ar71xx_prom_argc = fw_arg0;
162         ar71xx_prom_argv = (char **)fw_arg1;
163         ar71xx_prom_envp = (char **)fw_arg2;
164
165         p = ar71xx_prom_getenv("board");
166         if (!p)
167                 p = ar71xx_prom_getargv("board");
168         if (p)
169                 ar71xx_mach_type = find_board_byname(p);
170
171         p = ar71xx_prom_getenv("ethaddr");
172         if (!p)
173                 p = ar71xx_prom_getargv("kmac");
174         if (p)
175                 ar71xx_parse_mac_addr(p);
176 }
177
178 void __init prom_init(void)
179 {
180         printk(KERN_DEBUG "prom: fw_arg0=%08x, fw_arg1=%08x, "
181                         "fw_arg2=%08x, fw_arg3=%08x\n",
182                         (unsigned int)fw_arg0, (unsigned int)fw_arg1,
183                         (unsigned int)fw_arg2, (unsigned int)fw_arg3);
184
185         ar71xx_mach_type = AR71XX_MACH_GENERIC;
186
187         if (ar71xx_prom_init_myloader())
188                 return;
189
190         ar71xx_prom_init_generic();
191 }
192
193 void __init prom_free_prom_memory(void)
194 {
195         /* We do not have to prom memory to free */
196 }
197
198 #define UART_READ(r) \
199         __raw_readl((void __iomem *)(KSEG1ADDR(AR71XX_UART_BASE) + 4 * (r)))
200
201 #define UART_WRITE(r, v) \
202         __raw_writel((v), (void __iomem *)(KSEG1ADDR(AR71XX_UART_BASE) + 4*(r)))
203
204 void prom_putchar(unsigned char ch)
205 {
206         while (((UART_READ(UART_LSR)) & UART_LSR_THRE) == 0);
207         UART_WRITE(UART_TX, ch);
208         while (((UART_READ(UART_LSR)) & UART_LSR_THRE) == 0);
209 }