# !/usr/bin/python
# -*- coding: cp1252 -*-
#
##################################################################################
#
# Copyright 2015 FĂ©lix Brezo and Yaiza Rubio (i3visio, [email protected])
#
# This program is part of OSRFramework. You can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
##################################################################################
from osrframework.transforms.lib.maltego import *
import sys
import json
import osrframework.transforms.lib.constants as constants
def expandPropertiesFromI3visioEntity(argv):
'''
Method that expands the properties from a given i3visio entity. It is useful to create new Entities based on the contents of the properties.
:param argv: the serialized entity.
:return: Nothing is returned but the code of the entities is created.
'''
me = MaltegoTransform(argv)
newEntities = []
try:
# Trying to recover pending entities if they exist.
# If so, it means that it is NOT the first time that this info is recovered
if str(me.getVar("@number_pending")) != "0" and me.getVar("@number_pending") != None:
entitiesToShow = me.getVar("@pending")
#print entitiesToShow
newEntities = json.loads(entitiesToShow)
me.addListOfEntities(newEntities)
else:
# We will try to expand the rest of the attributes
for field in constants.I3VISIO_FIELDS:
value = me.getVar(field)
#print field, value
if value != None and value != " " and value != "":
aux = {}
aux["type"] = field
aux["value"] = value
aux["attributes"] = []
newEntities.append(aux)
except:
# In case of errors, we will try to expand the rest of the attributes
for field in constants.I3VISIO_FIELDS:
value = me.getVar(field)
#print field, value
if value != None and value != " " and value != "":
aux = {}
aux["type"] = field
aux["value"] = value
aux["attributes"] = []
newEntities.append(aux)
me.addListOfEntities(newEntities)
# Returning the output text...
me.returnOutput()
if __name__ == "__main__":
expandPropertiesFromI3visioEntity(sys.argv)