blob: 7223ba151271f16d0eedad83b49a0961b59fc674 [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
Serhiy Storchaka59bdf632015-03-12 11:12:51 +020071import re
Jeremy Hylton6459c8d2001-10-11 17:47:22 +000072import socket
Brett Cannon1eaf0742008-09-02 01:25:16 +000073from sys import py3kwarning
Jeremy Hylton8acf1e02002-03-08 19:35:51 +000074from urlparse import urlsplit
Bill Janssenc4592642007-08-31 19:02:23 +000075import warnings
Brett Cannon1eaf0742008-09-02 01:25:16 +000076with warnings.catch_warnings():
77 if py3kwarning:
78 warnings.filterwarnings("ignore", ".*mimetools has been removed",
79 DeprecationWarning)
Brett Cannon0a4128e2008-08-16 21:56:03 +000080 import mimetools
Guido van Rossum23acc951994-02-21 16:36:04 +000081
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000082try:
Greg Steindd6eefb2000-07-18 09:09:48 +000083 from cStringIO import StringIO
Greg Stein5e0fa402000-06-26 08:28:01 +000084except ImportError:
Greg Steindd6eefb2000-07-18 09:09:48 +000085 from StringIO import StringIO
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000086
Thomas Woutersa6900e82007-08-30 21:54:39 +000087__all__ = ["HTTP", "HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000088 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000089 "UnknownTransferEncoding", "UnimplementedFileMode",
90 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
91 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
Georg Brandl6aab16e2006-02-17 19:17:25 +000092 "BadStatusLine", "error", "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000093
Guido van Rossum23acc951994-02-21 16:36:04 +000094HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000095HTTPS_PORT = 443
96
Greg Stein5e0fa402000-06-26 08:28:01 +000097_UNKNOWN = 'UNKNOWN'
98
99# connection states
100_CS_IDLE = 'Idle'
101_CS_REQ_STARTED = 'Request-started'
102_CS_REQ_SENT = 'Request-sent'
103
Martin v. Löwis39a31782004-09-18 09:03:49 +0000104# status codes
105# informational
106CONTINUE = 100
107SWITCHING_PROTOCOLS = 101
108PROCESSING = 102
109
110# successful
111OK = 200
112CREATED = 201
113ACCEPTED = 202
114NON_AUTHORITATIVE_INFORMATION = 203
115NO_CONTENT = 204
116RESET_CONTENT = 205
117PARTIAL_CONTENT = 206
118MULTI_STATUS = 207
119IM_USED = 226
120
121# redirection
122MULTIPLE_CHOICES = 300
123MOVED_PERMANENTLY = 301
124FOUND = 302
125SEE_OTHER = 303
126NOT_MODIFIED = 304
127USE_PROXY = 305
128TEMPORARY_REDIRECT = 307
129
130# client error
131BAD_REQUEST = 400
132UNAUTHORIZED = 401
133PAYMENT_REQUIRED = 402
134FORBIDDEN = 403
135NOT_FOUND = 404
136METHOD_NOT_ALLOWED = 405
137NOT_ACCEPTABLE = 406
138PROXY_AUTHENTICATION_REQUIRED = 407
139REQUEST_TIMEOUT = 408
140CONFLICT = 409
141GONE = 410
142LENGTH_REQUIRED = 411
143PRECONDITION_FAILED = 412
144REQUEST_ENTITY_TOO_LARGE = 413
145REQUEST_URI_TOO_LONG = 414
146UNSUPPORTED_MEDIA_TYPE = 415
147REQUESTED_RANGE_NOT_SATISFIABLE = 416
148EXPECTATION_FAILED = 417
149UNPROCESSABLE_ENTITY = 422
150LOCKED = 423
151FAILED_DEPENDENCY = 424
152UPGRADE_REQUIRED = 426
153
154# server error
155INTERNAL_SERVER_ERROR = 500
156NOT_IMPLEMENTED = 501
157BAD_GATEWAY = 502
158SERVICE_UNAVAILABLE = 503
159GATEWAY_TIMEOUT = 504
160HTTP_VERSION_NOT_SUPPORTED = 505
161INSUFFICIENT_STORAGE = 507
162NOT_EXTENDED = 510
163
Georg Brandl6aab16e2006-02-17 19:17:25 +0000164# Mapping status codes to official W3C names
165responses = {
166 100: 'Continue',
167 101: 'Switching Protocols',
168
169 200: 'OK',
170 201: 'Created',
171 202: 'Accepted',
172 203: 'Non-Authoritative Information',
173 204: 'No Content',
174 205: 'Reset Content',
175 206: 'Partial Content',
176
177 300: 'Multiple Choices',
178 301: 'Moved Permanently',
179 302: 'Found',
180 303: 'See Other',
181 304: 'Not Modified',
182 305: 'Use Proxy',
183 306: '(Unused)',
184 307: 'Temporary Redirect',
185
186 400: 'Bad Request',
187 401: 'Unauthorized',
188 402: 'Payment Required',
189 403: 'Forbidden',
190 404: 'Not Found',
191 405: 'Method Not Allowed',
192 406: 'Not Acceptable',
193 407: 'Proxy Authentication Required',
194 408: 'Request Timeout',
195 409: 'Conflict',
196 410: 'Gone',
197 411: 'Length Required',
198 412: 'Precondition Failed',
199 413: 'Request Entity Too Large',
200 414: 'Request-URI Too Long',
201 415: 'Unsupported Media Type',
202 416: 'Requested Range Not Satisfiable',
203 417: 'Expectation Failed',
204
205 500: 'Internal Server Error',
206 501: 'Not Implemented',
207 502: 'Bad Gateway',
208 503: 'Service Unavailable',
209 504: 'Gateway Timeout',
210 505: 'HTTP Version Not Supported',
211}
212
Georg Brandl80ba8e82005-09-29 20:16:07 +0000213# maximal amount of data to read at one time in _safe_read
214MAXAMOUNT = 1048576
215
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000216# maximal line length when calling readline().
217_MAXLINE = 65536
218
Berker Peksagb7414e02014-08-05 07:15:57 +0300219# maximum amount of headers accepted
220_MAXHEADERS = 100
221
Serhiy Storchaka59bdf632015-03-12 11:12:51 +0200222# Header name/value ABNF (http://tools.ietf.org/html/rfc7230#section-3.2)
223#
224# VCHAR = %x21-7E
225# obs-text = %x80-FF
226# header-field = field-name ":" OWS field-value OWS
227# field-name = token
228# field-value = *( field-content / obs-fold )
229# field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
230# field-vchar = VCHAR / obs-text
231#
232# obs-fold = CRLF 1*( SP / HTAB )
233# ; obsolete line folding
234# ; see Section 3.2.4
235
236# token = 1*tchar
237#
238# tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*"
239# / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~"
240# / DIGIT / ALPHA
241# ; any VCHAR, except delimiters
242#
243# VCHAR defined in http://tools.ietf.org/html/rfc5234#appendix-B.1
244
245# the patterns for both name and value are more leniant than RFC
246# definitions to allow for backwards compatibility
247_is_legal_header_name = re.compile(r'\A[^:\s][^:\r\n]*\Z').match
248_is_illegal_header_value = re.compile(r'\n(?![ \t])|\r(?![ \t\n])').search
249
R David Murrayb4b000f2015-03-22 15:15:44 -0400250# We always set the Content-Length header for these methods because some
251# servers will otherwise respond with a 411
252_METHODS_EXPECTING_BODY = {'PATCH', 'POST', 'PUT'}
253
Berker Peksagb7414e02014-08-05 07:15:57 +0300254
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000255class HTTPMessage(mimetools.Message):
256
257 def addheader(self, key, value):
258 """Add header for field key handling repeats."""
259 prev = self.dict.get(key)
260 if prev is None:
261 self.dict[key] = value
262 else:
263 combined = ", ".join((prev, value))
264 self.dict[key] = combined
265
266 def addcontinue(self, key, more):
267 """Add more field data from a continuation line."""
268 prev = self.dict[key]
269 self.dict[key] = prev + "\n " + more
270
271 def readheaders(self):
272 """Read header lines.
273
274 Read header lines up to the entirely blank line that terminates them.
275 The (normally blank) line that ends the headers is skipped, but not
276 included in the returned list. If a non-header line ends the headers,
277 (which is an error), an attempt is made to backspace over it; it is
278 never included in the returned list.
279
280 The variable self.status is set to the empty string if all went well,
281 otherwise it is an error message. The variable self.headers is a
282 completely uninterpreted list of lines contained in the header (so
283 printing them will reproduce the header exactly as it appears in the
284 file).
285
286 If multiple header fields with the same name occur, they are combined
287 according to the rules in RFC 2616 sec 4.2:
288
289 Appending each subsequent field-value to the first, each separated
290 by a comma. The order in which header fields with the same field-name
291 are received is significant to the interpretation of the combined
292 field value.
293 """
294 # XXX The implementation overrides the readheaders() method of
295 # rfc822.Message. The base class design isn't amenable to
296 # customized behavior here so the method here is a copy of the
297 # base class code with a few small changes.
298
299 self.dict = {}
300 self.unixfrom = ''
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000301 self.headers = hlist = []
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000302 self.status = ''
303 headerseen = ""
304 firstline = 1
305 startofline = unread = tell = None
306 if hasattr(self.fp, 'unread'):
307 unread = self.fp.unread
308 elif self.seekable:
309 tell = self.fp.tell
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000310 while True:
Berker Peksagb7414e02014-08-05 07:15:57 +0300311 if len(hlist) > _MAXHEADERS:
312 raise HTTPException("got more than %d headers" % _MAXHEADERS)
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000313 if tell:
314 try:
315 startofline = tell()
316 except IOError:
317 startofline = tell = None
318 self.seekable = 0
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000319 line = self.fp.readline(_MAXLINE + 1)
320 if len(line) > _MAXLINE:
321 raise LineTooLong("header line")
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000322 if not line:
323 self.status = 'EOF in headers'
324 break
325 # Skip unix From name time lines
326 if firstline and line.startswith('From '):
327 self.unixfrom = self.unixfrom + line
328 continue
329 firstline = 0
330 if headerseen and line[0] in ' \t':
331 # XXX Not sure if continuation lines are handled properly
332 # for http and/or for repeating headers
333 # It's a continuation line.
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000334 hlist.append(line)
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000335 self.addcontinue(headerseen, line.strip())
336 continue
337 elif self.iscomment(line):
338 # It's a comment. Ignore it.
339 continue
340 elif self.islast(line):
341 # Note! No pushback here! The delimiter line gets eaten.
342 break
343 headerseen = self.isheader(line)
344 if headerseen:
345 # It's a legal header line, save it.
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000346 hlist.append(line)
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000347 self.addheader(headerseen, line[len(headerseen)+1:].strip())
348 continue
Benjamin Petersonbfd976f2015-01-25 23:34:42 -0500349 elif headerseen is not None:
350 # An empty header name. These aren't allowed in HTTP, but it's
351 # probably a benign mistake. Don't add the header, just keep
352 # going.
353 continue
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000354 else:
355 # It's not a header line; throw it back and stop here.
356 if not self.dict:
357 self.status = 'No headers'
358 else:
359 self.status = 'Non-header line where header expected'
360 # Try to undo the read.
361 if unread:
362 unread(line)
363 elif tell:
364 self.fp.seek(startofline)
365 else:
366 self.status = self.status + '; bad seek'
367 break
Greg Stein5e0fa402000-06-26 08:28:01 +0000368
369class HTTPResponse:
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000370
371 # strict: If true, raise BadStatusLine if the status line can't be
372 # parsed as a valid HTTP/1.0 or 1.1 status line. By default it is
Skip Montanaro186bec22002-07-25 16:10:38 +0000373 # false because it prevents clients from talking to HTTP/0.9
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000374 # servers. Note that a response with a sufficiently corrupted
375 # status line will look like an HTTP/0.9 response.
376
377 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
378
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +0000379 def __init__(self, sock, debuglevel=0, strict=0, method=None, buffering=False):
380 if buffering:
381 # The caller won't be using any sock.recv() calls, so buffering
Gregory P. Smith77db0162009-08-20 09:38:43 +0000382 # is fine and recommended for performance.
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +0000383 self.fp = sock.makefile('rb')
384 else:
385 # The buffer size is specified as zero, because the headers of
386 # the response are read with readline(). If the reads were
387 # buffered the readline() calls could consume some of the
388 # response, which make be read via a recv() on the underlying
389 # socket.
390 self.fp = sock.makefile('rb', 0)
Jeremy Hylton30f86742000-09-18 22:50:38 +0000391 self.debuglevel = debuglevel
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000392 self.strict = strict
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000393 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000394
Greg Steindd6eefb2000-07-18 09:09:48 +0000395 self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000396
Greg Steindd6eefb2000-07-18 09:09:48 +0000397 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000398 self.version = _UNKNOWN # HTTP-Version
399 self.status = _UNKNOWN # Status-Code
400 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000401
Tim Peters07e99cb2001-01-14 23:47:14 +0000402 self.chunked = _UNKNOWN # is "chunked" being used?
403 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
404 self.length = _UNKNOWN # number of bytes left in response
405 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000406
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000407 def _read_status(self):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000408 # Initialize with Simple-Response defaults
Christian Heimes671138f2012-09-25 13:29:30 +0200409 line = self.fp.readline(_MAXLINE + 1)
410 if len(line) > _MAXLINE:
411 raise LineTooLong("header line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000412 if self.debuglevel > 0:
413 print "reply:", repr(line)
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000414 if not line:
415 # Presumably, the server closed the connection before
416 # sending a valid response.
417 raise BadStatusLine(line)
Greg Steindd6eefb2000-07-18 09:09:48 +0000418 try:
Guido van Rossum34735a62000-12-15 15:09:42 +0000419 [version, status, reason] = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000420 except ValueError:
421 try:
Guido van Rossum34735a62000-12-15 15:09:42 +0000422 [version, status] = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000423 reason = ""
424 except ValueError:
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000425 # empty version will cause next test to fail and status
426 # will be treated as 0.9 response.
427 version = ""
428 if not version.startswith('HTTP/'):
429 if self.strict:
430 self.close()
431 raise BadStatusLine(line)
432 else:
433 # assume it's a Simple-Response from an 0.9 server
434 self.fp = LineAndFileWrapper(line, self.fp)
435 return "HTTP/0.9", 200, ""
Greg Stein5e0fa402000-06-26 08:28:01 +0000436
Jeremy Hylton23d40472001-04-13 14:57:08 +0000437 # The status code is a three-digit number
438 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000439 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000440 if status < 100 or status > 999:
441 raise BadStatusLine(line)
442 except ValueError:
443 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000444 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000445
Jeremy Hylton39c03802002-07-12 14:04:09 +0000446 def begin(self):
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000447 if self.msg is not None:
448 # we've already started reading the response
449 return
450
451 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000452 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000453 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000454 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000455 break
456 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000457 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000458 skip = self.fp.readline(_MAXLINE + 1)
459 if len(skip) > _MAXLINE:
460 raise LineTooLong("header line")
461 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000462 if not skip:
463 break
464 if self.debuglevel > 0:
465 print "header:", skip
Tim Petersc411dba2002-07-16 21:35:23 +0000466
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000467 self.status = status
468 self.reason = reason.strip()
Greg Steindd6eefb2000-07-18 09:09:48 +0000469 if version == 'HTTP/1.0':
470 self.version = 10
Jeremy Hylton110941a2000-10-12 19:58:36 +0000471 elif version.startswith('HTTP/1.'):
Tim Peters07e99cb2001-01-14 23:47:14 +0000472 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Jeremy Hylton110941a2000-10-12 19:58:36 +0000473 elif version == 'HTTP/0.9':
474 self.version = 9
Greg Steindd6eefb2000-07-18 09:09:48 +0000475 else:
476 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000477
Jeremy Hylton110941a2000-10-12 19:58:36 +0000478 if self.version == 9:
Georg Brandl0aade9a2005-06-26 22:06:54 +0000479 self.length = None
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000480 self.chunked = 0
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000481 self.will_close = 1
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000482 self.msg = HTTPMessage(StringIO())
Jeremy Hylton110941a2000-10-12 19:58:36 +0000483 return
484
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000485 self.msg = HTTPMessage(self.fp, 0)
Jeremy Hylton30f86742000-09-18 22:50:38 +0000486 if self.debuglevel > 0:
487 for hdr in self.msg.headers:
488 print "header:", hdr,
Greg Stein5e0fa402000-06-26 08:28:01 +0000489
Greg Steindd6eefb2000-07-18 09:09:48 +0000490 # don't let the msg keep an fp
491 self.msg.fp = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000492
Greg Steindd6eefb2000-07-18 09:09:48 +0000493 # are we using the chunked-style of transfer encoding?
494 tr_enc = self.msg.getheader('transfer-encoding')
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000495 if tr_enc and tr_enc.lower() == "chunked":
Greg Steindd6eefb2000-07-18 09:09:48 +0000496 self.chunked = 1
497 self.chunk_left = None
498 else:
499 self.chunked = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000500
Greg Steindd6eefb2000-07-18 09:09:48 +0000501 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000502 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000503
Greg Steindd6eefb2000-07-18 09:09:48 +0000504 # do we have a Content-Length?
505 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
506 length = self.msg.getheader('content-length')
507 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000508 try:
509 self.length = int(length)
510 except ValueError:
511 self.length = None
Georg Brandl8c460d52008-02-24 00:14:24 +0000512 else:
513 if self.length < 0: # ignore nonsensical negative lengths
514 self.length = None
Greg Steindd6eefb2000-07-18 09:09:48 +0000515 else:
516 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000517
Greg Steindd6eefb2000-07-18 09:09:48 +0000518 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000519 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000520 100 <= status < 200 or # 1xx codes
521 self._method == 'HEAD'):
Greg Steindd6eefb2000-07-18 09:09:48 +0000522 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000523
Greg Steindd6eefb2000-07-18 09:09:48 +0000524 # if the connection remains open, and we aren't using chunked, and
525 # a content-length was not provided, then assume that the connection
526 # WILL close.
527 if not self.will_close and \
528 not self.chunked and \
529 self.length is None:
530 self.will_close = 1
Greg Stein5e0fa402000-06-26 08:28:01 +0000531
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000532 def _check_close(self):
Jeremy Hylton2c178252004-08-07 16:28:14 +0000533 conn = self.msg.getheader('connection')
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000534 if self.version == 11:
535 # An HTTP/1.1 proxy is assumed to stay open unless
536 # explicitly closed.
537 conn = self.msg.getheader('connection')
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000538 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000539 return True
540 return False
541
Jeremy Hylton2c178252004-08-07 16:28:14 +0000542 # Some HTTP/1.0 implementations have support for persistent
543 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000544
Georg Brandl26458c12007-12-08 10:56:39 +0000545 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000546 if self.msg.getheader('keep-alive'):
547 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000548
Jeremy Hylton2c178252004-08-07 16:28:14 +0000549 # At least Akamai returns a "Connection: Keep-Alive" header,
550 # which was supposed to be sent by the client.
551 if conn and "keep-alive" in conn.lower():
552 return False
553
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000554 # Proxy-Connection is a netscape hack.
555 pconn = self.msg.getheader('proxy-connection')
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000556 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000557 return False
558
559 # otherwise, assume it will close
560 return True
561
Greg Steindd6eefb2000-07-18 09:09:48 +0000562 def close(self):
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +0300563 fp = self.fp
564 if fp:
Greg Steindd6eefb2000-07-18 09:09:48 +0000565 self.fp = None
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +0300566 fp.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000567
Greg Steindd6eefb2000-07-18 09:09:48 +0000568 def isclosed(self):
569 # NOTE: it is possible that we will not ever call self.close(). This
570 # case occurs when will_close is TRUE, length is None, and we
571 # read up to the last byte, but NOT past it.
572 #
573 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
574 # called, meaning self.isclosed() is meaningful.
575 return self.fp is None
576
Jeremy Hylton2c178252004-08-07 16:28:14 +0000577 # XXX It would be nice to have readline and __iter__ for this, too.
578
Greg Steindd6eefb2000-07-18 09:09:48 +0000579 def read(self, amt=None):
580 if self.fp is None:
581 return ''
582
Senthil Kumaraned9204342010-04-28 17:20:43 +0000583 if self._method == 'HEAD':
Senthil Kumarandfaced52010-06-04 16:32:14 +0000584 self.close()
Senthil Kumaraned9204342010-04-28 17:20:43 +0000585 return ''
586
Greg Steindd6eefb2000-07-18 09:09:48 +0000587 if self.chunked:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000588 return self._read_chunked(amt)
Tim Peters230a60c2002-11-09 05:08:07 +0000589
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000590 if amt is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000591 # unbounded read
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000592 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000593 s = self.fp.read()
594 else:
Antoine Pitroud66c0ee2013-02-02 22:49:34 +0100595 try:
596 s = self._safe_read(self.length)
597 except IncompleteRead:
598 self.close()
599 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000600 self.length = 0
Tim Peters07e99cb2001-01-14 23:47:14 +0000601 self.close() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000602 return s
603
604 if self.length is not None:
605 if amt > self.length:
606 # clip the read to the "end of response"
607 amt = self.length
Greg Steindd6eefb2000-07-18 09:09:48 +0000608
609 # we do not use _safe_read() here because this may be a .will_close
610 # connection, and the user is reading more bytes than will be provided
611 # (for example, reading in 1k chunks)
612 s = self.fp.read(amt)
Serhiy Storchakac97f5ed2013-12-17 21:49:48 +0200613 if not s and amt:
Antoine Pitroud66c0ee2013-02-02 22:49:34 +0100614 # Ideally, we would raise IncompleteRead if the content-length
615 # wasn't satisfied, but it might break compatibility.
616 self.close()
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000617 if self.length is not None:
618 self.length -= len(s)
Facundo Batista70665902007-10-18 03:16:03 +0000619 if not self.length:
620 self.close()
Antoine Pitrou4113d2b2012-12-15 19:11:54 +0100621
Greg Steindd6eefb2000-07-18 09:09:48 +0000622 return s
623
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000624 def _read_chunked(self, amt):
625 assert self.chunked != _UNKNOWN
626 chunk_left = self.chunk_left
Chris Withers68c6e922009-09-04 16:12:32 +0000627 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000628 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000629 if chunk_left is None:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000630 line = self.fp.readline(_MAXLINE + 1)
631 if len(line) > _MAXLINE:
632 raise LineTooLong("chunk size")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000633 i = line.find(';')
634 if i >= 0:
635 line = line[:i] # strip chunk-extensions
Georg Brandl23635032008-02-24 00:03:22 +0000636 try:
637 chunk_left = int(line, 16)
638 except ValueError:
639 # close the connection as protocol synchronisation is
640 # probably lost
641 self.close()
Chris Withers68c6e922009-09-04 16:12:32 +0000642 raise IncompleteRead(''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000643 if chunk_left == 0:
644 break
645 if amt is None:
Chris Withers68c6e922009-09-04 16:12:32 +0000646 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000647 elif amt < chunk_left:
Chris Withers68c6e922009-09-04 16:12:32 +0000648 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000649 self.chunk_left = chunk_left - amt
Chris Withers68c6e922009-09-04 16:12:32 +0000650 return ''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000651 elif amt == chunk_left:
Chris Withers68c6e922009-09-04 16:12:32 +0000652 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000653 self._safe_read(2) # toss the CRLF at the end of the chunk
654 self.chunk_left = None
Chris Withers68c6e922009-09-04 16:12:32 +0000655 return ''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000656 else:
Chris Withers68c6e922009-09-04 16:12:32 +0000657 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000658 amt -= chunk_left
659
660 # we read the whole chunk, get another
661 self._safe_read(2) # toss the CRLF at the end of the chunk
662 chunk_left = None
663
664 # read and discard trailer up to the CRLF terminator
665 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000666 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000667 line = self.fp.readline(_MAXLINE + 1)
668 if len(line) > _MAXLINE:
669 raise LineTooLong("trailer line")
Martin v. Löwisdad88dc2008-02-12 18:47:34 +0000670 if not line:
671 # a vanishingly small number of sites EOF without
672 # sending the trailer
673 break
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000674 if line == '\r\n':
675 break
676
677 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000678 self.close()
679
Chris Withers68c6e922009-09-04 16:12:32 +0000680 return ''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000681
Greg Steindd6eefb2000-07-18 09:09:48 +0000682 def _safe_read(self, amt):
683 """Read the number of bytes requested, compensating for partial reads.
684
685 Normally, we have a blocking socket, but a read() can be interrupted
686 by a signal (resulting in a partial read).
687
688 Note that we cannot distinguish between EOF and an interrupt when zero
689 bytes have been read. IncompleteRead() will be raised in this
690 situation.
691
692 This function should be used when <amt> bytes "should" be present for
693 reading. If the bytes are truly not available (due to EOF), then the
694 IncompleteRead exception can be used to detect the problem.
695 """
Gregory P. Smith8cabfa32009-08-15 22:39:03 +0000696 # NOTE(gps): As of svn r74426 socket._fileobject.read(x) will never
697 # return less than x bytes unless EOF is encountered. It now handles
698 # signal interruptions (socket.error EINTR) internally. This code
699 # never caught that exception anyways. It seems largely pointless.
700 # self.fp.read(amt) will work fine.
Georg Brandl80ba8e82005-09-29 20:16:07 +0000701 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000702 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000703 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000704 if not chunk:
Benjamin Peterson7d49bba2009-03-02 22:41:42 +0000705 raise IncompleteRead(''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000706 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000707 amt -= len(chunk)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000708 return ''.join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000709
Senthil Kumarand389cb52010-09-21 01:38:15 +0000710 def fileno(self):
711 return self.fp.fileno()
712
Greg Steindd6eefb2000-07-18 09:09:48 +0000713 def getheader(self, name, default=None):
714 if self.msg is None:
715 raise ResponseNotReady()
716 return self.msg.getheader(name, default)
Greg Stein5e0fa402000-06-26 08:28:01 +0000717
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000718 def getheaders(self):
719 """Return list of (header, value) tuples."""
720 if self.msg is None:
721 raise ResponseNotReady()
722 return self.msg.items()
723
Greg Stein5e0fa402000-06-26 08:28:01 +0000724
725class HTTPConnection:
726
Greg Steindd6eefb2000-07-18 09:09:48 +0000727 _http_vsn = 11
728 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000729
Greg Steindd6eefb2000-07-18 09:09:48 +0000730 response_class = HTTPResponse
731 default_port = HTTP_PORT
732 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000733 debuglevel = 0
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000734 strict = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000735
Facundo Batista4f1b1ed2008-05-29 16:39:26 +0000736 def __init__(self, host, port=None, strict=None,
Gregory P. Smith9d325212010-01-03 02:06:07 +0000737 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Facundo Batista07c78be2007-03-23 18:54:07 +0000738 self.timeout = timeout
Gregory P. Smith9d325212010-01-03 02:06:07 +0000739 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000740 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000741 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000742 self.__response = None
743 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000744 self._method = None
Senthil Kumarane266f252009-05-24 09:14:50 +0000745 self._tunnel_host = None
746 self._tunnel_port = None
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000747 self._tunnel_headers = {}
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000748 if strict is not None:
749 self.strict = strict
Greg Stein5e0fa402000-06-26 08:28:01 +0000750
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700751 (self.host, self.port) = self._get_hostport(host, port)
752
753 # This is stored as an instance variable to allow unittests
754 # to replace with a suitable mock
755 self._create_connection = socket.create_connection
756
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000757 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700758 """ Set up host and port for HTTP CONNECT tunnelling.
759
760 In a connection that uses HTTP Connect tunneling, the host passed to the
761 constructor is used as proxy server that relays all communication to the
762 endpoint passed to set_tunnel. This is done by sending a HTTP CONNECT
763 request to the proxy server when the connection is established.
764
Senthil Kumaran812b9752015-01-24 12:58:10 -0800765 This method must be called before the HTTP connection has been
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700766 established.
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000767
768 The headers argument should be a mapping of extra HTTP headers
769 to send with the CONNECT request.
770 """
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700771 # Verify if this is required.
772 if self.sock:
773 raise RuntimeError("Can't setup tunnel for established connection.")
774
Serhiy Storchaka9d1de8a2015-05-28 22:37:13 +0300775 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000776 if headers:
777 self._tunnel_headers = headers
778 else:
779 self._tunnel_headers.clear()
Senthil Kumarane266f252009-05-24 09:14:50 +0000780
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700781 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000782 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000783 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000784 j = host.rfind(']') # ipv6 addresses have [...]
785 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000786 try:
787 port = int(host[i+1:])
788 except ValueError:
Łukasz Langa7a153902011-10-18 17:16:00 +0200789 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
790 port = self.default_port
791 else:
792 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000793 host = host[:i]
794 else:
795 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000796 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000797 host = host[1:-1]
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700798 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000799
Jeremy Hylton30f86742000-09-18 22:50:38 +0000800 def set_debuglevel(self, level):
801 self.debuglevel = level
802
Senthil Kumarane266f252009-05-24 09:14:50 +0000803 def _tunnel(self):
Serhiy Storchaka9d1de8a2015-05-28 22:37:13 +0300804 self.send("CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host,
805 self._tunnel_port))
Senthil Kumaran7713acf2009-12-20 06:05:13 +0000806 for header, value in self._tunnel_headers.iteritems():
807 self.send("%s: %s\r\n" % (header, value))
808 self.send("\r\n")
Senthil Kumarane266f252009-05-24 09:14:50 +0000809 response = self.response_class(self.sock, strict = self.strict,
810 method = self._method)
811 (version, code, message) = response._read_status()
812
Martin Panterb75a0e92015-09-07 01:18:47 +0000813 if version == "HTTP/0.9":
814 # HTTP/0.9 doesn't support the CONNECT verb, so if httplib has
815 # concluded HTTP/0.9 is being used something has gone wrong.
816 self.close()
817 raise socket.error("Invalid response from tunnel request")
Senthil Kumarane266f252009-05-24 09:14:50 +0000818 if code != 200:
819 self.close()
Antoine Pitrou92331d52009-10-14 19:04:48 +0000820 raise socket.error("Tunnel connection failed: %d %s" % (code,
821 message.strip()))
Senthil Kumarane266f252009-05-24 09:14:50 +0000822 while True:
Antoine Pitroud7b6ac62010-12-18 18:18:21 +0000823 line = response.fp.readline(_MAXLINE + 1)
824 if len(line) > _MAXLINE:
825 raise LineTooLong("header line")
Senthil Kumaran02671852012-04-23 23:46:46 +0800826 if not line:
827 # for sites which EOF without sending trailer
828 break
829 if line == '\r\n':
830 break
Senthil Kumarane266f252009-05-24 09:14:50 +0000831
832
Greg Steindd6eefb2000-07-18 09:09:48 +0000833 def connect(self):
834 """Connect to the host and port specified in __init__."""
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700835 self.sock = self._create_connection((self.host,self.port),
836 self.timeout, self.source_address)
Greg Stein5e0fa402000-06-26 08:28:01 +0000837
Senthil Kumarane266f252009-05-24 09:14:50 +0000838 if self._tunnel_host:
839 self._tunnel()
840
Greg Steindd6eefb2000-07-18 09:09:48 +0000841 def close(self):
842 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000843 self.__state = _CS_IDLE
Serhiy Storchaka1aa2c0f2015-04-10 13:24:10 +0300844 try:
845 sock = self.sock
846 if sock:
847 self.sock = None
848 sock.close() # close it manually... there may be other refs
849 finally:
850 response = self.__response
851 if response:
852 self.__response = None
853 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000854
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000855 def send(self, data):
856 """Send `data' to the server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000857 if self.sock is None:
858 if self.auto_open:
859 self.connect()
860 else:
861 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000862
Jeremy Hylton30f86742000-09-18 22:50:38 +0000863 if self.debuglevel > 0:
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000864 print "send:", repr(data)
Victor Stinner2c6aee92010-07-24 02:46:16 +0000865 blocksize = 8192
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000866 if hasattr(data,'read') and not isinstance(data, array):
Victor Stinner2c6aee92010-07-24 02:46:16 +0000867 if self.debuglevel > 0: print "sendIng a read()able"
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000868 datablock = data.read(blocksize)
869 while datablock:
870 self.sock.sendall(datablock)
871 datablock = data.read(blocksize)
Victor Stinner2c6aee92010-07-24 02:46:16 +0000872 else:
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000873 self.sock.sendall(data)
Greg Stein5e0fa402000-06-26 08:28:01 +0000874
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000875 def _output(self, s):
876 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000877
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000878 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000879 """
880 self._buffer.append(s)
881
Jeremy Hylton0381f482008-11-29 01:09:35 +0000882 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000883 """Send the currently buffered request and clear the buffer.
884
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000885 Appends an extra \\r\\n to the buffer.
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000886 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000887 """
888 self._buffer.extend(("", ""))
889 msg = "\r\n".join(self._buffer)
890 del self._buffer[:]
Jeremy Hylton0381f482008-11-29 01:09:35 +0000891 # If msg and message_body are sent in a single send() call,
892 # it will avoid performance problems caused by the interaction
Ezio Melotti24b07bc2011-03-15 18:55:01 +0200893 # between delayed ack and the Nagle algorithm.
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000894 if isinstance(message_body, str):
Jeremy Hylton0381f482008-11-29 01:09:35 +0000895 msg += message_body
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000896 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000897 self.send(msg)
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +0000898 if message_body is not None:
899 #message_body was not a string (i.e. it is a file) and
900 #we must run the risk of Nagle
901 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000902
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000903 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000904 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000905
Greg Steindd6eefb2000-07-18 09:09:48 +0000906 `method' specifies an HTTP request method, e.g. 'GET'.
907 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000908 `skip_host' if True does not add automatically a 'Host:' header
909 `skip_accept_encoding' if True does not add automatically an
910 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000911 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000912
Greg Stein616a58d2003-06-24 06:35:19 +0000913 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000914 if self.__response and self.__response.isclosed():
915 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000916
Tim Peters58eb11c2004-01-18 20:29:55 +0000917
Greg Steindd6eefb2000-07-18 09:09:48 +0000918 # in certain cases, we cannot issue another request on this connection.
919 # this occurs when:
920 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
921 # 2) a response to a previous request has signalled that it is going
922 # to close the connection upon completion.
923 # 3) the headers for the previous response have not been read, thus
924 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
925 #
926 # if there is no prior response, then we can request at will.
927 #
928 # if point (2) is true, then we will have passed the socket to the
929 # response (effectively meaning, "there is no prior response"), and
930 # will open a new one when a new request is made.
931 #
932 # Note: if a prior response exists, then we *can* start a new request.
933 # We are not allowed to begin fetching the response to this new
934 # request, however, until that prior response is complete.
935 #
936 if self.__state == _CS_IDLE:
937 self.__state = _CS_REQ_STARTED
938 else:
939 raise CannotSendRequest()
Greg Stein5e0fa402000-06-26 08:28:01 +0000940
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000941 # Save the method we use, we need it later in the response phase
942 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000943 if not url:
944 url = '/'
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000945 hdr = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000946
Senthil Kumaran452b0ce2010-10-02 10:43:45 +0000947 self._output(hdr)
Greg Stein5e0fa402000-06-26 08:28:01 +0000948
Greg Steindd6eefb2000-07-18 09:09:48 +0000949 if self._http_vsn == 11:
950 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000951
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000952 if not skip_host:
953 # this header is issued *only* for HTTP/1.1
954 # connections. more specifically, this means it is
955 # only issued when the client uses the new
956 # HTTPConnection() class. backwards-compat clients
957 # will be using HTTP/1.0 and those clients may be
958 # issuing this header themselves. we should NOT issue
959 # it twice; some web servers (such as Apache) barf
960 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000961
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000962 # If we need a non-standard port,include it in the
963 # header. If the request is going through a proxy,
964 # but the host of the actual URL, not the host of the
965 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000966
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000967 netloc = ''
968 if url.startswith('http'):
969 nil, netloc, nil, nil, nil = urlsplit(url)
970
971 if netloc:
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000972 try:
973 netloc_enc = netloc.encode("ascii")
974 except UnicodeEncodeError:
975 netloc_enc = netloc.encode("idna")
976 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000977 else:
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700978 if self._tunnel_host:
979 host = self._tunnel_host
980 port = self._tunnel_port
981 else:
982 host = self.host
983 port = self.port
984
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000985 try:
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700986 host_enc = host.encode("ascii")
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000987 except UnicodeEncodeError:
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700988 host_enc = host.encode("idna")
Senthil Kumaran501bfd82010-11-14 03:31:52 +0000989 # Wrap the IPv6 Host Header with [] (RFC 2732)
990 if host_enc.find(':') >= 0:
991 host_enc = "[" + host_enc + "]"
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700992 if port == self.default_port:
Georg Brandla2ac2ef2006-05-03 18:03:22 +0000993 self.putheader('Host', host_enc)
994 else:
Senthil Kumaran36f28f72014-05-16 18:51:46 -0700995 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000996
Greg Steindd6eefb2000-07-18 09:09:48 +0000997 # note: we are assuming that clients will not attempt to set these
998 # headers since *this* library must deal with the
999 # consequences. this also means that when the supporting
1000 # libraries are updated to recognize other forms, then this
1001 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001002
Greg Steindd6eefb2000-07-18 09:09:48 +00001003 # we only want a Content-Encoding of "identity" since we don't
1004 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001005 if not skip_accept_encoding:
1006 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001007
Greg Steindd6eefb2000-07-18 09:09:48 +00001008 # we can accept "chunked" Transfer-Encodings, but no others
1009 # NOTE: no TE header implies *only* "chunked"
1010 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001011
Greg Steindd6eefb2000-07-18 09:09:48 +00001012 # if TE is supplied in the header, then it must appear in a
1013 # Connection header.
1014 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001015
Greg Steindd6eefb2000-07-18 09:09:48 +00001016 else:
1017 # For HTTP/1.0, the server will assume "not chunked"
1018 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001019
Brett Cannone3d0bf72008-11-15 22:40:44 +00001020 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001021 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001022
Greg Steindd6eefb2000-07-18 09:09:48 +00001023 For example: h.putheader('Accept', 'text/html')
1024 """
1025 if self.__state != _CS_REQ_STARTED:
1026 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001027
Serhiy Storchaka59bdf632015-03-12 11:12:51 +02001028 header = '%s' % header
1029 if not _is_legal_header_name(header):
1030 raise ValueError('Invalid header name %r' % (header,))
1031
1032 values = [str(v) for v in values]
1033 for one_value in values:
1034 if _is_illegal_header_value(one_value):
1035 raise ValueError('Invalid header value %r' % (one_value,))
1036
1037 hdr = '%s: %s' % (header, '\r\n\t'.join(values))
Senthil Kumaran452b0ce2010-10-02 10:43:45 +00001038 self._output(hdr)
Greg Stein5e0fa402000-06-26 08:28:01 +00001039
Jeremy Hylton0381f482008-11-29 01:09:35 +00001040 def endheaders(self, message_body=None):
1041 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001042
Jeremy Hylton0381f482008-11-29 01:09:35 +00001043 This method sends the request to the server. The optional
Senthil Kumaran41d663f2011-10-05 23:52:49 +08001044 message_body argument can be used to pass a message body
Jeremy Hylton0381f482008-11-29 01:09:35 +00001045 associated with the request. The message body will be sent in
Senthil Kumaran41d663f2011-10-05 23:52:49 +08001046 the same packet as the message headers if it is string, otherwise it is
1047 sent as a separate packet.
Jeremy Hylton0381f482008-11-29 01:09:35 +00001048 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001049 if self.__state == _CS_REQ_STARTED:
1050 self.__state = _CS_REQ_SENT
1051 else:
1052 raise CannotSendHeader()
Jeremy Hylton0381f482008-11-29 01:09:35 +00001053 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001054
Greg Steindd6eefb2000-07-18 09:09:48 +00001055 def request(self, method, url, body=None, headers={}):
1056 """Send a complete request to the server."""
Victor Stinner2c6aee92010-07-24 02:46:16 +00001057 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +00001058
R David Murrayb4b000f2015-03-22 15:15:44 -04001059 def _set_content_length(self, body, method):
1060 # Set the content-length based on the body. If the body is "empty", we
1061 # set Content-Length: 0 for methods that expect a body (RFC 7230,
1062 # Section 3.3.2). If the body is set for other methods, we set the
1063 # header provided we can figure out what the length is.
Jeremy Hylton0381f482008-11-29 01:09:35 +00001064 thelen = None
R David Murrayb4b000f2015-03-22 15:15:44 -04001065 if body is None and method.upper() in _METHODS_EXPECTING_BODY:
1066 thelen = '0'
1067 elif body is not None:
Jeremy Hylton0381f482008-11-29 01:09:35 +00001068 try:
R David Murrayb4b000f2015-03-22 15:15:44 -04001069 thelen = str(len(body))
Serhiy Storchaka80573bb2015-05-16 18:58:41 +03001070 except (TypeError, AttributeError):
R David Murrayb4b000f2015-03-22 15:15:44 -04001071 # If this is a file-like object, try to
1072 # fstat its file descriptor
1073 try:
1074 thelen = str(os.fstat(body.fileno()).st_size)
1075 except (AttributeError, OSError):
1076 # Don't send a length if this failed
1077 if self.debuglevel > 0: print "Cannot stat!!"
Jeremy Hylton0381f482008-11-29 01:09:35 +00001078
1079 if thelen is not None:
1080 self.putheader('Content-Length', thelen)
1081
Greg Steindd6eefb2000-07-18 09:09:48 +00001082 def _send_request(self, method, url, body, headers):
Victor Stinner2c6aee92010-07-24 02:46:16 +00001083 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +00001084 header_names = dict.fromkeys([k.lower() for k in headers])
1085 skips = {}
1086 if 'host' in header_names:
1087 skips['skip_host'] = 1
1088 if 'accept-encoding' in header_names:
1089 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001090
Jeremy Hylton2c178252004-08-07 16:28:14 +00001091 self.putrequest(method, url, **skips)
1092
R David Murrayb4b000f2015-03-22 15:15:44 -04001093 if 'content-length' not in header_names:
1094 self._set_content_length(body, method)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +00001095 for hdr, value in headers.iteritems():
Greg Steindd6eefb2000-07-18 09:09:48 +00001096 self.putheader(hdr, value)
Kristján Valur Jónsson7e876f52009-01-09 20:23:16 +00001097 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001098
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001099 def getresponse(self, buffering=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001100 "Get the response from the server."
Greg Stein5e0fa402000-06-26 08:28:01 +00001101
Greg Stein616a58d2003-06-24 06:35:19 +00001102 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001103 if self.__response and self.__response.isclosed():
1104 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001105
Greg Steindd6eefb2000-07-18 09:09:48 +00001106 #
1107 # if a prior response exists, then it must be completed (otherwise, we
1108 # cannot read this response's header to determine the connection-close
1109 # behavior)
1110 #
1111 # note: if a prior response existed, but was connection-close, then the
1112 # socket and response were made independent of this HTTPConnection
1113 # object since a new request requires that we open a whole new
1114 # connection
1115 #
1116 # this means the prior response had one of two states:
1117 # 1) will_close: this connection was reset and the prior socket and
1118 # response operate independently
1119 # 2) persistent: the response was retained and we await its
1120 # isclosed() status to become true.
1121 #
1122 if self.__state != _CS_REQ_SENT or self.__response:
1123 raise ResponseNotReady()
Greg Stein5e0fa402000-06-26 08:28:01 +00001124
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001125 args = (self.sock,)
1126 kwds = {"strict":self.strict, "method":self._method}
Jeremy Hylton30f86742000-09-18 22:50:38 +00001127 if self.debuglevel > 0:
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001128 args += (self.debuglevel,)
1129 if buffering:
1130 #only add this keyword if non-default, for compatibility with
1131 #other response_classes.
1132 kwds["buffering"] = True;
1133 response = self.response_class(*args, **kwds)
Greg Stein5e0fa402000-06-26 08:28:01 +00001134
Serhiy Storchakad862db02014-12-01 13:07:28 +02001135 try:
1136 response.begin()
1137 assert response.will_close != _UNKNOWN
1138 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001139
Serhiy Storchakad862db02014-12-01 13:07:28 +02001140 if response.will_close:
1141 # this effectively passes the connection to the response
1142 self.close()
1143 else:
1144 # remember this, so we can tell when it is complete
1145 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001146
Serhiy Storchakad862db02014-12-01 13:07:28 +02001147 return response
1148 except:
1149 response.close()
1150 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001151
Greg Stein5e0fa402000-06-26 08:28:01 +00001152
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001153class HTTP:
Greg Steindd6eefb2000-07-18 09:09:48 +00001154 "Compatibility class with httplib.py from 1.5."
Greg Stein5e0fa402000-06-26 08:28:01 +00001155
Greg Steindd6eefb2000-07-18 09:09:48 +00001156 _http_vsn = 10
1157 _http_vsn_str = 'HTTP/1.0'
Greg Stein5e0fa402000-06-26 08:28:01 +00001158
Greg Steindd6eefb2000-07-18 09:09:48 +00001159 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +00001160
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001161 _connection_class = HTTPConnection
1162
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001163 def __init__(self, host='', port=None, strict=None):
Greg Steindd6eefb2000-07-18 09:09:48 +00001164 "Provide a default host, since the superclass requires one."
Greg Stein5e0fa402000-06-26 08:28:01 +00001165
Greg Steindd6eefb2000-07-18 09:09:48 +00001166 # some joker passed 0 explicitly, meaning default port
1167 if port == 0:
1168 port = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001169
Andrew Svetlov4bb142b2012-12-18 21:27:37 +02001170 # Note that we may pass an empty string as the host; this will raise
Greg Steindd6eefb2000-07-18 09:09:48 +00001171 # an error when we attempt to connect. Presumably, the client code
1172 # will call connect before then, with a proper host.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001173 self._setup(self._connection_class(host, port, strict))
Greg Stein5e0fa402000-06-26 08:28:01 +00001174
Greg Stein81937a42001-08-18 09:20:23 +00001175 def _setup(self, conn):
1176 self._conn = conn
1177
1178 # set up delegation to flesh out interface
1179 self.send = conn.send
1180 self.putrequest = conn.putrequest
Brett Cannone3d0bf72008-11-15 22:40:44 +00001181 self.putheader = conn.putheader
Greg Stein81937a42001-08-18 09:20:23 +00001182 self.endheaders = conn.endheaders
1183 self.set_debuglevel = conn.set_debuglevel
1184
1185 conn._http_vsn = self._http_vsn
1186 conn._http_vsn_str = self._http_vsn_str
Greg Stein5e0fa402000-06-26 08:28:01 +00001187
Greg Steindd6eefb2000-07-18 09:09:48 +00001188 self.file = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001189
Greg Steindd6eefb2000-07-18 09:09:48 +00001190 def connect(self, host=None, port=None):
1191 "Accept arguments to set the host/port, since the superclass doesn't."
Greg Stein5e0fa402000-06-26 08:28:01 +00001192
Greg Steindd6eefb2000-07-18 09:09:48 +00001193 if host is not None:
Senthil Kumaran812b9752015-01-24 12:58:10 -08001194 (self._conn.host, self._conn.port) = self._conn._get_hostport(host, port)
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001195 self._conn.connect()
Greg Stein5e0fa402000-06-26 08:28:01 +00001196
Greg Steindd6eefb2000-07-18 09:09:48 +00001197 def getfile(self):
1198 "Provide a getfile, since the superclass' does not use this concept."
1199 return self.file
Greg Stein5e0fa402000-06-26 08:28:01 +00001200
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001201 def getreply(self, buffering=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001202 """Compat definition since superclass does not define it.
Greg Stein5e0fa402000-06-26 08:28:01 +00001203
Greg Steindd6eefb2000-07-18 09:09:48 +00001204 Returns a tuple consisting of:
1205 - server status code (e.g. '200' if all goes well)
1206 - server "reason" corresponding to status code
1207 - any RFC822 headers in the response from the server
1208 """
1209 try:
Kristján Valur Jónsson3c43fcb2009-01-11 16:23:37 +00001210 if not buffering:
1211 response = self._conn.getresponse()
1212 else:
1213 #only add this keyword if non-default for compatibility
1214 #with other connection classes
1215 response = self._conn.getresponse(buffering)
Greg Steindd6eefb2000-07-18 09:09:48 +00001216 except BadStatusLine, e:
1217 ### hmm. if getresponse() ever closes the socket on a bad request,
1218 ### then we are going to have problems with self.sock
Greg Stein5e0fa402000-06-26 08:28:01 +00001219
Greg Steindd6eefb2000-07-18 09:09:48 +00001220 ### should we keep this behavior? do people use it?
1221 # keep the socket open (as a file), and return it
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001222 self.file = self._conn.sock.makefile('rb', 0)
Greg Stein5e0fa402000-06-26 08:28:01 +00001223
Greg Steindd6eefb2000-07-18 09:09:48 +00001224 # close our socket -- we want to restart after any protocol error
1225 self.close()
Greg Stein5e0fa402000-06-26 08:28:01 +00001226
Greg Steindd6eefb2000-07-18 09:09:48 +00001227 self.headers = None
1228 return -1, e.line, None
Greg Stein5e0fa402000-06-26 08:28:01 +00001229
Greg Steindd6eefb2000-07-18 09:09:48 +00001230 self.headers = response.msg
1231 self.file = response.fp
1232 return response.status, response.reason, response.msg
Greg Stein5e0fa402000-06-26 08:28:01 +00001233
Greg Steindd6eefb2000-07-18 09:09:48 +00001234 def close(self):
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001235 self._conn.close()
Greg Stein5e0fa402000-06-26 08:28:01 +00001236
Greg Steindd6eefb2000-07-18 09:09:48 +00001237 # note that self.file == response.fp, which gets closed by the
1238 # superclass. just clear the object ref here.
1239 ### hmm. messy. if status==-1, then self.file is owned by us.
1240 ### well... we aren't explicitly closing, but losing this ref will
1241 ### do it
1242 self.file = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001243
Bill Janssen426ea0a2007-08-29 22:35:05 +00001244try:
1245 import ssl
1246except ImportError:
1247 pass
1248else:
1249 class HTTPSConnection(HTTPConnection):
1250 "This class allows communication via SSL."
1251
1252 default_port = HTTPS_PORT
1253
1254 def __init__(self, host, port=None, key_file=None, cert_file=None,
Gregory P. Smith9d325212010-01-03 02:06:07 +00001255 strict=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Benjamin Peterson227f6e02014-12-07 13:41:26 -05001256 source_address=None, context=None):
Gregory P. Smith9d325212010-01-03 02:06:07 +00001257 HTTPConnection.__init__(self, host, port, strict, timeout,
1258 source_address)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001259 self.key_file = key_file
1260 self.cert_file = cert_file
Benjamin Petersonfcfb18e2014-11-23 11:42:45 -06001261 if context is None:
Benjamin Petersone3e7d402014-11-23 21:02:02 -06001262 context = ssl._create_default_https_context()
Benjamin Petersonfcfb18e2014-11-23 11:42:45 -06001263 if key_file or cert_file:
1264 context.load_cert_chain(cert_file, key_file)
1265 self._context = context
Bill Janssen426ea0a2007-08-29 22:35:05 +00001266
1267 def connect(self):
1268 "Connect to a host on a given (SSL) port."
1269
Benjamin Petersonfcfb18e2014-11-23 11:42:45 -06001270 HTTPConnection.connect(self)
1271
Senthil Kumarane266f252009-05-24 09:14:50 +00001272 if self._tunnel_host:
Benjamin Petersonfcfb18e2014-11-23 11:42:45 -06001273 server_hostname = self._tunnel_host
1274 else:
1275 server_hostname = self.host
Benjamin Petersonfcfb18e2014-11-23 11:42:45 -06001276
1277 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson31aa69e2014-11-23 20:13:31 -06001278 server_hostname=server_hostname)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001279
Thomas Woutersa6900e82007-08-30 21:54:39 +00001280 __all__.append("HTTPSConnection")
Bill Janssen426ea0a2007-08-29 22:35:05 +00001281
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001282 class HTTPS(HTTP):
1283 """Compatibility with 1.5 httplib interface
1284
1285 Python 1.5.2 did not have an HTTPS class, but it defined an
1286 interface for sending http requests that is also useful for
Tim Peters5ceadc82001-01-13 19:16:21 +00001287 https.
Jeremy Hylton29b8d5a2000-08-01 17:33:32 +00001288 """
1289
Martin v. Löwisd7bf9742000-09-21 22:09:47 +00001290 _connection_class = HTTPSConnection
Tim Peters5ceadc82001-01-13 19:16:21 +00001291
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001292 def __init__(self, host='', port=None, key_file=None, cert_file=None,
Benjamin Petersonb2064732014-11-23 20:55:24 -06001293 strict=None, context=None):
Greg Stein81937a42001-08-18 09:20:23 +00001294 # provide a default host, pass the X509 cert info
1295
1296 # urf. compensate for bad input.
1297 if port == 0:
1298 port = None
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001299 self._setup(self._connection_class(host, port, key_file,
Benjamin Petersonb2064732014-11-23 20:55:24 -06001300 cert_file, strict,
1301 context=context))
Greg Stein81937a42001-08-18 09:20:23 +00001302
1303 # we never actually use these for anything, but we keep them
1304 # here for compatibility with post-1.5.2 CVS.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001305 self.key_file = key_file
1306 self.cert_file = cert_file
Greg Stein81937a42001-08-18 09:20:23 +00001307
Greg Stein5e0fa402000-06-26 08:28:01 +00001308
Bill Janssen426ea0a2007-08-29 22:35:05 +00001309 def FakeSocket (sock, sslobj):
Bill Janssenc4592642007-08-31 19:02:23 +00001310 warnings.warn("FakeSocket is deprecated, and won't be in 3.x. " +
Bill Janssen98d19da2007-09-10 21:51:02 +00001311 "Use the result of ssl.wrap_socket() directly instead.",
Bill Janssenc4592642007-08-31 19:02:23 +00001312 DeprecationWarning, stacklevel=2)
Bill Janssen426ea0a2007-08-29 22:35:05 +00001313 return sslobj
1314
1315
Greg Stein5e0fa402000-06-26 08:28:01 +00001316class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001317 # Subclasses that define an __init__ must call Exception.__init__
1318 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001319 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001320
1321class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001322 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001323
Skip Montanaro9d389972002-03-24 16:53:50 +00001324class InvalidURL(HTTPException):
1325 pass
1326
Greg Stein5e0fa402000-06-26 08:28:01 +00001327class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001328 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001329 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001330 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001331
1332class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001333 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001334
Greg Stein5e0fa402000-06-26 08:28:01 +00001335class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001336 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001337
1338class IncompleteRead(HTTPException):
Benjamin Peterson7d49bba2009-03-02 22:41:42 +00001339 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001340 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001341 self.partial = partial
Benjamin Peterson7d49bba2009-03-02 22:41:42 +00001342 self.expected = expected
1343 def __repr__(self):
1344 if self.expected is not None:
1345 e = ', %i more expected' % self.expected
1346 else:
1347 e = ''
1348 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1349 def __str__(self):
1350 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001351
1352class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001353 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001354
1355class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001356 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001357
1358class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001359 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001360
1361class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001362 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001363
1364class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001365 def __init__(self, line):
Dirkjan Ochtmanebc73dc2010-02-24 04:49:00 +00001366 if not line:
1367 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001368 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001369 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001370
Antoine Pitroud7b6ac62010-12-18 18:18:21 +00001371class LineTooLong(HTTPException):
1372 def __init__(self, line_type):
1373 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1374 % (_MAXLINE, line_type))
1375
Greg Stein5e0fa402000-06-26 08:28:01 +00001376# for backwards compatibility
1377error = HTTPException
1378
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001379class LineAndFileWrapper:
1380 """A limited file-like object for HTTP/0.9 responses."""
1381
1382 # The status-line parsing code calls readline(), which normally
1383 # get the HTTP status line. For a 0.9 response, however, this is
1384 # actually the first line of the body! Clients need to get a
1385 # readable file object that contains that line.
1386
1387 def __init__(self, line, file):
1388 self._line = line
1389 self._file = file
1390 self._line_consumed = 0
1391 self._line_offset = 0
1392 self._line_left = len(line)
1393
1394 def __getattr__(self, attr):
1395 return getattr(self._file, attr)
1396
1397 def _done(self):
1398 # called when the last byte is read from the line. After the
1399 # call, all read methods are delegated to the underlying file
Skip Montanaro74b9a7a2003-02-25 17:48:15 +00001400 # object.
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001401 self._line_consumed = 1
1402 self.read = self._file.read
1403 self.readline = self._file.readline
1404 self.readlines = self._file.readlines
1405
1406 def read(self, amt=None):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001407 if self._line_consumed:
1408 return self._file.read(amt)
1409 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001410 if amt is None or amt > self._line_left:
1411 s = self._line[self._line_offset:]
1412 self._done()
1413 if amt is None:
1414 return s + self._file.read()
1415 else:
Tim Petersc411dba2002-07-16 21:35:23 +00001416 return s + self._file.read(amt - len(s))
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001417 else:
1418 assert amt <= self._line_left
1419 i = self._line_offset
1420 j = i + amt
1421 s = self._line[i:j]
1422 self._line_offset = j
1423 self._line_left -= amt
1424 if self._line_left == 0:
1425 self._done()
1426 return s
Tim Petersc411dba2002-07-16 21:35:23 +00001427
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001428 def readline(self):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001429 if self._line_consumed:
1430 return self._file.readline()
1431 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001432 s = self._line[self._line_offset:]
1433 self._done()
1434 return s
1435
1436 def readlines(self, size=None):
Hye-Shik Chang39aef792004-06-05 13:30:56 +00001437 if self._line_consumed:
1438 return self._file.readlines(size)
1439 assert self._line_left
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001440 L = [self._line[self._line_offset:]]
1441 self._done()
1442 if size is None:
1443 return L + self._file.readlines()
1444 else:
1445 return L + self._file.readlines(size)