ruby_smb 0.0.18 → 0.0.19

data/lib/ruby_smb/smb1/packet/trans2/find_first2_request.rb

@@ -0,0 +1,77 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        # A Trans2 FIND_FIRST2 Request Packet as defined in
+        # [2.2.6.2.1](https://msdn.microsoft.com/en-us/library/ee441987.aspx)
+        class FindFirst2Request < RubySMB::GenericPacket
+          class ParameterBlock < RubySMB::SMB1::Packet::Trans2::Request::ParameterBlock
+          end
+
+          # The Trans2 Parameter Block for this particular Subcommand
+          class Trans2Parameters < BinData::Record
+            endian :little
+            smb_file_attributes :search_attributes, label: 'File Attributes'
+            uint16              :search_count,      label: 'Search Count'
+
+            struct :flags do
+              bit3  :reserved,    label: 'Reserved Space'
+              bit1  :backup,      label: 'With Backup Intent'
+              bit1  :continue,    label: 'Continue From Last'
+              bit1  :resume_keys, label: 'Return Resume Keys'
+              bit1  :close_eos,   label: 'Close at End of Search'
+              bit1  :close,       label: 'Close Search After This Request'
+
+              bit8  :reserved2,   label: 'Reserved Space'
+            end
+
+            uint16    :information_level, label: 'Information Level'
+            uint32    :storage_type,      label: 'Search Storage type'
+
+            choice :filename, :copy_on_change => true, selection: -> { self.smb_header.flags2.unicode } do
+              stringz16 1, label: 'FileName'
+              stringz   0, label: 'FileName'
+            end
+
+            # Returns the length of the Trans2Parameters struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The Trans2 Data Blcok for this particular Subcommand
+          class Trans2Data < BinData::Record
+            smb_gea_list :extended_attribute_list, label: 'Get Extended Attribute List',
+              onlyif: -> { parent.trans2_parameters.information_level == FindInformationLevel::SMB_INFO_QUERY_EAS_FROM_LIST}
+
+            # Returns the length of the Trans2Data struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
+          class DataBlock < RubySMB::SMB1::Packet::Trans2::DataBlock
+            uint8              :name,               label: 'Name', initial_value: 0x00
+            string             :pad1,               length: -> { pad1_length }
+            trans2_parameters  :trans2_parameters,  label: 'Trans2 Parameters'
+            string             :pad2,               length: -> { pad2_length }
+            trans2_data        :trans2_data,        label: 'Trans2 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_TRANSACTION2
+            parameter_block.setup << RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_FIRST2
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/find_first2_response.rb

@@ -0,0 +1,87 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        # This class represents an SMB1 Trans2 FIND_FIRST2 Response Packet as defined in
+        # [2.2.6.2.2 Response](https://msdn.microsoft.com/en-us/library/ee441704.aspx)
+        class FindFirst2Response < RubySMB::GenericPacket
+          class ParameterBlock < RubySMB::SMB1::Packet::Trans2::Response::ParameterBlock
+          end
+
+          # The Trans2 Parameter Block for this particular Subcommand
+          class Trans2Parameters < BinData::Record
+            endian  :little
+
+            uint16  :sid,               label: 'Search ID'
+            uint16  :search_count,      label: 'Search Count'
+            uint16  :eos,               label: 'End of Search'
+            uint16  :ea_error_offset,   label: 'Offset to EA Error'
+            uint16  :last_name_offset,  label: 'Last Name Offset'
+
+            # Returns the length of the Trans2Parameters struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The Trans2 Data Blcok for this particular Subcommand
+          class Trans2Data < BinData::Record
+            rest :buffer, label: 'Results Buffer'
+
+            # Returns the length of the Trans2Data struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
+          class DataBlock < RubySMB::SMB1::Packet::Trans2::DataBlock
+            uint8              :name,               label: 'Name', initial_value: 0x00
+            string             :pad1,               length: -> { pad1_length }
+            trans2_parameters  :trans2_parameters,  label: 'Trans2 Parameters'
+            string             :pad2,               length: -> { pad2_length }
+            trans2_data        :trans2_data,        label: 'Trans2 Data', length: 0
+          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_TRANSACTION2
+            parameter_block.setup << RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_FIRST2
+            smb_header.flags.reply = 1
+          end
+
+          # Returns the File Information in an array of appropriate
+          # structs for the given FileInformationClass. Pulled out of
+          # the string buffer.
+          #
+          # @param klass [Class] the FileInformationClass class to read the data as
+          # @return [array<BinData::Record>] An array of structs holding the requested information
+          def results(klass, unicode:)
+            information_classes = []
+            blob = data_block.trans2_data.buffer.to_binary_s.dup
+            until blob.empty?
+              length = blob[0, 4].unpack('V').first
+
+              data = if length.zero?
+                       blob.slice!(0, blob.length)
+                     else
+                       blob.slice!(0, length)
+                     end
+
+              file_info = klass.new
+              file_info.unicode = unicode
+              information_classes << file_info.read(data)
+            end
+            information_classes
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/find_information_level.rb

@@ -0,0 +1,32 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        # FIND information levels are used in TRANS2_FIND_FIRST2 (section 2.2.6.2) and
+        # TRANS2_FIND_NEXT2 (section 2.2.6.3) subcommand requests to indicate the level
+        # of information that a server MUST respond with for each file matching the
+        # request's search criteria.
+        module FindInformationLevel
+          # Return creation, access, and last write timestamps, size and file attributes along with the file name.
+          SMB_INFO_STANDARD                 = 0x0001
+          # Return the SMB_INFO_STANDARD data along with the size of a file's extended attributes (EAs).
+          SMB_INFO_QUERY_EA_SIZE            = 0x0002
+          # Return the SMB_INFO_QUERY_EA_SIZE data along with a specific list of a file's EAs. The requested EAs are provided in the Trans2_Data block of the request.
+          SMB_INFO_QUERY_EAS_FROM_LIST      = 0x0003
+          # Return 64-bit format versions of: creation, access, last write, and last attribute change timestamps; size. In addition, return extended file attributes and file name.
+          SMB_FIND_FILE_DIRECTORY_INFO      = 0x0101
+          # Returns the SMB_FIND_FILE_DIRECTORY_INFO data along with the size of a file's EAs.
+          SMB_FIND_FILE_FULL_DIRECTORY_INFO = 0x0102
+          # Returns the name(s) of the file(s).
+          SMB_FIND_FILE_NAMES_INFO          = 0x0103
+          # Returns a combination of the data from SMB_FIND_FILE_FULL_DIRECTORY_INFO and SMB_FIND_FILE_NAMES_INFO.
+          SMB_FIND_FILE_BOTH_DIRECTORY_INFO = 0x0104
+
+
+          require 'ruby_smb/smb1/packet/trans2/find_information_level/find_file_full_directory_info'
+
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/find_information_level/find_file_full_directory_info.rb

@@ -0,0 +1,45 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        module FindInformationLevel
+          # The SMB_FIND_FILE_FULL_DIRECTORY_INFO class as defined in
+          # [2.2.8.1.5 SMB_FIND_FILE_FULL_DIRECTORY_INFO](https://msdn.microsoft.com/en-us/library/ff470042.aspx)
+          class FindFileFullDirectoryInfo < BinData::Record
+            CLASS_LEVEL = FindInformationLevel::SMB_FIND_FILE_FULL_DIRECTORY_INFO
+
+            endian :little
+
+            uint32                  :next_offset,         label: 'Next Entry Offset'
+            uint32                  :file_index,          label: 'File Index'
+            file_time               :create_time,         label: 'Create Time'
+            file_time               :last_access,         label: 'Last Accessed Time'
+            file_time               :last_write,          label: 'Last Write Time'
+            file_time               :last_change,         label: 'Last Attribute Change Time'
+            uint64                  :end_of_file,         label: 'End of File'
+            uint64                  :allocation_size,     label: 'Allocated Size'
+            smb_ext_file_attributes :ext_file_attributes, label: 'Extended File Attributes'
+            uint32                  :file_name_length,    label: 'File Name Length', initial_value: -> { file_name.do_num_bytes }
+            uint32                  :ea_size,             label: 'Extended Attributes Size'
+
+            choice :file_name, :copy_on_change => true, selection: -> { unicode } do
+              string16 true,  label: 'File Name', read_length: -> { file_name_length }
+              stringz  false, label: 'File Name', read_length: -> { file_name_length }
+            end
+
+            # Set unicode encoding for filename
+            # @!attribute [rw] unicode
+            #   @return [Boolean]
+            attr_accessor :unicode
+
+            def initialize_instance
+              super
+              @unicode = false
+            end
+
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/find_next2_request.rb

@@ -0,0 +1,77 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        # A Trans2 FIND_NEXT2 Request Packet as defined in
+        # [2.2.6.3.1 Request](https://msdn.microsoft.com/en-us/library/ee441844.aspx)
+        class FindNext2Request < RubySMB::GenericPacket
+          class ParameterBlock < RubySMB::SMB1::Packet::Trans2::Request::ParameterBlock
+          end
+
+          # The Trans2 Parameter Block for this particular Subcommand
+          class Trans2Parameters < BinData::Record
+            endian  :little
+
+            uint16  :sid,               label: 'Search ID'
+            uint16  :search_count,      label: 'Search Count'
+            uint16  :information_level, label: 'Information Level'
+            uint32  :resume_key,        label: 'Resume Key'
+
+            struct :flags do
+              bit3  :reserved,    label: 'Reserved Space'
+              bit1  :backup,      label: 'With Backup Intent'
+              bit1  :continue,    label: 'Continue From Last'
+              bit1  :resume_keys, label: 'Return Resume Keys'
+              bit1  :close_eos,   label: 'Close at End of Search'
+              bit1  :close,       label: 'Close Search After This Request'
+
+              bit8  :reserved2,   label: 'Reserved Space'
+            end
+
+            choice :filename, :copy_on_change => true, selection: -> { @obj.parent.parent.parent.smb_header.flags2.unicode } do
+              stringz16 1, label: 'FileName'
+              stringz   0, label: 'FileName'
+            end
+
+            # Returns the length of the Trans2Parameters struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The Trans2 Data Blcok for this particular Subcommand
+          class Trans2Data < BinData::Record
+            smb_gea_list :extended_attribute_list, label: 'Get Extended Attribute List',
+              onlyif: -> { parent.trans2_parameters.information_level == FindInformationLevel::SMB_INFO_QUERY_EAS_FROM_LIST}
+
+            # Returns the length of the Trans2Data struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
+          class DataBlock < RubySMB::SMB1::Packet::Trans2::DataBlock
+            uint8              :name,               label: 'Name', initial_value: 0x00
+            string             :pad1,               length: -> { pad1_length }
+            trans2_parameters  :trans2_parameters,  label: 'Trans2 Parameters'
+            string             :pad2,               length: -> { pad2_length }
+            trans2_data        :trans2_data,        label: 'Trans2 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_TRANSACTION2
+            parameter_block.setup << RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_NEXT2
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/find_next2_response.rb

@@ -0,0 +1,86 @@
+module RubySMB
+  module SMB1
+    module Packet
+      module Trans2
+        # This class represents an SMB1 Trans2 FIND_NEXT2 Response Packet as defined in
+        # [2.2.6.3.2 Response](https://msdn.microsoft.com/en-us/library/ee441871.aspx)
+        class FindNext2Response < RubySMB::GenericPacket
+          class ParameterBlock < RubySMB::SMB1::Packet::Trans2::Response::ParameterBlock
+          end
+
+          # The Trans2 Parameter Block for this particular Subcommand
+          class Trans2Parameters < BinData::Record
+            endian  :little
+
+            uint16  :search_count,      label: 'Search Count'
+            uint16  :eos,               label: 'End of Search'
+            uint16  :ea_error_offset,   label: 'Offset to EA Error'
+            uint16  :last_name_offset,  label: 'Last Name Offset'
+
+            # Returns the length of the Trans2Parameters struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The Trans2 Data Blcok for this particular Subcommand
+          class Trans2Data < BinData::Record
+            rest :buffer, label: 'Results Buffer'
+
+            # Returns the length of the Trans2Data struct
+            # in number of bytes
+            def length
+              do_num_bytes
+            end
+          end
+
+          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
+          class DataBlock < RubySMB::SMB1::Packet::Trans2::DataBlock
+            uint8              :name,               label: 'Name', initial_value: 0x00
+            string             :pad1,               length: -> { pad1_length }
+            trans2_parameters  :trans2_parameters,  label: 'Trans2 Parameters'
+            string             :pad2,               length: -> { pad2_length }
+            trans2_data        :trans2_data,        label: 'Trans2 Data', length: 0
+          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_TRANSACTION2
+            parameter_block.setup << RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_NEXT2
+            smb_header.flags.reply = 1
+          end
+
+          # Returns the File Information in an array of appropriate
+          # structs for the given FileInformationClass. Pulled out of
+          # the string buffer.
+          #
+          # @param klass [Class] the FileInformationClass class to read the data as
+          # @return [array<BinData::Record>] An array of structs holding the requested information
+          def results(klass, unicode:)
+            information_classes = []
+            blob = data_block.trans2_data.buffer.to_binary_s.dup
+            until blob.empty?
+              length = blob[0, 4].unpack('V').first
+
+              data = if length.zero?
+                       blob.slice!(0, blob.length)
+                     else
+                       blob.slice!(0, length)
+                     end
+
+              file_info = klass.new
+              file_info.unicode = unicode
+              information_classes << file_info.read(data)
+            end
+            information_classes
+          end
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb1/packet/trans2/open2_request.rb

@@ -2,14 +2,13 @@ module RubySMB
   module SMB1
     module Packet
       module Trans2
-
         # A Trans2 OPEN2 Request Packet as defined in
         # [2.2.6.1.1 Request](https://msdn.microsoft.com/en-us/library/ee441733.aspx)
         class Open2Request < RubySMB::GenericPacket
-
           class ParameterBlock < RubySMB::SMB1::Packet::Trans2::Request::ParameterBlock
           end
 
+          # The Trans2 Parameter Block for this particular Subcommand
           class Trans2Parameters < BinData::Record
             endian :little
             open2_flags         :flags,           label: 'Flags'
@@ -19,27 +18,29 @@ module RubySMB
             utime               :creation_time,   label: 'Creation Time'
             open2_open_mode     :open_mode,       label: 'Open Mode'
             uint32              :allocation_size, label: 'Allocation Size'
-            array               :reserved2,        initial_length: 5 do
+            array               :reserved2, initial_length: 5 do
               uint16 initial_value: 0x0000
             end
-            stringz             :filename,        label: 'Filename'
+            stringz :filename, label: 'Filename'
 
             # Returns the length of the Trans2Parameters struct
             # in number of bytes
             def length
-              self.do_num_bytes
+              do_num_bytes
             end
           end
 
+          # The Trans2 Data Blcok for this particular Subcommand
           class Trans2Data < BinData::Record
-            smb_fea_list  :extended_attribute_list, label: 'Extended Attribute List'
+            smb_fea_list :extended_attribute_list, label: 'Extended Attribute List'
           end
 
+          # The {RubySMB::SMB1::DataBlock} specific to this packet type.
           class DataBlock < RubySMB::SMB1::Packet::Trans2::DataBlock
-            uint8              :name,               label: 'Name',              initial_value: 0x00
-            string             :pad1,               length: lambda { pad1_length }
+            uint8              :name,               label: 'Name', initial_value: 0x00
+            string             :pad1,               length: -> { pad1_length }
             trans2_parameters  :trans2_parameters,  label: 'Trans2 Parameters'
-            string             :pad2,               length: lambda { pad2_length }
+            string             :pad2,               length: -> { pad2_length }
             trans2_data        :trans2_data,        label: 'Trans2 Data'
           end
 
@@ -47,7 +48,6 @@ module RubySMB
           parameter_block   :parameter_block
           data_block        :data_block
 
-
           def initialize_instance
             super
             smb_header.command = RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2