Add references to stream when possible

Author: jvazquez-r7

lib/rex/java/serialization/model/class_desc.rb

@@ -28,6 +28,7 @@ def decode(io)
             end
 
             self.description = content
+            stream.add_reference(self) unless stream.nil?
             self
           end
 

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

@@ -11,6 +11,7 @@ class LongUtf < Utf
           # @return [self] if deserialization succeeds
           # @return [nil] if deserialization doesn't succeed
           def decode(io)
+            stream.add_reference(self) unless stream.nil?
             raw_length = io.read(8)
             if raw_length.nil? || raw_length.length != 8
               raise ::RuntimeError, 'Failed to unserialize LongUtf'

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

@@ -31,6 +31,7 @@ def initialize(stream = nil)
           # @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)

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

@@ -70,6 +70,7 @@ def encode
             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|

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

@@ -27,6 +27,7 @@ def initialize(stream = nil)
           # @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

lib/rex/java/serialization/model/new_object.rb

@@ -27,6 +27,7 @@ def initialize(stream = nil)
           # @raise [RuntimeError] if deserialization doesn't succeed
           def decode(io)
             self.class_desc = ClassDesc.decode(io, stream)
+            stream.add_reference(self) unless stream.nil?
             if class_desc.description.class == Rex::Java::Serialization::Model::NewClassDesc
               self.class_data = decode_class_data(io, class_desc)
             end

lib/rex/java/serialization/model/stream.rb

@@ -16,12 +16,18 @@ class Stream < Element
           # @!attribute contents
           #   @return [Array] The stream's contents
           attr_accessor :contents
+          attr_accessor :references
 
           def initialize(stream = nil)
             super(stream)
             self.magic = STREAM_MAGIC
             self.version = STREAM_VERSION
             self.contents = []
+            self.references = []
+          end
+
+          def add_reference(ref)
+            self.references.push(ref)
           end
 
           # Deserializes a Java::Serialization::Model::Stream

lib/rex/java/serialization/model/utf.rb

@@ -25,6 +25,7 @@ def initialize(stream = nil, contents = '')
           # @return [self] if deserialization succeeds
           # @raise [RuntimeError] if deserialization doesn't succeed
           def decode(io)
+            stream.add_reference(self) unless stream.nil?
             raw_length = io.read(2)
             if raw_length.nil? || raw_length.length != 2
               raise ::RuntimeError, 'Failed to unserialize Utf'