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)