ruby_smb 3.1.1 → 3.1.2

data/spec/lib/ruby_smb/fscc/file_information/file_rename_information_spec.rb

@@ -3,7 +3,7 @@ require 'spec_helper'
 RSpec.describe RubySMB::Fscc::FileInformation::FileRenameInformation do
   it 'references the correct class level' do
     expect(described_class).to be_const_defined(:CLASS_LEVEL)
-    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileInformation::FILE_ID_FULL_DIRECTORY_INFORMATION
+    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileInformation::FILE_RENAME_INFORMATION
   end
 
   subject(:struct) { described_class.new }

data/spec/lib/ruby_smb/fscc/file_information/file_standard_information_spec.rb

@@ -0,0 +1,41 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileStandardInformation do
+  it 'references the correct class level' do
+    expect(described_class).to be_const_defined(:CLASS_LEVEL)
+    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileInformation::FILE_STANDARD_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :allocation_size }
+  it { should respond_to :end_of_file }
+  it { should respond_to :number_of_links }
+  it { should respond_to :delete_pending }
+  it { should respond_to :directory }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the allocation size in a Int64 field' do
+    expect(struct.allocation_size).to be_a BinData::Int64le
+  end
+
+  it 'tracks the file size in a Int64 field' do
+    expect(struct.end_of_file).to be_a BinData::Int64le
+  end
+
+  it 'tracks the number of links in a Uint32 field' do
+    expect(struct.number_of_links).to be_a BinData::Uint32le
+  end
+
+  it 'tracks if a delete is pending in a Int8 field' do
+    expect(struct.delete_pending).to be_a BinData::Int8
+  end
+
+  it 'tracks if it is a directory in a Int8 field' do
+    expect(struct.directory).to be_a BinData::Int8
+  end
+
+end

data/spec/lib/ruby_smb/fscc/file_information/file_stream_information_spec.rb

@@ -0,0 +1,51 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileStreamInformation do
+  it 'references the correct class level' do
+    expect(described_class).to be_const_defined(:CLASS_LEVEL)
+    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileInformation::FILE_STREAM_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_entry_offset }
+  it { should respond_to :stream_name_length }
+  it { should respond_to :stream_size }
+  it { should respond_to :stream_allocation_size }
+  it { should respond_to :stream_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the next entry offset in a Uint32 field' do
+    expect(struct.next_entry_offset).to be_a BinData::Uint32le
+  end
+
+  it 'tracks the stream name length in a Uint32 field' do
+    expect(struct.stream_name_length).to be_a BinData::Uint32le
+  end
+
+  it 'tracks the stream size in a Int64 field' do
+    expect(struct.stream_size).to be_a BinData::Int64le
+  end
+
+  it 'tracks the stream allocation size in a Int64 field' do
+    expect(struct.stream_allocation_size).to be_a BinData::Int64le
+  end
+
+  it 'tracks the stream name in a String16 field' do
+    expect(struct.stream_name).to be_a RubySMB::Field::String16
+  end
+
+  it 'tracks the length of the stream_name field' do
+    struct.stream_name = 'Hello.txt'
+    expect(struct.stream_name_length).to eq struct.stream_name.do_num_bytes
+  end
+
+  it 'automatically encodes the stream name in UTF-16LE' do
+    name = 'Hello_world.txt'
+    struct.stream_name = name
+    expect(struct.stream_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+end

data/spec/lib/ruby_smb/fscc/file_information_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation do
+
+  describe '#name' do
+    it 'maps constant names to their value' do
+      expect(described_class.name(RubySMB::Fscc::FileInformation::FILE_DIRECTORY_INFORMATION)).to eq :FILE_DIRECTORY_INFORMATION
+    end
+
+    it 'returns nil for values that do not exist' do
+      expect(described_class.name(-1)).to be_nil
+    end
+  end
+end

data/spec/lib/ruby_smb/fscc/file_system_information/file_fs_attribute_information_spec.rb

@@ -0,0 +1,46 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileSystemInformation::FileFsAttributeInformation do
+  it 'references the correct class level' do
+    expect(described_class).to be_const_defined(:CLASS_LEVEL)
+    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileSystemInformation::FILE_FS_ATTRIBUTE_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :file_system_attributes }
+  it { should respond_to :maximum_component_name_length }
+  it { should respond_to :file_system_name_length }
+  it { should respond_to :file_system_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the file system attributes in a struct field' do
+    expect(struct.file_system_attributes).to be_a BinData::Struct
+  end
+
+  it 'tracks the maximum component name length in a Int32 field' do
+    expect(struct.maximum_component_name_length).to be_a BinData::Int32le
+  end
+
+  it 'tracks the file system name length in a Uint32 field' do
+    expect(struct.file_system_name_length).to be_a BinData::Uint32le
+  end
+
+  it 'tracks the file system name in a String16 field' do
+    expect(struct.file_system_name).to be_a RubySMB::Field::String16
+  end
+
+  it 'tracks the length of the file_system_name field' do
+    struct.file_system_name = 'NTFS'
+    expect(struct.file_system_name_length).to eq struct.file_system_name.do_num_bytes
+  end
+
+  it 'automatically encodes the file system name in UTF-16LE' do
+    name = 'NTFS'
+    struct.file_system_name = name
+    expect(struct.file_system_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+end

data/spec/lib/ruby_smb/fscc/file_system_information/file_fs_volume_information_spec.rb

@@ -0,0 +1,51 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileSystemInformation::FileFsVolumeInformation do
+  it 'references the correct class level' do
+    expect(described_class).to be_const_defined(:CLASS_LEVEL)
+    expect(described_class::CLASS_LEVEL).to be RubySMB::Fscc::FileSystemInformation::FILE_FS_VOLUME_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :volume_creation_time }
+  it { should respond_to :volume_serial_number }
+  it { should respond_to :volume_label_length }
+  it { should respond_to :supports_objects }
+  it { should respond_to :volume_label }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the volume creation time in a FileTime field' do
+    expect(struct.volume_creation_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the volume serial number in a Uint32 field' do
+    expect(struct.volume_serial_number).to be_a BinData::Uint32le
+  end
+
+  it 'tracks the volume label length in a Uint32 field' do
+    expect(struct.volume_label_length).to be_a BinData::Uint32le
+  end
+
+  it 'tracks if it supports objects in a Uint8 field' do
+    expect(struct.supports_objects).to be_a BinData::Uint8
+  end
+
+  it 'tracks the volume label in a String16 field' do
+    expect(struct.volume_label).to be_a RubySMB::Field::String16
+  end
+
+  it 'tracks the length of the volume_label field' do
+    struct.volume_label = 'NTFS'
+    expect(struct.volume_label_length).to eq struct.volume_label.do_num_bytes
+  end
+
+  it 'automatically encodes the file system name in UTF-16LE' do
+    name = 'NTFS'
+    struct.volume_label = name
+    expect(struct.volume_label.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+end

data/spec/lib/ruby_smb/fscc/file_system_information_spec.rb

@@ -0,0 +1,14 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileSystemInformation do
+
+  describe '#name' do
+    it 'maps constant names to their value' do
+      expect(described_class.name(RubySMB::Fscc::FileSystemInformation::FILE_FS_VOLUME_INFORMATION)).to eq :FILE_FS_VOLUME_INFORMATION
+    end
+
+    it 'returns nil for values that do not exist' do
+      expect(described_class.name(-1)).to be_nil
+    end
+  end
+end

data/spec/lib/ruby_smb/server/server_client_spec.rb

@@ -5,6 +5,7 @@ RSpec.describe RubySMB::Server::ServerClient do
   subject(:server_client) { described_class.new(server, dispatcher) }
 
   it { is_expected.to respond_to :dialect }
+  it { is_expected.to respond_to :dispatcher }
   it { is_expected.to respond_to :session_table }
 
   describe '#disconnect!' do
@@ -31,6 +32,20 @@ RSpec.describe RubySMB::Server::ServerClient do
     end
   end
 
+  describe '#peerhost' do
+    it 'returns the peer IP address' do
+      expect(server_client).to receive(:getpeername).and_return(Socket.sockaddr_in(4444, '127.0.0.1'))
+      expect(server_client.peerhost).to eq '127.0.0.1'
+    end
+  end
+
+  describe '#peerport' do
+    it 'returns the peer IP port' do
+      expect(server_client).to receive(:getpeername).and_return(Socket.sockaddr_in(4444, '127.0.0.1'))
+      expect(server_client.peerport).to eq 4444
+    end
+  end
+
   describe '#process_gss' do
     before(:each) do
       expect(server_client.instance_eval { @gss_authenticator }).to receive(:process).and_call_original