mirror of
https://github.com/brain-hackers/u-boot-brain
synced 2024-08-17 07:28:51 +09:00
![Sam Protsenko](/assets/img/avatar_default.png)
In fastboot_*_get_part_info() functions we can use stronger typing by expecting const strings. Signed-off-by: Sam Protsenko <semen.protsenko@linaro.org> Reviewed-by: Lukasz Majewski <lukma@denx.de> Reviewed-by: Igor Opaniuk <igor.opaniuk@toradex.com>
262 lines
6.0 KiB
C
262 lines
6.0 KiB
C
// SPDX-License-Identifier: GPL-2.0+
|
|
/*
|
|
* Copyright 2014 Broadcom Corporation.
|
|
* Copyright 2015 Free Electrons.
|
|
*/
|
|
|
|
#include <config.h>
|
|
#include <common.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);
|
|
}
|