X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=lttng-statedump-impl.c;h=657031ec9e535990683d37c24bbb64ae6e75ced0;hb=1c6722fbd85d139bffcd6fbeaffba177bdf251e0;hp=a40ef760902db32ac657072fb7cb881ef40f04df;hpb=6388029c11796f2416f772c1f9aca2ec925774e2;p=lttng-modules.git

diff --git a/lttng-statedump-impl.c b/lttng-statedump-impl.c
index a40ef760..657031ec 100644
--- a/lttng-statedump-impl.c
+++ b/lttng-statedump-impl.c
@@ -34,10 +34,10 @@
 #include <linux/device.h>
 #include <linux/fdtable.h>
 #include <linux/irq.h>
+#include <linux/genhd.h>
 
 #include <lttng-events.h>
 #include <lttng-tracer.h>
-#include <wrapper/genhd.h>
 
 /* Define the tracepoints, but do not build the probes */
 #define CREATE_TRACE_POINTS
@@ -114,19 +114,11 @@ enum lttng_process_status {
 static
 int lttng_enumerate_block_devices(struct lttng_session *session)
 {
-	struct class *ptr_block_class;
-	struct device_type *ptr_disk_type;
 	struct class_dev_iter iter;
 	struct device *dev;
 
-	ptr_block_class = wrapper_get_block_class();
-	if (!ptr_block_class)
-		return -ENOSYS;
-	ptr_disk_type = wrapper_get_disk_type();
-	if (!ptr_disk_type) {
-		return -ENOSYS;
-	}
-	class_dev_iter_init(&iter, ptr_block_class, NULL, ptr_disk_type);
+	class_dev_iter_init(&iter, gendisk_block_class(), NULL,
+			    gendisk_device_type());
 	while ((dev = class_dev_iter_next(&iter))) {
 		struct disk_part_iter piter;
 		struct gendisk *disk = dev_to_disk(dev);
@@ -145,7 +137,7 @@ int lttng_enumerate_block_devices(struct lttng_session *session)
 			char name_buf[BDEVNAME_SIZE];
 			char *p;
 
-			p = wrapper_disk_name(disk, part->partno, name_buf);
+			p = gendisk_name(disk, part->partno, name_buf);
 			if (!p) {
 				disk_part_iter_exit(&piter);
 				class_dev_iter_exit(&iter);