ruby_smb 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ddd72f55b7cfbe6907e179398762e3ea353e4b4b
-  data.tar.gz: 565f474fa8140500252ac2f9cada13566d8f3566
+  metadata.gz: 2c631886979a7433c911311d4a86e396714d492c
+  data.tar.gz: 2239c4167668f032308e16d047b40e0f4c1587cb
 SHA512:
-  metadata.gz: 4a49da2c17d8eb6384a64932eed74f64365f45c5400c3291b289ec49de946852f276864ab5f5a3b8fc07bfd5a00c966ab076fd356f61f6b2af096eb871ed42a8
-  data.tar.gz: cbd0b9a55074976eaa380125d07592c06640b5139d46da8be55eff603a03304f07f8cb87b9b296d13eae88adabc9f80998491781c93fd7a6083c094aa767ccdf
+  metadata.gz: 1fcc1eacddf7c95375b966772d580357ba321d31c157d277e4c4306e448696f0f1f1aa210f06b127db7f2ffad83b5086f30608e8cbeb0ad370339dd3e1630731
+  data.tar.gz: 9cb22ee1ce93621365fba1f3bb67f0855a92331e28fc388b42d10e7b32716e587bfb29dda783d8c5bf617b388735f02abdc805a84b438aa1bdad70317f93d4f5

data/lib/ruby_smb/dispatcher/socket.rb

@@ -66,7 +66,9 @@ module RubySMB
         end
 
         begin
-          nbss_header = RubySMB::Nbss::SessionHeader.read(@tcp_socket)
+          nbss_data = @tcp_socket.read(4)
+          raise IOError if nbss_data.nil?
+          nbss_header = RubySMB::Nbss::SessionHeader.read(nbss_data)
         rescue IOError
           raise ::RubySMB::Error::NetBiosSessionService, 'NBSS Header is missing'
         end

data/lib/ruby_smb/version.rb

@@ -1,3 +1,3 @@
 module RubySMB
-  VERSION = '1.0.2'.freeze
+  VERSION = '1.0.3'.freeze
 end

data/spec/lib/ruby_smb/dispatcher/socket_spec.rb

@@ -85,7 +85,7 @@ RSpec.describe RubySMB::Dispatcher::Socket do
       end
 
       it 'reads the nbss header using Nbss::SessionHeader structure' do
-        expect(RubySMB::Nbss::SessionHeader).to receive(:read).with(response_socket).and_return(session_header)
+        expect(RubySMB::Nbss::SessionHeader).to receive(:read).with(nbss).and_return(session_header)
         smb_socket.recv_packet
       end
 
@@ -105,6 +105,7 @@ RSpec.describe RubySMB::Dispatcher::Socket do
         packet_length = 10
         session_header.packet_length = packet_length
         allow(RubySMB::Nbss::SessionHeader).to receive(:read).and_return(session_header)
+        expect(response_socket).to receive(:read).with(4).and_return(session_header)
         expect(response_socket).to receive(:read).with(packet_length).and_return('A' * packet_length)
         smb_socket.recv_packet
       end
@@ -116,6 +117,7 @@ RSpec.describe RubySMB::Dispatcher::Socket do
           session_header.packet_length = packet_length
           allow(RubySMB::Nbss::SessionHeader).to receive(:read).and_return(session_header)
 
+          expect(response_socket).to receive(:read).with(4).and_return(session_header)
           loop do
             expect(response_socket).to receive(:read).with(packet_length).and_return('A' * returned_length).once
             packet_length -= returned_length

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_smb
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - David Maloney
@@ -91,7 +91,7 @@ cert_chain:
   G+Hmcg1v810agasPdoydE0RTVZgEOOMoQ07qu7JFXVWZ9ZQpHT7qJATWL/b2csFG
   8mVuTXnyJOKRJA==
   -----END CERTIFICATE-----
-date: 2018-06-28 00:00:00.000000000 Z
+date: 2018-07-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redcarpet