From 5b4f313e27e6ef228a64dfe00de0bbbdbaf21a8a Mon Sep 17 00:00:00 2001
From: Yihui Xiong <yihui.xiong@hotmail.com>
Date: Mon, 7 Aug 2017 09:36:02 +0800
Subject: [PATCH] build for current kernel

---
 install.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/install.sh b/install.sh
index d6a7d5a..d949b01 100755
--- a/install.sh
+++ b/install.sh
@@ -18,7 +18,7 @@ apt-get -y install  dkms
 
 # locate currently installed kernels (may be different to running kernel if
 # it's just been updated)
-kernels=$(ls /lib/modules | sed "s/^/-k /")
+kernel="$(uname -r)"
 
 function install_module {
   src=$1
@@ -35,7 +35,7 @@ function install_module {
   mkdir -p /usr/src/$mod-$ver
   cp -a $src/* /usr/src/$mod-$ver/
   dkms add -m $mod -v $ver
-  dkms build $kernels -m $mod -v $ver && dkms install $kernels -m $mod -v $ver
+  dkms build -k $kernel -m $mod -v $ver && dkms install -k $kernel -m $mod -v $ver
 
   mkdir -p /var/lib/dkms/$mod/$ver/$marker
 }