1*4882a593Smuzhiyun /*
2*4882a593Smuzhiyun * Copyright (c) 2017 Paweł Jarosz <paweljarosz3691@gmail.com>
3*4882a593Smuzhiyun *
4*4882a593Smuzhiyun * SPDX-License-Identifier: GPL-2.0+
5*4882a593Smuzhiyun */
6*4882a593Smuzhiyun
7*4882a593Smuzhiyun #include "imagetool.h"
8*4882a593Smuzhiyun #include <image.h>
9*4882a593Smuzhiyun #include <rc4.h>
10*4882a593Smuzhiyun #include "mkimage.h"
11*4882a593Smuzhiyun #include "rkcommon.h"
12*4882a593Smuzhiyun
13*4882a593Smuzhiyun #define DIV_ROUND_UP(n, d) (((n) + (d) - 1) / (d))
14*4882a593Smuzhiyun
15*4882a593Smuzhiyun enum {
16*4882a593Smuzhiyun RKNAND_SECT_LEN = RK_BLK_SIZE * 4,
17*4882a593Smuzhiyun };
18*4882a593Smuzhiyun
19*4882a593Smuzhiyun struct rknand_info {
20*4882a593Smuzhiyun uint32_t pagesize;
21*4882a593Smuzhiyun uint32_t itersize;
22*4882a593Smuzhiyun uint32_t tplsplsize;
23*4882a593Smuzhiyun };
24*4882a593Smuzhiyun
25*4882a593Smuzhiyun struct rknand_info ninfo;
26*4882a593Smuzhiyun
rknand_set_header(void * buf,struct stat * sbuf,int ifd,struct image_tool_params * params)27*4882a593Smuzhiyun static void rknand_set_header(void *buf, struct stat *sbuf, int ifd,
28*4882a593Smuzhiyun struct image_tool_params *params)
29*4882a593Smuzhiyun {
30*4882a593Smuzhiyun int sector;
31*4882a593Smuzhiyun unsigned int size;
32*4882a593Smuzhiyun
33*4882a593Smuzhiyun size = params->orig_file_size;
34*4882a593Smuzhiyun
35*4882a593Smuzhiyun rkcommon_set_header(buf, sbuf, ifd, params);
36*4882a593Smuzhiyun
37*4882a593Smuzhiyun /*
38*4882a593Smuzhiyun * Spread the image out so we only use the first 2KB of each pagesize
39*4882a593Smuzhiyun * region. This is a feature of the NAND format required by the Rockchip
40*4882a593Smuzhiyun * boot ROM.
41*4882a593Smuzhiyun */
42*4882a593Smuzhiyun if (params->vflag)
43*4882a593Smuzhiyun fprintf(stderr, "Spreading nand image from %u to %u\n",
44*4882a593Smuzhiyun size, params->file_size);
45*4882a593Smuzhiyun
46*4882a593Smuzhiyun for (sector = ninfo.tplsplsize / ninfo.itersize - 1; sector >= 0;
47*4882a593Smuzhiyun sector--) {
48*4882a593Smuzhiyun memmove(buf + ninfo.pagesize + sector * ninfo.itersize,
49*4882a593Smuzhiyun buf + RK_SPL_HDR_START + sector * RKNAND_SECT_LEN,
50*4882a593Smuzhiyun RKNAND_SECT_LEN);
51*4882a593Smuzhiyun
52*4882a593Smuzhiyun memset(buf + ninfo.pagesize + sector * ninfo.itersize +
53*4882a593Smuzhiyun RKNAND_SECT_LEN, 0xFF,
54*4882a593Smuzhiyun ninfo.itersize - RKNAND_SECT_LEN);
55*4882a593Smuzhiyun }
56*4882a593Smuzhiyun
57*4882a593Smuzhiyun /* Fill up padded area of the header. */
58*4882a593Smuzhiyun memset(buf + RK_SPL_HDR_START, 0xFF, ninfo.pagesize - RK_SPL_HDR_START);
59*4882a593Smuzhiyun }
60*4882a593Smuzhiyun
rknand_check_image_type(uint8_t type)61*4882a593Smuzhiyun static int rknand_check_image_type(uint8_t type)
62*4882a593Smuzhiyun {
63*4882a593Smuzhiyun if (type == IH_TYPE_RKNAND)
64*4882a593Smuzhiyun return EXIT_SUCCESS;
65*4882a593Smuzhiyun else
66*4882a593Smuzhiyun return EXIT_FAILURE;
67*4882a593Smuzhiyun }
68*4882a593Smuzhiyun
rknand_vrec_header(struct image_tool_params * params,struct image_type_params * tparams)69*4882a593Smuzhiyun static int rknand_vrec_header(struct image_tool_params *params,
70*4882a593Smuzhiyun struct image_type_params *tparams)
71*4882a593Smuzhiyun {
72*4882a593Smuzhiyun int tplsplsize;
73*4882a593Smuzhiyun uint32_t skippages;
74*4882a593Smuzhiyun int ret;
75*4882a593Smuzhiyun
76*4882a593Smuzhiyun rkcommon_vrec_header(params, tparams);
77*4882a593Smuzhiyun
78*4882a593Smuzhiyun ret = sscanf(params->extraparams, "%u,%u", &ninfo.pagesize, &skippages);
79*4882a593Smuzhiyun if (ret != 2 || (ninfo.pagesize % RKNAND_SECT_LEN)) {
80*4882a593Smuzhiyun fprintf(stderr, "%s: Wrong nand params\n", params->cmdname);
81*4882a593Smuzhiyun exit(EXIT_FAILURE);
82*4882a593Smuzhiyun }
83*4882a593Smuzhiyun
84*4882a593Smuzhiyun ninfo.itersize = ninfo.pagesize * (skippages + 1);
85*4882a593Smuzhiyun
86*4882a593Smuzhiyun tplsplsize = params->file_size - RK_SPL_HDR_START;
87*4882a593Smuzhiyun ninfo.tplsplsize =
88*4882a593Smuzhiyun DIV_ROUND_UP(tplsplsize, RKNAND_SECT_LEN) * ninfo.itersize;
89*4882a593Smuzhiyun
90*4882a593Smuzhiyun /* Padded file size = padded header + padded tpl & spl. */
91*4882a593Smuzhiyun params->file_size = ninfo.pagesize + ninfo.tplsplsize;
92*4882a593Smuzhiyun
93*4882a593Smuzhiyun return 0;
94*4882a593Smuzhiyun }
95*4882a593Smuzhiyun
96*4882a593Smuzhiyun /*
97*4882a593Smuzhiyun * rknand parameters
98*4882a593Smuzhiyun */
99*4882a593Smuzhiyun U_BOOT_IMAGE_TYPE(
100*4882a593Smuzhiyun rknand,
101*4882a593Smuzhiyun "Rockchip NAND Boot Image support",
102*4882a593Smuzhiyun 0,
103*4882a593Smuzhiyun NULL,
104*4882a593Smuzhiyun rkcommon_check_params,
105*4882a593Smuzhiyun /* TODO: Support rknand in there helpers */
106*4882a593Smuzhiyun NULL, //rkcommon_verify_header,
107*4882a593Smuzhiyun NULL, //rkcommon_print_header,
108*4882a593Smuzhiyun rknand_set_header,
109*4882a593Smuzhiyun NULL,
110*4882a593Smuzhiyun rknand_check_image_type,
111*4882a593Smuzhiyun NULL,
112*4882a593Smuzhiyun rknand_vrec_header
113*4882a593Smuzhiyun );
114