156 lines
5.5 KiB
Ruby
156 lines
5.5 KiB
Ruby
# -*- coding: binary -*-
|
|
|
|
module Rex
|
|
module Java
|
|
module Serialization
|
|
module Model
|
|
module Contents
|
|
include Rex::Java::Serialization
|
|
|
|
# Deserializes a content
|
|
#
|
|
# @param io [IO] the io to read from
|
|
# @return [Rex::Java::Serialization::Model::Element] if deserialization succeeds
|
|
# @raise [RuntimeError] if deserialization doesn't succeed or unsupported content
|
|
def decode_content(io, stream)
|
|
opcode = io.read(1)
|
|
raise ::RuntimeError, 'Failed to unserialize content' if opcode.nil?
|
|
opcode = opcode.unpack('C')[0]
|
|
content = nil
|
|
|
|
case opcode
|
|
when TC_BLOCKDATA
|
|
content = BlockData.decode(io, stream)
|
|
when TC_BLOCKDATALONG
|
|
content = BlockDataLong.decode(io, stream)
|
|
when TC_ENDBLOCKDATA
|
|
content = EndBlockData.decode(io, stream)
|
|
when TC_OBJECT
|
|
content = NewObject.decode(io, stream)
|
|
when TC_CLASS
|
|
content = ClassDesc.decode(io, stream)
|
|
when TC_ARRAY
|
|
content = NewArray.decode(io, stream)
|
|
when TC_STRING
|
|
content = Utf.decode(io, stream)
|
|
stream.add_reference(content) unless stream.nil?
|
|
when TC_LONGSTRING
|
|
content = LongUtf.decode(io, stream)
|
|
stream.add_reference(content) unless stream.nil?
|
|
when TC_ENUM
|
|
content = NewEnum.decode(io, stream)
|
|
when TC_CLASSDESC
|
|
content = NewClassDesc.decode(io, stream)
|
|
when TC_PROXYCLASSDESC
|
|
raise ::RuntimeError, 'Failed to unserialize unsupported TC_PROXYCLASSDESC content'
|
|
when TC_REFERENCE
|
|
content = Reference.decode(io, stream)
|
|
when TC_NULL
|
|
content = NullReference.decode(io, stream)
|
|
when TC_EXCEPTION
|
|
raise ::RuntimeError, 'Failed to unserialize unsupported TC_EXCEPTION content'
|
|
when TC_RESET
|
|
content = Reset.decode(io, stream)
|
|
else
|
|
raise ::RuntimeError, 'Failed to unserialize content'
|
|
end
|
|
|
|
content
|
|
end
|
|
|
|
# Serializes a content
|
|
#
|
|
# @param content [Rex::Java::Serialization::Model::Element] the content to serialize
|
|
# @return [String] if serialization succeeds
|
|
# @raise [RuntimeError] if serialization doesn't succeed
|
|
def encode_content(content)
|
|
encoded = ''
|
|
|
|
case content
|
|
when BlockData
|
|
encoded << [TC_BLOCKDATA].pack('C')
|
|
when BlockDataLong
|
|
encoded << [TC_BLOCKDATALONG].pack('C')
|
|
when EndBlockData
|
|
encoded << [TC_ENDBLOCKDATA].pack('C')
|
|
when NewObject
|
|
encoded << [TC_OBJECT].pack('C')
|
|
when ClassDesc
|
|
encoded << [TC_CLASS].pack('C')
|
|
when NewArray
|
|
encoded << [TC_ARRAY].pack('C')
|
|
when Utf
|
|
encoded << [TC_STRING].pack('C')
|
|
when LongUtf
|
|
encoded << [TC_LONGSTRING].pack('C')
|
|
when NewEnum
|
|
encoded << [TC_ENUM].pack('C')
|
|
when NewClassDesc
|
|
encoded << [TC_CLASSDESC].pack('C')
|
|
when NullReference
|
|
encoded << [TC_NULL].pack('C')
|
|
when Reset
|
|
encoded << [TC_RESET].pack('C')
|
|
when Reference
|
|
encoded << [TC_REFERENCE].pack('C')
|
|
else
|
|
raise ::RuntimeError, 'Failed to serialize content'
|
|
end
|
|
|
|
encoded << content.encode
|
|
encoded
|
|
end
|
|
|
|
# Creates a print-friendly string representation
|
|
#
|
|
# @param content [Rex::Java::Serialization::Model::Element] the content to print
|
|
# @return [String]
|
|
def print_content(content)
|
|
str = ''
|
|
|
|
case content
|
|
when BlockData
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when BlockDataLong
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when EndBlockData
|
|
str << "#{print_class(content)}"
|
|
when NewObject
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when ClassDesc
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when NewArray
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when Utf
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when LongUtf
|
|
str << "#{print_class(content)} { #{content.to_s} } "
|
|
when NewEnum
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when NewClassDesc
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
when NullReference
|
|
str << "#{print_class(content)}"
|
|
when Reset
|
|
str << "#{print_class(content)}"
|
|
when Reference
|
|
str << "#{print_class(content)} { #{content.to_s} }"
|
|
else
|
|
raise ::RuntimeError, 'Failed to serialize content'
|
|
end
|
|
|
|
str
|
|
end
|
|
|
|
# Creates a print-friendly string representation of the content class
|
|
#
|
|
# @param content [Rex::Java::Serialization::Model::Element] the content
|
|
# @return [String]
|
|
def print_class(content)
|
|
content.class.name.split('::').last
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end |