ruby_smb 0.0.18 → 0.0.19

data/spec/lib/ruby_smb/field/smb_gea_spec.rb

@@ -0,0 +1,22 @@
+RSpec.describe RubySMB::Field::SmbGea do
+  subject(:gea) { described_class.new }
+
+  it { is_expected.to respond_to :attribute_name_length }
+  it { is_expected.to respond_to :attribute_name }
+  it { is_expected.to respond_to :null_pad }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#attribute_name_length' do
+    it 'reflects the size of the attribute_name field' do
+      gea.attribute_name = 'TEST'
+      expect(gea.attribute_name_length).to eq 4
+    end
+  end
+
+  it 'always has a null pad at the end' do
+    expect(gea.null_pad).to eq 0x00
+  end
+end

data/spec/lib/ruby_smb/field/stringz16_spec.rb

@@ -4,14 +4,14 @@ RSpec.describe RubySMB::Field::Stringz16 do
   subject(:stringz16) { described_class.new }
 
   it 'starts as an empty string' do
-    expect(stringz16).to eq ""
+    expect(stringz16).to eq ''
   end
 
   context 'with a value already set' do
-    let(:abcd) { described_class.new("ABCD") }
+    let(:abcd) { described_class.new('ABCD') }
 
     it 'should be UTF-16le' do
-      expect(abcd).to eq "ABCD".encode("utf-16le")
+      expect(abcd).to eq 'ABCD'.encode('utf-16le')
     end
 
     it 'should include the NULL terminator on binary output' do
@@ -23,12 +23,11 @@ RSpec.describe RubySMB::Field::Stringz16 do
     let(:null_terminator) { described_class.new("ABCD\x00EFG") }
 
     it 'drops everything after the null terminator' do
-      expect(null_terminator).to eq "ABCD".encode("utf-16le")
+      expect(null_terminator).to eq 'ABCD'.encode('utf-16le')
     end
   end
 
   context 'when reading data in' do
-
     it 'stops at the first double null byte' do
       io = StringIO.new("A\x00B\x00C\x00D\x00\x00\x00E\x00F\x00")
       stringz16.read(io)
@@ -38,13 +37,12 @@ RSpec.describe RubySMB::Field::Stringz16 do
     it 'handles a zero length string' do
       io = StringIO.new("\x00\x00A\x00")
       stringz16.read(io)
-      expect(stringz16).to eq ""
+      expect(stringz16).to eq ''
     end
 
     it 'fails if no null terminator is found' do
       io = StringIO.new("A\x00B\x00C\x00D\x00")
-      expect{ stringz16.read(io) }.to raise_error(EOFError)
+      expect { stringz16.read(io) }.to raise_error(EOFError)
     end
   end
-
-end
+end

data/spec/lib/ruby_smb/field/utime_spec.rb

@@ -1,3 +1,5 @@
+require 'date'
+
 RSpec.describe RubySMB::Field::Utime do
   subject(:time) { described_class.read(binary_filetime) }
 
@@ -42,13 +44,13 @@ RSpec.describe RubySMB::Field::Utime do
     it 'will take a Time object correctly but lose Nanoseconds' do
       empty_filetime.set time_filetime
       val = empty_filetime.get
-      expect(val / 10_000_000).to eq (int_filetime / 10_000_000)
+      expect(val / 10_000_000).to eq(int_filetime / 10_000_000)
     end
 
     it 'will take a DateTime object correctly but lose Nanoseconds' do
       empty_filetime.set datetime_filetime
       val = empty_filetime.get
-      expect(val / 10_000_000).to eq (int_filetime / 10_000_000)
+      expect(val / 10_000_000).to eq(int_filetime / 10_000_000)
     end
 
     it 'will accept a raw integer value and set it' do

data/spec/lib/ruby_smb/{field → fscc}/ea_info_array_spec.rb

@@ -1,21 +1,21 @@
 require 'spec_helper'
 
-RSpec.describe RubySMB::Field::EaInfoArray do
+RSpec.describe RubySMB::Fscc::EaInfoArray do
   subject(:info_array) { described_class.new(type: :file_full_ea_info) }
   let(:first_ea) {
-    ea = RubySMB::Field::FileFullEaInfo.new
+    ea = RubySMB::Fscc::FileFullEaInfo.new
     ea.ea_name  = 'First'
     ea.ea_value = 'First Value'
     ea
   }
   let(:second_ea) {
-    ea = RubySMB::Field::FileFullEaInfo.new
+    ea = RubySMB::Fscc::FileFullEaInfo.new
     ea.ea_name  = 'Second'
     ea.ea_value = 'Second Value'
     ea
   }
   let(:additional_ea) {
-    ea = RubySMB::Field::FileFullEaInfo.new
+    ea = RubySMB::Fscc::FileFullEaInfo.new
     ea.ea_name  = 'Additional'
     ea.ea_value = 'Additional Value'
     ea
@@ -40,12 +40,10 @@ RSpec.describe RubySMB::Field::EaInfoArray do
   end
 
   it 'raises an error when inserting an invalid object' do
-    expect{ info_array << "Foo" }.to raise_error(ArgumentError)
+    expect { info_array << 'Foo' }.to raise_error(ArgumentError)
   end
 
   it 'raises an error when setting an invalid value' do
-    expect{ info_array[0] = "Foo" }.to raise_error(ArgumentError)
+    expect { info_array[0] = 'Foo' }.to raise_error(ArgumentError)
   end
-
-
-end
+end

data/spec/lib/ruby_smb/{field → fscc}/file_full_ea_info_spec.rb

@@ -1,6 +1,6 @@
 require 'spec_helper'
 
-RSpec.describe RubySMB::Field::FileFullEaInfo do
+RSpec.describe RubySMB::Fscc::FileFullEaInfo do
   subject(:info) { described_class.new }
 
   it { is_expected.to respond_to :next_entry_offset }
@@ -31,5 +31,4 @@ RSpec.describe RubySMB::Field::FileFullEaInfo do
       expect(info.ea_value_length).to eq info.ea_value.do_num_bytes
     end
   end
-
-end
+end

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

@@ -0,0 +1,71 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileBothDirectoryInformation 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_BOTH_DIRECTORY_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_offset }
+  it { should respond_to :file_index }
+  it { should respond_to :create_time }
+  it { should respond_to :last_access }
+  it { should respond_to :last_write }
+  it { should respond_to :last_change }
+  it { should respond_to :end_of_file }
+  it { should respond_to :allocation_size }
+  it { should respond_to :file_attributes }
+  it { should respond_to :file_name_length }
+  it { should respond_to :ea_size }
+  it { should respond_to :short_name }
+  it { should respond_to :short_name_length }
+  it { should respond_to :file_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the creation time in a Filetime field' do
+    expect(struct.create_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last access time in a Filetime field' do
+    expect(struct.last_access).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last write time in a Filetime field' do
+    expect(struct.last_write).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last modified time in a Filetime field' do
+    expect(struct.last_change).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'contains the file attributes of the file' do
+    expect(struct.file_attributes).to be_a RubySMB::Fscc::FileAttributes
+  end
+
+  it 'tracks the length of the file_name field' do
+    struct.file_name = 'Hello.txt'
+    expect(struct.file_name_length).to eq struct.file_name.do_num_bytes
+  end
+
+  it 'automatically encodes the file name in UTF-16LE' do
+    name = 'Hello_world.txt'
+    struct.file_name = name
+    expect(struct.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+
+  describe 'reading in from a blob' do
+    it 'uses the file_name_length to know when to stop reading' do
+      name = 'Hello_world.txt'
+      struct.file_name = name
+      blob = struct.to_binary_s
+      blob << 'AAAA'
+      new_from_blob = described_class.read(blob)
+      expect(new_from_blob.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+    end
+  end
+end

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

@@ -0,0 +1,68 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileDirectoryInformation 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_DIRECTORY_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_offset }
+  it { should respond_to :file_index }
+  it { should respond_to :create_time }
+  it { should respond_to :last_access }
+  it { should respond_to :last_write }
+  it { should respond_to :last_change }
+  it { should respond_to :end_of_file }
+  it { should respond_to :allocation_size }
+  it { should respond_to :file_attributes }
+  it { should respond_to :file_name_length }
+  it { should respond_to :file_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the creation time in a Filetime field' do
+    expect(struct.create_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last access time in a Filetime field' do
+    expect(struct.last_access).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last write time in a Filetime field' do
+    expect(struct.last_write).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last modified time in a Filetime field' do
+    expect(struct.last_change).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'contains the file attributes of the file' do
+    expect(struct.file_attributes).to be_a RubySMB::Fscc::FileAttributes
+  end
+
+  it 'tracks the length of the file_name field' do
+    struct.file_name = 'Hello.txt'
+    expect(struct.file_name_length).to eq struct.file_name.do_num_bytes
+  end
+
+  it 'automatically encodes the file name in UTF-16LE' do
+    name = 'Hello_world.txt'
+    struct.file_name = name
+    expect(struct.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+
+  describe 'reading in from a blob' do
+    it 'uses the file_name_length to know when to stop reading' do
+      name = 'Hello_world.txt'
+      struct.file_name = name
+      blob = struct.to_binary_s
+      blob << 'AAAA'
+      new_from_blob = described_class.read(blob)
+      expect(new_from_blob.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+    end
+  end
+end

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

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileDispositionInformation 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_DISPOSITION_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :delete_pending }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#delete_pending' do
+    it 'is a 8-bit field' do
+      expect(struct.delete_pending).to be_a BinData::Uint8
+    end
+
+    it 'should have a default value of 0' do
+      expect(struct.delete_pending).to eq 0
+    end
+  end
+end

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

@@ -0,0 +1,69 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileFullDirectoryInformation 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_FULL_DIRECTORY_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_offset }
+  it { should respond_to :file_index }
+  it { should respond_to :create_time }
+  it { should respond_to :last_access }
+  it { should respond_to :last_write }
+  it { should respond_to :last_change }
+  it { should respond_to :end_of_file }
+  it { should respond_to :allocation_size }
+  it { should respond_to :file_attributes }
+  it { should respond_to :file_name_length }
+  it { should respond_to :ea_size }
+  it { should respond_to :file_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the creation time in a Filetime field' do
+    expect(struct.create_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last access time in a Filetime field' do
+    expect(struct.last_access).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last write time in a Filetime field' do
+    expect(struct.last_write).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last modified time in a Filetime field' do
+    expect(struct.last_change).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'contains the file attributes of the file' do
+    expect(struct.file_attributes).to be_a RubySMB::Fscc::FileAttributes
+  end
+
+  it 'tracks the length of the file_name field' do
+    struct.file_name = 'Hello.txt'
+    expect(struct.file_name_length).to eq struct.file_name.do_num_bytes
+  end
+
+  it 'automatically encodes the file name in UTF-16LE' do
+    name = 'Hello_world.txt'
+    struct.file_name = name
+    expect(struct.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+
+  describe 'reading in from a blob' do
+    it 'uses the file_name_length to know when to stop reading' do
+      name = 'Hello_world.txt'
+      struct.file_name = name
+      blob = struct.to_binary_s
+      blob << 'AAAA'
+      new_from_blob = described_class.read(blob)
+      expect(new_from_blob.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+    end
+  end
+end

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

@@ -0,0 +1,72 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::Fscc::FileInformation::FileIdBothDirectoryInformation 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_BOTH_DIRECTORY_INFORMATION
+  end
+
+  subject(:struct) { described_class.new }
+
+  it { should respond_to :next_offset }
+  it { should respond_to :file_index }
+  it { should respond_to :create_time }
+  it { should respond_to :last_access }
+  it { should respond_to :last_write }
+  it { should respond_to :last_change }
+  it { should respond_to :end_of_file }
+  it { should respond_to :allocation_size }
+  it { should respond_to :file_attributes }
+  it { should respond_to :file_name_length }
+  it { should respond_to :ea_size }
+  it { should respond_to :short_name }
+  it { should respond_to :short_name_length }
+  it { should respond_to :file_id }
+  it { should respond_to :file_name }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'tracks the creation time in a Filetime field' do
+    expect(struct.create_time).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last access time in a Filetime field' do
+    expect(struct.last_access).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last write time in a Filetime field' do
+    expect(struct.last_write).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'tracks the last modified time in a Filetime field' do
+    expect(struct.last_change).to be_a RubySMB::Field::FileTime
+  end
+
+  it 'contains the file attributes of the file' do
+    expect(struct.file_attributes).to be_a RubySMB::Fscc::FileAttributes
+  end
+
+  it 'tracks the length of the file_name field' do
+    struct.file_name = 'Hello.txt'
+    expect(struct.file_name_length).to eq struct.file_name.do_num_bytes
+  end
+
+  it 'automatically encodes the file name in UTF-16LE' do
+    name = 'Hello_world.txt'
+    struct.file_name = name
+    expect(struct.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+  end
+
+  describe 'reading in from a blob' do
+    it 'uses the file_name_length to know when to stop reading' do
+      name = 'Hello_world.txt'
+      struct.file_name = name
+      blob = struct.to_binary_s
+      blob << 'AAAA'
+      new_from_blob = described_class.read(blob)
+      expect(new_from_blob.file_name.force_encoding('utf-16le')).to eq name.encode('utf-16le')
+    end
+  end
+end