liminix/pkgs/mac80211/default.nix

# make out-of-tree modules given a backported kernel source tree

{
  extraConfig ? { },
  drivers ? [ ],
  kernel-backport,
  stdenv,
  writeText,
  klibBuild ? null,
  buildPackages,
  fetchFromGitHub,

  liminix,
  lib,
}:
let
  arch = stdenv.hostPlatform.linuxArch;
  openwrtSrc = fetchFromGitHub {
    name = "openwrt-source";
    repo = "openwrt";
    owner = "openwrt";
    rev = "a5265497a4f6da158e95d6a450cb2cb6dc085cab";
    hash = "sha256-YYi4gkpLjbOK7bM2MGQjAyEBuXJ9JNXoz/JEmYf8xE8=";
  };
  inherit (liminix.services) oneshot;
  inherit (lib.lists) foldl;
  configs = {
    ath9k.kconfig = {
      WLAN_VENDOR_ATH = "y";
      ATH_COMMON = "m";
      ATH9K = "m";
      ATH9K_AHB = "y";
      BACKPORTED_ATH9K_AHB = "y";
    };
    ath9k_pci = {
      module = "ath9k";
      kconfig = {
        WLAN_VENDOR_ATH = "y";
        ATH_COMMON = "m";
        ATH9K = "m";
        ATH9K_PCI = "y";
      };
    };
    ath10k_pci.kconfig = {
      WLAN_VENDOR_ATH = "y";
      ATH_COMMON = "m";
      ATH10K = "m";
      ATH10K_PCI = "y";
      ATH10K_DEBUG = "y";
    };
    rt2800soc.kconfig = {
      WLAN_VENDOR_RALINK = "y";
      RT2800SOC = "m";
      RT2X00 = "m";
    };
    mt7603e.kconfig = {
      # XXX find a better name for this
      WLAN_VENDOR_RALINK = "y";
      WLAN_VENDOR_MEDIATEK = "y";
      MT7603E = "y";
    };

    mt7915e.kconfig = {
      MT7915E = "m";
    };
    mt7615e.kconfig = {
      MT7615E = "m";
      MT7622_WMAC = "y";
    };
    mac80211_hwsim.kconfig = {
      MAC80211_HWSIM = "y";
    };
  };
  kconfig =
    (foldl (config: d: (config // configs.${d}.kconfig)) {
      WLAN = "y";
      CFG80211 = "m";
      MAC80211 = "m";

      # (nixwrt comment) I am reluctant to have to enable this but
      # can't transmit on 5GHz bands without it (they are all marked
      # NO-IR)
      CFG80211_CERTIFICATION_ONUS = "y";
      # (nixwrt comment) can't get signed regdb to work rn, it just
      # gives me "loaded regulatory.db is malformed or signature is
      # missing/invalid"
      CFG80211_REQUIRE_SIGNED_REGDB = "n"; # depends on ONUS

      CFG80211_CRDA_SUPPORT = "n";

      MAC80211_MESH = "y";
    } drivers)
    // extraConfig;

  writeConfig =
    name: config:
    writeText name (
      builtins.concatStringsSep "\n" (
        lib.mapAttrsToList (
          name: value: (if value == "n" then "# CPTCFG_${name} is not set" else "CPTCFG_${name}=${value}")
        ) config
      )
    );
  kconfigFile = writeConfig "backports_kconfig" kconfig;
  src = kernel-backport;
  CROSS_COMPILE = stdenv.cc.bintools.targetPrefix;
  CC = "${buildPackages.stdenv.cc}/bin/gcc";
  module = stdenv.mkDerivation {
    name = "mac80211";
    inherit src;

    hardeningDisable = [ "all" ];
    nativeBuildInputs =
      [ buildPackages.stdenv.cc ]
      ++ (with buildPackages.pkgs; [
        bc
        bison
        flex
        pkg-config
        openssl
        which
        kmod
        cpio
      ]);
    inherit CC CROSS_COMPILE;
    ARCH = arch;
    dontStrip = true;
    dontPatchELF = true;
    phases = [
      "unpackPhase"
      "patchFromOpenwrt"
      "configurePhase"
      "checkConfigurationPhase"
      "buildPhase"
      "installPhase"
    ];

    patchFromOpenwrt = ''
      mac80211=${openwrtSrc}/package/kernel/mac80211
      echo $mac80211
      for i in $(ls $mac80211/patches/build/ | grep -v 015-ipw200); do
        echo $i; (patch -p1 -N <$mac80211/patches/build/$i)
      done
      for i in $(ls $mac80211/patches/rt2x00/ | grep -v 009-rt2x00-don-t | grep -v 013-rt2x00-set-correct | grep -v 014 | grep -v 015 | grep -v 016); do
        echo $i; (patch -p1 -N <$mac80211/patches/rt2x00/$i)
      done
      for i in $mac80211/patches/ath/*; do echo $i; (patch -p1 -N <$i) ;done
      for i in $mac80211/patches/ath9k/*; do echo $i; (patch -p1 -N <$i) ;done
    '';

    configurePhase = ''
      cp ${kconfigFile} .config
      cp ${kconfigFile} .config.orig
      chmod +w .config .config.orig
      make V=1 CC=${CC} SHELL=`type -p bash` LEX=flex KLIB_BUILD=${klibBuild} olddefconfig
    '';

    checkConfigurationPhase = ''
      echo Checking required config items:
      if comm -2 -3 <(awk -F= '/CPTCFG/ {print $1}' ${kconfigFile} |sort) <(awk -F= '/CPTCFG/ {print $1}' .config|sort) |grep '.'    ; then
        echo -e "^^^ Some configuration lost :-(\nPerhaps you have mutually incompatible settings, or have disabled options on which these depend.\n"
        exit 0
      fi
      echo "OK"
    '';

    KBUILD_BUILD_HOST = "liminix.builder";

    buildPhase = ''
      patchShebangs scripts/
      echo  ${klibBuild}
      make V=1 SHELL=`type -p bash` KLIB_BUILD=${klibBuild} modules
      find . -name \*.ko | xargs ${CROSS_COMPILE}strip --strip-debug
    '';

    installPhase = ''
      mkdir -p $out/lib/modules/0.0
      find . -name \*.ko | cpio --make-directories -p $out/lib/modules/0.0
      depmod -b $out -v 0.0
      touch $out/load.sh
      for i in ${
        lib.concatStringsSep " " (
          map (
            d:
            let
              c = {
                module = d;
              } // configs.${d};
            in
            c.module
          ) drivers
        )
      }; do
        modprobe -S 0.0 -d $out --show-depends $i >> $out/load.sh
      done
      tac < $out/load.sh | sed 's/^insmod/rmmod/g' > $out/unload.sh
    '';
  };
in
oneshot {
  name = "wlan.module";
  up = "sh ${module}/load.sh";
  down = "sh ${module}/unload.sh";
}