lib/ruby_skynet/common.rb in ruby_skynet-0.4.0.pre vs lib/ruby_skynet/common.rb in ruby_skynet-0.4.0.pre2

@@ -7,15 +7,19 @@
     def self.read_bson_document(socket)
       bytebuf = BSON::ByteBuffer.new
       # Read 4 byte size of following BSON document
       bytes = socket.read(4)
 
+      # No more data
+      return unless bytes
+
       # Read BSON document
       sz = bytes.unpack("V")[0]
       raise "Invalid Data received from server:#{bytes.inspect}" unless sz
 
       bytebuf.append!(bytes)
       bytebuf.append!(socket.read(sz - 4))
+      raise "Celluloid is not returning #{sz} requested bytes. #{bytebuf.length} bytes returned" unless sz == bytebuf.length
       return BSON.deserialize(bytebuf)
     end
 
   end
 end
\ No newline at end of file