ruby_smb 0.0.18 → 0.0.19

data/lib/ruby_smb/smb2/packet/echo_request.rb

@@ -1,7 +1,6 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 Echo Request Packet as defined in
       # [2.2.28 SMB2 ECHO Request](https://msdn.microsoft.com/en-us/library/cc246540.aspx)
       class EchoRequest < RubySMB::GenericPacket
@@ -14,8 +13,7 @@ module RubySMB
           super
           smb2_header.command = RubySMB::SMB2::Commands::ECHO
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/echo_response.rb

@@ -1,7 +1,6 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 Echo response Packet as defined in
       # [2.2.29 SMB2 ECHO Response](https://msdn.microsoft.com/en-us/library/cc246541.aspx)
       class EchoResponse < RubySMB::GenericPacket
@@ -15,8 +14,7 @@ module RubySMB
           smb2_header.command = RubySMB::SMB2::Commands::ECHO
           smb2_header.flags.reply = 1
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/error_packet.rb

@@ -1,9 +1,7 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 Error packet for when an incomplete response comes back
-
       class ErrorPacket < RubySMB::GenericPacket
         endian       :little
         smb2_header  :smb2_header
@@ -12,4 +10,4 @@ module RubySMB
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/ioctl_request.rb

@@ -0,0 +1,54 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Ioctl Request Packet as defined in
+      # [2.2.31 SMB2 IOCTL Request](https://msdn.microsoft.com/en-us/library/cc246545.aspx)
+      class IoctlRequest < RubySMB::GenericPacket
+        endian :little
+
+        smb2_header   :smb2_header
+        uint16        :structure_size,      label: 'Structure Size',      initial_value: 57
+        uint16        :reserved1,           label: 'Reserved Space'
+        uint32        :ctl_code,            label: 'Control Code'
+        smb2_fileid   :file_id,             label: 'File Id'
+        uint32        :input_offset,        label: 'Input Offset',        initial_value: -> { calc_input_offset }
+        uint32        :input_count,         label: 'Input Count',         initial_value: -> { buffer.do_num_bytes }
+        uint32        :max_input_response,  label: 'Max Input Response'
+        uint32        :output_offset,       label: 'Output Offset',       initial_value: -> { input_offset + output_count }
+        uint32        :output_count,        label: 'Output Count'
+        uint32        :max_output_response, label: 'Max Output response', initial_value: 65536
+
+        struct :flags do
+          bit7  :reserved1, label: 'Reserved Space'
+          bit1  :is_fsctl,  label: 'FSCTL not IOCTL'
+
+          bit8  :reserved2, label: 'Reserved Space'
+          bit8  :reserved3, label: 'Reserved Space'
+          bit8  :reserved4, label: 'Reserved Space'
+        end
+
+        uint32  :reserved2, label: 'Reserved Space'
+        string  :buffer,    label: 'Input Buffer', read_length: -> { input_count + output_count }
+
+        def initialize_instance
+          super
+          smb2_header.command = RubySMB::SMB2::Commands::IOCTL
+        end
+
+        # Calculates the value for the input_offset field.
+        # If the input buffer is empty then this should be set to 0,
+        # otherwise it should return the absolute offset of the input buffer.
+        #
+        # @return [Integer] the value to store in #input_offset
+        def calc_input_offset
+          if input_count.zero?
+            0
+          else
+            buffer.abs_offset
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/ioctl_response.rb

@@ -0,0 +1,39 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Ioctl Response Packet as defined in
+      # [2.2.32 SMB2 IOCTL Response](https://msdn.microsoft.com/en-us/library/cc246548.aspx)
+      class IoctlResponse < RubySMB::GenericPacket
+        endian :little
+
+        smb2_header   :smb2_header
+        uint16        :structure_size,      label: 'Structure Size',      initial_value: 49
+        uint16        :reserved1,           label: 'Reserved Space'
+        uint32        :ctl_code,            label: 'Control Code'
+        smb2_fileid   :file_id,             label: 'File Id'
+        uint32        :input_offset,        label: 'Input Offset'
+        uint32        :input_count,         label: 'Input Count'
+        uint32        :output_offset,       label: 'Output Offset'
+        uint32        :output_count,        label: 'Output Count'
+        uint32        :flags,               label: 'Flags'
+        uint32        :reserved2,           label: 'Reserved Space'
+        string        :buffer,              label: 'Input Buffer',        read_length: -> { input_count + output_count }
+
+        def initialize_instance
+          super
+          smb2_header.flags.reply = 1
+          smb2_header.command     = RubySMB::SMB2::Commands::IOCTL
+        end
+
+        def input_data
+          to_binary_s[input_offset, input_count]
+        end
+
+        def output_data
+          to_binary_s[output_offset, output_count]
+        end
+
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/logoff_request.rb

@@ -1,7 +1,6 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 LOGOFF Request Packet as defined in
       # [2.2.7 SMB2 LOGOFF Request](https://msdn.microsoft.com/en-us/library/cc246565.aspx)
       class LogoffRequest < RubySMB::GenericPacket
@@ -10,13 +9,11 @@ module RubySMB
         uint16       :structure_size, label: 'Structure Size', initial_value: 4
         uint16       :reserved,       label: 'Reserved Space'
 
-
         def initialize_instance
           super
           smb2_header.command = RubySMB::SMB2::Commands::LOGOFF
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/logoff_response.rb

@@ -1,7 +1,6 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 LOGOFF Response Packet as defined in
       # [2.2.8 SMB2 LOGOFF Response](https://msdn.microsoft.com/en-us/library/cc246566.aspx)
       class LogoffResponse < RubySMB::GenericPacket
@@ -10,14 +9,12 @@ module RubySMB
         uint16       :structure_size, label: 'Structure Size', initial_value: 4
         uint16       :reserved,       label: 'Reserved Space'
 
-
         def initialize_instance
           super
           smb2_header.command = RubySMB::SMB2::Commands::LOGOFF
           smb2_header.flags.reply = 1
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/negotiate_request.rb

@@ -25,7 +25,7 @@ module RubySMB
         # @param [Fixnum] the numeric code for the dialect you wish to add
         # @return [Array<Fixnum>] the array of all currently selected dialects
         def add_dialect(dialect)
-          return ArgumentError, 'Must be a number' unless dialect.is_a? Fixnum
+          return ArgumentError, 'Must be a number' unless dialect.is_a? Integer
           self.dialect_count += 1
           dialects << dialect
         end

data/lib/ruby_smb/smb2/packet/negotiate_response.rb

@@ -6,7 +6,7 @@ module RubySMB
       class NegotiateResponse < RubySMB::GenericPacket
         endian              :little
         smb2_header         :smb2_header
-        uint16              :structure_size,   label: 'Structure Size', initial_value: 65
+        uint16              :structure_size, label: 'Structure Size', initial_value: 65
         smb2_security_mode  :security_mode
         uint16              :dialect_revision,          label: 'Dialect Revision'
         uint16              :negotiate_context_count,   label: 'Negotiate Context Count',      initial_value: 0

data/lib/ruby_smb/smb2/packet/query_directory_request.rb

@@ -0,0 +1,35 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Query Directory Request Packet as defined in
+      # [2.2.33 SMB2 QUERY_DIRECTORY Request](https://msdn.microsoft.com/en-us/library/cc246551.aspx)
+      class QueryDirectoryRequest < RubySMB::GenericPacket
+        endian       :little
+        smb2_header  :smb2_header
+        uint16       :structure_size,          label: 'Structure Size', initial_value: 33
+        uint8        :file_information_class,  label: 'File Information Class'
+
+        struct :flags do
+          bit3  :reserved2,       label: 'Reserved Space'
+          bit1  :reopen,          label: 'Reopen Search'
+          bit1  :reserved,        label: 'Reserved Space'
+          bit1  :index_specified, label: 'Start at Specified Index'
+          bit1  :return_single,   label: 'Return Single Entry'
+          bit1  :restart_scans,   label: 'Restart Enumeration from Start'
+        end
+
+        uint32        :file_index,    label: 'File Index'
+        smb2_fileid   :file_id,       label: 'File ID'
+        uint16        :name_offset,   label: 'File Name Offset',      initial_value: -> { name.abs_offset }
+        uint16        :name_length,   label: 'File Name Length',      initial_value: -> { name.do_num_bytes }
+        uint32        :output_length, label: 'Output Buffer Length'
+        string16      :name,          label: 'Name/Search Pattern'
+
+        def initialize_instance
+          super
+          smb2_header.command = RubySMB::SMB2::Commands::QUERY_DIRECTORY
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/query_directory_response.rb

@@ -0,0 +1,45 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Query Directory Response Packet as defined in
+      # [2.2.34 SMB2 QUERY_DIRECTORY Response](https://msdn.microsoft.com/en-us/library/cc246552.aspx)
+      class QueryDirectoryResponse < RubySMB::GenericPacket
+        endian       :little
+        smb2_header  :smb2_header
+        uint16       :structure_size,  label: 'Structure Size',       initial_value: 9
+        uint16       :buffer_offset,   label: 'Output Buffer Offset', initial_value: -> { buffer.abs_offset }
+        uint32       :buffer_length,   label: 'Output Buffer Length', initial_value: -> { buffer.do_num_bytes }
+        string       :buffer,          read_length: -> { buffer_length }
+
+        def initialize_instance
+          super
+          smb2_header.command = RubySMB::SMB2::Commands::QUERY_DIRECTORY
+          smb2_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)
+          information_classes = []
+          blob = 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
+
+            information_classes << klass.read(data)
+          end
+          information_classes
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/read_request.rb

@@ -0,0 +1,30 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Read Request Packet as defined in
+      # [2.2.19 SMB2 READ Request](https://msdn.microsoft.com/en-us/library/cc246527.aspx)
+      class ReadRequest < RubySMB::GenericPacket
+        endian :little
+
+        smb2_header           :smb2_header
+        uint16                :structure_size,        label: 'Structure Size',  initial_value: 49
+        uint8                 :padding,               label: 'Padding',         initial_value: 0x50
+        uint8                 :flags,                 label: 'Flags'
+        uint32                :read_length,           label: 'Read Length'
+        uint64                :offset,                label: 'Read Offset'
+        smb2_fileid           :file_id,               label: 'File ID'
+        uint32                :min_bytes,             label: 'Minimum Count'
+        uint32                :channel,               label: 'Channel'
+        uint32                :remaining_bytes,       label: 'Remaining Bytes'
+        uint16                :channel_offset,        label: 'Read Channel Info Offset'
+        uint16                :channel_length,        label: 'Read Channel Info Length'
+        string                :buffer,                label: 'Read Channel info Buffer', initial_value: 0x00
+
+        def initialize_instance
+          super
+          smb2_header.command = RubySMB::SMB2::Commands::READ
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/read_response.rb

@@ -0,0 +1,26 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Read Response Packet as defined in
+      # [2.2.20 SMB2 READ Response](https://msdn.microsoft.com/en-us/library/cc246531.aspx)
+      class ReadResponse < RubySMB::GenericPacket
+        endian :little
+
+        smb2_header   :smb2_header
+        uint16        :structure_size,  label: 'Structure Size',      initial_value: 17
+        uint8         :data_offset,     label: 'Data Buffer Offset',  initial_value: -> { buffer.abs_offset }
+        uint8         :reserved,        label: 'Reserved Space'
+        uint32        :data_length,     label: 'Data Buffer Length'
+        uint32        :data_remaining,  label: 'Data Remaining'
+        uint32        :reserved2,       label: 'Reserved Space'
+        string        :buffer,          label: 'Data Buffer', length: -> { data_length }
+
+        def initialize_instance
+          super
+          smb2_header.command     = RubySMB::SMB2::Commands::READ
+          smb2_header.flags.reply = 1
+        end
+      end
+    end
+  end
+end

data/lib/ruby_smb/smb2/packet/session_setup_request.rb

@@ -1,7 +1,6 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 SessionSetupRequest Packet as defined in
       # [2.2.5 SMB2 SESSION_SETUP Request](https://msdn.microsoft.com/en-us/library/cc246563.aspx)
       class SessionSetupRequest < RubySMB::GenericPacket
@@ -15,8 +14,7 @@ module RubySMB
         uint16                     :security_buffer_offset,  label: 'Security Buffer Offset',  initial_value: 0x58
         uint16                     :security_buffer_length,  label: 'Security Buffer Length'
         uint64                     :previous_session_id,     label: 'Previous Session ID'
-        string                     :buffer,                  label: 'Security Buffer',         length: lambda { self.security_buffer_length }
-
+        string                     :buffer,                  label: 'Security Buffer', length: -> { security_buffer_length }
 
         def initialize_instance
           super
@@ -46,8 +44,7 @@ module RubySMB
           self.security_buffer_length = gss_blob.length
           self.buffer = gss_blob
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/session_setup_response.rb

@@ -1,18 +1,16 @@
 module RubySMB
   module SMB2
     module Packet
-
       # An SMB2 SessionSetupResponse Packet as defined in
       # [2.2.6 SMB2 SESSION_SETUP Response](https://msdn.microsoft.com/en-us/library/cc246564.aspx)
       class SessionSetupResponse < RubySMB::GenericPacket
         endian :little
         smb2_header         :smb2_header
-        uint16              :structure_size,          label: 'Structure Size',          initial_value: 9
+        uint16              :structure_size, label: 'Structure Size', initial_value: 9
         session_flags       :session_flags
-        uint16              :security_buffer_offset,  label: 'Security Buffer Offset',  initial_value: 0x48
+        uint16              :security_buffer_offset,  label: 'Security Buffer Offset', initial_value: 0x48
         uint16              :security_buffer_length,  label: 'Security Buffer Length'
-        string              :buffer,                  label: 'Security Buffer',         length: lambda { self.security_buffer_length }
-
+        string              :buffer,                  label: 'Security Buffer', length: -> { security_buffer_length }
 
         def initialize_instance
           super
@@ -31,8 +29,7 @@ module RubySMB
           self.security_buffer_length = gss_blob.length
           self.buffer = gss_blob
         end
-
       end
     end
   end
-end
+end

data/lib/ruby_smb/smb2/packet/set_info_request.rb

@@ -0,0 +1,58 @@
+module RubySMB
+  module SMB2
+    module Packet
+      # An SMB2 Set Info Request Packet as defined in
+      # [2.2.39 SMB2 SET_INFO Request](https://msdn.microsoft.com/en-us/library/cc246560.aspx)
+      class SetInfoRequest < RubySMB::GenericPacket
+        include RubySMB::Fscc::FileInformation
+
+        endian :little
+
+        smb2_header :smb2_header
+        uint16      :structure_size,  label: 'Structure Size', initial_value: 33
+        # Constants defined in RubySMB::SMB2::InfoType
+        uint8       :info_type,       label: 'Info Type',      initial_value: RubySMB::SMB2::InfoType::SMB2_0_INFO_FILE
+        uint8       :file_info_class, label: 'File Info Class'
+        uint32      :buffer_length,   label: 'Buffer Length',  initial_value: -> { buffer.do_num_bytes }
+        uint16      :buffer_offset,   label: 'Buffer Offset',  initial_value: -> { buffer.abs_offset }
+        uint16      :reserved,        label: 'Reserved',       initial_value: 0
+
+        struct :additional_information do
+          bit1  :reserved,                       label: 'Reserved Space'
+          bit1  :scope_security_information,     label: 'Scope Security Information'
+          bit1  :attribute_security_information, label: 'Attribute Security Information'
+          bit1  :label_security_information,     label: 'Label Security Information'
+          bit1  :sacl_security_information,      label: 'SACL Security Information'
+          bit1  :dacl_security_information,      label: 'DACL Security Information'
+          bit1  :group_security_information,     label: 'Group Security Information'
+          bit1  :owner_security_information,     label: 'Owner Security Information'
+          # byte boundary
+          bit8  :reserved2,                      label: 'Reserved Space'
+          # byte boundary
+          bit7  :reserved3,                      label: 'Reserved Space'
+          bit1  :backup_security_information,    label: 'Backup Security Information'
+          # byte boundary
+          bit8  :reserved4,                      label: 'Reserved Space'
+        end
+
+        smb2_fileid :file_id, label: 'File ID'
+
+        choice :buffer, label: 'Buffer', selection: -> { file_info_class } do
+          file_directory_information         FILE_DIRECTORY_INFORMATION,         label: 'File Directory Information'
+          file_full_directory_information    FILE_FULL_DIRECTORY_INFORMATION,    label: 'File Full Directory Information'
+          file_both_directory_information    FILE_BOTH_DIRECTORY_INFORMATION,    label: 'File Both Directory Information'
+          file_rename_information            FILE_RENAME_INFORMATION,            label: 'File Rename Information'
+          file_names_information             FILE_NAMES_INFORMATION,             label: 'File Names Information'
+          file_disposition_information       FILE_DISPOSITION_INFORMATION,       label: 'File Disposition Information'
+          file_id_both_directory_information FILE_ID_BOTH_DIRECTORY_INFORMATION, label: 'File Id Both Directory Information'
+          file_id_full_directory_information FILE_ID_FULL_DIRECTORY_INFORMATION, label: 'File Id Full Directory Information'
+        end
+
+        def initialize_instance
+          super
+          smb2_header.command = RubySMB::SMB2::Commands::SET_INFO
+        end
+      end
+    end
+  end
+end