From ad8226d6312eebaa8c6ca6f5e65dea47bcff97a9 Mon Sep 17 00:00:00 2001
From: Yaro Kasear <git@kasear.net>
Date: Mon, 28 Apr 2025 10:31:03 -0500
Subject: [PATCH] Add timing logs to main function for performance tracking
 during pcap analysis

---
 enrich.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/enrich.py b/enrich.py
index 63edf3c..4161424 100755
--- a/enrich.py
+++ b/enrich.py
@@ -19,6 +19,7 @@ from enrichment.csv_handler import (
     write_ssid_sidecar
 )
 from enrichment.merg_ssid_summaries import merge_ssid_summaries
+import time
 
 def parse_args():
     parser = argparse.ArgumentParser()
@@ -112,6 +113,7 @@ def analyze_pcap(pcapng_path, start_ts, end_ts, ap_bssid, ap_channel):
     )
 
 def main():
+    total_start_time = time.perf_counter()
     args = parse_args()
     cap = pyshark.FileCapture(
         args.pcapng,
@@ -155,6 +157,8 @@ def main():
             enriched_rows.append(row)
             continue
 
+        start_time = time.perf_counter()
+
         result = analyze_pcap(args.pcapng, tstart, tend, ap_bssid, ap_channel)
         (
             clients_ap, clients_chan, aps_chan,
@@ -163,6 +167,10 @@ def main():
             average_signal, max_ssid_signal, num_channels_ssid,
             ssid_summary, packet_count
         ) = result
+
+        elapsed_time = time.perf_counter() - start_time
+        print(f"[+] Analyzed {ap_bssid} in {elapsed_time:.2f} seconds")
+
         row.update({
             'ClientsOnAP': clients_ap,
             'ClientsOnChannel': clients_chan,
@@ -190,5 +198,8 @@ def main():
 
     print(f"[+] Enrichment complete: {args.output}")
 
+    total_elapsed_time = time.perf_counter() - total_start_time
+    print(f"[+] Total time taken: {total_elapsed_time:.2f} seconds")
+
 if __name__ == "__main__":
     main()