rapid7/ruby_smb

module RubySMB
  module SMB1
    module Packet
      module Trans2
        # This class represents a generic SMB1 Trans2 Response Packet as defined in
        # [2.2.4.46.2 Response](https://msdn.microsoft.com/en-us/library/ee441550.aspx)
        class Response < RubySMB::GenericPacket
          COMMAND = RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2

          # The {RubySMB::SMB1::ParameterBlock} specific to this packet type.
          class ParameterBlock < RubySMB::SMB1::ParameterBlock
            uint16  :total_parameter_count, label: 'Total Parameter Count(bytes)'
            uint16  :total_data_count,        label: 'Total Data Count(bytes)'
            uint16  :reserved,                label: 'Reserved Space',                 initial_value: 0x00
            uint16  :parameter_count,         label: 'Parameter Count(bytes)',         initial_value: -> { parent.data_block.trans2_parameters.length }
            uint16  :parameter_offset,        label: 'Parameter Offset',               initial_value: -> { parent.data_block.trans2_parameters.abs_offset }
            uint16  :parameter_displacement,  label: 'Parameter Displacement'
            uint16  :data_count,              label: 'Data Count(bytes)',              initial_value: -> { parent.data_block.trans2_data.length }
            uint16  :data_offset,             label: 'Data Offset',                    initial_value: -> { parent.data_block.trans2_data.abs_offset }
            uint16  :data_displacement,       label: 'Data Displacement'
            uint8   :setup_count,             label: 'Setup Count',                    initial_value: -> { setup.length }
            uint8   :reserved2,               label: 'Reserved Space',                 initial_value: 0x00

            array :setup, type: :uint16, initial_length: :setup_count
          end

          require 'ruby_smb/smb1/packet/trans2/find_first2_response'
          require 'ruby_smb/smb1/packet/trans2/find_next2_response'
          require 'ruby_smb/smb1/packet/trans2/open2_response'
          require 'ruby_smb/smb1/packet/trans2/query_file_information_response'
          require 'ruby_smb/smb1/packet/trans2/query_path_information_response'
          require 'ruby_smb/smb1/packet/trans2/set_file_information_response'
          require 'ruby_smb/smb1/packet/trans2/query_fs_information_response'

          smb_header        :smb_header
          parameter_block   :parameter_block
          choice            :data_block, selection: -> { parameter_block.setup.first || :default } do
            open2_response_data_block                  Subcommands::OPEN2
            find_first2_response_data_block            Subcommands::FIND_FIRST2
            find_next2_response_data_block             Subcommands::FIND_NEXT2
            query_file_information_response_data_block Subcommands::QUERY_FILE_INFORMATION
            query_path_information_response_data_block Subcommands::QUERY_PATH_INFORMATION
            set_file_information_response_data_block   Subcommands::SET_FILE_INFORMATION
            query_fs_information_response_data_block   Subcommands::QUERY_FS_INFORMATION
            data_block                                 :default
          end

          def initialize_instance
            super
            smb_header.flags.reply = 1
          end
        end
      end
    end
  end
end