1 files changed, 54 insertions, 0 deletions
diff --git a/slixmpp/xmlstream/filesocket.py b/slixmpp/xmlstream/filesocket.py
new file mode 100644
index 00000000..1b52251e
--- /dev/null
+++ b/slixmpp/xmlstream/filesocket.py
@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+"""
+    slixmpp.xmlstream.filesocket
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    This module is a shim for correcting deficiencies in the file
+    socket implementation of Python2.6.
+
+    Part of Slixmpp: The Slick XMPP Library
+
+    :copyright: (c) 2011 Nathanael C. Fritz
+    :license: MIT, see LICENSE for more details
+"""
+
+from socket import _fileobject
+import errno
+import socket
+
+
+class FileSocket(_fileobject):
+
+    """Create a file object wrapper for a socket to work around
+    issues present in Python 2.6 when using sockets as file objects.
+
+    The parser for :class:`~xml.etree.cElementTree` requires a file, but
+    we will be reading from the XMPP connection socket instead.
+    """
+
+    def read(self, size=4096):
+        """Read data from the socket as if it were a file."""
+        if self._sock is None:
+            return None
+        while True:
+            try:
+                data = self._sock.recv(size)
+                break
+            except socket.error as serr:
+                if serr.errno != errno.EINTR:
+                    raise
+        if data is not None:
+            return data
+
+
+class Socket26(socket.socket):
+
+    """A custom socket implementation that uses our own FileSocket class
+    to work around issues in Python 2.6 when using sockets as files.
+    """
+
+    def makefile(self, mode='r', bufsize=-1):
+        """makefile([mode[, bufsize]]) -> file object
+        Return a regular file object corresponding to the socket.  The mode
+        and bufsize arguments are as for the built-in open() function."""
+        return FileSocket(self._sock, mode, bufsize)