mirror of
https://github.com/ytisf/theZoo.git
synced 2026-04-28 12:03:10 -07:00
Updating to v0.43 - minor fixes
This commit is contained in:
nativy
@@ -51,10 +51,10 @@ class EULA:
|
||||
print '|infect you machines will live and dangerous malwares!. |'
|
||||
print '|___________________________________________________________________________|'
|
||||
print globals.bcolors.WHITE
|
||||
eula_answer = raw_input('Type YES in captial letters to accept this EULA.\n >')
|
||||
eula_answer = raw_input('Type YES in captial letters to accept this EULA.\n > ')
|
||||
if eula_answer == 'YES':
|
||||
new = open(globals.vars.eula_file, 'a')
|
||||
new.write(eula_answer)
|
||||
else:
|
||||
print 'You need to accept the EULA.\nExiting the program.'
|
||||
sys.exit(1)
|
||||
sys.exit(0)
|
||||
@@ -8,163 +8,204 @@ from imports.updatehandler import Updater
|
||||
|
||||
|
||||
class Controller:
|
||||
def __init__(self):
|
||||
self.modules = None
|
||||
self.currentmodule = ''
|
||||
self.commands = [ ("search", "searching for malwares using given parameter with 'set'."),
|
||||
("list all", "lists all available modules"),
|
||||
("set", "sets options for the search"),
|
||||
("get", "downloads the malware"),
|
||||
("update-db", "updates the databse"),
|
||||
("back", "removes currently chosen malware"),
|
||||
("help", "displays this help..."),
|
||||
("exit", "exits...")]
|
||||
def __init__(self):
|
||||
self.modules = None
|
||||
self.currentmodule = ''
|
||||
self.commands = [("search", "searching for malwares using given parameter with 'set'."),
|
||||
("list all", "lists all available modules"),
|
||||
("set", "sets options for the search"),
|
||||
("get", "downloads the malware"),
|
||||
("report-mal", "report a malware you found"),
|
||||
("update-db", "updates the databse"),
|
||||
("back", "removes currently chosen malware and filters"),
|
||||
("help", "displays this help..."),
|
||||
("exit", "exits...")]
|
||||
|
||||
self.searchmeth = [ ("arch","which architecture etc; x86, x64, arm7 so on..."),
|
||||
("plat","platform: win32, win64, mac, android so on..."),
|
||||
("lang","c, cpp, vbs, bin so on..."),
|
||||
("vip", "1 or 0")]
|
||||
self.searchmeth = [("arch", "which architecture etc; x86, x64, arm7 so on..."),
|
||||
("plat", "platform: win32, win64, mac, android so on..."),
|
||||
("lang", "c, cpp, vbs, bin so on..."),
|
||||
("vip", "1 or 0")]
|
||||
|
||||
self.modules = self.GetPayloads()
|
||||
self.modules = self.GetPayloads()
|
||||
|
||||
#print 'im at init'
|
||||
self.plat = ''
|
||||
self.arch = ''
|
||||
self.lang = ''
|
||||
self.type = ''
|
||||
self.vip = ''
|
||||
self.plat = ''
|
||||
self.arch = ''
|
||||
self.lang = ''
|
||||
self.type = ''
|
||||
self.vip = ''
|
||||
|
||||
def GetPayloads(self):
|
||||
m = []
|
||||
csvReader = csv.reader(open(globals.vars.main_csv_file, 'rb'), delimiter=',')
|
||||
for row in csvReader:
|
||||
m.append(row)
|
||||
return m
|
||||
def GetPayloads(self):
|
||||
m = []
|
||||
csvReader = csv.reader(open(globals.vars.main_csv_file, 'rb'), delimiter=',')
|
||||
for row in csvReader:
|
||||
m.append(row)
|
||||
return m
|
||||
|
||||
def MainMenu(self):
|
||||
if len(self.currentmodule) > 0:
|
||||
g = int(self.currentmodule) - 1
|
||||
just_print = self.modules[int(g)][int(globals.vars.column_for_name)]
|
||||
cmd = raw_input(
|
||||
globals.bcolors.GREEN + 'mdb ' + globals.bcolors.RED + str(just_print) + globals.bcolors.GREEN + '#> ' + globals.bcolors.WHITE).strip()
|
||||
else:
|
||||
cmd = raw_input(globals.bcolors.GREEN + 'mdb ' + globals.bcolors.GREEN + '#> ' + globals.bcolors.WHITE).strip()
|
||||
def MainMenu(self):
|
||||
# This will give you the nice prompt you like to much
|
||||
if len(self.currentmodule) > 0:
|
||||
g = int(self.currentmodule) - 1
|
||||
just_print = self.modules[int(g)][int(globals.vars.column_for_name)]
|
||||
cmd = raw_input(
|
||||
globals.bcolors.GREEN + 'mdb ' + globals.bcolors.RED + str(
|
||||
just_print) + globals.bcolors.GREEN + '#> ' + globals.bcolors.WHITE).strip()
|
||||
else:
|
||||
cmd = raw_input(
|
||||
globals.bcolors.GREEN + 'mdb ' + globals.bcolors.GREEN + '#> ' + globals.bcolors.WHITE).strip()
|
||||
|
||||
try:
|
||||
while cmd == "":
|
||||
#print 'no cmd'
|
||||
self.MainMenu()
|
||||
try:
|
||||
while cmd == "":
|
||||
#print 'no cmd'
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'help':
|
||||
print " Available commands:\n"
|
||||
for (cmd, desc) in self.commands:
|
||||
print "\t%s\t%s" % ('{0: <12}'.format(cmd), desc)
|
||||
print ''
|
||||
self.MainMenu()
|
||||
if cmd == 'help':
|
||||
print " Available commands:\n"
|
||||
for (cmd, desc) in self.commands:
|
||||
print "\t%s\t%s" % ('{0: <12}'.format(cmd), desc)
|
||||
print ''
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'search':
|
||||
ar = self.modules
|
||||
manySearch = manysearches.MuchSearch()
|
||||
if cmd == 'search':
|
||||
ar = self.modules
|
||||
manySearch = manysearches.MuchSearch()
|
||||
|
||||
# function to sort by arch
|
||||
if len(self.arch) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_arch, self.arch)
|
||||
# function to sort by plat
|
||||
if len(self.plat) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_plat, self.plat)
|
||||
# function to sort by lang
|
||||
if len(self.lang) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_pl, self.lang)
|
||||
if len(self.type) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_type, self.type)
|
||||
if len(self.vip) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_vip, self.vip)
|
||||
printController = manysearches.MuchSearch()
|
||||
printController.PrintPayloads(ar)
|
||||
self.MainMenu()
|
||||
# function to sort by arch
|
||||
if len(self.arch) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_arch, self.arch)
|
||||
# function to sort by plat
|
||||
if len(self.plat) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_plat, self.plat)
|
||||
# function to sort by lang
|
||||
if len(self.lang) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_pl, self.lang)
|
||||
if len(self.type) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_type, self.type)
|
||||
if len(self.vip) > 0:
|
||||
ar = manySearch.sort(ar, globals.vars.column_for_vip, self.vip)
|
||||
printController = manysearches.MuchSearch()
|
||||
printController.PrintPayloads(ar)
|
||||
self.MainMenu()
|
||||
|
||||
if re.match('^set', cmd):
|
||||
try:
|
||||
cmd = re.split('\s+', cmd)
|
||||
print cmd[1] + ' => ' + cmd[2]
|
||||
if cmd[1] == 'arch':
|
||||
self.arch = cmd[2]
|
||||
if cmd[1] == 'plat':
|
||||
self.plat = cmd[2]
|
||||
if cmd[1] == 'lang':
|
||||
self.lang = cmd[2]
|
||||
if cmd[1] == 'type':
|
||||
self.type = cmd[2]
|
||||
except:
|
||||
print 'Need to use the set method with two arguments.'
|
||||
cmd = ''
|
||||
self.MainMenu()
|
||||
if re.match('^set', cmd):
|
||||
try:
|
||||
cmd = re.split('\s+', cmd)
|
||||
print cmd[1] + ' => ' + cmd[2]
|
||||
if cmd[1] == 'arch':
|
||||
self.arch = cmd[2]
|
||||
if cmd[1] == 'plat':
|
||||
self.plat = cmd[2]
|
||||
if cmd[1] == 'lang':
|
||||
self.lang = cmd[2]
|
||||
if cmd[1] == 'type':
|
||||
self.type = cmd[2]
|
||||
except:
|
||||
print 'Need to use the set method with two arguments.'
|
||||
cmd = ''
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'show':
|
||||
if len(self.currentmodule) == 0:
|
||||
print "No modules have been chosen. Use 'use' command."
|
||||
if len(self.currentmodule) > 0:
|
||||
print 'Currently selected Module: ' + self.currentmodule
|
||||
print '\tarch => ' + str(self.arch)
|
||||
print '\tplat => ' + str(self.plat)
|
||||
print '\tlang => ' + str(self.lang)
|
||||
print '\ttype => ' + str(self.type)
|
||||
print ''
|
||||
self.MainMenu()
|
||||
if cmd == 'show':
|
||||
if len(self.currentmodule) == 0:
|
||||
print "No modules have been chosen. Use 'use' command."
|
||||
if len(self.currentmodule) > 0:
|
||||
print 'Currently selected Module: ' + self.currentmodule
|
||||
print '\tarch => ' + str(self.arch)
|
||||
print '\tplat => ' + str(self.plat)
|
||||
print '\tlang => ' + str(self.lang)
|
||||
print '\ttype => ' + str(self.type)
|
||||
print ''
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'exit':
|
||||
sys.exit(1)
|
||||
if cmd == 'exit':
|
||||
sys.exit(1)
|
||||
|
||||
if cmd == 'update-db':
|
||||
updateHandler = Updater()
|
||||
updateHandler.get_maldb_ver()
|
||||
self.MainMenu()
|
||||
if cmd == 'update-db':
|
||||
updateHandler = Updater()
|
||||
updateHandler.get_maldb_ver()
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'get':
|
||||
updateHandler = Updater()
|
||||
try:
|
||||
updateHandler.get_malware(self.currentmodule, self.modules)
|
||||
self.MainMenu()
|
||||
except:
|
||||
print globals.bcolors.RED + '[-]' + globals.bcolors.WHITE + 'Error getting malware.'
|
||||
self.MainMenu()
|
||||
if cmd == 'report-mal':
|
||||
rprt_name = raw_input("Name of malware: ")
|
||||
rprt_type = raw_input("Type of malware: ")
|
||||
rprt_version = raw_input("Version: ")
|
||||
rprt_lang = raw_input("Language: ")
|
||||
rprt_src = raw_input("Source / Binary (s/b): ")
|
||||
rprt_arch = raw_input("Win32, ARM etc. ? ")
|
||||
rprt_reporter = raw_input("Your name for a thanks note on theZoo.\nPlease notice that this will be public!\n\nName: ")
|
||||
rprt_comments = raw_input("Comments? ")
|
||||
|
||||
if re.match('^use', cmd):
|
||||
try:
|
||||
cmd = re.split('\s+', cmd)
|
||||
self.currentmodule = cmd[1]
|
||||
cmd = ''
|
||||
except:
|
||||
print 'The use method needs an argument.'
|
||||
self.MainMenu()
|
||||
report = ("//%s//\n" % rprt_name)
|
||||
report += ("///type/%s///\n" % rprt_type)
|
||||
report += ("///ver/%s///\n" % rprt_version)
|
||||
report += ("///lang/%s///\n" % rprt_lang)
|
||||
report += ("///src/%s///\n" % rprt_src)
|
||||
report += ("///arch/%s///\n" % rprt_arch)
|
||||
report += ("//reporter/%s//\n" % rprt_reporter)
|
||||
report += ("//comments/%s//\n" % rprt_comments)
|
||||
|
||||
if cmd == 'back':
|
||||
print 'im at back - WTF?'
|
||||
self.arch = ''
|
||||
self.plat = ''
|
||||
self.lang = ''
|
||||
self.type = ''
|
||||
self.currentmodule = ''
|
||||
self.MainMenu()
|
||||
# Just to avoid bots spamming us...
|
||||
email = "info"
|
||||
email += "\x40"
|
||||
email += "morirt\x2ecom"
|
||||
print "-------------- Begin of theZoo Report --------------"
|
||||
print report
|
||||
print "-------------- Ending of theZoo Report --------------"
|
||||
print "To avoid compromising your privacy we have chose this method of reporting."
|
||||
print "If you have not stated your name we will not write a thanks in our README."
|
||||
print "Your email will remain private in scenario and will not be published."
|
||||
print ""
|
||||
print "Please create an archive file with the structure as in the README file"
|
||||
print "And attach it to the email. "
|
||||
print("Please send this report to %s" % email)
|
||||
|
||||
if cmd == 'list all':
|
||||
print "\nAvailable Payloads:"
|
||||
array = self.modules
|
||||
i = 0
|
||||
print "ID\tName\tType"
|
||||
print '-----------------'
|
||||
for element in array:
|
||||
answer = array[i][globals.vars.column_for_uid]
|
||||
answer += '\t%s' % ('{0: <12}'.format(array[i][globals.vars.column_for_name]))
|
||||
answer += '\t%s' % ('{0: <12}'.format(array[i][globals.vars.column_for_type]))
|
||||
print answer
|
||||
i=i+1
|
||||
self.MainMenu()
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'quit':
|
||||
print ":("
|
||||
sys.exit(1)
|
||||
# 'get' command. Not yet fully operational
|
||||
if cmd == 'get':
|
||||
updateHandler = Updater()
|
||||
try:
|
||||
updateHandler.get_malware(self.currentmodule, self.modules)
|
||||
self.MainMenu()
|
||||
except:
|
||||
print globals.bcolors.RED + '[-]' + globals.bcolors.WHITE + 'Error getting malware.'
|
||||
self.MainMenu()
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print ("i'll just go now...")
|
||||
sys.exit()
|
||||
# If used the 'use' command
|
||||
if re.match('^use', cmd):
|
||||
try:
|
||||
cmd = re.split('\s+', cmd)
|
||||
self.currentmodule = cmd[1]
|
||||
cmd = ''
|
||||
except:
|
||||
print 'The use method needs an argument.'
|
||||
self.MainMenu()
|
||||
|
||||
# Rests all current data
|
||||
if cmd == 'back':
|
||||
self.arch = ''
|
||||
self.plat = ''
|
||||
self.lang = ''
|
||||
self.type = ''
|
||||
self.currentmodule = ''
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'list all':
|
||||
print "\nAvailable Payloads:"
|
||||
array = self.modules
|
||||
i = 0
|
||||
print "ID\tName\tType"
|
||||
print '-----------------'
|
||||
for element in array:
|
||||
answer = array[i][globals.vars.column_for_uid]
|
||||
answer += '\t%s' % ('{0: <12}'.format(array[i][globals.vars.column_for_name]))
|
||||
answer += '\t%s' % ('{0: <12}'.format(array[i][globals.vars.column_for_type]))
|
||||
print answer
|
||||
i = i + 1
|
||||
self.MainMenu()
|
||||
|
||||
if cmd == 'quit':
|
||||
print ":("
|
||||
sys.exit(1)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print ("i'll just go now...")
|
||||
sys.exit()
|
||||
|
||||
@@ -39,11 +39,12 @@ class Updater:
|
||||
return 0
|
||||
|
||||
curr_maldb_ver = f
|
||||
response = urllib2.urlopen(globals.vars.giturl+ globals.vars.maldb_ver_file)
|
||||
response = urllib2.urlopen(globals.vars.giturl + globals.vars.maldb_ver_file)
|
||||
new_maldb_ver = response.read()
|
||||
if new_maldb_ver == curr_maldb_ver:
|
||||
print globals.bcolors.GREEN + '[+]' + globals.bcolors.WHITE + " No need for an update.\n" + globals.bcolors.GREEN + '[+]' + globals.bcolors.WHITE + " You are at " + new_maldb_ver + " which is the latest version."
|
||||
sys.exit(1)
|
||||
|
||||
# Write the new DB version into the file
|
||||
f = open(globals.vars.maldb_ver_file, 'w')
|
||||
f.write(new_maldb_ver)
|
||||
|
||||
Reference in New Issue
Block a user