Skip to content

Commit edb92d2

Browse files
committed
✨ add do_set
1 parent 9eafe7a commit edb92d2

File tree

2 files changed

+11
-1
lines changed

2 files changed

+11
-1
lines changed

lib/ExploitOptions.py

+6
Original file line numberDiff line numberDiff line change
@@ -12,3 +12,9 @@ def get_option(self, name):
1212
if option.name == name:
1313
return option.value
1414
return None
15+
16+
def set_option(self, name, value):
17+
for idx, option in enumerate(self.options):
18+
if option.name == name:
19+
option.value = value
20+
self.options[idx] = option

lib/pocket.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,11 @@ def do_list(self, args):
2525
self.poutput("list all module")
2626

2727
def do_set(self, args):
28-
pass
28+
if not self.module_instance:
29+
raise ModuleNotUseException()
30+
31+
[arg, value] = args.split(" ")
32+
self.module_instance.options.set_option(arg, value)
2933

3034
def do_use(self, module_name):
3135
module_file = "{ROOT}/modules/{MODULE}.py".format(ROOT=ROOT_PATH, MODULE=module_name)

0 commit comments

Comments
 (0)