| /* |
| * (C) Copyright 2011 Quantenna Communications Inc. |
| * |
| * See file CREDITS for list of people who contributed to this |
| * project. |
| * |
| * This program is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU General Public License as |
| * published by the Free Software Foundation; either version 2 of |
| * the License, or (at your option) any later version. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License |
| * along with this program; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, |
| * MA 02111-1307 USA |
| */ |
| |
| #include <common.h> |
| #include <config.h> |
| #include <command.h> |
| #include <asm/types.h> |
| #include <net.h> |
| #include "ruby.h" |
| |
| #define UBOOT_ENV_IPADDR "ipaddr" |
| #define UBOOT_ENV_SERVERIP "serverip" |
| #define UBOOT_ENV_BOOTFILE "bootfile" |
| #define UBOOT_ENV_NETRETRY "netretry" |
| #define UBOOT_ENV_STR_MAX 32 |
| |
| int do_bootp(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); |
| int do_tftpb(cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]); |
| |
| static char prev_netretry[UBOOT_ENV_STR_MAX] = {0}; |
| static char prev_serverip[UBOOT_ENV_STR_MAX] = {0}; |
| static char prev_bootfile[UBOOT_ENV_STR_MAX] = {0}; |
| static char prev_ipaddr[UBOOT_ENV_STR_MAX] = {0}; |
| |
| static int get_stage2_bootp(void) |
| { |
| setenv(UBOOT_ENV_IPADDR, NULL); |
| setenv(UBOOT_ENV_SERVERIP, NULL); |
| setenv(UBOOT_ENV_BOOTFILE, prev_bootfile); |
| return do_bootp(NULL, 0, 1, NULL); |
| } |
| |
| static int get_stage2_tftp(void) |
| { |
| setenv(UBOOT_ENV_IPADDR, prev_ipaddr); |
| setenv(UBOOT_ENV_SERVERIP, prev_serverip); |
| setenv(UBOOT_ENV_BOOTFILE, prev_bootfile); |
| return do_tftpb(NULL, 0, 1, NULL); |
| } |
| |
| int get_stage2_image(uint32_t method_req, uint32_t *method_used) |
| { |
| int ret; |
| const char *prev_netretry_env; |
| const char *p; |
| |
| /* |
| * temporary copies of ipaddr etc |
| * bootp needs these to be blank, tftp needs these set |
| */ |
| if ((p = getenv(UBOOT_ENV_IPADDR)) != NULL) { |
| strcpy(prev_ipaddr, p); |
| } |
| if ((p = getenv(UBOOT_ENV_SERVERIP)) != NULL) { |
| strcpy(prev_serverip, p); |
| } |
| if ((p = getenv(UBOOT_ENV_BOOTFILE)) != NULL) { |
| strcpy(prev_bootfile, p); |
| } |
| if ((prev_netretry_env = getenv(UBOOT_ENV_NETRETRY)) != NULL) { |
| strcpy(prev_netretry, prev_netretry_env); |
| } |
| |
| switch (method_req) { |
| case RUBY_BOOT_METHOD_BOOTP: |
| ret = get_stage2_bootp(); |
| if (ret == 0) { |
| *method_used = RUBY_BOOT_METHOD_BOOTP; |
| } |
| break; |
| case RUBY_BOOT_METHOD_TFTP: |
| ret = get_stage2_tftp(); |
| if (ret == 0) { |
| *method_used = RUBY_BOOT_METHOD_TFTP; |
| } |
| break; |
| case RUBY_BOOT_METHOD_TRYLOOP: |
| default: |
| setenv(UBOOT_ENV_NETRETRY, "no"); |
| while (1) { |
| ret = get_stage2_tftp(); |
| if (ret == 0) { |
| *method_used = RUBY_BOOT_METHOD_TFTP; |
| break; |
| } else if (had_ctrlc()) { |
| ret = -1; |
| break; |
| } |
| |
| ret = get_stage2_bootp(); |
| if (ret == 0) { |
| *method_used = RUBY_BOOT_METHOD_BOOTP; |
| break; |
| } else if (had_ctrlc()) { |
| ret = -1; |
| break; |
| } |
| } |
| } |
| |
| setenv(UBOOT_ENV_NETRETRY, prev_netretry_env ? prev_netretry : NULL); |
| |
| return ret; |
| } |
| |
| void qtn_parse_early_flash_config(int overwrite) |
| { |
| struct early_flash_config *ef = get_early_flash_config(TEXT_BASE); |
| char tmp[20]; |
| if (ef->ipaddr) { |
| if (overwrite || getenv(UBOOT_ENV_IPADDR) == NULL) { |
| ip_to_string(ef->ipaddr, tmp); |
| setenv(UBOOT_ENV_IPADDR, tmp); |
| printf("Using '%s' from early config: %s\n", UBOOT_ENV_IPADDR, getenv(UBOOT_ENV_IPADDR)); |
| } |
| } |
| if (ef->serverip) { |
| if (overwrite || getenv(UBOOT_ENV_SERVERIP) == NULL) { |
| ip_to_string(ef->serverip, tmp); |
| setenv(UBOOT_ENV_SERVERIP, tmp); |
| printf("Using '%s' from early config: %s\n", UBOOT_ENV_SERVERIP, getenv(UBOOT_ENV_SERVERIP)); |
| } |
| } |
| } |
| |
| #ifndef RUBY_MINI |
| |
| static const char *qtn_netboot_method_str(uint32_t method) { |
| switch (method) { |
| case RUBY_BOOT_METHOD_TFTP: |
| return "tftp"; |
| case RUBY_BOOT_METHOD_BOOTP: |
| return "bootp"; |
| case RUBY_BOOT_METHOD_TRYLOOP: |
| return "loop"; |
| default: |
| return "unknown"; |
| } |
| } |
| |
| int do_qtn_netboot (cmd_tbl_t *cmdtp, int flag, int argc, char *argv[]) |
| { |
| uint32_t method_req = RUBY_BOOT_METHOD_TRYLOOP; |
| uint32_t this_method_used = 0; |
| int run = 1; |
| |
| if (argc >= 2) { |
| uint32_t i; |
| for (i = 0; i < RUBY_BOOT_METHOD_MAX; i++) { |
| if (strcmp(argv[1], qtn_netboot_method_str(i)) == 0) { |
| method_req = i; |
| } |
| } |
| if (strcmp(argv[1], "tftps1") == 0) { |
| method_req = RUBY_BOOT_METHOD_TFTP; |
| qtn_parse_early_flash_config(1); |
| } else if (strcmp(argv[1], "stage1") == 0) { |
| struct early_flash_config *ef = get_early_flash_config(TEXT_BASE); |
| method_req = ef->method; |
| qtn_parse_early_flash_config(1); |
| } else if (strcmp(argv[1], "loops1") == 0) { |
| method_req = RUBY_BOOT_METHOD_TRYLOOP; |
| qtn_parse_early_flash_config(1); |
| } else if (strcmp(argv[1], "loadvars") == 0) { |
| run = 0; |
| qtn_parse_early_flash_config(1); |
| } |
| } |
| |
| if (run) { |
| return get_stage2_image(method_req, &this_method_used); |
| } else { |
| return 0; |
| } |
| } |
| |
| U_BOOT_CMD(qtn_netboot, CONFIG_SYS_MAXARGS, 0, do_qtn_netboot, |
| "boot with tftp/bootp loop", |
| "Optional argument specifies the method. Alternatives are:\n" |
| " tftp - use tftp to load image\n" |
| " bootp - use bootp to load image\n" |
| " loop - use tftp/bootp loop, starting with tftp\n" |
| " tftps1 - use 'ipaddr' and 'serverip' from start of text area\n" |
| " (possibly set by the 1st stage bootloader, or\n" |
| " hard-coded into the image), then use tftp to load image\n" |
| " loops1 - like tftps1, but uses tftp/bootp loop \n" |
| " stage1 - like tftps1, but uses the method specified in start\n" |
| " of text area\n" |
| " loadvars - load 'ipaddr' 'serverip' from text area, without loading\n" |
| ); |
| |
| #endif |