kernmod/build.sh

#!/bin/bash

# Initial variables
PACKAGE="helloworld"
VERSION="0.1"
REVISION="1"
SRCDIR="/vagrant/src"
SCRATCH="/vagrant/scratch"
OUTDIR="$SCRATCH"
TEMPDIR="$(mktemp -d)"
INSTALL=0
BUILDDIR="$TEMPDIR/${PACKAGE}_$VERSION-$REVISION"

# Test for distribution of GNU/Linux
# 0 = unknown
# 1 = debian
# 2 = rocky/centos/rhel
function check_distro {
  if [ -f /etc/debian_version ]; then echo 1
  elif [ -f /etc/redhat-release ]; then echo 2
  else echo 0
  fi
}

# Install Linux headers for current debian kernel
function debian_headers {
  export DEBIAN_FRONTEND=noninteractive
  apt-get update
  apt-get install -y linux-headers-$(uname -r)
}

# Build debian package
function build_deb {
  [ $INSTALL -eq 1 ] && debian_headers
  mkdir -p "$BUILDDIR/usr/src/${PACKAGE}-$VERSION"
  mkdir -p "$BUILDDIR/etc"
  mkdir -p "$BUILDDIR/DEBIAN"
  cp -r $SRCDIR/usr/src/* "$BUILDDIR/usr/src/${PACKAGE}-$VERSION"
  cp -r $SRCDIR/etc/* "$BUILDDIR/etc"
  cp -r $SRCDIR/DEBIAN/* "$BUILDDIR/DEBIAN"
  cd "$TEMPDIR"
  dpkg-deb --build "${PACKAGE}_$VERSION-$REVISION"
}

# Install Linux headers for current rhel kernel
function rhel_headers {
  KERNEL_VERSION="$(uname -r | rev | cut -d '.' -f 2- | rev)"
  yum install -y kernel-headers-"$KERNEL_VERSION" kernel-devel
}

# Build redhat package
function build_rpm {
  EL_VER="$(uname -r | awk 'match($0,/el[0-9]/) {print substr($0, RSTART, RLENGTH)}')"
  [ $INSTALL -eq 1 ] && rhel_headers
  yum install -y rpm-build rpmdevtools rpmlint
  rpmdev-setuptree
  mkdir -p "$BUILDDIR/${PACKAGE}-${VERSION}"
  cp -r $SRCDIR/* "$BUILDDIR/${PACKAGE}-${VERSION}"
  rm -rf ~/rpmbuild/SOURCES/$PACKAGE-$VERSION.tar.gz
  (cd $BUILDDIR &&
  tar -czvf ~/rpmbuild/SOURCES/$PACKAGE-$VERSION.tar.gz $PACKAGE-$VERSION)
  cp $SRCDIR/$PACKAGE.spec ~/rpmbuild/SPECS/
  rpmlint ~/rpmbuild/SPECS/$PACKAGE.spec &&
  rpmbuild -bb ~/rpmbuild/SPECS/$PACKAGE.spec
}

# Display details on module
function info_mod {
  modinfo "$PACKAGE"
  cat /proc/modules | grep "$PACKAGE"
  rmmod "$PACKAGE"
  modprobe "$PACKAGE"
  cat /var/log/messages | grep "$PACKAGE"
}

# Build and install (optional)
function build_install {
  DISTRO="$(check_distro)"
  if [ $DISTRO -eq 0 ]; then
    echo "ERROR: GNU/Linux distribution not detected"
    exit -1
  elif [ $DISTRO -eq 1 ]; then
    build_deb
    [ $INSTALL -eq 1 ] && apt-get install -y "./${PACKAGE}_$VERSION-$REVISION.deb"
  elif [ $DISTRO -eq 2 ]; then
    build_rpm
    if [ $INSTALL -eq 1 ]; then
      yum install -y epel-release
      yum install -y dkms
      rpm -i "~/rpmbuild/RPMS/noarch/$PACKAGE-$VERSION-$REVISION.$EL_VER.noarch.rpm"
    fi
  fi
  info_mod
}

# Build and install helloworld module or module(s) in $SCRATCH
function main_routine {
  if [ ! -z "$(ls -Al $SCRATCH | grep -e ^d)" ]; then
    cd "$SCRATCH"
    for d in */ ; do
      if [ -f "$(basename $d)/override.sh" ]; then
        SRCDIR="$(pwd)/$(basename $d)"
        . "$(basename $d)/override.sh"
        build_install
      fi
    done
  else
    INSTALL=1
    build_install
  fi
}

# Program starts here unless KERNMOD_MAIN is not 0
if [ "${KERNMOD_MAIN:-0}" -eq 0 ]; then
  set -x
  main_routine
fi