2 files changed, 11 insertions, 6 deletions

diff --git a/drivers/mtd/devices/block2mtd.c b/drivers/mtd/devices/block2mtd.c
index 519d942e7940..7b72a1b36115 100644
--- a/drivers/mtd/devices/block2mtd.c
+++ b/drivers/mtd/devices/block2mtd.c
@@ -241,6 +241,7 @@ static struct block2mtd_dev *add_device(char *devname, int erase_size)
 {
         struct block_device *bdev;
         struct block2mtd_dev *dev;
+        char *name;
 
         if (!devname)
                 return NULL;
@@ -279,12 +280,13 @@ static struct block2mtd_dev *add_device(char *devname, int erase_size)
 
         /* Setup the MTD structure */
         /* make the name contain the block device in */
-        dev->mtd.name = kmalloc(sizeof("block2mtd: ") + strlen(devname),
+        name = kmalloc(sizeof("block2mtd: ") + strlen(devname) + 1,
                         GFP_KERNEL);
-        if (!dev->mtd.name)
+        if (!name)
                 goto devinit_err;
 
-        sprintf(dev->mtd.name, "block2mtd: %s", devname);
+        sprintf(name, "block2mtd: %s", devname);
+        dev->mtd.name = name;
 
         dev->mtd.size = dev->blkdev->bd_inode->i_size & PAGE_MASK;
         dev->mtd.erasesize = erase_size;
diff --git a/drivers/mtd/mtdchar.c b/drivers/mtd/mtdchar.c
index 129d429cd2da..aef9f4b687c9 100644
--- a/drivers/mtd/mtdchar.c
+++ b/drivers/mtd/mtdchar.c
@@ -28,10 +28,13 @@ static void mtd_notify_add(struct mtd_info* mtd)
         if (!mtd)
                 return;
 
-        device_create(mtd_class, NULL, MKDEV(MTD_CHAR_MAJOR, mtd->index*2), "mtd%d", mtd->index);
+        device_create_drvdata(mtd_class, NULL,
+                              MKDEV(MTD_CHAR_MAJOR, mtd->index*2),
+                              NULL, "mtd%d", mtd->index);
 
-        device_create(mtd_class, NULL,
-                      MKDEV(MTD_CHAR_MAJOR, mtd->index*2+1), "mtd%dro", mtd->index);
+        device_create_drvdata(mtd_class, NULL,
+                              MKDEV(MTD_CHAR_MAJOR, mtd->index*2+1),
+                              NULL, "mtd%dro", mtd->index);
 }
 
 static void mtd_notify_remove(struct mtd_info* mtd)