mirror of
https://github.com/AsahiLinux/u-boot
synced 2024-12-05 11:00:15 +00:00
17ead040d4
A large number of files include <flash.h> as it used to be how various SPI flash related functions were found, or for other reasons entirely. In order to migrate some further CONFIG symbols to Kconfig we need to not include flash.h in cases where we don't have a NOR flash of some sort enabled. Furthermore, in cases where we are in common code and it doesn't make sense to try and further refactor the code itself in to new files we need to guard this inclusion. Signed-off-by: Tom Rini <trini@konsulko.com>
262 lines
6 KiB
C
262 lines
6 KiB
C
// SPDX-License-Identifier: GPL-2.0+
|
|
/*
|
|
* Copyright 2014 Broadcom Corporation.
|
|
* Copyright 2015 Free Electrons.
|
|
*/
|
|
|
|
#include <config.h>
|
|
#include <common.h>
|
|
#include <blk.h>
|
|
|
|
#include <fastboot.h>
|
|
#include <image-sparse.h>
|
|
|
|
#include <linux/mtd/mtd.h>
|
|
#include <jffs2/jffs2.h>
|
|
#include <nand.h>
|
|
|
|
struct fb_nand_sparse {
|
|
struct mtd_info *mtd;
|
|
struct part_info *part;
|
|
};
|
|
|
|
__weak int board_fastboot_erase_partition_setup(char *name)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
__weak int board_fastboot_write_partition_setup(char *name)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
static int fb_nand_lookup(const char *partname,
|
|
struct mtd_info **mtd,
|
|
struct part_info **part,
|
|
char *response)
|
|
{
|
|
struct mtd_device *dev;
|
|
int ret;
|
|
u8 pnum;
|
|
|
|
ret = mtdparts_init();
|
|
if (ret) {
|
|
pr_err("Cannot initialize MTD partitions\n");
|
|
fastboot_fail("cannot init mtdparts", response);
|
|
return ret;
|
|
}
|
|
|
|
ret = find_dev_and_part(partname, &dev, &pnum, part);
|
|
if (ret) {
|
|
pr_err("cannot find partition: '%s'", partname);
|
|
fastboot_fail("cannot find partition", response);
|
|
return ret;
|
|
}
|
|
|
|
if (dev->id->type != MTD_DEV_TYPE_NAND) {
|
|
pr_err("partition '%s' is not stored on a NAND device",
|
|
partname);
|
|
fastboot_fail("not a NAND device", response);
|
|
return -EINVAL;
|
|
}
|
|
|
|
*mtd = get_nand_dev_by_index(dev->id->num);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int _fb_nand_erase(struct mtd_info *mtd, struct part_info *part)
|
|
{
|
|
nand_erase_options_t opts;
|
|
int ret;
|
|
|
|
memset(&opts, 0, sizeof(opts));
|
|
opts.offset = part->offset;
|
|
opts.length = part->size;
|
|
opts.quiet = 1;
|
|
|
|
printf("Erasing blocks 0x%llx to 0x%llx\n",
|
|
part->offset, part->offset + part->size);
|
|
|
|
ret = nand_erase_opts(mtd, &opts);
|
|
if (ret)
|
|
return ret;
|
|
|
|
printf("........ erased 0x%llx bytes from '%s'\n",
|
|
part->size, part->name);
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int _fb_nand_write(struct mtd_info *mtd, struct part_info *part,
|
|
void *buffer, u32 offset,
|
|
size_t length, size_t *written)
|
|
{
|
|
int flags = WITH_WR_VERIFY;
|
|
|
|
#ifdef CONFIG_FASTBOOT_FLASH_NAND_TRIMFFS
|
|
flags |= WITH_DROP_FFS;
|
|
#endif
|
|
|
|
return nand_write_skip_bad(mtd, offset, &length, written,
|
|
part->size - (offset - part->offset),
|
|
buffer, flags);
|
|
}
|
|
|
|
static lbaint_t fb_nand_sparse_write(struct sparse_storage *info,
|
|
lbaint_t blk, lbaint_t blkcnt, const void *buffer)
|
|
{
|
|
struct fb_nand_sparse *sparse = info->priv;
|
|
size_t written;
|
|
int ret;
|
|
|
|
ret = _fb_nand_write(sparse->mtd, sparse->part, (void *)buffer,
|
|
blk * info->blksz,
|
|
blkcnt * info->blksz, &written);
|
|
if (ret < 0) {
|
|
printf("Failed to write sparse chunk\n");
|
|
return ret;
|
|
}
|
|
|
|
/* TODO - verify that the value "written" includes the "bad-blocks" ... */
|
|
|
|
/*
|
|
* the return value must be 'blkcnt' ("good-blocks") plus the
|
|
* number of "bad-blocks" encountered within this space...
|
|
*/
|
|
return written / info->blksz;
|
|
}
|
|
|
|
static lbaint_t fb_nand_sparse_reserve(struct sparse_storage *info,
|
|
lbaint_t blk, lbaint_t blkcnt)
|
|
{
|
|
int bad_blocks = 0;
|
|
|
|
/*
|
|
* TODO - implement a function to determine the total number
|
|
* of blocks which must be used in order to reserve the specified
|
|
* number ("blkcnt") of "good-blocks", starting at "blk"...
|
|
* ( possibly something like the "check_skip_len()" function )
|
|
*/
|
|
|
|
/*
|
|
* the return value must be 'blkcnt' ("good-blocks") plus the
|
|
* number of "bad-blocks" encountered within this space...
|
|
*/
|
|
return blkcnt + bad_blocks;
|
|
}
|
|
|
|
/**
|
|
* fastboot_nand_get_part_info() - Lookup NAND partion by name
|
|
*
|
|
* @part_name: Named device to lookup
|
|
* @part_info: Pointer to returned part_info pointer
|
|
* @response: Pointer to fastboot response buffer
|
|
*/
|
|
int fastboot_nand_get_part_info(const char *part_name,
|
|
struct part_info **part_info, char *response)
|
|
{
|
|
struct mtd_info *mtd = NULL;
|
|
|
|
return fb_nand_lookup(part_name, &mtd, part_info, response);
|
|
}
|
|
|
|
/**
|
|
* fastboot_nand_flash_write() - Write image to NAND for fastboot
|
|
*
|
|
* @cmd: Named device to write image to
|
|
* @download_buffer: Pointer to image data
|
|
* @download_bytes: Size of image data
|
|
* @response: Pointer to fastboot response buffer
|
|
*/
|
|
void fastboot_nand_flash_write(const char *cmd, void *download_buffer,
|
|
u32 download_bytes, char *response)
|
|
{
|
|
struct part_info *part;
|
|
struct mtd_info *mtd = NULL;
|
|
int ret;
|
|
|
|
ret = fb_nand_lookup(cmd, &mtd, &part, response);
|
|
if (ret) {
|
|
pr_err("invalid NAND device");
|
|
fastboot_fail("invalid NAND device", response);
|
|
return;
|
|
}
|
|
|
|
ret = board_fastboot_write_partition_setup(part->name);
|
|
if (ret)
|
|
return;
|
|
|
|
if (is_sparse_image(download_buffer)) {
|
|
struct fb_nand_sparse sparse_priv;
|
|
struct sparse_storage sparse;
|
|
|
|
sparse_priv.mtd = mtd;
|
|
sparse_priv.part = part;
|
|
|
|
sparse.blksz = mtd->writesize;
|
|
sparse.start = part->offset / sparse.blksz;
|
|
sparse.size = part->size / sparse.blksz;
|
|
sparse.write = fb_nand_sparse_write;
|
|
sparse.reserve = fb_nand_sparse_reserve;
|
|
sparse.mssg = fastboot_fail;
|
|
|
|
printf("Flashing sparse image at offset " LBAFU "\n",
|
|
sparse.start);
|
|
|
|
sparse.priv = &sparse_priv;
|
|
ret = write_sparse_image(&sparse, cmd, download_buffer,
|
|
response);
|
|
if (!ret)
|
|
fastboot_okay(NULL, response);
|
|
} else {
|
|
printf("Flashing raw image at offset 0x%llx\n",
|
|
part->offset);
|
|
|
|
ret = _fb_nand_write(mtd, part, download_buffer, part->offset,
|
|
download_bytes, NULL);
|
|
|
|
printf("........ wrote %u bytes to '%s'\n",
|
|
download_bytes, part->name);
|
|
}
|
|
|
|
if (ret) {
|
|
fastboot_fail("error writing the image", response);
|
|
return;
|
|
}
|
|
|
|
fastboot_okay(NULL, response);
|
|
}
|
|
|
|
/**
|
|
* fastboot_nand_flash_erase() - Erase NAND for fastboot
|
|
*
|
|
* @cmd: Named device to erase
|
|
* @response: Pointer to fastboot response buffer
|
|
*/
|
|
void fastboot_nand_erase(const char *cmd, char *response)
|
|
{
|
|
struct part_info *part;
|
|
struct mtd_info *mtd = NULL;
|
|
int ret;
|
|
|
|
ret = fb_nand_lookup(cmd, &mtd, &part, response);
|
|
if (ret) {
|
|
pr_err("invalid NAND device");
|
|
fastboot_fail("invalid NAND device", response);
|
|
return;
|
|
}
|
|
|
|
ret = board_fastboot_erase_partition_setup(part->name);
|
|
if (ret)
|
|
return;
|
|
|
|
ret = _fb_nand_erase(mtd, part);
|
|
if (ret) {
|
|
pr_err("failed erasing from device %s", mtd->name);
|
|
fastboot_fail("failed erasing from device", response);
|
|
return;
|
|
}
|
|
|
|
fastboot_okay(NULL, response);
|
|
}
|