From 4610a7110aa0326956a65e43e105092e06cfc631 Mon Sep 17 00:00:00 2001
From: Vasily Evseenko <svpcom@gmail.com>
Date: Wed, 1 Jan 2025 17:26:36 +0300
Subject: [PATCH] Add wifi cards autodetect

---
 .gitignore                    |  1 +
 scripts/default/wifibroadcast |  5 ++++-
 scripts/wfb-nics              | 11 +++++++++++
 setup.py                      |  3 ++-
 4 files changed, 18 insertions(+), 2 deletions(-)
 create mode 100755 scripts/wfb-nics

diff --git a/.gitignore b/.gitignore
index 5a94d88..b15c1ea 100644
--- a/.gitignore
+++ b/.gitignore
@@ -10,6 +10,7 @@ _trial_temp/
 wfb_rx
 wfb_tx
 wfb_tx_cmd
+wfb_tun
 gs.key
 drone.key
 wfb_keygen
diff --git a/scripts/default/wifibroadcast b/scripts/default/wifibroadcast
index 2cfe5c4..4a06087 100644
--- a/scripts/default/wifibroadcast
+++ b/scripts/default/wifibroadcast
@@ -1,6 +1,9 @@
 #WFB_NICS="wlx00c0caa578a9"
 #WFB_NICS="wlan1 wlan2"
+
 # For multi-link setup you can specify path to custom config
 # and/or list of cards in /etc/default/wifibroadcast.<profile_name>
 # WIFIBROADCAST_CFG=/etc/wifibroadcast_linkXXX.cfg
-WFB_NICS="wlan0"
+
+# Autodetect local cards supported by wfb-ng (8812au and 8812eu)
+WFB_NICS="$(wfb-nics)"
diff --git a/scripts/wfb-nics b/scripts/wfb-nics
new file mode 100755
index 0000000..2d08faa
--- /dev/null
+++ b/scripts/wfb-nics
@@ -0,0 +1,11 @@
+#!/bin/bash
+
+# Autodetect 8812au and 8812eu cards
+
+for i in $(find /sys/class/net/ -maxdepth 1 -type l | sort)
+do
+   if udevadm info $i | grep -qE 'ID_NET_DRIVER=(rtl88xxau_wfb|rtl88x2eu)'
+   then
+         echo $(basename $i)
+   fi
+done
diff --git a/setup.py b/setup.py
index 97c5c3b..49ca468 100644
--- a/setup.py
+++ b/setup.py
@@ -48,7 +48,8 @@ setup(
                                       'wfb-server=wfb_ng.server:main',
                                       'wfb-log-parser=wfb_ng.log_parser:main']},
     package_data={'wfb_ng.conf': ['master.cfg', 'site.cfg']},
-    data_files = [('/usr/bin', ['wfb_tx', 'wfb_rx', 'wfb_keygen', 'wfb_tx_cmd', 'scripts/wfb-cli-x11']),
+    data_files = [('/usr/bin', ['wfb_tx', 'wfb_rx', 'wfb_keygen', 'wfb_tx_cmd',
+                                'scripts/wfb-cli-x11', 'scripts/wfb-nics']),
                   ('/lib/systemd/system', ['scripts/wifibroadcast.service',
                                            'scripts/wifibroadcast@.service',
                                            'scripts/wfb-cluster.service',