Added kernel paths for other distros

1 files changed, 12 insertions, 1 deletions

diff --git a/Makefile b/Makefile
index e78339a..da1d1b2 100644
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,18 @@
 TARGET	        := $(shell uname -r)
-KERNEL_BUILD	:= /usr/src/linux-headers-$(TARGET)
 DKMS_ROOT_PATH  := /usr/src/zenpower-0.1.0
 
+ifneq ("","$(wildcard /usr/src/linux-headers-$(TARGET)/*)")
+# Ubuntu
+KERNEL_BUILD	:= /usr/src/linux-headers-$(TARGET)
+else
+ifneq ("","$(wildcard /usr/src/kernels/$(TARGET)/*)")
+# Fedora
+KERNEL_BUILD	:= /usr/src/kernels/$(TARGET)
+else
+KERNEL_BUILD	:= $(KERNEL_MODULES)/build
+endif
+endif
+
 obj-m	:= $(patsubst %,%.o,zenpower)
 obj-ko	:= $(patsubst %,%.ko,zenpower)