async def _create_ability(self,
ability_id,
tactic=None,
technique_name=None,
technique_id=None,
name=None,
test=None,
description=None,
executor=None,
platform=None,
cleanup=None,
payloads=None,
parsers=None,
requirements=None,
privilege=None,
timeout=60,
access=None,
buckets=None,
repeatable=False,
code=None,
language=None,
build_target=None,
variations=None,
**kwargs):
ps = []
for module in parsers:
ps.append(
Parser.load(dict(module=module,
parserconfigs=parsers[module])))
rs = []
for requirement in requirements:
for module in requirement:
rs.append(
Requirement.load(
dict(module=module,
relationship_match=requirement[module])))
ability = Ability(ability_id=ability_id,
name=name,
test=test,
tactic=tactic,
technique_id=technique_id,
technique=technique_name,
code=code,
language=language,
executor=executor,
platform=platform,
description=description,
build_target=build_target,
cleanup=cleanup,
payloads=payloads,
parsers=ps,
requirements=rs,
privilege=privilege,
timeout=timeout,
repeatable=repeatable,
variations=variations,
buckets=buckets,
**kwargs)
ability.access = access
return await self.store(ability)
async def _create_ability(self,
ability_id,
tactic=None,
technique_name=None,
technique_id=None,
name=None,
test=None,
description=None,
executor=None,
platform=None,
cleanup=None,
payloads=None,
parsers=None,
requirements=None,
privilege=None,
timeout=60,
access=None,
buckets=None,
repeatable=False,
code=None,
language=None,
build_target=None,
variations=None):
ps = []
for module in parsers:
pcs = [(ParserConfig(**m)) for m in parsers[module]]
ps.append(Parser(module=module, parserconfigs=pcs))
rs = []
for requirement in requirements:
for module in requirement:
relation = [
Relationship(source=r['source'],
edge=r.get('edge'),
target=r.get('target'))
for r in requirement[module]
]
rs.append(Requirement(module=module, relationships=relation))
ability = Ability(ability_id=ability_id,
name=name,
test=test,
tactic=tactic,
technique_id=technique_id,
technique=technique_name,
code=code,
language=language,
executor=executor,
platform=platform,
description=description,
build_target=build_target,
cleanup=cleanup,
payloads=payloads,
parsers=ps,
requirements=rs,
privilege=privilege,
timeout=timeout,
repeatable=repeatable,
variations=variations,
buckets=buckets)
ability.access = access
return await self.store(ability)