firefox/src/rb/lib/selenium/webdriver/firefox/extension_connection.rb in selenium-webdriver-0.0.6 vs firefox/src/rb/lib/selenium/webdriver/firefox/extension_connection.rb in selenium-webdriver-0.0.7

@@ -64,11 +64,19 @@
             if received
               resp += received
             end
           end
 
-          length      = resp.split(":")[1].lstrip!.to_i
-          json_string = @socket.recv length
+          length         = Integer(resp.split(":").last.strip)
+          json_string    = ''
+          bytes_received = 0
+
+          until bytes_received == length
+            read_string = @socket.recv(length - bytes_received)
+
+            bytes_received += read_string.length
+            json_string << read_string
+          end
 
           if json_string.empty?
             raise Error::WebDriverError, "empty response from extension"
           end