If keep_bio_blkcg is enabled, assign the io_context and the blkcg of
current task to bio before processing the bio.
Signed-off-by: Hou Tao <houtao1@xxxxxxxxxx>
---
drivers/md/dm-thin.c | 5 +++++
drivers/md/dm-thin.h | 17 +++++++++++++++++
2 files changed, 22 insertions(+)
create mode 100644 drivers/md/dm-thin.h
diff --git a/drivers/md/dm-thin.c b/drivers/md/dm-thin.c
index 140cdae..0efbdbe 100644
--- a/drivers/md/dm-thin.c
+++ b/drivers/md/dm-thin.c
@@ -22,6 +22,8 @@
#include <linux/sort.h>
#include <linux/rbtree.h>
+#include "dm-thin.h"
+
#define DM_MSG_PREFIX "thin"
/*
@@ -2629,6 +2631,9 @@ static int thin_bio_map(struct dm_target *ti, struct bio *bio)
struct dm_bio_prison_cell *virt_cell, *data_cell;
struct dm_cell_key key;
+ if (keep_pool_bio_blkcg(tc->pool))
+ thin_keep_bio_blkcg(bio);
+
thin_hook_bio(tc, bio);
if (tc->requeue_mode) {
diff --git a/drivers/md/dm-thin.h b/drivers/md/dm-thin.h
new file mode 100644
index 0000000..09e920a
--- /dev/null
+++ b/drivers/md/dm-thin.h
@@ -0,0 +1,17 @@
+#ifndef DM_THIN_H
+#define DM_THIN_H
+
+#include <linux/bio.h>
+
+#ifdef CONFIG_BLK_CGROUP
+static inline void thin_keep_bio_blkcg(struct bio *bio)
+{
+ if (!bio->bi_css)
+ bio_associate_current(bio);
+}
+#else
+static inline void thin_keep_bio_blkcg(struct bio *bio) {}
+#endif /* CONFIG_BLK_CGROUP */
+
+#endif
+
--
2.5.0
--
To unsubscribe from this list: send the line "unsubscribe linux-block" in
the body of a message to majordomo@xxxxxxxxxxxxxxx
More majordomo info at http://vger.kernel.org/majordomo-info.html
