Commit | Line | Data |
---|---|---|
1da177e4 | 1 | /* |
a1452a37 | 2 | * Interface to Linux block layer for MTD 'translation layers'. |
1da177e4 | 3 | * |
a1452a37 DW |
4 | * Copyright © 2003-2010 David Woodhouse <dwmw2@infradead.org> |
5 | * | |
6 | * This program is free software; you can redistribute it and/or modify | |
7 | * it under the terms of the GNU General Public License as published by | |
8 | * the Free Software Foundation; either version 2 of the License, or | |
9 | * (at your option) any later version. | |
10 | * | |
11 | * This program is distributed in the hope that it will be useful, | |
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 | * GNU General Public License for more details. | |
15 | * | |
16 | * You should have received a copy of the GNU General Public License | |
17 | * along with this program; if not, write to the Free Software | |
18 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
1da177e4 LT |
19 | * |
20 | */ | |
21 | ||
22 | #include <linux/kernel.h> | |
23 | #include <linux/slab.h> | |
24 | #include <linux/module.h> | |
25 | #include <linux/list.h> | |
26 | #include <linux/fs.h> | |
27 | #include <linux/mtd/blktrans.h> | |
28 | #include <linux/mtd/mtd.h> | |
29 | #include <linux/blkdev.h> | |
30 | #include <linux/blkpg.h> | |
31 | #include <linux/spinlock.h> | |
32 | #include <linux/hdreg.h> | |
33 | #include <linux/init.h> | |
48b19268 | 34 | #include <linux/mutex.h> |
99f9b243 | 35 | #include <linux/kthread.h> |
1da177e4 | 36 | #include <asm/uaccess.h> |
1da177e4 | 37 | |
356d70f1 | 38 | #include "mtdcore.h" |
1da177e4 | 39 | |
356d70f1 | 40 | static LIST_HEAD(blktrans_majors); |
048d8719 ML |
41 | static DEFINE_MUTEX(blktrans_ref_mutex); |
42 | ||
7f53f12f | 43 | static void blktrans_dev_release(struct kref *kref) |
048d8719 ML |
44 | { |
45 | struct mtd_blktrans_dev *dev = | |
46 | container_of(kref, struct mtd_blktrans_dev, ref); | |
47 | ||
48 | dev->disk->private_data = NULL; | |
e4d64cab | 49 | blk_cleanup_queue(dev->rq); |
048d8719 ML |
50 | put_disk(dev->disk); |
51 | list_del(&dev->list); | |
52 | kfree(dev); | |
53 | } | |
54 | ||
55 | static struct mtd_blktrans_dev *blktrans_dev_get(struct gendisk *disk) | |
56 | { | |
57 | struct mtd_blktrans_dev *dev; | |
58 | ||
59 | mutex_lock(&blktrans_ref_mutex); | |
60 | dev = disk->private_data; | |
61 | ||
62 | if (!dev) | |
63 | goto unlock; | |
64 | kref_get(&dev->ref); | |
65 | unlock: | |
66 | mutex_unlock(&blktrans_ref_mutex); | |
67 | return dev; | |
68 | } | |
69 | ||
7f53f12f | 70 | static void blktrans_dev_put(struct mtd_blktrans_dev *dev) |
048d8719 ML |
71 | { |
72 | mutex_lock(&blktrans_ref_mutex); | |
73 | kref_put(&dev->ref, blktrans_dev_release); | |
74 | mutex_unlock(&blktrans_ref_mutex); | |
75 | } | |
1da177e4 | 76 | |
1da177e4 LT |
77 | |
78 | static int do_blktrans_request(struct mtd_blktrans_ops *tr, | |
79 | struct mtd_blktrans_dev *dev, | |
80 | struct request *req) | |
81 | { | |
82 | unsigned long block, nsect; | |
83 | char *buf; | |
84 | ||
83096ebf | 85 | block = blk_rq_pos(req) << 9 >> tr->blkshift; |
1011c1b9 | 86 | nsect = blk_rq_cur_bytes(req) >> tr->blkshift; |
19187672 | 87 | |
1da177e4 LT |
88 | buf = req->buffer; |
89 | ||
33659ebb | 90 | if (req->cmd_type != REQ_TYPE_FS) |
f06d9a2b | 91 | return -EIO; |
1da177e4 | 92 | |
83096ebf TH |
93 | if (blk_rq_pos(req) + blk_rq_cur_sectors(req) > |
94 | get_capacity(req->rq_disk)) | |
f06d9a2b | 95 | return -EIO; |
1da177e4 | 96 | |
33659ebb | 97 | if (req->cmd_flags & REQ_DISCARD) |
1122a26f CH |
98 | return tr->discard(dev, block, nsect); |
99 | ||
1da177e4 LT |
100 | switch(rq_data_dir(req)) { |
101 | case READ: | |
19187672 | 102 | for (; nsect > 0; nsect--, block++, buf += tr->blksize) |
1da177e4 | 103 | if (tr->readsect(dev, block, buf)) |
f06d9a2b | 104 | return -EIO; |
2d4dc890 | 105 | rq_flush_dcache_pages(req); |
f06d9a2b | 106 | return 0; |
1da177e4 LT |
107 | case WRITE: |
108 | if (!tr->writesect) | |
f06d9a2b | 109 | return -EIO; |
1da177e4 | 110 | |
2d4dc890 | 111 | rq_flush_dcache_pages(req); |
19187672 | 112 | for (; nsect > 0; nsect--, block++, buf += tr->blksize) |
1da177e4 | 113 | if (tr->writesect(dev, block, buf)) |
f06d9a2b TH |
114 | return -EIO; |
115 | return 0; | |
1da177e4 | 116 | default: |
9a292308 | 117 | printk(KERN_NOTICE "Unknown request %u\n", rq_data_dir(req)); |
f06d9a2b | 118 | return -EIO; |
1da177e4 LT |
119 | } |
120 | } | |
121 | ||
c7519dbf JL |
122 | int mtd_blktrans_cease_background(struct mtd_blktrans_dev *dev) |
123 | { | |
124 | if (kthread_should_stop()) | |
125 | return 1; | |
126 | ||
7bf7e370 | 127 | return dev->bg_stop; |
c7519dbf JL |
128 | } |
129 | EXPORT_SYMBOL_GPL(mtd_blktrans_cease_background); | |
130 | ||
1da177e4 LT |
131 | static int mtd_blktrans_thread(void *arg) |
132 | { | |
a8638622 | 133 | struct mtd_blktrans_dev *dev = arg; |
c7519dbf | 134 | struct mtd_blktrans_ops *tr = dev->tr; |
a8638622 | 135 | struct request_queue *rq = dev->rq; |
1498ada7 | 136 | struct request *req = NULL; |
c7519dbf | 137 | int background_done = 0; |
1da177e4 | 138 | |
1da177e4 | 139 | spin_lock_irq(rq->queue_lock); |
1498ada7 | 140 | |
3e67fe45 | 141 | while (!kthread_should_stop()) { |
f06d9a2b | 142 | int res; |
1da177e4 | 143 | |
7bf7e370 | 144 | dev->bg_stop = false; |
9934c8c0 | 145 | if (!req && !(req = blk_fetch_request(rq))) { |
c7519dbf JL |
146 | if (tr->background && !background_done) { |
147 | spin_unlock_irq(rq->queue_lock); | |
148 | mutex_lock(&dev->lock); | |
149 | tr->background(dev); | |
150 | mutex_unlock(&dev->lock); | |
151 | spin_lock_irq(rq->queue_lock); | |
152 | /* | |
153 | * Do background processing just once per idle | |
154 | * period. | |
155 | */ | |
7bf7e370 | 156 | background_done = !dev->bg_stop; |
c7519dbf JL |
157 | continue; |
158 | } | |
1da177e4 | 159 | set_current_state(TASK_INTERRUPTIBLE); |
12aebf3e ML |
160 | |
161 | if (kthread_should_stop()) | |
162 | set_current_state(TASK_RUNNING); | |
163 | ||
1da177e4 | 164 | spin_unlock_irq(rq->queue_lock); |
1da177e4 | 165 | schedule(); |
1da177e4 | 166 | spin_lock_irq(rq->queue_lock); |
1da177e4 LT |
167 | continue; |
168 | } | |
169 | ||
1da177e4 LT |
170 | spin_unlock_irq(rq->queue_lock); |
171 | ||
48b19268 | 172 | mutex_lock(&dev->lock); |
a8638622 | 173 | res = do_blktrans_request(dev->tr, dev, req); |
48b19268 | 174 | mutex_unlock(&dev->lock); |
1da177e4 LT |
175 | |
176 | spin_lock_irq(rq->queue_lock); | |
177 | ||
1498ada7 TH |
178 | if (!__blk_end_request_cur(req, res)) |
179 | req = NULL; | |
c7519dbf JL |
180 | |
181 | background_done = 0; | |
1da177e4 | 182 | } |
1498ada7 TH |
183 | |
184 | if (req) | |
185 | __blk_end_request_all(req, -EIO); | |
186 | ||
1da177e4 LT |
187 | spin_unlock_irq(rq->queue_lock); |
188 | ||
3e67fe45 | 189 | return 0; |
1da177e4 LT |
190 | } |
191 | ||
192 | static void mtd_blktrans_request(struct request_queue *rq) | |
193 | { | |
048d8719 ML |
194 | struct mtd_blktrans_dev *dev; |
195 | struct request *req = NULL; | |
196 | ||
197 | dev = rq->queuedata; | |
1da177e4 | 198 | |
048d8719 ML |
199 | if (!dev) |
200 | while ((req = blk_fetch_request(rq)) != NULL) | |
201 | __blk_end_request_all(req, -ENODEV); | |
7bf7e370 AB |
202 | else { |
203 | dev->bg_stop = true; | |
048d8719 | 204 | wake_up_process(dev->thread); |
7bf7e370 | 205 | } |
048d8719 | 206 | } |
1da177e4 | 207 | |
af0e2a0a | 208 | static int blktrans_open(struct block_device *bdev, fmode_t mode) |
1da177e4 | 209 | { |
048d8719 | 210 | struct mtd_blktrans_dev *dev = blktrans_dev_get(bdev->bd_disk); |
008c751e | 211 | int ret = 0; |
048d8719 ML |
212 | |
213 | if (!dev) | |
6e9624b8 | 214 | return -ERESTARTSYS; /* FIXME: busy loop! -arnd*/ |
048d8719 ML |
215 | |
216 | mutex_lock(&dev->lock); | |
217 | ||
342ff28f | 218 | if (dev->open) |
048d8719 | 219 | goto unlock; |
048d8719 | 220 | |
008c751e ML |
221 | kref_get(&dev->ref); |
222 | __module_get(dev->tr->owner); | |
223 | ||
94735ec4 AB |
224 | if (!dev->mtd) |
225 | goto unlock; | |
226 | ||
227 | if (dev->tr->open) { | |
228 | ret = dev->tr->open(dev); | |
229 | if (ret) | |
230 | goto error_put; | |
008c751e | 231 | } |
048d8719 | 232 | |
94735ec4 AB |
233 | ret = __get_mtd_device(dev->mtd); |
234 | if (ret) | |
235 | goto error_release; | |
236 | ||
048d8719 | 237 | unlock: |
342ff28f | 238 | dev->open++; |
048d8719 ML |
239 | mutex_unlock(&dev->lock); |
240 | blktrans_dev_put(dev); | |
1da177e4 | 241 | return ret; |
94735ec4 AB |
242 | |
243 | error_release: | |
244 | if (dev->tr->release) | |
245 | dev->tr->release(dev); | |
246 | error_put: | |
247 | module_put(dev->tr->owner); | |
248 | kref_put(&dev->ref, blktrans_dev_release); | |
249 | mutex_unlock(&dev->lock); | |
250 | blktrans_dev_put(dev); | |
251 | return ret; | |
1da177e4 LT |
252 | } |
253 | ||
af0e2a0a | 254 | static int blktrans_release(struct gendisk *disk, fmode_t mode) |
1da177e4 | 255 | { |
048d8719 | 256 | struct mtd_blktrans_dev *dev = blktrans_dev_get(disk); |
008c751e | 257 | int ret = 0; |
1da177e4 | 258 | |
048d8719 ML |
259 | if (!dev) |
260 | return ret; | |
1da177e4 | 261 | |
048d8719 ML |
262 | mutex_lock(&dev->lock); |
263 | ||
008c751e | 264 | if (--dev->open) |
048d8719 ML |
265 | goto unlock; |
266 | ||
008c751e ML |
267 | kref_put(&dev->ref, blktrans_dev_release); |
268 | module_put(dev->tr->owner); | |
269 | ||
270 | if (dev->mtd) { | |
271 | ret = dev->tr->release ? dev->tr->release(dev) : 0; | |
272 | __put_mtd_device(dev->mtd); | |
273 | } | |
048d8719 ML |
274 | unlock: |
275 | mutex_unlock(&dev->lock); | |
276 | blktrans_dev_put(dev); | |
1da177e4 LT |
277 | return ret; |
278 | } | |
279 | ||
a885c8c4 CH |
280 | static int blktrans_getgeo(struct block_device *bdev, struct hd_geometry *geo) |
281 | { | |
048d8719 ML |
282 | struct mtd_blktrans_dev *dev = blktrans_dev_get(bdev->bd_disk); |
283 | int ret = -ENXIO; | |
284 | ||
285 | if (!dev) | |
286 | return ret; | |
287 | ||
288 | mutex_lock(&dev->lock); | |
a885c8c4 | 289 | |
048d8719 ML |
290 | if (!dev->mtd) |
291 | goto unlock; | |
292 | ||
293 | ret = dev->tr->getgeo ? dev->tr->getgeo(dev, geo) : 0; | |
294 | unlock: | |
295 | mutex_unlock(&dev->lock); | |
296 | blktrans_dev_put(dev); | |
297 | return ret; | |
a885c8c4 | 298 | } |
1da177e4 | 299 | |
af0e2a0a | 300 | static int blktrans_ioctl(struct block_device *bdev, fmode_t mode, |
1da177e4 LT |
301 | unsigned int cmd, unsigned long arg) |
302 | { | |
048d8719 ML |
303 | struct mtd_blktrans_dev *dev = blktrans_dev_get(bdev->bd_disk); |
304 | int ret = -ENXIO; | |
305 | ||
306 | if (!dev) | |
307 | return ret; | |
308 | ||
309 | mutex_lock(&dev->lock); | |
310 | ||
311 | if (!dev->mtd) | |
312 | goto unlock; | |
1da177e4 LT |
313 | |
314 | switch (cmd) { | |
315 | case BLKFLSBUF: | |
048d8719 | 316 | ret = dev->tr->flush ? dev->tr->flush(dev) : 0; |
007c2d87 | 317 | break; |
1da177e4 | 318 | default: |
048d8719 | 319 | ret = -ENOTTY; |
1da177e4 | 320 | } |
048d8719 ML |
321 | unlock: |
322 | mutex_unlock(&dev->lock); | |
323 | blktrans_dev_put(dev); | |
324 | return ret; | |
1da177e4 LT |
325 | } |
326 | ||
83d5cde4 | 327 | static const struct block_device_operations mtd_blktrans_ops = { |
1da177e4 | 328 | .owner = THIS_MODULE, |
af0e2a0a AV |
329 | .open = blktrans_open, |
330 | .release = blktrans_release, | |
8a6cfeb6 | 331 | .ioctl = blktrans_ioctl, |
a885c8c4 | 332 | .getgeo = blktrans_getgeo, |
1da177e4 LT |
333 | }; |
334 | ||
335 | int add_mtd_blktrans_dev(struct mtd_blktrans_dev *new) | |
336 | { | |
337 | struct mtd_blktrans_ops *tr = new->tr; | |
71a928c0 | 338 | struct mtd_blktrans_dev *d; |
1da177e4 LT |
339 | int last_devnum = -1; |
340 | struct gendisk *gd; | |
a8638622 | 341 | int ret; |
1da177e4 | 342 | |
b3561ea9 | 343 | if (mutex_trylock(&mtd_table_mutex)) { |
48b19268 | 344 | mutex_unlock(&mtd_table_mutex); |
1da177e4 LT |
345 | BUG(); |
346 | } | |
347 | ||
048d8719 | 348 | mutex_lock(&blktrans_ref_mutex); |
71a928c0 | 349 | list_for_each_entry(d, &tr->devs, list) { |
1da177e4 LT |
350 | if (new->devnum == -1) { |
351 | /* Use first free number */ | |
352 | if (d->devnum != last_devnum+1) { | |
353 | /* Found a free devnum. Plug it in here */ | |
354 | new->devnum = last_devnum+1; | |
355 | list_add_tail(&new->list, &d->list); | |
356 | goto added; | |
357 | } | |
358 | } else if (d->devnum == new->devnum) { | |
359 | /* Required number taken */ | |
048d8719 | 360 | mutex_unlock(&blktrans_ref_mutex); |
1da177e4 LT |
361 | return -EBUSY; |
362 | } else if (d->devnum > new->devnum) { | |
363 | /* Required number was free */ | |
364 | list_add_tail(&new->list, &d->list); | |
365 | goto added; | |
97894cda | 366 | } |
1da177e4 LT |
367 | last_devnum = d->devnum; |
368 | } | |
a8638622 ML |
369 | |
370 | ret = -EBUSY; | |
1da177e4 LT |
371 | if (new->devnum == -1) |
372 | new->devnum = last_devnum+1; | |
373 | ||
4d3a8534 BH |
374 | /* Check that the device and any partitions will get valid |
375 | * minor numbers and that the disk naming code below can cope | |
376 | * with this number. */ | |
377 | if (new->devnum > (MINORMASK >> tr->part_bits) || | |
048d8719 ML |
378 | (tr->part_bits && new->devnum >= 27 * 26)) { |
379 | mutex_unlock(&blktrans_ref_mutex); | |
a8638622 | 380 | goto error1; |
048d8719 | 381 | } |
1da177e4 | 382 | |
1da177e4 LT |
383 | list_add_tail(&new->list, &tr->devs); |
384 | added: | |
048d8719 ML |
385 | mutex_unlock(&blktrans_ref_mutex); |
386 | ||
ce37ab42 | 387 | mutex_init(&new->lock); |
048d8719 | 388 | kref_init(&new->ref); |
1da177e4 LT |
389 | if (!tr->writesect) |
390 | new->readonly = 1; | |
391 | ||
a8638622 ML |
392 | /* Create gendisk */ |
393 | ret = -ENOMEM; | |
1da177e4 | 394 | gd = alloc_disk(1 << tr->part_bits); |
a8638622 ML |
395 | |
396 | if (!gd) | |
397 | goto error2; | |
398 | ||
399 | new->disk = gd; | |
400 | gd->private_data = new; | |
1da177e4 LT |
401 | gd->major = tr->major; |
402 | gd->first_minor = (new->devnum) << tr->part_bits; | |
403 | gd->fops = &mtd_blktrans_ops; | |
97894cda | 404 | |
65a8de36 TP |
405 | if (tr->part_bits) |
406 | if (new->devnum < 26) | |
407 | snprintf(gd->disk_name, sizeof(gd->disk_name), | |
408 | "%s%c", tr->name, 'a' + new->devnum); | |
409 | else | |
410 | snprintf(gd->disk_name, sizeof(gd->disk_name), | |
411 | "%s%c%c", tr->name, | |
412 | 'a' - 1 + new->devnum / 26, | |
413 | 'a' + new->devnum % 26); | |
414 | else | |
415 | snprintf(gd->disk_name, sizeof(gd->disk_name), | |
416 | "%s%d", tr->name, new->devnum); | |
1da177e4 | 417 | |
19187672 | 418 | set_capacity(gd, (new->size * tr->blksize) >> 9); |
1da177e4 | 419 | |
a8638622 ML |
420 | /* Create the request queue */ |
421 | spin_lock_init(&new->queue_lock); | |
422 | new->rq = blk_init_queue(mtd_blktrans_request, &new->queue_lock); | |
423 | ||
424 | if (!new->rq) | |
425 | goto error3; | |
426 | ||
427 | new->rq->queuedata = new; | |
428 | blk_queue_logical_block_size(new->rq, tr->blksize); | |
429 | ||
16f7eca5 DM |
430 | queue_flag_set_unlocked(QUEUE_FLAG_NONROT, new->rq); |
431 | ||
115ee88c JL |
432 | if (tr->discard) { |
433 | queue_flag_set_unlocked(QUEUE_FLAG_DISCARD, new->rq); | |
434 | new->rq->limits.max_discard_sectors = UINT_MAX; | |
435 | } | |
a8638622 ML |
436 | |
437 | gd->queue = new->rq; | |
438 | ||
439 | /* Create processing thread */ | |
440 | /* TODO: workqueue ? */ | |
441 | new->thread = kthread_run(mtd_blktrans_thread, new, | |
442 | "%s%d", tr->name, new->mtd->index); | |
443 | if (IS_ERR(new->thread)) { | |
444 | ret = PTR_ERR(new->thread); | |
445 | goto error4; | |
446 | } | |
d694846b | 447 | gd->driverfs_dev = &new->mtd->dev; |
1da177e4 LT |
448 | |
449 | if (new->readonly) | |
450 | set_disk_ro(gd, 1); | |
451 | ||
452 | add_disk(gd); | |
026ec578 | 453 | |
133fa8c7 ML |
454 | if (new->disk_attributes) { |
455 | ret = sysfs_create_group(&disk_to_dev(gd)->kobj, | |
026ec578 | 456 | new->disk_attributes); |
133fa8c7 ML |
457 | WARN_ON(ret); |
458 | } | |
1da177e4 | 459 | return 0; |
a8638622 ML |
460 | error4: |
461 | blk_cleanup_queue(new->rq); | |
462 | error3: | |
463 | put_disk(new->disk); | |
464 | error2: | |
465 | list_del(&new->list); | |
466 | error1: | |
a8638622 | 467 | return ret; |
1da177e4 LT |
468 | } |
469 | ||
470 | int del_mtd_blktrans_dev(struct mtd_blktrans_dev *old) | |
471 | { | |
048d8719 ML |
472 | unsigned long flags; |
473 | ||
b3561ea9 | 474 | if (mutex_trylock(&mtd_table_mutex)) { |
48b19268 | 475 | mutex_unlock(&mtd_table_mutex); |
1da177e4 LT |
476 | BUG(); |
477 | } | |
478 | ||
026ec578 ML |
479 | if (old->disk_attributes) |
480 | sysfs_remove_group(&disk_to_dev(old->disk)->kobj, | |
481 | old->disk_attributes); | |
482 | ||
dba76c03 ML |
483 | /* Stop new requests to arrive */ |
484 | del_gendisk(old->disk); | |
485 | ||
486 | ||
a8638622 ML |
487 | /* Stop the thread */ |
488 | kthread_stop(old->thread); | |
489 | ||
048d8719 ML |
490 | /* Kill current requests */ |
491 | spin_lock_irqsave(&old->queue_lock, flags); | |
492 | old->rq->queuedata = NULL; | |
493 | blk_start_queue(old->rq); | |
494 | spin_unlock_irqrestore(&old->queue_lock, flags); | |
048d8719 | 495 | |
008c751e ML |
496 | /* If the device is currently open, tell trans driver to close it, |
497 | then put mtd device, and don't touch it again */ | |
048d8719 | 498 | mutex_lock(&old->lock); |
008c751e ML |
499 | if (old->open) { |
500 | if (old->tr->release) | |
501 | old->tr->release(old); | |
502 | __put_mtd_device(old->mtd); | |
048d8719 ML |
503 | } |
504 | ||
048d8719 ML |
505 | old->mtd = NULL; |
506 | ||
507 | mutex_unlock(&old->lock); | |
508 | blktrans_dev_put(old); | |
1da177e4 LT |
509 | return 0; |
510 | } | |
511 | ||
512 | static void blktrans_notify_remove(struct mtd_info *mtd) | |
513 | { | |
71a928c0 CM |
514 | struct mtd_blktrans_ops *tr; |
515 | struct mtd_blktrans_dev *dev, *next; | |
1da177e4 | 516 | |
71a928c0 CM |
517 | list_for_each_entry(tr, &blktrans_majors, list) |
518 | list_for_each_entry_safe(dev, next, &tr->devs, list) | |
1da177e4 LT |
519 | if (dev->mtd == mtd) |
520 | tr->remove_dev(dev); | |
1da177e4 LT |
521 | } |
522 | ||
523 | static void blktrans_notify_add(struct mtd_info *mtd) | |
524 | { | |
71a928c0 | 525 | struct mtd_blktrans_ops *tr; |
1da177e4 LT |
526 | |
527 | if (mtd->type == MTD_ABSENT) | |
528 | return; | |
529 | ||
71a928c0 | 530 | list_for_each_entry(tr, &blktrans_majors, list) |
1da177e4 | 531 | tr->add_mtd(tr, mtd); |
1da177e4 LT |
532 | } |
533 | ||
534 | static struct mtd_notifier blktrans_notifier = { | |
535 | .add = blktrans_notify_add, | |
536 | .remove = blktrans_notify_remove, | |
537 | }; | |
97894cda | 538 | |
1da177e4 LT |
539 | int register_mtd_blktrans(struct mtd_blktrans_ops *tr) |
540 | { | |
f1332ba2 BH |
541 | struct mtd_info *mtd; |
542 | int ret; | |
1da177e4 | 543 | |
97894cda | 544 | /* Register the notifier if/when the first device type is |
1da177e4 LT |
545 | registered, to prevent the link/init ordering from fucking |
546 | us over. */ | |
547 | if (!blktrans_notifier.list.next) | |
548 | register_mtd_user(&blktrans_notifier); | |
549 | ||
1da177e4 | 550 | |
48b19268 | 551 | mutex_lock(&mtd_table_mutex); |
1da177e4 LT |
552 | |
553 | ret = register_blkdev(tr->major, tr->name); | |
6fe4c590 | 554 | if (ret < 0) { |
1da177e4 LT |
555 | printk(KERN_WARNING "Unable to register %s block device on major %d: %d\n", |
556 | tr->name, tr->major, ret); | |
48b19268 | 557 | mutex_unlock(&mtd_table_mutex); |
1da177e4 LT |
558 | return ret; |
559 | } | |
eae9acd1 | 560 | |
6fe4c590 FL |
561 | if (ret) |
562 | tr->major = ret; | |
563 | ||
19187672 | 564 | tr->blkshift = ffs(tr->blksize) - 1; |
1da177e4 | 565 | |
1da177e4 LT |
566 | INIT_LIST_HEAD(&tr->devs); |
567 | list_add(&tr->list, &blktrans_majors); | |
568 | ||
f1332ba2 BH |
569 | mtd_for_each_device(mtd) |
570 | if (mtd->type != MTD_ABSENT) | |
571 | tr->add_mtd(tr, mtd); | |
1da177e4 | 572 | |
48b19268 | 573 | mutex_unlock(&mtd_table_mutex); |
1da177e4 LT |
574 | return 0; |
575 | } | |
576 | ||
577 | int deregister_mtd_blktrans(struct mtd_blktrans_ops *tr) | |
578 | { | |
71a928c0 | 579 | struct mtd_blktrans_dev *dev, *next; |
1da177e4 | 580 | |
48b19268 | 581 | mutex_lock(&mtd_table_mutex); |
1da177e4 | 582 | |
1da177e4 LT |
583 | /* Remove it from the list of active majors */ |
584 | list_del(&tr->list); | |
585 | ||
71a928c0 | 586 | list_for_each_entry_safe(dev, next, &tr->devs, list) |
1da177e4 | 587 | tr->remove_dev(dev); |
1da177e4 | 588 | |
1da177e4 | 589 | unregister_blkdev(tr->major, tr->name); |
48b19268 | 590 | mutex_unlock(&mtd_table_mutex); |
1da177e4 | 591 | |
373ebfbf | 592 | BUG_ON(!list_empty(&tr->devs)); |
1da177e4 LT |
593 | return 0; |
594 | } | |
595 | ||
596 | static void __exit mtd_blktrans_exit(void) | |
597 | { | |
598 | /* No race here -- if someone's currently in register_mtd_blktrans | |
599 | we're screwed anyway. */ | |
600 | if (blktrans_notifier.list.next) | |
601 | unregister_mtd_user(&blktrans_notifier); | |
602 | } | |
603 | ||
604 | module_exit(mtd_blktrans_exit); | |
605 | ||
606 | EXPORT_SYMBOL_GPL(register_mtd_blktrans); | |
607 | EXPORT_SYMBOL_GPL(deregister_mtd_blktrans); | |
608 | EXPORT_SYMBOL_GPL(add_mtd_blktrans_dev); | |
609 | EXPORT_SYMBOL_GPL(del_mtd_blktrans_dev); | |
610 | ||
611 | MODULE_AUTHOR("David Woodhouse <dwmw2@infradead.org>"); | |
612 | MODULE_LICENSE("GPL"); | |
613 | MODULE_DESCRIPTION("Common interface to block layer for MTD 'translation layers'"); |