Raspberry Pi: Add uncompressed wireless FW to Ubuntu releases

Ubuntu uses compressed "*.zst" firmware of which the kernel has
issue.

Signed-off-by: Patrick Yavitz <pyavitz@gmail.com>

Author: pyavitz

config/sources/families/bcm2711.conf

@@ -180,12 +180,6 @@ function post_family_tweaks_bsp__add_hooks_to_move_kernel_initrd_and_dtb() {
 	run_host_command_logged chmod a+x "${destination}"/etc/initramfs/post-update.d/zzz-update-initramfs
 }
 
-function post_family_tweaks_bsp__rpi_firmware() {
-	display_alert "$BOARD" "Installing firmware" "info"
-	git clone https://github.com/pyavitz/firmware.git --depth=1 -q "${destination}"/lib/firmware/updates/brcm
-	rm -fdr "${destination}"/lib/firmware/updates/brcm/{.git,README.md}
-}
-
 function post_family_tweaks_bsp__add_x11_config() {
 	display_alert "rpi5b" "Adding X11 configuration" "info"
 	run_host_command_logged mkdir -p "${destination}"/etc/X11/xorg.conf.d/
@@ -260,6 +254,15 @@ function pre_install_distribution_specific__add_rpi_packages() {
 	fi
 }
 
+function pre_install_distribution_specific__ubuntu_raspi_firmware() {
+	# Ubuntu uses compressed "*.zst" firmware of which the kernel has issue.
+	if [[ "${DISTRIBUTION}" == "Ubuntu" ]]; then
+		display_alert "$DISTRIBUTION" "Adding uncompressed raspi-firmware" "info"
+		chroot_sdcard git clone --depth=1 -q https://github.com/pyavitz/firmware.git -b brcm lib/firmware/updates
+		rm -fdr "${SDCARD}"/lib/firmware/updates/.git
+	fi
+}
+
 function pre_install_distribution_specific__unblock_rfkill() {
 	# Create a systemd service to unblock rfkill
 	cat > "${SDCARD}/etc/systemd/system/unblock-rfkill.service" <<- EOT