ruby_smb 0.0.8

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

@@ -0,0 +1,147 @@
+RSpec.describe RubySMB::SMB2::Packet::NegotiateResponse 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 :security_mode }
+  it { is_expected.to respond_to :dialect_revision }
+  it { is_expected.to respond_to :negotiate_context_count }
+  it { is_expected.to respond_to :server_guid }
+  it { is_expected.to respond_to :capabilities }
+  it { is_expected.to respond_to :max_transact_size }
+  it { is_expected.to respond_to :max_read_size }
+  it { is_expected.to respond_to :max_write_size }
+  it { is_expected.to respond_to :system_time }
+  it { is_expected.to respond_to :server_start_time }
+  it { is_expected.to respond_to :security_buffer_offset }
+  it { is_expected.to respond_to :security_buffer_length }
+  it { is_expected.to respond_to :negotiate_context_offset }
+  it { is_expected.to respond_to :security_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::NEGOTIATE
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    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 65 as per the SMB2 spec' do
+      expect(packet.structure_size).to eq 65
+    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 '#dialect_revision' do
+    it 'is a 16-bit unsigned integer' do
+      expect(packet.dialect_revision).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#negotiate_context_count' do
+    it 'is a 16-bit unsigned integer' do
+      expect(packet.negotiate_context_count).to be_a BinData::Uint16le
+    end
+
+    it 'has a default value of 0' do
+      expect(packet.negotiate_context_count).to eq 0
+    end
+  end
+
+  describe '#server_guid' do
+    it 'should be a binary string' do
+      expect(packet.server_guid).to be_a BinData::String
+    end
+
+    it 'should be 16-bytes' do
+      expect(packet.server_guid.do_num_bytes).to eq 16
+    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 '#max_transact_size' do
+    it 'is a 32-bit unsigned integer' do
+      expect(packet.max_transact_size).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#max_read_size' do
+    it 'is a 32-bit unsigned integer' do
+      expect(packet.max_read_size).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#max_write_size' do
+    it 'is a 32-bit unsigned integer' do
+      expect(packet.max_write_size).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#system_time' do
+    it 'should be a Filetime field' do
+      expect(packet.system_time).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#server_start_time' do
+    it 'should be a Filetime field' do
+      expect(packet.server_start_time).to be_a RubySMB::Field::FileTime
+    end
+  end
+
+  describe '#security_buffer_offset' do
+    it 'is a 16-bit unsigned integer' do
+      expect(packet.security_buffer_offset).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#security_buffer_length' do
+    it 'is a 16-bit unsigned integer' do
+      expect(packet.security_buffer_length).to be_a BinData::Uint16le
+    end
+
+    it 'should be the length of the security_buffer field' do
+      packet.security_buffer = 'foobar'
+      expect(packet.security_buffer_length).to eq 6
+    end
+  end
+
+  describe '#negotiate_context_offset' do
+    it 'is a 32-bit unsigned integer' do
+      expect(packet.negotiate_context_offset).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#security_buffer' do
+    it 'should be a binary string' do
+      expect(packet.security_buffer).to be_a BinData::String
+    end
+  end
+end

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

@@ -0,0 +1,79 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::SessionSetupRequest 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 :flags }
+  it { is_expected.to respond_to :security_mode }
+  it { is_expected.to respond_to :capabilities }
+  it { is_expected.to respond_to :channel }
+  it { is_expected.to respond_to :security_buffer_offset }
+  it { is_expected.to respond_to :security_buffer_length }
+  it { is_expected.to respond_to :previous_session_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 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::SESSION_SETUP
+    end
+
+    it 'should not have the response flag set' do
+      expect(header.flags.reply).to eq 0
+    end
+  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.buffer).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.security_buffer_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.buffer).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.security_buffer_length).to eq fake_message.length
+    end
+  end
+end

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

@@ -0,0 +1,54 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB2::Packet::SessionSetupResponse 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 :session_flags }
+  it { is_expected.to respond_to :security_buffer_offset }
+  it { is_expected.to respond_to :security_buffer_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::SESSION_SETUP
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    end
+  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.buffer).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.security_buffer_length).to eq fake_message.length
+    end
+  end
+end

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

@@ -0,0 +1,127 @@
+RSpec.describe RubySMB::SMB2::SMB2Header do
+  subject(:header) { described_class.new }
+
+  it { is_expected.to respond_to :protocol }
+  it { is_expected.to respond_to :structure_size }
+  it { is_expected.to respond_to :credit_charge }
+  it { is_expected.to respond_to :nt_status }
+  it { is_expected.to respond_to :command }
+  it { is_expected.to respond_to :credits }
+  it { is_expected.to respond_to :flags }
+  it { is_expected.to respond_to :next_command }
+  it { is_expected.to respond_to :message_id }
+  it { is_expected.to respond_to :process_id }
+  it { is_expected.to respond_to :tree_id }
+  it { is_expected.to respond_to :session_id }
+  it { is_expected.to respond_to :signature }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#protocol' do
+    it 'is a 32-bit field' do
+      expect(header.protocol).to be_a BinData::Bit32
+    end
+
+    it 'has a default value of the SMB2 ID' do
+      expect(header.protocol).to eq RubySMB::SMB2::SMB2_PROTOCOL_ID
+    end
+  end
+
+  describe '#structure_size' do
+    it 'is a 16-bit unsigned integer' do
+      expect(header.structure_size).to be_a BinData::Uint16le
+    end
+
+    it 'has a default value of 64 as per the SMB2 spec' do
+      expect(header.structure_size).to eq 64
+    end
+  end
+
+  describe '#credit_charge' do
+    it 'is a 16-bit unsigned integer' do
+      expect(header.credit_charge).to be_a BinData::Uint16le
+    end
+
+    it 'has a default value of 0' do
+      expect(header.credit_charge).to eq 0
+    end
+  end
+
+  describe '#nt_status' do
+    it 'is a NTStatus field' do
+      expect(header.nt_status).to be_a RubySMB::Field::NtStatus
+    end
+
+    it 'has a default value of 0' do
+      expect(header.nt_status).to eq 0
+    end
+  end
+
+  describe '#command' do
+    it 'is a 16-bit unsigned integer' do
+      expect(header.command).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#credits' do
+    it 'is a 16-bit unsigned integer' do
+      expect(header.credits).to be_a BinData::Uint16le
+    end
+  end
+
+  describe '#flags' do
+    it 'is an smb2_header_flags field' do
+      expect(header.flags).to be_a RubySMB::SMB2::BitField::Smb2HeaderFlags
+    end
+  end
+
+  describe '#next_command' do
+    it 'is a 32-bit unsigned integer' do
+      expect(header.next_command).to be_a BinData::Uint32le
+    end
+
+    it 'has a default value of 0' do
+      expect(header.next_command).to eq 0
+    end
+  end
+
+  describe '#message_id' do
+    it 'is a 64-bit unsigned integer' do
+      expect(header.message_id).to be_a BinData::Uint64le
+    end
+  end
+
+  describe '#process_id' do
+    it 'is a 32-bit unsigned integer' do
+      expect(header.process_id).to be_a BinData::Uint32le
+    end
+
+    it 'has a default value of 0x0000feff' do
+      expect(header.process_id).to eq 0x0000feff
+    end
+  end
+
+  describe '#tree_id' do
+    it 'is a 32-bit unsigned integer' do
+      expect(header.tree_id).to be_a BinData::Uint32le
+    end
+  end
+
+  describe '#session_id' do
+    it 'is a 64-bit unsigned integer' do
+      expect(header.session_id).to be_a BinData::Uint64le
+    end
+  end
+
+  describe '#signature' do
+    it 'is a binary string' do
+      expect(header.signature).to be_a BinData::String
+    end
+
+    it 'is 16 bytes' do
+      expect(header.signature.do_num_bytes).to eq 16
+    end
+  end
+end

data/spec/lib/ruby_smb_spec.rb

@@ -0,0 +1,2 @@
+RSpec.describe RubySMB do
+end

data/spec/spec_helper.rb

@@ -0,0 +1,100 @@
+require 'simplecov'
+
+SimpleCov.start do
+  add_filter "/spec/"
+end
+
+require 'coveralls'
+require 'ruby_smb'
+
+if ENV['TRAVIS'] == 'true'
+  # don't generate local report as it is inaccessible on travis-ci, which is
+  # why coveralls is being used.
+  SimpleCov.formatter = Coveralls::SimpleCov::Formatter
+else
+  SimpleCov.formatter = SimpleCov::Formatter::HTMLFormatter
+end
+
+Dir['./spec/support/**/*.rb'].sort.each { |f| require f }
+
+# This file was generated by the `rspec --init` command. Conventionally, all
+# specs live under a `spec` directory, which RSpec adds to the `$LOAD_PATH`.
+# The generated `.rspec` file contains `--require spec_helper` which will cause
+# this file to always be loaded, without a need to explicitly require it in any
+# files.
+#
+# Given that it is always loaded, you are encouraged to keep this file as
+# light-weight as possible. Requiring heavyweight dependencies from this file
+# will add to the boot time of your test suite on EVERY test run, even for an
+# individual file that may not need all of that loaded. Instead, consider making
+# a separate helper file that requires the additional dependencies and performs
+# the additional setup, and require it from the spec files that actually need
+# it.
+#
+# The `.rspec` file also contains a few flags that are not defaults but that
+# users commonly want.
+#
+# See http://rubydoc.info/gems/rspec-core/RSpec/Core/Configuration
+RSpec.configure do |config|
+  # rspec-expectations config goes here. You can use an alternate
+  # assertion/expectation library such as wrong or the stdlib/minitest
+  # assertions if you prefer.
+  config.expect_with :rspec do |expectations|
+    expectations.syntax = :expect
+  end
+
+  # rspec-mocks config goes here. You can use an alternate test double
+  # library (such as bogus or mocha) by changing the `mock_with` option here.
+  config.mock_with :rspec do |mocks|
+    # Prevents you from mocking or stubbing a method that does not exist on
+    # a real object. This is generally recommended, and will default to
+    # `true` in RSpec 4.
+    mocks.verify_partial_doubles = true
+    mocks.syntax = :expect
+  end
+
+  # These two settings work together to allow you to limit a spec run
+  # to individual examples or groups you care about by tagging them with
+  # `:focus` metadata. When nothing is tagged with `:focus`, all examples
+  # get run.
+  config.filter_run :focus
+  config.run_all_when_everything_filtered = true
+
+  # Limits the available syntax to the non-monkey patched syntax that is
+  # recommended. For more details, see:
+  #   - http://myronmars.to/n/dev-blog/2012/06/rspecs-new-expectation-syntax
+  #   - http://teaisaweso.me/blog/2013/05/27/rspecs-new-message-expectation-syntax/
+  #   - http://myronmars.to/n/dev-blog/2014/05/notable-changes-in-rspec-3#new__config_option_to_disable_rspeccore_monkey_patching
+  config.disable_monkey_patching!
+
+  # This setting enables warnings. It's recommended, but in some cases may
+  # be too noisy due to issues in dependencies.
+  config.warnings = false
+
+  # Many RSpec users commonly either run the entire suite or an individual
+  # file, and it's useful to allow more verbose output when running an
+  # individual spec file.
+  if config.files_to_run.one?
+    # Use the documentation formatter for detailed output,
+    # unless a formatter has already been configured
+    # (e.g. via a command-line flag).
+    config.default_formatter = 'doc'
+
+    # RSpec filters the backtrace by default so as not to be so noisy.
+    # This causes the full backtrace to be printed when running a single
+    # spec file (e.g. to troubleshoot a particular spec failure).
+    config.full_backtrace = true
+  end
+
+  # Run specs in random order to surface order dependencies. If you find an
+  # order dependency and want to debug it, you can fix the order by providing
+  # the seed, which is printed after each run.
+  #     --seed 1234
+  config.order = :random
+
+  # Seed global randomization in this process using the `--seed` CLI option.
+  # Setting this allows you to use `--seed` to deterministically reproduce
+  # test failures related to randomization by passing the same `--seed` value
+  # as the one that triggered the failure.
+  Kernel.srand config.seed
+end

data/spec/support/mock_socket_dispatcher.rb

@@ -0,0 +1,8 @@
+require 'ruby_smb/dispatcher'
+class MockSocketDispatcher < RubySMB::Dispatcher::Base
+  def recv_packet
+    ''
+  end
+
+  def send_packet(packet); end
+end

data/spec/support/shared/examples/bit_field_single_flag.rb

@@ -0,0 +1,14 @@
+RSpec.shared_examples 'bit field with one flag set' do |flag, pack, value|
+  it "has a value of #{value} when only #{flag} is set" do
+    subject.field_names.each do |sub_field|
+      if sub_field == flag
+        subject.send(sub_field.to_s.concat('=').to_sym, 1)
+      else
+        subject.send(sub_field.to_s.concat('=').to_sym, 0)
+      end
+    end
+    field_val = subject.to_binary_s
+    field_val = field_val.unpack(pack).first
+    expect(field_val).to eq value
+  end
+end