Files
sigma-rules/detection_rules/schemas/__init__.py
T
Ross Wolf 8c92ae7348 Add ATT&CK subtechniques to the schema (#337)
* Add ATT&CK subtechniques to the schema
* Switch subtechniques to the 7.11 schema
* Make technique still required
* Lint fixes
* Cleanup EQL constant
* Trim more cruft
* Restore EQL for 710

Co-authored-by: Justin Ibarra <brokensound77@users.noreply.github.com>
2020-12-08 14:57:30 -07:00

60 lines
1.7 KiB
Python

# Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
# or more contributor license agreements. Licensed under the Elastic License;
# you may not use this file except in compliance with the Elastic License.
from .base import TomlMetadata
from .rta_schema import validate_rta_mapping
from ..semver import Version
# import all of the schema versions
from .v7_8 import ApiSchema78
from .v7_9 import ApiSchema79
from .v7_10 import ApiSchema710
from .v7_11 import ApiSchema711
__all__ = (
"all_schemas",
"downgrade",
"CurrentSchema",
"validate_rta_mapping",
"TomlMetadata",
)
all_schemas = [
ApiSchema78,
ApiSchema79,
ApiSchema710,
ApiSchema711,
]
CurrentSchema = all_schemas[-1]
def downgrade(api_contents: dict, target_version: str):
"""Downgrade a rule to a target stack version."""
# truncate to (major, minor)
target_version_str = target_version
target_version = Version(target_version)[:2]
versions = set(Version(schema_cls.STACK_VERSION) for schema_cls in all_schemas)
role = api_contents.get("type")
check_versioned = "version" in api_contents
if target_version not in versions:
raise ValueError(f"Unable to downgrade from {CurrentSchema.STACK_VERSION} to {target_version_str}")
current_schema = None
for target_schema in reversed(all_schemas):
if check_versioned:
target_schema = target_schema.versioned()
if current_schema is not None:
api_contents = current_schema.downgrade(target_schema, api_contents, role)
current_schema = target_schema
if Version(current_schema.STACK_VERSION) == target_version:
break
return api_contents