2 * Copyright (c) International Business Machines Corp., 2006
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See
12 * the GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18 * Author: Artem Bityutskiy (Битюцкий Артём), Joern Engel
22 * This is a small driver which implements fake MTD devices on top of UBI
23 * volumes. This sounds strange, but it is in fact quite useful to make
24 * MTD-oriented software (including all the legacy software) work on top of
27 * Gluebi emulates MTD devices of "MTD_UBIVOLUME" type. Their minimal I/O unit
28 * size (@mtd->writesize) is equivalent to the UBI minimal I/O unit. The
29 * eraseblock size is equivalent to the logical eraseblock size of the volume.
32 #include <linux/err.h>
33 #include <linux/list.h>
34 #include <linux/slab.h>
35 #include <linux/sched.h>
36 #include <linux/math64.h>
37 #include <linux/module.h>
38 #include <linux/mutex.h>
39 #include <linux/mtd/ubi.h>
40 #include <linux/mtd/mtd.h>
41 #include "ubi-media.h"
43 #define err_msg(fmt, ...) \
44 printk(KERN_DEBUG "gluebi (pid %d): %s: " fmt "\n", \
45 current->pid, __func__, ##__VA_ARGS__)
48 * struct gluebi_device - a gluebi device description data structure.
49 * @mtd: emulated MTD device description object
50 * @refcnt: gluebi device reference count
51 * @desc: UBI volume descriptor
52 * @ubi_num: UBI device number this gluebi device works on
53 * @vol_id: ID of UBI volume this gluebi device works on
54 * @list: link in a list of gluebi devices
56 struct gluebi_device
{
59 struct ubi_volume_desc
*desc
;
62 struct list_head list
;
65 /* List of all gluebi devices */
66 static LIST_HEAD(gluebi_devices
);
67 static DEFINE_MUTEX(devices_mutex
);
70 * find_gluebi_nolock - find a gluebi device.
71 * @ubi_num: UBI device number
74 * This function seraches for gluebi device corresponding to UBI device
75 * @ubi_num and UBI volume @vol_id. Returns the gluebi device description
76 * object in case of success and %NULL in case of failure. The caller has to
77 * have the &devices_mutex locked.
79 static struct gluebi_device
*find_gluebi_nolock(int ubi_num
, int vol_id
)
81 struct gluebi_device
*gluebi
;
83 list_for_each_entry(gluebi
, &gluebi_devices
, list
)
84 if (gluebi
->ubi_num
== ubi_num
&& gluebi
->vol_id
== vol_id
)
90 * gluebi_get_device - get MTD device reference.
91 * @mtd: the MTD device description object
93 * This function is called every time the MTD device is being opened and
94 * implements the MTD get_device() operation. Returns zero in case of success
95 * and a negative error code in case of failure.
97 static int gluebi_get_device(struct mtd_info
*mtd
)
99 struct gluebi_device
*gluebi
;
100 int ubi_mode
= UBI_READONLY
;
102 if (!try_module_get(THIS_MODULE
))
105 if (mtd
->flags
& MTD_WRITEABLE
)
106 ubi_mode
= UBI_READWRITE
;
108 gluebi
= container_of(mtd
, struct gluebi_device
, mtd
);
109 mutex_lock(&devices_mutex
);
110 if (gluebi
->refcnt
> 0) {
112 * The MTD device is already referenced and this is just one
113 * more reference. MTD allows many users to open the same
114 * volume simultaneously and do not distinguish between
115 * readers/writers/exclusive openers as UBI does. So we do not
116 * open the UBI volume again - just increase the reference
117 * counter and return.
120 mutex_unlock(&devices_mutex
);
125 * This is the first reference to this UBI volume via the MTD device
126 * interface. Open the corresponding volume in read-write mode.
128 gluebi
->desc
= ubi_open_volume(gluebi
->ubi_num
, gluebi
->vol_id
,
130 if (IS_ERR(gluebi
->desc
)) {
131 mutex_unlock(&devices_mutex
);
132 module_put(THIS_MODULE
);
133 return PTR_ERR(gluebi
->desc
);
136 mutex_unlock(&devices_mutex
);
141 * gluebi_put_device - put MTD device reference.
142 * @mtd: the MTD device description object
144 * This function is called every time the MTD device is being put. Returns
145 * zero in case of success and a negative error code in case of failure.
147 static void gluebi_put_device(struct mtd_info
*mtd
)
149 struct gluebi_device
*gluebi
;
151 gluebi
= container_of(mtd
, struct gluebi_device
, mtd
);
152 mutex_lock(&devices_mutex
);
154 if (gluebi
->refcnt
== 0)
155 ubi_close_volume(gluebi
->desc
);
156 module_put(THIS_MODULE
);
157 mutex_unlock(&devices_mutex
);
161 * gluebi_read - read operation of emulated MTD devices.
162 * @mtd: MTD device description object
163 * @from: absolute offset from where to read
164 * @len: how many bytes to read
165 * @retlen: count of read bytes is returned here
166 * @buf: buffer to store the read data
168 * This function returns zero in case of success and a negative error code in
171 static int gluebi_read(struct mtd_info
*mtd
, loff_t from
, size_t len
,
172 size_t *retlen
, unsigned char *buf
)
174 int err
= 0, lnum
, offs
, total_read
;
175 struct gluebi_device
*gluebi
;
177 if (len
< 0 || from
< 0 || from
+ len
> mtd
->size
)
180 gluebi
= container_of(mtd
, struct gluebi_device
, mtd
);
182 lnum
= div_u64_rem(from
, mtd
->erasesize
, &offs
);
185 size_t to_read
= mtd
->erasesize
- offs
;
187 if (to_read
> total_read
)
188 to_read
= total_read
;
190 err
= ubi_read(gluebi
->desc
, lnum
, buf
, offs
, to_read
);
196 total_read
-= to_read
;
200 *retlen
= len
- total_read
;
205 * gluebi_write - write operation of emulated MTD devices.
206 * @mtd: MTD device description object
207 * @to: absolute offset where to write
208 * @len: how many bytes to write
209 * @retlen: count of written bytes is returned here
210 * @buf: buffer with data to write
212 * This function returns zero in case of success and a negative error code in
215 static int gluebi_write(struct mtd_info
*mtd
, loff_t to
, size_t len
,
216 size_t *retlen
, const u_char
*buf
)
218 int err
= 0, lnum
, offs
, total_written
;
219 struct gluebi_device
*gluebi
;
221 if (len
< 0 || to
< 0 || len
+ to
> mtd
->size
)
224 gluebi
= container_of(mtd
, struct gluebi_device
, mtd
);
226 if (!(mtd
->flags
& MTD_WRITEABLE
))
229 lnum
= div_u64_rem(to
, mtd
->erasesize
, &offs
);
231 if (len
% mtd
->writesize
|| offs
% mtd
->writesize
)
235 while (total_written
) {
236 size_t to_write
= mtd
->erasesize
- offs
;
238 if (to_write
> total_written
)
239 to_write
= total_written
;
241 err
= ubi_write(gluebi
->desc
, lnum
, buf
, offs
, to_write
);
247 total_written
-= to_write
;
251 *retlen
= len
- total_written
;
256 * gluebi_erase - erase operation of emulated MTD devices.
257 * @mtd: the MTD device description object
258 * @instr: the erase operation description
260 * This function calls the erase callback when finishes. Returns zero in case
261 * of success and a negative error code in case of failure.
263 static int gluebi_erase(struct mtd_info
*mtd
, struct erase_info
*instr
)
265 int err
, i
, lnum
, count
;
266 struct gluebi_device
*gluebi
;
268 if (instr
->addr
< 0 || instr
->addr
> mtd
->size
- mtd
->erasesize
)
270 if (instr
->len
< 0 || instr
->addr
+ instr
->len
> mtd
->size
)
272 if (mtd_mod_by_ws(instr
->addr
, mtd
) || mtd_mod_by_ws(instr
->len
, mtd
))
275 lnum
= mtd_div_by_eb(instr
->addr
, mtd
);
276 count
= mtd_div_by_eb(instr
->len
, mtd
);
278 gluebi
= container_of(mtd
, struct gluebi_device
, mtd
);
280 if (!(mtd
->flags
& MTD_WRITEABLE
))
283 for (i
= 0; i
< count
- 1; i
++) {
284 err
= ubi_leb_unmap(gluebi
->desc
, lnum
+ i
);
289 * MTD erase operations are synchronous, so we have to make sure the
290 * physical eraseblock is wiped out.
292 * Thus, perform leb_erase instead of leb_unmap operation - leb_erase
293 * will wait for the end of operations
295 err
= ubi_leb_erase(gluebi
->desc
, lnum
+ i
);
299 instr
->state
= MTD_ERASE_DONE
;
300 mtd_erase_callback(instr
);
304 instr
->state
= MTD_ERASE_FAILED
;
305 instr
->fail_addr
= (long long)lnum
* mtd
->erasesize
;
310 * gluebi_create - create a gluebi device for an UBI volume.
311 * @di: UBI device description object
312 * @vi: UBI volume description object
314 * This function is called when a new UBI volume is created in order to create
315 * corresponding fake MTD device. Returns zero in case of success and a
316 * negative error code in case of failure.
318 static int gluebi_create(struct ubi_device_info
*di
,
319 struct ubi_volume_info
*vi
)
321 struct gluebi_device
*gluebi
, *g
;
322 struct mtd_info
*mtd
;
324 gluebi
= kzalloc(sizeof(struct gluebi_device
), GFP_KERNEL
);
329 mtd
->name
= kmemdup(vi
->name
, vi
->name_len
+ 1, GFP_KERNEL
);
335 gluebi
->vol_id
= vi
->vol_id
;
336 gluebi
->ubi_num
= vi
->ubi_num
;
337 mtd
->type
= MTD_UBIVOLUME
;
339 mtd
->flags
= MTD_WRITEABLE
;
340 mtd
->owner
= THIS_MODULE
;
341 mtd
->writesize
= di
->min_io_size
;
342 mtd
->erasesize
= vi
->usable_leb_size
;
343 mtd
->read
= gluebi_read
;
344 mtd
->write
= gluebi_write
;
345 mtd
->erase
= gluebi_erase
;
346 mtd
->get_device
= gluebi_get_device
;
347 mtd
->put_device
= gluebi_put_device
;
350 * In case of dynamic a volume, MTD device size is just volume size. In
351 * case of a static volume the size is equivalent to the amount of data
354 if (vi
->vol_type
== UBI_DYNAMIC_VOLUME
)
355 mtd
->size
= (unsigned long long)vi
->usable_leb_size
* vi
->size
;
357 mtd
->size
= vi
->used_bytes
;
359 /* Just a sanity check - make sure this gluebi device does not exist */
360 mutex_lock(&devices_mutex
);
361 g
= find_gluebi_nolock(vi
->ubi_num
, vi
->vol_id
);
363 err_msg("gluebi MTD device %d form UBI device %d volume %d "
364 "already exists", g
->mtd
.index
, vi
->ubi_num
,
366 mutex_unlock(&devices_mutex
);
368 if (mtd_device_register(mtd
, NULL
, 0)) {
369 err_msg("cannot add MTD device");
375 mutex_lock(&devices_mutex
);
376 list_add_tail(&gluebi
->list
, &gluebi_devices
);
377 mutex_unlock(&devices_mutex
);
382 * gluebi_remove - remove a gluebi device.
383 * @vi: UBI volume description object
385 * This function is called when an UBI volume is removed and it removes
386 * corresponding fake MTD device. Returns zero in case of success and a
387 * negative error code in case of failure.
389 static int gluebi_remove(struct ubi_volume_info
*vi
)
392 struct mtd_info
*mtd
;
393 struct gluebi_device
*gluebi
;
395 mutex_lock(&devices_mutex
);
396 gluebi
= find_gluebi_nolock(vi
->ubi_num
, vi
->vol_id
);
398 err_msg("got remove notification for unknown UBI device %d "
399 "volume %d", vi
->ubi_num
, vi
->vol_id
);
401 } else if (gluebi
->refcnt
)
404 list_del(&gluebi
->list
);
405 mutex_unlock(&devices_mutex
);
410 err
= mtd_device_unregister(mtd
);
412 err_msg("cannot remove fake MTD device %d, UBI device %d, "
413 "volume %d, error %d", mtd
->index
, gluebi
->ubi_num
,
414 gluebi
->vol_id
, err
);
415 mutex_lock(&devices_mutex
);
416 list_add_tail(&gluebi
->list
, &gluebi_devices
);
417 mutex_unlock(&devices_mutex
);
427 * gluebi_updated - UBI volume was updated notifier.
428 * @vi: volume info structure
430 * This function is called every time an UBI volume is updated. It does nothing
431 * if te volume @vol is dynamic, and changes MTD device size if the
432 * volume is static. This is needed because static volumes cannot be read past
433 * data they contain. This function returns zero in case of success and a
434 * negative error code in case of error.
436 static int gluebi_updated(struct ubi_volume_info
*vi
)
438 struct gluebi_device
*gluebi
;
440 mutex_lock(&devices_mutex
);
441 gluebi
= find_gluebi_nolock(vi
->ubi_num
, vi
->vol_id
);
443 mutex_unlock(&devices_mutex
);
444 err_msg("got update notification for unknown UBI device %d "
445 "volume %d", vi
->ubi_num
, vi
->vol_id
);
449 if (vi
->vol_type
== UBI_STATIC_VOLUME
)
450 gluebi
->mtd
.size
= vi
->used_bytes
;
451 mutex_unlock(&devices_mutex
);
456 * gluebi_resized - UBI volume was re-sized notifier.
457 * @vi: volume info structure
459 * This function is called every time an UBI volume is re-size. It changes the
460 * corresponding fake MTD device size. This function returns zero in case of
461 * success and a negative error code in case of error.
463 static int gluebi_resized(struct ubi_volume_info
*vi
)
465 struct gluebi_device
*gluebi
;
467 mutex_lock(&devices_mutex
);
468 gluebi
= find_gluebi_nolock(vi
->ubi_num
, vi
->vol_id
);
470 mutex_unlock(&devices_mutex
);
471 err_msg("got update notification for unknown UBI device %d "
472 "volume %d", vi
->ubi_num
, vi
->vol_id
);
475 gluebi
->mtd
.size
= vi
->used_bytes
;
476 mutex_unlock(&devices_mutex
);
481 * gluebi_notify - UBI notification handler.
482 * @nb: registered notifier block
483 * @l: notification type
484 * @ptr: pointer to the &struct ubi_notification object
486 static int gluebi_notify(struct notifier_block
*nb
, unsigned long l
,
489 struct ubi_notification
*nt
= ns_ptr
;
492 case UBI_VOLUME_ADDED
:
493 gluebi_create(&nt
->di
, &nt
->vi
);
495 case UBI_VOLUME_REMOVED
:
496 gluebi_remove(&nt
->vi
);
498 case UBI_VOLUME_RESIZED
:
499 gluebi_resized(&nt
->vi
);
501 case UBI_VOLUME_UPDATED
:
502 gluebi_updated(&nt
->vi
);
510 static struct notifier_block gluebi_notifier
= {
511 .notifier_call
= gluebi_notify
,
514 static int __init
ubi_gluebi_init(void)
516 return ubi_register_volume_notifier(&gluebi_notifier
, 0);
519 static void __exit
ubi_gluebi_exit(void)
521 struct gluebi_device
*gluebi
, *g
;
523 list_for_each_entry_safe(gluebi
, g
, &gluebi_devices
, list
) {
525 struct mtd_info
*mtd
= &gluebi
->mtd
;
527 err
= mtd_device_unregister(mtd
);
529 err_msg("error %d while removing gluebi MTD device %d, "
530 "UBI device %d, volume %d - ignoring", err
,
531 mtd
->index
, gluebi
->ubi_num
, gluebi
->vol_id
);
535 ubi_unregister_volume_notifier(&gluebi_notifier
);
538 module_init(ubi_gluebi_init
);
539 module_exit(ubi_gluebi_exit
);
540 MODULE_DESCRIPTION("MTD emulation layer over UBI volumes");
541 MODULE_AUTHOR("Artem Bityutskiy, Joern Engel");
542 MODULE_LICENSE("GPL");