metasploit-framework/tools/update_payload_cached_sizes.rb

30 lines
830 B
Ruby
Raw Normal View History

#!/usr/bin/env ruby
#
# This script updates the CachedSize constants in payload modules
#
msfbase = __FILE__
while File.symlink?(msfbase)
msfbase = File.expand_path(File.readlink(msfbase), File.dirname(msfbase))
end
$:.unshift(File.expand_path(File.join(File.dirname(msfbase), '..', 'lib')))
require 'msfenv'
$:.unshift(ENV['MSF_LOCAL_LIB']) if ENV['MSF_LOCAL_LIB']
require 'rex'
require 'msf/ui'
require 'msf/base'
require 'msf/util/payload_cached_size'
# Initialize the simplified framework instance.
framework = Msf::Simple::Framework.create('DisableDatabase' => true)
framework.payloads.each_module do |name, mod|
next if Msf::Util::PayloadCachedSize.is_cached_size_accurate?(mod)
$stdout.puts "[*] Updating the CacheSize for #{mod.file_path}..."
Msf::Util::PayloadCachedSize.update_module_cached_size(mod)
end