#!/bin/sh
# Build and install the patched bwn driver (if_bwn.ko), the patched PCIe
# bridge (bhndb_pci.ko) and the Broadcom v4 firmware (bwn_v4_n.ko) on a
# clean FreeBSD 15.1 system.
#
# Usage: sudo ./build.sh [SRC_ROOT]
# Default SRC_ROOT=/usr/src
#
# What it does:
#   1. Builds if_bwn.ko   (sys/modules/bwn)        -> /boot/modules/if_bwn.ko
#   2. Builds bhndb_pci.ko(sys/modules/bhnd/bhndb_pci) -> /boot/kernel/bhndb_pci.ko
#   3. Builds bwn_v4_n.ko (src/firmware)          -> /boot/modules/bwn_v4_n.ko
#   4. Adds "bwn_v4_n_load=YES" to /boot/loader.conf (Fix 5) if missing.
#
# All changes take effect on the NEXT reboot (modules are only loaded at
# boot). The running kernel is NOT touched, so an active SSH session (e.g.
# over the fallback rtwn0/wlan0 interface) is never disrupted by this script.
#
# After reboot, bwn0 attaches on the BCM43224 and uses legacy INTx (the
# NO_MSI quirk), so TX interrupts are delivered and the "device timeout" /
# detach-panic / heap-corruption issues are resolved.
set -e
SRC="${1:-/usr/src}"
HERE="$(cd "$(dirname "$0")" && pwd)"

if [ ! -d "$SRC/sys" ]; then
	echo "ERROR: $SRC/sys not found. Point SRC_ROOT at a FreeBSD 15.1 src tree." >&2
	exit 1
fi

echo "==> [1/3] Building if_bwn.ko"
make -C "$SRC/sys/modules/bwn" SRCTOP="$SRC" SYSDIR="$SRC/sys" clean
make -C "$SRC/sys/modules/bwn" SRCTOP="$SRC" SYSDIR="$SRC/sys"
install -m 444 -o root -g wheel "$SRC/sys/modules/bwn/if_bwn.ko" /boot/modules/if_bwn.ko
sync
echo "    -> /boot/modules/if_bwn.ko"

echo "==> [2/3] Building bhndb_pci.ko (NO_MSI quirk)"
make -C "$SRC/sys/modules/bhnd/bhndb_pci" SRCTOP="$SRC" SYSDIR="$SRC/sys" clean
make -C "$SRC/sys/modules/bhnd/bhndb_pci" SRCTOP="$SRC" SYSDIR="$SRC/sys"
install -m 444 -o root -g wheel "$SRC/sys/modules/bhnd/bhndb_pci/bhndb_pci.ko" /boot/kernel/bhndb_pci.ko
sync
echo "    -> /boot/kernel/bhndb_pci.ko"

echo "==> [3/3] Building + installing bwn_v4_n.ko (firmware)"
make -C "$HERE/src/firmware" SRCTOP="$SRC" SYSDIR="$SRC/sys" clean
make -C "$HERE/src/firmware" SRCTOP="$SRC" SYSDIR="$SRC/sys"
install -m 444 -o root -g wheel "$HERE/src/firmware/bwn_v4_n.ko" /boot/modules/bwn_v4_n.ko
sync
echo "    -> /boot/modules/bwn_v4_n.ko"

echo "==> Adding firmware load to /boot/loader.conf (Fix 5)"
if ! grep -q 'bwn_v4_n_load="YES"' /boot/loader.conf 2>/dev/null; then
	cp /boot/loader.conf "$HERE/rc.conf.backups/loader.conf.before_build_$(date +%Y%m%d%H%M%S)"
	printf '\n# bwn driver: load Broadcom v4 firmware at loader time (Fix 5)\nbwn_v4_n_load="YES"\n' >> /boot/loader.conf
	echo "    added bwn_v4_n_load=YES"
else
	echo "    already present, skipped"
fi

echo
echo "==> All modules built and installed. REBOOT to activate."
echo "    After reboot, create the interface with:"
echo "        ifconfig wlan create wlandev bwn0 [wap]"