def __init__(self,
domain='enterprise',
source='taxii',
local=None,
config=None):
"""
Sets up exporting system, builds underlying matrix
:param domain: Which domain to utilize for the underlying matrix layout
:param source: Use the taxii server or local data
:param local: Optional path to local stix data
:param config: Optional pre-existing SVGConfig object
"""
self.raw_handle = SvgTemplates(domain=domain,
source=source,
local=local)
if config != None and isinstance(config, SVGConfig):
self.config = config
else:
self.config = SVGConfig()
class ToSvg:
def __init__(self,
domain='enterprise',
source='taxii',
local=None,
config=None):
"""
Sets up exporting system, builds underlying matrix
:param domain: Which domain to utilize for the underlying matrix layout
:param source: Use the taxii server or local data
:param local: Optional path to local stix data
:param config: Optional pre-existing SVGConfig object
"""
self.raw_handle = SvgTemplates(domain=domain,
source=source,
local=local)
if config != None and isinstance(config, SVGConfig):
self.config = config
else:
self.config = SVGConfig()
def to_svg(self, layer, filepath='example.svg'):
"""
Generate a svg file based on the input layer
:param layer: Input attack layer object to transform
:param filepath: Desired output svg location
:return: (meta) svg file at the targeted output location
"""
if layer is not None:
if not isinstance(layer, Layer):
raise TypeError
if layer is None:
raise NoLayer
included_subs = []
if layer.layer.techniques:
for entry in layer.layer.techniques:
if self.config.showSubtechniques == "expanded":
if entry.showSubtechniques:
if entry.tactic:
included_subs.append(
(entry.techniqueID, entry.tactic))
else:
included_subs.append((entry.techniqueID, False))
elif self.config.showSubtechniques == "all":
if entry.tactic:
included_subs.append((entry.techniqueID, entry.tactic))
else:
included_subs.append((entry.techniqueID, False))
else: # none displayed
pass
excluded = []
if layer.layer.hideDisabled:
for entry in layer.layer.techniques:
if entry.enabled == False:
if entry.tactic:
excluded.append((entry.techniqueID, entry.tactic))
else:
excluded.append((entry.techniqueID, False))
scores = []
colors = []
if layer.layer.techniques:
for entry in layer.layer.techniques:
if entry.score:
if entry.tactic:
scores.append(
(entry.techniqueID, entry.tactic, entry.score))
else:
scores.append((entry.techniqueID, False, entry.score))
elif entry.color:
if entry.tactic:
colors.append(
(entry.techniqueID, entry.tactic, entry.color))
else:
colors.append((entry.techniqueID, False, entry.color))
sName = True
sID = False
sort = 0
if layer.layer.layout:
sName = layer.layer.layout.showName
sID = layer.layer.layout.showID
if layer.layer.sorting:
sort = layer.layer.sorting
d = self.raw_handle.export(showName=sName,
showID=sID,
sort=sort,
scores=scores,
subtechs=included_subs,
colors=colors,
exclude=excluded,
lhandle=layer.layer,
config=self.config)
d.saveSvg(filepath)