ruby_smb 0.0.20 → 0.0.21

data/spec/lib/ruby_smb/smb1/packet/trans/response_spec.rb

@@ -0,0 +1,85 @@
+require 'spec_helper'
+
+RSpec.describe RubySMB::SMB1::Packet::Trans::Response 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
+
+    it 'should have the command set to SMB_COM_NEGOTIATE' do
+      expect(header.command).to eq RubySMB::SMB1::Commands::SMB_COM_TRANSACTION
+    end
+
+    it 'should have the response flag set' do
+      expect(header.flags.reply).to eq 1
+    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 { is_expected.to respond_to :total_parameter_count }
+    it { is_expected.to respond_to :total_data_count }
+    it { is_expected.to respond_to :parameter_count }
+    it { is_expected.to respond_to :parameter_offset }
+    it { is_expected.to respond_to :parameter_displacement }
+    it { is_expected.to respond_to :data_count }
+    it { is_expected.to respond_to :data_offset }
+    it { is_expected.to respond_to :parameter_displacement }
+    it { is_expected.to respond_to :setup_count }
+
+    describe 'parameter_count' do
+      it 'is a count of bytes in the data_block trans_parameters field' do
+        packet.data_block.trans_parameters = "\x00\x01\x02\x03"
+        expect(parameter_block.parameter_count).to eq 4
+      end
+    end
+
+    describe 'parameter_offset' do
+      it ' contains the absolute_offset to the data_block trans_parameters field' do
+        expect(parameter_block.parameter_offset).to eq packet.data_block.trans_parameters.abs_offset
+      end
+    end
+
+    describe 'data_count' do
+      it 'is a count of bytes in the data_block trans_data field' do
+        packet.data_block.trans_data = "\x00\x01\x02\x03"
+        expect(parameter_block.data_count).to eq 4
+      end
+    end
+
+    describe 'data_offset' do
+      it 'contains the absolute_offset to the data_block trans_data field' do
+        expect(parameter_block.data_offset).to eq packet.data_block.trans_data.abs_offset
+      end
+    end
+  end
+
+  describe '#data_block' do
+    subject(:data_block) { packet.data_block }
+
+    it 'is a Trans DataBlock' do
+      expect(data_block).to be_a RubySMB::SMB1::Packet::Trans::DataBlock
+    end
+
+    it { is_expected.to respond_to :trans_parameters }
+    it { is_expected.to respond_to :trans_data }
+
+    it 'should keep #trans_parameters 4-byte aligned' do
+      expect(data_block.trans_parameters.abs_offset % 4).to eq 0
+    end
+
+    it 'should keep #trans_data 4-byte aligned' do
+      data_block.trans_parameters = 'a'
+      expect(data_block.trans_data.abs_offset % 4).to eq 0
+    end
+  end
+end

data/spec/lib/ruby_smb/smb1/packet/trans2/open2_response_spec.rb

@@ -10,7 +10,7 @@ RSpec.describe RubySMB::SMB1::Packet::Trans2::Open2Response do
       expect(header).to be_a RubySMB::SMB1::SMBHeader
     end
 
-    it 'should have the command set to SMB_COM_NEGOTIATE' do
+    it 'should have the command set to SMB_COM_TRANSACTION2' do
       expect(header.command).to eq RubySMB::SMB1::Commands::SMB_COM_TRANSACTION2
     end
 

data/spec/lib/ruby_smb/smb1/pipe_spec.rb

@@ -0,0 +1,65 @@
+RSpec.describe RubySMB::SMB1::Pipe do
+
+  let(:peek_nmpipe_response) {
+    packet = RubySMB::SMB1::Packet::Trans::PeekNmpipeResponse.new
+    packet.data_block.trans_parameters.read("\x10\x20\x00\x00\x03\x00")
+    packet
+  }
+
+  describe RubySMB::SMB1::Pipe do
+    it { expect(described_class).to be < RubySMB::SMB1::File }
+  end
+
+  let(:dispatcher) { RubySMB::Dispatcher::Socket.new(double('socket')) }
+  let(:client) { RubySMB::Client.new(dispatcher, username: 'msfadmin', password: 'msfadmin') }
+  let(:connect_response) {
+    packet = RubySMB::SMB1::Packet::TreeConnectResponse.new
+    packet.smb_header.tid = 2051
+    packet.parameter_block.guest_access_rights.read("\xff\x01\x1f\x00")
+    packet.parameter_block.access_rights.read("\xff\x01\x1f\x01")
+    packet
+  }
+  let(:tree) { RubySMB::SMB1::Tree.new(client: client, share: '\\1.2.3.4\IPC$', response: connect_response) }
+  let(:nt_create_andx_response) {
+    response = RubySMB::SMB1::Packet::NtCreateAndxResponse.new
+    response.parameter_block.ext_file_attributes = { normal: 1 }
+    response.parameter_block.fid = 0x4000
+    response.parameter_block.last_access_time = DateTime.parse("2017-09-20T1:1:1")
+    response.parameter_block.last_change_time = DateTime.parse("2017-09-22T2:2:2")
+    response.parameter_block.last_write_time  = DateTime.parse("2017-09-25T3:3:3")
+    response.parameter_block.end_of_file = 53
+    response.parameter_block.allocation_size = 4096
+    response
+  }
+  let(:filename) { 'msf-pipe' }
+
+  subject(:pipe) {
+    described_class.new(tree: tree, response: nt_create_andx_response, name: filename)
+  }
+
+  describe '#peek_available' do
+    it 'reads the correct number of bytes available' do
+      allow(pipe).to receive(:peek) { peek_nmpipe_response }
+      allow(pipe).to receive(:peek_available) { pipe.peek.data_block.trans_parameters.read_data_available }
+      expect(pipe.peek_available).to eq(0x2010)
+    end
+  end
+
+  describe '#peek_state' do
+    it 'reads the correct state of the pipe' do
+      allow(pipe).to receive(:peek) { peek_nmpipe_response }
+      allow(pipe).to receive(:peek_state) { pipe.peek.data_block.trans_parameters.pipe_state }
+      expect(pipe.peek_state).to eq(RubySMB::SMB1::Pipe::STATUS_OK)
+    end
+  end
+
+  describe '#is_connected?' do
+    it 'identifies that the pipe is connected from the status' do
+      allow(pipe).to receive(:peek) { peek_nmpipe_response }
+      allow(pipe).to receive(:peek_state) { pipe.peek.data_block.trans_parameters.pipe_state }
+      allow(pipe).to receive(:is_connected?) { pipe.peek_state == RubySMB::SMB1::Pipe::STATUS_OK }
+      expect(pipe.is_connected?).to eq(true)
+    end
+  end
+
+end

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

@@ -0,0 +1,64 @@
+RSpec.describe RubySMB::SMB2::Pipe do
+
+  let(:sock) { double('Socket', peeraddr: '192.168.1.5') }
+  let(:dispatcher) { RubySMB::Dispatcher::Socket.new(sock) }
+
+  let(:client) { RubySMB::Client.new(dispatcher, username: 'msfadmin', password: 'msfadmin') }
+  let(:tree_id) { 2049 }
+  let(:path) { '\\192.168.1.1\IPC$' }
+  let(:connect_response) {
+    packet = RubySMB::SMB2::Packet::TreeConnectResponse.new
+    packet.smb2_header.tree_id = tree_id
+    packet.maximal_access.read("\xff\x01\x1f\x00")
+    packet.share_type = 0x01
+    packet
+  }
+
+  let(:tree) { RubySMB::SMB2::Tree.new(client: client, share: path, response: connect_response) }
+  let(:file_id) { RubySMB::Field::Smb2Fileid.read('\x6d\x01\x00\x00\x00\x00\x00\x00\x\x01\x00\x00\x00\xff\xff\xff\xff') }
+  let(:time) { DateTime.now }
+  let(:create_response) {
+    RubySMB::SMB2::Packet::CreateResponse.new(
+      file_id: file_id,
+      end_of_file: 108,
+      allocation_size: 112,
+      last_access: time,
+      last_change: time,
+      last_write: time
+    )
+  }
+
+  let(:ioctl_response) {
+    packet = RubySMB::SMB2::Packet::IoctlResponse.new
+    packet.buffer = "\x03\x00\x00\x00" + "\x10\x20\x30\x40" + "\x00\x00\x00\x00" + "\x00\x00\x00\x00"
+    packet
+  }
+
+  subject(:pipe) { described_class.new(name: 'msf-pipe', response: create_response, tree: tree) }
+
+  describe '#peek_available' do
+    it 'reads the correct number of bytes available' do
+      allow(pipe).to receive(:peek) { ioctl_response }
+      allow(pipe).to receive(:peek_available) { pipe.peek.buffer.unpack('VV')[1] }
+      expect(pipe.peek_available).to eq(0x40302010)
+    end
+  end
+  
+  describe '#peek_state' do
+    it 'reads the correct state of the pipe' do
+      allow(pipe).to receive(:peek) { ioctl_response }
+      allow(pipe).to receive(:peek_state)  { pipe.peek.buffer.unpack('V')[0] }
+      expect(pipe.peek_state).to eq(RubySMB::SMB2::Pipe::STATUS_CONNECTED)
+    end
+  end
+
+  describe '#is_connected?' do
+    it 'identifies that the pipe is connected from the status' do
+      allow(pipe).to receive(:peek) { ioctl_response }
+      allow(pipe).to receive(:peek_state)  { pipe.peek.buffer.unpack('V')[0] }
+      allow(pipe).to receive(:is_connected?) { pipe.peek_state == RubySMB::SMB2::Pipe::STATUS_CONNECTED }
+      expect(pipe.is_connected?).to eq(true)
+    end
+  end
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_smb
 version: !ruby/object:Gem::Version
-  version: 0.0.20
+  version: 0.0.21
 platform: ruby
 authors:
 - David Maloney
@@ -89,7 +89,7 @@ cert_chain:
   G+Hmcg1v810agasPdoydE0RTVZgEOOMoQ07qu7JFXVWZ9ZQpHT7qJATWL/b2csFG
   8mVuTXnyJOKRJA==
   -----END CERTIFICATE-----
-date: 2018-02-22 00:00:00.000000000 Z
+date: 2018-03-05 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet
@@ -229,6 +229,7 @@ files:
 - examples/negotiate.rb
 - examples/negotiate_with_netbios_service.rb
 - examples/net_share_enum_all.rb
+- examples/pipes.rb
 - examples/read_file.rb
 - examples/rename_file.rb
 - examples/tree_connect.rb
@@ -310,6 +311,7 @@ files:
 - lib/ruby_smb/smb1/bit_field/smb_file_attributes.rb
 - lib/ruby_smb/smb1/bit_field/smb_nmpipe_status.rb
 - lib/ruby_smb/smb1/bit_field/trans2_flags.rb
+- lib/ruby_smb/smb1/bit_field/trans_flags.rb
 - lib/ruby_smb/smb1/bit_field/tree_connect_flags.rb
 - lib/ruby_smb/smb1/commands.rb
 - lib/ruby_smb/smb1/create_actions.rb
@@ -342,6 +344,13 @@ files:
 - lib/ruby_smb/smb1/packet/session_setup_legacy_response.rb
 - lib/ruby_smb/smb1/packet/session_setup_request.rb
 - lib/ruby_smb/smb1/packet/session_setup_response.rb
+- lib/ruby_smb/smb1/packet/trans.rb
+- lib/ruby_smb/smb1/packet/trans/data_block.rb
+- lib/ruby_smb/smb1/packet/trans/peek_nmpipe_request.rb
+- lib/ruby_smb/smb1/packet/trans/peek_nmpipe_response.rb
+- lib/ruby_smb/smb1/packet/trans/request.rb
+- lib/ruby_smb/smb1/packet/trans/response.rb
+- lib/ruby_smb/smb1/packet/trans/subcommands.rb
 - lib/ruby_smb/smb1/packet/trans2.rb
 - lib/ruby_smb/smb1/packet/trans2/data_block.rb
 - lib/ruby_smb/smb1/packet/trans2/find_first2_request.rb
@@ -365,6 +374,7 @@ files:
 - lib/ruby_smb/smb1/packet/write_andx_request.rb
 - lib/ruby_smb/smb1/packet/write_andx_response.rb
 - lib/ruby_smb/smb1/parameter_block.rb
+- lib/ruby_smb/smb1/pipe.rb
 - lib/ruby_smb/smb1/resource_type.rb
 - lib/ruby_smb/smb1/smb_header.rb
 - lib/ruby_smb/smb1/tree.rb
@@ -411,6 +421,7 @@ files:
 - lib/ruby_smb/smb2/packet/tree_disconnect_response.rb
 - lib/ruby_smb/smb2/packet/write_request.rb
 - lib/ruby_smb/smb2/packet/write_response.rb
+- lib/ruby_smb/smb2/pipe.rb
 - lib/ruby_smb/smb2/smb2_header.rb
 - lib/ruby_smb/smb2/tree.rb
 - lib/ruby_smb/version.rb
@@ -469,6 +480,7 @@ files:
 - spec/lib/ruby_smb/smb1/bit_field/smb_file_attributes_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/smb_nmpipe_status_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/trans2_flags_spec.rb
+- spec/lib/ruby_smb/smb1/bit_field/trans_flags_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/tree_connect_flags_spec.rb
 - spec/lib/ruby_smb/smb1/data_block_spec.rb
 - spec/lib/ruby_smb/smb1/dialect_spec.rb
@@ -495,6 +507,10 @@ files:
 - spec/lib/ruby_smb/smb1/packet/session_setup_legacy_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/session_setup_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/session_setup_response_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/peek_nmpipe_request_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/peek_nmpipe_response_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/request_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_first2_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_first2_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_information_level/find_file_full_directory_info_spec.rb
@@ -514,6 +530,7 @@ files:
 - spec/lib/ruby_smb/smb1/packet/write_andx_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/write_andx_response_spec.rb
 - spec/lib/ruby_smb/smb1/parameter_block_spec.rb
+- spec/lib/ruby_smb/smb1/pipe_spec.rb
 - spec/lib/ruby_smb/smb1/smb_header_spec.rb
 - spec/lib/ruby_smb/smb1/tree_spec.rb
 - spec/lib/ruby_smb/smb2/bit_field/directory_access_mask_spec.rb
@@ -551,6 +568,7 @@ files:
 - spec/lib/ruby_smb/smb2/packet/tree_disconnect_response_spec.rb
 - spec/lib/ruby_smb/smb2/packet/write_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/write_response_spec.rb
+- spec/lib/ruby_smb/smb2/pipe_spec.rb
 - spec/lib/ruby_smb/smb2/smb2_header_spec.rb
 - spec/lib/ruby_smb/smb2/tree_spec.rb
 - spec/lib/ruby_smb_spec.rb
@@ -636,6 +654,7 @@ test_files:
 - spec/lib/ruby_smb/smb1/bit_field/smb_file_attributes_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/smb_nmpipe_status_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/trans2_flags_spec.rb
+- spec/lib/ruby_smb/smb1/bit_field/trans_flags_spec.rb
 - spec/lib/ruby_smb/smb1/bit_field/tree_connect_flags_spec.rb
 - spec/lib/ruby_smb/smb1/data_block_spec.rb
 - spec/lib/ruby_smb/smb1/dialect_spec.rb
@@ -662,6 +681,10 @@ test_files:
 - spec/lib/ruby_smb/smb1/packet/session_setup_legacy_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/session_setup_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/session_setup_response_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/peek_nmpipe_request_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/peek_nmpipe_response_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/request_spec.rb
+- spec/lib/ruby_smb/smb1/packet/trans/response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_first2_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_first2_response_spec.rb
 - spec/lib/ruby_smb/smb1/packet/trans2/find_information_level/find_file_full_directory_info_spec.rb
@@ -681,6 +704,7 @@ test_files:
 - spec/lib/ruby_smb/smb1/packet/write_andx_request_spec.rb
 - spec/lib/ruby_smb/smb1/packet/write_andx_response_spec.rb
 - spec/lib/ruby_smb/smb1/parameter_block_spec.rb
+- spec/lib/ruby_smb/smb1/pipe_spec.rb
 - spec/lib/ruby_smb/smb1/smb_header_spec.rb
 - spec/lib/ruby_smb/smb1/tree_spec.rb
 - spec/lib/ruby_smb/smb2/bit_field/directory_access_mask_spec.rb
@@ -718,6 +742,7 @@ test_files:
 - spec/lib/ruby_smb/smb2/packet/tree_disconnect_response_spec.rb
 - spec/lib/ruby_smb/smb2/packet/write_request_spec.rb
 - spec/lib/ruby_smb/smb2/packet/write_response_spec.rb
+- spec/lib/ruby_smb/smb2/pipe_spec.rb
 - spec/lib/ruby_smb/smb2/smb2_header_spec.rb
 - spec/lib/ruby_smb/smb2/tree_spec.rb
 - spec/lib/ruby_smb_spec.rb