Contents

module RubySMB
  module SMB1
    module Packet
      module Trans
        # Extends the {RubySMB::SMB1::DataBlock} to include padding methods
        # that all Trans DataBlocks will need to handle proper byte alignment.
        class DataBlock < RubySMB::SMB1::DataBlock
          # Controls whether the padding fields will be used
          # @!attribute [rw] enable_padding
          #   @return [Boolean]
          attr_accessor :enable_padding

          def initialize_instance
            super
            @enable_padding = true
          end

          private

          # Determines the correct length for the padding in front of
          # trans_parameters. It should always force a 4-byte alignment.
          def pad1_length
            if enable_padding
              offset = if respond_to?(:name)
                         (name.abs_offset + 1) % 4
                       else
                         (byte_count.abs_offset + 2) % 4
                       end
              (4 - offset) % 4
            else
              0
            end
          end

          # Determines the correct length for the padding in front of
          # trans_data. It should always force a 4-byte alignment.
          def pad2_length
            if enable_padding
              offset = (trans_parameters.abs_offset + trans_parameters.length) % 4
              (4 - offset) % 4
            else
              0
            end
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
ruby_smb-0.0.21	lib/ruby_smb/smb1/packet/trans/data_block.rb