75 lines
1.8 KiB
Ruby
75 lines
1.8 KiB
Ruby
require 'msf/base'
|
|
|
|
module Msf
|
|
module Simple
|
|
|
|
###
|
|
#
|
|
# Payload
|
|
# -------
|
|
#
|
|
# Simple payload wrapper class for performing generation.
|
|
#
|
|
###
|
|
class Payload
|
|
|
|
#
|
|
# Generate a payload with the mad skillz. The payload can be generated in
|
|
# a number of ways.
|
|
#
|
|
# opts can have:
|
|
#
|
|
# Encoder => A encoder module instance.
|
|
# Badchars => A string of bad characters.
|
|
# Format => The format to represent the data as: ruby, perl, c, raw
|
|
# Options => A hash of options to set.
|
|
# OptionStr => A string of options in VAR=VAL form separated by
|
|
# whitespace.
|
|
# NoComment => Disables prepention of a comment
|
|
#
|
|
# raises:
|
|
#
|
|
# BadcharError => If the supplied encoder fails to encode the payload
|
|
# NoKeyError => No valid encoder key could be found
|
|
# ArgumentParseError => Options were supplied improperly
|
|
#
|
|
def self.generate(payload, opts)
|
|
# If options were supplied, import them into the payload's
|
|
# datastore
|
|
if (opts['Option'])
|
|
payload.datastore.import_options_from_hash(opts['Options'])
|
|
elsif (opts['OptionStr'])
|
|
payload.datastore.import_options_from_s(opts['OptionStr'])
|
|
end
|
|
|
|
# Generate the payload
|
|
buf = payload.generate
|
|
|
|
# If an encoder was specified, encode the generated payload
|
|
if (opts['Encoder'])
|
|
buf = opts['Encoder'].encode(buf, opts['Badchars'])
|
|
end
|
|
|
|
fmt = opts['Format'] || 'raw'
|
|
|
|
# Serialize the generated payload to some sort of format
|
|
buf = Buffer.transform(buf, fmt)
|
|
|
|
# Prepend a comment
|
|
if (fmt != 'raw' and opts['NoComment'] != true)
|
|
((ds = payload.datastore.to_s) and ds.length > 0) ? ds += "\n" : ds = ''
|
|
|
|
buf = Buffer.comment(
|
|
"#{payload.refname} - http://www.metasploit.com\n" +
|
|
"#{ds}" +
|
|
((opts['Encoder']) ? "Encoder=" + opts['Encoder'].refname + "\n" : ''), fmt) + buf
|
|
end
|
|
|
|
return buf
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|