ruby_smb 0.0.8 → 0.0.9

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

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::TreeDisconnectResponse 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_TREE_DISCONNECT
+    end
+
+    it 'should not 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 'is a standard ParameterBlock' do
+      expect(parameter_block).to be_a RubySMB::SMB1::ParameterBlock
+    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
+  end
+
+
+end

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

@@ -0,0 +1,55 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Tree do
+  let(:dispatcher) { RubySMB::Dispatcher::Socket.new(nil) }
+
+  let(:client) { RubySMB::Client.new(dispatcher, username: 'msfadmin', password: 'msfadmin') }
+  let(:tree_id) { 2049 }
+  let(:path) { '\\192.168.1.1\example' }
+  let(:response) {
+    packet = RubySMB::SMB1::Packet::TreeConnectResponse.new
+    packet.smb_header.tid = tree_id
+    packet.parameter_block.access_rights.read("\xff\x01\x1f\x00")
+    packet.data_block.service = 'A:'
+    packet
+  }
+  subject(:tree) {
+    described_class.new(client:client, share:path, response:response )
+  }
+
+  it { is_expected.to respond_to :client }
+  it { is_expected.to respond_to :guest_permissions }
+  it { is_expected.to respond_to :permissions }
+  it { is_expected.to respond_to :share }
+  it { is_expected.to respond_to :id }
+
+  it 'inherits the client that spawned it' do
+    expect(tree.client).to eq client
+  end
+
+  it 'inherits the permissions from the response packet' do
+    expect(tree.permissions).to eq response.parameter_block.access_rights
+  end
+
+  it 'inherits the Tree id from the response packet' do
+    expect(tree.id).to eq response.smb_header.tid
+  end
+
+  describe '#disconnect!' do
+    let(:disco_req) { RubySMB::SMB1::Packet::TreeDisconnectRequest.new }
+    let(:disco_resp) { RubySMB::SMB1::Packet::TreeDisconnectResponse.new }
+
+    it 'sends a TreeDisconnectRequest with the Tree ID in the header' do
+      allow(RubySMB::SMB1::Packet::TreeDisconnectRequest).to receive(:new).and_return(disco_req)
+      modified_req = disco_req
+      modified_req.smb_header.tid = tree.id
+      expect(client).to receive(:send_recv).with(modified_req).and_return(disco_resp.to_binary_s)
+      tree.disconnect!
+    end
+
+    it 'returns the NTStatus code from the response' do
+      allow(client).to receive(:send_recv).and_return(disco_resp.to_binary_s)
+      expect(tree.disconnect!).to eq disco_resp.status_code
+    end
+  end
+end

data/spec/lib/ruby_smb/smb2/bit_field/directory_access_mask_spec.rb

@@ -0,0 +1,190 @@
+RSpec.describe RubySMB::SMB2::BitField::DirectoryAccessMask do
+  subject(:flags) { described_class.new }
+
+  it { is_expected.to respond_to :read_attr }
+  it { is_expected.to respond_to :delete_child }
+  it { is_expected.to respond_to :traverse }
+  it { is_expected.to respond_to :write_ea }
+  it { is_expected.to respond_to :read_ea }
+  it { is_expected.to respond_to :add_subdir }
+  it { is_expected.to respond_to :add_file }
+  it { is_expected.to respond_to :list }
+  it { is_expected.to respond_to :write_attr }
+  it { is_expected.to respond_to :synchronize }
+  it { is_expected.to respond_to :write_owner }
+  it { is_expected.to respond_to :write_dac }
+  it { is_expected.to respond_to :read_control }
+  it { is_expected.to respond_to :delete_access }
+  it { is_expected.to respond_to :generic_read }
+  it { is_expected.to respond_to :generic_write }
+  it { is_expected.to respond_to :generic_execute }
+  it { is_expected.to respond_to :generic_all }
+  it { is_expected.to respond_to :maximum }
+  it { is_expected.to respond_to :system_security }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#list' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.list).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :list, 'V', 0x00000001
+  end
+
+
+  describe '#add_file' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.add_file).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :add_file, 'V', 0x00000002
+  end
+
+  describe '#add_subdir' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.add_subdir).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :add_subdir, 'V', 0x00000004
+  end
+
+  describe '#read_ea' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_ea).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_ea, 'V', 0x00000008
+  end
+
+  describe '#write_ea' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_ea).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_ea, 'V', 0x00000010
+  end
+
+  describe '#traverse' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.traverse).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :traverse, 'V', 0x00000020
+  end
+
+  describe '#delete_child' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.delete_child).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :delete_child, 'V', 0x00000040
+  end
+
+  describe '#read_attr' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_attr).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_attr, 'V', 0x00000080
+  end
+
+  describe '#write_attr' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_attr).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_attr, 'V', 0x00000100
+  end
+
+  describe '#delete_access' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.delete_access).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :delete_access, 'V', 0x00010000
+  end
+
+  describe '#read_control' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_control).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_control, 'V', 0x00020000
+  end
+
+  describe '#write_dac' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_dac).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_dac, 'V', 0x00040000
+  end
+
+  describe '#write_owner' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_owner).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_owner, 'V', 0x00080000
+  end
+
+  describe '#synchronize' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.synchronize).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :synchronize, 'V', 0x00100000
+  end
+
+  describe '#system_security' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.system_security).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :system_security, 'V', 0x01000000
+  end
+
+  describe '#maximum' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.maximum).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :maximum, 'V', 0x02000000
+  end
+
+  describe '#generic_all' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_all).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_all, 'V', 0x10000000
+  end
+
+  describe '#generic_execute' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_execute).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_execute, 'V', 0x20000000
+  end
+
+  describe '#generic_write' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_write).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_write, 'V', 0x40000000
+  end
+
+  describe '#generic_read' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_read).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_read, 'V', 0x80000000
+  end
+
+end

data/spec/lib/ruby_smb/smb2/bit_field/file_access_mask_spec.rb

@@ -0,0 +1,190 @@
+RSpec.describe RubySMB::SMB2::BitField::FileAccessMask do
+  subject(:flags) { described_class.new }
+
+  it { is_expected.to respond_to :read_attr }
+  it { is_expected.to respond_to :delete_child }
+  it { is_expected.to respond_to :execute }
+  it { is_expected.to respond_to :write_ea }
+  it { is_expected.to respond_to :read_ea }
+  it { is_expected.to respond_to :append_data }
+  it { is_expected.to respond_to :write_data }
+  it { is_expected.to respond_to :read_data }
+  it { is_expected.to respond_to :write_attr }
+  it { is_expected.to respond_to :synchronize }
+  it { is_expected.to respond_to :write_owner }
+  it { is_expected.to respond_to :write_dac }
+  it { is_expected.to respond_to :read_control }
+  it { is_expected.to respond_to :delete_access }
+  it { is_expected.to respond_to :generic_read }
+  it { is_expected.to respond_to :generic_write }
+  it { is_expected.to respond_to :generic_execute }
+  it { is_expected.to respond_to :generic_all }
+  it { is_expected.to respond_to :maximum }
+  it { is_expected.to respond_to :system_security }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#read_data' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_data).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_data, 'V', 0x00000001
+  end
+
+
+  describe '#write_data' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_data).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_data, 'V', 0x00000002
+  end
+
+  describe '#append_data' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.append_data).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :append_data, 'V', 0x00000004
+  end
+
+  describe '#read_ea' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_ea).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_ea, 'V', 0x00000008
+  end
+
+  describe '#write_ea' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_ea).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_ea, 'V', 0x00000010
+  end
+
+  describe '#execute' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.execute).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :execute, 'V', 0x00000020
+  end
+
+  describe '#delete_child' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.delete_child).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :delete_child, 'V', 0x00000040
+  end
+
+  describe '#read_attr' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_attr).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_attr, 'V', 0x00000080
+  end
+
+  describe '#write_attr' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_attr).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_attr, 'V', 0x00000100
+  end
+
+  describe '#delete_access' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.delete_access).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :delete_access, 'V', 0x00010000
+  end
+
+  describe '#read_control' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.read_control).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :read_control, 'V', 0x00020000
+  end
+
+  describe '#write_dac' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_dac).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_dac, 'V', 0x00040000
+  end
+
+  describe '#write_owner' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.write_owner).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :write_owner, 'V', 0x00080000
+  end
+
+  describe '#synchronize' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.synchronize).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :synchronize, 'V', 0x00100000
+  end
+
+  describe '#system_security' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.system_security).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :system_security, 'V', 0x01000000
+  end
+
+  describe '#maximum' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.maximum).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :maximum, 'V', 0x02000000
+  end
+
+  describe '#generic_all' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_all).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_all, 'V', 0x10000000
+  end
+
+  describe '#generic_execute' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_execute).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_execute, 'V', 0x20000000
+  end
+
+  describe '#generic_write' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_write).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_write, 'V', 0x40000000
+  end
+
+  describe '#generic_read' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.generic_read).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :generic_read, 'V', 0x80000000
+  end
+
+end