873e52a2a1
- adapter.py: add completed_at field to C2TaskStatus dataclass - mythic.py: implement get_task() (GraphQL task query) and get_task_output() (response query + decode_response_text concat) - fake.py: deterministic state progression via per-instance call counter; get_task_output raises C2Error until completed - mapping.py: apply_task_to_simulation() idempotent output mapper (mapping_applied anchor prevents double-writes) - migration 0007: add mapping_applied BOOLEAN NOT NULL DEFAULT false to c2_task - c2_task model: mapping_applied column added - api/c2.py: GET /api/simulations/<id>/c2/tasks poll-on-read endpoint; refreshes incomplete tasks from C2, fetches output on completion, applies mapping, skips re-polling for completed tasks; best-effort (C2Error on individual task skipped, returns 200 with stale status) - 51 new tests (396 total); pytest/ruff/mypy all green Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""C2Task model — link between a Mimic simulation and a Mythic task."""
|
|
from __future__ import annotations
|
|
|
|
import enum
|
|
from datetime import UTC, datetime
|
|
|
|
from backend.app.extensions import db
|
|
|
|
|
|
class C2TaskSource(str, enum.Enum):
|
|
MIMIC = "mimic"
|
|
IMPORT = "import"
|
|
|
|
|
|
class C2Task(db.Model): # type: ignore[name-defined]
|
|
__tablename__ = "c2_task"
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
simulation_id = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey("simulations.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
mythic_task_display_id = db.Column(db.Integer, nullable=False)
|
|
callback_display_id = db.Column(db.Integer, nullable=False)
|
|
command = db.Column(db.Text, nullable=False)
|
|
params = db.Column(db.Text, nullable=True)
|
|
status = db.Column(db.Text, nullable=False)
|
|
completed = db.Column(db.Boolean, nullable=False, default=False)
|
|
output = db.Column(db.Text, nullable=True)
|
|
source = db.Column(
|
|
db.Enum(C2TaskSource, name="c2task_source"),
|
|
nullable=False,
|
|
)
|
|
created_at = db.Column(
|
|
db.DateTime, nullable=False, default=lambda: datetime.now(UTC)
|
|
)
|
|
completed_at = db.Column(db.DateTime, nullable=True)
|
|
mapping_applied = db.Column(db.Boolean, nullable=False, default=False)
|
|
|
|
simulation = db.relationship(
|
|
"Simulation",
|
|
backref=db.backref("c2_tasks", cascade="all, delete-orphan", lazy="dynamic"),
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<C2Task simulation_id={self.simulation_id} mythic_id={self.mythic_task_display_id}>"
|