Patch by Kenneth Johansson, 30 Jun 2003:
[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 #if (CONFIG_COMMANDS & CFG_CMD_NET)
32
33
34 extern int do_bootm (cmd_tbl_t *, int, int, char *[]);
35
36 static int netboot_common (int, cmd_tbl_t *, int , char *[]);
37
38 int do_bootp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
39 {
40         return netboot_common (BOOTP, cmdtp, argc, argv);
41 }
42
43 U_BOOT_CMD(
44         bootp,  3,      1,      do_bootp,
45         "bootp   - boot image via network using BootP/TFTP protocol\n",
46         "[loadAddress] [bootfilename]\n"
47 );
48
49 int do_tftpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
50 {
51         return netboot_common (TFTP, cmdtp, argc, argv);
52 }
53
54 U_BOOT_CMD(
55         tftpboot,       3,      1,      do_tftpb,
56         "tftpboot- boot image via network using TFTP protocol\n",
57         "[loadAddress] [bootfilename]\n"
58 );
59
60 int do_rarpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
61 {
62         return netboot_common (RARP, cmdtp, argc, argv);
63 }
64
65 U_BOOT_CMD(
66         rarpboot,       3,      1,      do_rarpb,
67         "rarpboot- boot image via network using RARP/TFTP protocol\n",
68         "[loadAddress] [bootfilename]\n"
69 );
70
71 #if (CONFIG_COMMANDS & CFG_CMD_DHCP)
72 int do_dhcp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
73 {
74         return netboot_common(DHCP, cmdtp, argc, argv);
75 }
76
77 U_BOOT_CMD(
78         dhcp,   3,      1,      do_dhcp,
79         "dhcp    - invoke DHCP client to obtain IP/boot params\n",
80         "\n"
81 );
82 #endif  /* CFG_CMD_DHCP */
83
84 static void netboot_update_env(void)
85 {
86     char tmp[16] ;
87
88     if (NetOurGatewayIP) {
89         ip_to_string (NetOurGatewayIP, tmp);
90         setenv("gatewayip", tmp);
91     }
92
93     if (NetOurSubnetMask) {
94         ip_to_string (NetOurSubnetMask, tmp);
95         setenv("netmask", tmp);
96     }
97
98     if (NetOurHostName[0])
99         setenv("hostname", NetOurHostName);
100
101     if (NetOurRootPath[0])
102         setenv("rootpath", NetOurRootPath);
103
104     if (NetOurIP) {
105         ip_to_string (NetOurIP, tmp);
106         setenv("ipaddr", tmp);
107     }
108
109     if (NetServerIP) {
110         ip_to_string (NetServerIP, tmp);
111         setenv("serverip", tmp);
112     }
113
114     if (NetOurDNSIP) {
115         ip_to_string (NetOurDNSIP, tmp);
116         setenv("dnsip", tmp);
117     }
118
119     if (NetOurNISDomain[0])
120         setenv("domain", NetOurNISDomain);
121
122 }
123 static int
124 netboot_common (int proto, cmd_tbl_t *cmdtp, int argc, char *argv[])
125 {
126         char *s;
127         int   rcode = 0;
128         int   size;
129
130         /* pre-set load_addr */
131         if ((s = getenv("loadaddr")) != NULL) {
132                 load_addr = simple_strtoul(s, NULL, 16);
133         }
134
135         switch (argc) {
136         case 1:
137                 break;
138
139         case 2: /* only one arg - accept two forms:
140                  * just load address, or just boot file name.
141                  * The latter form must be written "filename" here.
142                  */
143                 if (argv[1][0] == '"') {        /* just boot filename */
144                         copy_filename (BootFile, argv[1], sizeof(BootFile));
145                 } else {                        /* load address */
146                         load_addr = simple_strtoul(argv[1], NULL, 16);
147                 }
148                 break;
149
150         case 3: load_addr = simple_strtoul(argv[1], NULL, 16);
151                 copy_filename (BootFile, argv[2], sizeof(BootFile));
152
153                 break;
154
155         default: printf ("Usage:\n%s\n", cmdtp->usage);
156                 return 1;
157         }
158
159         if ((size = NetLoop(proto)) < 0)
160                 return 1;
161
162         /* NetLoop ok, update environment */
163         netboot_update_env();
164
165         /* done if no file was loaded (no errors though) */
166         if (size == 0)
167                 return 0;
168
169         /* flush cache */
170         flush_cache(load_addr, size);
171
172         /* Loading ok, check if we should attempt an auto-start */
173         if (((s = getenv("autostart")) != NULL) && (strcmp(s,"yes") == 0)) {
174                 char *local_args[2];
175                 local_args[0] = argv[0];
176                 local_args[1] = NULL;
177
178                 printf ("Automatic boot of image at addr 0x%08lX ...\n",
179                         load_addr);
180                 rcode = do_bootm (cmdtp, 0, 1, local_args);
181         }
182
183 #ifdef CONFIG_AUTOSCRIPT
184         if (((s = getenv("autoscript")) != NULL) && (strcmp(s,"yes") == 0)) {
185                 printf("Running autoscript at addr 0x%08lX ...\n", load_addr);
186                 rcode = autoscript (load_addr);
187         }
188 #endif
189         return rcode;
190 }
191
192 #if (CONFIG_COMMANDS & CFG_CMD_PING)
193 int do_ping (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
194 {
195         if (argc < 2)
196                 return -1;
197
198         NetPingIP = string_to_ip(argv[1]);
199         if (NetPingIP == 0) {
200                 printf ("Usage:\n%s\n", cmdtp->usage);
201                 return -1;
202         }
203
204         if (NetLoop(PING) < 0) {
205                 printf("ping failed; host %s is not alive\n", argv[1]);
206                 return 1;
207         }
208
209         printf("host %s is alive\n", argv[1]);
210
211         return 0;
212 }
213 #endif  /* CFG_CMD_PING */
214
215 #endif  /* CFG_CMD_NET */