discover.py 2.57 KB
Newer Older
Mati's avatar
Mati committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
#!/usr/bin/python

"""
Inguma Penetration Testing Toolkit
Copyright (c) 2006, 2007 Joxean Koret, joxeankoret [at] yahoo.es

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; version 2
of the License.

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, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
"""

import sys

from inguma_module import *

def doDiscover(vars):

    targets = ()

    if not vars:
        print "INTERNAL ERROR: No one variable was passed"
        return False
    else:
        target = vars[0]
        port = vars[1]
        covert = vars[2]
        timeout = vars[3]
        waittime = vars[4]
        wizard = vars[5]
        modules = vars[6]

    try:
        if target == "":
            mTarget = raw_input("Target: ")
        else:
            mTarget = target
    
        vars = (mTarget, port, covert, timeout, waittime, wizard)

        print 
        print "Available modules:"
        print

        index = 0
        for x in modules:
            print "\t",index+1,x.name,"  \t",x.brief_description
            index += 1

        print
        res = raw_input("Select module [all]: ")

        try:
            if res.lower() == "all" or res.lower() == "":
                for x in modules:
                    targets += runModule(vars, x)
            else:
                targets += runModule(vars, modules[int(res)-1])
        except KeyboardInterrupt:
            print "Aborted."
        except EOFError:
            print "Aborted."
        except:
            print "Internal error:",sys.exc_info()[1]

        tmp = (mTarget, )
        tmp += cleanTargets(targets)
        ret = cleanTargets(tmp)

        return ret
    except KeyboardInterrupt:
        print "Aborted."
    except EOFError:
        print "Aborted."
    except:
        print "Error from module:",sys.exc_info()[1]
        return

def cleanTargets(targets):

    ret = ()
    
    for x in targets:
        bFound = False
        for y in ret:
            if y == x:
                bFound = True
                break
            else:
                bFound = False
                
        if not bFound:
            ret += (x, )

    return ret