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