Navigation

Changeset 3807

Timestamp:

09/23/06 08:53:02 (2 years ago)

Author:

mtredinnick

Message:

A corrected version of r3805.

Files:

r3806	r3807
101	101	Stuart Langridge <http://www.kryogenix.org/>
102	102	Eugene Lazutkin <http://lazutkin.com/blog/>
	103	Jeong-Min Lee
103	104	Christopher Lenz <http://www.cmlenz.net/>
104	105	limodou

r3806	r3807
5	5	from django import http
6	6	from pprint import pformat
	7	from shutil import copyfileobj
	8	try:
	9	from cStringIO import StringIO
	10	except ImportError:
	11	from StringIO import StringIO
7	12
8	13	# See http://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html
…	…
50	55	505: 'HTTP VERSION NOT SUPPORTED',
51	56	}
	57
	58	def safe_copyfileobj(fsrc, fdst, length=16*1024, size=0):
	59	"""
	60	A version of shutil.copyfileobj that will not read more than 'size' bytes.
	61	This makes it safe from clients sending more than CONTENT_LENGTH bytes of
	62	data in the body.
	63	"""
	64	if not size:
	65	return copyfileobj(fsrc, fdst, length)
	66	while size > 0:
	67	buf = fsrc.read(min(length, size))
	68	if not buf:
	69	break
	70	fdst.write(buf)
	71	size -= len(buf)
52	72
53	73	class WSGIRequest(http.HttpRequest):
…	…
120	140	return self._raw_post_data
121	141	except AttributeError:
122		self._raw_post_data = self.environ['wsgi.input'].read(int(self.environ["CONTENT_LENGTH"]))
	142	buf = StringIO()
	143	content_length = int(self.environ['CONTENT_LENGTH'])
	144	safe_copyfileobj(self.environ['wsgi.input'], buf, size=content_length)
	145	self._raw_post_data = buf.getvalue()
	146	buf.close()
123	147	return self._raw_post_data
124	148