Signed-off-by: Sascha Hauer <[email protected]>
---
 common/imx-bbu-nand-fcb.c | 66 +++++++++++++++++++++++++++++++++--------------
 1 file changed, 46 insertions(+), 20 deletions(-)

diff --git a/common/imx-bbu-nand-fcb.c b/common/imx-bbu-nand-fcb.c
index 3eb9e9b..26b0b29 100644
--- a/common/imx-bbu-nand-fcb.c
+++ b/common/imx-bbu-nand-fcb.c
@@ -300,12 +300,45 @@ static int imx_bbu_erase(struct mtd_info *mtd)
        return 0;
 }
 
-static int imx_bbu_write_firmware(struct mtd_info *mtd, unsigned block,
-               unsigned num_blocks, void *buf, size_t len)
+/**
+ * imx_bbu_firmware_max_blocks - get max number of blocks for firmware
+ * @mtd: The mtd device
+ *
+ * We use 4 blocks for FCB/DBBT, the rest of the partition is
+ * divided into two equally sized firmware slots. This function
+ * returns the number of blocks available for one firmware slot.
+ * The actually usable size may be smaller due to bad blocks.
+ */
+static int imx_bbu_firmware_max_blocks(struct mtd_info *mtd)
+{
+       return (mtd_div_by_eb(mtd->size, mtd) - 4) / 2;
+}
+
+/**
+ * imx_bbu_firmware_start_block - get start block for a firmware slot
+ * @mtd: The mtd device
+ * @num: The slot number (0 or 1)
+ *
+ * We use 4 blocks for FCB/DBBT, the rest of the partition is
+ * divided into two equally sized firmware slots. This function
+ * returns the start block for the given firmware slot.
+ */
+static int imx_bbu_firmware_start_block(struct mtd_info *mtd, int num)
+{
+       return 4 + num * imx_bbu_firmware_max_blocks(mtd);
+}
+
+static int imx_bbu_write_firmware(struct mtd_info *mtd, unsigned num, void 
*buf,
+                                 size_t len)
 {
-       uint64_t offset = block * mtd->erasesize;
        int ret;
        size_t written;
+       int num_blocks = imx_bbu_firmware_max_blocks(mtd);
+       int block = imx_bbu_firmware_start_block(mtd, num);
+       uint64_t offset = block * mtd->erasesize;
+
+       pr_info("writing firmware %d to block %d (ofs 0x%08x)\n",
+                       num, block, block * mtd->erasesize);
 
        while (len > 0) {
                int now = min(len, mtd->erasesize);
@@ -364,7 +397,7 @@ static int imx_bbu_nand_update(struct bbu_handler *handler, 
struct bbu_data *dat
                container_of(handler, struct imx_nand_fcb_bbu_handler, handler);
        struct cdev *bcb_cdev;
        struct mtd_info *mtd;
-       int ret, block_fw1, block_fw2;
+       int ret;
        struct fcb_block *fcb;
        struct dbbt_block *dbbt;
        void *fcb_raw_page, *dbbt_page, *dbbt_data_page;
@@ -374,7 +407,8 @@ static int imx_bbu_nand_update(struct bbu_handler *handler, 
struct bbu_data *dat
        unsigned fw_size, partition_size;
        int i;
        enum filetype filetype;
-       unsigned num_blocks_fcb_dbbt, num_blocks, num_blocks_fw;
+       unsigned num_blocks_fw;
+       int pages_per_block;
 
        filetype = file_detect_type(data->image, data->len);
 
@@ -392,6 +426,7 @@ static int imx_bbu_nand_update(struct bbu_handler *handler, 
struct bbu_data *dat
 
        mtd = bcb_cdev->mtd;
        partition_size = mtd->size;
+       pages_per_block = mtd->erasesize / mtd->writesize;
 
        fcb_raw_page = xzalloc(mtd->writesize + mtd->oobsize);
 
@@ -411,17 +446,8 @@ static int imx_bbu_nand_update(struct bbu_handler 
*handler, struct bbu_data *dat
        fw = xzalloc(fw_size);
        memcpy(fw, data->image, data->len);
 
-       num_blocks_fcb_dbbt = 4;
-       num_blocks = partition_size / mtd->erasesize;
-       num_blocks_fw = (num_blocks - num_blocks_fcb_dbbt) / 2;
-
-       block_fw1 = num_blocks_fcb_dbbt;
-       block_fw2 = num_blocks_fcb_dbbt + num_blocks_fw;
+       num_blocks_fw = imx_bbu_firmware_max_blocks(mtd);
 
-       pr_info("writing first firmware to block %d (ofs 0x%08x)\n",
-                       block_fw1, block_fw1 * mtd->erasesize);
-       pr_info("writing second firmware to block %d (ofs 0x%08x)\n",
-                       block_fw2, block_fw2 * mtd->erasesize);
        pr_info("maximum size per firmware: 0x%08x bytes\n",
                        num_blocks_fw * mtd->erasesize);
 
@@ -436,16 +462,16 @@ static int imx_bbu_nand_update(struct bbu_handler 
*handler, struct bbu_data *dat
        if (ret)
                goto out;
 
-       ret = imx_bbu_write_firmware(mtd, block_fw1, num_blocks_fw, fw, 
fw_size);
+       ret = imx_bbu_write_firmware(mtd, 0, fw, fw_size);
        if (ret < 0)
                goto out;
 
-       ret = imx_bbu_write_firmware(mtd, block_fw2, num_blocks_fw, fw, 
fw_size);
+       ret = imx_bbu_write_firmware(mtd, 1, fw, fw_size);
        if (ret < 0)
                goto out;
 
-       fcb->Firmware1_startingPage = block_fw1 * mtd->erasesize / 
mtd->writesize;
-       fcb->Firmware2_startingPage = block_fw2 * mtd->erasesize / 
mtd->writesize;
+       fcb->Firmware1_startingPage = imx_bbu_firmware_start_block(mtd, 0) * 
pages_per_block;
+       fcb->Firmware2_startingPage = imx_bbu_firmware_start_block(mtd, 1) * 
pages_per_block;
        fcb->PagesInFirmware1 = ALIGN(data->len, mtd->writesize) / 
mtd->writesize;
        fcb->PagesInFirmware2 = fcb->PagesInFirmware1;
 
@@ -465,7 +491,7 @@ static int imx_bbu_nand_update(struct bbu_handler *handler, 
struct bbu_data *dat
        dbbt->FingerPrint = 0x54424244;
        dbbt->Version = 0x01000000;
 
-       ret = dbbt_data_create(mtd, dbbt_data_page, block_fw2 + num_blocks_fw);
+       ret = dbbt_data_create(mtd, dbbt_data_page, partition_size / 
mtd->erasesize);
        if (ret < 0)
                goto out;
 
-- 
2.7.0


_______________________________________________
barebox mailing list
[email protected]
http://lists.infradead.org/mailman/listinfo/barebox

Reply via email to