bwatters
31 lines
794 B
Ruby
31 lines
794 B
Ruby
# -*- coding => binary -*-
|
|
|
|
#
|
|
module Msf::Payload::Custom
|
|
|
|
def stage_payload(_opts = {})
|
|
return nil if datastore['SHELLCODE_FILE'].blank?
|
|
|
|
File.binread(datastore['SHELLCODE_FILE'])
|
|
end
|
|
|
|
def setup_handler
|
|
if datastore['SHELLCODE_FILE'].blank?
|
|
fail_with(Msf::Module::Failure::BadConfig, "No SHELLCODE_FILE provided")
|
|
end
|
|
begin
|
|
# read the file before we start the handler to make sure that it is valid
|
|
test = File.binread(datastore['SHELLCODE_FILE'])
|
|
rescue => e
|
|
print_error("Unable to read #{datastore['SHELLCODE_FILE']}:")
|
|
elog("Unable to read #{datastore['SHELLCODE_FILE']}:", error: e)
|
|
fail_with(Msf::Module::Failure::BadConfig, "Bad SHELLCODE_FILE provided")
|
|
end
|
|
super
|
|
end
|
|
|
|
def read_stage_size?
|
|
true
|
|
end
|
|
end
|