ruby_smb 1.0.3 → 2.0.1

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