Ticket #2613: wsgi.py-memoryerror.diff

core/handlers/wsgi.py

@@ -4 +4 @@
 from django.utils import datastructures
 from django import http
 from pprint import pformat
+try:
+    from cStringIO import StringIO
+except ImportError:
+    from StringIO import StringIO
 
+from shutil import copyfileobj
+def copyfileobj2(fsrc, fdst, length=16*1024, howmany=0):
+    """copy data from file-like object fsrc to file-like object fdst"""
+    if howmany == 0:
+        return copyfileobj(fsrc, fdst, length)
+    remain = howmany
+    while remain > 0:
+        buf = fsrc.read(min(length, remain))
+        if not buf:
+            break
+        fdst.write(buf)
+        remain -= len(buf)
+
 # See http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html
 STATUS_CODE_TEXT = {
     100: 'CONTINUE',
@@ -119 +136 @@
         try:
             return self._raw_post_data
         except AttributeError:
-            self._raw_post_data = self.environ['wsgi.input'].read(int(self.environ["CONTENT_LENGTH"]))
+            s = StringIO()
+            copyfileobj2(self.environ['wsgi.input'], s,
+                howmany=int(self.environ['CONTENT_LENGTH']))
+            self._raw_post_data = s.getvalue()
+            s.close()
             return self._raw_post_data
 
     GET = property(_get_get, _set_get)