ruby_smb 0.0.18 → 0.0.19

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

@@ -0,0 +1,64 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::CreateResponse 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::CREATE
+    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 89' do
+    expect(packet.structure_size).to eq 89
+  end
+
+  it 'tracks the creation time in a Filetime field' do
+    expect(packet.create_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last access time in a Filetime field' do
+    expect(packet.last_access).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last write time in a Filetime field' do
+    expect(packet.last_write).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last modified time in a Filetime field' do
+    expect(packet.last_change).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'contains the file attributes of the file' do
+    expect(packet.file_attributes).to be_a RubySMB::Fscc::FileAttributes
+  end
+
+  it 'has the handles to the file in an SMB2_FILEID' do
+    expect(packet.file_id).to be_a RubySMB::Field::Smb2Fileid
+  end
+
+  it 'tracks the offset to #context in #context_offset' do
+    expect(packet.context_offset).to eq packet.context.abs_offset
+  end
+
+  it 'tracks the length of #context in #context_length' do
+    expect(packet.context_length).to eq packet.context.do_num_bytes
+  end
+end

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

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

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

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

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

@@ -0,0 +1,48 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::IoctlResponse 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 :ctl_code }
+  it { is_expected.to respond_to :file_id }
+  it { is_expected.to respond_to :input_offset }
+  it { is_expected.to respond_to :input_count }
+  it { is_expected.to respond_to :output_offset }
+  it { is_expected.to respond_to :output_count }
+  it { is_expected.to respond_to :flags }
+  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_IOCTL' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::IOCTL
+    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 49
+  end
+
+  describe '#file_id' do
+    it 'should be an SMB FileID field' do
+      expect(packet.file_id).to be_a RubySMB::Field::Smb2Fileid
+    end
+  end
+
+
+end

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

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

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

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

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

@@ -0,0 +1,80 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::QueryDirectoryRequest 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_QUERY_DIRECTORY' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::QUERY_DIRECTORY
+    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 33' do
+    expect(packet.structure_size).to eq 33
+  end
+
+  describe '#flags' do
+    subject(:flags) { packet.flags }
+
+    describe '#restart_scans' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(flags.restart_scans).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :restart_scans, 'C', 0x01
+    end
+
+    describe '#return_single' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(flags.return_single).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :return_single, 'C', 0x02
+    end
+
+    describe '#index_specified' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(flags.index_specified).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :index_specified, 'C', 0x04
+    end
+
+    describe '#reopen' do
+      it 'should be a 1-bit field per the SMB spec' do
+        expect(flags.reopen).to be_a BinData::Bit1
+      end
+
+      it_behaves_like 'bit field with one flag set', :reopen, 'C', 0x10
+    end
+  end
+
+  it 'has an SMB2 FILEID field' do
+    expect(packet.file_id).to be_a RubySMB::Field::Smb2Fileid
+  end
+
+  it 'has an offset pointer to the name field' do
+    expect(packet.name_offset).to eq packet.name.abs_offset
+  end
+
+  it 'has a length value for the name field' do
+    expect(packet.name_length).to eq packet.name.do_num_bytes
+  end
+end

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

@@ -0,0 +1,64 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::QueryDirectoryResponse 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_QUERY_DIRECTORY' do
+      expect(header.command).to eq RubySMB::SMB2::Commands::QUERY_DIRECTORY
+    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 33' do
+    expect(packet.structure_size).to eq 9
+  end
+
+  it 'has an offset pointer to the buffer field' do
+    expect(packet.buffer_offset).to eq packet.buffer.abs_offset
+  end
+
+  it 'has a length value for the buffer field' do
+    expect(packet.buffer_length).to eq packet.buffer.do_num_bytes
+  end
+
+  describe '#results' do
+    let(:names1) {
+      names = RubySMB::Fscc::FileInformation::FileNamesInformation.new
+      names.file_name = 'test.txt'
+      names.next_offset = names.do_num_bytes
+      names
+    }
+
+    let(:names2) {
+      names = RubySMB::Fscc::FileInformation::FileNamesInformation.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 Fileinformation structs' do
+      packet.buffer = names_blob
+      expect(packet.results(RubySMB::Fscc::FileInformation::FileNamesInformation)).to eq names_array
+    end
+  end
+end

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

@@ -0,0 +1,43 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::ReadRequest 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 :padding }
+  it { is_expected.to respond_to :flags }
+  it { is_expected.to respond_to :read_length }
+  it { is_expected.to respond_to :offset }
+  it { is_expected.to respond_to :file_id }
+  it { is_expected.to respond_to :min_bytes }
+  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::READ
+    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
+end

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

@@ -0,0 +1,50 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::ReadResponse 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 :data_length }
+  it { is_expected.to respond_to :data_remaining }
+  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::READ
+    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 17
+  end
+
+  it 'stores the offset to the data buffer' do
+    expect(packet.data_offset).to eq packet.buffer.abs_offset
+  end
+
+  describe '#data_length' do
+    it 'sets the length of the actual buffer' do
+      packet.data_length = 12
+      packet.buffer = 'hello world!'
+      expect(packet.buffer.length).to eq 12
+      packet.data_length = 5
+      expect(packet.buffer).to eq 'hello'
+    end
+  end
+end

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

@@ -1,7 +1,6 @@
 require 'spec_helper'
 
 RSpec.describe RubySMB::SMB2::Packet::SessionSetupRequest do
-
   subject(:packet) { described_class.new }
 
   it { is_expected.to respond_to :smb2_header }
@@ -36,7 +35,7 @@ RSpec.describe RubySMB::SMB2::Packet::SessionSetupRequest do
   end
 
   describe '#set_type1_blob' do
-    let(:fake_message) { "foo" }
+    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)
@@ -57,7 +56,7 @@ RSpec.describe RubySMB::SMB2::Packet::SessionSetupRequest do
   end
 
   describe '#set_type3_blob' do
-    let(:fake_message) { "foo" }
+    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)
@@ -76,4 +75,4 @@ RSpec.describe RubySMB::SMB2::Packet::SessionSetupRequest do
       expect(packet.security_buffer_length).to eq fake_message.length
     end
   end
-end
+end