blkmtd.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820
  1. /*
  2. * $Id: blkmtd.c,v 1.27 2005/11/07 11:14:24 gleixner Exp $
  3. *
  4. * blkmtd.c - use a block device as a fake MTD
  5. *
  6. * Author: Simon Evans <spse@secret.org.uk>
  7. *
  8. * Copyright (C) 2001,2002 Simon Evans
  9. *
  10. * Licence: GPL
  11. *
  12. * How it works:
  13. * The driver uses raw/io to read/write the device and the page
  14. * cache to cache access. Writes update the page cache with the
  15. * new data and mark it dirty and add the page into a BIO which
  16. * is then written out.
  17. *
  18. * It can be loaded Read-Only to prevent erases and writes to the
  19. * medium.
  20. *
  21. */
  22. #include <linux/config.h>
  23. #include <linux/module.h>
  24. #include <linux/fs.h>
  25. #include <linux/blkdev.h>
  26. #include <linux/bio.h>
  27. #include <linux/pagemap.h>
  28. #include <linux/list.h>
  29. #include <linux/init.h>
  30. #include <linux/mtd/mtd.h>
  31. #include <linux/mutex.h>
  32. #define err(format, arg...) printk(KERN_ERR "blkmtd: " format "\n" , ## arg)
  33. #define info(format, arg...) printk(KERN_INFO "blkmtd: " format "\n" , ## arg)
  34. #define warn(format, arg...) printk(KERN_WARNING "blkmtd: " format "\n" , ## arg)
  35. #define crit(format, arg...) printk(KERN_CRIT "blkmtd: " format "\n" , ## arg)
  36. /* Default erase size in K, always make it a multiple of PAGE_SIZE */
  37. #define CONFIG_MTD_BLKDEV_ERASESIZE (128 << 10) /* 128KiB */
  38. #define VERSION "$Revision: 1.27 $"
  39. /* Info for the block device */
  40. struct blkmtd_dev {
  41. struct list_head list;
  42. struct block_device *blkdev;
  43. struct mtd_info mtd_info;
  44. struct mutex wrbuf_mutex;
  45. };
  46. /* Static info about the MTD, used in cleanup_module */
  47. static LIST_HEAD(blkmtd_device_list);
  48. static void blkmtd_sync(struct mtd_info *mtd);
  49. #define MAX_DEVICES 4
  50. /* Module parameters passed by insmod/modprobe */
  51. static char *device[MAX_DEVICES]; /* the block device to use */
  52. static int erasesz[MAX_DEVICES]; /* optional default erase size */
  53. static int ro[MAX_DEVICES]; /* optional read only flag */
  54. static int sync;
  55. MODULE_LICENSE("GPL");
  56. MODULE_AUTHOR("Simon Evans <spse@secret.org.uk>");
  57. MODULE_DESCRIPTION("Emulate an MTD using a block device");
  58. module_param_array(device, charp, NULL, 0);
  59. MODULE_PARM_DESC(device, "block device to use");
  60. module_param_array(erasesz, int, NULL, 0);
  61. MODULE_PARM_DESC(erasesz, "optional erase size to use in KiB. eg 4=4KiB.");
  62. module_param_array(ro, bool, NULL, 0);
  63. MODULE_PARM_DESC(ro, "1=Read only, writes and erases cause errors");
  64. module_param(sync, bool, 0);
  65. MODULE_PARM_DESC(sync, "1=Synchronous writes");
  66. /* completion handler for BIO reads */
  67. static int bi_read_complete(struct bio *bio, unsigned int bytes_done, int error)
  68. {
  69. if (bio->bi_size)
  70. return 1;
  71. complete((struct completion*)bio->bi_private);
  72. return 0;
  73. }
  74. /* completion handler for BIO writes */
  75. static int bi_write_complete(struct bio *bio, unsigned int bytes_done, int error)
  76. {
  77. const int uptodate = test_bit(BIO_UPTODATE, &bio->bi_flags);
  78. struct bio_vec *bvec = bio->bi_io_vec + bio->bi_vcnt - 1;
  79. if (bio->bi_size)
  80. return 1;
  81. if(!uptodate)
  82. err("bi_write_complete: not uptodate\n");
  83. do {
  84. struct page *page = bvec->bv_page;
  85. DEBUG(3, "Cleaning up page %ld\n", page->index);
  86. if (--bvec >= bio->bi_io_vec)
  87. prefetchw(&bvec->bv_page->flags);
  88. if (uptodate) {
  89. SetPageUptodate(page);
  90. } else {
  91. ClearPageUptodate(page);
  92. SetPageError(page);
  93. }
  94. clear_page_dirty(page);
  95. unlock_page(page);
  96. page_cache_release(page);
  97. } while (bvec >= bio->bi_io_vec);
  98. complete((struct completion*)bio->bi_private);
  99. return 0;
  100. }
  101. /* read one page from the block device */
  102. static int blkmtd_readpage(struct blkmtd_dev *dev, struct page *page)
  103. {
  104. struct bio *bio;
  105. struct completion event;
  106. int err = -ENOMEM;
  107. if(PageUptodate(page)) {
  108. DEBUG(2, "blkmtd: readpage page %ld is already upto date\n", page->index);
  109. unlock_page(page);
  110. return 0;
  111. }
  112. ClearPageUptodate(page);
  113. ClearPageError(page);
  114. bio = bio_alloc(GFP_KERNEL, 1);
  115. if(bio) {
  116. init_completion(&event);
  117. bio->bi_bdev = dev->blkdev;
  118. bio->bi_sector = page->index << (PAGE_SHIFT-9);
  119. bio->bi_private = &event;
  120. bio->bi_end_io = bi_read_complete;
  121. if(bio_add_page(bio, page, PAGE_SIZE, 0) == PAGE_SIZE) {
  122. submit_bio(READ_SYNC, bio);
  123. wait_for_completion(&event);
  124. err = test_bit(BIO_UPTODATE, &bio->bi_flags) ? 0 : -EIO;
  125. bio_put(bio);
  126. }
  127. }
  128. if(err)
  129. SetPageError(page);
  130. else
  131. SetPageUptodate(page);
  132. flush_dcache_page(page);
  133. unlock_page(page);
  134. return err;
  135. }
  136. /* write out the current BIO and wait for it to finish */
  137. static int blkmtd_write_out(struct bio *bio)
  138. {
  139. struct completion event;
  140. int err;
  141. if(!bio->bi_vcnt) {
  142. bio_put(bio);
  143. return 0;
  144. }
  145. init_completion(&event);
  146. bio->bi_private = &event;
  147. bio->bi_end_io = bi_write_complete;
  148. submit_bio(WRITE_SYNC, bio);
  149. wait_for_completion(&event);
  150. DEBUG(3, "submit_bio completed, bi_vcnt = %d\n", bio->bi_vcnt);
  151. err = test_bit(BIO_UPTODATE, &bio->bi_flags) ? 0 : -EIO;
  152. bio_put(bio);
  153. return err;
  154. }
  155. /**
  156. * blkmtd_add_page - add a page to the current BIO
  157. * @bio: bio to add to (NULL to alloc initial bio)
  158. * @blkdev: block device
  159. * @page: page to add
  160. * @pagecnt: pages left to add
  161. *
  162. * Adds a page to the current bio, allocating it if necessary. If it cannot be
  163. * added, the current bio is written out and a new one is allocated. Returns
  164. * the new bio to add or NULL on error
  165. */
  166. static struct bio *blkmtd_add_page(struct bio *bio, struct block_device *blkdev,
  167. struct page *page, int pagecnt)
  168. {
  169. retry:
  170. if(!bio) {
  171. bio = bio_alloc(GFP_KERNEL, pagecnt);
  172. if(!bio)
  173. return NULL;
  174. bio->bi_sector = page->index << (PAGE_SHIFT-9);
  175. bio->bi_bdev = blkdev;
  176. }
  177. if(bio_add_page(bio, page, PAGE_SIZE, 0) != PAGE_SIZE) {
  178. blkmtd_write_out(bio);
  179. bio = NULL;
  180. goto retry;
  181. }
  182. return bio;
  183. }
  184. /**
  185. * write_pages - write block of data to device via the page cache
  186. * @dev: device to write to
  187. * @buf: data source or NULL if erase (output is set to 0xff)
  188. * @to: offset into output device
  189. * @len: amount to data to write
  190. * @retlen: amount of data written
  191. *
  192. * Grab pages from the page cache and fill them with the source data.
  193. * Non page aligned start and end result in a readin of the page and
  194. * part of the page being modified. Pages are added to the bio and then written
  195. * out.
  196. */
  197. static int write_pages(struct blkmtd_dev *dev, const u_char *buf, loff_t to,
  198. size_t len, size_t *retlen)
  199. {
  200. int pagenr, offset;
  201. size_t start_len = 0, end_len;
  202. int pagecnt = 0;
  203. int err = 0;
  204. struct bio *bio = NULL;
  205. size_t thislen = 0;
  206. pagenr = to >> PAGE_SHIFT;
  207. offset = to & ~PAGE_MASK;
  208. DEBUG(2, "blkmtd: write_pages: buf = %p to = %ld len = %zd pagenr = %d offset = %d\n",
  209. buf, (long)to, len, pagenr, offset);
  210. /* see if we have to do a partial write at the start */
  211. if(offset) {
  212. start_len = ((offset + len) > PAGE_SIZE) ? PAGE_SIZE - offset : len;
  213. len -= start_len;
  214. }
  215. /* calculate the length of the other two regions */
  216. end_len = len & ~PAGE_MASK;
  217. len -= end_len;
  218. if(start_len)
  219. pagecnt++;
  220. if(len)
  221. pagecnt += len >> PAGE_SHIFT;
  222. if(end_len)
  223. pagecnt++;
  224. mutex_lock(&dev->wrbuf_mutex);
  225. DEBUG(3, "blkmtd: write: start_len = %zd len = %zd end_len = %zd pagecnt = %d\n",
  226. start_len, len, end_len, pagecnt);
  227. if(start_len) {
  228. /* do partial start region */
  229. struct page *page;
  230. DEBUG(3, "blkmtd: write: doing partial start, page = %d len = %zd offset = %d\n",
  231. pagenr, start_len, offset);
  232. BUG_ON(!buf);
  233. page = read_cache_page(dev->blkdev->bd_inode->i_mapping, pagenr, (filler_t *)blkmtd_readpage, dev);
  234. lock_page(page);
  235. if(PageDirty(page)) {
  236. err("to = %lld start_len = %zd len = %zd end_len = %zd pagenr = %d\n",
  237. to, start_len, len, end_len, pagenr);
  238. BUG();
  239. }
  240. memcpy(page_address(page)+offset, buf, start_len);
  241. set_page_dirty(page);
  242. SetPageUptodate(page);
  243. buf += start_len;
  244. thislen = start_len;
  245. bio = blkmtd_add_page(bio, dev->blkdev, page, pagecnt);
  246. if(!bio) {
  247. err = -ENOMEM;
  248. err("bio_add_page failed\n");
  249. goto write_err;
  250. }
  251. pagecnt--;
  252. pagenr++;
  253. }
  254. /* Now do the main loop to a page aligned, n page sized output */
  255. if(len) {
  256. int pagesc = len >> PAGE_SHIFT;
  257. DEBUG(3, "blkmtd: write: whole pages start = %d, count = %d\n",
  258. pagenr, pagesc);
  259. while(pagesc) {
  260. struct page *page;
  261. /* see if page is in the page cache */
  262. DEBUG(3, "blkmtd: write: grabbing page %d from page cache\n", pagenr);
  263. page = grab_cache_page(dev->blkdev->bd_inode->i_mapping, pagenr);
  264. if(PageDirty(page)) {
  265. BUG();
  266. }
  267. if(!page) {
  268. warn("write: cannot grab cache page %d", pagenr);
  269. err = -ENOMEM;
  270. goto write_err;
  271. }
  272. if(!buf) {
  273. memset(page_address(page), 0xff, PAGE_SIZE);
  274. } else {
  275. memcpy(page_address(page), buf, PAGE_SIZE);
  276. buf += PAGE_SIZE;
  277. }
  278. bio = blkmtd_add_page(bio, dev->blkdev, page, pagecnt);
  279. if(!bio) {
  280. err = -ENOMEM;
  281. err("bio_add_page failed\n");
  282. goto write_err;
  283. }
  284. pagenr++;
  285. pagecnt--;
  286. set_page_dirty(page);
  287. SetPageUptodate(page);
  288. pagesc--;
  289. thislen += PAGE_SIZE;
  290. }
  291. }
  292. if(end_len) {
  293. /* do the third region */
  294. struct page *page;
  295. DEBUG(3, "blkmtd: write: doing partial end, page = %d len = %zd\n",
  296. pagenr, end_len);
  297. BUG_ON(!buf);
  298. page = read_cache_page(dev->blkdev->bd_inode->i_mapping, pagenr, (filler_t *)blkmtd_readpage, dev);
  299. lock_page(page);
  300. if(PageDirty(page)) {
  301. err("to = %lld start_len = %zd len = %zd end_len = %zd pagenr = %d\n",
  302. to, start_len, len, end_len, pagenr);
  303. BUG();
  304. }
  305. memcpy(page_address(page), buf, end_len);
  306. set_page_dirty(page);
  307. SetPageUptodate(page);
  308. DEBUG(3, "blkmtd: write: writing out partial end\n");
  309. thislen += end_len;
  310. bio = blkmtd_add_page(bio, dev->blkdev, page, pagecnt);
  311. if(!bio) {
  312. err = -ENOMEM;
  313. err("bio_add_page failed\n");
  314. goto write_err;
  315. }
  316. pagenr++;
  317. }
  318. DEBUG(3, "blkmtd: write: got %d vectors to write\n", bio->bi_vcnt);
  319. write_err:
  320. if(bio)
  321. blkmtd_write_out(bio);
  322. DEBUG(2, "blkmtd: write: end, retlen = %zd, err = %d\n", *retlen, err);
  323. mutex_unlock(&dev->wrbuf_mutex);
  324. if(retlen)
  325. *retlen = thislen;
  326. return err;
  327. }
  328. /* erase a specified part of the device */
  329. static int blkmtd_erase(struct mtd_info *mtd, struct erase_info *instr)
  330. {
  331. struct blkmtd_dev *dev = mtd->priv;
  332. struct mtd_erase_region_info *einfo = mtd->eraseregions;
  333. int numregions = mtd->numeraseregions;
  334. size_t from;
  335. u_long len;
  336. int err = -EIO;
  337. size_t retlen;
  338. instr->state = MTD_ERASING;
  339. from = instr->addr;
  340. len = instr->len;
  341. /* check erase region has valid start and length */
  342. DEBUG(2, "blkmtd: erase: dev = `%s' from = 0x%zx len = 0x%lx\n",
  343. mtd->name+9, from, len);
  344. while(numregions) {
  345. DEBUG(3, "blkmtd: checking erase region = 0x%08X size = 0x%X num = 0x%x\n",
  346. einfo->offset, einfo->erasesize, einfo->numblocks);
  347. if(from >= einfo->offset
  348. && from < einfo->offset + (einfo->erasesize * einfo->numblocks)) {
  349. if(len == einfo->erasesize
  350. && ( (from - einfo->offset) % einfo->erasesize == 0))
  351. break;
  352. }
  353. numregions--;
  354. einfo++;
  355. }
  356. if(!numregions) {
  357. /* Not a valid erase block */
  358. err("erase: invalid erase request 0x%lX @ 0x%08zX", len, from);
  359. instr->state = MTD_ERASE_FAILED;
  360. err = -EIO;
  361. }
  362. if(instr->state != MTD_ERASE_FAILED) {
  363. /* do the erase */
  364. DEBUG(3, "Doing erase from = %zd len = %ld\n", from, len);
  365. err = write_pages(dev, NULL, from, len, &retlen);
  366. if(err || retlen != len) {
  367. err("erase failed err = %d", err);
  368. instr->state = MTD_ERASE_FAILED;
  369. } else {
  370. instr->state = MTD_ERASE_DONE;
  371. }
  372. }
  373. DEBUG(3, "blkmtd: erase: checking callback\n");
  374. mtd_erase_callback(instr);
  375. DEBUG(2, "blkmtd: erase: finished (err = %d)\n", err);
  376. return err;
  377. }
  378. /* read a range of the data via the page cache */
  379. static int blkmtd_read(struct mtd_info *mtd, loff_t from, size_t len,
  380. size_t *retlen, u_char *buf)
  381. {
  382. struct blkmtd_dev *dev = mtd->priv;
  383. int err = 0;
  384. int offset;
  385. int pagenr, pages;
  386. size_t thislen = 0;
  387. DEBUG(2, "blkmtd: read: dev = `%s' from = %lld len = %zd buf = %p\n",
  388. mtd->name+9, from, len, buf);
  389. if(from > mtd->size)
  390. return -EINVAL;
  391. if(from + len > mtd->size)
  392. len = mtd->size - from;
  393. pagenr = from >> PAGE_SHIFT;
  394. offset = from - (pagenr << PAGE_SHIFT);
  395. pages = (offset+len+PAGE_SIZE-1) >> PAGE_SHIFT;
  396. DEBUG(3, "blkmtd: read: pagenr = %d offset = %d, pages = %d\n",
  397. pagenr, offset, pages);
  398. while(pages) {
  399. struct page *page;
  400. int cpylen;
  401. DEBUG(3, "blkmtd: read: looking for page: %d\n", pagenr);
  402. page = read_cache_page(dev->blkdev->bd_inode->i_mapping, pagenr, (filler_t *)blkmtd_readpage, dev);
  403. if(IS_ERR(page)) {
  404. err = -EIO;
  405. goto readerr;
  406. }
  407. cpylen = (PAGE_SIZE > len) ? len : PAGE_SIZE;
  408. if(offset+cpylen > PAGE_SIZE)
  409. cpylen = PAGE_SIZE-offset;
  410. memcpy(buf + thislen, page_address(page) + offset, cpylen);
  411. offset = 0;
  412. len -= cpylen;
  413. thislen += cpylen;
  414. pagenr++;
  415. pages--;
  416. if(!PageDirty(page))
  417. page_cache_release(page);
  418. }
  419. readerr:
  420. if(retlen)
  421. *retlen = thislen;
  422. DEBUG(2, "blkmtd: end read: retlen = %zd, err = %d\n", thislen, err);
  423. return err;
  424. }
  425. /* write data to the underlying device */
  426. static int blkmtd_write(struct mtd_info *mtd, loff_t to, size_t len,
  427. size_t *retlen, const u_char *buf)
  428. {
  429. struct blkmtd_dev *dev = mtd->priv;
  430. int err;
  431. if(!len)
  432. return 0;
  433. DEBUG(2, "blkmtd: write: dev = `%s' to = %lld len = %zd buf = %p\n",
  434. mtd->name+9, to, len, buf);
  435. if(to >= mtd->size) {
  436. return -ENOSPC;
  437. }
  438. if(to + len > mtd->size) {
  439. len = mtd->size - to;
  440. }
  441. err = write_pages(dev, buf, to, len, retlen);
  442. if(err > 0)
  443. err = 0;
  444. DEBUG(2, "blkmtd: write: end, err = %d\n", err);
  445. return err;
  446. }
  447. /* sync the device - wait until the write queue is empty */
  448. static void blkmtd_sync(struct mtd_info *mtd)
  449. {
  450. /* Currently all writes are synchronous */
  451. }
  452. static void free_device(struct blkmtd_dev *dev)
  453. {
  454. DEBUG(2, "blkmtd: free_device() dev = %p\n", dev);
  455. if(dev) {
  456. kfree(dev->mtd_info.eraseregions);
  457. kfree(dev->mtd_info.name);
  458. if(dev->blkdev) {
  459. invalidate_inode_pages(dev->blkdev->bd_inode->i_mapping);
  460. close_bdev_excl(dev->blkdev);
  461. }
  462. kfree(dev);
  463. }
  464. }
  465. /* For a given size and initial erase size, calculate the number
  466. * and size of each erase region. Goes round the loop twice,
  467. * once to find out how many regions, then allocates space,
  468. * then round the loop again to fill it in.
  469. */
  470. static struct mtd_erase_region_info *calc_erase_regions(
  471. size_t erase_size, size_t total_size, int *regions)
  472. {
  473. struct mtd_erase_region_info *info = NULL;
  474. DEBUG(2, "calc_erase_regions, es = %zd size = %zd regions = %d\n",
  475. erase_size, total_size, *regions);
  476. /* Make any user specified erasesize be a power of 2
  477. and at least PAGE_SIZE */
  478. if(erase_size) {
  479. int es = erase_size;
  480. erase_size = 1;
  481. while(es != 1) {
  482. es >>= 1;
  483. erase_size <<= 1;
  484. }
  485. if(erase_size < PAGE_SIZE)
  486. erase_size = PAGE_SIZE;
  487. } else {
  488. erase_size = CONFIG_MTD_BLKDEV_ERASESIZE;
  489. }
  490. *regions = 0;
  491. do {
  492. int tot_size = total_size;
  493. int er_size = erase_size;
  494. int count = 0, offset = 0, regcnt = 0;
  495. while(tot_size) {
  496. count = tot_size / er_size;
  497. if(count) {
  498. tot_size = tot_size % er_size;
  499. if(info) {
  500. DEBUG(2, "adding to erase info off=%d er=%d cnt=%d\n",
  501. offset, er_size, count);
  502. (info+regcnt)->offset = offset;
  503. (info+regcnt)->erasesize = er_size;
  504. (info+regcnt)->numblocks = count;
  505. (*regions)++;
  506. }
  507. regcnt++;
  508. offset += (count * er_size);
  509. }
  510. while(er_size > tot_size)
  511. er_size >>= 1;
  512. }
  513. if(info == NULL) {
  514. info = kmalloc(regcnt * sizeof(struct mtd_erase_region_info), GFP_KERNEL);
  515. if(!info)
  516. break;
  517. }
  518. } while(!(*regions));
  519. DEBUG(2, "calc_erase_regions done, es = %zd size = %zd regions = %d\n",
  520. erase_size, total_size, *regions);
  521. return info;
  522. }
  523. extern dev_t __init name_to_dev_t(const char *line);
  524. static struct blkmtd_dev *add_device(char *devname, int readonly, int erase_size)
  525. {
  526. struct block_device *bdev;
  527. int mode;
  528. struct blkmtd_dev *dev;
  529. if(!devname)
  530. return NULL;
  531. /* Get a handle on the device */
  532. #ifdef MODULE
  533. mode = (readonly) ? O_RDONLY : O_RDWR;
  534. bdev = open_bdev_excl(devname, mode, NULL);
  535. #else
  536. mode = (readonly) ? FMODE_READ : FMODE_WRITE;
  537. bdev = open_by_devnum(name_to_dev_t(devname), mode);
  538. #endif
  539. if(IS_ERR(bdev)) {
  540. err("error: cannot open device %s", devname);
  541. DEBUG(2, "blkmtd: opening bdev returned %ld\n", PTR_ERR(bdev));
  542. return NULL;
  543. }
  544. DEBUG(1, "blkmtd: found a block device major = %d, minor = %d\n",
  545. MAJOR(bdev->bd_dev), MINOR(bdev->bd_dev));
  546. if(MAJOR(bdev->bd_dev) == MTD_BLOCK_MAJOR) {
  547. err("attempting to use an MTD device as a block device");
  548. blkdev_put(bdev);
  549. return NULL;
  550. }
  551. dev = kmalloc(sizeof(struct blkmtd_dev), GFP_KERNEL);
  552. if(dev == NULL) {
  553. blkdev_put(bdev);
  554. return NULL;
  555. }
  556. memset(dev, 0, sizeof(struct blkmtd_dev));
  557. dev->blkdev = bdev;
  558. if(!readonly) {
  559. mutex_init(&dev->wrbuf_mutex);
  560. }
  561. dev->mtd_info.size = dev->blkdev->bd_inode->i_size & PAGE_MASK;
  562. /* Setup the MTD structure */
  563. /* make the name contain the block device in */
  564. dev->mtd_info.name = kmalloc(sizeof("blkmtd: ") + strlen(devname), GFP_KERNEL);
  565. if(dev->mtd_info.name == NULL)
  566. goto devinit_err;
  567. sprintf(dev->mtd_info.name, "blkmtd: %s", devname);
  568. dev->mtd_info.eraseregions = calc_erase_regions(erase_size, dev->mtd_info.size,
  569. &dev->mtd_info.numeraseregions);
  570. if(dev->mtd_info.eraseregions == NULL)
  571. goto devinit_err;
  572. dev->mtd_info.erasesize = dev->mtd_info.eraseregions->erasesize;
  573. DEBUG(1, "blkmtd: init: found %d erase regions\n",
  574. dev->mtd_info.numeraseregions);
  575. if(readonly) {
  576. dev->mtd_info.type = MTD_ROM;
  577. dev->mtd_info.flags = MTD_CAP_ROM;
  578. } else {
  579. dev->mtd_info.type = MTD_RAM;
  580. dev->mtd_info.flags = MTD_CAP_RAM;
  581. dev->mtd_info.erase = blkmtd_erase;
  582. dev->mtd_info.write = blkmtd_write;
  583. dev->mtd_info.writev = default_mtd_writev;
  584. dev->mtd_info.sync = blkmtd_sync;
  585. }
  586. dev->mtd_info.read = blkmtd_read;
  587. dev->mtd_info.readv = default_mtd_readv;
  588. dev->mtd_info.priv = dev;
  589. dev->mtd_info.owner = THIS_MODULE;
  590. list_add(&dev->list, &blkmtd_device_list);
  591. if (add_mtd_device(&dev->mtd_info)) {
  592. /* Device didnt get added, so free the entry */
  593. list_del(&dev->list);
  594. goto devinit_err;
  595. } else {
  596. info("mtd%d: [%s] erase_size = %dKiB %s",
  597. dev->mtd_info.index, dev->mtd_info.name + strlen("blkmtd: "),
  598. dev->mtd_info.erasesize >> 10,
  599. readonly ? "(read-only)" : "");
  600. }
  601. return dev;
  602. devinit_err:
  603. free_device(dev);
  604. return NULL;
  605. }
  606. /* Cleanup and exit - sync the device and kill of the kernel thread */
  607. static void __devexit cleanup_blkmtd(void)
  608. {
  609. struct list_head *temp1, *temp2;
  610. /* Remove the MTD devices */
  611. list_for_each_safe(temp1, temp2, &blkmtd_device_list) {
  612. struct blkmtd_dev *dev = list_entry(temp1, struct blkmtd_dev,
  613. list);
  614. blkmtd_sync(&dev->mtd_info);
  615. del_mtd_device(&dev->mtd_info);
  616. info("mtd%d: [%s] removed", dev->mtd_info.index,
  617. dev->mtd_info.name + strlen("blkmtd: "));
  618. list_del(&dev->list);
  619. free_device(dev);
  620. }
  621. }
  622. #ifndef MODULE
  623. /* Handle kernel boot params */
  624. static int __init param_blkmtd_device(char *str)
  625. {
  626. int i;
  627. for(i = 0; i < MAX_DEVICES; i++) {
  628. device[i] = str;
  629. DEBUG(2, "blkmtd: device setup: %d = %s\n", i, device[i]);
  630. strsep(&str, ",");
  631. }
  632. return 1;
  633. }
  634. static int __init param_blkmtd_erasesz(char *str)
  635. {
  636. int i;
  637. for(i = 0; i < MAX_DEVICES; i++) {
  638. char *val = strsep(&str, ",");
  639. if(val)
  640. erasesz[i] = simple_strtoul(val, NULL, 0);
  641. DEBUG(2, "blkmtd: erasesz setup: %d = %d\n", i, erasesz[i]);
  642. }
  643. return 1;
  644. }
  645. static int __init param_blkmtd_ro(char *str)
  646. {
  647. int i;
  648. for(i = 0; i < MAX_DEVICES; i++) {
  649. char *val = strsep(&str, ",");
  650. if(val)
  651. ro[i] = simple_strtoul(val, NULL, 0);
  652. DEBUG(2, "blkmtd: ro setup: %d = %d\n", i, ro[i]);
  653. }
  654. return 1;
  655. }
  656. static int __init param_blkmtd_sync(char *str)
  657. {
  658. if(str[0] == '1')
  659. sync = 1;
  660. return 1;
  661. }
  662. __setup("blkmtd_device=", param_blkmtd_device);
  663. __setup("blkmtd_erasesz=", param_blkmtd_erasesz);
  664. __setup("blkmtd_ro=", param_blkmtd_ro);
  665. __setup("blkmtd_sync=", param_blkmtd_sync);
  666. #endif
  667. /* Startup */
  668. static int __init init_blkmtd(void)
  669. {
  670. int i;
  671. info("version " VERSION);
  672. /* Check args - device[0] is the bare minimum*/
  673. if(!device[0]) {
  674. err("error: missing `device' name\n");
  675. return -EINVAL;
  676. }
  677. for(i = 0; i < MAX_DEVICES; i++)
  678. add_device(device[i], ro[i], erasesz[i] << 10);
  679. if(list_empty(&blkmtd_device_list))
  680. return -EINVAL;
  681. return 0;
  682. }
  683. module_init(init_blkmtd);
  684. module_exit(cleanup_blkmtd);