def complete_use(self, text, line, begidx, endidx):
if text:
return [
i.lower() for i in PAYLOADS.keys()
if i.lower().startswith(text.lower())
]
return [i.lower() for i in PAYLOADS.keys()]
def initializePoc(folders=[]):
pocNumber = 0
if not os.path.isdir(paths.POC_PATH):
os.makedirs(paths.POC_PATH)
folders.append(paths.POC_PATH)
for folder in folders:
files = os.listdir(folder)
for file_i in files:
if file_i.endswith(".py") and "__init__" not in file_i:
PAYLOADS.update({ntpath.splitext(file_i)[0]: os.path.join(folder, file_i)})
def initializePoc(folders=[]):
pocNumber = 0
if not os.path.isdir(paths.POC_PATH):
os.makedirs(paths.POC_PATH)
folders.append(paths.POC_PATH)
for folder in folders:
files = os.listdir(folder)
for file_i in files:
if file_i.endswith(".py") and "__init__" not in file_i:
PAYLOADS.update(
{ntpath.splitext(file_i)[0]: os.path.join(folder, file_i)})
def complete_search(self, text, line, begidx, endidx):
selist = []
selist.extend(self.all_2_list(PAYLOADS.keys()))
selist.extend(self.all_2_list(cache))
selist = list(set(selist))
if text:
return [i for i in selist if text in i]
return [i for i in selist]
def complete_search(self, text, line, begidx, endidx):
selist = []
selist.extend(self.all_2_list(PAYLOADS.keys()))
selist.extend(self.all_2_list(cache))
selist = list(set(selist))
if text:
return [i for i in selist if text in i]
return [i for i in selist]
def do_search(self, args):
if not args:
return
selist = []
selist.extend(self.all_2_list(PAYLOADS.keys()))
selist.extend(self.all_2_list(cache))
selist = list(set(selist))
res = [i for i in selist if args in i]
for i in res:
self.ui.hilight.hilight(i, args,self.ui.GREEN, self.ui.RED)
def do_search(self, args):
if not args:
return
selist = []
selist.extend(self.all_2_list(PAYLOADS.keys()))
selist.extend(self.all_2_list(cache))
selist = list(set(selist))
res = [i for i in selist if args in i]
for i in res:
self.ui.hilight.hilight(i, args, self.ui.GREEN, self.ui.RED)
def do_use(self, args):
if PAYLOADS.has_key(args):
self.load_module(args, PAYLOADS.get(args))
else:
self.ui.notify.error("Load payload %s error." % args)
def complete_use(self, text, line, begidx, endidx):
if text:
return [i.lower() for i in PAYLOADS.keys() if i.lower().startswith(text.lower())]
return [i.lower() for i in PAYLOADS.keys()]
def do_use(self, args):
if PAYLOADS.has_key(args):
self.load_module(args, PAYLOADS.get(args))
else:
self.ui.notify.error("Load payload %s error." % args)
