]> git.kernelconcepts.de Git - karo-tx-uboot.git/blobdiff - common/cmd_pxe.c
Merge branch 'master' of git://www.denx.de/git/u-boot-imx
[karo-tx-uboot.git] / common / cmd_pxe.c
index 59483a75c8fd7960b7dcb300a03ef5e3bebf37bf..c27ec354cc597ba808f8667943623adee23b75dd 100644 (file)
@@ -1,19 +1,9 @@
 /*
  * Copyright 2010-2011 Calxeda, Inc.
  *
- * 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 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, see <http://www.gnu.org/licenses/>.
+ * SPDX-License-Identifier:    GPL-2.0+
  */
+
 #include <common.h>
 #include <command.h>
 #include <malloc.h>
 #define MAX_TFTP_PATH_LEN 127
 
 const char *pxe_default_paths[] = {
+#ifdef CONFIG_SYS_SOC
        "default-" CONFIG_SYS_ARCH "-" CONFIG_SYS_SOC,
+#endif
        "default-" CONFIG_SYS_ARCH,
        "default",
        NULL
 };
 
+static bool is_pxe;
+
 /*
  * Like getenv, but prints an error if envvar isn't defined in the
  * environment.  It always returns what getenv does, so it can be used in
@@ -65,7 +59,7 @@ static int format_mac_pxe(char *outbuf, size_t outbuf_len)
        uchar ethaddr[6];
 
        if (outbuf_len < 21) {
-               printf("outbuf is too small (%d < 21)\n", outbuf_len);
+               printf("outbuf is too small (%zd < 21)\n", outbuf_len);
 
                return -EINVAL;
        }
@@ -92,7 +86,8 @@ static int get_bootfile_path(const char *file_path, char *bootfile_path,
        char *bootfile, *last_slash;
        size_t path_len = 0;
 
-       if (file_path[0] == '/')
+       /* Only syslinux allows absolute paths */
+       if (file_path[0] == '/' && !is_pxe)
                goto ret;
 
        bootfile = from_env("bootfile");
@@ -108,7 +103,7 @@ static int get_bootfile_path(const char *file_path, char *bootfile_path,
        path_len = (last_slash - bootfile) + 1;
 
        if (bootfile_path_size < path_len) {
-               printf("bootfile_path too small. (%d < %d)\n",
+               printf("bootfile_path too small. (%zd < %zd)\n",
                                bootfile_path_size, path_len);
 
                return -1;
@@ -122,16 +117,16 @@ static int get_bootfile_path(const char *file_path, char *bootfile_path,
        return 1;
 }
 
-static int (*do_getfile)(const char *file_path, char *file_addr);
+static int (*do_getfile)(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr);
 
-static int do_get_tftp(const char *file_path, char *file_addr)
+static int do_get_tftp(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
 {
        char *tftp_argv[] = {"tftp", NULL, NULL, NULL};
 
        tftp_argv[1] = file_addr;
        tftp_argv[2] = (void *)file_path;
 
-       if (do_tftpb(NULL, 0, 3, tftp_argv))
+       if (do_tftpb(cmdtp, 0, 3, tftp_argv))
                return -ENOENT;
 
        return 1;
@@ -139,27 +134,27 @@ static int do_get_tftp(const char *file_path, char *file_addr)
 
 static char *fs_argv[5];
 
-static int do_get_ext2(const char *file_path, char *file_addr)
+static int do_get_ext2(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
 {
 #ifdef CONFIG_CMD_EXT2
        fs_argv[0] = "ext2load";
        fs_argv[3] = file_addr;
        fs_argv[4] = (void *)file_path;
 
-       if (!do_ext2load(NULL, 0, 5, fs_argv))
+       if (!do_ext2load(cmdtp, 0, 5, fs_argv))
                return 1;
 #endif
        return -ENOENT;
 }
 
-static int do_get_fat(const char *file_path, char *file_addr)
+static int do_get_fat(cmd_tbl_t *cmdtp, const char *file_path, char *file_addr)
 {
 #ifdef CONFIG_CMD_FAT
        fs_argv[0] = "fatload";
        fs_argv[3] = file_addr;
        fs_argv[4] = (void *)file_path;
 
-       if (!do_fat_fsload(NULL, 0, 5, fs_argv))
+       if (!do_fat_fsload(cmdtp, 0, 5, fs_argv))
                return 1;
 #endif
        return -ENOENT;
@@ -173,7 +168,7 @@ static int do_get_fat(const char *file_path, char *file_addr)
  *
  * Returns 1 for success, or < 0 on error.
  */
-static int get_relfile(const char *file_path, void *file_addr)
+static int get_relfile(cmd_tbl_t *cmdtp, const char *file_path, void *file_addr)
 {
        size_t path_len;
        char relfile[MAX_TFTP_PATH_LEN+1];
@@ -202,7 +197,7 @@ static int get_relfile(const char *file_path, void *file_addr)
 
        sprintf(addr_buf, "%p", file_addr);
 
-       return do_getfile(relfile, addr_buf);
+       return do_getfile(cmdtp, relfile, addr_buf);
 }
 
 /*
@@ -212,13 +207,13 @@ static int get_relfile(const char *file_path, void *file_addr)
  *
  * Returns 1 on success, or < 0 for error.
  */
-static int get_pxe_file(const char *file_path, void *file_addr)
+static int get_pxe_file(cmd_tbl_t *cmdtp, const char *file_path, void *file_addr)
 {
        unsigned long config_file_size;
        char *tftp_filesize;
        int err;
 
-       err = get_relfile(file_path, file_addr);
+       err = get_relfile(cmdtp, file_path, file_addr);
 
        if (err < 0)
                return err;
@@ -249,7 +244,7 @@ static int get_pxe_file(const char *file_path, void *file_addr)
  *
  * Returns 1 on success or < 0 on error.
  */
-static int get_pxelinux_path(const char *file, void *pxefile_addr_r)
+static int get_pxelinux_path(cmd_tbl_t *cmdtp, const char *file, void *pxefile_addr_r)
 {
        size_t base_len = strlen(PXELINUX_DIR);
        char path[MAX_TFTP_PATH_LEN+1];
@@ -262,7 +257,7 @@ static int get_pxelinux_path(const char *file, void *pxefile_addr_r)
 
        sprintf(path, PXELINUX_DIR "%s", file);
 
-       return get_pxe_file(path, pxefile_addr_r);
+       return get_pxe_file(cmdtp, path, pxefile_addr_r);
 }
 
 /*
@@ -270,7 +265,7 @@ static int get_pxelinux_path(const char *file, void *pxefile_addr_r)
  *
  * Returns 1 on success or < 0 on error.
  */
-static int pxe_uuid_path(void *pxefile_addr_r)
+static int pxe_uuid_path(cmd_tbl_t *cmdtp, void *pxefile_addr_r)
 {
        char *uuid_str;
 
@@ -279,7 +274,7 @@ static int pxe_uuid_path(void *pxefile_addr_r)
        if (!uuid_str)
                return -ENOENT;
 
-       return get_pxelinux_path(uuid_str, pxefile_addr_r);
+       return get_pxelinux_path(cmdtp, uuid_str, pxefile_addr_r);
 }
 
 /*
@@ -288,7 +283,7 @@ static int pxe_uuid_path(void *pxefile_addr_r)
  *
  * Returns 1 on success or < 0 on error.
  */
-static int pxe_mac_path(void *pxefile_addr_r)
+static int pxe_mac_path(cmd_tbl_t *cmdtp, void *pxefile_addr_r)
 {
        char mac_str[21];
        int err;
@@ -298,7 +293,7 @@ static int pxe_mac_path(void *pxefile_addr_r)
        if (err < 0)
                return err;
 
-       return get_pxelinux_path(mac_str, pxefile_addr_r);
+       return get_pxelinux_path(cmdtp, mac_str, pxefile_addr_r);
 }
 
 /*
@@ -308,7 +303,7 @@ static int pxe_mac_path(void *pxefile_addr_r)
  *
  * Returns 1 on success or < 0 on error.
  */
-static int pxe_ipaddr_paths(void *pxefile_addr_r)
+static int pxe_ipaddr_paths(cmd_tbl_t *cmdtp, void *pxefile_addr_r)
 {
        char ip_addr[9];
        int mask_pos, err;
@@ -316,7 +311,7 @@ static int pxe_ipaddr_paths(void *pxefile_addr_r)
        sprintf(ip_addr, "%08X", ntohl(NetOurIP));
 
        for (mask_pos = 7; mask_pos >= 0;  mask_pos--) {
-               err = get_pxelinux_path(ip_addr, pxefile_addr_r);
+               err = get_pxelinux_path(cmdtp, ip_addr, pxefile_addr_r);
 
                if (err > 0)
                        return err;
@@ -367,16 +362,16 @@ do_pxe_get(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
         * Keep trying paths until we successfully get a file we're looking
         * for.
         */
-       if (pxe_uuid_path((void *)pxefile_addr_r) > 0 ||
-           pxe_mac_path((void *)pxefile_addr_r) > 0 ||
-           pxe_ipaddr_paths((void *)pxefile_addr_r) > 0) {
+       if (pxe_uuid_path(cmdtp, (void *)pxefile_addr_r) > 0 ||
+           pxe_mac_path(cmdtp, (void *)pxefile_addr_r) > 0 ||
+           pxe_ipaddr_paths(cmdtp, (void *)pxefile_addr_r) > 0) {
                printf("Config file found\n");
 
                return 0;
        }
 
        while (pxe_default_paths[i]) {
-               if (get_pxelinux_path(pxe_default_paths[i],
+               if (get_pxelinux_path(cmdtp, pxe_default_paths[i],
                                      (void *)pxefile_addr_r) > 0) {
                        printf("Config file found\n");
                        return 0;
@@ -396,7 +391,7 @@ do_pxe_get(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
  *
  * Returns 1 on success or < 0 on error.
  */
-static int get_relfile_envaddr(const char *file_path, const char *envaddr_name)
+static int get_relfile_envaddr(cmd_tbl_t *cmdtp, const char *file_path, const char *envaddr_name)
 {
        unsigned long file_addr;
        char *envaddr;
@@ -409,7 +404,7 @@ static int get_relfile_envaddr(const char *file_path, const char *envaddr_name)
        if (strict_strtoul(envaddr, 16, &file_addr) < 0)
                return -EINVAL;
 
-       return get_relfile(file_path, (void *)file_addr);
+       return get_relfile(cmdtp, file_path, (void *)file_addr);
 }
 
 /*
@@ -450,6 +445,7 @@ struct pxe_label {
        char *append;
        char *initrd;
        char *fdt;
+       int ipappend;
        int attempted;
        int localboot;
        int localboot_val;
@@ -579,11 +575,15 @@ static int label_localboot(struct pxe_label *label)
  * If the label specifies an 'append' line, its contents will overwrite that
  * of the 'bootargs' environment variable.
  */
-static int label_boot(struct pxe_label *label)
+static int label_boot(cmd_tbl_t *cmdtp, struct pxe_label *label)
 {
        char *bootm_argv[] = { "bootm", NULL, NULL, NULL, NULL };
        char initrd_str[22];
+       char mac_str[29] = "";
+       char ip_str[68] = "";
+       char *bootargs;
        int bootm_argc = 3;
+       int len = 0;
 
        label_print(label);
 
@@ -602,7 +602,7 @@ static int label_boot(struct pxe_label *label)
        }
 
        if (label->initrd) {
-               if (get_relfile_envaddr(label->initrd, "ramdisk_addr_r") < 0) {
+               if (get_relfile_envaddr(cmdtp, label->initrd, "ramdisk_addr_r") < 0) {
                        printf("Skipping %s for failure retrieving initrd\n",
                                        label->name);
                        return 1;
@@ -616,15 +616,45 @@ static int label_boot(struct pxe_label *label)
                bootm_argv[2] = "-";
        }
 
-       if (get_relfile_envaddr(label->kernel, "kernel_addr_r") < 0) {
+       if (get_relfile_envaddr(cmdtp, label->kernel, "kernel_addr_r") < 0) {
                printf("Skipping %s for failure retrieving kernel\n",
                                label->name);
                return 1;
        }
 
-       if (label->append) {
-               setenv("bootargs", label->append);
-               printf("append: %s\n", label->append);
+       if (label->ipappend & 0x1) {
+               sprintf(ip_str, " ip=%s:%s:%s:%s",
+                       getenv("ipaddr"), getenv("serverip"),
+                       getenv("gatewayip"), getenv("netmask"));
+               len += strlen(ip_str);
+       }
+
+       if (label->ipappend & 0x2) {
+               int err;
+               strcpy(mac_str, " BOOTIF=");
+               err = format_mac_pxe(mac_str + 8, sizeof(mac_str) - 8);
+               if (err < 0)
+                       mac_str[0] = '\0';
+               len += strlen(mac_str);
+       }
+
+       if (label->append)
+               len += strlen(label->append);
+
+       if (len) {
+               bootargs = malloc(len + 1);
+               if (!bootargs)
+                       return 1;
+               bootargs[0] = '\0';
+               if (label->append)
+                       strcpy(bootargs, label->append);
+               strcat(bootargs, ip_str);
+               strcat(bootargs, mac_str);
+
+               setenv("bootargs", bootargs);
+               printf("append: %s\n", bootargs);
+
+               free(bootargs);
        }
 
        bootm_argv[1] = getenv("kernel_addr_r");
@@ -646,7 +676,7 @@ static int label_boot(struct pxe_label *label)
 
        /* if fdt label is defined then get fdt from server */
        if (bootm_argv[3] && label->fdt) {
-               if (get_relfile_envaddr(label->fdt, "fdt_addr_r") < 0) {
+               if (get_relfile_envaddr(cmdtp, label->fdt, "fdt_addr_r") < 0) {
                        printf("Skipping %s for failure retrieving fdt\n",
                                        label->name);
                        return 1;
@@ -657,11 +687,11 @@ static int label_boot(struct pxe_label *label)
        if (bootm_argv[3])
                bootm_argc = 4;
 
-       do_bootm(NULL, 0, bootm_argc, bootm_argv);
+       do_bootm(cmdtp, 0, bootm_argc, bootm_argv);
 
 #ifdef CONFIG_CMD_BOOTZ
        /* Try booting a zImage if do_bootm returns */
-       do_bootz(NULL, 0, bootm_argc, bootm_argv);
+       do_bootz(cmdtp, 0, bootm_argc, bootm_argv);
 #endif
        return 1;
 }
@@ -687,6 +717,7 @@ enum token_type {
        T_INCLUDE,
        T_FDT,
        T_ONTIMEOUT,
+       T_IPAPPEND,
        T_INVALID
 };
 
@@ -716,6 +747,7 @@ static const struct token keywords[] = {
        {"include", T_INCLUDE},
        {"fdt", T_FDT},
        {"ontimeout", T_ONTIMEOUT,},
+       {"ipappend", T_IPAPPEND,},
        {NULL, T_INVALID}
 };
 
@@ -921,7 +953,7 @@ static int parse_integer(char **c, int *dst)
        return 1;
 }
 
-static int parse_pxefile_top(char *p, struct pxe_menu *cfg, int nest_level);
+static int parse_pxefile_top(cmd_tbl_t *cmdtp, char *p, struct pxe_menu *cfg, int nest_level);
 
 /*
  * Parse an include statement, and retrieve and parse the file it mentions.
@@ -931,7 +963,7 @@ static int parse_pxefile_top(char *p, struct pxe_menu *cfg, int nest_level);
  * include, nest_level has already been incremented and doesn't need to be
  * incremented here.
  */
-static int handle_include(char **c, char *base,
+static int handle_include(cmd_tbl_t *cmdtp, char **c, char *base,
                                struct pxe_menu *cfg, int nest_level)
 {
        char *include_path;
@@ -946,14 +978,14 @@ static int handle_include(char **c, char *base,
                return err;
        }
 
-       err = get_pxe_file(include_path, base);
+       err = get_pxe_file(cmdtp, include_path, base);
 
        if (err < 0) {
                printf("Couldn't retrieve %s\n", include_path);
                return err;
        }
 
-       return parse_pxefile_top(base, cfg, nest_level);
+       return parse_pxefile_top(cmdtp, base, cfg, nest_level);
 }
 
 /*
@@ -966,7 +998,7 @@ static int handle_include(char **c, char *base,
  * nest_level should be 1 when parsing the top level pxe file, 2 when parsing
  * a file it includes, 3 when parsing a file included by that file, and so on.
  */
-static int parse_menu(char **c, struct pxe_menu *cfg, char *b, int nest_level)
+static int parse_menu(cmd_tbl_t *cmdtp, char **c, struct pxe_menu *cfg, char *b, int nest_level)
 {
        struct token t;
        char *s = *c;
@@ -981,7 +1013,7 @@ static int parse_menu(char **c, struct pxe_menu *cfg, char *b, int nest_level)
                break;
 
        case T_INCLUDE:
-               err = handle_include(c, b + strlen(b) + 1, cfg,
+               err = handle_include(cmdtp, c, b + strlen(b) + 1, cfg,
                                                nest_level + 1);
                break;
 
@@ -1107,6 +1139,10 @@ static int parse_label(char **c, struct pxe_menu *cfg)
                        err = parse_integer(c, &label->localboot_val);
                        break;
 
+               case T_IPAPPEND:
+                       err = parse_integer(c, &label->ipappend);
+                       break;
+
                case T_EOL:
                        break;
 
@@ -1139,7 +1175,7 @@ static int parse_label(char **c, struct pxe_menu *cfg)
  *
  * Returns 1 on success, < 0 on error.
  */
-static int parse_pxefile_top(char *p, struct pxe_menu *cfg, int nest_level)
+static int parse_pxefile_top(cmd_tbl_t *cmdtp, char *p, struct pxe_menu *cfg, int nest_level)
 {
        struct token t;
        char *s, *b, *label_name;
@@ -1161,7 +1197,7 @@ static int parse_pxefile_top(char *p, struct pxe_menu *cfg, int nest_level)
                switch (t.type) {
                case T_MENU:
                        cfg->prompt = 1;
-                       err = parse_menu(&p, cfg, b, nest_level);
+                       err = parse_menu(cmdtp, &p, cfg, b, nest_level);
                        break;
 
                case T_TIMEOUT:
@@ -1186,7 +1222,7 @@ static int parse_pxefile_top(char *p, struct pxe_menu *cfg, int nest_level)
                        break;
 
                case T_INCLUDE:
-                       err = handle_include(&p, b + ALIGN(strlen(b), 4), cfg,
+                       err = handle_include(cmdtp, &p, b + ALIGN(strlen(b), 4), cfg,
                                                        nest_level + 1);
                        break;
 
@@ -1243,7 +1279,7 @@ static void destroy_pxe_menu(struct pxe_menu *cfg)
  * files it includes). The resulting pxe_menu struct can be free()'d by using
  * the destroy_pxe_menu() function.
  */
-static struct pxe_menu *parse_pxefile(char *menucfg)
+static struct pxe_menu *parse_pxefile(cmd_tbl_t *cmdtp, char *menucfg)
 {
        struct pxe_menu *cfg;
 
@@ -1256,7 +1292,7 @@ static struct pxe_menu *parse_pxefile(char *menucfg)
 
        INIT_LIST_HEAD(&cfg->labels);
 
-       if (parse_pxefile_top(menucfg, cfg, 1) < 0) {
+       if (parse_pxefile_top(cmdtp, menucfg, cfg, 1) < 0) {
                destroy_pxe_menu(cfg);
                return NULL;
        }
@@ -1322,7 +1358,7 @@ static struct menu *pxe_menu_to_menu(struct pxe_menu *cfg)
 /*
  * Try to boot any labels we have yet to attempt to boot.
  */
-static void boot_unattempted_labels(struct pxe_menu *cfg)
+static void boot_unattempted_labels(cmd_tbl_t *cmdtp, struct pxe_menu *cfg)
 {
        struct list_head *pos;
        struct pxe_label *label;
@@ -1331,7 +1367,7 @@ static void boot_unattempted_labels(struct pxe_menu *cfg)
                label = list_entry(pos, struct pxe_label, list);
 
                if (!label->attempted)
-                       label_boot(label);
+                       label_boot(cmdtp, label);
        }
 }
 
@@ -1347,7 +1383,7 @@ static void boot_unattempted_labels(struct pxe_menu *cfg)
  * If this function returns, there weren't any labels that successfully
  * booted, or the user interrupted the menu selection via ctrl+c.
  */
-static void handle_pxe_menu(struct pxe_menu *cfg)
+static void handle_pxe_menu(cmd_tbl_t *cmdtp, struct pxe_menu *cfg)
 {
        void *choice;
        struct menu *m;
@@ -1373,14 +1409,14 @@ static void handle_pxe_menu(struct pxe_menu *cfg)
         */
 
        if (err == 1) {
-               err = label_boot(choice);
+               err = label_boot(cmdtp, choice);
                if (!err)
                        return;
        } else if (err != -ENOENT) {
                return;
        }
 
-       boot_unattempted_labels(cfg);
+       boot_unattempted_labels(cmdtp, cfg);
 }
 
 /*
@@ -1413,14 +1449,14 @@ do_pxe_boot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                return 1;
        }
 
-       cfg = parse_pxefile((char *)(pxefile_addr_r));
+       cfg = parse_pxefile(cmdtp, (char *)(pxefile_addr_r));
 
        if (cfg == NULL) {
                printf("Error parsing config file\n");
                return 1;
        }
 
-       handle_pxe_menu(cfg);
+       handle_pxe_menu(cmdtp, cfg);
 
        destroy_pxe_menu(cfg);
 
@@ -1439,6 +1475,8 @@ int do_pxe(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
        if (argc < 2)
                return CMD_RET_USAGE;
 
+       is_pxe = true;
+
        /* drop initial "pxe" arg */
        argc--;
        argv++;
@@ -1471,6 +1509,8 @@ int do_sysboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
        char *filename;
        int prompt = 0;
 
+       is_pxe = false;
+
        if (strstr(argv[1], "-p")) {
                prompt = 1;
                argc--;
@@ -1511,12 +1551,12 @@ int do_sysboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
                return 1;
        }
 
-       if (get_pxe_file(filename, (void *)pxefile_addr_r) < 0) {
+       if (get_pxe_file(cmdtp, filename, (void *)pxefile_addr_r) < 0) {
                printf("Error reading config file\n");
                return 1;
        }
 
-       cfg = parse_pxefile((char *)(pxefile_addr_r));
+       cfg = parse_pxefile(cmdtp, (char *)(pxefile_addr_r));
 
        if (cfg == NULL) {
                printf("Error parsing config file\n");
@@ -1526,7 +1566,7 @@ int do_sysboot(cmd_tbl_t *cmdtp, int flag, int argc, char * const argv[])
        if (prompt)
                cfg->prompt = 1;
 
-       handle_pxe_menu(cfg);
+       handle_pxe_menu(cmdtp, cfg);
 
        destroy_pxe_menu(cfg);