ruby_smb 3.1.1 → 3.1.2

data/spec/lib/ruby_smb/server/share/provider/virtual_disk/virtual_stat_spec.rb

@@ -0,0 +1,207 @@
+RSpec.describe RubySMB::Server::Share::Provider::VirtualDisk::VirtualStat do
+  subject(:virtual_stat) { described_class.new }
+
+  it { is_expected.to respond_to :atime }
+  it { is_expected.to respond_to :blksize }
+  it { is_expected.to respond_to :blockdev? }
+  it { is_expected.to respond_to :blocks }
+  it { is_expected.to respond_to :chardev? }
+  it { is_expected.to respond_to :ctime }
+  it { is_expected.to respond_to :dev }
+  it { is_expected.to respond_to :directory? }
+  it { is_expected.to respond_to :executable? }
+  it { is_expected.to respond_to :file? }
+  it { is_expected.to respond_to :ftype }
+  it { is_expected.to respond_to :gid }
+  it { is_expected.to respond_to :grpowned? }
+  it { is_expected.to respond_to :ino }
+  it { is_expected.to respond_to :mode }
+  it { is_expected.to respond_to :mtime }
+  it { is_expected.to respond_to :nlink }
+  it { is_expected.to respond_to :owned? }
+  it { is_expected.to respond_to :pipe? }
+  it { is_expected.to respond_to :readable? }
+  it { is_expected.to respond_to :setgid? }
+  it { is_expected.to respond_to :setuid? }
+  it { is_expected.to respond_to :size }
+  it { is_expected.to respond_to :socket? }
+  it { is_expected.to respond_to :sticky? }
+  it { is_expected.to respond_to :symlink? }
+  it { is_expected.to respond_to :uid }
+  it { is_expected.to respond_to :world_readable? }
+  it { is_expected.to respond_to :world_writable? }
+  it { is_expected.to respond_to :writable? }
+  it { is_expected.to respond_to :zero? }
+
+  describe '#initialize' do
+    it 'defaults to being a directory' do
+      expect(virtual_stat.directory?).to be true
+    end
+
+    it 'throws an exception when it is both a file and a directory' do
+      expect { described_class.new(directory?: true, file?: true) }.to raise_error(ArgumentError)
+    end
+
+    it 'throws an exception when it is neither a file or a directory' do
+      expect { described_class.new(directory?: false, file?: false) }.to raise_error(ArgumentError)
+    end
+
+    it 'is a directory when it is a directory' do
+      expect(described_class.new(directory?: true).directory?).to be true
+    end
+
+    it 'is a directory when not a file' do
+      expect(described_class.new(file?: false).directory?).to be true
+    end
+
+    it 'is a file when it is a file' do
+      expect(described_class.new(file?: true).file?).to be true
+    end
+
+    it 'is a file when not a directory' do
+      expect(described_class.new(directory?: false).file?).to be true
+    end
+  end
+
+  describe '#atime' do
+    it 'is expected to default to now' do
+      now = Time.now
+      expect(Time).to receive(:now).and_return(now)
+      expect(virtual_stat.atime).to eq now
+    end
+  end
+
+  describe '#ctime' do
+    it 'is expected to default to now' do
+      now = Time.now
+      expect(Time).to receive(:now).and_return(now)
+      expect(virtual_stat.ctime).to eq now
+    end
+  end
+
+  describe '#mtime' do
+    it 'is expected to default to now' do
+      now = Time.now
+      expect(Time).to receive(:now).and_return(now)
+      expect(virtual_stat.mtime).to eq now
+    end
+  end
+
+  describe '#birthtime' do
+    it 'is expected to default to now' do
+      now = Time.now
+      expect(Time).to receive(:now).and_return(now)
+      expect(virtual_stat.birthtime).to eq now
+    end
+  end
+
+  describe '#uid' do
+    it 'is expected to default to the current process uid' do
+      expect(virtual_stat.uid).to eq Process.uid
+    end
+  end
+
+  describe '#owned?' do
+    it 'is expected to return true when uid is this process' do
+      expect(virtual_stat).to receive(:uid).and_return(Process.uid)
+      expect(virtual_stat.owned?).to be true
+    end
+
+    it 'is expected to return false when uid is not this process' do
+      expect(virtual_stat).to receive(:uid)
+      expect(virtual_stat.owned?).to be false
+    end
+  end
+
+  describe '#gid' do
+    it 'is expected to default to the current process gid' do
+      expect(virtual_stat.gid).to eq Process.gid
+    end
+  end
+
+  describe '#grpowned?' do
+    it 'is expected to return true when gid is this process' do
+      expect(virtual_stat).to receive(:gid).and_return(Process.gid)
+      expect(virtual_stat.grpowned?).to be true
+    end
+
+    it 'is expected to return false when gid is not this process' do
+      expect(virtual_stat).to receive(:gid)
+      expect(virtual_stat.grpowned?).to be false
+    end
+  end
+
+  describe '#readable?' do
+    it 'is expected to readable by default' do
+      expect(virtual_stat.readable?).to be true
+    end
+  end
+
+  describe '#writable?' do
+    it 'is expected to writable by default' do
+      expect(virtual_stat.writable?).to be true
+    end
+  end
+
+  describe '#ftype' do
+    it 'is expected to be a String' do
+      expect(virtual_stat.ftype).to be_a String
+    end
+  end
+
+  # test each and every permission represented by the 12 possible bits
+  (0..12).each.map { |bit| (1 << bit) & 0o7777 }.each do |mode|
+    context "when mode is 0o#{mode.to_s(8).rjust(4, '0')}" do
+      before(:each) { expect(virtual_stat).to receive(:mode).at_least(:once).and_return(mode) }
+
+      describe '#executable?' do
+        it 'is correct' do
+          expect(virtual_stat.executable?).to eq mode & 0o111 != 0
+        end
+      end
+
+      describe '#readable?' do
+        it 'is correct' do
+          expect(virtual_stat.readable?).to eq mode & 0o444 != 0
+        end
+      end
+
+      describe '#setgid?' do
+        it 'is correct' do
+          expect(virtual_stat.setgid?).to eq mode & 0o2000 != 0
+        end
+      end
+
+      describe '#setuid?' do
+        it 'is correct' do
+          expect(virtual_stat.setuid?).to eq mode & 0o4000 != 0
+        end
+      end
+
+      describe '#sticky?' do
+        it 'is correct' do
+          expect(virtual_stat.sticky?).to eq mode & 0o1000 != 0
+        end
+      end
+
+      describe '#world_readable?' do
+        it 'is correct' do
+          expect(virtual_stat.world_readable?).to eq mode & 0o004 != 0
+        end
+      end
+
+      describe '#world_writable?' do
+        it 'is correct' do
+          expect(virtual_stat.world_writable?).to eq mode & 0o002 != 0
+        end
+      end
+
+      describe '#writable?' do
+        it 'is correct' do
+          expect(virtual_stat.writable?).to eq mode & 0o222 != 0
+        end
+      end
+    end
+  end
+end
+

data/spec/lib/ruby_smb/server/share/provider/virtual_disk_spec.rb

@@ -0,0 +1,122 @@
+RSpec.describe RubySMB::Server::Share::Provider::VirtualDisk do
+  let(:name) { 'vdisk' }
+  subject(:virtual_disk) { described_class.new(name) }
+
+  it { is_expected.to respond_to :name }
+  it { is_expected.to respond_to :path }
+  it { is_expected.to respond_to :type }
+  context 'it provides the basic Hash methods' do
+    it { is_expected.to respond_to :[] }
+    it { is_expected.to respond_to :each }
+    it { is_expected.to respond_to :each_key }
+    it { is_expected.to respond_to :each_value }
+    it { is_expected.to respond_to :keys }
+    it { is_expected.to respond_to :length }
+    it { is_expected.to respond_to :values }
+  end
+
+  describe '#add' do
+    let(:path) { '/path' }
+
+    it 'does not throw an exception if the stat object is not a file' do
+      stat = RubySMB::Server::Share::Provider::VirtualDisk::VirtualStat.new(file?: false)
+      vp = RubySMB::Server::Share::Provider::VirtualDisk::VirtualPathname.new({}, path, stat: stat)
+      expect { virtual_disk.add(vp) }.not_to raise_error
+    end
+
+    it 'throws an exception if the path is not absolute' do
+      vp = RubySMB::Server::Share::Provider::VirtualDisk::VirtualPathname.new({}, path[1..])
+      expect { virtual_disk.add(vp) }.to raise_error(ArgumentError)
+    end
+
+    it 'automatically creates parent directories' do
+      expect(virtual_disk.length).to eq 1 # starts with just the root folder
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualPathname).to receive(:new).exactly(3).times.and_call_original
+      virtual_disk.add(RubySMB::Server::Share::Provider::VirtualDisk::VirtualPathname.new({}, File.join(path, 'sub', 'dir')))
+      expect(virtual_disk.length).to eq 4 # ends with the added item and 2 parent folders
+    end
+  end
+
+  describe '#add_dynamic_file' do
+    let(:path) { 'path' }
+    let(:content) { 'Hello World!' }
+    let(:block) { Proc.new { content } }
+
+    before(:each) { allow(virtual_disk).to receive(:add) }
+
+    it 'normalizes the path argument' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualDynamicFile).to receive(:new).with(virtual_disk, File::SEPARATOR + path, stat: nil, &block)
+      virtual_disk.add_dynamic_file(path + '/dir/..', &block)
+    end
+
+    it 'initializes a new VirtualDynamicFile instance' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualDynamicFile).to receive(:new)
+      virtual_disk.add_dynamic_file(path, &block)
+    end
+
+    it 'adds the mapped file' do
+      expect(virtual_disk).to receive(:add)
+      expect(virtual_disk.add_dynamic_file(path, &block))
+    end
+
+    it 'throws an exception if the stat object is not a file' do
+      stat = RubySMB::Server::Share::Provider::VirtualDisk::VirtualStat.new(file?: false)
+      expect { virtual_disk.add_dynamic_file(path, stat: stat, &block) }. to raise_error(ArgumentError)
+    end
+  end
+
+  describe '#add_mapped_file' do
+    let(:path) { 'path' }
+    let(:mapped_path) { 'mapped_path' }
+
+    before(:each) { allow(virtual_disk).to receive(:add) }
+
+    it 'normalizes the path argument' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualMappedFile).to receive(:new).with(virtual_disk, File::SEPARATOR + path, mapped_path)
+      virtual_disk.add_mapped_file(path + '/dir/..', mapped_path)
+    end
+
+    it 'initializes a new VirtualMappedFile instance' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualMappedFile).to receive(:new)
+      virtual_disk.add_mapped_file(path, mapped_path)
+    end
+
+    it 'adds the mapped file' do
+      expect(virtual_disk).to receive(:add)
+      expect(virtual_disk.add_mapped_file(path, mapped_path))
+    end
+
+    it 'throws an exception if the stat object is not a file' do
+      stat = RubySMB::Server::Share::Provider::VirtualDisk::VirtualStat.new(file?: false)
+      expect { virtual_disk.add_mapped_file(path, mapped_path, stat: stat) }. to raise_error(ArgumentError)
+    end
+  end
+
+  describe '#add_static_file' do
+    let(:path) { 'path' }
+    let(:content) { 'Hello World!' }
+
+    before(:each) { allow(virtual_disk).to receive(:add) }
+
+    it 'normalizes the path argument' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualStaticFile).to receive(:new).with(virtual_disk, File::SEPARATOR + path, content, stat: nil)
+      virtual_disk.add_static_file(path + '/dir/..', content)
+    end
+
+    it 'initializes a new VirtualStaticFile instance' do
+      expect(RubySMB::Server::Share::Provider::VirtualDisk::VirtualStaticFile).to receive(:new)
+      virtual_disk.add_static_file(path, content)
+    end
+
+    it 'adds the mapped file' do
+      expect(virtual_disk).to receive(:add)
+      expect(virtual_disk.add_static_file(path, content))
+    end
+
+    it 'throws an exception if the stat object is not a file' do
+      stat = RubySMB::Server::Share::Provider::VirtualDisk::VirtualStat.new(file?: false)
+      expect { virtual_disk.add_static_file(path, content, stat: stat) }.to raise_error(ArgumentError)
+    end
+  end
+end
+

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_smb
 version: !ruby/object:Gem::Version
-  version: 3.1.1
+  version: 3.1.2
 platform: ruby
 authors:
 - Metasploit Hackers
@@ -97,7 +97,7 @@ cert_chain:
   EknWpNgVhohbot1lfVAMmIhdtOVaRVcQQixWPwprDj/ydB8ryDMDosIMcw+fkoXU
   9GJsSaSRRYQ9UUkVL27b64okU8D48m8=
   -----END CERTIFICATE-----
-date: 2022-04-21 00:00:00.000000000 Z
+date: 2022-05-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet
@@ -279,6 +279,7 @@ files:
 - examples/read_registry_key_value.rb
 - examples/rename_file.rb
 - examples/tree_connect.rb
+- examples/virtual_file_server.rb
 - examples/write_file.rb
 - lib/ruby_smb.rb
 - lib/ruby_smb/client.rb
@@ -436,6 +437,10 @@ files:
 - lib/ruby_smb/fscc/file_attributes.rb
 - lib/ruby_smb/fscc/file_full_ea_info.rb
 - lib/ruby_smb/fscc/file_information.rb
+- lib/ruby_smb/fscc/file_information/file_access_information.rb
+- lib/ruby_smb/fscc/file_information/file_alignment_information.rb
+- lib/ruby_smb/fscc/file_information/file_all_information.rb
+- lib/ruby_smb/fscc/file_information/file_basic_information.rb
 - lib/ruby_smb/fscc/file_information/file_both_directory_information.rb
 - lib/ruby_smb/fscc/file_information/file_directory_information.rb
 - lib/ruby_smb/fscc/file_information/file_disposition_information.rb
@@ -443,9 +448,15 @@ files:
 - lib/ruby_smb/fscc/file_information/file_full_directory_information.rb
 - lib/ruby_smb/fscc/file_information/file_id_both_directory_information.rb
 - lib/ruby_smb/fscc/file_information/file_id_full_directory_information.rb
+- lib/ruby_smb/fscc/file_information/file_internal_information.rb
+- lib/ruby_smb/fscc/file_information/file_mode_information.rb
+- lib/ruby_smb/fscc/file_information/file_name_information.rb
 - lib/ruby_smb/fscc/file_information/file_names_information.rb
 - lib/ruby_smb/fscc/file_information/file_network_open_information.rb
+- lib/ruby_smb/fscc/file_information/file_normalized_name_information.rb
+- lib/ruby_smb/fscc/file_information/file_position_information.rb
 - lib/ruby_smb/fscc/file_information/file_rename_information.rb
+- lib/ruby_smb/fscc/file_information/file_standard_information.rb
 - lib/ruby_smb/fscc/file_information/file_stream_information.rb
 - lib/ruby_smb/fscc/file_system_information.rb
 - lib/ruby_smb/fscc/file_system_information/file_fs_attribute_information.rb
@@ -481,6 +492,10 @@ files:
 - lib/ruby_smb/server/share/provider/disk/processor/read.rb
 - lib/ruby_smb/server/share/provider/pipe.rb
 - lib/ruby_smb/server/share/provider/processor.rb
+- lib/ruby_smb/server/share/provider/virtual_disk.rb
+- lib/ruby_smb/server/share/provider/virtual_disk/virtual_file.rb
+- lib/ruby_smb/server/share/provider/virtual_disk/virtual_pathname.rb
+- lib/ruby_smb/server/share/provider/virtual_disk/virtual_stat.rb
 - lib/ruby_smb/signing.rb
 - lib/ruby_smb/smb1.rb
 - lib/ruby_smb/smb1/andx_block.rb
@@ -752,14 +767,31 @@ files:
 - spec/lib/ruby_smb/field/utime_spec.rb
 - spec/lib/ruby_smb/fscc/ea_info_array_spec.rb
 - spec/lib/ruby_smb/fscc/file_full_ea_info_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_access_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_alignment_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_all_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_basic_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_both_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_disposition_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_ea_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_full_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_id_both_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_id_full_directory_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_internal_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_mode_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_name_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_names_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_network_open_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_normalized_name_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_position_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_rename_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_standard_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_stream_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information/file_fs_attribute_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information/file_fs_volume_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information_spec.rb
 - spec/lib/ruby_smb/fscc/fscc_file_attributes_spec.rb
 - spec/lib/ruby_smb/generic_packet_spec.rb
 - spec/lib/ruby_smb/gss/provider/ntlm/account_spec.rb
@@ -774,6 +806,9 @@ files:
 - spec/lib/ruby_smb/server/session_spec.rb
 - spec/lib/ruby_smb/server/share/provider/disk_spec.rb
 - spec/lib/ruby_smb/server/share/provider/pipe_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk/virtual_pathname_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk/virtual_stat_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk_spec.rb
 - spec/lib/ruby_smb/server/share/provider_spec.rb
 - spec/lib/ruby_smb/server_spec.rb
 - spec/lib/ruby_smb/smb1/andx_block_spec.rb
@@ -1042,14 +1077,31 @@ test_files:
 - spec/lib/ruby_smb/field/utime_spec.rb
 - spec/lib/ruby_smb/fscc/ea_info_array_spec.rb
 - spec/lib/ruby_smb/fscc/file_full_ea_info_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_access_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_alignment_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_all_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_basic_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_both_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_disposition_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_ea_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_full_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_id_both_directory_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_id_full_directory_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_internal_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_mode_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_name_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_names_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_network_open_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_normalized_name_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_position_information_spec.rb
 - spec/lib/ruby_smb/fscc/file_information/file_rename_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_standard_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information/file_stream_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information/file_fs_attribute_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information/file_fs_volume_information_spec.rb
+- spec/lib/ruby_smb/fscc/file_system_information_spec.rb
 - spec/lib/ruby_smb/fscc/fscc_file_attributes_spec.rb
 - spec/lib/ruby_smb/generic_packet_spec.rb
 - spec/lib/ruby_smb/gss/provider/ntlm/account_spec.rb
@@ -1064,6 +1116,9 @@ test_files:
 - spec/lib/ruby_smb/server/session_spec.rb
 - spec/lib/ruby_smb/server/share/provider/disk_spec.rb
 - spec/lib/ruby_smb/server/share/provider/pipe_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk/virtual_pathname_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk/virtual_stat_spec.rb
+- spec/lib/ruby_smb/server/share/provider/virtual_disk_spec.rb
 - spec/lib/ruby_smb/server/share/provider_spec.rb
 - spec/lib/ruby_smb/server_spec.rb
 - spec/lib/ruby_smb/smb1/andx_block_spec.rb