drivers/dfu/dfu_nand.c

   1 /*
   2  * dfu_nand.c -- DFU for NAND routines.
   3  *
   4  * Copyright (C) 2012-2013 Texas Instruments, Inc.
   5  *
   6  * Based on dfu_mmc.c which is:
   7  * Copyright (C) 2012 Samsung Electronics
   8  * author: Lukasz Majewski <l.majewski@samsung.com>
   9  *
  10  * SPDX-License-Identifier:     GPL-2.0+
  11  */
  12
  13 #include <common.h>
  14 #include <malloc.h>
  15 #include <errno.h>
  16 #include <div64.h>
  17 #include <dfu.h>
  18 #include <linux/mtd/mtd.h>
  19 #include <jffs2/load_kernel.h>
  20 #include <nand.h>
  21
  22 static int nand_block_op(enum dfu_op op, struct dfu_entity *dfu,
  23                         u64 offset, void *buf, long *len)
  24 {
  25         loff_t start, lim;
  26         size_t count, actual;
  27         int ret;
  28         struct mtd_info *mtd;
  29
  30         /* if buf == NULL return total size of the area */
  31         if (buf == NULL) {
  32                 *len = dfu->data.nand.size;
  33                 return 0;
  34         }
  35
  36         start = dfu->data.nand.start + offset + dfu->bad_skip;
  37         lim = dfu->data.nand.start + dfu->data.nand.size - start;
  38         count = *len;
  39
  40         if (nand_curr_device < 0 ||
  41             nand_curr_device >= CONFIG_SYS_MAX_NAND_DEVICE ||
  42             !nand_info[nand_curr_device]->name) {
  43                 printf("%s: invalid nand device\n", __func__);
  44                 return -1;
  45         }
  46
  47         mtd = nand_info[nand_curr_device];
  48
  49         if (op == DFU_OP_READ) {
  50                 ret = nand_read_skip_bad(mtd, start, &count, &actual,
  51                                          lim, buf);
  52         } else {
  53                 nand_erase_options_t opts;
  54
  55                 memset(&opts, 0, sizeof(opts));
  56                 opts.offset = start;
  57                 opts.length = count;
  58                 opts.spread = 1;
  59                 opts.quiet = 1;
  60                 opts.lim = lim;
  61                 /* first erase */
  62                 ret = nand_erase_opts(mtd, &opts);
  63                 if (ret)
  64                         return ret;
  65                 /* then write */
  66                 ret = nand_write_skip_bad(mtd, start, &count, &actual,
  67                                           lim, buf, WITH_WR_VERIFY);
  68         }
  69
  70         if (ret != 0) {
  71                 printf("%s: nand_%s_skip_bad call failed at %llx!\n",
  72                        __func__, op == DFU_OP_READ ? "read" : "write",
  73                        start);
  74                 return ret;
  75         }
  76
  77         /*
  78          * Find out where we stopped writing data.  This can be deeper into
  79          * the NAND than we expected due to having to skip bad blocks.  So
  80          * we must take this into account for the next write, if any.
  81          */
  82         if (actual > count)
  83                 dfu->bad_skip += actual - count;
  84
  85         return ret;
  86 }
  87
  88 static inline int nand_block_write(struct dfu_entity *dfu,
  89                 u64 offset, void *buf, long *len)
  90 {
  91         return nand_block_op(DFU_OP_WRITE, dfu, offset, buf, len);
  92 }
  93
  94 static inline int nand_block_read(struct dfu_entity *dfu,
  95                 u64 offset, void *buf, long *len)
  96 {
  97         return nand_block_op(DFU_OP_READ, dfu, offset, buf, len);
  98 }
  99
 100 static int dfu_write_medium_nand(struct dfu_entity *dfu,
 101                 u64 offset, void *buf, long *len)
 102 {
 103         int ret = -1;
 104
 105         switch (dfu->layout) {
 106         case DFU_RAW_ADDR:
 107                 ret = nand_block_write(dfu, offset, buf, len);
 108                 break;
 109         default:
 110                 printf("%s: Layout (%s) not (yet) supported!\n", __func__,
 111                        dfu_get_layout(dfu->layout));
 112         }
 113
 114         return ret;
 115 }
 116
 117 long dfu_get_medium_size_nand(struct dfu_entity *dfu)
 118 {
 119         return dfu->data.nand.size;
 120 }
 121
 122 static int dfu_read_medium_nand(struct dfu_entity *dfu, u64 offset, void *buf,
 123                 long *len)
 124 {
 125         int ret = -1;
 126
 127         switch (dfu->layout) {
 128         case DFU_RAW_ADDR:
 129                 ret = nand_block_read(dfu, offset, buf, len);
 130                 break;
 131         default:
 132                 printf("%s: Layout (%s) not (yet) supported!\n", __func__,
 133                        dfu_get_layout(dfu->layout));
 134         }
 135
 136         return ret;
 137 }
 138
 139 static int dfu_flush_medium_nand(struct dfu_entity *dfu)
 140 {
 141         int ret = 0;
 142         u64 off;
 143
 144         /* in case of ubi partition, erase rest of the partition */
 145         if (dfu->data.nand.ubi) {
 146                 struct mtd_info *mtd;
 147                 nand_erase_options_t opts;
 148
 149                 if (nand_curr_device < 0 ||
 150                     nand_curr_device >= CONFIG_SYS_MAX_NAND_DEVICE ||
 151                     !nand_info[nand_curr_device]->name) {
 152                         printf("%s: invalid nand device\n", __func__);
 153                         return -1;
 154                 }
 155
 156                 mtd = nand_info[nand_curr_device];
 157
 158                 memset(&opts, 0, sizeof(opts));
 159                 off = dfu->offset;
 160                 if ((off & (mtd->erasesize - 1)) != 0) {
 161                         /*
 162                          * last write ended with unaligned length
 163                          * sector is erased, jump to next
 164                          */
 165                         off = off & ~((mtd->erasesize - 1));
 166                         off += mtd->erasesize;
 167                 }
 168                 opts.offset = dfu->data.nand.start + off +
 169                                 dfu->bad_skip;
 170                 opts.length = dfu->data.nand.start +
 171                                 dfu->data.nand.size - opts.offset;
 172                 ret = nand_erase_opts(mtd, &opts);
 173                 if (ret != 0)
 174                         printf("Failure erase: %d\n", ret);
 175         }
 176
 177         return ret;
 178 }
 179
 180 unsigned int dfu_polltimeout_nand(struct dfu_entity *dfu)
 181 {
 182         /*
 183          * Currently, Poll Timeout != 0 is only needed on nand
 184          * ubi partition, as the not used sectors need an erase
 185          */
 186         if (dfu->data.nand.ubi)
 187                 return DFU_MANIFEST_POLL_TIMEOUT;
 188
 189         return DFU_DEFAULT_POLL_TIMEOUT;
 190 }
 191
 192 int dfu_fill_entity_nand(struct dfu_entity *dfu, char *devstr, char *s)
 193 {
 194         char *st;
 195         int ret, dev, part;
 196
 197         dfu->data.nand.ubi = 0;
 198         dfu->dev_type = DFU_DEV_NAND;
 199         st = strsep(&s, " ");
 200         if (!strcmp(st, "raw")) {
 201                 dfu->layout = DFU_RAW_ADDR;
 202                 dfu->data.nand.start = simple_strtoul(s, &s, 16);
 203                 s++;
 204                 dfu->data.nand.size = simple_strtoul(s, &s, 16);
 205         } else if ((!strcmp(st, "part")) || (!strcmp(st, "partubi"))) {
 206                 char mtd_id[32];
 207                 struct mtd_device *mtd_dev;
 208                 u8 part_num;
 209                 struct part_info *pi;
 210
 211                 dfu->layout = DFU_RAW_ADDR;
 212
 213                 dev = simple_strtoul(s, &s, 10);
 214                 s++;
 215                 part = simple_strtoul(s, &s, 10);
 216
 217                 sprintf(mtd_id, "%s%d,%d", "nand", dev, part - 1);
 218                 printf("using id '%s'\n", mtd_id);
 219
 220                 mtdparts_init();
 221
 222                 ret = find_dev_and_part(mtd_id, &mtd_dev, &part_num, &pi);
 223                 if (ret != 0) {
 224                         printf("Could not locate '%s'\n", mtd_id);
 225                         return -1;
 226                 }
 227
 228                 dfu->data.nand.start = pi->offset;
 229                 dfu->data.nand.size = pi->size;
 230                 if (!strcmp(st, "partubi"))
 231                         dfu->data.nand.ubi = 1;
 232         } else {
 233                 printf("%s: Memory layout (%s) not supported!\n", __func__, st);
 234                 return -1;
 235         }
 236
 237         dfu->get_medium_size = dfu_get_medium_size_nand;
 238         dfu->read_medium = dfu_read_medium_nand;
 239         dfu->write_medium = dfu_write_medium_nand;
 240         dfu->flush_medium = dfu_flush_medium_nand;
 241         dfu->poll_timeout = dfu_polltimeout_nand;
 242
 243         /* initial state */
 244         dfu->inited = 0;
 245
 246         return 0;
 247 }