Fix: Disable IBT around indirect function calls

[lttng-modules.git] / src / wrapper / page_alloc.c

diff --git a/src/wrapper/page_alloc.c b/src/wrapper/page_alloc.c
index 81835cb2a04191fbd1aef3f45eae20cbc0a12cd6..8a05c62ee552cf31bebfc0b410f6a138397ac2ad 100644 (file)
--- a/src/wrapper/page_alloc.c
+++ b/src/wrapper/page_alloc.c
@@ -37,7 +37,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;
        }