Skip to content
Snippets Groups Projects
Select Git revision
  • master default protected
  • stable
  • extraversion
  • early-display
  • variant-emmc-nvme-boot
  • 2024-07-19
  • 2024-06-30
  • 2023-10-18
  • 2023-10-10
  • 2023-07-04
  • 2023-01-25
  • v3
  • variant-emmc-nvme-boot
  • 2020-06-01
14 results

update.c

Blame
    • Tom Rini's avatar
      83d290c5
      SPDX: Convert all of our single license tags to Linux Kernel style · 83d290c5
      Tom Rini authored
      
      When U-Boot started using SPDX tags we were among the early adopters and
      there weren't a lot of other examples to borrow from.  So we picked the
      area of the file that usually had a full license text and replaced it
      with an appropriate SPDX-License-Identifier: entry.  Since then, the
      Linux Kernel has adopted SPDX tags and they place it as the very first
      line in a file (except where shebangs are used, then it's second line)
      and with slightly different comment styles than us.
      
      In part due to community overlap, in part due to better tag visibility
      and in part for other minor reasons, switch over to that style.
      
      This commit changes all instances where we have a single declared
      license in the tag as both the before and after are identical in tag
      contents.  There's also a few places where I found we did not have a tag
      and have introduced one.
      
      Signed-off-by: default avatarTom Rini <trini@konsulko.com>
      83d290c5
      History
      SPDX: Convert all of our single license tags to Linux Kernel style
      Tom Rini authored
      
      When U-Boot started using SPDX tags we were among the early adopters and
      there weren't a lot of other examples to borrow from.  So we picked the
      area of the file that usually had a full license text and replaced it
      with an appropriate SPDX-License-Identifier: entry.  Since then, the
      Linux Kernel has adopted SPDX tags and they place it as the very first
      line in a file (except where shebangs are used, then it's second line)
      and with slightly different comment styles than us.
      
      In part due to community overlap, in part due to better tag visibility
      and in part for other minor reasons, switch over to that style.
      
      This commit changes all instances where we have a single declared
      license in the tag as both the before and after are identical in tag
      contents.  There's also a few places where I found we did not have a tag
      and have introduced one.
      
      Signed-off-by: default avatarTom Rini <trini@konsulko.com>
    update.c 8.00 KiB
    // SPDX-License-Identifier: GPL-2.0+
    /*
     * (C) Copyright 2008 Semihalf
     *
     * Written by: Rafal Czubak <rcz@semihalf.com>
     *             Bartlomiej Sieka <tur@semihalf.com>
     */
    
    #include <common.h>
    
    #if !(defined(CONFIG_FIT) && defined(CONFIG_OF_LIBFDT))
    #error "CONFIG_FIT and CONFIG_OF_LIBFDT are required for auto-update feature"
    #endif
    
    #if defined(CONFIG_UPDATE_TFTP) && !defined(CONFIG_MTD_NOR_FLASH)
    #error "CONFIG_UPDATE_TFTP and !CONFIG_MTD_NOR_FLASH needed for legacy behaviour"
    #endif
    
    #include <command.h>
    #include <flash.h>
    #include <net.h>
    #include <net/tftp.h>
    #include <malloc.h>
    #include <dfu.h>
    #include <errno.h>
    #include <mtd/cfi_flash.h>
    
    /* env variable holding the location of the update file */
    #define UPDATE_FILE_ENV		"updatefile"
    
    /* set configuration defaults if needed */
    #ifndef CONFIG_UPDATE_LOAD_ADDR
    #define CONFIG_UPDATE_LOAD_ADDR	0x100000
    #endif
    
    #ifndef CONFIG_UPDATE_TFTP_MSEC_MAX
    #define CONFIG_UPDATE_TFTP_MSEC_MAX	100
    #endif
    
    #ifndef CONFIG_UPDATE_TFTP_CNT_MAX
    #define CONFIG_UPDATE_TFTP_CNT_MAX	0
    #endif
    
    extern ulong tftp_timeout_ms;
    extern int tftp_timeout_count_max;
    extern ulong load_addr;
    #ifdef CONFIG_MTD_NOR_FLASH
    extern flash_info_t flash_info[];
    static uchar *saved_prot_info;
    #endif
    static int update_load(char *filename, ulong msec_max, int cnt_max, ulong addr)
    {
    	int size, rv;
    	ulong saved_timeout_msecs;
    	int saved_timeout_count;
    	char *saved_netretry, *saved_bootfile;
    
    	rv = 0;
    	/* save used globals and env variable */
    	saved_timeout_msecs = tftp_timeout_ms;
    	saved_timeout_count = tftp_timeout_count_max;
    	saved_netretry = strdup(env_get("netretry"));
    	saved_bootfile = strdup(net_boot_file_name);
    
    	/* set timeouts for auto-update */
    	tftp_timeout_ms = msec_max;
    	tftp_timeout_count_max = cnt_max;
    
    	/* we don't want to retry the connection if errors occur */
    	env_set("netretry", "no");
    
    	/* download the update file */
    	load_addr = addr;
    	copy_filename(net_boot_file_name, filename, sizeof(net_boot_file_name));
    	size = net_loop(TFTPGET);
    
    	if (size < 0)
    		rv = 1;
    	else if (size > 0)
    		flush_cache(addr, size);
    
    	/* restore changed globals and env variable */
    	tftp_timeout_ms = saved_timeout_msecs;
    	tftp_timeout_count_max = saved_timeout_count;
    
    	env_set("netretry", saved_netretry);
    	if (saved_netretry != NULL)
    		free(saved_netretry);
    
    	if (saved_bootfile != NULL) {
    		copy_filename(net_boot_file_name, saved_bootfile,
    			      sizeof(net_boot_file_name));
    		free(saved_bootfile);
    	}
    
    	return rv;
    }
    
    #ifdef CONFIG_MTD_NOR_FLASH
    static int update_flash_protect(int prot, ulong addr_first, ulong addr_last)
    {
    	uchar *sp_info_ptr;
    	ulong s;
    	int i, bank, cnt;
    	flash_info_t *info;
    
    	sp_info_ptr = NULL;
    
    	if (prot == 0) {
    		saved_prot_info =
    			calloc(CONFIG_SYS_MAX_FLASH_BANKS * CONFIG_SYS_MAX_FLASH_SECT, 1);
    		if (!saved_prot_info)
    			return 1;
    	}
    
    	for (bank = 0; bank < CONFIG_SYS_MAX_FLASH_BANKS; ++bank) {
    		cnt = 0;
    		info = &flash_info[bank];
    
    		/* Nothing to do if the bank doesn't exist */
    		if (info->sector_count == 0)
    			return 0;
    
    		/* Point to current bank protection information */
    		sp_info_ptr = saved_prot_info + (bank * CONFIG_SYS_MAX_FLASH_SECT);
    
    		/*
    		 * Adjust addr_first or addr_last if we are on bank boundary.
    		 * Address space between banks must be continuous for other
    		 * flash functions (like flash_sect_erase or flash_write) to
    		 * succeed. Banks must also be numbered in correct order,
    		 * according to increasing addresses.
    		 */
    		if (addr_last > info->start[0] + info->size - 1)
    			addr_last = info->start[0] + info->size - 1;
    		if (addr_first < info->start[0])
    			addr_first = info->start[0];
    
    		for (i = 0; i < info->sector_count; i++) {
    			/* Save current information about protected sectors */
    			if (prot == 0) {
    				s = info->start[i];
    				if ((s >= addr_first) && (s <= addr_last))
    					sp_info_ptr[i] = info->protect[i];
    
    			}
    
    			/* Protect/unprotect sectors */
    			if (sp_info_ptr[i] == 1) {
    #if defined(CONFIG_SYS_FLASH_PROTECTION)
    				if (flash_real_protect(info, i, prot))
    					return 1;
    #else
    				info->protect[i] = prot;
    #endif
    				cnt++;
    			}
    		}
    
    		if (cnt) {
    			printf("%sProtected %d sectors\n",
    						prot ? "": "Un-", cnt);
    		}
    	}
    
    	if((prot == 1) && saved_prot_info)
    		free(saved_prot_info);
    
    	return 0;
    }
    #endif
    
    static int update_flash(ulong addr_source, ulong addr_first, ulong size)
    {
    #ifdef CONFIG_MTD_NOR_FLASH
    	ulong addr_last = addr_first + size - 1;
    
    	/* round last address to the sector boundary */
    	if (flash_sect_roundb(&addr_last) > 0)
    		return 1;
    
    	if (addr_first >= addr_last) {
    		printf("Error: end address exceeds addressing space\n");
    		return 1;
    	}
    
    	/* remove protection on processed sectors */
    	if (update_flash_protect(0, addr_first, addr_last) > 0) {
    		printf("Error: could not unprotect flash sectors\n");
    		return 1;
    	}
    
    	printf("Erasing 0x%08lx - 0x%08lx", addr_first, addr_last);
    	if (flash_sect_erase(addr_first, addr_last) > 0) {
    		printf("Error: could not erase flash\n");
    		return 1;
    	}
    
    	printf("Copying to flash...");
    	if (flash_write((char *)addr_source, addr_first, size) > 0) {
    		printf("Error: could not copy to flash\n");
    		return 1;
    	}
    	printf("done\n");
    
    	/* enable protection on processed sectors */
    	if (update_flash_protect(1, addr_first, addr_last) > 0) {
    		printf("Error: could not protect flash sectors\n");
    		return 1;
    	}
    #endif
    	return 0;
    }
    
    static int update_fit_getparams(const void *fit, int noffset, ulong *addr,
    						ulong *fladdr, ulong *size)
    {
    	const void *data;
    
    	if (fit_image_get_data(fit, noffset, &data, (size_t *)size))
    		return 1;
    
    	if (fit_image_get_load(fit, noffset, (ulong *)fladdr))
    		return 1;
    
    	*addr = (ulong)data;
    
    	return 0;
    }
    
    int update_tftp(ulong addr, char *interface, char *devstring)
    {
    	char *filename, *env_addr, *fit_image_name;
    	ulong update_addr, update_fladdr, update_size;
    	int images_noffset, ndepth, noffset;
    	bool update_tftp_dfu;
    	int ret = 0;
    	void *fit;
    
    	if (interface == NULL && devstring == NULL) {
    		update_tftp_dfu = false;
    	} else if (interface && devstring) {
    		update_tftp_dfu = true;
    	} else {
    		pr_err("Interface: %s and devstring: %s not supported!\n",
    		      interface, devstring);
    		return -EINVAL;
    	}
    
    	/* use already present image */
    	if (addr)
    		goto got_update_file;
    
    	printf("Auto-update from TFTP: ");
    
    	/* get the file name of the update file */
    	filename = env_get(UPDATE_FILE_ENV);
    	if (filename == NULL) {
    		printf("failed, env. variable '%s' not found\n",
    							UPDATE_FILE_ENV);
    		return 1;
    	}
    
    	printf("trying update file '%s'\n", filename);
    
    	/* get load address of downloaded update file */
    	env_addr = env_get("loadaddr");
    	if (env_addr)
    		addr = simple_strtoul(env_addr, NULL, 16);
    	else
    		addr = CONFIG_UPDATE_LOAD_ADDR;
    
    
    	if (update_load(filename, CONFIG_UPDATE_TFTP_MSEC_MAX,
    					CONFIG_UPDATE_TFTP_CNT_MAX, addr)) {
    		printf("Can't load update file, aborting auto-update\n");
    		return 1;
    	}
    
    got_update_file:
    	fit = (void *)addr;
    
    	if (!fit_check_format((void *)fit)) {
    		printf("Bad FIT format of the update file, aborting "
    							"auto-update\n");
    		return 1;
    	}
    
    	/* process updates */
    	images_noffset = fdt_path_offset(fit, FIT_IMAGES_PATH);
    
    	ndepth = 0;
    	noffset = fdt_next_node(fit, images_noffset, &ndepth);
    	while (noffset >= 0 && ndepth > 0) {
    		if (ndepth != 1)
    			goto next_node;
    
    		fit_image_name = (char *)fit_get_name(fit, noffset, NULL);
    		printf("Processing update '%s' :", fit_image_name);
    
    		if (!fit_image_verify(fit, noffset)) {
    			printf("Error: invalid update hash, aborting\n");
    			ret = 1;
    			goto next_node;
    		}
    
    		printf("\n");
    		if (update_fit_getparams(fit, noffset, &update_addr,
    					&update_fladdr, &update_size)) {
    			printf("Error: can't get update parameteres, "
    								"aborting\n");
    			ret = 1;
    			goto next_node;
    		}
    
    		if (!update_tftp_dfu) {
    			if (update_flash(update_addr, update_fladdr,
    					 update_size)) {
    				printf("Error: can't flash update, aborting\n");
    				ret = 1;
    				goto next_node;
    			}
    		} else if (fit_image_check_type(fit, noffset,
    						IH_TYPE_FIRMWARE)) {
    			ret = dfu_tftp_write(fit_image_name, update_addr,
    					     update_size, interface, devstring);
    			if (ret)
    				return ret;
    		}
    next_node:
    		noffset = fdt_next_node(fit, noffset, &ndepth);
    	}
    
    	return ret;
    }