mirror of
https://github.com/ytisf/theZoo.git
synced 2025-12-05 20:19:57 -08:00
Auto-complete now works for Windows & bug fixes
This commit is contained in:
Sheksa
BIN
imports/_rlsetup.pyd
Normal file
BIN
imports/_rlsetup.pyd
Normal file
Binary file not shown.
@@ -1,8 +1,10 @@
|
||||
import sys
|
||||
import re
|
||||
import rlcompleter
|
||||
import readline
|
||||
|
||||
try:
|
||||
import readline
|
||||
except ImportError:
|
||||
from imports import winreadline as readline
|
||||
|
||||
import globals
|
||||
from imports import manysearches
|
||||
@@ -24,7 +26,7 @@ class Controller:
|
||||
("help", "Displays this help..."),
|
||||
("exit", "Exits...")]
|
||||
|
||||
self.commandsWithoutDiscription = ['search', 'list all', 'use', 'get', 'report-mal', 'update-db', 'help', 'exit']
|
||||
self.commandsWithoutDescription = {'search':'', 'list all':'', 'use':'', 'get':'', 'report-mal':'', 'update-db':'', 'help':'', 'exit':''}
|
||||
|
||||
self.searchmeth = [("arch", "which architecture etc; x86, x64, arm7 so on..."),
|
||||
("plat",
|
||||
@@ -33,7 +35,7 @@ class Controller:
|
||||
("vip", "1 or 0")]
|
||||
|
||||
self.modules = self.GetPayloads()
|
||||
completer = Completer(self.commandsWithoutDiscription)
|
||||
completer = globals.Completer(self.commandsWithoutDescription)
|
||||
|
||||
readline.parse_and_bind("tab: complete")
|
||||
readline.set_completer(completer.complete)
|
||||
@@ -44,6 +46,7 @@ class Controller:
|
||||
|
||||
def MainMenu(self):
|
||||
# This will give you the nice prompt you like so much
|
||||
while (True): # Dont hate
|
||||
if len(self.currentmodule) > 0:
|
||||
g = int(self.currentmodule) - 1
|
||||
just_print = self.modules[
|
||||
@@ -54,17 +57,20 @@ class Controller:
|
||||
else:
|
||||
cmd = raw_input(
|
||||
globals.bcolors.GREEN + 'mdb ' + globals.bcolors.GREEN + '#> ' + globals.bcolors.WHITE).strip()
|
||||
self.actOnCommand(cmd)
|
||||
|
||||
def actOnCommand(self, cmd):
|
||||
try:
|
||||
while cmd == "":
|
||||
# print 'no cmd'
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
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()
|
||||
return
|
||||
|
||||
# Checks if normal or freestyle search
|
||||
if re.match('^search', cmd):
|
||||
@@ -74,7 +80,7 @@ class Controller:
|
||||
manySearch.sort(args)
|
||||
except:
|
||||
print 'Uh oh, Invalid query.'
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
if cmd == 'exit':
|
||||
sys.exit(1)
|
||||
@@ -83,7 +89,7 @@ class Controller:
|
||||
update_handler = Updater()
|
||||
db_ver = update_handler.get_maldb_ver()
|
||||
update_handler.update_db(db_ver)
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
if cmd == 'report-mal':
|
||||
rprt_name = raw_input("Name of malware: ")
|
||||
@@ -121,7 +127,7 @@ class Controller:
|
||||
print "And attach it to the email. "
|
||||
print("Please send this report to %s" % email)
|
||||
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
if cmd == 'get':
|
||||
update_handler = Updater()
|
||||
@@ -129,7 +135,7 @@ class Controller:
|
||||
update_handler.get_malware(self.currentmodule)
|
||||
except:
|
||||
print globals.bcolors.RED + '[-] ' + globals.bcolors.WHITE + 'Error getting malware.'
|
||||
self.MainMenu()
|
||||
return
|
||||
# If used the 'use' command
|
||||
if re.match('^use', cmd):
|
||||
try:
|
||||
@@ -138,7 +144,7 @@ class Controller:
|
||||
cmd = ''
|
||||
except:
|
||||
print 'The use method needs an argument.'
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
if cmd == 'list all':
|
||||
print "\nAvailable Payloads:"
|
||||
@@ -154,7 +160,7 @@ class Controller:
|
||||
'{0: <12}'.format(array[i][globals.vars.column_for_type]))
|
||||
print answer
|
||||
i = i + 1
|
||||
self.MainMenu()
|
||||
return
|
||||
|
||||
if cmd == 'quit':
|
||||
print ":("
|
||||
|
||||
343
imports/winreadline.py
Normal file
343
imports/winreadline.py
Normal file
@@ -0,0 +1,343 @@
|
||||
"""readline.py -- Readline Alternative for Windows
|
||||
Copyright 2001, Chris Gonnerman and Alex Martelli
|
||||
"""
|
||||
#
|
||||
# By obtaining, using, and/or copying this software and/or its
|
||||
# associated documentation, you agree that you have read, understood,
|
||||
# and will comply with the following terms and conditions:
|
||||
#
|
||||
# Permission to use, copy, modify, and distribute this software and its
|
||||
# associated documentation for any purpose and without fee is hereby
|
||||
# granted, provided that the above copyright notice appears in all
|
||||
# copies, and that both that copyright notice and this permission notice
|
||||
# appear in supporting documentation.
|
||||
#
|
||||
# THE AUTHOR DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE,
|
||||
# INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO
|
||||
# EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, INDIRECT OR
|
||||
# CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF
|
||||
# USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
||||
# OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
||||
# PERFORMANCE OF THIS SOFTWARE.
|
||||
|
||||
# CREDITS:
|
||||
#
|
||||
# Small additions for history-file stuff by Alex Martelli
|
||||
#
|
||||
# Completer by Jrgen Hermann
|
||||
|
||||
__version__ = "1.7"
|
||||
|
||||
import sys, os, msvcrt, _rlsetup
|
||||
|
||||
_history = []
|
||||
|
||||
def set_history_length(n):
|
||||
global _history_length
|
||||
if n < 0:
|
||||
_history_length = sys.maxint
|
||||
else:
|
||||
_history_length = n
|
||||
|
||||
def get_history_length():
|
||||
if _history_length == sys.maxint:
|
||||
return -1
|
||||
return _history_length
|
||||
|
||||
set_history_length(int(os.getenv('PYHISTMAX','-1')))
|
||||
|
||||
def read_history_file(filename):
|
||||
def clean(s, p=['']):
|
||||
if s==p[0]:
|
||||
return 0
|
||||
p[0]=s
|
||||
return s
|
||||
fp = open(filename)
|
||||
all_lines = fp.read().splitlines()
|
||||
_history[:] = filter(clean, all_lines)[:_history_length]
|
||||
fp.close()
|
||||
|
||||
def write_history_file(filename):
|
||||
fp = open(filename, "w")
|
||||
fp.write('\n'.join(_history+['']))
|
||||
fp.close()
|
||||
|
||||
_pyhistfile = os.getenv('PYHISTFILE', sys.prefix+'\\pyhist.txt')
|
||||
|
||||
try:
|
||||
read_history_file(_pyhistfile)
|
||||
except IOError:
|
||||
pass
|
||||
|
||||
input = sys.stdin
|
||||
output = sys.stderr
|
||||
|
||||
_step = 10
|
||||
_maxlen = 70
|
||||
|
||||
_kstable = {
|
||||
'BackSpace': '\010',
|
||||
'Delete': '!S',
|
||||
'Home': '!G',
|
||||
'End': '!O',
|
||||
'Ctrl-Home': '!w',
|
||||
'Ctrl-End': '!u',
|
||||
'Right': '!M',
|
||||
'Left': '!K',
|
||||
'Tab': '\011',
|
||||
'Up': '!H',
|
||||
'Down': '!P',
|
||||
}
|
||||
|
||||
_completer = None
|
||||
|
||||
class _NullOutput:
|
||||
def write(self, s):
|
||||
pass
|
||||
|
||||
debug = _NullOutput()
|
||||
# debug = open("debug.txt", "w")
|
||||
|
||||
def BS(n):
|
||||
debug.write("BS(%d)\n" % n)
|
||||
output.write('\x08' * n)
|
||||
|
||||
class ReadlineBuffer:
|
||||
|
||||
def __init__(self):
|
||||
self.s = ""
|
||||
self.p = 0
|
||||
self.o = 0
|
||||
|
||||
def home_cursor(buf):
|
||||
debug.write("home_cursor() p = %d, o = %d\n" % (buf.p, buf.o))
|
||||
BS(buf.p - buf.o)
|
||||
if buf.o > 0:
|
||||
BS(1)
|
||||
|
||||
def paint_tail(buf):
|
||||
debug.write("paint_tail()\n")
|
||||
tail = buf.s[buf.p:min(len(buf.s), buf.maxlen + buf.o)]
|
||||
output.write(tail)
|
||||
output.write(" \x08\x08")
|
||||
if len(buf.s) > (buf.maxlen + buf.o):
|
||||
output.write(">\x08")
|
||||
BS(len(tail))
|
||||
|
||||
def rewrite_buffer(buf):
|
||||
debug.write("rewrite_buffer()\n")
|
||||
debug.write("buf.o = %d\n" % buf.o)
|
||||
if buf.o > 0:
|
||||
output.write("<")
|
||||
win = buf.s[buf.o:buf.maxlen+buf.o]
|
||||
clr = max(buf.maxlen - len(win), 0)
|
||||
debug.write("win = " + win + "\n")
|
||||
debug.write("clr = %d\n" % clr)
|
||||
output.write(win+(" "*(clr + 2)))
|
||||
BS(2)
|
||||
if len(buf.s) > (buf.maxlen + buf.o):
|
||||
output.write(">\x08")
|
||||
BS(max(buf.maxlen - (buf.p - buf.o), 0))
|
||||
|
||||
def correct_offset(buf):
|
||||
new_o = buf.o
|
||||
while new_o and buf.p < (new_o + buf.step):
|
||||
new_o = new_o - buf.step
|
||||
while buf.p > (buf.maxlen + new_o - buf.step):
|
||||
new_o = new_o + buf.step
|
||||
if new_o == buf.o:
|
||||
return
|
||||
buf.home_cursor()
|
||||
buf.o = new_o
|
||||
buf.rewrite_buffer()
|
||||
|
||||
def new_line(buf):
|
||||
debug.write("new_line()\n")
|
||||
buf.p = len(buf.s)
|
||||
if len(buf.s) > (buf.maxlen - buf.step):
|
||||
buf.o = (((len(buf.s) - buf.maxlen) / buf.step) + 1) * buf.step
|
||||
else:
|
||||
buf.o = 0
|
||||
buf.rewrite_buffer()
|
||||
|
||||
# dummy functions
|
||||
|
||||
def set_completer(function=None):
|
||||
global _completer
|
||||
_completer = function
|
||||
|
||||
def parse_and_bind(line):
|
||||
pass
|
||||
|
||||
def read_init_file(filename=None):
|
||||
pass
|
||||
|
||||
|
||||
# This makes it easier to tell if this code loaded correctly.
|
||||
# Python "hides" any errors loading readline.
|
||||
# Change to a true value to hide.
|
||||
_issued = None
|
||||
|
||||
def readline(step = _step, maxlen = _maxlen,
|
||||
history = _history, histfile = _pyhistfile):
|
||||
"""readline(step, maxlen, history, histfile)
|
||||
Read a line from the console.
|
||||
|
||||
step: Number of columns to side-scroll per step.
|
||||
maxlen: Maximum length of string to read.
|
||||
history: List of strings comprising the current history.
|
||||
histfile: History file name (for saving history), None to disable.
|
||||
"""
|
||||
|
||||
global _issued
|
||||
|
||||
if not _issued:
|
||||
_issued = 1
|
||||
|
||||
buf = ReadlineBuffer()
|
||||
buf.maxlen = maxlen
|
||||
buf.step = step
|
||||
|
||||
pos = len(history)
|
||||
|
||||
c_state = 0
|
||||
|
||||
while 1:
|
||||
c_new_state = 0
|
||||
|
||||
buf.p = max(min(buf.p, len(buf.s)), 0)
|
||||
|
||||
buf.correct_offset()
|
||||
|
||||
c = msvcrt.getch()
|
||||
|
||||
if c == '\0' or c == '\xe0':
|
||||
c = "!" + msvcrt.getch()
|
||||
|
||||
if c == '\r' or c == '\n':
|
||||
buf.s = buf.s + '\n'
|
||||
break
|
||||
|
||||
elif c == '\x1b':
|
||||
buf.s = ""
|
||||
|
||||
elif c == '\x04' or c == '\x1a':
|
||||
buf.s = ""
|
||||
break
|
||||
|
||||
elif c == _kstable["BackSpace"]:
|
||||
if c_state:
|
||||
buf.home_cursor()
|
||||
buf.s, buf.p = c_save
|
||||
buf.rewrite_buffer()
|
||||
elif buf.s:
|
||||
buf.s = buf.s[:buf.p-1] + buf.s[buf.p:]
|
||||
buf.p = buf.p - 1
|
||||
BS(1)
|
||||
buf.paint_tail()
|
||||
|
||||
elif c == _kstable["Delete"]:
|
||||
if buf.s:
|
||||
buf.s = buf.s[:buf.p] + buf.s[buf.p+1:]
|
||||
buf.paint_tail()
|
||||
|
||||
elif c == _kstable["Up"]:
|
||||
if pos:
|
||||
pos = pos - 1
|
||||
buf.home_cursor()
|
||||
buf.s = history[pos]
|
||||
buf.new_line()
|
||||
|
||||
elif c == _kstable["Ctrl-Home"]:
|
||||
if pos:
|
||||
pos = 0
|
||||
buf.home_cursor()
|
||||
buf.s = history[pos]
|
||||
buf.new_line()
|
||||
|
||||
elif c == _kstable["Down"]:
|
||||
if pos < len(history):
|
||||
pos = pos + 1
|
||||
buf.home_cursor()
|
||||
if pos < len(history):
|
||||
buf.s = history[pos]
|
||||
else:
|
||||
buf.s = ""
|
||||
buf.new_line()
|
||||
|
||||
elif c == _kstable["Ctrl-End"]:
|
||||
if pos < len(history):
|
||||
pos = len(history)
|
||||
buf.home_cursor()
|
||||
buf.s = ""
|
||||
buf.p = 0
|
||||
|
||||
elif c == _kstable["Home"] or c == '\001':
|
||||
buf.home_cursor()
|
||||
buf.p = 0
|
||||
buf.o = 0
|
||||
buf.paint_tail()
|
||||
|
||||
elif c == _kstable["End"] or c == '\005':
|
||||
buf.home_cursor()
|
||||
debug.write("END buf.p = %d, buf.o = %d\n" % (buf.p, buf.o))
|
||||
buf.new_line()
|
||||
debug.write("END\n")
|
||||
|
||||
elif c == _kstable["Left"] or c == '\002':
|
||||
buf.p = buf.p - 1
|
||||
if buf.p >= 0:
|
||||
BS(1)
|
||||
|
||||
elif c == _kstable["Right"] or c == '\006':
|
||||
if buf.s[buf.p:]:
|
||||
output.write(buf.s[buf.p])
|
||||
buf.p = buf.p + 1
|
||||
|
||||
elif (c == _kstable["Tab"] or c == '\011') and _completer:
|
||||
if c_state == 0:
|
||||
c_save = (buf.s, buf.p)
|
||||
c_pos = buf.p
|
||||
while c_pos > 0 and (buf.s[c_pos-1].isalnum()
|
||||
or buf.s[c_pos-1] in '_.'):
|
||||
c_pos -= 1
|
||||
c_text = buf.s[c_pos:buf.p]
|
||||
|
||||
try:
|
||||
c_result = _completer(c_text, c_state)
|
||||
except:
|
||||
c_result = None
|
||||
|
||||
if c_result:
|
||||
buf.home_cursor()
|
||||
c_new_state = c_state + 1
|
||||
buf.s = c_save[0][:c_pos] + c_result + c_save[0][c_save[1]:]
|
||||
buf.p = c_pos + len(c_result)
|
||||
buf.rewrite_buffer()
|
||||
|
||||
elif len(c) == 1 and c >= " ":
|
||||
output.write(c)
|
||||
buf.s = buf.s[:buf.p] + c + buf.s[buf.p:]
|
||||
buf.p = buf.p + len(c)
|
||||
buf.paint_tail()
|
||||
|
||||
c_state = c_new_state
|
||||
|
||||
output.write("\n")
|
||||
|
||||
if buf.s != "\n":
|
||||
if not history or history[-1] != buf.s[:-1]:
|
||||
history.append(buf.s[:-1])
|
||||
history = history[-1*_history_length:]
|
||||
if histfile and type(histfile) is type(""):
|
||||
fp = open(histfile, "w")
|
||||
for line in history:
|
||||
if line:
|
||||
fp.write(line + "\n")
|
||||
fp.close()
|
||||
|
||||
return buf.s
|
||||
|
||||
_rlsetup.install_readline(readline)
|
||||
|
||||
# end of file.
|
||||
Reference in New Issue
Block a user