drivers/mtd/mtdpstore.c

   1 // SPDX-License-Identifier: GPL-2.0
   2
   3 #define dev_fmt(fmt) "mtdoops-pstore: " fmt
   4
   5 #include <linux/kernel.h>
   6 #include <linux/module.h>
   7 #include <linux/pstore_blk.h>
   8 #include <linux/mtd/mtd.h>
   9 #include <linux/bitops.h>
  10 #include <linux/slab.h>
  11
  12 static struct mtdpstore_context {
  13         int index;
  14         struct pstore_blk_config info;
  15         struct pstore_device_info dev;
  16         struct mtd_info *mtd;
  17         unsigned long *rmmap;           /* removed bit map */
  18         unsigned long *usedmap;         /* used bit map */
  19         /*
  20          * used for panic write
  21          * As there are no block_isbad for panic case, we should keep this
  22          * status before panic to ensure panic_write not failed.
  23          */
  24         unsigned long *badmap;          /* bad block bit map */
  25 } oops_cxt;
  26
  27 static int mtdpstore_block_isbad(struct mtdpstore_context *cxt, loff_t off)
  28 {
  29         int ret;
  30         struct mtd_info *mtd = cxt->mtd;
  31         u64 blknum;
  32
  33         off = ALIGN_DOWN(off, mtd->erasesize);
  34         blknum = div_u64(off, mtd->erasesize);
  35
  36         if (test_bit(blknum, cxt->badmap))
  37                 return true;
  38         ret = mtd_block_isbad(mtd, off);
  39         if (ret < 0) {
  40                 dev_err(&mtd->dev, "mtd_block_isbad failed, aborting\n");
  41                 return ret;
  42         } else if (ret > 0) {
  43                 set_bit(blknum, cxt->badmap);
  44                 return true;
  45         }
  46         return false;
  47 }
  48
  49 static inline int mtdpstore_panic_block_isbad(struct mtdpstore_context *cxt,
  50                 loff_t off)
  51 {
  52         struct mtd_info *mtd = cxt->mtd;
  53         u64 blknum;
  54
  55         off = ALIGN_DOWN(off, mtd->erasesize);
  56         blknum = div_u64(off, mtd->erasesize);
  57         return test_bit(blknum, cxt->badmap);
  58 }
  59
  60 static inline void mtdpstore_mark_used(struct mtdpstore_context *cxt,
  61                 loff_t off)
  62 {
  63         struct mtd_info *mtd = cxt->mtd;
  64         u64 zonenum = div_u64(off, cxt->info.kmsg_size);
  65
  66         dev_dbg(&mtd->dev, "mark zone %llu used\n", zonenum);
  67         set_bit(zonenum, cxt->usedmap);
  68 }
  69
  70 static inline void mtdpstore_mark_unused(struct mtdpstore_context *cxt,
  71                 loff_t off)
  72 {
  73         struct mtd_info *mtd = cxt->mtd;
  74         u64 zonenum = div_u64(off, cxt->info.kmsg_size);
  75
  76         dev_dbg(&mtd->dev, "mark zone %llu unused\n", zonenum);
  77         clear_bit(zonenum, cxt->usedmap);
  78 }
  79
  80 static inline void mtdpstore_block_mark_unused(struct mtdpstore_context *cxt,
  81                 loff_t off)
  82 {
  83         struct mtd_info *mtd = cxt->mtd;
  84         u32 zonecnt = mtd->erasesize / cxt->info.kmsg_size;
  85         u64 zonenum;
  86
  87         off = ALIGN_DOWN(off, mtd->erasesize);
  88         zonenum = div_u64(off, cxt->info.kmsg_size);
  89         while (zonecnt > 0) {
  90                 dev_dbg(&mtd->dev, "mark zone %llu unused\n", zonenum);
  91                 clear_bit(zonenum, cxt->usedmap);
  92                 zonenum++;
  93                 zonecnt--;
  94         }
  95 }
  96
  97 static inline int mtdpstore_is_used(struct mtdpstore_context *cxt, loff_t off)
  98 {
  99         u64 zonenum = div_u64(off, cxt->info.kmsg_size);
 100         u64 blknum = div_u64(off, cxt->mtd->erasesize);
 101
 102         if (test_bit(blknum, cxt->badmap))
 103                 return true;
 104         return test_bit(zonenum, cxt->usedmap);
 105 }
 106
 107 static int mtdpstore_block_is_used(struct mtdpstore_context *cxt,
 108                 loff_t off)
 109 {
 110         struct mtd_info *mtd = cxt->mtd;
 111         u32 zonecnt = mtd->erasesize / cxt->info.kmsg_size;
 112         u64 zonenum;
 113
 114         off = ALIGN_DOWN(off, mtd->erasesize);
 115         zonenum = div_u64(off, cxt->info.kmsg_size);
 116         while (zonecnt > 0) {
 117                 if (test_bit(zonenum, cxt->usedmap))
 118                         return true;
 119                 zonenum++;
 120                 zonecnt--;
 121         }
 122         return false;
 123 }
 124
 125 static int mtdpstore_is_empty(struct mtdpstore_context *cxt, char *buf,
 126                 size_t size)
 127 {
 128         struct mtd_info *mtd = cxt->mtd;
 129         size_t sz;
 130         int i;
 131
 132         sz = min_t(uint32_t, size, mtd->writesize / 4);
 133         for (i = 0; i < sz; i++) {
 134                 if (buf[i] != (char)0xFF)
 135                         return false;
 136         }
 137         return true;
 138 }
 139
 140 static void mtdpstore_mark_removed(struct mtdpstore_context *cxt, loff_t off)
 141 {
 142         struct mtd_info *mtd = cxt->mtd;
 143         u64 zonenum = div_u64(off, cxt->info.kmsg_size);
 144
 145         dev_dbg(&mtd->dev, "mark zone %llu removed\n", zonenum);
 146         set_bit(zonenum, cxt->rmmap);
 147 }
 148
 149 static void mtdpstore_block_clear_removed(struct mtdpstore_context *cxt,
 150                 loff_t off)
 151 {
 152         struct mtd_info *mtd = cxt->mtd;
 153         u32 zonecnt = mtd->erasesize / cxt->info.kmsg_size;
 154         u64 zonenum;
 155
 156         off = ALIGN_DOWN(off, mtd->erasesize);
 157         zonenum = div_u64(off, cxt->info.kmsg_size);
 158         while (zonecnt > 0) {
 159                 clear_bit(zonenum, cxt->rmmap);
 160                 zonenum++;
 161                 zonecnt--;
 162         }
 163 }
 164
 165 static int mtdpstore_block_is_removed(struct mtdpstore_context *cxt,
 166                 loff_t off)
 167 {
 168         struct mtd_info *mtd = cxt->mtd;
 169         u32 zonecnt = mtd->erasesize / cxt->info.kmsg_size;
 170         u64 zonenum;
 171
 172         off = ALIGN_DOWN(off, mtd->erasesize);
 173         zonenum = div_u64(off, cxt->info.kmsg_size);
 174         while (zonecnt > 0) {
 175                 if (test_bit(zonenum, cxt->rmmap))
 176                         return true;
 177                 zonenum++;
 178                 zonecnt--;
 179         }
 180         return false;
 181 }
 182
 183 static int mtdpstore_erase_do(struct mtdpstore_context *cxt, loff_t off)
 184 {
 185         struct mtd_info *mtd = cxt->mtd;
 186         struct erase_info erase;
 187         int ret;
 188
 189         off = ALIGN_DOWN(off, cxt->mtd->erasesize);
 190         dev_dbg(&mtd->dev, "try to erase off 0x%llx\n", off);
 191         erase.len = cxt->mtd->erasesize;
 192         erase.addr = off;
 193         ret = mtd_erase(cxt->mtd, &erase);
 194         if (!ret)
 195                 mtdpstore_block_clear_removed(cxt, off);
 196         else
 197                 dev_err(&mtd->dev, "erase of region [0x%llx, 0x%llx] on \"%s\" failed\n",
 198                        (unsigned long long)erase.addr,
 199                        (unsigned long long)erase.len, cxt->info.device);
 200         return ret;
 201 }
 202
 203 /*
 204  * called while removing file
 205  *
 206  * Avoiding over erasing, do erase block only when the whole block is unused.
 207  * If the block contains valid log, do erase lazily on flush_removed() when
 208  * unregister.
 209  */
 210 static ssize_t mtdpstore_erase(size_t size, loff_t off)
 211 {
 212         struct mtdpstore_context *cxt = &oops_cxt;
 213
 214         if (mtdpstore_block_isbad(cxt, off))
 215                 return -EIO;
 216
 217         mtdpstore_mark_unused(cxt, off);
 218
 219         /* If the block still has valid data, mtdpstore do erase lazily */
 220         if (likely(mtdpstore_block_is_used(cxt, off))) {
 221                 mtdpstore_mark_removed(cxt, off);
 222                 return 0;
 223         }
 224
 225         /* all zones are unused, erase it */
 226         return mtdpstore_erase_do(cxt, off);
 227 }
 228
 229 /*
 230  * What is security for mtdpstore?
 231  * As there is no erase for panic case, we should ensure at least one zone
 232  * is writable. Otherwise, panic write will fail.
 233  * If zone is used, write operation will return -ENOMSG, which means that
 234  * pstore/blk will try one by one until gets an empty zone. So, it is not
 235  * needed to ensure the next zone is empty, but at least one.
 236  */
 237 static int mtdpstore_security(struct mtdpstore_context *cxt, loff_t off)
 238 {
 239         int ret = 0, i;
 240         struct mtd_info *mtd = cxt->mtd;
 241         u32 zonenum = (u32)div_u64(off, cxt->info.kmsg_size);
 242         u32 zonecnt = (u32)div_u64(cxt->mtd->size, cxt->info.kmsg_size);
 243         u32 blkcnt = (u32)div_u64(cxt->mtd->size, cxt->mtd->erasesize);
 244         u32 erasesize = cxt->mtd->erasesize;
 245
 246         for (i = 0; i < zonecnt; i++) {
 247                 u32 num = (zonenum + i) % zonecnt;
 248
 249                 /* found empty zone */
 250                 if (!test_bit(num, cxt->usedmap))
 251                         return 0;
 252         }
 253
 254         /* If there is no any empty zone, we have no way but to do erase */
 255         while (blkcnt--) {
 256                 div64_u64_rem(off + erasesize, cxt->mtd->size, (u64 *)&off);
 257
 258                 if (mtdpstore_block_isbad(cxt, off))
 259                         continue;
 260
 261                 ret = mtdpstore_erase_do(cxt, off);
 262                 if (!ret) {
 263                         mtdpstore_block_mark_unused(cxt, off);
 264                         break;
 265                 }
 266         }
 267
 268         if (ret)
 269                 dev_err(&mtd->dev, "all blocks bad!\n");
 270         dev_dbg(&mtd->dev, "end security\n");
 271         return ret;
 272 }
 273
 274 static ssize_t mtdpstore_write(const char *buf, size_t size, loff_t off)
 275 {
 276         struct mtdpstore_context *cxt = &oops_cxt;
 277         struct mtd_info *mtd = cxt->mtd;
 278         size_t retlen;
 279         int ret;
 280
 281         if (mtdpstore_block_isbad(cxt, off))
 282                 return -ENOMSG;
 283
 284         /* zone is used, please try next one */
 285         if (mtdpstore_is_used(cxt, off))
 286                 return -ENOMSG;
 287
 288         dev_dbg(&mtd->dev, "try to write off 0x%llx size %zu\n", off, size);
 289         ret = mtd_write(cxt->mtd, off, size, &retlen, (u_char *)buf);
 290         if (ret < 0 || retlen != size) {
 291                 dev_err(&mtd->dev, "write failure at %lld (%zu of %zu written), err %d\n",
 292                                 off, retlen, size, ret);
 293                 return -EIO;
 294         }
 295         mtdpstore_mark_used(cxt, off);
 296
 297         mtdpstore_security(cxt, off);
 298         return retlen;
 299 }
 300
 301 static inline bool mtdpstore_is_io_error(int ret)
 302 {
 303         return ret < 0 && !mtd_is_bitflip(ret) && !mtd_is_eccerr(ret);
 304 }
 305
 306 /*
 307  * All zones will be read as pstore/blk will read zone one by one when do
 308  * recover.
 309  */
 310 static ssize_t mtdpstore_read(char *buf, size_t size, loff_t off)
 311 {
 312         struct mtdpstore_context *cxt = &oops_cxt;
 313         struct mtd_info *mtd = cxt->mtd;
 314         size_t retlen, done;
 315         int ret;
 316
 317         if (mtdpstore_block_isbad(cxt, off))
 318                 return -ENOMSG;
 319
 320         dev_dbg(&mtd->dev, "try to read off 0x%llx size %zu\n", off, size);
 321         for (done = 0, retlen = 0; done < size; done += retlen) {
 322                 retlen = 0;
 323
 324                 ret = mtd_read(cxt->mtd, off + done, size - done, &retlen,
 325                                 (u_char *)buf + done);
 326                 if (mtdpstore_is_io_error(ret)) {
 327                         dev_err(&mtd->dev, "read failure at %lld (%zu of %zu read), err %d\n",
 328                                         off + done, retlen, size - done, ret);
 329                         /* the zone may be broken, try next one */
 330                         return -ENOMSG;
 331                 }
 332
 333                 /*
 334                  * ECC error. The impact on log data is so small. Maybe we can
 335                  * still read it and try to understand. So mtdpstore just hands
 336                  * over what it gets and user can judge whether the data is
 337                  * valid or not.
 338                  */
 339                 if (mtd_is_eccerr(ret)) {
 340                         dev_err(&mtd->dev, "ecc error at %lld (%zu of %zu read), err %d\n",
 341                                         off + done, retlen, size - done, ret);
 342                         /* driver may not set retlen when ecc error */
 343                         retlen = retlen == 0 ? size - done : retlen;
 344                 }
 345         }
 346
 347         if (mtdpstore_is_empty(cxt, buf, size))
 348                 mtdpstore_mark_unused(cxt, off);
 349         else
 350                 mtdpstore_mark_used(cxt, off);
 351
 352         mtdpstore_security(cxt, off);
 353         return retlen;
 354 }
 355
 356 static ssize_t mtdpstore_panic_write(const char *buf, size_t size, loff_t off)
 357 {
 358         struct mtdpstore_context *cxt = &oops_cxt;
 359         struct mtd_info *mtd = cxt->mtd;
 360         size_t retlen;
 361         int ret;
 362
 363         if (mtdpstore_panic_block_isbad(cxt, off))
 364                 return -ENOMSG;
 365
 366         /* zone is used, please try next one */
 367         if (mtdpstore_is_used(cxt, off))
 368                 return -ENOMSG;
 369
 370         ret = mtd_panic_write(cxt->mtd, off, size, &retlen, (u_char *)buf);
 371         if (ret < 0 || size != retlen) {
 372                 dev_err(&mtd->dev, "panic write failure at %lld (%zu of %zu read), err %d\n",
 373                                 off, retlen, size, ret);
 374                 return -EIO;
 375         }
 376         mtdpstore_mark_used(cxt, off);
 377
 378         return retlen;
 379 }
 380
 381 static void mtdpstore_notify_add(struct mtd_info *mtd)
 382 {
 383         int ret;
 384         struct mtdpstore_context *cxt = &oops_cxt;
 385         struct pstore_blk_config *info = &cxt->info;
 386         unsigned long longcnt;
 387
 388         if (!strcmp(mtd->name, info->device))
 389                 cxt->index = mtd->index;
 390
 391         if (mtd->index != cxt->index || cxt->index < 0)
 392                 return;
 393
 394         dev_dbg(&mtd->dev, "found matching MTD device %s\n", mtd->name);
 395
 396         if (mtd->size < info->kmsg_size * 2) {
 397                 dev_err(&mtd->dev, "MTD partition %d not big enough\n",
 398                                 mtd->index);
 399                 return;
 400         }
 401         /*
 402          * kmsg_size must be aligned to 4096 Bytes, which is limited by
 403          * psblk. The default value of kmsg_size is 64KB. If kmsg_size
 404          * is larger than erasesize, some errors will occur since mtdpsotre
 405          * is designed on it.
 406          */
 407         if (mtd->erasesize < info->kmsg_size) {
 408                 dev_err(&mtd->dev, "eraseblock size of MTD partition %d too small\n",
 409                                 mtd->index);
 410                 return;
 411         }
 412         if (unlikely(info->kmsg_size % mtd->writesize)) {
 413                 dev_err(&mtd->dev, "record size %lu KB must align to write size %d KB\n",
 414                                 info->kmsg_size / 1024,
 415                                 mtd->writesize / 1024);
 416                 return;
 417         }
 418
 419         longcnt = BITS_TO_LONGS(div_u64(mtd->size, info->kmsg_size));
 420         cxt->rmmap = kcalloc(longcnt, sizeof(long), GFP_KERNEL);
 421         cxt->usedmap = kcalloc(longcnt, sizeof(long), GFP_KERNEL);
 422
 423         longcnt = BITS_TO_LONGS(div_u64(mtd->size, mtd->erasesize));
 424         cxt->badmap = kcalloc(longcnt, sizeof(long), GFP_KERNEL);
 425
 426         cxt->dev.total_size = mtd->size;
 427         /* just support dmesg right now */
 428         cxt->dev.flags = PSTORE_FLAGS_DMESG;
 429         cxt->dev.read = mtdpstore_read;
 430         cxt->dev.write = mtdpstore_write;
 431         cxt->dev.erase = mtdpstore_erase;
 432         cxt->dev.panic_write = mtdpstore_panic_write;
 433
 434         ret = register_pstore_device(&cxt->dev);
 435         if (ret) {
 436                 dev_err(&mtd->dev, "mtd%d register to psblk failed\n",
 437                                 mtd->index);
 438                 return;
 439         }
 440         cxt->mtd = mtd;
 441         dev_info(&mtd->dev, "Attached to MTD device %d\n", mtd->index);
 442 }
 443
 444 static int mtdpstore_flush_removed_do(struct mtdpstore_context *cxt,
 445                 loff_t off, size_t size)
 446 {
 447         struct mtd_info *mtd = cxt->mtd;
 448         u_char *buf;
 449         int ret;
 450         size_t retlen;
 451         struct erase_info erase;
 452
 453         buf = kmalloc(mtd->erasesize, GFP_KERNEL);
 454         if (!buf)
 455                 return -ENOMEM;
 456
 457         /* 1st. read to cache */
 458         ret = mtd_read(mtd, off, mtd->erasesize, &retlen, buf);
 459         if (mtdpstore_is_io_error(ret))
 460                 goto free;
 461
 462         /* 2nd. erase block */
 463         erase.len = mtd->erasesize;
 464         erase.addr = off;
 465         ret = mtd_erase(mtd, &erase);
 466         if (ret)
 467                 goto free;
 468
 469         /* 3rd. write back */
 470         while (size) {
 471                 unsigned int zonesize = cxt->info.kmsg_size;
 472
 473                 /* there is valid data on block, write back */
 474                 if (mtdpstore_is_used(cxt, off)) {
 475                         ret = mtd_write(mtd, off, zonesize, &retlen, buf);
 476                         if (ret)
 477                                 dev_err(&mtd->dev, "write failure at %lld (%zu of %u written), err %d\n",
 478                                                 off, retlen, zonesize, ret);
 479                 }
 480
 481                 off += zonesize;
 482                 size -= min_t(unsigned int, zonesize, size);
 483         }
 484
 485 free:
 486         kfree(buf);
 487         return ret;
 488 }
 489
 490 /*
 491  * What does mtdpstore_flush_removed() do?
 492  * When user remove any log file on pstore filesystem, mtdpstore should do
 493  * something to ensure log file removed. If the whole block is no longer used,
 494  * it's nice to erase the block. However if the block still contains valid log,
 495  * what mtdpstore can do is to erase and write the valid log back.
 496  */
 497 static int mtdpstore_flush_removed(struct mtdpstore_context *cxt)
 498 {
 499         struct mtd_info *mtd = cxt->mtd;
 500         int ret;
 501         loff_t off;
 502         u32 blkcnt = (u32)div_u64(mtd->size, mtd->erasesize);
 503
 504         for (off = 0; blkcnt > 0; blkcnt--, off += mtd->erasesize) {
 505                 ret = mtdpstore_block_isbad(cxt, off);
 506                 if (ret)
 507                         continue;
 508
 509                 ret = mtdpstore_block_is_removed(cxt, off);
 510                 if (!ret)
 511                         continue;
 512
 513                 ret = mtdpstore_flush_removed_do(cxt, off, mtd->erasesize);
 514                 if (ret)
 515                         return ret;
 516         }
 517         return 0;
 518 }
 519
 520 static void mtdpstore_notify_remove(struct mtd_info *mtd)
 521 {
 522         struct mtdpstore_context *cxt = &oops_cxt;
 523
 524         if (mtd->index != cxt->index || cxt->index < 0)
 525                 return;
 526
 527         mtdpstore_flush_removed(cxt);
 528
 529         unregister_pstore_device(&cxt->dev);
 530         kfree(cxt->badmap);
 531         kfree(cxt->usedmap);
 532         kfree(cxt->rmmap);
 533         cxt->mtd = NULL;
 534         cxt->index = -1;
 535 }
 536
 537 static struct mtd_notifier mtdpstore_notifier = {
 538         .add    = mtdpstore_notify_add,
 539         .remove = mtdpstore_notify_remove,
 540 };
 541
 542 static int __init mtdpstore_init(void)
 543 {
 544         int ret;
 545         struct mtdpstore_context *cxt = &oops_cxt;
 546         struct pstore_blk_config *info = &cxt->info;
 547
 548         ret = pstore_blk_get_config(info);
 549         if (unlikely(ret))
 550                 return ret;
 551
 552         if (strlen(info->device) == 0) {
 553                 pr_err("mtd device must be supplied (device name is empty)\n");
 554                 return -EINVAL;
 555         }
 556         if (!info->kmsg_size) {
 557                 pr_err("no backend enabled (kmsg_size is 0)\n");
 558                 return -EINVAL;
 559         }
 560
 561         /* Setup the MTD device to use */
 562         ret = kstrtoint((char *)info->device, 0, &cxt->index);
 563         if (ret)
 564                 cxt->index = -1;
 565
 566         register_mtd_user(&mtdpstore_notifier);
 567         return 0;
 568 }
 569 module_init(mtdpstore_init);
 570
 571 static void __exit mtdpstore_exit(void)
 572 {
 573         unregister_mtd_user(&mtdpstore_notifier);
 574 }
 575 module_exit(mtdpstore_exit);
 576
 577 MODULE_LICENSE("GPL");
 578 MODULE_AUTHOR("WeiXiong Liao <liaoweixiong@allwinnertech.com>");
 579 MODULE_DESCRIPTION("MTD backend for pstore/blk");