]>
Commit | Line | Data |
---|---|---|
fa1df308 BS |
1 | /* |
2 | * (C) Copyright 2007 Schindler Lift Inc. | |
86b116b1 | 3 | * (C) Copyright 2007 DENX Software Engineering |
fa1df308 BS |
4 | * |
5 | * Author: Michel Marti <mma@objectxp.com> | |
6 | * Adapted for U-Boot 1.2 by Piotr Kruszynski <ppk@semihalf.com>: | |
86b116b1 BS |
7 | * - code clean-up |
8 | * - bugfix for overwriting bootargs by user | |
fa1df308 BS |
9 | * |
10 | * See file CREDITS for list of people who contributed to this | |
11 | * project. | |
12 | * | |
13 | * This program is free software; you can redistribute it and/or | |
14 | * modify it under the terms of the GNU General Public License as | |
15 | * published by the Free Software Foundation; either version 2 of | |
16 | * the License, or (at your option) any later version. | |
17 | * | |
18 | * This program is distributed in the hope that it will be useful, | |
19 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
20 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
21 | * GNU General Public License for more details. | |
22 | * | |
23 | * You should have received a copy of the GNU General Public License | |
24 | * along with this program; if not, write to the Free Software | |
25 | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, | |
26 | * MA 02111-1307 USA | |
27 | */ | |
28 | ||
29 | #include <common.h> | |
30 | #include <command.h> | |
31 | #include <malloc.h> | |
32 | #include <image.h> | |
33 | #include <usb.h> | |
34 | #include <fat.h> | |
35 | ||
36 | #include "fwupdate.h" | |
37 | ||
38 | extern int do_bootm(cmd_tbl_t *, int, int, char *[]); | |
39 | extern long do_fat_read(const char *, void *, unsigned long, int); | |
40 | extern int do_fat_fsload(cmd_tbl_t *, int, int, char *[]); | |
41 | ||
42 | static int load_rescue_image(ulong); | |
43 | ||
86b116b1 | 44 | void cm5200_fwupdate(void) |
fa1df308 BS |
45 | { |
46 | cmd_tbl_t *bcmd; | |
47 | char *rsargs; | |
48 | char *tmp = NULL; | |
49 | char ka[16]; | |
50 | char *argv[3] = { "bootm", ka, NULL }; | |
51 | ||
52 | /* Check if rescue system is disabled... */ | |
53 | if (getenv("norescue")) { | |
54 | printf(LOG_PREFIX "Rescue System disabled.\n"); | |
55 | return; | |
56 | } | |
57 | ||
58 | /* Check if we have a USB storage device and load image */ | |
59 | if (load_rescue_image(LOAD_ADDR)) | |
60 | return; | |
61 | ||
62 | bcmd = find_cmd("bootm"); | |
63 | if (!bcmd) | |
64 | return; | |
65 | ||
9b55a253 | 66 | sprintf(ka, "%lx", (ulong)LOAD_ADDR); |
fa1df308 BS |
67 | |
68 | /* prepare our bootargs */ | |
69 | rsargs = getenv("rs-args"); | |
70 | if (!rsargs) | |
71 | rsargs = RS_BOOTARGS; | |
72 | else { | |
73 | tmp = malloc(strlen(rsargs+1)); | |
74 | if (!tmp) { | |
75 | printf(LOG_PREFIX "Memory allocation failed\n"); | |
76 | return; | |
77 | } | |
78 | strcpy(tmp, rsargs); | |
79 | rsargs = tmp; | |
80 | } | |
81 | ||
82 | setenv("bootargs", rsargs); | |
83 | ||
84 | if (rsargs == tmp) | |
85 | free(rsargs); | |
86 | ||
87 | printf(LOG_PREFIX "Starting update system (bootargs=%s)...\n", rsargs); | |
88 | do_bootm(bcmd, 0, 2, argv); | |
89 | } | |
90 | ||
91 | static int load_rescue_image(ulong addr) | |
92 | { | |
93 | disk_partition_t info; | |
94 | int devno; | |
95 | int partno; | |
96 | int i; | |
97 | char fwdir[64]; | |
98 | char nxri[128]; | |
99 | char *tmp; | |
100 | char dev[7]; | |
101 | char addr_str[16]; | |
102 | char *argv[6] = { "fatload", "usb", dev, addr_str, nxri, NULL }; | |
103 | block_dev_desc_t *stor_dev = NULL; | |
104 | cmd_tbl_t *bcmd; | |
105 | ||
106 | /* Get name of firmware directory */ | |
107 | tmp = getenv("fw-dir"); | |
108 | ||
109 | /* Copy it into fwdir */ | |
110 | strncpy(fwdir, tmp ? tmp : FW_DIR, sizeof(fwdir)); | |
111 | fwdir[sizeof(fwdir) - 1] = 0; /* Terminate string */ | |
112 | ||
113 | printf(LOG_PREFIX "Checking for firmware image directory '%s' on USB" | |
114 | " storage...\n", fwdir); | |
115 | usb_stop(); | |
116 | if (usb_init() != 0) | |
117 | return 1; | |
118 | ||
119 | /* Check for storage device */ | |
120 | if (usb_stor_scan(1) != 0) { | |
121 | usb_stop(); | |
122 | return 1; | |
123 | } | |
124 | ||
125 | /* Detect storage device */ | |
126 | for (devno = 0; devno < USB_MAX_STOR_DEV; devno++) { | |
127 | stor_dev = usb_stor_get_dev(devno); | |
128 | if (stor_dev->type != DEV_TYPE_UNKNOWN) | |
129 | break; | |
130 | } | |
131 | if (!stor_dev || stor_dev->type == DEV_TYPE_UNKNOWN) { | |
132 | printf(LOG_PREFIX "No valid storage device found...\n"); | |
133 | usb_stop(); | |
134 | return 1; | |
135 | } | |
136 | ||
137 | /* Detect partition */ | |
138 | for (partno = -1, i = 0; i < 6; i++) { | |
139 | if (get_partition_info(stor_dev, i, &info) == 0) { | |
140 | if (fat_register_device(stor_dev, i) == 0) { | |
141 | /* Check if rescue image is present */ | |
142 | FW_DEBUG("Looking for firmware directory '%s'" | |
143 | " on partition %d\n", fwdir, i); | |
144 | if (do_fat_read(fwdir, NULL, 0, LS_NO) == -1) { | |
145 | FW_DEBUG("No NX rescue image on " | |
146 | "partition %d.\n", i); | |
86b116b1 | 147 | partno = -2; |
fa1df308 BS |
148 | } else { |
149 | partno = i; | |
150 | FW_DEBUG("Partition %d contains " | |
151 | "firmware directory\n", partno); | |
152 | break; | |
153 | } | |
154 | } | |
155 | } | |
156 | } | |
157 | ||
86b116b1 BS |
158 | if (partno < 0) { |
159 | switch (partno) { | |
160 | case -1: | |
161 | printf(LOG_PREFIX "Error: No valid (FAT) partition " | |
162 | "detected\n"); | |
163 | break; | |
164 | case -2: | |
165 | printf(LOG_PREFIX "Error: No NX rescue image on FAT " | |
166 | "partition\n"); | |
167 | break; | |
168 | default: | |
169 | printf(LOG_PREFIX "Error: Failed with code %d\n", | |
170 | partno); | |
171 | } | |
fa1df308 BS |
172 | usb_stop(); |
173 | return 1; | |
174 | } | |
175 | ||
176 | /* Load the rescue image */ | |
177 | bcmd = find_cmd("fatload"); | |
178 | if (!bcmd) { | |
179 | printf(LOG_PREFIX "Error - 'fatload' command not present.\n"); | |
180 | usb_stop(); | |
181 | return 1; | |
182 | } | |
183 | ||
184 | tmp = getenv("nx-rescue-image"); | |
185 | sprintf(nxri, "%s/%s", fwdir, tmp ? tmp : RESCUE_IMAGE); | |
186 | sprintf(dev, "%d:%d", devno, partno); | |
187 | sprintf(addr_str, "%lx", addr); | |
188 | ||
189 | FW_DEBUG("fat_fsload device='%s', addr='%s', file: %s\n", | |
190 | dev, addr_str, nxri); | |
191 | ||
192 | if (do_fat_fsload(bcmd, 0, 5, argv) != 0) { | |
193 | usb_stop(); | |
194 | return 1; | |
195 | } | |
196 | ||
197 | /* Stop USB */ | |
198 | usb_stop(); | |
199 | return 0; | |
200 | } |