yaro/wifi_test
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for --no-enrich flag in runtest.sh to skip data enrichment

Browse source
This commit is contained in:
Yaro Kasear 2025-04-30 13:50:59 -05:00
parent 027890896e
commit 2a227d7222
2 changed files with 107 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

30
runtest.sh
View file

@ -4,6 +4,20 @@ set -uo pipefail
trap 'echo "[✖] Execution halted at line $LINENO. Please consult your nearest bash therapist." >&2' ERR
trap 'echo "[✌️] Script exited cleanly. Have a burrito." >&2' EXIT
NO_ENRICH=0
for arg in "$@"; do
case $arg in
--no-enrich)
NO_ENRICH=1
shift
;;
*)
echo "Unknown argument: $arg"
exit 1
;;
esac
done
IFS=$'\n\t'
# === Start logging with 'script' safely ===
@ -188,13 +202,17 @@ done
log "Stopping kismet..."
sudo systemctl stop kismet
log "Enriching data..."
KISMET_LOG=$(find "$KISMET_LOG_DIR" -type f -name "*.pcapng" -printf "%T@ %p\n" | sort -n | tail -1 | cut -d' ' -f2-)
[ ! -f "$KISMET_LOG" ] && die "Packet capture not found."
if [ "$NO_ENRICH" -eq 0 ]; then
log "Enriching data..."
KISMET_LOG=$(find "$KISMET_LOG_DIR" -type f -name "*.pcapng" -printf "%T@ %p\n" | sort -n | tail -1 | cut -d' ' -f2-)
[ ! -f "$KISMET_LOG" ] && die "Packet capture not found."
SECONDS=0
python3 "$SCRIPT_DIRECTORY/enrich.py" --csv "$TEST_FILE" --pcapng "$KISMET_LOG" --output "$ENRICHED_FILE"
log "Enrichment took $SECONDS seconds"
SECONDS=0
python3 "$SCRIPT_DIRECTORY/enrich.py" --csv "$TEST_FILE" --pcapng "$KISMET_LOG" --output "$ENRICHED_FILE"
log "Enrichment took $SECONDS seconds"
else
log "Skipping enrichment as per --no-enrich flag."
fi
ATTACHMENTS=()
[ -f "$ENRICHED_FILE" ] && ATTACHMENTS+=("$ENRICHED_FILE")