ruby_smb 0.0.8

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

@@ -0,0 +1,100 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::SessionSetupRequest 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_SESSION_SETUP
+    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::ParameterBlock
+    end
+
+    it { is_expected.to respond_to :andx_block }
+    it { is_expected.to respond_to :max_buffer_size }
+    it { is_expected.to respond_to :max_mpx_count }
+    it { is_expected.to respond_to :vc_number }
+    it { is_expected.to respond_to :session_key }
+    it { is_expected.to respond_to :security_blob_length }
+    it { is_expected.to respond_to :capabilities }
+
+    it 'has an AndXBlock' do
+      expect(parameter_block.andx_block).to be_a RubySMB::SMB1::AndXBlock
+    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 :security_blob }
+    it { is_expected.to respond_to :native_os }
+    it { is_expected.to respond_to :native_lan_man }
+
+  end
+
+  describe '#set_type1_blob' do
+    let(:fake_message) { "foo" }
+
+    it 'calls the #gss_type1 method to create a blob' do
+      expect(RubySMB::Gss).to receive(:gss_type1).with(fake_message).and_return(fake_message)
+      packet.set_type1_blob(fake_message)
+    end
+
+    it 'sets the security blob to the result from the GSS call' do
+      expect(RubySMB::Gss).to receive(:gss_type1).with(fake_message).and_return(fake_message)
+      packet.set_type1_blob(fake_message)
+      expect(packet.data_block.security_blob).to eq fake_message
+    end
+
+    it 'sets the security_blob_length field automatically' do
+      expect(RubySMB::Gss).to receive(:gss_type1).with(fake_message).and_return(fake_message)
+      packet.set_type1_blob(fake_message)
+      expect(packet.parameter_block.security_blob_length).to eq fake_message.length
+    end
+  end
+
+  describe '#set_type3_blob' do
+    let(:fake_message) { "foo" }
+
+    it 'calls the #gss_type3 method to create a blob' do
+      expect(RubySMB::Gss).to receive(:gss_type3).with(fake_message).and_return(fake_message)
+      packet.set_type3_blob(fake_message)
+    end
+
+    it 'sets the security blob to the result from the GSS call' do
+      expect(RubySMB::Gss).to receive(:gss_type3).with(fake_message).and_return(fake_message)
+      packet.set_type3_blob(fake_message)
+      expect(packet.data_block.security_blob).to eq fake_message
+    end
+
+    it 'sets the security_blob_length field automatically' do
+      expect(RubySMB::Gss).to receive(:gss_type3).with(fake_message).and_return(fake_message)
+      packet.set_type3_blob(fake_message)
+      expect(packet.parameter_block.security_blob_length).to eq fake_message.length
+    end
+  end
+
+
+
+end

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

@@ -0,0 +1,72 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::SessionSetupResponse 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_SESSION_SETUP
+    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 'is a standard ParameterBlock' do
+      expect(parameter_block).to be_a RubySMB::SMB1::ParameterBlock
+    end
+
+    it { is_expected.to respond_to :andx_block }
+    it { is_expected.to respond_to :action }
+    it { is_expected.to respond_to :security_blob_length }
+
+    it 'has an AndXBlock' do
+      expect(parameter_block.andx_block).to be_a RubySMB::SMB1::AndXBlock
+    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 :security_blob }
+    it { is_expected.to respond_to :native_os }
+    it { is_expected.to respond_to :native_lan_man }
+  end
+
+  describe '#set_type2_blob' do
+    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)
+      packet.set_type2_blob(fake_message)
+    end
+
+    it 'sets the security blob to the result from the GSS call' do
+      expect(RubySMB::Gss).to receive(:gss_type2).with(fake_message).and_return(fake_message)
+      packet.set_type2_blob(fake_message)
+      expect(packet.data_block.security_blob).to eq fake_message
+    end
+
+    it 'sets the security_blob_length field automatically' do
+      expect(RubySMB::Gss).to receive(:gss_type2).with(fake_message).and_return(fake_message)
+      packet.set_type2_blob(fake_message)
+      expect(packet.parameter_block.security_blob_length).to eq fake_message.length
+    end
+  end
+
+end

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

@@ -0,0 +1,26 @@
+RSpec.describe RubySMB::SMB1::ParameterBlock do
+  subject(:parameter_block) { described_class.new }
+
+  it { is_expected.to respond_to :word_count }
+
+  describe 'byte_count' do
+    it 'should be a 8-bit field per the SMB spec' do
+      expect(parameter_block.word_count).to be_a BinData::Uint8
+    end
+
+    it 'should equal the size of the rest of the block in words' do
+      remaining_size = ((parameter_block.do_num_bytes - 1) / 2).ceil
+      expect(parameter_block.word_count).to eq remaining_size
+    end
+  end
+
+  describe 'class method #calculate_word_count' do
+    it 'always returns 0' do
+      expect(described_class.calculate_word_count).to eq 0
+    end
+  end
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+end

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

@@ -0,0 +1,96 @@
+RSpec.describe RubySMB::SMB1::SMBHeader do
+  subject(:header) { described_class.new }
+
+  it { is_expected.to respond_to :protocol }
+  it { is_expected.to respond_to :command }
+  it { is_expected.to respond_to :nt_status }
+  it { is_expected.to respond_to :flags }
+
+  it { is_expected.to respond_to :pid_high }
+  it { is_expected.to respond_to :security_features }
+  it { is_expected.to respond_to :reserved }
+  it { is_expected.to respond_to :tid }
+  it { is_expected.to respond_to :pid_low }
+  it { is_expected.to respond_to :uid }
+  it { is_expected.to respond_to :mid }
+
+  describe 'protocol' do
+    it 'should be a 32-bit field per the SMB spec' do
+      expect(header.protocol).to be_a BinData::Bit32
+    end
+
+    it 'should be hardcoded to SMB_PROTOCOL_ID by default per the SMB spec' do
+      expect(header.protocol).to eq RubySMB::SMB1::SMB_PROTOCOL_ID
+    end
+  end
+
+  describe 'command' do
+    it 'should be a 8-bit field per the SMB spec' do
+      expect(header.command).to be_a BinData::Bit8
+    end
+  end
+
+  describe 'nt_status' do
+    it 'should be a NTStatus field' do
+      expect(header.nt_status).to be_a RubySMB::Field::NtStatus
+    end
+  end
+
+  describe 'flags' do
+    it 'should be a HeaderFlags BitField' do
+      expect(header.flags).to be_a RubySMB::SMB1::BitField::HeaderFlags
+    end
+  end
+
+  describe 'flags2' do
+    it 'should be a HeaderFlags2 BitField' do
+      expect(header.flags2).to be_a RubySMB::SMB1::BitField::HeaderFlags2
+    end
+  end
+
+  describe 'pid_high' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.pid_high).to be_a BinData::Bit16
+    end
+  end
+
+  describe 'security_features' do
+    it 'should be a 8-byte string per the SMB spec' do
+      expect(header.security_features).to be_a BinData::String
+    end
+  end
+
+  describe 'reserved' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.reserved).to be_a BinData::Bit16
+    end
+  end
+
+  describe 'tid' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.tid).to be_a BinData::Bit16
+    end
+  end
+
+  describe 'pid_low' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.pid_low).to be_a BinData::Bit16
+    end
+  end
+
+  describe 'uid' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.uid).to be_a BinData::Bit16
+    end
+  end
+
+  describe 'mid' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(header.mid).to be_a BinData::Bit16
+    end
+  end
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+end

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

@@ -0,0 +1,81 @@
+RSpec.describe RubySMB::SMB2::BitField::Smb2HeaderFlags do
+  subject(:flags) { described_class.new }
+
+  it { is_expected.to respond_to :reserved1 }
+  it { is_expected.to respond_to :replay_operation }
+  it { is_expected.to respond_to :dfs_operation }
+  it { is_expected.to respond_to :reserved2 }
+  it { is_expected.to respond_to :reserved3 }
+  it { is_expected.to respond_to :signed }
+  it { is_expected.to respond_to :related_operations }
+  it { is_expected.to respond_to :async_command }
+  it { is_expected.to respond_to :reply }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#reserved1' do
+    it 'should be a 2-bit field per the SMB spec' do
+      expect(flags.reserved1).to be_a BinData::Bit2
+    end
+  end
+
+  describe '#replay_operation' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.replay_operation).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :replay_operation, 'V', 0x20000000
+  end
+
+  describe '#dfs_operation' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.dfs_operation).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :dfs_operation, 'V', 0x10000000
+  end
+
+  describe '#reserved2' do
+    it 'should be a 2-byte field per the SMB spec' do
+      expect(flags.reserved2).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#reserved3' do
+    it 'should be a 4-bit field per the SMB spec' do
+      expect(flags.reserved3).to be_a BinData::Bit4
+    end
+  end
+
+  describe '#signed' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.signed).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :signed, 'V', 0x00000008
+  end
+
+  describe '#related_operations' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.related_operations).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :related_operations, 'V', 0x00000004
+  end
+
+  describe '#async_command' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.async_command).to be_a BinData::Bit1
+    end
+  end
+
+  describe '#reply' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.reply).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :reply, 'V', 0x00000001
+  end
+end

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

@@ -0,0 +1,28 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::BitField::SessionFlags do
+  subject(:flags) { described_class.new }
+
+  it { is_expected.to respond_to :guest }
+  it { is_expected.to respond_to :null }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#guest' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.guest).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :guest, 'v', 0x00000001
+  end
+
+  describe '#null' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.null).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :null, 'v', 0x00000002
+  end
+end

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

@@ -0,0 +1,72 @@
+RSpec.describe RubySMB::SMB2::BitField::Smb2Capabilities do
+  subject(:capabilities) { described_class.new }
+
+  it { is_expected.to respond_to :reserved2 }
+  it { is_expected.to respond_to :encryption }
+  it { is_expected.to respond_to :directory_leasing }
+  it { is_expected.to respond_to :persistent_handles }
+  it { is_expected.to respond_to :multi_channel }
+  it { is_expected.to respond_to :large_mtu }
+  it { is_expected.to respond_to :leasing }
+  it { is_expected.to respond_to :dfs }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#encryption' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.encryption).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :encryption, 'V', 0x00000040
+  end
+
+  describe '#directory_leasing' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.directory_leasing).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :directory_leasing, 'V', 0x00000020
+  end
+
+  describe '#persistent_handles' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.persistent_handles).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :persistent_handles, 'V', 0x00000010
+  end
+
+  describe '#multi_channel' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.multi_channel).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :multi_channel, 'V', 0x00000008
+  end
+
+  describe '#large_mtu' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.large_mtu).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :large_mtu, 'V', 0x00000004
+  end
+
+  describe '#leasing' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.leasing).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :leasing, 'V', 0x00000002
+  end
+
+  describe '#dfs' do
+    it 'is a 1-bit flag' do
+      expect(capabilities.dfs).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :dfs, 'V', 0x00000001
+  end
+end

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

@@ -0,0 +1,22 @@
+RSpec.describe RubySMB::SMB2::BitField::Smb2SecurityMode do
+  subject(:security_mode) { described_class.new }
+
+  it { is_expected.to respond_to :signing_required }
+  it { is_expected.to respond_to :signing_enabled }
+
+  describe '#signing_required' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.signing_required).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :signing_required, 'v', 0x0002
+  end
+
+  describe '#signing_enabled' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.signing_enabled).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :signing_enabled, 'v', 0x0001
+  end
+end

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

@@ -0,0 +1,122 @@
+RSpec.describe RubySMB::SMB2::Packet::NegotiateRequest 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 :dialect_count }
+  it { is_expected.to respond_to :security_mode }
+  it { is_expected.to respond_to :reserved1 }
+  it { is_expected.to respond_to :capabilities }
+  it { is_expected.to respond_to :client_guid }
+  it { is_expected.to respond_to :client_start_time }
+  it { is_expected.to respond_to :dialects }
+
+  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::NEGOTIATE
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  end
+
+  describe '#structure_size' do
+    it 'should be a 16-bit unsigned integer' do
+      expect(packet.structure_size).to be_a BinData::Uint16le
+    end
+
+    it 'should have a default value of 36 as per the SMB2 spec' do
+      expect(packet.structure_size).to eq 36
+    end
+  end
+
+  describe '#dialect_count' do
+    it 'should be a 16-bit unsigned integer' do
+      expect(packet.dialect_count).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#security_mode' do
+    it 'should be a SMB2 Security Mode BitField' do
+      expect(packet.security_mode).to be_a RubySMB::SMB2::BitField::Smb2SecurityMode
+    end
+  end
+
+  describe '#capabailities' do
+    it 'should be a SMB2 Capabilities BitField' do
+      expect(packet.capabilities).to be_a RubySMB::SMB2::BitField::Smb2Capabilities
+    end
+  end
+
+  describe '#client_guid' do
+    it 'should be a binary string' do
+      expect(packet.client_guid).to be_a BinData::String
+    end
+
+    it 'should be 16-bytes' do
+      expect(packet.client_guid.do_num_bytes).to eq 16
+    end
+  end
+
+  describe '#client_start_time' do
+    it 'should be a Filetime field' do
+      expect(packet.client_start_time).to be_a RubySMB::Field::FileTime
+    end
+
+    it 'should have a default value of 0 as per the SMB2 spec' do
+      expect(packet.client_start_time).to eq 0
+    end
+  end
+
+  describe '#dialects' do
+    it 'is an array field as per the SMB spec' do
+      expect(packet.dialects).to be_a BinData::Array
+    end
+  end
+
+  describe '#add_dialect' do
+    it 'adds the dialect to the Dialects array' do
+      packet.add_dialect 0x0201
+      expect(packet.dialects).to include(0x0201)
+    end
+
+    it 'updates the #dialect_count field' do
+      packet.add_dialect 0x0201
+      expect(packet.dialect_count).to eq 1
+    end
+  end
+
+  describe '#set_dialects' do
+    before(:each) do
+      packet.add_dialect 0x0201
+    end
+
+    let(:dialect_set) { [0x0202, 0x0210, 0x0300] }
+
+    it 'removes the existing dialects' do
+      packet.set_dialects dialect_set
+      expect(packet.dialects).to_not include(0x0201)
+    end
+
+    it 'sets #dialects to exacty what is supplied' do
+      packet.set_dialects dialect_set
+      expect(packet.dialects).to match_array(dialect_set)
+    end
+
+    it 'sets the #dialect_count field correctly' do
+      packet.set_dialects dialect_set
+      expect(packet.dialect_count).to eq 3
+    end
+  end
+end