ruby_smb 1.0.0 → 1.1.0

data/spec/lib/ruby_smb/smb2/tree_spec.rb

@@ -48,6 +48,13 @@ RSpec.describe RubySMB::SMB2::Tree do
       tree.disconnect!
     end
 
+    it 'raises an InvalidPacket exception if the response is not valid' do
+      allow(client).to receive(:send_recv)
+      allow(RubySMB::SMB2::Packet::TreeDisconnectResponse).to receive(:read).and_return(disco_resp)
+      allow(disco_resp).to receive(:valid?).and_return(false)
+      expect { tree.disconnect! }.to raise_error(RubySMB::Error::InvalidPacket)
+    end
+
     it 'returns the NTStatus code from the response' do
       allow(client).to receive(:send_recv).and_return(disco_resp.to_binary_s)
       expect(tree.disconnect!).to eq disco_resp.status_code
@@ -129,6 +136,15 @@ RSpec.describe RubySMB::SMB2::Tree do
       expect(client).to receive(:send_recv).with(create_req).and_return(create_response.to_binary_s)
       tree.open_directory
     end
+
+    context 'when the response is not a valid packet' do
+      it 'raises an InvalidPacket exception' do
+        allow(client).to receive(:send_recv)
+        allow(RubySMB::SMB2::Packet::CreateResponse).to receive(:read).and_return(create_response)
+        create_response.smb2_header.command = RubySMB::SMB2::Commands::LOGOFF
+        expect { tree.open_directory }.to raise_error(RubySMB::Error::InvalidPacket)
+      end
+    end
   end
 
   describe '#list' do
@@ -202,6 +218,13 @@ RSpec.describe RubySMB::SMB2::Tree do
       expect(tree.list).to eq([file1])
     end
 
+    context 'when the response is not a valid packet' do
+      it 'raises an InvalidPacket exception' do
+        query_dir_res.smb2_header.command = RubySMB::SMB2::Commands::LOGOFF
+        expect { tree.list }.to raise_error(RubySMB::Error::InvalidPacket)
+      end
+    end
+
     context 'when multiple requests are needed to retrieve the full directory list' do
       before :example do
         query_dir_res.smb2_header.nt_status = WindowsError::NTStatus::STATUS_SUCCESS.value
@@ -238,4 +261,195 @@ RSpec.describe RubySMB::SMB2::Tree do
     end
 
   end
+
+  describe '#open_file' do
+    let(:create_request)      { RubySMB::SMB2::Packet::CreateRequest.new }
+    let(:create_response) { RubySMB::SMB2::Packet::CreateResponse.new }
+    let(:filename) { "test_file\x00".encode('UTF-16LE') }
+
+    before :each do
+      allow(RubySMB::SMB2::Packet::CreateRequest).to receive(:new).and_return(create_request)
+      allow(RubySMB::SMB2::Packet::CreateResponse).to receive(:read).and_return(create_response)
+    end
+
+    it 'calls #set_header_fields' do
+      allow(client).to receive(:send_recv).and_return(create_response.to_binary_s)
+      expect(tree).to receive(:set_header_fields).with(create_request).and_call_original
+      tree.open_file(filename: filename)
+    end
+
+    describe 'filename' do
+      it 'takes the filename as an argument' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.name).to eq(filename)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename)
+      end
+    end
+
+    describe 'attributes' do
+      it 'has the correct default fields set' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.file_attributes.directory).to eq(0)
+          expect(packet.file_attributes.normal).to eq(1)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename)
+      end
+
+      it 'can take the Attributes as an argument' do
+        attributes = { normal: 0 }
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.file_attributes.normal).to eq(0)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, attributes: attributes)
+      end
+    end
+
+    describe 'options' do
+      it 'has the correct default fields set' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.create_options.directory_file).to eq(0)
+          expect(packet.create_options.non_directory_file).to eq(1)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename)
+      end
+
+      it 'can take the Create Options as an argument' do
+        options = RubySMB::SMB1::BitField::CreateOptions.new(directory_file: 1)
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.create_options.directory_file).to eq(1)
+          expect(packet.create_options.non_directory_file).to eq(0)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, options: options)
+      end
+    end
+
+    describe 'disposition' do
+      it 'defaults to FILE_OPEN' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.create_disposition).to eq(RubySMB::Dispositions::FILE_OPEN)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename)
+      end
+
+      it 'can take the Disposition as an argument' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.create_disposition).to eq(RubySMB::Dispositions::FILE_OPEN_IF)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, disposition: RubySMB::Dispositions::FILE_OPEN_IF)
+      end
+    end
+
+    describe 'impersonation level' do
+      it 'defaults to SEC_IMPERSONATE' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.impersonation_level).to eq(RubySMB::ImpersonationLevels::SEC_IMPERSONATE)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename)
+      end
+
+      it 'can take the Impersonation Level as an argument' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.impersonation_level).to eq(RubySMB::ImpersonationLevels::SEC_DELEGATE)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, impersonation: RubySMB::ImpersonationLevels::SEC_DELEGATE)
+      end
+    end
+
+    describe 'RWD access permissions' do
+      it 'will set the read permission from the parameters' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.share_access.read_access).to    eq(1)
+          expect(packet.desired_access.read_data).to    eq(1)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, read: true)
+      end
+
+      it 'will set the write permission from the parameters' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.share_access.write_access).to   eq(1)
+          expect(packet.desired_access.write_data).to  eq(1)
+          expect(packet.desired_access.append_data).to eq(1)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, write: true)
+      end
+
+      it 'will set the delete permission from the parameters' do
+        allow(client).to receive(:send_recv) do |packet|
+          expect(packet.share_access.delete_access).to    eq(1)
+          expect(packet.desired_access.delete_access).to eq(1)
+          create_response.to_binary_s
+        end
+        tree.open_file(filename: filename, delete: true)
+      end
+    end
+
+    it 'sets #requested_oplock to 0xFF' do
+      allow(client).to receive(:send_recv) do |packet|
+        expect(packet.requested_oplock).to eq(0xFF)
+        create_response.to_binary_s
+      end
+      tree.open_file(filename: filename, delete: true)
+    end
+
+    it 'sends the CreateRequest request packet and gets the expected CreateResponse response back' do
+      expect(client).to receive(:send_recv).with(create_request).and_return(create_response.to_binary_s)
+      tree.open_file(filename: filename)
+    end
+
+    context 'when sending the request packet and gets a response back' do
+      before :example do
+        allow(client).to receive(:send_recv).with(create_request).and_return(create_response.to_binary_s)
+      end
+
+      context 'when it is a file' do
+        it 'returns the expected RubySMB::SMB2::File object' do
+          file_obj = RubySMB::SMB2::File.new(name: filename, tree: tree, response: create_response)
+          expect(RubySMB::SMB2::File).to receive(:new).with(name: filename, tree: tree, response: create_response).and_return(file_obj)
+          expect(tree.open_file(filename: filename)).to eq(file_obj)
+        end
+      end
+
+      context 'when it is a pipe' do
+        it 'returns the expected RubySMB::SMB2::Pipe object' do
+          response.share_type = 0x02
+          pipe_obj = RubySMB::SMB2::Pipe.new(name: filename, tree: tree, response: create_response)
+          expect(RubySMB::SMB2::Pipe).to receive(:new).with(name: filename, tree: tree, response: create_response).and_return(pipe_obj)
+          expect(tree.open_file(filename: filename)).to eq(pipe_obj)
+        end
+      end
+
+      context 'when it is an unsupported share type' do
+        it 'raises a RubySMBError exception' do
+          response.share_type = 0x03
+          expect { tree.open_file(filename: filename) }.to raise_error(RubySMB::Error::RubySMBError)
+        end
+      end
+
+      context 'when the response is not a CreateResponse packet' do
+        it 'raises an InvalidPacket exception' do
+          create_response.smb2_header.command = RubySMB::SMB2::Commands::LOGOFF
+          expect { tree.open_file(filename: filename) }.to raise_error(RubySMB::Error::InvalidPacket)
+        end
+      end
+
+      context 'when the response status code is not STATUS_SUCCESS' do
+        it 'raises an UnexpectedStatusCode exception' do
+          create_response.smb2_header.nt_status = WindowsError::NTStatus::STATUS_INVALID_HANDLE.value
+          expect { tree.open_file(filename: filename) }.to raise_error(RubySMB::Error::UnexpectedStatusCode)
+        end
+      end
+    end
+  end
 end

metadata

@@ -1,9 +1,10 @@
 --- !ruby/object:Gem::Specification
 name: ruby_smb
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
+- Metasploit Hackers
 - David Maloney
 - James Lee
 - Dev Mohanty
@@ -91,7 +92,7 @@ cert_chain:
   G+Hmcg1v810agasPdoydE0RTVZgEOOMoQ07qu7JFXVWZ9ZQpHT7qJATWL/b2csFG
   8mVuTXnyJOKRJA==
   -----END CERTIFICATE-----
-date: 2018-06-25 00:00:00.000000000 Z
+date: 2019-06-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet
@@ -207,10 +208,7 @@ dependencies:
         version: '0'
 description: ''
 email:
-- DMaloney@rapid7.com
-- egypt@metasploit.com
-- dev_mohanty@rapid7.com
-- paristvinternet-github@yahoo.com
+- msfdev@metasploit.com
 executables: []
 extensions: []
 extra_rdoc_files: []
@@ -229,12 +227,15 @@ files:
 - examples/append_file.rb
 - examples/authenticate.rb
 - examples/delete_file.rb
+- examples/enum_registry_key.rb
+- examples/enum_registry_values.rb
 - examples/list_directory.rb
 - examples/negotiate.rb
 - examples/negotiate_with_netbios_service.rb
 - examples/net_share_enum_all.rb
 - examples/pipes.rb
 - examples/read_file.rb
+- examples/read_registry_key_value.rb
 - examples/rename_file.rb
 - examples/tree_connect.rb
 - examples/write_file.rb
@@ -246,6 +247,7 @@ files:
 - lib/ruby_smb/client/signing.rb
 - lib/ruby_smb/client/tree_connect.rb
 - lib/ruby_smb/client/utils.rb
+- lib/ruby_smb/client/winreg.rb
 - lib/ruby_smb/dcerpc.rb
 - lib/ruby_smb/dcerpc/bind.rb
 - lib/ruby_smb/dcerpc/bind_ack.rb
@@ -256,9 +258,26 @@ files:
 - lib/ruby_smb/dcerpc/ptypes.rb
 - lib/ruby_smb/dcerpc/request.rb
 - lib/ruby_smb/dcerpc/response.rb
+- lib/ruby_smb/dcerpc/rrp_unicode_string.rb
 - lib/ruby_smb/dcerpc/srvsvc.rb
 - lib/ruby_smb/dcerpc/srvsvc/net_share_enum_all.rb
 - lib/ruby_smb/dcerpc/uuid.rb
+- lib/ruby_smb/dcerpc/winreg.rb
+- lib/ruby_smb/dcerpc/winreg/close_key_request.rb
+- lib/ruby_smb/dcerpc/winreg/close_key_response.rb
+- lib/ruby_smb/dcerpc/winreg/enum_key_request.rb
+- lib/ruby_smb/dcerpc/winreg/enum_key_response.rb
+- lib/ruby_smb/dcerpc/winreg/enum_value_request.rb
+- lib/ruby_smb/dcerpc/winreg/enum_value_response.rb
+- lib/ruby_smb/dcerpc/winreg/open_key_request.rb
+- lib/ruby_smb/dcerpc/winreg/open_key_response.rb
+- lib/ruby_smb/dcerpc/winreg/open_root_key_request.rb
+- lib/ruby_smb/dcerpc/winreg/open_root_key_response.rb
+- lib/ruby_smb/dcerpc/winreg/query_info_key_request.rb
+- lib/ruby_smb/dcerpc/winreg/query_info_key_response.rb
+- lib/ruby_smb/dcerpc/winreg/query_value_request.rb
+- lib/ruby_smb/dcerpc/winreg/query_value_response.rb
+- lib/ruby_smb/dcerpc/winreg/regsam.rb
 - lib/ruby_smb/dispatcher.rb
 - lib/ruby_smb/dispatcher/base.rb
 - lib/ruby_smb/dispatcher/socket.rb
@@ -324,7 +343,6 @@ files:
 - lib/ruby_smb/smb1/commands.rb
 - lib/ruby_smb/smb1/create_actions.rb
 - lib/ruby_smb/smb1/data_block.rb
-- lib/ruby_smb/smb1/dcerpc.rb
 - lib/ruby_smb/smb1/dialect.rb
 - lib/ruby_smb/smb1/file.rb
 - lib/ruby_smb/smb1/oplock_levels.rb
@@ -402,7 +420,6 @@ files:
 - lib/ruby_smb/smb2/bit_field/smb2_security_mode_single.rb
 - lib/ruby_smb/smb2/commands.rb
 - lib/ruby_smb/smb2/create_context.rb
-- lib/ruby_smb/smb2/dcerpc.rb
 - lib/ruby_smb/smb2/file.rb
 - lib/ruby_smb/smb2/info_type.rb
 - lib/ruby_smb/smb2/packet.rb
@@ -441,12 +458,32 @@ files:
 - spec/lib/ruby_smb/client_spec.rb
 - spec/lib/ruby_smb/dcerpc/bind_ack_spec.rb
 - spec/lib/ruby_smb/dcerpc/bind_spec.rb
+- spec/lib/ruby_smb/dcerpc/ndr_spec.rb
 - spec/lib/ruby_smb/dcerpc/p_syntax_id_t_spec.rb
 - spec/lib/ruby_smb/dcerpc/pdu_header_spec.rb
 - spec/lib/ruby_smb/dcerpc/request_spec.rb
 - spec/lib/ruby_smb/dcerpc/response_spec.rb
+- spec/lib/ruby_smb/dcerpc/rrp_unicode_string_spec.rb
 - spec/lib/ruby_smb/dcerpc/srvsvc/net_share_enum_all_spec.rb
+- spec/lib/ruby_smb/dcerpc/srvsvc_spec.rb
 - spec/lib/ruby_smb/dcerpc/uuid_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/close_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/close_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_value_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_value_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_root_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_root_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_info_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_info_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_value_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_value_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/regsam_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg_spec.rb
+- spec/lib/ruby_smb/dcerpc_spec.rb
 - spec/lib/ruby_smb/dispatcher/base_spec.rb
 - spec/lib/ruby_smb/dispatcher/socket_spec.rb
 - spec/lib/ruby_smb/field/extended_attribute_flag_spec.rb
@@ -503,7 +540,7 @@ files:
 - spec/lib/ruby_smb/smb1/packet/close_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/echo_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/echo_response_spec.rb
-- spec/lib/ruby_smb/smb1/packet/error_packet_spec.rb
+- spec/lib/ruby_smb/smb1/packet/empty_packet_spec.rb
 - spec/lib/ruby_smb/smb1/packet/logoff_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/logoff_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/negotiate_request_spec.rb
@@ -566,6 +603,7 @@ files:
 - spec/lib/ruby_smb/smb2/packet/create_response_spec.rb
 - spec/lib/ruby_smb/smb2/packet/echo_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/echo_response_spec.rb
+- spec/lib/ruby_smb/smb2/packet/error_packet_spec.rb
 - spec/lib/ruby_smb/smb2/packet/ioctl_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/logoff_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/logoff_response_spec.rb
@@ -604,7 +642,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.2.0
+      version: 2.3.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
@@ -620,12 +658,32 @@ test_files:
 - spec/lib/ruby_smb/client_spec.rb
 - spec/lib/ruby_smb/dcerpc/bind_ack_spec.rb
 - spec/lib/ruby_smb/dcerpc/bind_spec.rb
+- spec/lib/ruby_smb/dcerpc/ndr_spec.rb
 - spec/lib/ruby_smb/dcerpc/p_syntax_id_t_spec.rb
 - spec/lib/ruby_smb/dcerpc/pdu_header_spec.rb
 - spec/lib/ruby_smb/dcerpc/request_spec.rb
 - spec/lib/ruby_smb/dcerpc/response_spec.rb
+- spec/lib/ruby_smb/dcerpc/rrp_unicode_string_spec.rb
 - spec/lib/ruby_smb/dcerpc/srvsvc/net_share_enum_all_spec.rb
+- spec/lib/ruby_smb/dcerpc/srvsvc_spec.rb
 - spec/lib/ruby_smb/dcerpc/uuid_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/close_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/close_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_value_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/enum_value_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_root_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/open_root_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_info_key_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_info_key_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_value_request_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/query_value_response_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg/regsam_spec.rb
+- spec/lib/ruby_smb/dcerpc/winreg_spec.rb
+- spec/lib/ruby_smb/dcerpc_spec.rb
 - spec/lib/ruby_smb/dispatcher/base_spec.rb
 - spec/lib/ruby_smb/dispatcher/socket_spec.rb
 - spec/lib/ruby_smb/field/extended_attribute_flag_spec.rb
@@ -682,7 +740,7 @@ test_files:
 - spec/lib/ruby_smb/smb1/packet/close_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/echo_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/echo_response_spec.rb
-- spec/lib/ruby_smb/smb1/packet/error_packet_spec.rb
+- spec/lib/ruby_smb/smb1/packet/empty_packet_spec.rb
 - spec/lib/ruby_smb/smb1/packet/logoff_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/logoff_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/negotiate_request_spec.rb
@@ -745,6 +803,7 @@ test_files:
 - spec/lib/ruby_smb/smb2/packet/create_response_spec.rb
 - spec/lib/ruby_smb/smb2/packet/echo_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/echo_response_spec.rb
+- spec/lib/ruby_smb/smb2/packet/error_packet_spec.rb
 - spec/lib/ruby_smb/smb2/packet/ioctl_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/logoff_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/logoff_response_spec.rb

data/lib/ruby_smb/smb1/dcerpc.rb

@@ -1,67 +0,0 @@
-module RubySMB
-  module SMB1
-    module Dcerpc
-
-      def net_share_enum_all(host)
-        bind(endpoint: RubySMB::Dcerpc::Srvsvc)
-
-        response = request(RubySMB::Dcerpc::Srvsvc::NET_SHARE_ENUM_ALL, host: host)
-
-        shares = RubySMB::Dcerpc::Srvsvc::NetShareEnumAll.parse_response(response.stub.to_binary_s)
-        shares.map{|s|{name: s[0], type: s[1], comment: s[2]}}
-      end
-
-      def bind(options={})
-        bind_req = RubySMB::Dcerpc::Bind.new(options)
-        write(data: bind_req.to_binary_s)
-        @size = 1024
-        dcerpc_raw_response = read()
-        begin
-          dcerpc_response = RubySMB::Dcerpc::BindAck.read(dcerpc_raw_response)
-        rescue IOError
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Error reading the DCERPC response"
-        end
-        unless dcerpc_response.pdu_header.ptype == RubySMB::Dcerpc::PTypes::BIND_ACK
-          raise RubySMB::Dcerpc::Error::BindError, "Not a BindAck packet"
-        end
-
-        res_list = dcerpc_response.p_result_list
-        if res_list.n_results == 0 ||
-           res_list.p_results[0].result != RubySMB::Dcerpc::BindAck::ACCEPTANCE
-          raise RubySMB::Dcerpc::Error::BindError,
-            "Bind Failed (Result: #{res_list.p_results[0].result}, Reason: #{res_list.p_results[0].reason})"
-        end
-        dcerpc_response
-      end
-
-      def request(opnum, options={})
-        dcerpc_request = RubySMB::Dcerpc::Request.new({ :opnum => opnum }, options)
-        request = RubySMB::SMB1::Packet::Trans::TransactNmpipeRequest.new(options)
-        @tree.set_header_fields(request)
-        request.set_fid(@fid)
-        request.data_block.trans_data.write_data = dcerpc_request.to_binary_s
-
-        trans_nmpipe_raw_response = @tree.client.send_recv(request)
-        trans_nmpipe_response = RubySMB::SMB1::Packet::Trans::TransactNmpipeResponse.read(trans_nmpipe_raw_response)
-        unless trans_nmpipe_response.smb_header.command == RubySMB::SMB1::Commands::SMB_COM_TRANSACTION
-          raise RubySMB::Error::InvalidPacket, 'Not a Trans packet'
-        end
-        unless trans_nmpipe_response.status_code == WindowsError::NTStatus::STATUS_SUCCESS
-          raise RubySMB::Error::UnexpectedStatusCode, trans_nmpipe_response.status_code.name
-        end
-
-        begin
-          dcerpc_response = RubySMB::Dcerpc::Response.read(trans_nmpipe_response.data_block.trans_data.read_data)
-        rescue IOError
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Error reading the DCERPC response"
-        end
-        unless dcerpc_response.pdu_header.ptype == RubySMB::Dcerpc::PTypes::RESPONSE
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Not a Response packet"
-        end
-        dcerpc_response
-      end
-
-    end
-  end
-end
-

data/lib/ruby_smb/smb2/dcerpc.rb

@@ -1,70 +0,0 @@
-module RubySMB
-  module SMB2
-    module Dcerpc
-
-      def net_share_enum_all(host)
-        bind(endpoint: RubySMB::Dcerpc::Srvsvc)
-
-        response = request(RubySMB::Dcerpc::Srvsvc::NET_SHARE_ENUM_ALL, host: host)
-
-        shares = RubySMB::Dcerpc::Srvsvc::NetShareEnumAll.parse_response(response.stub.to_binary_s)
-        shares.map{|s|{name: s[0], type: s[1], comment: s[2]}}
-      end
-
-      def bind(options={})
-        bind_req = RubySMB::Dcerpc::Bind.new(options)
-        write(data: bind_req.to_binary_s)
-        @size = 1024
-        dcerpc_raw_response = read()
-        begin
-          dcerpc_response = RubySMB::Dcerpc::BindAck.read(dcerpc_raw_response)
-        rescue IOError
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Error reading the DCERPC response"
-        end
-        unless dcerpc_response.pdu_header.ptype == RubySMB::Dcerpc::PTypes::BIND_ACK
-          raise RubySMB::Dcerpc::Error::BindError, "Not a BindAck packet"
-        end
-
-        res_list = dcerpc_response.p_result_list
-        if res_list.n_results == 0 ||
-           res_list.p_results[0].result != RubySMB::Dcerpc::BindAck::ACCEPTANCE
-          raise RubySMB::Dcerpc::Error::BindError,
-            "Bind Failed (Result: #{res_list.p_results[0].result}, Reason: #{res_list.p_results[0].reason})"
-        end
-        dcerpc_response
-      end
-
-      def request(opnum, options={})
-        dcerpc_request = RubySMB::Dcerpc::Request.new({ :opnum => opnum }, options)
-        ioctl_response = ioctl_send_recv(dcerpc_request, options)
-        begin
-          dcerpc_response = RubySMB::Dcerpc::Response.read(ioctl_response.output_data)
-        rescue IOError
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Error reading the DCERPC response"
-        end
-        unless dcerpc_response.pdu_header.ptype == RubySMB::Dcerpc::PTypes::RESPONSE
-          raise RubySMB::Dcerpc::Error::InvalidPacket, "Not a Response packet"
-        end
-        dcerpc_response
-      end
-
-      def ioctl_send_recv(action, options={})
-        request = set_header_fields(RubySMB::SMB2::Packet::IoctlRequest.new(options))
-        request.ctl_code = 0x0011C017
-        request.flags.is_fsctl = 0x00000001
-        request.buffer = action.to_binary_s
-        ioctl_raw_response = @tree.client.send_recv(request)
-        ioctl_response = RubySMB::SMB2::Packet::IoctlResponse.read(ioctl_raw_response)
-        unless ioctl_response.smb2_header.command == RubySMB::SMB2::Commands::IOCTL
-          raise RubySMB::Error::InvalidPacket, 'Not a IoctlResponse packet'
-        end
-        unless ioctl_response.status_code == WindowsError::NTStatus::STATUS_SUCCESS
-          raise RubySMB::Error::UnexpectedStatusCode, ioctl_response.status_code.name
-        end
-        ioctl_response
-      end
-
-    end
-  end
-end
-

data/spec/lib/ruby_smb/smb1/packet/error_packet_spec.rb

@@ -1,37 +0,0 @@
-require 'spec_helper'
-
-RSpec.describe RubySMB::SMB1::Packet::EmptyPacket do
-  subject(:packet) { described_class.new }
-
-  describe '#smb_header' do
-    subject(:header) { packet.smb_header }
-
-    it 'is a standard SMB Header' do
-      expect(header).to be_a RubySMB::SMB1::SMBHeader
-    end
-  end
-
-  describe '#parameter_block' do
-    subject(:parameter_block) { packet.parameter_block }
-
-    it 'is a standard ParameterBlock' do
-      expect(parameter_block).to be_a RubySMB::SMB1::ParameterBlock
-    end
-
-    it 'should be empty' do
-      expect(parameter_block.to_binary_s).to eq "\x00"
-    end
-  end
-
-  describe '#data_block' do
-    subject(:data_block) { packet.data_block }
-
-    it 'is a standard DataBlock' do
-      expect(data_block).to be_a RubySMB::SMB1::DataBlock
-    end
-
-    it 'should be empty' do
-      expect(data_block.to_binary_s).to eq "\x00\x00"
-    end
-  end
-end