Fix: build x86 kvm probes on kernels >= 4.1-rc1

[lttng-modules.git] / probes / lttng-probe-kvm-x86.c

diff --git a/probes/lttng-probe-kvm-x86.c b/probes/lttng-probe-kvm-x86.c
index a15a4e958250119675fe5eb426fd24071ab916b6..9b2054d806e088487caccb7ec6449a4f51c10675 100644 (file)
--- a/probes/lttng-probe-kvm-x86.c
+++ b/probes/lttng-probe-kvm-x86.c
@@ -23,6 +23,13 @@
 #include <linux/module.h>
 #include <linux/kvm_host.h>
 #include "../lttng-tracer.h"
+#include "../lttng-kernel-version.h"
+
+#if (LINUX_VERSION_CODE >= KERNEL_VERSION(4,1,0))
+#include <kvm/iodev.h>
+#else /* #if (LINUX_VERSION_CODE >= KERNEL_VERSION(4,1,0)) */
+#include <../../virt/kvm/iodev.h>
+#endif /* #else #if (LINUX_VERSION_CODE >= KERNEL_VERSION(4,1,0)) */
 
 /*
  * Create the tracepoint static inlines from the kernel to validate that our