File size: 702 Bytes
a0fe78f
 
 
 
26aeea9
 
 
a0fe78f
 
 
 
26aeea9
 
 
 
 
a0fe78f
 
 
26aeea9
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from .sql_injection           import SQLInjectionTask
from .spearphish_credential   import SpearphishCredentialTask
from .cloud_identity_intrusion import CloudIdentityIntrusionTask
from .ai_tool_exploitation     import AIToolExploitationTask

# Registry: task_id → class
TASK_REGISTRY = {
    SQLInjectionTask.TASK_ID:          SQLInjectionTask,
    SpearphishCredentialTask.TASK_ID:  SpearphishCredentialTask,
    CloudIdentityIntrusionTask.TASK_ID: CloudIdentityIntrusionTask,
    AIToolExploitationTask.TASK_ID:    AIToolExploitationTask,
}

__all__ = [
    "TASK_REGISTRY",
    "SQLInjectionTask",
    "SpearphishCredentialTask",
    "CloudIdentityIntrusionTask",
    "AIToolExploitationTask",
]