diff options
Diffstat (limited to 'sleekxmpp')
| -rw-r--r-- | sleekxmpp/test/livesocket.py | 2 | ||||
| -rw-r--r-- | sleekxmpp/test/mocksocket.py | 1 | ||||
| -rw-r--r-- | sleekxmpp/xmlstream/xmlstream.py | 36 |
3 files changed, 26 insertions, 13 deletions
diff --git a/sleekxmpp/test/livesocket.py b/sleekxmpp/test/livesocket.py index 7dd4693c..80d63307 100644 --- a/sleekxmpp/test/livesocket.py +++ b/sleekxmpp/test/livesocket.py @@ -138,7 +138,7 @@ class TestLiveSocket(object): """ with self.send_queue_lock: self.send_queue.put(data) - self.socket.send(data) + return self.socket.send(data) # ------------------------------------------------------------------ # File Socket diff --git a/sleekxmpp/test/mocksocket.py b/sleekxmpp/test/mocksocket.py index a2af8d6b..0920b7ea 100644 --- a/sleekxmpp/test/mocksocket.py +++ b/sleekxmpp/test/mocksocket.py @@ -121,6 +121,7 @@ class TestSocket(object): if self.disconnected: raise socket.error self.send_queue.put(data) + return len(data) # ------------------------------------------------------------------ # File Socket diff --git a/sleekxmpp/xmlstream/xmlstream.py b/sleekxmpp/xmlstream/xmlstream.py index 3ce278f0..c886c3bd 100644 --- a/sleekxmpp/xmlstream/xmlstream.py +++ b/sleekxmpp/xmlstream/xmlstream.py @@ -915,7 +915,15 @@ class XMLStream(object): if now: log.debug("SEND (IMMED): %s" % data) try: - self.socket.send(data.encode('utf-8')) + data = data.encode('utf-8') + total = len(data) + sent = 0 + count = 0 + while sent < total and not self.stop.is_set(): + sent += self.socket.send(data[sent:]) + count += 1 + if count > 1: + log.debug('SENT: %d chunks' % count) except Socket.error as serr: self.event('socket_error', serr) log.warning("Failed to send %s" % data) @@ -1222,7 +1230,7 @@ class XMLStream(object): Extract stanzas from the send queue and send them on the stream. """ try: - while not self.stop.isSet(): + while not self.stop.is_set(): self.session_started_event.wait() if self.__failed_send_stanza is not None: data = self.__failed_send_stanza @@ -1234,22 +1242,26 @@ class XMLStream(object): continue log.debug("SEND: %s" % data) try: - self.socket.send(data.encode('utf-8')) + enc_data = data.encode('utf-8') + total = len(enc_data) + sent = 0 + count = 0 + while sent < total and not self.stop.is_set(): + sent += self.socket.send(enc_data[sent:]) + count += 1 + if count > 1: + log.debug('SENT: %d chunks' % count) self.send_queue.task_done() except Socket.error as serr: self.event('socket_error', serr) log.warning("Failed to send %s" % data) self.__failed_send_stanza = data self.disconnect(self.auto_reconnect) - except KeyboardInterrupt: - log.debug("Keyboard Escape Detected in _send_thread") - self.event('killed', direct=True) - self.disconnect() - return - except SystemExit: - self.disconnect() - self.event_queue.put(('quit', None, None)) - return + except Exception as ex: + log.exception('Unexpected error in send thread: %s' % ex) + self.exception(ex) + if not self.stop.is_set(): + self.disconnect(self.auto_reconnect) def exception(self, exception): """ |