rex 2.0.4 → 2.0.5

data/lib/rex/java/serialization/model/long_utf.rb

@@ -0,0 +1,48 @@
+# -*- coding: binary -*-
+
+module Rex
+  module Java
+    module Serialization
+      module Model
+        # This class provides a Long Utf string representation
+        class LongUtf < Utf
+
+          # Deserializes a Rex::Java::Serialization::Model::LongUtf
+          #
+          # @param io [IO] the io to read from
+          # @return [self] if deserialization succeeds
+          # @return [nil] if deserialization doesn't succeed
+          def decode(io)
+            raw_length = io.read(8)
+            if raw_length.nil? || raw_length.length != 8
+              raise ::RuntimeError, 'Failed to unserialize LongUtf'
+            end
+            self.length = raw_length.unpack('Q>')[0]
+
+            if length == 0
+              self.contents = ''
+            else
+              self.contents = io.read(length)
+              if contents.nil? || contents.length != length
+                raise ::RuntimeError, 'Failed to unserialize LongUtf'
+              end
+            end
+
+            self
+          end
+
+          # Serializes the Rex::Java::Serialization::Model::LongUtf
+          #
+          # @return [String]
+          def encode
+            encoded = [length].pack('Q>')
+            encoded << contents
+
+            encoded
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rex/java/serialization/model/new_array.rb

@@ -0,0 +1,225 @@
+# -*- coding: binary -*-
+
+module Rex
+  module Java
+    module Serialization
+      module Model
+        # This class provides a NewArray (Java Array) representation
+        class NewArray < Element
+
+          include Rex::Java::Serialization::Model::Contents
+
+          # @!attribute array_description
+          #   @return [Java::Serialization::Model::ClassDesc] The description of the array
+          attr_accessor :array_description
+          # @!attribute type
+          #   @return [String] The type of the array values
+          attr_accessor :type
+          # @!attribute values
+          #   @return [Array] The contents of the java array
+          attr_accessor :values
+
+          # @param stream [Rex::Java::Serialization::Model::Stream] the stream where it belongs to
+          def initialize(stream = nil)
+            super(stream)
+            self.array_description = nil
+            self.type = ''
+            self.values = []
+          end
+
+          # Deserializes a Rex::Java::Serialization::Model::NewArray
+          #
+          # @param io [IO] the io to read from
+          # @return [self] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode(io)
+            self.array_description = ClassDesc.decode(io, stream)
+            stream.add_reference(self) unless stream.nil?
+            self.type = array_type
+
+            values_length = decode_values_length(io)
+
+            values_length.times do
+              value = decode_value(io)
+              self.values << value
+            end
+
+            self
+          end
+
+          # Serializes the Rex::Java::Serialization::Model::NewArray
+          #
+          # @return [String] if serialization succeeds
+          # @raise [RuntimeError] if serialization doesn't succeed
+          def encode
+            unless array_description.class == ClassDesc
+              raise ::RuntimeError, 'Failed to serialize NewArray'
+            end
+
+            encoded = ''
+            encoded << array_description.encode
+
+            encoded << [values.length].pack('N')
+
+            values.each do |value|
+              encoded << encode_value(value)
+            end
+
+            encoded
+          end
+
+          # Creates a print-friendly string representation
+          #
+          # @return [String]
+          def to_s
+            str = "#{type}, "
+            values_data = values.collect {|v| "#{v}"}
+            str << "#{values_data}"
+          end
+
+          private
+
+          # Deserializes the NewArray length
+          #
+          # @param io [IO] the io to read from
+          # @return [Integer] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode_values_length(io)
+            values_length = io.read(4)
+            if values_length.nil? || values_length.length != 4
+              raise ::RuntimeError, 'Failed to unserialize NewArray'
+            end
+
+            values_length.unpack('N')[0]
+          end
+
+          # Extracts the NewArray data type
+          #
+          # @return [String]
+          # @raise [RuntimeError] if the NewArray description isn't valid
+          # @raise [RuntimeError] if the NewArray type isn't supported
+          def array_type
+            if array_description.nil?
+              raise ::RuntimeError, 'Empty NewArray description'
+            end
+
+            unless array_description.class == ClassDesc
+              raise ::RuntimeError, 'Unsupported NewArray description class'
+            end
+
+            desc = array_description.description
+
+            unless desc.class_name.contents[0] == '[' # Array
+              raise ::RuntimeError, 'Unsupported NewArray description'
+            end
+
+            decoded_type = desc.class_name.contents[1]
+            if PRIMITIVE_TYPE_CODES.keys.include?(decoded_type)
+              return PRIMITIVE_TYPE_CODES[decoded_type]
+            elsif decoded_type == 'L' # L : Object
+              return desc.class_name.contents[2..desc.class_name.contents.index(';')] # Object class
+            else
+              raise ::RuntimeError, 'Unsupported NewArray Type'
+            end
+          end
+
+          # Deserializes a NewArray value
+          #
+          # @param io [IO] the io to read from
+          # @return [Fixnum, Float] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization fails
+          def decode_value(io)
+            value = nil
+
+            case type
+            when 'byte'
+              value = io.read(1)
+              raise ::RuntimeError, 'Failed to deserialize NewArray value' if value.nil?
+              value = value.unpack('c')[0]
+            when 'char'
+              value = io.read(2)
+              unless value && value.length == 2
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('s>')[0]
+            when 'double'
+              value = io.read(8)
+              unless value && value.length == 8
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('G')[0]
+            when 'float'
+              value = io.read(4)
+              unless value && value.length == 4
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('g')[0]
+            when 'int'
+              value = io.read(4)
+              unless value && value.length == 4
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('l>')[0]
+            when 'long'
+              value = io.read(8)
+              unless value && value.length == 8
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('q>')[0]
+            when 'short'
+              value = io.read(2)
+              unless value && value.length == 2
+                raise ::RuntimeError, 'Failed to deserialize NewArray value'
+              end
+              value = value.unpack('s>')[0]
+            when 'boolean'
+              value = io.read(1)
+              raise ::RuntimeError, 'Failed to deserialize NewArray value' if value.nil?
+              value = value.unpack('c')[0]
+            else # object
+              value = decode_content(io, stream)
+            end
+
+            value
+          end
+
+          # Serializes an NewArray value
+          #
+          # @param value [Fixnum] the value to serialize
+          # @param value [Float] the value to serialize
+          # @return [String] the serialized value
+          # @raise [RuntimeError] if serialization fails
+          def encode_value(value)
+            res = ''
+
+            case type
+            when 'byte'
+              res = [value].pack('c')
+            when 'char'
+              res = [value].pack('s>')
+            when 'double'
+              res = [value].pack('G')
+            when 'float'
+              res = [value].pack('g')
+            when 'int'
+              res = [value].pack('l>')
+            when 'long'
+              res = [value].pack('q>')
+            when 'short'
+              res = [value].pack('s>')
+            when 'boolean'
+              res = [value].pack('c')
+            when Element
+              res = value.encode
+            else # object
+              res = encode_content(value)
+            end
+
+            res
+          end
+
+        end
+      end
+    end
+  end
+end

data/lib/rex/java/serialization/model/new_class_desc.rb

@@ -0,0 +1,155 @@
+# -*- coding: binary -*-
+
+module Rex
+  module Java
+    module Serialization
+      module Model
+        # This class provides a newClassDesc representation
+        class NewClassDesc < Element
+
+          include Rex::Java::Serialization
+
+          # @!attribute class_name
+          #   @return [Rex::Java::Serialization::Model::Utf] The name of the class
+          attr_accessor :class_name
+          # @!attribute name
+          #   @return [Integer] The java class serial version
+          attr_accessor :serial_version
+          # @!attribute flags
+          #   @return [Integer] The java class flags
+          attr_accessor :flags
+          # @!attribute fields
+          #   @return [Array] The java class fields
+          attr_accessor :fields
+          # @!attribute fields
+          #   @return [Rex::Java::Serialization::Model::Annotation] The java class annotations
+          attr_accessor :class_annotation
+          # @!attribute super_class
+          #   @return [Rex::Java::Serialization::Model::ClassDesc] The java class superclass description
+          attr_accessor :super_class
+
+          # @param stream [Rex::Java::Serialization::Model::Stream] the stream where it belongs to
+          def initialize(stream = nil)
+            super(stream)
+            self.class_name = nil
+            self.serial_version = 0
+            self.flags = 0
+            self.fields = []
+            self.class_annotation = nil
+            self.super_class = nil
+          end
+
+          # Deserializes a Rex::Java::Serialization::Model::ClassDescription
+          #
+          # @param io [IO] the io to read from
+          # @return [self] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode(io)
+            self.class_name = Utf.decode(io, stream)
+            self.serial_version = decode_serial_version(io)
+            stream.add_reference(self) unless stream.nil?
+            self.flags = decode_flags(io)
+            fields_length = decode_fields_length(io)
+            fields_length.times do
+              field = Field.decode(io, stream)
+              self.fields << field
+            end
+
+            self.class_annotation = Annotation.decode(io, stream)
+            self.super_class = ClassDesc.decode(io, stream)
+
+            self
+          end
+
+          # Serializes the Rex::Java::Serialization::Model::ClassDescription
+          #
+          # @return [String] if serialization succeeds
+          # @raise [RuntimeError] if serialization doesn't succeed
+          def encode
+            unless class_name.class == Rex::Java::Serialization::Model::Utf &&
+                    class_annotation.class == Rex::Java::Serialization::Model::Annotation &&
+                    super_class.class == Rex::Java::Serialization::Model::ClassDesc
+              raise ::RuntimeError, 'Filed to serialize NewClassDesc'
+            end
+            encoded = ''
+            encoded << class_name.encode
+            encoded << [serial_version].pack('Q>')
+            stream.add_reference(self) unless stream.nil?
+            encoded << [flags].pack('C')
+            encoded << [fields.length].pack('n')
+            fields.each do |field|
+              encoded << field.encode
+            end
+            encoded << class_annotation.encode
+            encoded << super_class.encode
+
+            encoded
+          end
+
+          # Creates a print-friendly string representation
+          #
+          # @return [String]
+          def to_s
+            str = "#{class_name}, [ "
+            fields_str = []
+            fields.each do |field|
+              fields_str << field.to_s
+            end
+            str << "#{fields_str.join(', ')} ]"
+
+            case super_class.description
+            when NewClassDesc
+              str << ", @super_class: #{super_class.description.class_name.to_s}"
+            when Reference
+              str << ", @super_class: #{super_class.description.to_s}"
+            end
+
+            str
+          end
+
+          private
+
+          # Deserializes a class serial version
+          #
+          # @param io [IO] the io to read from
+          # @return [Integer] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode_serial_version(io)
+            raw_serial = io.read(8)
+            if raw_serial.nil? || raw_serial.length != 8
+              raise ::RuntimeError, 'Failed to unserialize ClassDescription'
+            end
+
+            raw_serial.unpack('Q>')[0]
+          end
+
+          # Deserializes a class flags
+          #
+          # @param io [IO] the io to read from
+          # @return [Integer] if deserialization is possible
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode_flags(io)
+            raw_flags = io.read(1)
+            raise ::RuntimeError, 'Failed to unserialize ClassDescription' if raw_flags.nil?
+
+            raw_flags.unpack('C')[0]
+          end
+
+          # Deserializes a class fields length
+          #
+          # @param io [IO] the io to read from
+          # @return [Integer] if deserialization is possible
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode_fields_length(io)
+            fields_length = io.read(2)
+            if fields_length.nil? || fields_length.length != 2
+              raise ::RuntimeError, 'Failed to unserialize ClassDescription'
+            end
+
+            fields_length.unpack('n')[0]
+          end
+        end
+      end
+    end
+  end
+end

data/lib/rex/java/serialization/model/new_enum.rb

@@ -0,0 +1,79 @@
+# -*- coding: binary -*-
+
+module Rex
+  module Java
+    module Serialization
+      module Model
+        # This class provides a NewEnum (Java Enum) representation
+        class NewEnum < Element
+
+          include Rex::Java::Serialization::Model::Contents
+
+          # @!attribute enum_description
+          #   @return [Rex::Java::Serialization::Model::ClassDescription] The description of the enum
+          attr_accessor :enum_description
+          # @!attribute constant_name
+          #   @return [Rex::Java::Serialization::Model::Utf] The constant value in the Java Enum
+          attr_accessor :constant_name
+
+          # @param stream [Rex::Java::Serialization::Model::Stream] the stream where it belongs to
+          def initialize(stream = nil)
+            super(stream)
+            self.enum_description = nil
+            self.constant_name = nil
+          end
+
+          # Deserializes a Rex::Java::Serialization::Model::NewEnum
+          #
+          # @param io [IO] the io to read from
+          # @return [self] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succeed
+          def decode(io)
+            self.enum_description = ClassDesc.decode(io, stream)
+            stream.add_reference(self) unless stream.nil?
+            self.constant_name = decode_constant_name(io)
+
+            self
+          end
+
+          # Serializes the Rex::Java::Serialization::Model::NewEnum
+          #
+          # @return [String] if serialization succeeds
+          # @raise [RuntimeError] if serialization doesn't succeed
+          def encode
+            unless enum_description.class == ClassDesc &&
+                    constant_name.class == Utf
+              raise ::RuntimeError, 'Failed to serialize EnumDescription'
+            end
+
+            encoded = ''
+            encoded << enum_description.encode
+            encoded << encode_content(constant_name)
+            encoded
+          end
+
+          # Creates a print-friendly string representation
+          #
+          # @return [String]
+          def to_s
+            constant_name.to_s
+          end
+
+          private
+
+          # Deserializes the NewEnum constant name
+          #
+          # @param io [IO] the io to read from
+          # @return [Rex::Java::Serialization::Model::Utf] if deserialization succeeds
+          # @raise [RuntimeError] if deserialization doesn't succed
+          def decode_constant_name(io)
+            content = decode_content(io, stream)
+            raise ::RuntimeError, 'Failed to unserialize NewEnum' unless content.class == Rex::Java::Serialization::Model::Utf
+
+            content
+          end
+        end
+      end
+    end
+  end
+end