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 <cmd_net.h>
  30 #include <net.h>
  31
  32 #if (CONFIG_COMMANDS & CFG_CMD_NET)
  33
  34 # if (CONFIG_COMMANDS & CFG_CMD_AUTOSCRIPT)
  35 # include <cmd_autoscript.h>
  36 # endif
  37
  38 extern int do_bootm (cmd_tbl_t *, int, int, char *[]);
  39
  40 static int netboot_common (int, cmd_tbl_t *, int , char *[]);
  41
  42 int do_bootp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  43 {
  44         return netboot_common (BOOTP, cmdtp, argc, argv);
  45 }
  46
  47 int do_tftpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  48 {
  49         return netboot_common (TFTP, cmdtp, argc, argv);
  50 }
  51
  52 int do_rarpb (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  53 {
  54         return netboot_common (RARP, cmdtp, argc, argv);
  55 }
  56
  57 #if (CONFIG_COMMANDS & CFG_CMD_DHCP)
  58 int do_dhcp (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[])
  59 {
  60         return netboot_common(DHCP, cmdtp, argc, argv);
  61 }
  62 #endif  /* CFG_CMD_DHCP */
  63
  64 static void netboot_update_env(void)
  65 {
  66     char tmp[16] ;
  67
  68     if (NetOurGatewayIP) {
  69         ip_to_string (NetOurGatewayIP, tmp);
  70         setenv("gatewayip", tmp);
  71     }
  72
  73     if (NetOurSubnetMask) {
  74         ip_to_string (NetOurSubnetMask, tmp);
  75         setenv("netmask", tmp);
  76     }
  77
  78     if (NetOurHostName[0])
  79         setenv("hostname", NetOurHostName);
  80
  81     if (NetOurRootPath[0])
  82         setenv("rootpath", NetOurRootPath);
  83
  84     if (NetOurIP) {
  85         ip_to_string (NetOurIP, tmp);
  86         setenv("ipaddr", tmp);
  87     }
  88
  89     if (NetServerIP) {
  90         ip_to_string (NetServerIP, tmp);
  91         setenv("serverip", tmp);
  92     }
  93
  94     if (NetOurDNSIP) {
  95         ip_to_string (NetOurDNSIP, tmp);
  96         setenv("dnsip", tmp);
  97     }
  98 }
  99 static int
 100 netboot_common (int proto, cmd_tbl_t *cmdtp, int argc, char *argv[])
 101 {
 102         char *s;
 103         int   rcode = 0;
 104         int   size;
 105
 106         /* pre-set load_addr */
 107         if ((s = getenv("loadaddr")) != NULL) {
 108                 load_addr = simple_strtoul(s, NULL, 16);
 109         }
 110
 111         switch (argc) {
 112         case 1:
 113                 break;
 114
 115         case 2: /* only one arg - accept two forms:
 116                  * just load address, or just boot file name.
 117                  * The latter form must be written "filename" here.
 118                  */
 119                 if (argv[1][0] == '"') {        /* just boot filename */
 120                         copy_filename (BootFile, argv[1], sizeof(BootFile));
 121                 } else {                        /* load address */
 122                         load_addr = simple_strtoul(argv[1], NULL, 16);
 123                 }
 124                 break;
 125
 126         case 3: load_addr = simple_strtoul(argv[1], NULL, 16);
 127                 copy_filename (BootFile, argv[2], sizeof(BootFile));
 128
 129                 break;
 130
 131         default: printf ("Usage:\n%s\n", cmdtp->usage);
 132                 return 1;
 133         }
 134
 135         if ((size = NetLoop(proto)) == 0)
 136                 return 1;
 137
 138         /* NetLoop ok, update environment */
 139         netboot_update_env();
 140
 141         /* flush cache */
 142         flush_cache(load_addr, size);
 143
 144         /* Loading ok, check if we should attempt an auto-start */
 145         if (((s = getenv("autostart")) != NULL) && (strcmp(s,"yes") == 0)) {
 146                 char *local_args[2];
 147                 local_args[0] = argv[0];
 148                 local_args[1] = NULL;
 149
 150                 printf ("Automatic boot of image at addr 0x%08lX ...\n",
 151                         load_addr);
 152                 rcode = do_bootm (cmdtp, 0, 1, local_args);
 153         }
 154
 155 #ifdef CONFIG_AUTOSCRIPT
 156         if (((s = getenv("autoscript")) != NULL) && (strcmp(s,"yes") == 0)) {
 157                 printf("Running autoscript at addr 0x%08lX ...\n", load_addr);
 158                 rcode = autoscript (load_addr);
 159         }
 160 #endif
 161         return rcode;
 162 }
 163
 164 #endif  /* CFG_CMD_NET */