ruby_smb 3.1.2 → 3.1.5

data/spec/lib/ruby_smb/dcerpc/samr/samr_enumerate_users_in_domain_response_spec.rb

@@ -1,197 +1,3 @@
-RSpec.describe RubySMB::Dcerpc::Samr::SamprRidEnumeration do
-  subject(:packet) { described_class.new }
-
-  it { is_expected.to respond_to :relative_id }
-  it { is_expected.to respond_to :name }
-
-  it 'is little endian' do
-    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
-  end
-  it 'is a Ndr::NdrStruct' do
-    expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrStruct)
-  end
-  it 'is four-byte aligned' do
-    expect(packet.eval_parameter(:byte_align)).to eq(4)
-  end
-  describe '#relative_id' do
-    it 'is a NdrUint32 structure' do
-      expect(packet.relative_id).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
-    end
-  end
-  describe '#name' do
-    it 'is a RpcUnicodeString structure' do
-      expect(packet.name).to be_a RubySMB::Dcerpc::RpcUnicodeString
-    end
-  end
-  it 'reads itself' do
-    new_packet = described_class.new(relative_id: 4, name: 'Test String')
-    expected_output = {
-      relative_id: 4,
-      name: {
-         buffer_length: 22,
-         maximum_length: 22,
-         buffer: "Test String".encode('utf-16le')
-      }
-    }
-    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
-  end
-end
-
-RSpec.describe RubySMB::Dcerpc::Samr::SamprRidEnumerationArray do
-  subject(:packet) { described_class.new }
-
-  it 'is a Ndr::NdrConfArray' do
-    expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrConfArray)
-  end
-  it 'has element of type SamprRidEnumeration' do
-    packet << {relative_id: 4, name: ''}
-    expect(packet[0]).to be_a(RubySMB::Dcerpc::Samr::SamprRidEnumeration)
-  end
-  it 'reads itself' do
-    new_packet = described_class.new([
-      {relative_id: 4, name: 'Test1'},
-      {relative_id: 1, name: 'Test2'}
-    ])
-    expected_output = [
-    {
-      relative_id: 4,
-      name: {
-         buffer_length: 10,
-         maximum_length: 10,
-         buffer: "Test1".encode('utf-16le')
-      }
-    },
-    {
-      relative_id: 1,
-      name: {
-         buffer_length: 10,
-         maximum_length: 10,
-         buffer: "Test2".encode('utf-16le')
-      }
-    }]
-    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
-  end
-end
-
-RSpec.describe RubySMB::Dcerpc::Samr::PsamprRidEnumerationArray do
-  subject(:packet) { described_class.new }
-
-  it 'is a SamprRidEnumerationArray' do
-    expect(packet).to be_a(RubySMB::Dcerpc::Samr::SamprRidEnumerationArray)
-  end
-  it 'is a NdrPointer' do
-    expect(described_class).to be_a(RubySMB::Dcerpc::Ndr::PointerClassPlugin)
-    expect(packet).to be_a(RubySMB::Dcerpc::Ndr::PointerPlugin)
-  end
-  it 'is four-byte aligned' do
-    expect(packet.eval_parameter(:byte_align)).to eq(4)
-  end
-  it 'reads itself' do
-    new_packet = described_class.new([
-      {relative_id: 4, name: 'Test1'},
-      {relative_id: 1, name: 'Test2'}
-    ])
-    expected_output = [
-    {
-      relative_id: 4,
-      name: {
-         buffer_length: 10,
-         maximum_length: 10,
-         buffer: "Test1".encode('utf-16le')
-      }
-    },
-    {
-      relative_id: 1,
-      name: {
-         buffer_length: 10,
-         maximum_length: 10,
-         buffer: "Test2".encode('utf-16le')
-      }
-    }]
-    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
-  end
-end
-
-RSpec.describe RubySMB::Dcerpc::Samr::SamprEnumerationBuffer do
-  subject(:packet) { described_class.new }
-
-  it { is_expected.to respond_to :entries_read }
-  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
-  it 'is a Ndr::NdrStruct' do
-    expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrStruct)
-  end
-  it 'is four-byte aligned' do
-    expect(packet.eval_parameter(:byte_align)).to eq(4)
-  end
-  describe '#entries_read' do
-    it 'is a NdrUint32 structure' do
-      expect(packet.entries_read).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
-    end
-  end
-  describe '#buffer' do
-    it 'is a PsamprRidEnumerationArray structure' do
-      expect(packet.buffer).to be_a RubySMB::Dcerpc::Samr::PsamprRidEnumerationArray
-    end
-  end
-  it 'reads itself' do
-    new_packet = described_class.new(
-      entries_read: 3,
-      buffer: [
-        { relative_id: 500, name: "Administrator" },
-        { relative_id: 501, name: "Guest" },
-        { relative_id: 1001, name: "WIN-DP0M1BC768$" }
-      ]
-    )
-    expected_output = {
-      entries_read: 3,
-      buffer: [
-        {relative_id: 500, name: { buffer_length: 26, maximum_length: 26, buffer: "Administrator".encode('utf-16le') }},
-        {relative_id: 501, name: { buffer_length: 10, maximum_length: 10, buffer: "Guest".encode('utf-16le') }},
-        {relative_id: 1001, name: { buffer_length: 30, maximum_length: 30, buffer: "WIN-DP0M1BC768$".encode('utf-16le') }}
-      ]
-    }
-    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
-  end
-end
-
-RSpec.describe RubySMB::Dcerpc::Samr::PsamprEnumerationBuffer do
-  subject(:packet) { described_class.new }
-
-  it 'is a SamprEnumerationBuffer' do
-    expect(packet).to be_a(RubySMB::Dcerpc::Samr::SamprEnumerationBuffer)
-  end
-  it 'is a NdrPointer' do
-    expect(described_class).to be_a(RubySMB::Dcerpc::Ndr::PointerClassPlugin)
-    expect(packet).to be_a(RubySMB::Dcerpc::Ndr::PointerPlugin)
-  end
-  it 'is four-byte aligned' do
-    expect(packet.eval_parameter(:byte_align)).to eq(4)
-  end
-  it 'reads itself' do
-    new_packet = described_class.new(
-      entries_read: 3,
-      buffer: [
-        { relative_id: 500, name: "Administrator" },
-        { relative_id: 501, name: "Guest" },
-        { relative_id: 1001, name: "WIN-DP0M1BC768$" }
-      ]
-    )
-    expected_output = {
-      entries_read: 3,
-      buffer: [
-        {relative_id: 500, name: { buffer_length: 26, maximum_length: 26, buffer: "Administrator".encode('utf-16le') }},
-        {relative_id: 501, name: { buffer_length: 10, maximum_length: 10, buffer: "Guest".encode('utf-16le') }},
-        {relative_id: 1001, name: { buffer_length: 30, maximum_length: 30, buffer: "WIN-DP0M1BC768$".encode('utf-16le') }}
-      ]
-    }
-    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
-  end
-end
-
 RSpec.describe RubySMB::Dcerpc::Samr::SamrEnumerateUsersInDomainResponse do
   subject(:packet) { described_class.new }
 
@@ -262,4 +68,3 @@ RSpec.describe RubySMB::Dcerpc::Samr::SamrEnumerateUsersInDomainResponse do
     expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
   end
 end
-

data/spec/lib/ruby_smb/dcerpc/samr/samr_lookup_names_in_domain_request_spec.rb

@@ -0,0 +1,62 @@
+RSpec.describe RubySMB::Dcerpc::Samr::SamrLookupNamesInDomainRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :domain_handle }
+  it { is_expected.to respond_to :names_count }
+  it { is_expected.to respond_to :names }
+  it { is_expected.to respond_to :opnum }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'is a BinData::Record' do
+    expect(packet).to be_a(BinData::Record)
+  end
+
+  describe '#domain_handle' do
+    it 'is a SamprHandle structure' do
+      expect(packet.domain_handle).to be_a RubySMB::Dcerpc::Samr::SamprHandle
+    end
+  end
+
+  describe '#names_count' do
+    it 'is a NdrUint32 structure' do
+      expect(packet.names_count).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#names' do
+    it 'is a RpcUnicodeStringConfVarArray structure' do
+      expect(packet.names).to be_a RubySMB::Dcerpc::RpcUnicodeStringConfVarArray
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to SAMR_LOOKUP_NAMES_IN_DOMAIN constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::Samr::SAMR_LOOKUP_NAMES_IN_DOMAIN)
+    end
+  end
+
+  it 'reads itself' do
+    new_packet = described_class.new({
+      domain_handle: {
+        context_handle_attributes: 0,
+        context_handle_uuid: "fc873b90-d9a9-46a4-b9ea-f44bb1c272a7"
+      },
+      names_count: 1,
+      names: [ 'TEST' ]
+    })
+    expected_output = {
+      domain_handle: {
+        context_handle_attributes: 0,
+        context_handle_uuid: "fc873b90-d9a9-46a4-b9ea-f44bb1c272a7"
+      },
+      names_count: 1,
+      names: [
+        { buffer_length: 8, maximum_length: 8, buffer: "TEST".encode('utf-16le') }
+      ]
+    }
+    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+  end
+end

data/spec/lib/ruby_smb/dcerpc/samr/samr_lookup_names_in_domain_response_spec.rb

@@ -0,0 +1,54 @@
+RSpec.describe RubySMB::Dcerpc::Samr::SamrLookupNamesInDomainResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :relative_ids }
+  it { is_expected.to respond_to :use }
+  it { is_expected.to respond_to :error_status }
+  it { is_expected.to respond_to :opnum }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'is a BinData::Record' do
+    expect(packet).to be_a(BinData::Record)
+  end
+
+  describe '#relative_ids' do
+    it 'is a SamprHandle structure' do
+      expect(packet.relative_ids).to be_a RubySMB::Dcerpc::Samr::SamprUlongArray
+    end
+  end
+
+  describe '#use' do
+    it 'is a SamprHandle structure' do
+      expect(packet.use).to be_a RubySMB::Dcerpc::Samr::SamprUlongArray
+    end
+  end
+
+  describe '#error_status' do
+    it 'is a NdrUint32 structure' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to SAMR_LOOKUP_NAMES_IN_DOMAIN constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::Samr::SAMR_LOOKUP_NAMES_IN_DOMAIN)
+    end
+  end
+
+  it 'reads itself' do
+    new_packet = described_class.new({
+      relative_ids: { element_count: 2, elements: [ 500, 501 ] },
+      use: { element_count: 2, elements:  [ 1, 2 ] },
+      error_status: 0x11223344
+    })
+    expected_output = {
+      relative_ids: { element_count: 2, elements: [ 500, 501 ] },
+      use: { element_count: 2, elements:  [ 1, 2 ] },
+      error_status: 0x11223344
+    }
+    expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+  end
+end

data/spec/lib/ruby_smb/dcerpc/samr/samr_set_information_user2_request_spec.rb

@@ -0,0 +1,67 @@
+RSpec.describe RubySMB::Dcerpc::Samr::SamrSetInformationUser2Request do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :user_handle }
+  it { is_expected.to respond_to :user_information_class }
+  it { is_expected.to respond_to :buffer }
+  it { is_expected.to respond_to :opnum }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'is a BinData::Record' do
+    expect(packet).to be_a(BinData::Record)
+  end
+
+  describe '#user_handle' do
+    it 'is a SamprHandle structure' do
+      expect(packet.user_handle).to be_a RubySMB::Dcerpc::Samr::SamprHandle
+    end
+  end
+
+  describe '#user_information_class' do
+    it 'is a NdrUint16 structure' do
+      expect(packet.user_information_class).to be_a RubySMB::Dcerpc::Ndr::NdrUint16
+    end
+  end
+
+  describe '#buffer' do
+    it 'is a SamprUserInfoBuffer structure' do
+      expect(packet.buffer).to be_a RubySMB::Dcerpc::Samr::SamprUserInfoBuffer
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to SAMR_SET_INFORMATION_USER2 constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::Samr::SAMR_SET_INFORMATION_USER2)
+    end
+  end
+
+  it 'reads itself' do
+    new_class = described_class.new(
+      user_handle: {
+        context_handle_attributes: 0,
+        context_handle_uuid: '2ef54a87-e29e-4d24-90e9-9da49b94449e'
+      },
+      user_information_class: RubySMB::Dcerpc::Samr::USER_CONTROL_INFORMATION,
+      buffer: {
+        tag: RubySMB::Dcerpc::Samr::USER_CONTROL_INFORMATION,
+        member: { user_account_control: RubySMB::Dcerpc::Samr::USER_WORKSTATION_TRUST_ACCOUNT }
+      }
+    )
+    expect(packet.read(new_class.to_binary_s)).to eq(
+      {
+        user_handle: {
+          context_handle_attributes: 0,
+          context_handle_uuid: '2ef54a87-e29e-4d24-90e9-9da49b94449e'
+        },
+        user_information_class: RubySMB::Dcerpc::Samr::USER_CONTROL_INFORMATION,
+        buffer: {
+          tag: RubySMB::Dcerpc::Samr::USER_CONTROL_INFORMATION,
+          member: { user_account_control: RubySMB::Dcerpc::Samr::USER_WORKSTATION_TRUST_ACCOUNT }
+        }
+      }
+    )
+  end
+end

data/spec/lib/ruby_smb/dcerpc/samr/samr_set_information_user2_response_spec.rb

@@ -0,0 +1,35 @@
+RSpec.describe RubySMB::Dcerpc::Samr::SamrSetInformationUser2Response do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :error_status }
+  it { is_expected.to respond_to :opnum }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  it 'is a BinData::Record' do
+    expect(packet).to be_a(BinData::Record)
+  end
+
+  describe '#error_status' do
+    it 'is a NdrUint32 structure' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to SAMR_SET_INFORMATION_USER2 constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::Samr::SAMR_SET_INFORMATION_USER2)
+    end
+  end
+
+  it 'reads itself' do
+    new_class = described_class.new(error_status: 0x11223344)
+    expect(packet.read(new_class.to_binary_s)).to eq(
+      {
+        error_status: 0x11223344
+      }
+    )
+  end
+end

data/spec/lib/ruby_smb/dcerpc/samr_spec.rb

@@ -7,6 +7,200 @@ RSpec.describe RubySMB::Dcerpc::Samr do
     )
   end
 
+  describe described_class::SamprRidEnumeration do
+    subject(:packet) { described_class.new }
+
+    it { is_expected.to respond_to :relative_id }
+    it { is_expected.to respond_to :name }
+
+    it 'is little endian' do
+      expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+    end
+    it 'is a Ndr::NdrStruct' do
+      expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrStruct)
+    end
+    it 'is four-byte aligned' do
+      expect(packet.eval_parameter(:byte_align)).to eq(4)
+    end
+    describe '#relative_id' do
+      it 'is a NdrUint32 structure' do
+        expect(packet.relative_id).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+      end
+    end
+    describe '#name' do
+      it 'is a RpcUnicodeString structure' do
+        expect(packet.name).to be_a RubySMB::Dcerpc::RpcUnicodeString
+      end
+    end
+    it 'reads itself' do
+      new_packet = described_class.new(relative_id: 4, name: 'Test String')
+      expected_output = {
+        relative_id: 4,
+        name: {
+           buffer_length: 22,
+           maximum_length: 22,
+           buffer: "Test String".encode('utf-16le')
+        }
+      }
+      expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+    end
+  end
+
+  describe described_class::SamprRidEnumerationArray do
+    subject(:packet) { described_class.new }
+
+    it 'is a Ndr::NdrConfArray' do
+      expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrConfArray)
+    end
+    it 'has element of type SamprRidEnumeration' do
+      packet << {relative_id: 4, name: ''}
+      expect(packet[0]).to be_a(RubySMB::Dcerpc::Samr::SamprRidEnumeration)
+    end
+    it 'reads itself' do
+      new_packet = described_class.new([
+        {relative_id: 4, name: 'Test1'},
+        {relative_id: 1, name: 'Test2'}
+      ])
+      expected_output = [
+      {
+        relative_id: 4,
+        name: {
+           buffer_length: 10,
+           maximum_length: 10,
+           buffer: "Test1".encode('utf-16le')
+        }
+      },
+      {
+        relative_id: 1,
+        name: {
+           buffer_length: 10,
+           maximum_length: 10,
+           buffer: "Test2".encode('utf-16le')
+        }
+      }]
+      expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+    end
+  end
+
+  describe described_class::PsamprRidEnumerationArray do
+    subject(:packet) { described_class.new }
+
+    it 'is a SamprRidEnumerationArray' do
+      expect(packet).to be_a(RubySMB::Dcerpc::Samr::SamprRidEnumerationArray)
+    end
+    it 'is a NdrPointer' do
+      expect(described_class).to be_a(RubySMB::Dcerpc::Ndr::PointerClassPlugin)
+      expect(packet).to be_a(RubySMB::Dcerpc::Ndr::PointerPlugin)
+    end
+    it 'is four-byte aligned' do
+      expect(packet.eval_parameter(:byte_align)).to eq(4)
+    end
+    it 'reads itself' do
+      new_packet = described_class.new([
+        {relative_id: 4, name: 'Test1'},
+        {relative_id: 1, name: 'Test2'}
+      ])
+      expected_output = [
+      {
+        relative_id: 4,
+        name: {
+           buffer_length: 10,
+           maximum_length: 10,
+           buffer: "Test1".encode('utf-16le')
+        }
+      },
+      {
+        relative_id: 1,
+        name: {
+           buffer_length: 10,
+           maximum_length: 10,
+           buffer: "Test2".encode('utf-16le')
+        }
+      }]
+      expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+    end
+  end
+
+  describe described_class::SamprEnumerationBuffer do
+    subject(:packet) { described_class.new }
+
+    it { is_expected.to respond_to :entries_read }
+    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
+    it 'is a Ndr::NdrStruct' do
+      expect(packet).to be_a(RubySMB::Dcerpc::Ndr::NdrStruct)
+    end
+    it 'is four-byte aligned' do
+      expect(packet.eval_parameter(:byte_align)).to eq(4)
+    end
+    describe '#entries_read' do
+      it 'is a NdrUint32 structure' do
+        expect(packet.entries_read).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+      end
+    end
+    describe '#buffer' do
+      it 'is a PsamprRidEnumerationArray structure' do
+        expect(packet.buffer).to be_a RubySMB::Dcerpc::Samr::PsamprRidEnumerationArray
+      end
+    end
+    it 'reads itself' do
+      new_packet = described_class.new(
+        entries_read: 3,
+        buffer: [
+          { relative_id: 500, name: "Administrator" },
+          { relative_id: 501, name: "Guest" },
+          { relative_id: 1001, name: "WIN-DP0M1BC768$" }
+        ]
+      )
+      expected_output = {
+        entries_read: 3,
+        buffer: [
+          {relative_id: 500, name: { buffer_length: 26, maximum_length: 26, buffer: "Administrator".encode('utf-16le') }},
+          {relative_id: 501, name: { buffer_length: 10, maximum_length: 10, buffer: "Guest".encode('utf-16le') }},
+          {relative_id: 1001, name: { buffer_length: 30, maximum_length: 30, buffer: "WIN-DP0M1BC768$".encode('utf-16le') }}
+        ]
+      }
+      expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+    end
+  end
+
+  describe described_class::PsamprEnumerationBuffer do
+    subject(:packet) { described_class.new }
+
+    it 'is a SamprEnumerationBuffer' do
+      expect(packet).to be_a(RubySMB::Dcerpc::Samr::SamprEnumerationBuffer)
+    end
+    it 'is a NdrPointer' do
+      expect(described_class).to be_a(RubySMB::Dcerpc::Ndr::PointerClassPlugin)
+      expect(packet).to be_a(RubySMB::Dcerpc::Ndr::PointerPlugin)
+    end
+    it 'is four-byte aligned' do
+      expect(packet.eval_parameter(:byte_align)).to eq(4)
+    end
+    it 'reads itself' do
+      new_packet = described_class.new(
+        entries_read: 3,
+        buffer: [
+          { relative_id: 500, name: "Administrator" },
+          { relative_id: 501, name: "Guest" },
+          { relative_id: 1001, name: "WIN-DP0M1BC768$" }
+        ]
+      )
+      expected_output = {
+        entries_read: 3,
+        buffer: [
+          {relative_id: 500, name: { buffer_length: 26, maximum_length: 26, buffer: "Administrator".encode('utf-16le') }},
+          {relative_id: 501, name: { buffer_length: 10, maximum_length: 10, buffer: "Guest".encode('utf-16le') }},
+          {relative_id: 1001, name: { buffer_length: 30, maximum_length: 30, buffer: "WIN-DP0M1BC768$".encode('utf-16le') }}
+        ]
+      }
+      expect(packet.read(new_packet.to_binary_s)).to eq(expected_output)
+    end
+  end
+
   describe described_class::SamprHandle do
     it 'is a Ndr::NdrContextHandle' do
       expect(described_class).to be < RubySMB::Dcerpc::Ndr::NdrContextHandle