*/
 extern struct block_device *bdget_disk(struct gendisk *disk, int partno)
 {
-       dev_t devt = MKDEV(0, 0);
+       struct hd_struct *part;
+       struct block_device *bdev = NULL;
 
-       if (partno == 0)
-               devt = disk_devt(disk);
-       else {
-               struct hd_struct *part;
+       part = disk_get_part(disk, partno);
+       if (part && (part->nr_sects || partno == 0))
+               bdev = bdget(part_devt(part));
+       disk_put_part(part);
 
-               part = disk_get_part(disk, partno);
-               if (part && part->nr_sects)
-                       devt = part_devt(part);
-               disk_put_part(part);
-       }
-
-       if (likely(devt != MKDEV(0, 0)))
-               return bdget(devt);
-       return NULL;
+       return bdev;
 }
 EXPORT_SYMBOL(bdget_disk);
 
        class_dev_iter_init(&iter, &block_class, NULL, &disk_type);
        while ((dev = class_dev_iter_next(&iter))) {
                struct gendisk *disk = dev_to_disk(dev);
+               struct hd_struct *part;
 
                if (strcmp(dev->bus_id, name))
                        continue;
-               if (partno < 0 || partno >= disk_max_parts(disk))
-                       continue;
-
-               if (partno == 0)
-                       devt = disk_devt(disk);
-               else {
-                       struct hd_struct *part;
-
-                       part = disk_get_part(disk, partno);
-                       if (!part || !part->nr_sects) {
-                               disk_put_part(part);
-                               continue;
-                       }
 
+               part = disk_get_part(disk, partno);
+               if (part && (part->nr_sects || partno == 0)) {
                        devt = part_devt(part);
                        disk_put_part(part);
+                       break;
                }
-               break;
+               disk_put_part(part);
        }
        class_dev_iter_exit(&iter);
        return devt;
 
 #ifdef CONFIG_BLOCK
 
 #define kobj_to_dev(k)         container_of((k), struct device, kobj)
-#define dev_to_disk(device)    container_of((device), struct gendisk, __dev)
+#define dev_to_disk(device)    container_of((device), struct gendisk, part0.__dev)
 #define dev_to_part(device)    container_of((device), struct hd_struct, __dev)
-#define disk_to_dev(disk)      (&((disk)->__dev))
+#define disk_to_dev(disk)      (&(disk)->part0.__dev)
 #define part_to_dev(part)      (&((part)->__dev))
 
 extern struct device_type part_type;
 
        int flags;
        struct device *driverfs_dev;  // FIXME: remove
-       struct device __dev;
        struct kobject *holder_dir;
        struct kobject *slave_dir;
 
 
 static inline struct gendisk *part_to_disk(struct hd_struct *part)
 {
-       if (likely(part))
-               return dev_to_disk(part_to_dev(part)->parent);
+       if (likely(part)) {
+               if (part->partno)
+                       return dev_to_disk(part_to_dev(part)->parent);
+               else
+                       return dev_to_disk(part_to_dev(part));
+       }
        return NULL;
 }