cavisson-pythonagent 0.0.1__py3-none-any.whl

pythonagent/agent/probes/havoc/havoc_constants.py

@@ -0,0 +1,218 @@
+import json
+import logging
+from enum import IntEnum
+
+logger = logging.getLogger('pythonagent.agent')
+
+class HavocType(IntEnum):
+    INBOUND_SERVICE_DELAY = 1
+    INBOUND_SERVICE_FAILURE = 2
+    OUTBOUND_SERVICE_DELAY = 3
+    OUTBOUND_SERVICE_FAILURE = 4
+    METHOD_CALL_DELAY = 5
+    METHOD_CALL_FAILURE = 6
+    CUSTOM_MEMORY_LEAK = 7
+    CUSTOM_THREAD_LEAK = 9
+    APPLICATION_KILL = 10
+
+HAVOC_TYPE_NAMES = {
+    HavocType.INBOUND_SERVICE_DELAY: "InBoundService_Delay",
+    HavocType.INBOUND_SERVICE_FAILURE: "InBoundService_Failure",
+    HavocType.OUTBOUND_SERVICE_DELAY: "OutBoundService_Delay",
+    HavocType.OUTBOUND_SERVICE_FAILURE: "OutBoundService_Failure",
+    HavocType.METHOD_CALL_DELAY: "MethodCall_Delay",
+    HavocType.METHOD_CALL_FAILURE: "MethodCall_Failure",
+    HavocType.CUSTOM_MEMORY_LEAK: "Custom_Memory_Leak",
+    HavocType.CUSTOM_THREAD_LEAK: "Custom_Thread_Leak",
+    HavocType.APPLICATION_KILL: "Application_Kill"
+}
+
+HAVOC_TYPES = [
+    "InBoundService_Delay",
+    "InBoundService_Failure",
+    "OutBoundService_Delay",
+    "OutBoundService_Failure",
+    "MethodCall_Delay",
+    "MethodCall_Failure",
+    "Custom_Memory_Leak",
+    "Custom_Thread_Leak",
+    "Application_Kill"
+]
+
+# EXPIRE_DURATION_MS = 300000  # 5 minutes
+# EXPIRE_DURATION_MS = 3000  # 3 seconds
+EXPIRE_DURATION_MS = 1000  # 1 seconds
+
+
+SIZE_OF_OBJECTS_BYTES = 524288000  # 500 MB
+
+# Inbound Service Delay
+IB_SERV_DELAY_DICT = {
+   "enable": 0,
+   "havocType": 1,
+   "bTName": "",
+   "uRL": "/prod/path/to/resource",
+   "methodFQM": "",
+   "delayInSec": 10,
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+IB_SERV_DELAY_JSON = json.dumps(IB_SERV_DELAY_DICT)
+
+# Inbound Service Failure
+IB_SERV_FAIL_DICT = {
+   "enable": 0,
+   "havocType": 2,
+   "bTName": "",
+   "uRL": "/prod/path/to/resource",
+   "methodFQM": "",
+   "delayInSec": 10,
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+IB_SERV_FAIL_JSON = json.dumps(IB_SERV_FAIL_DICT)
+
+# Outbound Service Delay
+OB_SERV_DELAY_DICT = {
+   "enable": 0,
+   "havocType": 3,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "",
+   "delayInSec": 10,
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "sample_hostname",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+OB_SERV_DELAY_JSON = json.dumps(OB_SERV_DELAY_DICT)
+
+# Outbound Service Failure
+OB_SERV_FAIL_DICT = {
+   "enable": 0,
+   "havocType": 4,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "",
+   "delayInSec": 0,
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "sample_hostname",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+OB_SERV_FAIL_JSON = json.dumps(OB_SERV_FAIL_DICT)
+
+# Method Call Delay
+METH_CALL_DELAY_DICT = {
+   "enable": 0,
+   "havocType": 5,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "/var/task.lambda_handler",
+   "delayInSec": 10,
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+METH_CALL_DELAY_JSON = json.dumps(METH_CALL_DELAY_DICT)
+
+# Method Invocation Failure
+METH_INV_FAIL_DICT = {
+   "enable": 0,
+   "havocType": 6,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "/var/task.lambda_handler",
+   "delayInSec": "",
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "leaksINMB": "",
+   "memoryLeaksInSec": ""
+}
+METH_INV_FAIL_JSON = json.dumps(METH_INV_FAIL_DICT)
+
+# Custom Memory Leak
+CUSTOM_MEMORY_LEAK_DICT = {
+   "enable": 0,
+   "havocType": 7,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "/var/task.lambda_handler",
+   "delayInSec": "",
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "leaksINMB": 100,
+   "objectSize": 30,
+   "rampup": 0
+ 
+}
+CUSTOM_MEMORY_LEAK_JSON = json.dumps(CUSTOM_MEMORY_LEAK_DICT)
+
+# Custom Thread Leak
+CUSTOM_THREAD_LEAK_DICT = {
+   "enable": 0,
+   "havocType": 9,
+   "bTName": "",
+   "uRL": "",
+   "methodFQM": "",
+   "delayInSec": "",
+   "totalDurationInSec": 20,
+   "threshOld": {"count": 0, "percentage": 0, "currentcount": 0},
+   "protocol": "",
+   "hostname": "",
+   "backendName": "",
+   "numberOfThreads": 10,
+   "threadSleepTimeInMS": 5000
+}
+CUSTOM_THREAD_LEAK_JSON = json.dumps(CUSTOM_THREAD_LEAK_DICT)
+
+# Application Kill
+APPLICATION_KILL_DICT = {
+   "enable": 1,
+   "havocType": 10,
+   "methodMod": 1,
+   "methodFQM": "",
+   "customExitCode": 0,
+   "methodEntryOrExit": 0,
+   "killApplication": 1,
+   "latencyInSec": 0,
+   "totalDurationInSec": 0,
+   "threshOld": {"count": 0, "percentage": 100, "currentcount": 0}
+}
+APPLICATION_KILL_JSON = json.dumps(APPLICATION_KILL_DICT)
+
+
+HAVOC_PROFILES_JSON = [IB_SERV_DELAY_JSON, IB_SERV_FAIL_JSON,
+                       OB_SERV_DELAY_JSON, OB_SERV_FAIL_JSON,
+                       METH_CALL_DELAY_JSON, METH_INV_FAIL_JSON,
+                       CUSTOM_MEMORY_LEAK_JSON, CUSTOM_THREAD_LEAK_JSON,
+                       APPLICATION_KILL_JSON]
+
+logger.debug("HavocConstants: Havoc constants initialized")
+