Skip to content

Commit 9cbc7e4

Browse files
committed
Fix suggested stuff
hdm#2
1 parent c71428b commit 9cbc7e4

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

lib/msf/core/payload/stager.rb

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def initialize(info={})
1717
Msf::OptBool.new("EnableStageEncoding", [ false, "Encode the second stage payload", false ]),
1818
Msf::OptString.new("StageEncoder", [ false, "Encoder to use if EnableStageEncoding is set", nil ]),
1919
Msf::OptString.new("StageEncoderSaveRegisters", [ false, "Additional registers to preserve in the staged payload if EnableStageEncoding is set", "" ]),
20-
Msf::OptBool.new("FallbackToNoStageEncoding", [ false, "If encoders choosen in StageEncoder are not compatible to stage encoding fallback to no encoding otherwise fallback to automatic selected one", true ])
20+
Msf::OptBool.new("StageEncodingNoFallBack", [ false, "If encoders choosen in StageEncoder are not compatible to stage encoding fallback to no encoding otherwise fallback to automatic selected one", true ])
2121
], Msf::Payload::Stager)
2222

2323
end
@@ -101,7 +101,7 @@ def stage_over_connection?
101101
# @return [Boolean]
102102
def encode_stage?
103103
# Convert to string in case it hasn't been normalized
104-
datastore['EnableStageEncoding'] == "true" || !datastore["StageEncoder"].to_s.empty?
104+
!!(datastore['EnableStageEncoding'].to_s == "true" || datastore["StageEncoder"].to_s.length > 0)
105105
end
106106

107107
#
@@ -226,21 +226,21 @@ def encode_stage(stg)
226226
encode_stage_preserved_registers
227227
).strip
228228

229-
(stage_enc_mod || [nil]).each do |encoder|
229+
(stage_enc_mod || [nil]).each do |encoder_refname_from_user|
230230
# Generate an encoded version of the stage. We tell the encoding system
231231
# to save certain registers to ensure that it does not get clobbered.
232232
encp = Msf::EncodedPayload.create(
233233
self,
234234
'Raw' => stg,
235-
'Encoder' => encoder,
235+
'Encoder' => encoder_refname_from_user,
236236
'EncoderOptions' => { 'SaveRegisters' => saved_registers },
237237
'ForceSaveRegisters' => true,
238238
'ForceEncode' => true)
239239
if (encp.encoder == nil)
240-
print_warning("Encoder #{encoder} did not succeed")
241-
if !datastore['FallbackToNoStageEncoding']
240+
print_warning("Encoder #{encoder_refname_from_user} did not succeed")
241+
if !datastore['StageEncodingNoFallBack']
242242
print_warning("Fallback to automatic StageEncoder selection")
243-
encoder = nil
243+
encoder_refname_from_user = nil
244244
redo
245245
else
246246
print_warning("Fallback to no encoder")

0 commit comments

Comments
 (0)