ruby_smb 0.0.18 → 0.0.19

data/spec/lib/ruby_smb/smb2/packet/session_setup_response_spec.rb

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::SessionSetupResponse do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -32,7 +31,7 @@ RSpec.describe RubySMB::SMB2::Packet::SessionSetupResponse do
   end
 
   describe '#set_type2_blob' do
-    let(:fake_message) { "foo" }
+    let(:fake_message) { 'foo' }
 
     it 'calls the #gss_type2 method to create a blob' do
       expect(RubySMB::Gss).to receive(:gss_type2).with(fake_message).and_return(fake_message)
@@ -51,4 +50,4 @@ RSpec.describe RubySMB::SMB2::Packet::SessionSetupResponse do
       expect(packet.security_buffer_length).to eq fake_message.length
     end
   end
-end
+end

data/spec/lib/ruby_smb/smb2/packet/set_info_request_spec.rb

@@ -0,0 +1,205 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::SetInfoRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :smb2_header }
+  it { is_expected.to respond_to :structure_size }
+  it { is_expected.to respond_to :info_type }
+  it { is_expected.to respond_to :file_info_class }
+  it { is_expected.to respond_to :buffer_length }
+  it { is_expected.to respond_to :buffer_offset }
+  it { is_expected.to respond_to :reserved }
+  it { is_expected.to respond_to :additional_information }
+  it { is_expected.to respond_to :file_id }
+  it { is_expected.to respond_to :buffer }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#smb2_header' do
+    subject(:header) { packet.smb2_header }
+
+    it 'is a standard SMB2 Header' do
+      expect(header).to be_a RubySMB::SMB2::SMB2Header
+    end
+
+    it 'should have the command set to SMB_INFO' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::SET_INFO
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  end
+
+  describe '#structure_size' do
+    it 'is a 16-bit field' do
+      expect(packet.structure_size).to be_a BinData::Uint16le
+    end
+
+    it 'has a default value of 33' do
+      expect(packet.structure_size).to eq 33
+    end
+  end
+
+  describe '#info_type' do
+    it 'is a 8-bit field' do
+      expect(packet.info_type).to be_a BinData::Uint8
+    end
+
+   it 'has a default value of SMB2_0_INFO_FILE' do
+      expect(packet.info_type).to eq RubySMB::SMB2::InfoType::SMB2_0_INFO_FILE
+    end
+  end
+
+  describe '#file_info_class' do
+    it 'is a 8-bit field' do
+      expect(packet.file_info_class).to be_a BinData::Uint8
+    end
+  end
+
+  describe '#buffer_length' do
+    it 'is a 32-bit field' do
+      expect(packet.buffer_length).to be_a BinData::Uint32le
+    end
+
+    it 'is set to the buffer size' do
+      packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_FULL_DIRECTORY_INFORMATION
+      expect(packet.buffer_length).to eq packet.buffer.do_num_bytes
+    end
+  end
+
+  describe '#buffer_offset' do
+    it 'is a 16-bit field' do
+      expect(packet.buffer_offset).to be_a BinData::Uint16le
+    end
+
+    it 'is set to the buffer size' do
+      packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_FULL_DIRECTORY_INFORMATION
+      expect(packet.buffer_offset).to eq packet.buffer.abs_offset
+    end
+  end
+
+  describe '#additional_information' do
+    subject(:additional_information) { packet.additional_information }
+
+    it { is_expected.to respond_to :scope_security_information }
+    it { is_expected.to respond_to :attribute_security_information }
+    it { is_expected.to respond_to :label_security_information }
+    it { is_expected.to respond_to :sacl_security_information }
+    it { is_expected.to respond_to :dacl_security_information }
+    it { is_expected.to respond_to :group_security_information }
+    it { is_expected.to respond_to :owner_security_information }
+    it { is_expected.to respond_to :backup_security_information }
+
+    describe '#scope_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.scope_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :scope_security_information, 'V', 0x00000040
+    end
+
+    describe '#attribute_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.attribute_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :attribute_security_information, 'V', 0x00000020
+    end
+
+    describe '#label_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.label_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :label_security_information, 'V', 0x00000010
+    end
+
+    describe '#sacl_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.sacl_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :sacl_security_information, 'V', 0x00000008
+    end
+
+    describe '#dacl_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.dacl_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :dacl_security_information, 'V', 0x00000004
+    end
+
+    describe '#group_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.group_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :group_security_information, 'V', 0x00000002
+    end
+
+    describe '#owner_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.owner_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :owner_security_information, 'V', 0x00000001
+    end
+
+    describe '#backup_security_information' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(additional_information.backup_security_information).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :backup_security_information, 'V', 0x00010000
+    end
+  end
+
+  describe '#file_id' do
+    it 'is a Smb2Fileid field' do
+      expect(packet.file_id).to be_a RubySMB::Field::Smb2Fileid
+    end
+  end
+
+  describe '#buffer' do
+    context 'when file_info_class is set' do
+      it 'is set to a FileDirectoryInformation class with the FILE_DIRECTORY_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_DIRECTORY_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileDirectoryInformation.new
+      end
+      it 'is set to a FileFullDirectoryInformation class with the FILE_FULL_DIRECTORY_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_FULL_DIRECTORY_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileFullDirectoryInformation.new
+      end
+      it 'is set to a FileDispositionInformation class with the FILE_DISPOSITION_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_DISPOSITION_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileDispositionInformation.new
+      end
+      it 'is set to a FileIdFullDirectoryInformation class with the FILE_ID_FULL_DIRECTORY_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_ID_FULL_DIRECTORY_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileIdFullDirectoryInformation.new
+      end
+      it 'is set to a FileBothDirectoryInformation class with the FILE_BOTH_DIRECTORY_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_BOTH_DIRECTORY_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileBothDirectoryInformation.new
+      end
+      it 'is set to a FileIdBothDirectoryInformation class with the FILE_ID_BOTH_DIRECTORY_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_ID_BOTH_DIRECTORY_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileIdBothDirectoryInformation.new
+      end
+      it 'is set to a FileNamesInformation class with the FILE_NAMES_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_NAMES_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileNamesInformation.new
+      end
+      it 'is set to a FileRenameInformation class with the FILE_RENAME_INFORMATION code' do
+        packet.file_info_class = RubySMB::Fscc::FileInformation::FILE_RENAME_INFORMATION
+        expect(packet.buffer).to eq RubySMB::Fscc::FileInformation::FileRenameInformation.new
+      end
+    end
+  end
+
+end

data/spec/lib/ruby_smb/smb2/packet/set_info_response_spec.rb

@@ -0,0 +1,32 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::SetInfoResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :smb2_header }
+  it { is_expected.to respond_to :structure_size }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#smb2_header' do
+    subject(:header) { packet.smb2_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB2::SMB2Header
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::SET_INFO
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    end
+  end
+
+  it 'should have a structure size of 17' do
+    expect(packet.structure_size).to eq 2
+  end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::TreeConnectRequest do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -11,7 +10,6 @@ RSpec.describe RubySMB::SMB2::Packet::TreeConnectRequest do
   it { is_expected.to respond_to :path_length }
   it { is_expected.to respond_to :path }
 
-
   it 'is little endian' do
     expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
   end
@@ -33,8 +31,8 @@ RSpec.describe RubySMB::SMB2::Packet::TreeConnectRequest do
   end
 
   describe '#encode_path' do
-    let(:path) { "\\192.168.1.1\\example" }
-    let(:encoded_path) { path.encode("utf-16le").force_encoding("binary") }
+    let(:path) { '\\192.168.1.1\\example' }
+    let(:encoded_path) { path.encode('utf-16le').force_encoding('binary') }
 
     it 'sets the path string to a UTF-16LE version of the supplied string' do
       packet.encode_path(path)
@@ -46,4 +44,4 @@ RSpec.describe RubySMB::SMB2::Packet::TreeConnectRequest do
       expect(packet.path_length).to eq encoded_path.length
     end
   end
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::TreeConnectResponse do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -54,4 +53,4 @@ RSpec.describe RubySMB::SMB2::Packet::TreeConnectResponse do
       expect(packet.access_rights).to be_a RubySMB::SMB2::BitField::FileAccessMask
     end
   end
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::TreeDisconnectRequest do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -26,5 +25,4 @@ RSpec.describe RubySMB::SMB2::Packet::TreeDisconnectRequest do
       expect(header.flags.reply).to eq 0
     end
   end
-
-end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::TreeDisconnectResponse do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -26,5 +25,4 @@ RSpec.describe RubySMB::SMB2::Packet::TreeDisconnectResponse do
       expect(header.flags.reply).to eq 1
     end
   end
-
-end
+end

data/spec/lib/ruby_smb/smb2/packet/write_request_spec.rb

@@ -0,0 +1,51 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::WriteRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :smb2_header }
+  it { is_expected.to respond_to :structure_size }
+  it { is_expected.to respond_to :data_offset }
+  it { is_expected.to respond_to :flags }
+  it { is_expected.to respond_to :write_length }
+  it { is_expected.to respond_to :write_offset }
+  it { is_expected.to respond_to :file_id }
+  it { is_expected.to respond_to :channel }
+  it { is_expected.to respond_to :remaining_bytes }
+  it { is_expected.to respond_to :channel_offset }
+  it { is_expected.to respond_to :channel_length }
+  it { is_expected.to respond_to :buffer }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#smb2_header' do
+    subject(:header) { packet.smb2_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB2::SMB2Header
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::WRITE
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  end
+
+  it 'should have a structure size of 49' do
+    expect(packet.structure_size).to eq 49
+  end
+
+  it 'stores the offset of the write buffer in #data_offset' do
+    expect(packet.data_offset).to eq packet.buffer.abs_offset
+  end
+
+  it 'stores the length of the write buffer in #write_length' do
+    packet.buffer = 'hello'
+    expect(packet.write_length).to eq packet.buffer.length
+  end
+end

data/spec/lib/ruby_smb/smb2/packet/write_response_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::WriteResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :smb2_header }
+  it { is_expected.to respond_to :structure_size }
+  it { is_expected.to respond_to :write_count }
+  it { is_expected.to respond_to :remaining_bytes }
+  it { is_expected.to respond_to :channel_offset }
+  it { is_expected.to respond_to :channel_length }
+
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#smb2_header' do
+    subject(:header) { packet.smb2_header }
+
+    it 'is a standard SMB Header' do
+      expect(header).to be_a RubySMB::SMB2::SMB2Header
+    end
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::WRITE
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    end
+  end
+
+  it 'should have a structure size of 49' do
+    expect(packet.structure_size).to eq 17
+  end
+
+end

data/spec/lib/ruby_smb/smb2/tree_spec.rb

@@ -1,7 +1,7 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Tree do
-  let(:sock)  { double("Socket", peeraddr: '192.168.1.5' ) }
+  let(:sock) { double('Socket', peeraddr: '192.168.1.5') }
   let(:dispatcher) { RubySMB::Dispatcher::Socket.new(sock) }
 
   let(:client) { RubySMB::Client.new(dispatcher, username: 'msfadmin', password: 'msfadmin') }
@@ -14,8 +14,12 @@ RSpec.describe RubySMB::SMB2::Tree do
     packet.share_type = 0x01
     packet
   }
+
+  let(:disco_req) { RubySMB::SMB2::Packet::TreeDisconnectRequest.new }
+  let(:disco_resp) { RubySMB::SMB2::Packet::TreeDisconnectResponse.new }
+
   subject(:tree) {
-    described_class.new(client:client, share:path, response:response )
+    described_class.new(client: client, share: path, response: response)
   }
 
   it { is_expected.to respond_to :client }
@@ -36,9 +40,6 @@ RSpec.describe RubySMB::SMB2::Tree do
   end
 
   describe '#disconnect!' do
-    let(:disco_req) { RubySMB::SMB2::Packet::TreeDisconnectRequest.new }
-    let(:disco_resp) { RubySMB::SMB2::Packet::TreeDisconnectResponse.new }
-
     it 'sends a TreeDisconnectRequest with the Tree ID in the header' do
       allow(RubySMB::SMB2::Packet::TreeDisconnectRequest).to receive(:new).and_return(disco_req)
       modified_req = disco_req
@@ -52,4 +53,189 @@ RSpec.describe RubySMB::SMB2::Tree do
       expect(tree.disconnect!).to eq disco_resp.status_code
     end
   end
+
+  describe '#set_header_fields' do
+    let(:modified_request) { tree.set_header_fields(disco_req) }
+    it 'adds the TreeID to the header' do
+      expect(modified_request.smb2_header.tree_id).to eq tree.id
+    end
+
+    it 'sets the credit charge to 1' do
+      expect(modified_request.smb2_header.credit_charge).to eq 1
+    end
+
+    it 'sets the credits to 256' do
+      expect(modified_request.smb2_header.credits).to eq 256
+    end
+  end
+
+  describe '#open_directory_packet' do
+    describe 'directory name' do
+      it 'uses a null byte of nothing is passed in' do
+        expect(tree.open_directory_packet.name).to eq "\x00".encode('UTF-16LE')
+      end
+
+      it 'sets the #name_length to 0 if no name is passed in' do
+        expect(tree.open_directory_packet.name_length).to eq 0
+      end
+
+      it 'encodes any supplied file name in UTF-16LE' do
+        name = 'hello.txt'
+        expect(tree.open_directory_packet(directory: name).name).to eq name.encode('UTF-16LE')
+      end
+    end
+
+    describe 'disposition' do
+      it 'defaults to FILE_OPEN' do
+        expect(tree.open_directory_packet.create_disposition).to eq RubySMB::Dispositions::FILE_OPEN
+      end
+
+      it 'can take the Disposition as an argument' do
+        expect(tree.open_directory_packet(disposition: RubySMB::Dispositions::FILE_OPEN_IF).create_disposition).to eq RubySMB::Dispositions::FILE_OPEN_IF
+      end
+    end
+
+    describe 'impersonation level' do
+      it 'defaults to SEC_IMPERSONATE' do
+        expect(tree.open_directory_packet.impersonation_level).to eq RubySMB::ImpersonationLevels::SEC_IMPERSONATE
+      end
+
+      it 'can take the Impersonation Level as an argument' do
+        expect(tree.open_directory_packet(impersonation: RubySMB::ImpersonationLevels::SEC_DELEGATE).impersonation_level).to eq RubySMB::ImpersonationLevels::SEC_DELEGATE
+      end
+    end
+
+    describe 'RWD access permissions' do
+      it 'will set the read permission from the parameters' do
+        expect(tree.open_directory_packet(read: true).share_access.read_access).to eq 1
+      end
+
+      it 'will set the write permission from the parameters' do
+        expect(tree.open_directory_packet(write: true).share_access.write_access).to eq 1
+      end
+
+      it 'will set the delete permission from the parameters' do
+        expect(tree.open_directory_packet(delete: true).share_access.delete_access).to eq 1
+      end
+    end
+  end
+
+  describe '#open_directory' do
+    let(:create_req) { RubySMB::SMB2::Packet::CreateRequest.new }
+    let(:create_response) { RubySMB::SMB2::Packet::CreateResponse.new }
+
+    it 'sends the create request packet and gets a response back' do
+      allow(tree).to receive(:open_directory_packet).and_return(create_req)
+      expect(client).to receive(:send_recv).with(create_req).and_return(create_response.to_binary_s)
+      tree.open_directory
+    end
+  end
+
+  describe '#list' do
+    let(:create_res) { double('create response') }
+    let(:query_dir_req) { RubySMB::SMB2::Packet::QueryDirectoryRequest.new }
+    let(:query_dir_res) { RubySMB::SMB2::Packet::QueryDirectoryResponse.new }
+
+    before :example do
+      allow(tree).to receive(:open_directory).and_return(create_res)
+      allow(create_res).to receive(:file_id)
+      allow(RubySMB::SMB2::Packet::QueryDirectoryRequest).to receive(:new).and_return(query_dir_req)
+      allow(client).to receive(:send_recv)
+      allow(RubySMB::SMB2::Packet::QueryDirectoryResponse).to receive(:read).and_return(query_dir_res)
+      query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_NO_MORE_FILES.value
+    end
+
+    it 'calls #open_directory' do
+      dir = '/dir'
+      expect(tree).to receive(:open_directory).with(directory: dir).and_return(create_res)
+      tree.list(directory: dir)
+    end
+
+    it 'uses the File ID from the create response' do
+      file_id = RubySMB::Field::Smb2Fileid.new
+      allow(create_res).to receive(:file_id).and_return(file_id)
+      allow(client).to receive(:send_recv) do |packet|
+        expect(packet.file_id).to eq file_id
+      end
+      tree.list
+    end
+
+    it 'sets the default QueryDirectoryRequest values' do
+      allow(client).to receive(:send_recv) do |packet|
+        expect(packet.file_information_class).to eq RubySMB::Fscc::FileInformation::FileIdFullDirectoryInformation::CLASS_LEVEL
+        expect(packet.name).to eq '*'.encode('UTF-16LE')
+        expect(packet.output_length).to eq 65_535
+      end
+      tree.list
+    end
+
+    it 'sets QueryDirectoryRequest #name field to the pattern passed as argument' do
+      pattern = '/dir/*/'.encode('UTF-16LE')
+      allow(client).to receive(:send_recv) do |packet|
+        expect(packet.name).to eq pattern
+      end
+      tree.list(pattern: pattern)
+    end
+
+    it 'sets QueryDirectoryRequest #file_information_class field to the type passed as argument' do
+      type = RubySMB::Fscc::FileInformation::FileDirectoryInformation
+      allow(client).to receive(:send_recv) do |packet|
+        expect(packet.file_information_class).to eq type::CLASS_LEVEL
+      end
+      tree.list(type: type)
+    end
+
+    it 'calls #set_header_fields' do
+      expect(tree).to receive(:set_header_fields).with(query_dir_req).and_call_original
+      tree.list
+    end
+
+    let(:file1) { double('file information') }
+    let(:file2) { double('file information') }
+
+    it 'returns the expected file information' do
+      query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_SUCCESS.value
+      allow(query_dir_res).to receive(:results) do |_type|
+        query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_NO_MORE_FILES.value
+        [file1]
+      end
+      expect(tree.list).to eq([file1])
+    end
+
+    context 'when multiple requests are needed to retrieve the full directory list' do
+      before :example do
+        query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_SUCCESS.value
+        first_query = true
+        allow(query_dir_res).to receive(:results) do |_type|
+          if first_query
+            first_query = false
+            [file1]
+          else
+            query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_NO_MORE_FILES.value
+            [file2]
+          end
+        end
+      end
+
+      it 'returns the expected file information' do
+        expect(tree.list).to eq([file1] + [file2])
+      end
+
+      it 'resets the message ID between the requests' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.smb2_header.message_id).to eq 0
+          packet.smb2_header.message_id = 1
+        end
+        tree.list
+      end
+    end
+
+    context 'when an unexpected status code is received' do
+      it 'raises an exception' do
+        query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_FILE_NOT_AVAILABLE.value
+        expect { tree.list }.to raise_error(RubySMB::Error::UnexpectedStatusCode)
+      end
+    end
+
+  end
 end