ruby_smb 0.0.18 → 0.0.19

data/spec/lib/ruby_smb/smb1/packet/trans2/find_first2_request_spec.rb

@@ -0,0 +1,180 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::Trans2::FindFirst2Request do
+  subject(:packet) { described_class.new }
+
+  describe '#smb_header' do
+    subject(:header) { packet.smb_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB1::SMBHeader
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  end
+
+  describe '#parameter_block' do
+    subject(:parameter_block) { packet.parameter_block }
+
+    it 'is a standard ParameterBlock' do
+      expect(parameter_block).to be_a RubySMB::SMB1::Packet::Trans2::Request::ParameterBlock
+    end
+
+    it 'should have the setup set to the OPEN2 subcommand' do
+      expect(parameter_block.setup).to include RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_FIRST2
+    end
+  end
+
+  describe '#data_block' do
+    subject(:data_block) { packet.data_block }
+
+    it 'is a standard DataBlock' do
+      expect(data_block).to be_a RubySMB::SMB1::DataBlock
+    end
+
+    it { is_expected.to respond_to :name }
+    it { is_expected.to respond_to :trans2_parameters }
+    it { is_expected.to respond_to :trans2_data }
+
+    it 'should keep #trans2_parameters 4-byte aligned' do
+      expect(data_block.trans2_parameters.abs_offset % 4).to eq 0
+    end
+
+    it 'should keep #trans2_data 4-byte aligned' do
+      expect(data_block.trans2_data.abs_offset % 4).to eq 0
+    end
+
+    describe '#trans2_parameters' do
+      subject(:parameters) { data_block.trans2_parameters }
+
+      it { is_expected.to respond_to :search_attributes }
+      it { is_expected.to respond_to :search_count }
+      it { is_expected.to respond_to :flags }
+      it { is_expected.to respond_to :information_level }
+      it { is_expected.to respond_to :storage_type }
+      it { is_expected.to respond_to :filename }
+
+      describe '#search_attributes' do
+        it 'is a smb_file_attributes_field' do
+          expect(parameters.search_attributes).to be_a RubySMB::SMB1::BitField::SmbFileAttributes
+        end
+      end
+
+      describe '#flags' do
+        subject(:flags) { parameters.flags }
+
+        it { is_expected.to respond_to :backup }
+        it { is_expected.to respond_to :continue }
+        it { is_expected.to respond_to :resume_keys }
+        it { is_expected.to respond_to :close_eos }
+        it { is_expected.to respond_to :close }
+
+        describe 'close' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.close).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :close, 'v', 0x0001
+        end
+
+        describe 'close_eos' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.close_eos).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :close_eos, 'v', 0x0002
+        end
+
+        describe 'resume_keys' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.resume_keys).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :resume_keys, 'v', 0x0004
+        end
+
+        describe 'continue' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.continue).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :continue, 'v', 0x0008
+        end
+
+        describe 'backup' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.backup).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :backup, 'v', 0x0010
+        end
+      end
+
+      describe '#filename' do
+        let(:name) { 'hello.txt' }
+
+        before :example do
+          parameters.filename = name
+        end
+
+        context 'when unicode flag is set' do
+          before :example do
+            packet.smb_header.flags2.unicode = 1
+          end
+
+          it 'is terminated with unicode null-termination character' do
+            expect(parameters.filename.to_binary_s).to end_with("\x00\x00")
+          end
+
+          it 'is UTF-16LE encoded' do
+            expect(parameters.filename.encoding.name).to eq 'UTF-16LE'
+          end
+        end
+
+        context 'when unicode flag is not set' do
+          before :example do
+            packet.smb_header.flags2.unicode = 0
+          end
+
+          it 'is terminated with ASCII null-termination character' do
+            expect(parameters.filename.to_binary_s).to end_with("\x00")
+          end
+
+          it 'is ASCII-8BIT encoded' do
+            expect(parameters.filename.encoding.name).to eq 'ASCII-8BIT'
+          end
+        end
+      end
+    end
+
+    describe '#trans2_data' do
+      subject(:data) { data_block.trans2_data }
+
+      it { is_expected.to respond_to :extended_attribute_list }
+
+      describe '#extended_attribute_list' do
+        it 'is an smb_gea_list' do
+          expect(data.extended_attribute_list).to be_a RubySMB::Field::SmbGeaList
+        end
+
+        it 'only exists if the information level is SMB_INFO_QUERY_EAS_FROM_LIST' do
+          data_block.trans2_parameters.information_level =
+            RubySMB::SMB1::Packet::Trans2::FindInformationLevel::SMB_INFO_QUERY_EAS_FROM_LIST
+          expect(data.do_num_bytes).to eq RubySMB::Field::SmbGeaList.new.do_num_bytes
+        end
+
+        it 'does not exist if the information level is SMB_INFO_STANDARD' do
+          data_block.trans2_parameters.information_level =
+            RubySMB::SMB1::Packet::Trans2::FindInformationLevel::SMB_INFO_STANDARD
+          expect(data.do_num_bytes).to eq 0
+        end
+      end
+    end
+  end
+end

data/spec/lib/ruby_smb/smb1/packet/trans2/find_first2_response_spec.rb

@@ -0,0 +1,104 @@
+require 'spec_helper'
+
+include RubySMB::SMB1::Packet::Trans2::FindInformationLevel
+
+RSpec.describe RubySMB::SMB1::Packet::Trans2::FindFirst2Response do
+  subject(:packet) { described_class.new }
+
+  describe '#smb_header' do
+    subject(:header) { packet.smb_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB1::SMBHeader
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    end
+  end
+
+  describe '#parameter_block' do
+    subject(:parameter_block) { packet.parameter_block }
+
+    it 'should have the setup set to the OPEN2 subcommand' do
+      expect(parameter_block.setup).to include RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_FIRST2
+    end
+  end
+
+  describe '#data_block' do
+    subject(:data_block) { packet.data_block }
+
+    it { is_expected.to respond_to :name }
+    it { is_expected.to respond_to :trans2_parameters }
+    it { is_expected.to respond_to :trans2_data }
+
+    it 'should keep #trans2_parameters 4-byte aligned' do
+      expect(data_block.trans2_parameters.abs_offset % 4).to eq 0
+    end
+
+    it 'should keep #trans2_data 4-byte aligned' do
+      expect(data_block.trans2_data.abs_offset % 4).to eq 0
+    end
+
+    describe '#trans2_parameters' do
+      subject(:parameters) { data_block.trans2_parameters }
+
+      it { is_expected.to respond_to :sid }
+      it { is_expected.to respond_to :search_count }
+      it { is_expected.to respond_to :eos }
+      it { is_expected.to respond_to :ea_error_offset }
+      it { is_expected.to respond_to :last_name_offset }
+    end
+
+    describe '#trans2_data' do
+      subject(:data) { data_block.trans2_data }
+
+      it { is_expected.to respond_to :buffer }
+    end
+  end
+
+  describe '#results' do
+
+    let(:names1) {
+      names = FindFileFullDirectoryInfo.new
+      names.file_name = 'test.txt'
+      names.next_offset = names.do_num_bytes
+      names
+    }
+
+    let(:names2) {
+      names = FindFileFullDirectoryInfo.new
+      names.file_name = '..'
+      names
+    }
+
+    let(:names_array) { [names1, names2] }
+
+    let(:names_blob) { names_array.collect(&:to_binary_s).join('') }
+
+    it 'returns an array of parsed FindFileFullDirectoryInfo structs' do
+      packet.data_block.trans2_data.buffer = names_blob
+      expect(packet.results(FindFileFullDirectoryInfo, unicode: false)).to eq names_array
+    end
+
+    it 'sets the FindFileFullDirectoryInfo unicode attribute when unicode argument is true' do
+      packet.data_block.trans2_data.buffer = names1.to_binary_s
+      find_info = FindFileFullDirectoryInfo.new
+      allow(FindFileFullDirectoryInfo).to receive(:new).and_return find_info
+      expect(find_info).to receive(:unicode=).with(true).once
+      packet.results(FindFileFullDirectoryInfo, unicode: true)
+    end
+
+    it 'does not set the FindFileFullDirectoryInfo unicode attribute when unicode argument is false' do
+      packet.data_block.trans2_data.buffer = names1.to_binary_s
+      find_info = FindFileFullDirectoryInfo.new
+      allow(FindFileFullDirectoryInfo).to receive(:new).and_return find_info
+      expect(find_info).to receive(:unicode=).with(false).once
+      packet.results(FindFileFullDirectoryInfo, unicode: false)
+    end
+  end
+end

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

@@ -0,0 +1,128 @@
+RSpec.describe RubySMB::SMB1::Packet::Trans2::FindInformationLevel::FindFileFullDirectoryInfo do
+  it 'references the correct class level' do
+    expect(described_class).to be_const_defined(:CLASS_LEVEL)
+    expect(described_class::CLASS_LEVEL).to be RubySMB::SMB1::Packet::Trans2::FindInformationLevel::SMB_FIND_FILE_FULL_DIRECTORY_INFO
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_offset }
+  it { should respond_to :file_index }
+  it { should respond_to :create_time }
+  it { should respond_to :last_access }
+  it { should respond_to :last_write }
+  it { should respond_to :last_change }
+  it { should respond_to :end_of_file }
+  it { should respond_to :allocation_size }
+  it { should respond_to :ext_file_attributes }
+  it { should respond_to :file_name_length }
+  it { should respond_to :ea_size }
+  it { should respond_to :file_name }
+  it { should respond_to :unicode }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#next_offset' do
+    it 'is a 32-bit field' do
+      expect(struct.next_offset).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#file_index' do
+    it 'is a 32-bit field' do
+      expect(struct.file_index).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#create_time' do
+    it 'is a Filetime field' do
+      expect(struct.create_time).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#last_access' do
+    it 'is a Filetime field' do
+      expect(struct.last_access).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#last_write' do
+    it 'is a Filetime field' do
+      expect(struct.last_write).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#last_change' do
+    it 'is a Filetime field' do
+      expect(struct.last_change).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#end_of_file' do
+    it 'is a 64-bit field' do
+      expect(struct.end_of_file).to be_a BinData::Uint64le
+    end
+  end
+
+  describe '#ext_file_attributes' do
+    it 'contains the extended file attributes of the file' do
+      expect(struct.ext_file_attributes).to be_a RubySMB::SMB1::BitField::SmbExtFileAttributes
+    end
+  end
+
+  describe '#file_name_length' do
+    it 'is a 32-bit field' do
+      expect(struct.file_name_length).to be_a BinData::Uint32le
+    end
+
+    it 'tracks the length of the file_name field including the null-terminating character when unicode is not set' do
+      filename = "Hello.txt"
+      struct.file_name = filename
+      expect(struct.file_name_length).to eq (filename.bytes.size + 1)
+    end
+
+    it 'tracks the length of the file_name field without null-terminating characters when unicode is set' do
+      filename = "Hello.txt"
+      struct.unicode = true
+      struct.file_name = filename
+      expect(struct.file_name_length).to eq (filename.encode('UTF-16LE').bytes.size)
+    end
+  end
+
+  describe '#ea_size' do
+    it 'is a 32-bit field' do
+      expect(struct.ea_size).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#file_name' do
+    let(:filename) { "Hello.txt" }
+
+    before :example do
+      struct.file_name = filename
+    end
+
+    it 'is an unicode string field when unicode attribute is set to true' do
+      struct.unicode = true
+      expect(struct.file_name.encoding.name).to eq 'UTF-16LE'
+    end
+
+    it 'is an ASCII-8BIT field when unicode attribute is set to false' do
+      expect(struct.file_name.encoding.name).to eq 'ASCII-8BIT'
+    end
+
+    it 'is null-terminated when unicode attribute is set to false' do
+      expect(struct.file_name.to_binary_s).to eq "#{filename}\x00"
+    end
+  end
+
+  describe '#unicode' do
+    it 'is set to false by default' do
+      expect(struct.unicode).to be false
+    end
+  end
+
+end
+

data/spec/lib/ruby_smb/smb1/packet/trans2/find_next2_request_spec.rb

@@ -0,0 +1,174 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::Trans2::FindNext2Request do
+  subject(:packet) { described_class.new }
+
+  describe '#smb_header' do
+    subject(:header) { packet.smb_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB1::SMBHeader
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  end
+
+  describe '#parameter_block' do
+    subject(:parameter_block) { packet.parameter_block }
+
+    it 'is a standard ParameterBlock' do
+      expect(parameter_block).to be_a RubySMB::SMB1::Packet::Trans2::Request::ParameterBlock
+    end
+
+    it 'should have the setup set to the OPEN2 subcommand' do
+      expect(parameter_block.setup).to include RubySMB::SMB1::Packet::Trans2::Subcommands::FIND_NEXT2
+    end
+  end
+
+  describe '#data_block' do
+    subject(:data_block) { packet.data_block }
+
+    it 'is a standard DataBlock' do
+      expect(data_block).to be_a RubySMB::SMB1::DataBlock
+    end
+
+    it { is_expected.to respond_to :name }
+    it { is_expected.to respond_to :trans2_parameters }
+    it { is_expected.to respond_to :trans2_data }
+
+    it 'should keep #trans2_parameters 4-byte aligned' do
+      expect(data_block.trans2_parameters.abs_offset % 4).to eq 0
+    end
+
+    it 'should keep #trans2_data 4-byte aligned' do
+      expect(data_block.trans2_data.abs_offset % 4).to eq 0
+    end
+
+    describe '#trans2_parameters' do
+      subject(:parameters) { data_block.trans2_parameters }
+
+      it { is_expected.to respond_to :sid }
+      it { is_expected.to respond_to :search_count }
+      it { is_expected.to respond_to :flags }
+      it { is_expected.to respond_to :information_level }
+      it { is_expected.to respond_to :resume_key }
+      it { is_expected.to respond_to :filename }
+
+      describe '#flags' do
+        subject(:flags) { parameters.flags }
+
+        it { is_expected.to respond_to :backup }
+        it { is_expected.to respond_to :continue }
+        it { is_expected.to respond_to :resume_keys }
+        it { is_expected.to respond_to :close_eos }
+        it { is_expected.to respond_to :close }
+
+        describe 'close' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.close).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :close, 'v', 0x0001
+        end
+
+        describe 'close_eos' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.close_eos).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :close_eos, 'v', 0x0002
+        end
+
+        describe 'resume_keys' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.resume_keys).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :resume_keys, 'v', 0x0004
+        end
+
+        describe 'continue' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.continue).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :continue, 'v', 0x0008
+        end
+
+        describe 'backup' do
+          it 'should be a 1-bit field per the SMB spec' do
+            expect(flags.backup).to be_a BinData::Bit1
+          end
+
+          it_behaves_like 'bit field with one flag set', :backup, 'v', 0x0010
+        end
+      end
+
+      describe '#filename' do
+        let(:name) { 'hello.txt' }
+
+        before :example do
+          parameters.filename = name
+        end
+
+        context 'when unicode flag is set' do
+          before :example do
+            packet.smb_header.flags2.unicode = 1
+          end
+
+          it 'is terminated with unicode null-termination character' do
+            expect(parameters.filename.to_binary_s).to end_with("\x00\x00")
+          end
+
+          it 'is UTF-16LE encoded' do
+            expect(parameters.filename.encoding.name).to eq 'UTF-16LE'
+          end
+        end
+
+        context 'when unicode flag is not set' do
+          before :example do
+            packet.smb_header.flags2.unicode = 0
+          end
+
+          it 'is terminated with ASCII null-termination character' do
+            expect(parameters.filename.to_binary_s).to end_with("\x00")
+          end
+
+          it 'is ASCII-8BIT encoded' do
+            expect(parameters.filename.encoding.name).to eq 'ASCII-8BIT'
+          end
+        end
+      end
+    end
+
+    describe '#trans2_data' do
+      subject(:data) { data_block.trans2_data }
+
+      it { is_expected.to respond_to :extended_attribute_list }
+
+      describe '#extended_attribute_list' do
+        it 'is an smb_gea_list' do
+          expect(data.extended_attribute_list).to be_a RubySMB::Field::SmbGeaList
+        end
+
+        it 'only exists if the information level is SMB_INFO_QUERY_EAS_FROM_LIST' do
+          data_block.trans2_parameters.information_level =
+            RubySMB::SMB1::Packet::Trans2::FindInformationLevel::SMB_INFO_QUERY_EAS_FROM_LIST
+          expect(data.do_num_bytes).to eq RubySMB::Field::SmbGeaList.new.do_num_bytes
+        end
+
+        it 'does not exist if the information level is SMB_INFO_STANDARD' do
+          data_block.trans2_parameters.information_level =
+            RubySMB::SMB1::Packet::Trans2::FindInformationLevel::SMB_INFO_STANDARD
+          expect(data.do_num_bytes).to eq 0
+        end
+      end
+    end
+  end
+end