def install(self, install_prefix, distribution, configure=True, is_remote=False):
if isinstance(distribution, basestring) or not distribution:
distribution = MaltegoDistribution(distribution)
if not isinstance(distribution, MtzDistribution):
if distribution.version >= '3.4.0':
raise ValueError(INCOMPATIBLE)
print 'Installing transform package %s...' % self.name
install_prefix = self._init_install_prefix(install_prefix)
self._install_transforms(install_prefix, distribution)
self._install_entities(distribution)
self._install_machines(distribution)
if configure:
self.configure(install_prefix, remote=is_remote)
def uninstall(self, install_prefix, maltego_prefix=None):
distribution = MaltegoDistribution(maltego_prefix)
if distribution.version >= '3.4.0':
raise ValueError("""
!!!!!!!!!!!!!!!!!!!!!!!!!!! ERROR: NOT SUPPORTED !!!!!!!!!!!!!!!!!!!!!!!!!!!
Starting from Maltego Radium (v3.4.0) the 'canari uninstall-package'
command is no longer supported. Please use the Maltego interface to
uninstall packages.
!!!!!!!!!!!!!!!!!!!!!!!!!!! ERROR: NOT SUPPORTED !!!!!!!!!!!!!!!!!!!!!!!!!!!
""")
install_prefix = self._init_install_prefix(install_prefix)
self._uninstall_transforms(distribution)
self._uninstall_machines(distribution)
self.configure(install_prefix, load=False)
def install(self,
install_prefix,
distribution,
configure=True,
remote=False):
if isinstance(distribution, basestring) or not distribution:
distribution = MaltegoDistribution(distribution)
if not isinstance(distribution,
MtzDistribution) and distribution.version >= '3.4.0':
raise ValueError("""
!!!!!!!!!!!!!!!!!!!!!!!!!!! ERROR: NOT SUPPORTED !!!!!!!!!!!!!!!!!!!!!!!!!!!
Starting from Maltego Radium (v3.4.0) the 'canari install-package' command
is no longer supported. Please use the 'canari create-profile' command,
instead. This will create an importable config file (*.mtz) which can be
imported using the 'Import Configuration' option in Maltego. This option
can be found by clicking on the <Maltego icon> in the top left corner of
your Maltego window then scrolling to 'Import' then 'Import Configuration'.
NOTE: This command will automatically install and configure the
'canari.conf' file for you in the default location for your OS.
EXAMPLE:
shell> canari create-profile sploitego
...
shell> ls
sploitego.mtz <--- Import this file
!!!!!!!!!!!!!!!!!!!!!!!!!!! ERROR: NOT SUPPORTED !!!!!!!!!!!!!!!!!!!!!!!!!!!
""")
if not isinstance(distribution, MtzDistribution):
print 'Installing transform package %s...' % self.name
install_prefix = self._init_install_prefix(install_prefix)
self._install_transforms(install_prefix, distribution)
self._install_entities(distribution)
self._install_machines(distribution)
if configure:
self.configure(install_prefix, remote=remote)
def generate_entities(args):
opts = parse_args(args)
if os.path.exists(opts.outfile) and not opts.append and not \
parse_bool('%s already exists. Are you sure you want to overwrite it?' % repr(opts.outfile),
default='n'):
exit(-1)
distribution = None
if not opts.mtz_file:
distribution = MaltegoDistribution()
if distribution.version >= '3.4.0':
print("""
=========================== ERROR: NOT SUPPORTED ===========================
Starting from Maltego v3.4.0 the 'canari generate-entities' command can no
longer generate entity definition files from the Maltego configuration
directory. Entities can only be generated from export files (*.mtz). To
export entities navigate to the 'Manage' tab in Maltego, then click on the
'Export Entities' button and follow the prompts. Once the entities have
been exported, run the following command:
shell> canari generate-entities -m myentities.mtz
=========================== ERROR: NOT SUPPORTED ===========================
""")
exit(-1)
else:
distribution = MtzDistribution(opts.mtz_file)
namespaces = dict()
excluded_entities = []
if opts.append:
existing_entities = get_existing_entities(opts.outfile)
# excluded_entities.extend([e._type_ for e in existing_entities])
for entity_class in existing_entities:
excluded_entities.extend(entity_class._type_)
if entity_class._type_.endswith('Entity'):
namespaces[entity_class._namespace_] = entity_class.__name__
print 'Generating %s...' % repr(opts.outfile)
outfile = open(opts.outfile, 'ab' if opts.append else 'wb')
if opts.append:
outfile.write('\n\n')
else:
outfile.write(
'#!/usr/bin/env python\n\nfrom canari.maltego.entities import EntityField, Entity\n\n\n'
)
for entity_file in distribution.entity_files:
entity = MaltegoEntity.parse(distribution.read_file(entity_file))
if (opts.entity and entity.id
not in opts.entity) or entity.id in excluded_entities:
continue
namespace_entity = entity.id.split('.')
base_classname = None
namespace = '.'.join(namespace_entity[:-1])
name = namespace_entity[-1]
classname = name
if (opts.namespace and namespace
not in opts.namespace) or namespace in opts.exclude_namespace:
continue
print 'Generating entity definition for %s...' % entity_file
if namespace not in namespaces:
base_classname = '%sEntity' % (''.join(
[n.title() for n in namespace_entity[:-1]]))
namespaces[namespace] = base_classname
outfile.write('class %s(Entity):\n _namespace_ = %s\n\n' %
(base_classname, repr(namespace)))
else:
base_classname = namespaces[namespace]
for field in entity.properties.fields.itervalues():
fields = [
'name=%s' % repr(field.name),
'propname=%s' % repr(normalize_fn(field.name)),
'displayname=%s' % repr(field.displayname)
]
outfile.write('@EntityField(%s)\n' % ', '.join(fields))
outfile.write('class %s(%s):\n pass\n\n\n' %
(classname, base_classname))
outfile.close()
print 'done.'
def generate_entities(args):
opts = parse_args(args)
if os.path.exists(opts.outfile) and not opts.append and not \
parse_bool('%s already exists. Are you sure you want to overwrite it?' % repr(opts.outfile),
default='n'):
exit(-1)
distribution = None
if not opts.mtz_file:
distribution = MaltegoDistribution()
if distribution.version >= '3.4.0':
print("""
=========================== ERROR: NOT SUPPORTED ===========================
Starting from Maltego v3.4.0 the 'canari generate-entities' command can no
longer generate entity definition files from the Maltego configuration
directory. Entities can only be generated from export files (*.mtz). To
export entities navigate to the 'Manage' tab in Maltego, then click on the
'Export Entities' button and follow the prompts. Once the entities have
been exported, run the following command:
shell> canari generate-entities -m myentities.mtz
=========================== ERROR: NOT SUPPORTED ===========================
""")
exit(-1)
else:
distribution = MtzDistribution(opts.mtz_file)
namespaces = dict()
excluded_entities = []
if opts.append:
existing_entities = get_existing_entities(opts.outfile)
# excluded_entities.extend([e._type_ for e in existing_entities])
for entity_class in existing_entities:
excluded_entities.extend(entity_class._type_)
if entity_class._type_.endswith('Entity'):
namespaces[entity_class._namespace_] = entity_class.__name__
print 'Generating %s...' % repr(opts.outfile)
outfile = open(opts.outfile, 'ab' if opts.append else 'wb')
if opts.append:
outfile.write('\n\n')
else:
outfile.write('#!/usr/bin/env python\n\nfrom canari.maltego.entities import EntityField, Entity\n\n\n')
for entity_file in distribution.entity_files:
entity = MaltegoEntity.parse(
distribution.read_file(entity_file)
)
if (opts.entity and entity.id not in opts.entity) or entity.id in excluded_entities:
continue
namespace_entity = entity.id.split('.')
base_classname = None
namespace = '.'.join(namespace_entity[:-1])
name = namespace_entity[-1]
classname = name
if (opts.namespace and namespace not in opts.namespace) or namespace in opts.exclude_namespace:
continue
print 'Generating entity definition for %s...' % entity_file
if namespace not in namespaces:
base_classname = '%sEntity' % (''.join([n.title() for n in namespace_entity[:-1]]))
namespaces[namespace] = base_classname
outfile.write('class %s(Entity):\n _namespace_ = %s\n\n' % (base_classname, repr(namespace)))
else:
base_classname = namespaces[namespace]
for field in entity.properties.fields.itervalues():
fields = [
'name=%s' % repr(field.name),
'propname=%s' % repr(normalize_fn(field.name)),
'displayname=%s' % repr(field.displayname)
]
outfile.write('@EntityField(%s)\n' % ', '.join(fields))
outfile.write('class %s(%s):\n pass\n\n\n' % (classname, base_classname))
outfile.close()
print 'done.'