blob: 56c33413ee0780c0017802a8b324b3196dd0ca94 [file] [log] [blame]
Éric Araujo06efea32012-02-26 01:33:22 +01001r"""HTTP/1.1 client library
Guido van Rossum41999c11997-12-09 00:12:23 +00002
Greg Stein5e0fa402000-06-26 08:28:01 +00003<intro stuff goes here>
4<other stuff, too>
Guido van Rossum41999c11997-12-09 00:12:23 +00005
Andrew M. Kuchlingca2e7902006-07-30 00:27:34 +00006HTTPConnection goes through a number of "states", which define when a client
Greg Stein5e0fa402000-06-26 08:28:01 +00007may legally make another request or fetch the response for a particular
8request. This diagram details these state transitions:
Guido van Rossum41999c11997-12-09 00:12:23 +00009
Greg Stein5e0fa402000-06-26 08:28:01 +000010 (null)
11 |
12 | HTTPConnection()
13 v
14 Idle
15 |
16 | putrequest()
17 v
18 Request-started
19 |
20 | ( putheader() )* endheaders()
21 v
22 Request-sent
23 |
24 | response = getresponse()
25 v
26 Unread-response [Response-headers-read]
27 |\____________________
Tim Peters5ceadc82001-01-13 19:16:21 +000028 | |
29 | response.read() | putrequest()
30 v v
31 Idle Req-started-unread-response
32 ______/|
33 / |
34 response.read() | | ( putheader() )* endheaders()
35 v v
36 Request-started Req-sent-unread-response
37 |
38 | response.read()
39 v
40 Request-sent
Greg Stein5e0fa402000-06-26 08:28:01 +000041
42This diagram presents the following rules:
43 -- a second request may not be started until {response-headers-read}
44 -- a response [object] cannot be retrieved until {request-sent}
45 -- there is no differentiation between an unread response body and a
46 partially read response body
47
48Note: this enforcement is applied by the HTTPConnection class. The
49 HTTPResponse class does not enforce this state machine, which
50 implies sophisticated clients may accelerate the request/response
51 pipeline. Caution should be taken, though: accelerating the states
52 beyond the above pattern may imply knowledge of the server's
53 connection-close behavior for certain requests. For example, it
54 is impossible to tell whether the server will close the connection
55 UNTIL the response headers have been read; this means that further
56 requests cannot be placed into the pipeline until it is known that
57 the server will NOT be closing the connection.
58
59Logical State __state __response
60------------- ------- ----------
61Idle _CS_IDLE None
62Request-started _CS_REQ_STARTED None
63Request-sent _CS_REQ_SENT None
64Unread-response _CS_IDLE <response_class>
65Req-started-unread-response _CS_REQ_STARTED <response_class>
66Req-sent-unread-response _CS_REQ_SENT <response_class>
Guido van Rossum41999c11997-12-09 00:12:23 +000067"""
Guido van Rossum23acc951994-02-21 16:36:04 +000068
Antoine Pitrou72481782009-09-29 17:48:18 +000069from array import array
Victor Stinner2c6aee92010-07-24 02:46:16 +000070import os
Jeremy Hylton6459c8d2001-10-11 17:47:22 +000071import socket
Brett Cannon1eaf0742008-09-02 01:25:16 +000072from sys import py3kwarning
Jeremy Hylton8acf1e02002-03-08 19:35:51 +000073from urlparse import urlsplit
Bill Janssenc4592642007-08-31 19:02:23 +000074import warnings
Brett Cannon1eaf0742008-09-02 01:25:16 +000075with warnings.catch_warnings():
76 if py3kwarning:
77 warnings.filterwarnings("ignore", ".*mimetools has been removed",
78 DeprecationWarning)
Brett Cannon0a4128e2008-08-16 21:56:03 +000079 import mimetools
Guido van Rossum23acc951994-02-21 16:36:04 +000080
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000081try:
Greg Steindd6eefb2000-07-18 09:09:48 +000082 from cStringIO import StringIO
Greg Stein5e0fa402000-06-26 08:28:01 +000083except ImportError:
Greg Steindd6eefb2000-07-18 09:09:48 +000084 from StringIO import StringIO
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000085
Thomas Woutersa6900e82007-08-30 21:54:39 +000086__all__ = ["HTTP", "HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000087 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000088 "UnknownTransferEncoding", "UnimplementedFileMode",
89 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
90 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
Georg Brandl6aab16e2006-02-17 19:17:25 +000091 "BadStatusLine", "error", "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000092
Guido van Rossum23acc951994-02-21 16:36:04 +000093HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000094HTTPS_PORT = 443
95
Greg Stein5e0fa402000-06-26 08:28:01 +000096_UNKNOWN = 'UNKNOWN'
97
98# connection states
99_CS_IDLE = 'Idle'
100_CS_REQ_STARTED = 'Request-started'
101_CS_REQ_SENT = 'Request-sent'
102
Martin v. Löwis39a31782004-09-18 09:03:49 +0000103# status codes
104# informational
105CONTINUE = 100
106SWITCHING_PROTOCOLS = 101
107PROCESSING = 102
108
109# successful
110OK = 200
111CREATED = 201
112ACCEPTED = 202
113NON_AUTHORITATIVE_INFORMATION = 203
114NO_CONTENT = 204
115RESET_CONTENT = 205
116PARTIAL_CONTENT = 206
117MULTI_STATUS = 207
118IM_USED = 226
119
120# redirection
121MULTIPLE_CHOICES = 300
122MOVED_PERMANENTLY = 301
123FOUND = 302
124SEE_OTHER = 303
125NOT_MODIFIED = 304
126USE_PROXY = 305
127TEMPORARY_REDIRECT = 307
128
129# client error
130BAD_REQUEST = 400
131UNAUTHORIZED = 401
132PAYMENT_REQUIRED = 402
133FORBIDDEN = 403
134NOT_FOUND = 404
135METHOD_NOT_ALLOWED = 405
136NOT_ACCEPTABLE = 406
137PROXY_AUTHENTICATION_REQUIRED = 407
138REQUEST_TIMEOUT = 408
139CONFLICT = 409
140GONE = 410
141LENGTH_REQUIRED = 411
142PRECONDITION_FAILED = 412
143REQUEST_ENTITY_TOO_LARGE = 413
144REQUEST_URI_TOO_LONG = 414
145UNSUPPORTED_MEDIA_TYPE = 415
146REQUESTED_RANGE_NOT_SATISFIABLE = 416
147EXPECTATION_FAILED = 417
148UNPROCESSABLE_ENTITY = 422
149LOCKED = 423
150FAILED_DEPENDENCY = 424
151UPGRADE_REQUIRED = 426
152
153# server error
154INTERNAL_SERVER_ERROR = 500
155NOT_IMPLEMENTED = 501
156BAD_GATEWAY = 502
157SERVICE_UNAVAILABLE = 503
158GATEWAY_TIMEOUT = 504
159HTTP_VERSION_NOT_SUPPORTED = 505
160INSUFFICIENT_STORAGE = 507
161NOT_EXTENDED = 510
162
Georg Brandl6aab16e2006-02-17 19:17:25 +0000163# Mapping status codes to official W3C names
164responses = {
165 100: 'Continue',
166 101: 'Switching Protocols',
167
168 200: 'OK',
169 201: 'Created',
170 202: 'Accepted',
171 203: 'Non-Authoritative Information',
172 204: 'No Content',
173 205: 'Reset Content',
174 206: 'Partial Content',
175
176 300: 'Multiple Choices',
177 301: 'Moved Permanently',
178 302: 'Found',
179 303: 'See Other',
180 304: 'Not Modified',
181 305: 'Use Proxy',
182 306: '(Unused)',
183 307: 'Temporary Redirect',
184
185 400: 'Bad Request',
186 401: 'Unauthorized',
187 402: 'Payment Required',
188 403: 'Forbidden',
189 404: 'Not Found',
190 405: 'Method Not Allowed',
191 406: 'Not Acceptable',
192 407: 'Proxy Authentication Required',
193 408: 'Request Timeout',
194 409: 'Conflict',
195 410: 'Gone',
196 411: 'Length Required',
197 412: 'Precondition Failed',
198 413: 'Request Entity Too Large',
199 414: 'Request-URI Too Long',
200 415: 'Unsupported Media Type',
201 416: 'Requested Range Not Satisfiable',
202 417: 'Expectation Failed',
203
204 500: 'Internal Server Error',
205 501: 'Not Implemented',
206 502: 'Bad Gateway',
207 503: 'Service Unavailable',
208 504: 'Gateway Timeout',
209 505: 'HTTP Version Not Supported',
210}
211
Georg Brandl80ba8e82005-09-29 20:16:07 +0000212# maximal amount of data to read at one time in _safe_read
213MAXAMOUNT = 1048576
214
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000215# maximal line length when calling readline().
216_MAXLINE = 65536
217
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000218class HTTPMessage(mimetools.Message):
219
220 def addheader(self, key, value):
221 """Add header for field key handling repeats."""
222 prev = self.dict.get(key)
223 if prev is None:
224 self.dict[key] = value
225 else:
226 combined = ", ".join((prev, value))
227 self.dict[key] = combined
228
229 def addcontinue(self, key, more):
230 """Add more field data from a continuation line."""
231 prev = self.dict[key]
232 self.dict[key] = prev + "\n " + more
233
234 def readheaders(self):
235 """Read header lines.
236
237 Read header lines up to the entirely blank line that terminates them.
238 The (normally blank) line that ends the headers is skipped, but not
239 included in the returned list. If a non-header line ends the headers,
240 (which is an error), an attempt is made to backspace over it; it is
241 never included in the returned list.
242
243 The variable self.status is set to the empty string if all went well,
244 otherwise it is an error message. The variable self.headers is a
245 completely uninterpreted list of lines contained in the header (so
246 printing them will reproduce the header exactly as it appears in the
247 file).
248
249 If multiple header fields with the same name occur, they are combined
250 according to the rules in RFC 2616 sec 4.2:
251
252 Appending each subsequent field-value to the first, each separated
253 by a comma. The order in which header fields with the same field-name
254 are received is significant to the interpretation of the combined
255 field value.
256 """
257 # XXX The implementation overrides the readheaders() method of
258 # rfc822.Message. The base class design isn't amenable to
259 # customized behavior here so the method here is a copy of the
260 # base class code with a few small changes.
261
262 self.dict = {}
263 self.unixfrom = ''
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000264 self.headers = hlist = []
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000265 self.status = ''
266 headerseen = ""
267 firstline = 1
268 startofline = unread = tell = None
269 if hasattr(self.fp, 'unread'):
270 unread = self.fp.unread
271 elif self.seekable:
272 tell = self.fp.tell
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000273 while True:
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000274 if tell:
275 try:
276 startofline = tell()
277 except IOError:
278 startofline = tell = None
279 self.seekable = 0
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000280 line = self.fp.readline(_MAXLINE + 1)
281 if len(line) > _MAXLINE:
282 raise LineTooLong("header line")
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000283 if not line:
284 self.status = 'EOF in headers'
285 break
286 # Skip unix From name time lines
287 if firstline and line.startswith('From '):
288 self.unixfrom = self.unixfrom + line
289 continue
290 firstline = 0
291 if headerseen and line[0] in ' \t':
292 # XXX Not sure if continuation lines are handled properly
293 # for http and/or for repeating headers
294 # It's a continuation line.
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000295 hlist.append(line)
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000296 self.addcontinue(headerseen, line.strip())
297 continue
298 elif self.iscomment(line):
299 # It's a comment. Ignore it.
300 continue
301 elif self.islast(line):
302 # Note! No pushback here! The delimiter line gets eaten.
303 break
304 headerseen = self.isheader(line)
305 if headerseen:
306 # It's a legal header line, save it.
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000307 hlist.append(line)
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000308 self.addheader(headerseen, line[len(headerseen)+1:].strip())
309 continue
310 else:
311 # It's not a header line; throw it back and stop here.
312 if not self.dict:
313 self.status = 'No headers'
314 else:
315 self.status = 'Non-header line where header expected'
316 # Try to undo the read.
317 if unread:
318 unread(line)
319 elif tell:
320 self.fp.seek(startofline)
321 else:
322 self.status = self.status + '; bad seek'
323 break
Greg Stein5e0fa402000-06-26 08:28:01 +0000324
325class HTTPResponse:
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000326
327 # strict: If true, raise BadStatusLine if the status line can't be
328 # parsed as a valid HTTP/1.0 or 1.1 status line. By default it is
Skip Montanaro186bec22002-07-25 16:10:38 +0000329 # false because it prevents clients from talking to HTTP/0.9
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000330 # servers. Note that a response with a sufficiently corrupted
331 # status line will look like an HTTP/0.9 response.
332
333 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
334
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +0000335 def __init__(self, sock, debuglevel=0, strict=0, method=None, buffering=False):
336 if buffering:
337 # The caller won't be using any sock.recv() calls, so buffering
Gregory P. Smith77db0162009-08-20 09:38:43 +0000338 # is fine and recommended for performance.
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +0000339 self.fp = sock.makefile('rb')
340 else:
341 # The buffer size is specified as zero, because the headers of
342 # the response are read with readline(). If the reads were
343 # buffered the readline() calls could consume some of the
344 # response, which make be read via a recv() on the underlying
345 # socket.
346 self.fp = sock.makefile('rb', 0)
Jeremy Hylton30f86742000-09-18 22:50:38 +0000347 self.debuglevel = debuglevel
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000348 self.strict = strict
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000349 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000350
Greg Steindd6eefb2000-07-18 09:09:48 +0000351 self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000352
Greg Steindd6eefb2000-07-18 09:09:48 +0000353 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000354 self.version = _UNKNOWN # HTTP-Version
355 self.status = _UNKNOWN # Status-Code
356 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000357
Tim Peters07e99cb2001-01-14 23:47:14 +0000358 self.chunked = _UNKNOWN # is "chunked" being used?
359 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
360 self.length = _UNKNOWN # number of bytes left in response
361 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000362
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000363 def _read_status(self):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000364 # Initialize with Simple-Response defaults
Christian Heimes671138f2012-09-25 13:29:30 +0200365 line = self.fp.readline(_MAXLINE + 1)
366 if len(line) > _MAXLINE:
367 raise LineTooLong("header line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000368 if self.debuglevel > 0:
369 print "reply:", repr(line)
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000370 if not line:
371 # Presumably, the server closed the connection before
372 # sending a valid response.
373 raise BadStatusLine(line)
Greg Steindd6eefb2000-07-18 09:09:48 +0000374 try:
Guido van Rossum34735a62000-12-15 15:09:42 +0000375 [version, status, reason] = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000376 except ValueError:
377 try:
Guido van Rossum34735a62000-12-15 15:09:42 +0000378 [version, status] = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000379 reason = ""
380 except ValueError:
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000381 # empty version will cause next test to fail and status
382 # will be treated as 0.9 response.
383 version = ""
384 if not version.startswith('HTTP/'):
385 if self.strict:
386 self.close()
387 raise BadStatusLine(line)
388 else:
389 # assume it's a Simple-Response from an 0.9 server
390 self.fp = LineAndFileWrapper(line, self.fp)
391 return "HTTP/0.9", 200, ""
Greg Stein5e0fa402000-06-26 08:28:01 +0000392
Jeremy Hylton23d40472001-04-13 14:57:08 +0000393 # The status code is a three-digit number
394 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000395 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000396 if status < 100 or status > 999:
397 raise BadStatusLine(line)
398 except ValueError:
399 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000400 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000401
Jeremy Hylton39c03802002-07-12 14:04:09 +0000402 def begin(self):
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000403 if self.msg is not None:
404 # we've already started reading the response
405 return
406
407 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000408 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000409 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000410 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000411 break
412 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000413 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000414 skip = self.fp.readline(_MAXLINE + 1)
415 if len(skip) > _MAXLINE:
416 raise LineTooLong("header line")
417 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000418 if not skip:
419 break
420 if self.debuglevel > 0:
421 print "header:", skip
Tim Petersc411dba2002-07-16 21:35:23 +0000422
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000423 self.status = status
424 self.reason = reason.strip()
Greg Steindd6eefb2000-07-18 09:09:48 +0000425 if version == 'HTTP/1.0':
426 self.version = 10
Jeremy Hylton110941a2000-10-12 19:58:36 +0000427 elif version.startswith('HTTP/1.'):
Tim Peters07e99cb2001-01-14 23:47:14 +0000428 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Jeremy Hylton110941a2000-10-12 19:58:36 +0000429 elif version == 'HTTP/0.9':
430 self.version = 9
Greg Steindd6eefb2000-07-18 09:09:48 +0000431 else:
432 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000433
Jeremy Hylton110941a2000-10-12 19:58:36 +0000434 if self.version == 9:
Georg Brandl0aade9a2005-06-26 22:06:54 +0000435 self.length = None
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000436 self.chunked = 0
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000437 self.will_close = 1
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000438 self.msg = HTTPMessage(StringIO())
Jeremy Hylton110941a2000-10-12 19:58:36 +0000439 return
440
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000441 self.msg = HTTPMessage(self.fp, 0)
Jeremy Hylton30f86742000-09-18 22:50:38 +0000442 if self.debuglevel > 0:
443 for hdr in self.msg.headers:
444 print "header:", hdr,
Greg Stein5e0fa402000-06-26 08:28:01 +0000445
Greg Steindd6eefb2000-07-18 09:09:48 +0000446 # don't let the msg keep an fp
447 self.msg.fp = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000448
Greg Steindd6eefb2000-07-18 09:09:48 +0000449 # are we using the chunked-style of transfer encoding?
450 tr_enc = self.msg.getheader('transfer-encoding')
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000451 if tr_enc and tr_enc.lower() == "chunked":
Greg Steindd6eefb2000-07-18 09:09:48 +0000452 self.chunked = 1
453 self.chunk_left = None
454 else:
455 self.chunked = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000456
Greg Steindd6eefb2000-07-18 09:09:48 +0000457 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000458 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000459
Greg Steindd6eefb2000-07-18 09:09:48 +0000460 # do we have a Content-Length?
461 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
462 length = self.msg.getheader('content-length')
463 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000464 try:
465 self.length = int(length)
466 except ValueError:
467 self.length = None
Georg Brandl8c460d52008-02-24 00:14:24 +0000468 else:
469 if self.length < 0: # ignore nonsensical negative lengths
470 self.length = None
Greg Steindd6eefb2000-07-18 09:09:48 +0000471 else:
472 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000473
Greg Steindd6eefb2000-07-18 09:09:48 +0000474 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000475 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000476 100 <= status < 200 or # 1xx codes
477 self._method == 'HEAD'):
Greg Steindd6eefb2000-07-18 09:09:48 +0000478 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000479
Greg Steindd6eefb2000-07-18 09:09:48 +0000480 # if the connection remains open, and we aren't using chunked, and
481 # a content-length was not provided, then assume that the connection
482 # WILL close.
483 if not self.will_close and \
484 not self.chunked and \
485 self.length is None:
486 self.will_close = 1
Greg Stein5e0fa402000-06-26 08:28:01 +0000487
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000488 def _check_close(self):
Jeremy Hylton2c178252004-08-07 16:28:14 +0000489 conn = self.msg.getheader('connection')
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000490 if self.version == 11:
491 # An HTTP/1.1 proxy is assumed to stay open unless
492 # explicitly closed.
493 conn = self.msg.getheader('connection')
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000494 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000495 return True
496 return False
497
Jeremy Hylton2c178252004-08-07 16:28:14 +0000498 # Some HTTP/1.0 implementations have support for persistent
499 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000500
Georg Brandl26458c12007-12-08 10:56:39 +0000501 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000502 if self.msg.getheader('keep-alive'):
503 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000504
Jeremy Hylton2c178252004-08-07 16:28:14 +0000505 # At least Akamai returns a "Connection: Keep-Alive" header,
506 # which was supposed to be sent by the client.
507 if conn and "keep-alive" in conn.lower():
508 return False
509
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000510 # Proxy-Connection is a netscape hack.
511 pconn = self.msg.getheader('proxy-connection')
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000512 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000513 return False
514
515 # otherwise, assume it will close
516 return True
517
Greg Steindd6eefb2000-07-18 09:09:48 +0000518 def close(self):
519 if self.fp:
520 self.fp.close()
521 self.fp = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000522
Greg Steindd6eefb2000-07-18 09:09:48 +0000523 def isclosed(self):
524 # NOTE: it is possible that we will not ever call self.close(). This
525 # case occurs when will_close is TRUE, length is None, and we
526 # read up to the last byte, but NOT past it.
527 #
528 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
529 # called, meaning self.isclosed() is meaningful.
530 return self.fp is None
531
Jeremy Hylton2c178252004-08-07 16:28:14 +0000532 # XXX It would be nice to have readline and __iter__ for this, too.
533
Greg Steindd6eefb2000-07-18 09:09:48 +0000534 def read(self, amt=None):
535 if self.fp is None:
536 return ''
537
Senthil Kumaraned9204342010-04-28 17:20:43 +0000538 if self._method == 'HEAD':
Senthil Kumarandfaced52010-06-04 16:32:14 +0000539 self.close()
Senthil Kumaraned9204342010-04-28 17:20:43 +0000540 return ''
541
Greg Steindd6eefb2000-07-18 09:09:48 +0000542 if self.chunked:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000543 return self._read_chunked(amt)
Tim Peters230a60c2002-11-09 05:08:07 +0000544
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000545 if amt is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000546 # unbounded read
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000547 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000548 s = self.fp.read()
549 else:
Antoine Pitroud66c0ee2013-02-02 22:49:34 +0100550 try:
551 s = self._safe_read(self.length)
552 except IncompleteRead:
553 self.close()
554 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000555 self.length = 0
Tim Peters07e99cb2001-01-14 23:47:14 +0000556 self.close() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000557 return s
558
559 if self.length is not None:
560 if amt > self.length:
561 # clip the read to the "end of response"
562 amt = self.length
Greg Steindd6eefb2000-07-18 09:09:48 +0000563
564 # we do not use _safe_read() here because this may be a .will_close
565 # connection, and the user is reading more bytes than will be provided
566 # (for example, reading in 1k chunks)
567 s = self.fp.read(amt)
Serhiy Storchakac97f5ed2013-12-17 21:49:48 +0200568 if not s and amt:
Antoine Pitroud66c0ee2013-02-02 22:49:34 +0100569 # Ideally, we would raise IncompleteRead if the content-length
570 # wasn't satisfied, but it might break compatibility.
571 self.close()
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000572 if self.length is not None:
573 self.length -= len(s)
Facundo Batista70665902007-10-18 03:16:03 +0000574 if not self.length:
575 self.close()
Antoine Pitrou4113d2b2012-12-15 19:11:54 +0100576
Greg Steindd6eefb2000-07-18 09:09:48 +0000577 return s
578
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000579 def _read_chunked(self, amt):
580 assert self.chunked != _UNKNOWN
581 chunk_left = self.chunk_left
Chris Withers68c6e922009-09-04 16:12:32 +0000582 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000583 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000584 if chunk_left is None:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000585 line = self.fp.readline(_MAXLINE + 1)
586 if len(line) > _MAXLINE:
587 raise LineTooLong("chunk size")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000588 i = line.find(';')
589 if i >= 0:
590 line = line[:i] # strip chunk-extensions
Georg Brandl23635032008-02-24 00:03:22 +0000591 try:
592 chunk_left = int(line, 16)
593 except ValueError:
594 # close the connection as protocol synchronisation is
595 # probably lost
596 self.close()
Chris Withers68c6e922009-09-04 16:12:32 +0000597 raise IncompleteRead(''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000598 if chunk_left == 0:
599 break
600 if amt is None:
Chris Withers68c6e922009-09-04 16:12:32 +0000601 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000602 elif amt < chunk_left:
Chris Withers68c6e922009-09-04 16:12:32 +0000603 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000604 self.chunk_left = chunk_left - amt
Chris Withers68c6e922009-09-04 16:12:32 +0000605 return ''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000606 elif amt == chunk_left:
Chris Withers68c6e922009-09-04 16:12:32 +0000607 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000608 self._safe_read(2) # toss the CRLF at the end of the chunk
609 self.chunk_left = None
Chris Withers68c6e922009-09-04 16:12:32 +0000610 return ''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000611 else:
Chris Withers68c6e922009-09-04 16:12:32 +0000612 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000613 amt -= chunk_left
614
615 # we read the whole chunk, get another
616 self._safe_read(2) # toss the CRLF at the end of the chunk
617 chunk_left = None
618
619 # read and discard trailer up to the CRLF terminator
620 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000621 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000622 line = self.fp.readline(_MAXLINE + 1)
623 if len(line) > _MAXLINE:
624 raise LineTooLong("trailer line")
Martin v. Löwisdad88dc2008-02-12 18:47:34 +0000625 if not line:
626 # a vanishingly small number of sites EOF without
627 # sending the trailer
628 break
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000629 if line == '\r\n':
630 break
631
632 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000633 self.close()
634
Chris Withers68c6e922009-09-04 16:12:32 +0000635 return ''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000636
Greg Steindd6eefb2000-07-18 09:09:48 +0000637 def _safe_read(self, amt):
638 """Read the number of bytes requested, compensating for partial reads.
639
640 Normally, we have a blocking socket, but a read() can be interrupted
641 by a signal (resulting in a partial read).
642
643 Note that we cannot distinguish between EOF and an interrupt when zero
644 bytes have been read. IncompleteRead() will be raised in this
645 situation.
646
647 This function should be used when <amt> bytes "should" be present for
648 reading. If the bytes are truly not available (due to EOF), then the
649 IncompleteRead exception can be used to detect the problem.
650 """
Gregory P. Smith8cabfa32009-08-15 22:39:03 +0000651 # NOTE(gps): As of svn r74426 socket._fileobject.read(x) will never
652 # return less than x bytes unless EOF is encountered. It now handles
653 # signal interruptions (socket.error EINTR) internally. This code
654 # never caught that exception anyways. It seems largely pointless.
655 # self.fp.read(amt) will work fine.
Georg Brandl80ba8e82005-09-29 20:16:07 +0000656 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000657 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000658 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000659 if not chunk:
Benjamin Peterson7d49bba2009-03-02 22:41:42 +0000660 raise IncompleteRead(''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000661 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000662 amt -= len(chunk)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000663 return ''.join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000664
Senthil Kumarand389cb52010-09-21 01:38:15 +0000665 def fileno(self):
666 return self.fp.fileno()
667
Greg Steindd6eefb2000-07-18 09:09:48 +0000668 def getheader(self, name, default=None):
669 if self.msg is None:
670 raise ResponseNotReady()
671 return self.msg.getheader(name, default)
Greg Stein5e0fa402000-06-26 08:28:01 +0000672
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000673 def getheaders(self):
674 """Return list of (header, value) tuples."""
675 if self.msg is None:
676 raise ResponseNotReady()
677 return self.msg.items()
678
Greg Stein5e0fa402000-06-26 08:28:01 +0000679
680class HTTPConnection:
681
Greg Steindd6eefb2000-07-18 09:09:48 +0000682 _http_vsn = 11
683 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000684
Greg Steindd6eefb2000-07-18 09:09:48 +0000685 response_class = HTTPResponse
686 default_port = HTTP_PORT
687 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000688 debuglevel = 0
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000689 strict = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000690
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000691 def __init__(self, host, port=None, strict=None,
Gregory P. Smith9d325212010-01-03 02:06:07 +0000692 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Facundo Batista07c78be2007-03-23 18:54:07 +0000693 self.timeout = timeout
Gregory P. Smith9d325212010-01-03 02:06:07 +0000694 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000695 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000696 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000697 self.__response = None
698 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000699 self._method = None
Senthil Kumarane266f252009-05-24 09:14:50 +0000700 self._tunnel_host = None
701 self._tunnel_port = None
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000702 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000703
Greg Steindd6eefb2000-07-18 09:09:48 +0000704 self._set_hostport(host, port)
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000705 if strict is not None:
706 self.strict = strict
Greg Stein5e0fa402000-06-26 08:28:01 +0000707
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000708 def set_tunnel(self, host, port=None, headers=None):
709 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
710
711 The headers argument should be a mapping of extra HTTP headers
712 to send with the CONNECT request.
713 """
Senthil Kumarane266f252009-05-24 09:14:50 +0000714 self._tunnel_host = host
715 self._tunnel_port = port
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000716 if headers:
717 self._tunnel_headers = headers
718 else:
719 self._tunnel_headers.clear()
Senthil Kumarane266f252009-05-24 09:14:50 +0000720
Greg Steindd6eefb2000-07-18 09:09:48 +0000721 def _set_hostport(self, host, port):
722 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000723 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000724 j = host.rfind(']') # ipv6 addresses have [...]
725 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000726 try:
727 port = int(host[i+1:])
728 except ValueError:
Łukasz Langa7a153902011-10-18 17:16:00 +0200729 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
730 port = self.default_port
731 else:
732 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000733 host = host[:i]
734 else:
735 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000736 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000737 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000738 self.host = host
739 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000740
Jeremy Hylton30f86742000-09-18 22:50:38 +0000741 def set_debuglevel(self, level):
742 self.debuglevel = level
743
Senthil Kumarane266f252009-05-24 09:14:50 +0000744 def _tunnel(self):
745 self._set_hostport(self._tunnel_host, self._tunnel_port)
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000746 self.send("CONNECT %s:%d HTTP/1.0\r\n" % (self.host, self.port))
747 for header, value in self._tunnel_headers.iteritems():
748 self.send("%s: %s\r\n" % (header, value))
749 self.send("\r\n")
Senthil Kumarane266f252009-05-24 09:14:50 +0000750 response = self.response_class(self.sock, strict = self.strict,
751 method = self._method)
752 (version, code, message) = response._read_status()
753
754 if code != 200:
755 self.close()
Antoine Pitrou92331d52009-10-14 19:04:48 +0000756 raise socket.error("Tunnel connection failed: %d %s" % (code,
757 message.strip()))
Senthil Kumarane266f252009-05-24 09:14:50 +0000758 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000759 line = response.fp.readline(_MAXLINE + 1)
760 if len(line) > _MAXLINE:
761 raise LineTooLong("header line")
Senthil Kumaran02671852012-04-23 23:46:46 +0800762 if not line:
763 # for sites which EOF without sending trailer
764 break
765 if line == '\r\n':
766 break
Senthil Kumarane266f252009-05-24 09:14:50 +0000767
768
Greg Steindd6eefb2000-07-18 09:09:48 +0000769 def connect(self):
770 """Connect to the host and port specified in __init__."""
Georg Brandlf03facf2007-03-26 20:28:28 +0000771 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smith9d325212010-01-03 02:06:07 +0000772 self.timeout, self.source_address)
Greg Stein5e0fa402000-06-26 08:28:01 +0000773
Senthil Kumarane266f252009-05-24 09:14:50 +0000774 if self._tunnel_host:
775 self._tunnel()
776
Greg Steindd6eefb2000-07-18 09:09:48 +0000777 def close(self):
778 """Close the connection to the HTTP server."""
779 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000780 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000781 self.sock = None
782 if self.__response:
783 self.__response.close()
784 self.__response = None
785 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000786
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000787 def send(self, data):
788 """Send `data' to the server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000789 if self.sock is None:
790 if self.auto_open:
791 self.connect()
792 else:
793 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000794
Jeremy Hylton30f86742000-09-18 22:50:38 +0000795 if self.debuglevel > 0:
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000796 print "send:", repr(data)
Victor Stinner2c6aee92010-07-24 02:46:16 +0000797 blocksize = 8192
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000798 if hasattr(data,'read') and not isinstance(data, array):
Victor Stinner2c6aee92010-07-24 02:46:16 +0000799 if self.debuglevel > 0: print "sendIng a read()able"
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000800 datablock = data.read(blocksize)
801 while datablock:
802 self.sock.sendall(datablock)
803 datablock = data.read(blocksize)
Victor Stinner2c6aee92010-07-24 02:46:16 +0000804 else:
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000805 self.sock.sendall(data)
Greg Stein5e0fa402000-06-26 08:28:01 +0000806
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000807 def _output(self, s):
808 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000809
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000810 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000811 """
812 self._buffer.append(s)
813
Jeremy Hylton0381f482008-11-29 01:09:35 +0000814 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000815 """Send the currently buffered request and clear the buffer.
816
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000817 Appends an extra \\r\\n to the buffer.
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000818 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000819 """
820 self._buffer.extend(("", ""))
821 msg = "\r\n".join(self._buffer)
822 del self._buffer[:]
Jeremy Hylton0381f482008-11-29 01:09:35 +0000823 # If msg and message_body are sent in a single send() call,
824 # it will avoid performance problems caused by the interaction
Ezio Melotti24b07bc2011-03-15 18:55:01 +0200825 # between delayed ack and the Nagle algorithm.
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000826 if isinstance(message_body, str):
Jeremy Hylton0381f482008-11-29 01:09:35 +0000827 msg += message_body
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000828 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000829 self.send(msg)
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000830 if message_body is not None:
831 #message_body was not a string (i.e. it is a file) and
832 #we must run the risk of Nagle
833 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000834
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000835 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000836 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000837
Greg Steindd6eefb2000-07-18 09:09:48 +0000838 `method' specifies an HTTP request method, e.g. 'GET'.
839 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000840 `skip_host' if True does not add automatically a 'Host:' header
841 `skip_accept_encoding' if True does not add automatically an
842 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000843 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000844
Greg Stein616a58d2003-06-24 06:35:19 +0000845 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000846 if self.__response and self.__response.isclosed():
847 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000848
Tim Peters58eb11c2004-01-18 20:29:55 +0000849
Greg Steindd6eefb2000-07-18 09:09:48 +0000850 # in certain cases, we cannot issue another request on this connection.
851 # this occurs when:
852 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
853 # 2) a response to a previous request has signalled that it is going
854 # to close the connection upon completion.
855 # 3) the headers for the previous response have not been read, thus
856 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
857 #
858 # if there is no prior response, then we can request at will.
859 #
860 # if point (2) is true, then we will have passed the socket to the
861 # response (effectively meaning, "there is no prior response"), and
862 # will open a new one when a new request is made.
863 #
864 # Note: if a prior response exists, then we *can* start a new request.
865 # We are not allowed to begin fetching the response to this new
866 # request, however, until that prior response is complete.
867 #
868 if self.__state == _CS_IDLE:
869 self.__state = _CS_REQ_STARTED
870 else:
871 raise CannotSendRequest()
Greg Stein5e0fa402000-06-26 08:28:01 +0000872
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000873 # Save the method we use, we need it later in the response phase
874 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000875 if not url:
876 url = '/'
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000877 hdr = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000878
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000879 self._output(hdr)
Greg Stein5e0fa402000-06-26 08:28:01 +0000880
Greg Steindd6eefb2000-07-18 09:09:48 +0000881 if self._http_vsn == 11:
882 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000883
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000884 if not skip_host:
885 # this header is issued *only* for HTTP/1.1
886 # connections. more specifically, this means it is
887 # only issued when the client uses the new
888 # HTTPConnection() class. backwards-compat clients
889 # will be using HTTP/1.0 and those clients may be
890 # issuing this header themselves. we should NOT issue
891 # it twice; some web servers (such as Apache) barf
892 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000893
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000894 # If we need a non-standard port,include it in the
895 # header. If the request is going through a proxy,
896 # but the host of the actual URL, not the host of the
897 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000898
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000899 netloc = ''
900 if url.startswith('http'):
901 nil, netloc, nil, nil, nil = urlsplit(url)
902
903 if netloc:
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000904 try:
905 netloc_enc = netloc.encode("ascii")
906 except UnicodeEncodeError:
907 netloc_enc = netloc.encode("idna")
908 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000909 else:
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000910 try:
911 host_enc = self.host.encode("ascii")
912 except UnicodeEncodeError:
913 host_enc = self.host.encode("idna")
Senthil Kumaran501bfd82010-11-14 03:31:52 +0000914 # Wrap the IPv6 Host Header with [] (RFC 2732)
915 if host_enc.find(':') >= 0:
916 host_enc = "[" + host_enc + "]"
Gregory P. Smithc4691ec2008-07-07 05:09:12 +0000917 if self.port == self.default_port:
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000918 self.putheader('Host', host_enc)
919 else:
920 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000921
Greg Steindd6eefb2000-07-18 09:09:48 +0000922 # note: we are assuming that clients will not attempt to set these
923 # headers since *this* library must deal with the
924 # consequences. this also means that when the supporting
925 # libraries are updated to recognize other forms, then this
926 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000927
Greg Steindd6eefb2000-07-18 09:09:48 +0000928 # we only want a Content-Encoding of "identity" since we don't
929 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000930 if not skip_accept_encoding:
931 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000932
Greg Steindd6eefb2000-07-18 09:09:48 +0000933 # we can accept "chunked" Transfer-Encodings, but no others
934 # NOTE: no TE header implies *only* "chunked"
935 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000936
Greg Steindd6eefb2000-07-18 09:09:48 +0000937 # if TE is supplied in the header, then it must appear in a
938 # Connection header.
939 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000940
Greg Steindd6eefb2000-07-18 09:09:48 +0000941 else:
942 # For HTTP/1.0, the server will assume "not chunked"
943 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000944
Brett Cannone3d0bf72008-11-15 22:40:44 +0000945 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000946 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000947
Greg Steindd6eefb2000-07-18 09:09:48 +0000948 For example: h.putheader('Accept', 'text/html')
949 """
950 if self.__state != _CS_REQ_STARTED:
951 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000952
Senthil Kumaranaa5f49e2010-10-03 18:26:07 +0000953 hdr = '%s: %s' % (header, '\r\n\t'.join([str(v) for v in values]))
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000954 self._output(hdr)
Greg Stein5e0fa402000-06-26 08:28:01 +0000955
Jeremy Hylton0381f482008-11-29 01:09:35 +0000956 def endheaders(self, message_body=None):
957 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000958
Jeremy Hylton0381f482008-11-29 01:09:35 +0000959 This method sends the request to the server. The optional
Senthil Kumaran41d663f2011-10-05 23:52:49 +0800960 message_body argument can be used to pass a message body
Jeremy Hylton0381f482008-11-29 01:09:35 +0000961 associated with the request. The message body will be sent in
Senthil Kumaran41d663f2011-10-05 23:52:49 +0800962 the same packet as the message headers if it is string, otherwise it is
963 sent as a separate packet.
Jeremy Hylton0381f482008-11-29 01:09:35 +0000964 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000965 if self.__state == _CS_REQ_STARTED:
966 self.__state = _CS_REQ_SENT
967 else:
968 raise CannotSendHeader()
Jeremy Hylton0381f482008-11-29 01:09:35 +0000969 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000970
Greg Steindd6eefb2000-07-18 09:09:48 +0000971 def request(self, method, url, body=None, headers={}):
972 """Send a complete request to the server."""
Victor Stinner2c6aee92010-07-24 02:46:16 +0000973 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000974
Jeremy Hylton0381f482008-11-29 01:09:35 +0000975 def _set_content_length(self, body):
976 # Set the content-length based on the body.
977 thelen = None
978 try:
979 thelen = str(len(body))
980 except TypeError, te:
981 # If this is a file-like object, try to
982 # fstat its file descriptor
Jeremy Hylton0381f482008-11-29 01:09:35 +0000983 try:
984 thelen = str(os.fstat(body.fileno()).st_size)
985 except (AttributeError, OSError):
986 # Don't send a length if this failed
987 if self.debuglevel > 0: print "Cannot stat!!"
988
989 if thelen is not None:
990 self.putheader('Content-Length', thelen)
991
Greg Steindd6eefb2000-07-18 09:09:48 +0000992 def _send_request(self, method, url, body, headers):
Victor Stinner2c6aee92010-07-24 02:46:16 +0000993 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +0000994 header_names = dict.fromkeys([k.lower() for k in headers])
995 skips = {}
996 if 'host' in header_names:
997 skips['skip_host'] = 1
998 if 'accept-encoding' in header_names:
999 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001000
Jeremy Hylton2c178252004-08-07 16:28:14 +00001001 self.putrequest(method, url, **skips)
1002
Senthil Kumaran618802d2012-05-19 16:52:21 +08001003 if body is not None and 'content-length' not in header_names:
Jeremy Hylton0381f482008-11-29 01:09:35 +00001004 self._set_content_length(body)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +00001005 for hdr, value in headers.iteritems():
Greg Steindd6eefb2000-07-18 09:09:48 +00001006 self.putheader(hdr, value)
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +00001007 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001008
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001009 def getresponse(self, buffering=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001010 "Get the response from the server."
Greg Stein5e0fa402000-06-26 08:28:01 +00001011
Greg Stein616a58d2003-06-24 06:35:19 +00001012 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001013 if self.__response and self.__response.isclosed():
1014 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001015
Greg Steindd6eefb2000-07-18 09:09:48 +00001016 #
1017 # if a prior response exists, then it must be completed (otherwise, we
1018 # cannot read this response's header to determine the connection-close
1019 # behavior)
1020 #
1021 # note: if a prior response existed, but was connection-close, then the
1022 # socket and response were made independent of this HTTPConnection
1023 # object since a new request requires that we open a whole new
1024 # connection
1025 #
1026 # this means the prior response had one of two states:
1027 # 1) will_close: this connection was reset and the prior socket and
1028 # response operate independently
1029 # 2) persistent: the response was retained and we await its
1030 # isclosed() status to become true.
1031 #
1032 if self.__state != _CS_REQ_SENT or self.__response:
1033 raise ResponseNotReady()
Greg Stein5e0fa402000-06-26 08:28:01 +00001034
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001035 args = (self.sock,)
1036 kwds = {"strict":self.strict, "method":self._method}
Jeremy Hylton30f86742000-09-18 22:50:38 +00001037 if self.debuglevel > 0:
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001038 args += (self.debuglevel,)
1039 if buffering:
1040 #only add this keyword if non-default, for compatibility with
1041 #other response_classes.
1042 kwds["buffering"] = True;
1043 response = self.response_class(*args, **kwds)
Greg Stein5e0fa402000-06-26 08:28:01 +00001044
Jeremy Hylton39c03802002-07-12 14:04:09 +00001045 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001046 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001047 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001048
Greg Steindd6eefb2000-07-18 09:09:48 +00001049 if response.will_close:
Martin v. Löwis0af33882007-03-23 13:27:15 +00001050 # this effectively passes the connection to the response
1051 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001052 else:
1053 # remember this, so we can tell when it is complete
1054 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001055
Greg Steindd6eefb2000-07-18 09:09:48 +00001056 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001057
Greg Stein5e0fa402000-06-26 08:28:01 +00001058
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001059class HTTP:
Greg Steindd6eefb2000-07-18 09:09:48 +00001060 "Compatibility class with httplib.py from 1.5."
Greg Stein5e0fa402000-06-26 08:28:01 +00001061
Greg Steindd6eefb2000-07-18 09:09:48 +00001062 _http_vsn = 10
1063 _http_vsn_str = 'HTTP/1.0'
Greg Stein5e0fa402000-06-26 08:28:01 +00001064
Greg Steindd6eefb2000-07-18 09:09:48 +00001065 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +00001066
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001067 _connection_class = HTTPConnection
1068
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001069 def __init__(self, host='', port=None, strict=None):
Greg Steindd6eefb2000-07-18 09:09:48 +00001070 "Provide a default host, since the superclass requires one."
Greg Stein5e0fa402000-06-26 08:28:01 +00001071
Greg Steindd6eefb2000-07-18 09:09:48 +00001072 # some joker passed 0 explicitly, meaning default port
1073 if port == 0:
1074 port = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001075
Andrew Svetlov4bb142b2012-12-18 21:27:37 +02001076 # Note that we may pass an empty string as the host; this will raise
Greg Steindd6eefb2000-07-18 09:09:48 +00001077 # an error when we attempt to connect. Presumably, the client code
1078 # will call connect before then, with a proper host.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001079 self._setup(self._connection_class(host, port, strict))
Greg Stein5e0fa402000-06-26 08:28:01 +00001080
Greg Stein81937a42001-08-18 09:20:23 +00001081 def _setup(self, conn):
1082 self._conn = conn
1083
1084 # set up delegation to flesh out interface
1085 self.send = conn.send
1086 self.putrequest = conn.putrequest
Brett Cannone3d0bf72008-11-15 22:40:44 +00001087 self.putheader = conn.putheader
Greg Stein81937a42001-08-18 09:20:23 +00001088 self.endheaders = conn.endheaders
1089 self.set_debuglevel = conn.set_debuglevel
1090
1091 conn._http_vsn = self._http_vsn
1092 conn._http_vsn_str = self._http_vsn_str
Greg Stein5e0fa402000-06-26 08:28:01 +00001093
Greg Steindd6eefb2000-07-18 09:09:48 +00001094 self.file = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001095
Greg Steindd6eefb2000-07-18 09:09:48 +00001096 def connect(self, host=None, port=None):
1097 "Accept arguments to set the host/port, since the superclass doesn't."
Greg Stein5e0fa402000-06-26 08:28:01 +00001098
Greg Steindd6eefb2000-07-18 09:09:48 +00001099 if host is not None:
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001100 self._conn._set_hostport(host, port)
1101 self._conn.connect()
Greg Stein5e0fa402000-06-26 08:28:01 +00001102
Greg Steindd6eefb2000-07-18 09:09:48 +00001103 def getfile(self):
1104 "Provide a getfile, since the superclass' does not use this concept."
1105 return self.file
Greg Stein5e0fa402000-06-26 08:28:01 +00001106
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001107 def getreply(self, buffering=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001108 """Compat definition since superclass does not define it.
Greg Stein5e0fa402000-06-26 08:28:01 +00001109
Greg Steindd6eefb2000-07-18 09:09:48 +00001110 Returns a tuple consisting of:
1111 - server status code (e.g. '200' if all goes well)
1112 - server "reason" corresponding to status code
1113 - any RFC822 headers in the response from the server
1114 """
1115 try:
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001116 if not buffering:
1117 response = self._conn.getresponse()
1118 else:
1119 #only add this keyword if non-default for compatibility
1120 #with other connection classes
1121 response = self._conn.getresponse(buffering)
Greg Steindd6eefb2000-07-18 09:09:48 +00001122 except BadStatusLine, e:
1123 ### hmm. if getresponse() ever closes the socket on a bad request,
1124 ### then we are going to have problems with self.sock
Greg Stein5e0fa402000-06-26 08:28:01 +00001125
Greg Steindd6eefb2000-07-18 09:09:48 +00001126 ### should we keep this behavior? do people use it?
1127 # keep the socket open (as a file), and return it
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001128 self.file = self._conn.sock.makefile('rb', 0)
Greg Stein5e0fa402000-06-26 08:28:01 +00001129
Greg Steindd6eefb2000-07-18 09:09:48 +00001130 # close our socket -- we want to restart after any protocol error
1131 self.close()
Greg Stein5e0fa402000-06-26 08:28:01 +00001132
Greg Steindd6eefb2000-07-18 09:09:48 +00001133 self.headers = None
1134 return -1, e.line, None
Greg Stein5e0fa402000-06-26 08:28:01 +00001135
Greg Steindd6eefb2000-07-18 09:09:48 +00001136 self.headers = response.msg
1137 self.file = response.fp
1138 return response.status, response.reason, response.msg
Greg Stein5e0fa402000-06-26 08:28:01 +00001139
Greg Steindd6eefb2000-07-18 09:09:48 +00001140 def close(self):
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001141 self._conn.close()
Greg Stein5e0fa402000-06-26 08:28:01 +00001142
Greg Steindd6eefb2000-07-18 09:09:48 +00001143 # note that self.file == response.fp, which gets closed by the
1144 # superclass. just clear the object ref here.
1145 ### hmm. messy. if status==-1, then self.file is owned by us.
1146 ### well... we aren't explicitly closing, but losing this ref will
1147 ### do it
1148 self.file = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001149
Bill Janssen426ea0a2007-08-29 22:35:05 +00001150try:
1151 import ssl
1152except ImportError:
1153 pass
1154else:
1155 class HTTPSConnection(HTTPConnection):
1156 "This class allows communication via SSL."
1157
1158 default_port = HTTPS_PORT
1159
1160 def __init__(self, host, port=None, key_file=None, cert_file=None,
Gregory P. Smith9d325212010-01-03 02:06:07 +00001161 strict=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1162 source_address=None):
1163 HTTPConnection.__init__(self, host, port, strict, timeout,
1164 source_address)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001165 self.key_file = key_file
1166 self.cert_file = cert_file
1167
1168 def connect(self):
1169 "Connect to a host on a given (SSL) port."
1170
Gregory P. Smith9d325212010-01-03 02:06:07 +00001171 sock = socket.create_connection((self.host, self.port),
1172 self.timeout, self.source_address)
Senthil Kumarane266f252009-05-24 09:14:50 +00001173 if self._tunnel_host:
1174 self.sock = sock
1175 self._tunnel()
Bill Janssen98d19da2007-09-10 21:51:02 +00001176 self.sock = ssl.wrap_socket(sock, self.key_file, self.cert_file)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001177
Thomas Woutersa6900e82007-08-30 21:54:39 +00001178 __all__.append("HTTPSConnection")
Bill Janssen426ea0a2007-08-29 22:35:05 +00001179
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001180 class HTTPS(HTTP):
1181 """Compatibility with 1.5 httplib interface
1182
1183 Python 1.5.2 did not have an HTTPS class, but it defined an
1184 interface for sending http requests that is also useful for
Tim Peters5ceadc82001-01-13 19:16:21 +00001185 https.
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001186 """
1187
Martin v. Löwisd7bf9742000-09-21 22:09:47 +00001188 _connection_class = HTTPSConnection
Tim Peters5ceadc82001-01-13 19:16:21 +00001189
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001190 def __init__(self, host='', port=None, key_file=None, cert_file=None,
1191 strict=None):
Greg Stein81937a42001-08-18 09:20:23 +00001192 # provide a default host, pass the X509 cert info
1193
1194 # urf. compensate for bad input.
1195 if port == 0:
1196 port = None
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001197 self._setup(self._connection_class(host, port, key_file,
1198 cert_file, strict))
Greg Stein81937a42001-08-18 09:20:23 +00001199
1200 # we never actually use these for anything, but we keep them
1201 # here for compatibility with post-1.5.2 CVS.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001202 self.key_file = key_file
1203 self.cert_file = cert_file
Greg Stein81937a42001-08-18 09:20:23 +00001204
Greg Stein5e0fa402000-06-26 08:28:01 +00001205
Bill Janssen426ea0a2007-08-29 22:35:05 +00001206 def FakeSocket (sock, sslobj):
Bill Janssenc4592642007-08-31 19:02:23 +00001207 warnings.warn("FakeSocket is deprecated, and won't be in 3.x. " +
Bill Janssen98d19da2007-09-10 21:51:02 +00001208 "Use the result of ssl.wrap_socket() directly instead.",
Bill Janssenc4592642007-08-31 19:02:23 +00001209 DeprecationWarning, stacklevel=2)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001210 return sslobj
1211
1212
Greg Stein5e0fa402000-06-26 08:28:01 +00001213class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001214 # Subclasses that define an __init__ must call Exception.__init__
1215 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001216 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001217
1218class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001219 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001220
Skip Montanaro9d389972002-03-24 16:53:50 +00001221class InvalidURL(HTTPException):
1222 pass
1223
Greg Stein5e0fa402000-06-26 08:28:01 +00001224class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001225 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001226 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001227 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001228
1229class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001230 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001231
Greg Stein5e0fa402000-06-26 08:28:01 +00001232class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001233 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001234
1235class IncompleteRead(HTTPException):
Benjamin Peterson7d49bba2009-03-02 22:41:42 +00001236 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001237 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001238 self.partial = partial
Benjamin Peterson7d49bba2009-03-02 22:41:42 +00001239 self.expected = expected
1240 def __repr__(self):
1241 if self.expected is not None:
1242 e = ', %i more expected' % self.expected
1243 else:
1244 e = ''
1245 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1246 def __str__(self):
1247 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001248
1249class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001250 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001251
1252class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001253 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001254
1255class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001256 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001257
1258class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001259 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001260
1261class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001262 def __init__(self, line):
Dirkjan Ochtmanebc73dc2010-02-24 04:49:00 +00001263 if not line:
1264 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001265 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001266 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001267
Antoine Pitroud7b6ac62010-12-18 18:18:21 +00001268class LineTooLong(HTTPException):
1269 def __init__(self, line_type):
1270 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1271 % (_MAXLINE, line_type))
1272
Greg Stein5e0fa402000-06-26 08:28:01 +00001273# for backwards compatibility
1274error = HTTPException
1275
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001276class LineAndFileWrapper:
1277 """A limited file-like object for HTTP/0.9 responses."""
1278
1279 # The status-line parsing code calls readline(), which normally
1280 # get the HTTP status line. For a 0.9 response, however, this is
1281 # actually the first line of the body! Clients need to get a
1282 # readable file object that contains that line.
1283
1284 def __init__(self, line, file):
1285 self._line = line
1286 self._file = file
1287 self._line_consumed = 0
1288 self._line_offset = 0
1289 self._line_left = len(line)
1290
1291 def __getattr__(self, attr):
1292 return getattr(self._file, attr)
1293
1294 def _done(self):
1295 # called when the last byte is read from the line. After the
1296 # call, all read methods are delegated to the underlying file
Skip Montanaro74b9a7a2003-02-25 17:48:15 +00001297 # object.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001298 self._line_consumed = 1
1299 self.read = self._file.read
1300 self.readline = self._file.readline
1301 self.readlines = self._file.readlines
1302
1303 def read(self, amt=None):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001304 if self._line_consumed:
1305 return self._file.read(amt)
1306 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001307 if amt is None or amt > self._line_left:
1308 s = self._line[self._line_offset:]
1309 self._done()
1310 if amt is None:
1311 return s + self._file.read()
1312 else:
Tim Petersc411dba2002-07-16 21:35:23 +00001313 return s + self._file.read(amt - len(s))
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001314 else:
1315 assert amt <= self._line_left
1316 i = self._line_offset
1317 j = i + amt
1318 s = self._line[i:j]
1319 self._line_offset = j
1320 self._line_left -= amt
1321 if self._line_left == 0:
1322 self._done()
1323 return s
Tim Petersc411dba2002-07-16 21:35:23 +00001324
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001325 def readline(self):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001326 if self._line_consumed:
1327 return self._file.readline()
1328 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001329 s = self._line[self._line_offset:]
1330 self._done()
1331 return s
1332
1333 def readlines(self, size=None):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001334 if self._line_consumed:
1335 return self._file.readlines(size)
1336 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001337 L = [self._line[self._line_offset:]]
1338 self._done()
1339 if size is None:
1340 return L + self._file.readlines()
1341 else:
1342 return L + self._file.readlines(size)