X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=src%2Flttng-context-callstack-stackwalk-impl.h;h=5075e9850d6742f320a89e2aa412807c256fdfd0;hb=92e2c5feecb39cdd4796da89f2b684e395403040;hp=e73d1156ae7c0731e7918c194982e2446688f278;hpb=dfdb34fd8576fba33510491aef5cd5f6f67b56b8;p=lttng-modules.git

diff --git a/src/lttng-context-callstack-stackwalk-impl.h b/src/lttng-context-callstack-stackwalk-impl.h
index e73d1156..5075e985 100644
--- a/src/lttng-context-callstack-stackwalk-impl.h
+++ b/src/lttng-context-callstack-stackwalk-impl.h
@@ -152,6 +152,7 @@ size_t lttng_callstack_sequence_get_size(void *priv, struct lttng_kernel_probe_c
 	struct field_data *fdata = (struct field_data *) priv;
 	size_t orig_offset = offset;
 	int cpu = smp_processor_id();
+	struct irq_ibt_state irq_ibt_state;
 
 	/* do not write data if no space is available */
 	trace = stack_trace_context(fdata, cpu);
@@ -166,14 +167,18 @@ size_t lttng_callstack_sequence_get_size(void *priv, struct lttng_kernel_probe_c
 	switch (fdata->mode) {
 	case CALLSTACK_KERNEL:
 		/* do the real work and reserve space */
+		irq_ibt_state = wrapper_irq_ibt_save();
 		trace->nr_entries = save_func_kernel(trace->entries,
 						MAX_ENTRIES, 0);
+		wrapper_irq_ibt_restore(irq_ibt_state);
 		break;
 	case CALLSTACK_USER:
 		++per_cpu(callstack_user_nesting, cpu);
 		/* do the real work and reserve space */
+		irq_ibt_state = wrapper_irq_ibt_save();
 		trace->nr_entries = save_func_user(trace->entries,
 						MAX_ENTRIES);
+		wrapper_irq_ibt_restore(irq_ibt_state);
 		per_cpu(callstack_user_nesting, cpu)--;
 		break;
 	default: