X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=src%2Fwrapper%2Fpage_alloc.c;h=5e19d7b5c786accd74e0a9697ceb3af588dfc3ef;hb=92e2c5feecb39cdd4796da89f2b684e395403040;hp=a03838fad4f2b2d2f3ced26f80068305a0ba786f;hpb=dfdb34fd8576fba33510491aef5cd5f6f67b56b8;p=lttng-modules.git

diff --git a/src/wrapper/page_alloc.c b/src/wrapper/page_alloc.c
index a03838fa..5e19d7b5 100644
--- a/src/wrapper/page_alloc.c
+++ b/src/wrapper/page_alloc.c
@@ -30,7 +30,13 @@ unsigned long wrapper_get_pfnblock_flags_mask(struct page *page,
 {
 	WARN_ON_ONCE(!get_pfnblock_flags_mask_sym);
 	if (get_pfnblock_flags_mask_sym) {
-		return get_pfnblock_flags_mask_sym(page, pfn, end_bitidx, mask);
+		struct irq_ibt_state irq_ibt_state;
+		unsigned long ret;
+
+		irq_ibt_state = wrapper_irq_ibt_save();
+		ret = get_pfnblock_flags_mask_sym(page, pfn, end_bitidx, mask);
+		wrapper_irq_ibt_restore(irq_ibt_state);
+		return ret;
 	} else {
 		return -ENOSYS;
 	}