Skip to content

Commit 02e81d1

Browse files
committed
Add Enum-type options for external modules
1 parent ab36b5d commit 02e81d1

File tree

1 file changed

+7
-2
lines changed
  • lib/msf/core/modules/external

1 file changed

+7
-2
lines changed

lib/msf/core/modules/external/shim.rb

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,8 +45,13 @@ def self.mod_meta_common(mod, meta = {}, drop_rhost: true)
4545
end
4646

4747
meta[:options] = options.map do |n, o|
48-
"Opt#{o['type'].camelize}.new(#{n.dump},
49-
[#{o['required']}, #{o['description'].dump}, #{o['default'].inspect}])"
48+
if o['values']
49+
"Opt#{o['type'].camelize}.new(#{n.dump},
50+
[#{o['required']}, #{o['description'].dump}, #{o['default'].inspect}, #{o['values'].inspect}])"
51+
else
52+
"Opt#{o['type'].camelize}.new(#{n.dump},
53+
[#{o['required']}, #{o['description'].dump}, #{o['default'].inspect}])"
54+
end
5055
end.join(",\n ")
5156
meta
5257
end

0 commit comments

Comments
 (0)