Skip to content

Commit

Permalink
include attackInfo and exploit method (#28)
Browse files Browse the repository at this point in the history
  • Loading branch information
ldruschk authored May 23, 2021
1 parent 412977a commit 2f064c8
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 1 deletion.
10 changes: 10 additions & 0 deletions enochecker_core/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ class CheckerMethod(Enum):
PUTNOISE = "putnoise"
GETNOISE = "getnoise"
HAVOC = "havoc"
EXPLOIT = "exploit"

def __str__(self) -> str:
return self.value
Expand All @@ -32,12 +33,15 @@ class CheckerInfoMessage:
flag_variants: int
noise_variants: int
havoc_variants: int
exploit_variants: int


@dataclass
class CheckerResultMessage:
result: CheckerTaskResult
message: Optional[str]
attack_info: Optional[str] = None
flag: Optional[str] = None


@dataclass
Expand All @@ -60,6 +64,9 @@ class EnoLogMessage:
flag: Optional[str]
variant_id: Optional[int]
task_chain_id: Optional[str]
flag_regex: Optional[str]
flag_hash: Optional[str]
attack_info: Optional[str]


@dataclass
Expand All @@ -76,6 +83,9 @@ class CheckerTaskMessage:
timeout: int
round_length: int
task_chain_id: str
flag_regex: Optional[str] = None
flag_hash: Optional[str] = None
attack_info: Optional[str] = None


class BrokenServiceException(Exception):
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@

setuptools.setup(
name="enochecker_core",
version="0.9.0",
version="0.10.0",
author="Trolldemorted",
author_email="[email protected]",
description="Base library for enochecker libs",
Expand Down

0 comments on commit 2f064c8

Please sign in to comment.