ruby_smb 2.0.12 → 3.0.2

data/spec/lib/ruby_smb/dcerpc/netlogon/netr_server_authenticate3_request_spec.rb

@@ -19,8 +19,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerAuthenticate3Request do
   end
 
   describe '#account_name' do
-    it 'is a NdrString structure' do
-      expect(packet.account_name).to be_a RubySMB::Dcerpc::Ndr::NdrString
+    it 'is a NdrConfVarWideStringz structure' do
+      expect(packet.account_name).to be_a RubySMB::Dcerpc::Ndr::NdrConfVarWideStringz
     end
   end
 
@@ -31,8 +31,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerAuthenticate3Request do
   end
 
   describe '#computer_name' do
-    it 'is a NdrString structure' do
-      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrString
+    it 'is a NdrConfVarWideStringz structure' do
+      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrConfVarWideStringz
     end
   end
 

data/spec/lib/ruby_smb/dcerpc/netlogon/netr_server_password_set2_request_spec.rb

@@ -19,8 +19,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerPasswordSet2Request do
   end
 
   describe '#account_name' do
-    it 'is a NdrString structure' do
-      expect(packet.account_name).to be_a RubySMB::Dcerpc::Ndr::NdrString
+    it 'is a NdrConfVarWideStringz structure' do
+      expect(packet.account_name).to be_a RubySMB::Dcerpc::Ndr::NdrConfVarWideStringz
     end
   end
 
@@ -31,8 +31,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerPasswordSet2Request do
   end
 
   describe '#computer_name' do
-    it 'is a NdrString structure' do
-      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrString
+    it 'is a NdrConfVarWideStringz structure' do
+      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrConfVarWideStringz
     end
   end
 

data/spec/lib/ruby_smb/dcerpc/netlogon/netr_server_req_challenge_request_spec.rb

@@ -16,8 +16,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerReqChallengeRequest do
   end
 
   describe '#computer_name' do
-    it 'is a NdrString structure' do
-      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrString
+    it 'is a NdrConfVarWideStringz structure' do
+      expect(packet.computer_name).to be_a RubySMB::Dcerpc::Ndr::NdrConfVarWideStringz
     end
   end
 

data/spec/lib/ruby_smb/dcerpc/netlogon/netr_server_req_challenge_response_spec.rb

@@ -15,8 +15,8 @@ RSpec.describe RubySMB::Dcerpc::Netlogon::NetrServerReqChallengeResponse do
   end
 
   describe '#error_status' do
-    it 'is a 32-bit unsigned integer' do
-      expect(packet.error_status).to be_a BinData::Uint32le
+    it 'is a NDR 32-bit unsigned integer' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
     end
   end
 

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

@@ -9,6 +9,10 @@ RSpec.describe RubySMB::Dcerpc::PSyntaxIdT do
     expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
   end
 
+  it 'has :byte_align parameter set to the expected value' do
+    expect(described_class.default_parameters[:byte_align]).to eq(4)
+  end
+
   describe '#if_uuid' do
     it 'is a Uuid' do
       expect(packet.if_uuid).to be_a RubySMB::Dcerpc::Uuid
@@ -16,16 +20,26 @@ RSpec.describe RubySMB::Dcerpc::PSyntaxIdT do
   end
 
   describe '#if_ver_major' do
-    it 'should be a 16-bit unsigned integer' do
-      expect(packet.if_ver_major).to be_a BinData::Uint16le
+    it 'should be a NdrUint16' do
+      expect(packet.if_ver_major).to be_a RubySMB::Dcerpc::Ndr::NdrUint16
     end
   end
 
   describe '#if_ver_minor' do
-    it 'should be a 16-bit unsigned integer' do
-      expect(packet.if_ver_minor).to be_a BinData::Uint16le
+    it 'should be a NdrUint16' do
+      expect(packet.if_ver_minor).to be_a RubySMB::Dcerpc::Ndr::NdrUint16
     end
   end
 
+  it 'reads its own binary representation and output the same packet' do
+    packet = described_class.new(
+      if_uuid: '22fd0d58-357a-472d-9d03-5f19e09b3a92',
+      if_ver_major: rand(0xFF),
+      if_ver_minor: rand(0xFF)
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+
 end
 

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

@@ -37,9 +37,20 @@ RSpec.describe RubySMB::Dcerpc::PDUHeader do
   end
 
   describe '#pfc_flags' do
+    subject(:pfc_flags) { packet.pfc_flags }
+
     it 'should be a custom structure' do
-      expect(packet.pfc_flags).to be_a BinData::Struct
+      expect(pfc_flags).to be_a BinData::Struct
     end
+
+    it { is_expected.to respond_to :object_uuid }
+    it { is_expected.to respond_to :maybe }
+    it { is_expected.to respond_to :did_not_execute }
+    it { is_expected.to respond_to :conc_mpx }
+    it { is_expected.to respond_to :reserved_1 }
+    it { is_expected.to respond_to :support_header_sign }
+    it { is_expected.to respond_to :last_frag }
+    it { is_expected.to respond_to :first_frag }
   end
 
   describe '#packed_drep' do
@@ -79,6 +90,21 @@ RSpec.describe RubySMB::Dcerpc::PDUHeader do
     end
   end
 
+  it 'reads its own binary representation and output the same packet' do
+    packet = described_class.new(
+      rpc_vers: rand(0xFF),
+      rpc_vers_minor: rand(0xFF),
+      ptype: rand(0xFF),
+      pfc_flags: { support_header_sign: 1},
+      packed_drep: rand(0xFF),
+      frag_length: rand(0xFF),
+      auth_length: rand(0xFF),
+      call_id: rand(0xFF)
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+
 end
 
 

data/spec/lib/ruby_smb/dcerpc/print_system/driver_container_spec.rb

@@ -0,0 +1,41 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::DriverContainer do
+  subject(:struct) { described_class.new }
+
+  it { is_expected.to respond_to :level }
+  it { is_expected.to respond_to :tag }
+  it { is_expected.to respond_to :driver_info }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#level' do
+    it 'is a NdrUint32' do
+      expect(struct.level).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#tag' do
+    it 'is a NdrUint32' do
+      expect(struct.tag).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same struct' do
+    struct = described_class.new(
+      level: 2,
+      tag: 0,
+      driver_info: RubySMB::Dcerpc::PrintSystem::DriverInfo2.new(
+        c_version: 0,
+        p_name: 'p_name',
+        p_environment: 'p_environment',
+        p_driver_path: 'p_driver_path',
+        p_data_file: 'p_data_file',
+        p_config_file: 'p_config_file'
+      )
+    )
+    binary = struct.to_binary_s
+    expect(described_class.read(binary)).to eq(struct)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/driver_info2_spec.rb

@@ -0,0 +1,64 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::DriverInfo2 do
+  subject(:struct) { described_class.new }
+
+  it { is_expected.to respond_to :c_version }
+  it { is_expected.to respond_to :p_name }
+  it { is_expected.to respond_to :p_environment }
+  it { is_expected.to respond_to :p_driver_path }
+  it { is_expected.to respond_to :p_data_file }
+  it { is_expected.to respond_to :p_config_file }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#c_version' do
+    it 'is a NdrUint32' do
+      expect(struct.c_version).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#p_name' do
+    it 'is a NdrWideStringzPtr' do
+      expect(struct.p_name).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_environment' do
+    it 'is a NdrWideStringzPtr' do
+      expect(struct.p_environment).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_driver_path' do
+    it 'is a NdrWideStringzPtr' do
+      expect(struct.p_driver_path).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_data_file' do
+    it 'is a NdrWideStringzPtr' do
+      expect(struct.p_data_file).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_config_file' do
+    it 'is a NdrWideStringzPtr' do
+      expect(struct.p_config_file).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same struct' do
+    struct = described_class.new(
+      c_version: 0,
+      p_name: 'p_name',
+      p_environment: 'p_environment',
+      p_driver_path: 'p_driver_path',
+      p_data_file: 'p_data_file',
+      p_config_file: 'p_config_file'
+    )
+    binary = struct.to_binary_s
+    expect(described_class.read(binary)).to eq(struct)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_add_printer_driver_ex_request_spec.rb

@@ -0,0 +1,59 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcAddPrinterDriverExRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :p_name }
+  it { is_expected.to respond_to :p_driver_container }
+  it { is_expected.to respond_to :dw_file_copy_flags }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#p_name' do
+    it 'is a NdrWideStringzPtr' do
+      expect(packet.p_name).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_driver_container' do
+    it 'is a DriverContainer structure' do
+      expect(packet.p_driver_container).to be_a RubySMB::Dcerpc::PrintSystem::DriverContainer
+    end
+  end
+
+  describe '#dw_file_copy_flags' do
+    it 'is a NdrUint32' do
+      expect(packet.dw_file_copy_flags).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_ADD_PRINTER_DRIVER_EX constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_ADD_PRINTER_DRIVER_EX)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      p_name: 'p_name',
+      p_driver_container: RubySMB::Dcerpc::PrintSystem::DriverContainer.new(
+        RubySMB::Dcerpc::PrintSystem::DriverContainer.new(
+          level: 2,
+          tag: 2,
+          driver_info: RubySMB::Dcerpc::PrintSystem::DriverInfo2.new(
+            c_version: 0,
+            p_name: 'p_name',
+            p_environment: 'p_environment',
+            p_driver_path: 'p_driver_path',
+            p_data_file: 'p_data_file',
+            p_config_file: 'p_config_file'
+          )
+        )
+      ),
+      dw_file_copy_flags: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_add_printer_driver_ex_response_spec.rb

@@ -0,0 +1,30 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcAddPrinterDriverExResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :error_status }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#error_status' do
+    it 'is a NdrUint32' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_ADD_PRINTER_DRIVER_EX constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_ADD_PRINTER_DRIVER_EX)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      error_status: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_enum_printer_drivers_request_spec.rb

@@ -0,0 +1,62 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcEnumPrinterDriversRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :p_name }
+  it { is_expected.to respond_to :p_environment }
+  it { is_expected.to respond_to :level }
+  it { is_expected.to respond_to :p_drivers }
+  it { is_expected.to respond_to :cb_buf }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#p_name' do
+    it 'is a NdrWideStringzPtr' do
+      expect(packet.p_name).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_environment' do
+    it 'is a NdrWideStringzPtr' do
+      expect(packet.p_environment).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#level' do
+    it 'is a NdrUint32' do
+      expect(packet.level).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#p_drivers' do
+    it 'is a RprnByteArrayPtr' do
+      expect(packet.p_drivers).to be_a RubySMB::Dcerpc::PrintSystem::RprnByteArrayPtr
+    end
+  end
+
+  describe '#cb_buf' do
+    it 'is a NdrUint32' do
+      expect(packet.cb_buf).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_ENUM_PRINTER_DRIVERS constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_ENUM_PRINTER_DRIVERS)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      p_name: 'p_name',
+      p_environment: 'p_environment',
+      level: 0,
+      p_drivers: [0, 1, 2],
+      cb_buf: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_enum_printer_drivers_response_spec.rb

@@ -0,0 +1,54 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcEnumPrinterDriversResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :p_drivers }
+  it { is_expected.to respond_to :pcb_needed }
+  it { is_expected.to respond_to :pc_returned }
+  it { is_expected.to respond_to :error_status }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#p_drivers' do
+    it 'is a RprnByteArrayPtr' do
+      expect(packet.p_drivers).to be_a RubySMB::Dcerpc::PrintSystem::RprnByteArrayPtr
+    end
+  end
+
+  describe '#pcb_needed' do
+    it 'is a NdrUint32' do
+      expect(packet.pcb_needed).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#pc_returned' do
+    it 'is a NdrUint32' do
+      expect(packet.pc_returned).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#error_status' do
+    it 'is a NdrUint32' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_ENUM_PRINTER_DRIVERS constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_ENUM_PRINTER_DRIVERS)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      p_drivers: [0, 1, 2],
+      pcb_needed: 0,
+      pc_returned: 0,
+      error_status: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_get_printer_driver_directory_request_spec.rb

@@ -0,0 +1,62 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcGetPrinterDriverDirectoryRequest do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :p_name }
+  it { is_expected.to respond_to :p_environment }
+  it { is_expected.to respond_to :level}
+  it { is_expected.to respond_to :p_driver_directory }
+  it { is_expected.to respond_to :cb_buf }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#p_name' do
+    it 'is a NdrWideStringzPtr' do
+      expect(packet.p_name).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#p_environment' do
+    it 'is a NdrWideStringzPtr' do
+      expect(packet.p_environment).to be_a RubySMB::Dcerpc::Ndr::NdrWideStringzPtr
+    end
+  end
+
+  describe '#level' do
+    it 'is a NdrUint32' do
+      expect(packet.level).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#p_driver_directory' do
+    it 'is a RprnByteArrayPtr' do
+      expect(packet.p_driver_directory).to be_a RubySMB::Dcerpc::PrintSystem::RprnByteArrayPtr
+    end
+  end
+
+  describe '#cb_buf' do
+    it 'is a NdrUint32' do
+      expect(packet.cb_buf).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_GET_PRINTER_DRIVER_DIRECTORY constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_GET_PRINTER_DRIVER_DIRECTORY)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      p_name: 'p_name',
+      p_environment: 'p_environment',
+      level: 0,
+      p_driver_directory: [0, 1, 2],
+      cb_buf: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+

data/spec/lib/ruby_smb/dcerpc/print_system/rpc_get_printer_driver_directory_response_spec.rb

@@ -0,0 +1,46 @@
+RSpec.describe RubySMB::Dcerpc::PrintSystem::RpcGetPrinterDriverDirectoryResponse do
+  subject(:packet) { described_class.new }
+
+  it { is_expected.to respond_to :p_driver_directory }
+  it { is_expected.to respond_to :pcb_needed }
+  it { is_expected.to respond_to :error_status }
+
+  it 'is little endian' do
+    expect(described_class.fields.instance_variable_get(:@hints)[:endian]).to eq :little
+  end
+
+  describe '#p_driver_directory' do
+    it 'is a RprnByteArrayPtr' do
+      expect(packet.p_driver_directory).to be_a RubySMB::Dcerpc::PrintSystem::RprnByteArrayPtr
+    end
+  end
+
+  describe '#pcb_needed' do
+    it 'is a NdrUint32' do
+      expect(packet.pcb_needed).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#error_status' do
+    it 'is a NdrUint32' do
+      expect(packet.error_status).to be_a RubySMB::Dcerpc::Ndr::NdrUint32
+    end
+  end
+
+  describe '#initialize_instance' do
+    it 'sets #opnum to RPC_GET_PRINTER_DRIVER_DIRECTORY constant' do
+      expect(packet.opnum).to eq(RubySMB::Dcerpc::PrintSystem::RPC_GET_PRINTER_DRIVER_DIRECTORY)
+    end
+  end
+
+  it 'reads its own binary representation and outputs the same packet' do
+    packet = described_class.new(
+      p_driver_directory: [0, 1, 2],
+      pcb_needed: 0,
+      error_status: 0
+    )
+    binary = packet.to_binary_s
+    expect(described_class.read(binary)).to eq(packet)
+  end
+end
+