ruby_smb 0.0.8

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

@@ -0,0 +1,102 @@
+RSpec.describe RubySMB::SMB1::BitField::HeaderFlags do
+  subject(:flags) { described_class.new }
+
+  it { is_expected.to respond_to :reply }
+  it { is_expected.to respond_to :opbatch }
+  it { is_expected.to respond_to :oplock }
+  it { is_expected.to respond_to :canonicalized_paths }
+  it { is_expected.to respond_to :case_insensitive }
+  it { is_expected.to respond_to :reserved }
+  it { is_expected.to respond_to :buf_avail }
+  it { is_expected.to respond_to :lock_and_read_ok }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  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, 'C', 0x80
+  end
+
+  describe '#opbatch' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.opbatch).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 0' do
+      expect(flags.opbatch).to eq 0
+    end
+
+    it_behaves_like 'bit field with one flag set', :opbatch, 'C', 0x40
+  end
+
+  describe '#oplock' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.oplock).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 0' do
+      expect(flags.oplock).to eq 0
+    end
+
+    it_behaves_like 'bit field with one flag set', :oplock, 'C', 0x20
+  end
+
+  describe '#canonicalized_paths' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.canonicalized_paths).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 1' do
+      expect(flags.canonicalized_paths).to eq 1
+    end
+
+    it_behaves_like 'bit field with one flag set', :canonicalized_paths, 'C', 0x10
+  end
+
+  describe '#case_insensitive' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.case_insensitive).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 1' do
+      expect(flags.case_insensitive).to eq 1
+    end
+
+    it_behaves_like 'bit field with one flag set', :case_insensitive, 'C', 0x08
+  end
+
+  describe '#reserved' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.reserved).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 0' do
+      expect(flags.reserved).to eq 0
+    end
+  end
+
+  describe '#buf_avail' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.buf_avail).to be_a BinData::Bit1
+    end
+
+    it 'should have a default value of 0' do
+      expect(flags.buf_avail).to eq 0
+    end
+
+    it_behaves_like 'bit field with one flag set', :buf_avail, 'C', 0x02
+  end
+
+  describe '#lock_and_read_ok' do
+    it 'should be a 1-bit field per the SMB spec' do
+      expect(flags.lock_and_read_ok).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :lock_and_read_ok, 'C', 0x01
+  end
+end

data/spec/lib/ruby_smb/smb1/bit_field/security_mode_spec.rb

@@ -0,0 +1,44 @@
+RSpec.describe RubySMB::SMB1::BitField::SecurityMode do
+  subject(:security_mode) { described_class.new }
+
+  it { is_expected.to respond_to :user_security }
+  it { is_expected.to respond_to :encrypt_passwords }
+  it { is_expected.to respond_to :security_signatures_enabled }
+  it { is_expected.to respond_to :security_signatures_required }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#user_security' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.user_security).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :user_security, 'C', 0x01
+  end
+
+  describe '#encrypt_passwords' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.encrypt_passwords).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :encrypt_passwords, 'C', 0x02
+  end
+
+  describe '#security_signatures_enabled' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.security_signatures_enabled).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :security_signatures_enabled, 'C', 0x04
+  end
+
+  describe '#security_signatures_required' do
+    it 'is a 1-bit flag' do
+      expect(security_mode.security_signatures_required).to be_a BinData::Bit1
+    end
+
+    it_behaves_like 'bit field with one flag set', :security_signatures_required, 'C', 0x08
+  end
+end

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

@@ -0,0 +1,26 @@
+RSpec.describe RubySMB::SMB1::DataBlock do
+  subject(:data_block) { described_class.new }
+
+  it { is_expected.to respond_to :byte_count }
+
+  describe 'byte_count' do
+    it 'should be a 16-bit field per the SMB spec' do
+      expect(data_block.byte_count).to be_a BinData::Uint16le
+    end
+
+    it 'should equal the size of the rest of the block in bytes' do
+      remaining_size = data_block.do_num_bytes - 2
+      expect(data_block.byte_count).to eq remaining_size
+    end
+  end
+
+  describe 'class method #calculate_byte_count' do
+    it 'always returns 0' do
+      expect(described_class.calculate_byte_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/dialect_spec.rb

@@ -0,0 +1,26 @@
+RSpec.describe RubySMB::SMB1::Dialect do
+  subject(:dialect) { described_class.new }
+
+  it { is_expected.to respond_to :buffer_format }
+  it { is_expected.to respond_to :dialect_string }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe 'buffer_format' do
+    it 'should be a 8-bit field per the SMB spec' do
+      expect(dialect.buffer_format).to be_a BinData::Bit8
+    end
+
+    it 'should be hardcoded to 0x2 by default per the SMB spec' do
+      expect(dialect.buffer_format).to eq 0x2
+    end
+  end
+
+  describe 'dialect_string' do
+    it 'should be a null terminated string per the SMB spec' do
+      expect(dialect.dialect_string).to be_a BinData::Stringz
+    end
+  end
+end

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

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::ErrorPacket 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
+
+  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 'should be empty' do
+      expect(parameter_block.to_binary_s).to eq "\x00"
+    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 'should be empty' do
+      expect(data_block.to_binary_s).to eq "\x00\x00"
+    end
+  end
+
+end

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

@@ -0,0 +1,77 @@
+RSpec.describe RubySMB::SMB1::Packet::NegotiateRequest do
+  subject(:packet) { described_class.new }
+  let(:dialect_string) { 'NT LM 0.12' }
+
+  before(:each) do
+    packet.add_dialect(dialect_string)
+  end
+
+  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_NEGOTIATE
+    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
+  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 :dialects }
+
+    describe '#dialects' do
+      it 'is an array field as per the SMB spec' do
+        expect(data_block.dialects).to be_a BinData::Array
+      end
+    end
+  end
+
+  describe '#add_dialect' do
+    it 'adds a Dialect to the packet' do
+      expect { packet.add_dialect('foo') }.to change { packet.data_block.dialects.count }.by(1)
+    end
+
+    it 'uses the argument as the Dialect String' do
+      packet.add_dialect('bar')
+      dialects = packet.data_block.dialects.to_a
+      expect(dialects.last.dialect_string).to eq 'bar'
+    end
+  end
+
+  describe '#set_dialects' do
+    it 'clears out any existing dialects' do
+      expect { packet.set_dialects([]) }.to change { packet.data_block.dialects.count }.to(0)
+    end
+
+    it 'calls #add_dialect once for each string in the array' do
+      expect(packet).to receive(:add_dialect).exactly(3).times
+      packet.set_dialects(%w(foo bar baz))
+    end
+  end
+
+  describe '#dialects' do
+    it 'returns a ruby array of the dialect hashes' do
+      expect(packet.dialects).to eq [{:buffer_format=>2, :dialect_string=>"NT LM 0.12"}]
+    end
+  end
+end

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

@@ -0,0 +1,149 @@
+RSpec.describe RubySMB::SMB1::Packet::NegotiateResponseExtended 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_NEGOTIATE
+    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 :dialect_index }
+    it { is_expected.to respond_to :security_mode }
+    it { is_expected.to respond_to :max_mpx_count }
+    it { is_expected.to respond_to :max_number_vcs }
+    it { is_expected.to respond_to :max_buffer_size }
+    it { is_expected.to respond_to :max_raw_size }
+    it { is_expected.to respond_to :session_key }
+    it { is_expected.to respond_to :capabilities }
+    it { is_expected.to respond_to :system_time }
+    it { is_expected.to respond_to :server_time_zone }
+    it { is_expected.to respond_to :challenge_length }
+
+    describe '#dialect_index' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.dialect_index).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#security_mode' do
+      it 'is a SecurityMode bit-field' do
+        expect(parameter_block.security_mode).to be_a RubySMB::SMB1::BitField::SecurityMode
+      end
+    end
+
+    describe '#max_mpx_count' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.max_mpx_count).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#max_number_vcs' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.max_number_vcs).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#max_buffer_size' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.max_buffer_size).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#max_raw_size' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.max_raw_size).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#session_key' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.session_key).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#capabilities' do
+      it 'is a Capabilities bit-field' do
+        expect(parameter_block.capabilities).to be_a RubySMB::SMB1::BitField::Capabilities
+      end
+    end
+
+    describe '#system_time' do
+      it 'is a FileTime field' do
+        expect(parameter_block.system_time).to be_a RubySMB::Field::FileTime
+      end
+    end
+
+    describe '#server_time_zone' do
+      it 'is a 16-bit Signed Integer' do
+        expect(parameter_block.server_time_zone).to be_a BinData::Int16le
+      end
+    end
+
+    describe '#challenge_length' do
+      it 'is a 8-bit Unsigned Integer' do
+        expect(parameter_block.challenge_length).to be_a BinData::Uint8
+      end
+    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 :server_guid }
+    it { is_expected.to respond_to :security_blob }
+
+    describe '#server_guid' do
+      it 'is binary string field' do
+        expect(data_block.server_guid).to be_a BinData::String
+      end
+
+      it 'is exactly 16-bytes long' do
+        expect(data_block.server_guid.length).to eq 16
+      end
+    end
+
+    describe '#security_blob' do
+      it 'is a variable length "rest" field' do
+        expect(data_block.security_blob).to be_a BinData::Rest
+      end
+    end
+  end
+
+  describe '#valid?' do
+    it 'should return true if the command value ix 0x72' do
+      packet.parameter_block.capabilities.extended_security = 1
+      expect(packet.valid?).to be true
+    end
+
+    it 'should return false if the command value is not 0x72' do
+      packet.smb_header.command = 0xff
+      expect(packet.valid?).to be false
+    end
+
+    it 'should return false if the capabilities do not include extended security' do
+      packet.parameter_block.capabilities.extended_security = 0
+      expect(packet.valid?).to be false
+    end
+  end
+end

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

@@ -0,0 +1,150 @@
+RSpec.describe RubySMB::SMB1::Packet::NegotiateResponse 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_NEGOTIATE
+    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 :dialect_index }
+    it { is_expected.to respond_to :security_mode }
+    it { is_expected.to respond_to :max_mpx_count }
+    it { is_expected.to respond_to :max_number_vcs }
+    it { is_expected.to respond_to :max_buffer_size }
+    it { is_expected.to respond_to :max_raw_size }
+    it { is_expected.to respond_to :session_key }
+    it { is_expected.to respond_to :capabilities }
+    it { is_expected.to respond_to :system_time }
+    it { is_expected.to respond_to :server_time_zone }
+    it { is_expected.to respond_to :challenge_length }
+
+    describe '#dialect_index' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.dialect_index).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#security_mode' do
+      it 'is a SecurityMode bit-field' do
+        expect(parameter_block.security_mode).to be_a RubySMB::SMB1::BitField::SecurityMode
+      end
+    end
+
+    describe '#max_mpx_count' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.max_mpx_count).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#max_number_vcs' do
+      it 'is a 16-bit Unsigned Integer' do
+        expect(parameter_block.max_number_vcs).to be_a BinData::Uint16le
+      end
+    end
+
+    describe '#max_buffer_size' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.max_buffer_size).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#max_raw_size' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.max_raw_size).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#session_key' do
+      it 'is a 32-bit Unsigned Integer' do
+        expect(parameter_block.session_key).to be_a BinData::Uint32le
+      end
+    end
+
+    describe '#capabilities' do
+      it 'is a Capabilities bit-field' do
+        expect(parameter_block.capabilities).to be_a RubySMB::SMB1::BitField::Capabilities
+      end
+    end
+
+    describe '#system_time' do
+      it 'is a FileTime field' do
+        expect(parameter_block.system_time).to be_a RubySMB::Field::FileTime
+      end
+    end
+
+    describe '#server_time_zone' do
+      it 'is a 16-bit Signed Integer' do
+        expect(parameter_block.server_time_zone).to be_a BinData::Int16le
+      end
+    end
+
+    describe '#challenge_length' do
+      it 'is a 8-bit Unsigned Integer' do
+        expect(parameter_block.challenge_length).to be_a BinData::Uint8
+      end
+    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 :challenge }
+    it { is_expected.to respond_to :domain_name }
+    it { is_expected.to respond_to :server_name }
+
+    describe '#challenge' do
+      it 'is a sized string of bytes' do
+        expect(data_block.challenge).to be_a BinData::String
+      end
+
+      it 'is exactly 8-bytes long' do
+        expect(data_block.challenge.length).to eq 8
+      end
+    end
+
+    describe '#domain_name' do
+      it 'is a Unicode Null-terminated string' do
+        expect(data_block.domain_name).to be_a RubySMB::Field::Stringz16
+      end
+    end
+
+    describe '#server_name' do
+      it 'is a Unicode Null-terminated string' do
+        expect(data_block.server_name).to be_a RubySMB::Field::Stringz16
+      end
+    end
+  end
+
+  describe '#valid?' do
+    it 'should return true if the command value ix 0x72' do
+      expect(packet.valid?).to be true
+    end
+
+    it 'should return false if the command value is not 0x72' do
+      packet.smb_header.command = 0xff
+      expect(packet.valid?).to be false
+    end
+  end
+end