# !/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/>.
#
##################################################################################
import json
import sys
import re
from osrframework.transforms.lib.maltego import *
def uriToDomain(argv=None):
'''
Method that recovers the domain from a URI.
:param argv: the serialized entity.
'''
me = MaltegoTransform(argv)
# Recovering the Uri value
try:
uri = me.getVar("@value")
except:
uri = me.getValue()
# We add a trailing '/' to the uri if there is not a trailing one
if uri.count('/') == 2:
uri+='/'
domainRegExp = "(?:https?|s?ftp)://([a-zA-Z0-9\_\.\-]+)(?:\:|/)"
found = re.findall(domainRegExp, uri)
newEntities = []
if len(found) > 0:
aux = {}
aux["type"] = "i3visio.domain"
aux["value"] = found[0]
aux["attributes"] = []
newEntities.append(aux)
me.addListOfEntities(newEntities)
# Returning the output text...
me.returnOutput()
if __name__ == "__main__":
uriToDomain(sys.argv)