Fix encoded payloads test

This commit is contained in:
Dean Welch
2024-01-16 13:31:51 +00:00
parent 2cf045d3c4
commit 1c4258fd1e
5 changed files with 9 additions and 4 deletions
+1 -1
View File
@@ -237,7 +237,7 @@ class EncodedPayload
begin
eout = self.encoder.encode(eout, reqs['BadChars'], nil, pinst.platform)
rescue EncodingError
rescue EncodingError => e
wlog("#{err_start}: Encoder #{encoder.refname} failed: #{$!}", 'core', LEV_1)
dlog("#{err_start}: Call stack\n#{$@.join("\n")}", 'core', LEV_3)
next_encoder = true
@@ -17,6 +17,11 @@ RSpec.describe Msf::EncodedPayload do
module_type: 'encoder',
modules_path: modules_path,
)
allow(framework.encoders).to receive(:each_module_ranked)
.and_yield('x86/shikata_ga_nai', framework.encoders['x86/shikata_ga_nai'])
.and_yield('x86/xor_dynamic', framework.encoders['x86/xor_dynamic'])
.and_yield('x86/call4_dword_xor', framework.encoders['x86/call4_dword_xor'])
.and_yield('generic/none', framework.encoders['generic/none'])
end
let(:ancestor_reference_names) {
+1 -1
View File
@@ -250,7 +250,7 @@ RSpec.describe Msf::ModuleSet do
end
end
describe 'create' do
describe '#create' do
let(:module_refname) { 'module_refname' }
let(:framework) { instance_double(Msf::Framework) }
let(:module_manager) { instance_double(Msf::ModuleManager) }
+1 -1
View File
@@ -40,7 +40,7 @@ class MetasploitModule < Msf::Post
session_type = session.type
module_results = []
framework.modules.post.module_refnames.each do | refname |
framework.modules.post.module_refnames.each do |refname|
next unless refname.start_with?('test/') && refname != self.refname
mod = framework.modules.create(refname)
+1 -1
View File
@@ -23,7 +23,7 @@ require 'rex'
# Initialize the simplified framework instance.
$framework = Msf::Simple::Framework.create('DisableDatabase' => true)
# TODO: this is weird, merging module sets together for different module types could lead to unforseen issues
# XXX: this is weird, merging module sets together for different module types could lead to unforseen issues
all_modules = $framework.exploits.merge($framework.auxiliary)
all_ports = {}