ruby_smb 0.0.18 → 0.0.19

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

@@ -0,0 +1,102 @@
+require 'spec_helper'
+
+include RubySMB::SMB1::Packet::Trans2::FindInformationLevel
+
+RSpec.describe RubySMB::SMB1::Packet::Trans2::FindNext2Response 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_NEXT2
+    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 :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/open2_request_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Request do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -30,7 +29,6 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Request do
     it 'should have the setup set to the OPEN2 subcommand' do
       expect(parameter_block.setup).to include RubySMB::SMB1::Packet::Trans2::Subcommands::OPEN2
     end
-
   end
 
   describe '#data_block' do
@@ -105,8 +103,5 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Request do
         end
       end
     end
-
   end
-
-
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Response do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -49,8 +48,6 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Response do
     describe '#trans2_parameters' do
       subject(:parameters) { data_block.trans2_parameters }
 
-      it { is_expected.to be_a RubySMB::SMB1::Packet::Trans2::Open2Response::Trans2Parameters  }
-
       it { is_expected.to respond_to :fid }
       it { is_expected.to respond_to :file_attributes }
       it { is_expected.to respond_to :creation_time }
@@ -86,7 +83,7 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Response do
       end
 
       describe '#action_taken' do
-        subject(:action) { parameters.action_taken}
+        subject(:action) { parameters.action_taken }
 
         it 'is 2-bytes' do
           expect(action.do_num_bytes).to eq 2
@@ -102,6 +99,4 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Response do
       end
     end
   end
-
-
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::Trans2::RequestSecondary do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -70,8 +69,5 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::RequestSecondary do
     it 'is a standard DataBlock' do
       expect(data_block).to be_a RubySMB::SMB1::Packet::Trans2::Request::DataBlock
     end
-
   end
-
-
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::Trans2::Request do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -91,8 +90,5 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Request do
       data_block.trans2_parameters = 'a'
       expect(data_block.trans2_data.abs_offset % 4).to eq 0
     end
-
   end
-
-
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::Trans2::Response do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -83,6 +82,4 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Response do
       expect(data_block.trans2_data.abs_offset % 4).to eq 0
     end
   end
-
-
-end
+end

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

@@ -0,0 +1,98 @@
+include RubySMB::Fscc::FileInformation
+
+RSpec.describe RubySMB::SMB1::Packet::Trans2::SetFileInformationRequest 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_TRANSACTION2' 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::SET_FILE_INFORMATION
+    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 :fid }
+      it { is_expected.to respond_to :information_level }
+
+      describe '#fid' do
+        it 'is a 16-bit field' do
+          expect(parameters.fid).to be_a BinData::Uint16le
+        end
+      end
+
+      describe '#information_level' do
+        it 'is a 16-bit field' do
+          expect(parameters.information_level).to be_a BinData::Uint16le
+        end
+      end
+    end
+
+    describe '#trans2_data' do
+      subject(:data) { data_block.trans2_data }
+
+      it { is_expected.to respond_to :info_level_struct }
+
+      describe '#info_level_struct' do
+        context 'when #information_level field is FILE_DISPOSITION_INFORMATION with the pass-through capability'
+        it 'is a FileDispositionInformation structure' do
+          info_level = FILE_DISPOSITION_INFORMATION + SMB_INFO_PASSTHROUGH
+          data_block.trans2_parameters.information_level = info_level
+          file_info = FileDispositionInformation.new
+          expect(data.info_level_struct).to eq file_info
+        end
+
+        context 'when #information_level field is FILE_RENAME_INFORMATION with the pass-through capability'
+        it 'is a FileRenameInformation structure' do
+          info_level = FILE_RENAME_INFORMATION + SMB_INFO_PASSTHROUGH
+          data_block.trans2_parameters.information_level = info_level
+          file_info = FileRenameInformation.new
+          expect(data.info_level_struct).to eq file_info
+        end
+      end
+    end
+
+  end
+end
+

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

@@ -0,0 +1,56 @@
+RSpec.describe RubySMB::SMB1::Packet::Trans2::SetFileInformationResponse 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::SET_FILE_INFORMATION
+    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 :ea_error_offset }
+
+      describe '#ea_error_offset' do
+        it 'is a 16-bit field' do
+          expect(parameters.ea_error_offset).to be_a BinData::Uint16le
+        end
+      end
+    end
+  end
+
+end

data/spec/lib/ruby_smb/smb1/packet/tree_connect_request_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::TreeConnectRequest do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -46,8 +45,5 @@ RSpec.describe RubySMB::SMB1::Packet::TreeConnectRequest do
     it { is_expected.to respond_to :password }
     it { is_expected.to respond_to :path }
     it { is_expected.to respond_to :service }
-
   end
-
-
-end
+end

data/spec/lib/ruby_smb/smb1/packet/tree_connect_response_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::TreeConnectResponse do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -35,7 +34,6 @@ RSpec.describe RubySMB::SMB1::Packet::TreeConnectResponse do
     it 'has an AndXBlock' do
       expect(parameter_block.andx_block).to be_a RubySMB::SMB1::AndXBlock
     end
-
   end
 
   describe '#data_block' do
@@ -47,7 +45,6 @@ RSpec.describe RubySMB::SMB1::Packet::TreeConnectResponse do
 
     it { is_expected.to respond_to :service }
     it { is_expected.to respond_to :native_file_system }
-
   end
 
   context 'when the connect is to a directory' do
@@ -80,7 +77,5 @@ RSpec.describe RubySMB::SMB1::Packet::TreeConnectResponse do
     it 'returns a FileAccessMask from #guest_access_rights' do
       expect(file_response.guest_access_rights).to be_a RubySMB::SMB1::BitField::FileAccessMask
     end
-
   end
-
-end
+end

data/spec/lib/ruby_smb/smb1/packet/tree_disconnect_request_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::TreeDisconnectRequest do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -35,6 +34,4 @@ RSpec.describe RubySMB::SMB1::Packet::TreeDisconnectRequest do
       expect(data_block).to be_a RubySMB::SMB1::DataBlock
     end
   end
-
-
-end
+end

data/spec/lib/ruby_smb/smb1/packet/tree_disconnect_response_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB1::Packet::TreeDisconnectResponse do
-
   subject(:packet) { described_class.new }
 
   describe '#smb_header' do
@@ -35,6 +34,4 @@ RSpec.describe RubySMB::SMB1::Packet::TreeDisconnectResponse do
       expect(data_block).to be_a RubySMB::SMB1::DataBlock
     end
   end
-
-
-end
+end