[new uImage] Add new uImage format support to autoscript routine
[platform/kernel/u-boot.git] / common / cmd_net.c
1 /*
2  * (C) Copyright 2000
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * See file CREDITS for list of people who contributed to this
6  * project.
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License as
10  * published by the Free Software Foundation; either version 2 of
11  * the License, or (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21  * MA 02111-1307 USA
22  */
23
24 /*
25  * Boot support
26  */
27 #include <common.h>
28 #include <command.h>
29 #include <net.h>
30
31 extern int do_bootm (cmd_tbl_t *, int, int, char *[]);
32
33 static int netboot_common (proto_t, cmd_tbl_t *, int , char *[]);
34
35 int do_bootp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
36 {
37         return netboot_common (BOOTP, cmdtp, argc, argv);
38 }
39
40 U_BOOT_CMD(
41         bootp,  3,      1,      do_bootp,
42         "bootp\t- boot image via network using BootP/TFTP protocol\n",
43         "[loadAddress] [bootfilename]\n"
44 );
45
46 int do_tftpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
47 {
48         return netboot_common (TFTP, cmdtp, argc, argv);
49 }
50
51 U_BOOT_CMD(
52         tftpboot,       3,      1,      do_tftpb,
53         "tftpboot- boot image via network using TFTP protocol\n",
54         "[loadAddress] [[hostIPaddr:]bootfilename]\n"
55 );
56
57 int do_rarpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
58 {
59         return netboot_common (RARP, cmdtp, argc, argv);
60 }
61
62 U_BOOT_CMD(
63         rarpboot,       3,      1,      do_rarpb,
64         "rarpboot- boot image via network using RARP/TFTP protocol\n",
65         "[loadAddress] [bootfilename]\n"
66 );
67
68 #if defined(CONFIG_CMD_DHCP)
69 int do_dhcp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
70 {
71         return netboot_common(DHCP, cmdtp, argc, argv);
72 }
73
74 U_BOOT_CMD(
75         dhcp,   3,      1,      do_dhcp,
76         "dhcp\t- invoke DHCP client to obtain IP/boot params\n",
77         "\n"
78 );
79 #endif
80
81 #if defined(CONFIG_CMD_NFS)
82 int do_nfs (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
83 {
84         return netboot_common(NFS, cmdtp, argc, argv);
85 }
86
87 U_BOOT_CMD(
88         nfs,    3,      1,      do_nfs,
89         "nfs\t- boot image via network using NFS protocol\n",
90         "[loadAddress] [[hostIPaddr:]bootfilename]\n"
91 );
92 #endif
93
94 static void netboot_update_env (void)
95 {
96         char tmp[22];
97
98         if (NetOurGatewayIP) {
99                 ip_to_string (NetOurGatewayIP, tmp);
100                 setenv ("gatewayip", tmp);
101         }
102
103         if (NetOurSubnetMask) {
104                 ip_to_string (NetOurSubnetMask, tmp);
105                 setenv ("netmask", tmp);
106         }
107
108         if (NetOurHostName[0])
109                 setenv ("hostname", NetOurHostName);
110
111         if (NetOurRootPath[0])
112                 setenv ("rootpath", NetOurRootPath);
113
114         if (NetOurIP) {
115                 ip_to_string (NetOurIP, tmp);
116                 setenv ("ipaddr", tmp);
117         }
118
119         if (NetServerIP) {
120                 ip_to_string (NetServerIP, tmp);
121                 setenv ("serverip", tmp);
122         }
123
124         if (NetOurDNSIP) {
125                 ip_to_string (NetOurDNSIP, tmp);
126                 setenv ("dnsip", tmp);
127         }
128 #if defined(CONFIG_BOOTP_DNS2)
129         if (NetOurDNS2IP) {
130                 ip_to_string (NetOurDNS2IP, tmp);
131                 setenv ("dnsip2", tmp);
132         }
133 #endif
134         if (NetOurNISDomain[0])
135                 setenv ("domain", NetOurNISDomain);
136
137 #if defined(CONFIG_CMD_SNTP) \
138     && defined(CONFIG_BOOTP_TIMEOFFSET)
139         if (NetTimeOffset) {
140                 sprintf (tmp, "%d", NetTimeOffset);
141                 setenv ("timeoffset", tmp);
142         }
143 #endif
144 #if defined(CONFIG_CMD_SNTP) \
145     && defined(CONFIG_BOOTP_NTPSERVER)
146         if (NetNtpServerIP) {
147                 ip_to_string (NetNtpServerIP, tmp);
148                 setenv ("ntpserverip", tmp);
149         }
150 #endif
151 }
152
153 static int
154 netboot_common (proto_t proto, cmd_tbl_t *cmdtp, int argc, char *argv[])
155 {
156         char *s;
157         int   rcode = 0;
158         int   size;
159
160         /* pre-set load_addr */
161         if ((s = getenv("loadaddr")) != NULL) {
162                 load_addr = simple_strtoul(s, NULL, 16);
163         }
164
165         switch (argc) {
166         case 1:
167                 break;
168
169         case 2: /* only one arg - accept two forms:
170                  * just load address, or just boot file name.
171                  * The latter form must be written "filename" here.
172                  */
173                 if (argv[1][0] == '"') {        /* just boot filename */
174                         copy_filename (BootFile, argv[1], sizeof(BootFile));
175                 } else {                        /* load address */
176                         load_addr = simple_strtoul(argv[1], NULL, 16);
177                 }
178                 break;
179
180         case 3: load_addr = simple_strtoul(argv[1], NULL, 16);
181                 copy_filename (BootFile, argv[2], sizeof(BootFile));
182
183                 break;
184
185         default: printf ("Usage:\n%s\n", cmdtp->usage);
186                 show_boot_progress (-80);
187                 return 1;
188         }
189
190         show_boot_progress (80);
191         if ((size = NetLoop(proto)) < 0) {
192                 show_boot_progress (-81);
193                 return 1;
194         }
195
196         show_boot_progress (81);
197         /* NetLoop ok, update environment */
198         netboot_update_env();
199
200         /* done if no file was loaded (no errors though) */
201         if (size == 0) {
202                 show_boot_progress (-82);
203                 return 0;
204         }
205
206         /* flush cache */
207         flush_cache(load_addr, size);
208
209         /* Loading ok, check if we should attempt an auto-start */
210         if (((s = getenv("autostart")) != NULL) && (strcmp(s,"yes") == 0)) {
211                 char *local_args[2];
212                 local_args[0] = argv[0];
213                 local_args[1] = NULL;
214
215                 printf ("Automatic boot of image at addr 0x%08lX ...\n",
216                         load_addr);
217                 show_boot_progress (82);
218                 rcode = do_bootm (cmdtp, 0, 1, local_args);
219         }
220
221 #ifdef CONFIG_AUTOSCRIPT
222         if (((s = getenv("autoscript")) != NULL) && (strcmp(s,"yes") == 0)) {
223                 printf ("Running autoscript at addr 0x%08lX", load_addr);
224
225                 s = getenv ("autoscript_uname");
226                 if (s)
227                         printf (":%s ...\n", s);
228                 else
229                         puts (" ...\n");
230
231                 show_boot_progress (83);
232                 rcode = autoscript (load_addr, s);
233         }
234 #endif
235         if (rcode < 0)
236                 show_boot_progress (-83);
237         else
238                 show_boot_progress (84);
239         return rcode;
240 }
241
242 #if defined(CONFIG_CMD_PING)
243 int do_ping (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
244 {
245         if (argc < 2)
246                 return -1;
247
248         NetPingIP = string_to_ip(argv[1]);
249         if (NetPingIP == 0) {
250                 printf ("Usage:\n%s\n", cmdtp->usage);
251                 return -1;
252         }
253
254         if (NetLoop(PING) < 0) {
255                 printf("ping failed; host %s is not alive\n", argv[1]);
256                 return 1;
257         }
258
259         printf("host %s is alive\n", argv[1]);
260
261         return 0;
262 }
263
264 U_BOOT_CMD(
265         ping,   2,      1,      do_ping,
266         "ping\t- send ICMP ECHO_REQUEST to network host\n",
267         "pingAddress\n"
268 );
269 #endif
270
271 #if defined(CONFIG_CMD_CDP)
272
273 static void cdp_update_env(void)
274 {
275         char tmp[16];
276
277         if (CDPApplianceVLAN != htons(-1)) {
278                 printf("CDP offered appliance VLAN %d\n", ntohs(CDPApplianceVLAN));
279                 VLAN_to_string(CDPApplianceVLAN, tmp);
280                 setenv("vlan", tmp);
281                 NetOurVLAN = CDPApplianceVLAN;
282         }
283
284         if (CDPNativeVLAN != htons(-1)) {
285                 printf("CDP offered native VLAN %d\n", ntohs(CDPNativeVLAN));
286                 VLAN_to_string(CDPNativeVLAN, tmp);
287                 setenv("nvlan", tmp);
288                 NetOurNativeVLAN = CDPNativeVLAN;
289         }
290
291 }
292
293 int do_cdp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
294 {
295         int r;
296
297         r = NetLoop(CDP);
298         if (r < 0) {
299                 printf("cdp failed; perhaps not a CISCO switch?\n");
300                 return 1;
301         }
302
303         cdp_update_env();
304
305         return 0;
306 }
307
308 U_BOOT_CMD(
309         cdp,    1,      1,      do_cdp,
310         "cdp\t- Perform CDP network configuration\n",
311 );
312 #endif
313
314 #if defined(CONFIG_CMD_SNTP)
315 int do_sntp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
316 {
317         char *toff;
318
319         if (argc < 2) {
320                 NetNtpServerIP = getenv_IPaddr ("ntpserverip");
321                 if (NetNtpServerIP == 0) {
322                         printf ("ntpserverip not set\n");
323                         return (1);
324                 }
325         } else {
326                 NetNtpServerIP = string_to_ip(argv[1]);
327                 if (NetNtpServerIP == 0) {
328                         printf ("Bad NTP server IP address\n");
329                         return (1);
330                 }
331         }
332
333         toff = getenv ("timeoffset");
334         if (toff == NULL) NetTimeOffset = 0;
335         else NetTimeOffset = simple_strtol (toff, NULL, 10);
336
337         if (NetLoop(SNTP) < 0) {
338                 printf("SNTP failed: host %s not responding\n", argv[1]);
339                 return 1;
340         }
341
342         return 0;
343 }
344
345 U_BOOT_CMD(
346         sntp,   2,      1,      do_sntp,
347         "sntp\t- synchronize RTC via network\n",
348         "[NTP server IP]\n"
349 );
350 #endif