module RubySMB
  module SMB1
    module Packet
      module Trans
      # A SMB1 SMB_COM_TRANSACTION Response Packet as defined in
      # [2.2.4.33.2 Response](https://msdn.microsoft.com/en-us/library/ee442061.aspx)
        class Response < RubySMB::GenericPacket
          # A SMB1 Parameter Block
          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.trans_parameters.length }
            uint16        :parameter_offset,        label: 'Parameter Offset',               initial_value: -> { parent.data_block.trans_parameters.abs_offset }
            uint16        :parameter_displacement,  label: 'Parameter Displacement'
            uint16        :data_count,              label: 'Data Count(bytes)',              initial_value: -> { parent.data_block.trans_data.length }
            uint16        :data_offset,             label: 'Data Offset',                    initial_value: -> { parent.data_block.trans_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: 0
          end

          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
          class DataBlock < RubySMB::SMB1::Packet::Trans::DataBlock
            string :pad1,               length: -> { pad1_length }
            string :trans_parameters,   label: 'Trans Parameters'
            string :pad2,               length: -> { pad2_length }
            string :trans_data,         label: 'Trans Data'
          end

          smb_header        :smb_header
          parameter_block   :parameter_block
          data_block        :data_block

          def initialize_instance
            super
            smb_header.command = RubySMB::SMB1::Commands::SMB_COM_TRANSACTION
            smb_header.flags.reply = 1
          end
        end
      end
    end
  end
end