Skip to content

Commit d0732ee

Browse files
Add enabled checks in init, start, and stop. (#2292)
1 parent 54a39fc commit d0732ee

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

modules/auxiliary/AzSniffer.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,10 @@
2828
class AzSniffer(Auxiliary):
2929
def __init__(self):
3030
super().__init__()
31+
self.azsniffer_cfg = Config("auxiliary").get("AzSniffer")
32+
self.enabled = self.azsniffer_cfg.enabled
3133

32-
if not HAVE_AZURE:
34+
if not HAVE_AZURE or not self.enabled:
3335
return
3436

3537
self.cfg = Config("az")
@@ -58,6 +60,8 @@ def _get_credentials(self):
5860
return ClientSecretCredential(tenant_id=self.tenant_id, client_id=self.client_id, client_secret=self.client_secret)
5961

6062
def start(self):
63+
if not self.enabled:
64+
return
6165
self.capture_name = f"PacketCapture_{self.task.id}"
6266
custom_filters = []
6367
self.create_packet_capture(custom_filters)
@@ -96,6 +100,9 @@ def create_packet_capture(self, custom_filters):
96100
raise
97101

98102
def stop(self):
103+
if not self.enabled:
104+
return
105+
99106
if not self.capture_name:
100107
log.error("No packet capture to stop")
101108
return

0 commit comments

Comments
 (0)