metasploit-framework/lib/msf/core/encoder.rb

636 lines
15 KiB
Ruby
Raw Normal View History

# -*- coding: binary -*-
require 'msf/core'
module Msf
###
#
# This class is used to track the state of a single encoding operation
# from start to finish.
#
###
class EncoderState
#
# Initializes a new encoder state, optionally with a key.
#
def initialize(key = nil)
@orig_buf = nil
@buf = nil
reset(key)
end
#
# Reset the encoder state by initializing the encoded buffer to an empty
# string.
#
def reset(key = nil)
init_key(key)
self.encoded = ''
end
#
# Set the initial encoding key.
#
def init_key(key)
self.key = key
self.orig_key = key
end
#
# Set the raw buffer and the original buffer if one has not been set.
#
def buf=(buf)
@orig_buf = buf if (@orig_buf == nil or @buf == nil)
@buf = buf
end
attr_accessor :key # :nodoc:
attr_accessor :orig_key # :nodoc:
attr_reader :buf # :nodoc:
attr_reader :orig_buf # :nodoc:
attr_accessor :encoded # :nodoc:
attr_accessor :context # :nodoc:
attr_accessor :badchars # :nodoc:
# A boolean that indicates whether context encoding is enabled
attr_accessor :context_encoding
# The address that contains they key on the target machine
attr_accessor :context_address
# Decoder settings
attr_accessor :decoder_key_offset, :decoder_key_size, :decoder_key_pack # :nodoc:
attr_accessor :decoder_stub # :nodoc:
end
###
#
# This class is the base class that all encoders inherit from.
#
###
class Encoder < Module
#
# The type set that encoders can fall within. This classifies the type of
# output generated by the encoder in terms of the general character set
# that is used as well as other potential attributes.
#
module Type
#
# 'A' - 'Z', '0' - '9'
#
AlphanumUpper = "alpha_upper"
#
# 'a' - 'z', 'A' - 'Z', '0' - '9'
#
AlphanumMixed = "alpha_mixed"
#
# Unicode-safe 'A' - 'Z', '0' - '9'
#
AlphanumUnicodeUpper = "alpha_unicode_upper"
#
# Unicode-safe 'a' - 'z', 'A' - 'Z', '0' - '9'
#
AlphanumUnicodeMixed = "alpha_unicode_mixed"
#
# toupper/tolower safe ascii - not 'a' - 'z', 'A' - 'Z'
#
NonAlpha = "non_alpha"
#
# tolower safe ascii - not 'A' - 'Z' (more flexable than nonalpha)
#
NonUpper = "non_upper"
#
# tolower safe ascii UTF8-safe (<= 0x7f only)
#
NonUpperUtf8Safe = "non_upper_utf8_safe"
#
# tolower safe underscore safe for CVE-2012-2329 - PHP CGI apache_request_headers bof
#
NonUpperUnderscoreSafe = "non_upper_underscore"
#
# May result in the generation of any characters
#
Unspecified = "unspecified"
#
# The raw payload passed to the encoder will be the same as the encoded
# payload
#
Raw = "raw"
#
# Special Single Static Bit encoder
#
SingleStaticBit = "single_static_bit"
#
# Special printf(1) via PHP magic_quotes Command Encoder
#
PrintfPHPMagicQuotes = "printf_php_mq"
end
#
# Initializes an encoder module instance using the supplied information
# hash.
#
def initialize(info)
super({
'Platform' => '' # All platforms by default
}.update(info))
end
##
#
# Encoder information accessors that can be overriden
# by derived classes
#
##
#
# Returns MODULE_ENCODER to indicate that this is an encoder module.
#
def self.type
return MODULE_ENCODER
end
#
# Returns MODULE_ENCODER to indicate that this is an encoder module.
#
def type
return MODULE_ENCODER
end
#
# Returns the type or types of encoders that this specific module
# classifies as. If there is more than one type, the values should be
# separated by whitespace.
#
def encoder_type
module_info['EncoderType'] || Type::Unspecified
end
#
# Returns the decoder stub to use based on the supplied state.
#
def decoder_stub(state)
return decoder_hash['Stub'] || ''
end
#
# Returns the offset to the key associated with the decoder stub.
#
def decoder_key_offset
return decoder_hash['KeyOffset']
end
#
# Returns the size of the key, in bytes.
#
def decoder_key_size
return decoder_hash['KeySize']
end
#
# Returns the size of each logical encoding block, in bytes. This
# is typically the same as decoder_key_size.
#
def decoder_block_size
return decoder_hash['BlockSize']
end
#
# Returns the byte-packing character that should be used to encode
# the key.
#
def decoder_key_pack
return decoder_hash['KeyPack'] || 'V'
end
#
# Returns the module's decoder hash or an empty hash.
#
def decoder_hash
module_info['Decoder'] || {}
end
##
#
# Encoding
#
##
#
# This method generates an encoded version of the supplied buffer in buf
# using the bad characters as guides. On success, an encoded and
# functional version of the supplied buffer will be returned. Otherwise,
# an exception will be thrown if an error is encountered during the
# encoding process.
#
def encode(buf, badchars = nil, state = nil, platform = nil)
# Configure platform hints if necessary
init_platform(platform) if platform
# Initialize an empty set of bad characters
badchars = '' if (!badchars)
# Initialize the encoding state and key as necessary
if (state == nil)
state = EncoderState.new
end
# Prepend data to the buffer as necessary
buf = prepend_buf + buf
init_state(state)
# Save the buffer in the encoding state
state.badchars = badchars || ''
state.buf = buf
# If this encoder is key-based and we don't already have a key, find one
if ((decoder_key_size) and
(state.key == nil))
# Find a key that doesn't contain and wont generate any bad
# characters
state.init_key(obtain_key(buf, badchars, state))
if (state.key == nil)
raise NoKeyError, "A key could not be found for the #{self.name} encoder.", caller
end
end
# Reset the encoded buffer at this point since it may have been changed
# while finding a key.
state.encoded = ''
# Call encode_begin to do any encoder specific pre-processing
encode_begin(state)
# Perform the actual encoding operation with the determined state
do_encode(state)
# Call encoded_end to do any encoder specific post-processing
encode_end(state)
# Return the encoded buffer to the caller
return state.encoded
end
#
# Performs the actual encoding operation after the encoder state has been
# initialized and is ready to go.
#
def do_encode(state)
# Copy the decoder stub since we may need to modify it
stub = decoder_stub(state).dup
if (state.key != nil and state.decoder_key_offset)
# Substitute the decoder key in the copy of the decoder stub with the
# one that we found
real_key = state.key
# If we're using context encoding, the actual value we use for
# substitution is the context address, not the key we use for
# encoding
real_key = state.context_address if (state.context_encoding)
stub[state.decoder_key_offset,state.decoder_key_size] = [ real_key.to_i ].pack(state.decoder_key_pack)
else
stub = encode_finalize_stub(state, stub)
end
# Walk the buffer encoding each block along the way
offset = 0
if (decoder_block_size)
while (offset < state.buf.length)
block = state.buf[offset, decoder_block_size]
state.encoded += encode_block(state,
block + ("\x00" * (decoder_block_size - block.length)))
offset += decoder_block_size
end
else
state.encoded = encode_block(state, state.buf)
end
# Prefix the decoder stub to the encoded buffer
state.encoded = stub + state.encoded
# Last but not least, do one last badchar pass to see if the stub +
# encoded payload leads to any bad char issues...
if ((badchar_idx = has_badchars?(state.encoded, state.badchars)) != nil)
raise BadcharError.new(state.encoded, badchar_idx, stub.length, state.encoded[badchar_idx]),
"The #{self.name} encoder failed to encode without bad characters.",
caller
end
return true
end
##
#
# Buffer management
#
##
#
# Returns a string that should be prepended to the encoded version of the
# buffer before returning it to callers.
#
def prepend_buf
return ''
end
##
#
# Pre-processing, post-processing, and block encoding stubs
#
##
#
# Called when encoding is about to start immediately after the encoding
# state has been initialized.
#
def encode_begin(state)
return nil
end
#
# This callback allows a derived class to finalize a stub after a key have
# been selected. The finalized stub should be returned.
#
def encode_finalize_stub(state, stub)
stub
end
#
# Called after encoding has completed.
#
def encode_end(state)
return nil
end
#
# Called once for each block being encoded based on the attributes of the
# decoder.
#
def encode_block(state, block)
return block
end
#
# Provides the encoder with an opportunity to return the native format (as
# in the format the code will be in when it executes on the target). In
# general, the same buffer is returned to the caller. However, for things
# like unicode, the buffer is unicod encoded and then returned.
#
def to_native(buf)
buf
end
protected
#
# Initializes the encoding state supplied as an argument to the attributes
# that have been defined for this decoder stub, such as key offset, size,
# and pack.
#
def init_state(state)
# Update the state with default decoder information
state.decoder_key_offset = decoder_key_offset
state.decoder_key_size = decoder_key_size
state.decoder_key_pack = decoder_key_pack
state.decoder_stub = nil
# Restore the original buffer in case it was modified.
state.buf = state.orig_buf
end
#
# This provides a hook method for platform specific
# processing prior to the rest of encode() running
#
def init_platform(platform)
end
#
# Obtains the key to use during encoding. If context encoding is enabled,
# special steps are taken. Otherwise, the derived class is given an
# opportunity to find the key.
#
def obtain_key(buf, badchars, state)
if datastore['EnableContextEncoding']
return find_context_key(buf, badchars, state)
else
return find_key(buf, badchars, state)
end
end
#
# This method finds a compatible key for the supplied buffer based also on
# the supplied bad characters list. This is meant to make encoders more
# reliable and less prone to bad character failure by doing a fairly
# complete key search before giving up on an encoder.
#
def find_key(buf, badchars, state)
# Otherwise, we use the traditional method
key_bytes = [ ]
cur_key = [ ]
bad_keys = find_bad_keys(buf, badchars)
found = false
allset = [*(0..255)]
# Keep chugging until we find something...right
while (!found)
# Scan each byte position
0.upto(decoder_key_size - 1) { |index|
# Subtract the bad and leave the good
good_keys = allset - bad_keys[index].keys
# Was there anything left for this index?
if (good_keys.length == 0)
# Not much we can do about this :(
return nil
end
# Set the appropriate key byte
key_bytes[index] = good_keys[ rand(good_keys.length) ]
}
# Assume that we're going to rock this...
found = true
# Scan each byte and see what we've got going on to make sure
# no funny business is happening
key_bytes.each { |byte|
if (badchars.index(byte.chr) != nil)
found = false
end
}
found = find_key_verify(buf, key_bytes, badchars) if found
end
# Do we have all the key bytes accounted for?
if (key_bytes.length != decoder_key_size)
return nil
end
return key_bytes_to_integer(key_bytes)
end
#
# Parses a context information file in an effort to find a compatible key
#
def find_context_key(buf, badchars, state)
# Make sure our context information file is sane
if File.exists?(datastore['ContextInformationFile']) == false
raise NoKeyError, "A context information file must specified when using context encoding", caller
end
# Holds the address and key that we ultimately find
address = nil
key = nil
# Now, parse records from the information file searching for entries
# that are compatible with our bad character set
File.open(datastore['ContextInformationFile']) { |f|
begin
# Keep looping until we hit an EOF error or we find
# a compatible key
while key.nil?
# Read in the header
type, chunk_base_address, size = f.read(9).unpack('CNN')
offset = 0
# Read in the blob of data that will act as our key state
data = f.read(size)
# If the address doesn't contain bad characters, check to see
# the data itself will result in bad characters being generated
while data.length > decoder_key_size
# Extract the current set of key bytes
key_bytes = []
# My ruby is rusty
data[0, decoder_key_size].each_byte { |b|
key_bytes << b
}
# If the key verifies correctly, we need to check it's address
if find_key_verify(buf, key_bytes, badchars)
address = chunk_base_address + offset
# Pack it to byte form so that we can check each byte for
# bad characters
address_bytes = integer_to_key_bytes(address)
# Scan each byte and see what we've got going on to make sure
# no funny business is happening with the address
invalid_key = false
address_bytes.each { |byte|
if badchars.index(byte.chr)
invalid_key = true
end
}
if invalid_key == false
key = key_bytes_to_integer(key_bytes)
break
end
end
# If it didn't verify, then we need to proceed
data = data[1, data.length - 1]
offset += 1
end
end
rescue EOFError
end
}
# If the key is nil after all is said and done, then we failed to locate
# a compatible context-sensitive key
if key.nil?
raise NoKeyError, "No context key could be located in #{datastore['ContextInformationFile']}", caller
# Otherwise, we successfully determined the key, now we need to update
# the encoding state with our context address and set context encoding
# to true so that the encoders know to use it
else
ilog("#{refname}: Successfully found context address @ #{"%.8x" % address} using key #{"%.8x" % key}")
state.context_address = address
state.context_encoding = true
end
return key
end
#
# Returns the list of bad keys associated with this encoder.
#
def find_bad_keys(buf, badchars)
return Array.new(decoder_key_size) { Hash.new }
end
#
# Returns the index of any bad characters found in the supplied buffer.
#
def has_badchars?(buf, badchars)
badchars.each_byte { |badchar|
idx = buf.index(badchar.chr)
if (idx != nil)
return idx
end
}
return nil
end
#
# Convert individual key bytes into a single integer based on the
# decoder's key size and packing requirements
#
def key_bytes_to_integer(key_bytes)
return key_bytes_to_buffer(key_bytes).unpack(decoder_key_pack)[0]
end
#
# Convert individual key bytes into a byte buffer
#
def key_bytes_to_buffer(key_bytes)
return key_bytes.pack('C*')[0, decoder_key_size]
end
#
# Convert an integer into the individual key bytes based on the
# decoder's key size and packing requirements
#
def integer_to_key_bytes(integer)
return [ integer.to_i ].pack(decoder_key_pack).unpack('C*')[0, decoder_key_size]
end
#
# Determines if the key selected by find_key is usable
#
def find_key_verify(buf, key_bytes, badchars)
true
end
end
end
require 'msf/core/encoder/xor'
require 'msf/core/encoder/xor_additive_feedback'
require 'msf/core/encoder/alphanum'
require 'msf/core/encoder/nonalpha'
require 'msf/core/encoder/nonupper'