/*
|
* Copyright (c) 2017 Paweł Jarosz <paweljarosz3691@gmail.com>
|
*
|
* SPDX-License-Identifier: GPL-2.0+
|
*/
|
|
#include "imagetool.h"
|
#include <image.h>
|
#include <rc4.h>
|
#include "mkimage.h"
|
#include "rkcommon.h"
|
|
#define DIV_ROUND_UP(n, d) (((n) + (d) - 1) / (d))
|
|
enum {
|
RKNAND_SECT_LEN = RK_BLK_SIZE * 4,
|
};
|
|
struct rknand_info {
|
uint32_t pagesize;
|
uint32_t itersize;
|
uint32_t tplsplsize;
|
};
|
|
struct rknand_info ninfo;
|
|
static void rknand_set_header(void *buf, struct stat *sbuf, int ifd,
|
struct image_tool_params *params)
|
{
|
int sector;
|
unsigned int size;
|
|
size = params->orig_file_size;
|
|
rkcommon_set_header(buf, sbuf, ifd, params);
|
|
/*
|
* Spread the image out so we only use the first 2KB of each pagesize
|
* region. This is a feature of the NAND format required by the Rockchip
|
* boot ROM.
|
*/
|
if (params->vflag)
|
fprintf(stderr, "Spreading nand image from %u to %u\n",
|
size, params->file_size);
|
|
for (sector = ninfo.tplsplsize / ninfo.itersize - 1; sector >= 0;
|
sector--) {
|
memmove(buf + ninfo.pagesize + sector * ninfo.itersize,
|
buf + RK_SPL_HDR_START + sector * RKNAND_SECT_LEN,
|
RKNAND_SECT_LEN);
|
|
memset(buf + ninfo.pagesize + sector * ninfo.itersize +
|
RKNAND_SECT_LEN, 0xFF,
|
ninfo.itersize - RKNAND_SECT_LEN);
|
}
|
|
/* Fill up padded area of the header. */
|
memset(buf + RK_SPL_HDR_START, 0xFF, ninfo.pagesize - RK_SPL_HDR_START);
|
}
|
|
static int rknand_check_image_type(uint8_t type)
|
{
|
if (type == IH_TYPE_RKNAND)
|
return EXIT_SUCCESS;
|
else
|
return EXIT_FAILURE;
|
}
|
|
static int rknand_vrec_header(struct image_tool_params *params,
|
struct image_type_params *tparams)
|
{
|
int tplsplsize;
|
uint32_t skippages;
|
int ret;
|
|
rkcommon_vrec_header(params, tparams);
|
|
ret = sscanf(params->extraparams, "%u,%u", &ninfo.pagesize, &skippages);
|
if (ret != 2 || (ninfo.pagesize % RKNAND_SECT_LEN)) {
|
fprintf(stderr, "%s: Wrong nand params\n", params->cmdname);
|
exit(EXIT_FAILURE);
|
}
|
|
ninfo.itersize = ninfo.pagesize * (skippages + 1);
|
|
tplsplsize = params->file_size - RK_SPL_HDR_START;
|
ninfo.tplsplsize =
|
DIV_ROUND_UP(tplsplsize, RKNAND_SECT_LEN) * ninfo.itersize;
|
|
/* Padded file size = padded header + padded tpl & spl. */
|
params->file_size = ninfo.pagesize + ninfo.tplsplsize;
|
|
return 0;
|
}
|
|
/*
|
* rknand parameters
|
*/
|
U_BOOT_IMAGE_TYPE(
|
rknand,
|
"Rockchip NAND Boot Image support",
|
0,
|
NULL,
|
rkcommon_check_params,
|
/* TODO: Support rknand in there helpers */
|
NULL, //rkcommon_verify_header,
|
NULL, //rkcommon_print_header,
|
rknand_set_header,
|
NULL,
|
rknand_check_image_type,
|
NULL,
|
rknand_vrec_header
|
);
|
