blob: d226f63ac4ad52f40d59aa3af66690889e1d0e1f [file] [log] [blame]
Greg Stein5e0fa402000-06-26 08:28:01 +00001"""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
Thomas Wouters0e3f5912006-08-11 14:57:12 +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
Barry Warsaw820c1202008-06-12 04:06:45 +000069import email.parser
70import email.message
Jeremy Hylton636950f2009-03-28 04:34:21 +000071import io
72import os
73import socket
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000074import collections
Jeremy Hylton1afc1692008-06-18 20:49:58 +000075from urllib.parse import urlsplit
Thomas Wouters89d996e2007-09-08 17:39:28 +000076import warnings
Guido van Rossum23acc951994-02-21 16:36:04 +000077
Thomas Wouters47b49bf2007-08-30 22:15:33 +000078__all__ = ["HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000079 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000080 "UnknownTransferEncoding", "UnimplementedFileMode",
81 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
82 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
Georg Brandl6aab16e2006-02-17 19:17:25 +000083 "BadStatusLine", "error", "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000084
Guido van Rossum23acc951994-02-21 16:36:04 +000085HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000086HTTPS_PORT = 443
87
Greg Stein5e0fa402000-06-26 08:28:01 +000088_UNKNOWN = 'UNKNOWN'
89
90# connection states
91_CS_IDLE = 'Idle'
92_CS_REQ_STARTED = 'Request-started'
93_CS_REQ_SENT = 'Request-sent'
94
Martin v. Löwis39a31782004-09-18 09:03:49 +000095# status codes
96# informational
97CONTINUE = 100
98SWITCHING_PROTOCOLS = 101
99PROCESSING = 102
100
101# successful
102OK = 200
103CREATED = 201
104ACCEPTED = 202
105NON_AUTHORITATIVE_INFORMATION = 203
106NO_CONTENT = 204
107RESET_CONTENT = 205
108PARTIAL_CONTENT = 206
109MULTI_STATUS = 207
110IM_USED = 226
111
112# redirection
113MULTIPLE_CHOICES = 300
114MOVED_PERMANENTLY = 301
115FOUND = 302
116SEE_OTHER = 303
117NOT_MODIFIED = 304
118USE_PROXY = 305
119TEMPORARY_REDIRECT = 307
120
121# client error
122BAD_REQUEST = 400
123UNAUTHORIZED = 401
124PAYMENT_REQUIRED = 402
125FORBIDDEN = 403
126NOT_FOUND = 404
127METHOD_NOT_ALLOWED = 405
128NOT_ACCEPTABLE = 406
129PROXY_AUTHENTICATION_REQUIRED = 407
130REQUEST_TIMEOUT = 408
131CONFLICT = 409
132GONE = 410
133LENGTH_REQUIRED = 411
134PRECONDITION_FAILED = 412
135REQUEST_ENTITY_TOO_LARGE = 413
136REQUEST_URI_TOO_LONG = 414
137UNSUPPORTED_MEDIA_TYPE = 415
138REQUESTED_RANGE_NOT_SATISFIABLE = 416
139EXPECTATION_FAILED = 417
140UNPROCESSABLE_ENTITY = 422
141LOCKED = 423
142FAILED_DEPENDENCY = 424
143UPGRADE_REQUIRED = 426
144
145# server error
146INTERNAL_SERVER_ERROR = 500
147NOT_IMPLEMENTED = 501
148BAD_GATEWAY = 502
149SERVICE_UNAVAILABLE = 503
150GATEWAY_TIMEOUT = 504
151HTTP_VERSION_NOT_SUPPORTED = 505
152INSUFFICIENT_STORAGE = 507
153NOT_EXTENDED = 510
154
Georg Brandl6aab16e2006-02-17 19:17:25 +0000155# Mapping status codes to official W3C names
156responses = {
157 100: 'Continue',
158 101: 'Switching Protocols',
159
160 200: 'OK',
161 201: 'Created',
162 202: 'Accepted',
163 203: 'Non-Authoritative Information',
164 204: 'No Content',
165 205: 'Reset Content',
166 206: 'Partial Content',
167
168 300: 'Multiple Choices',
169 301: 'Moved Permanently',
170 302: 'Found',
171 303: 'See Other',
172 304: 'Not Modified',
173 305: 'Use Proxy',
174 306: '(Unused)',
175 307: 'Temporary Redirect',
176
177 400: 'Bad Request',
178 401: 'Unauthorized',
179 402: 'Payment Required',
180 403: 'Forbidden',
181 404: 'Not Found',
182 405: 'Method Not Allowed',
183 406: 'Not Acceptable',
184 407: 'Proxy Authentication Required',
185 408: 'Request Timeout',
186 409: 'Conflict',
187 410: 'Gone',
188 411: 'Length Required',
189 412: 'Precondition Failed',
190 413: 'Request Entity Too Large',
191 414: 'Request-URI Too Long',
192 415: 'Unsupported Media Type',
193 416: 'Requested Range Not Satisfiable',
194 417: 'Expectation Failed',
195
196 500: 'Internal Server Error',
197 501: 'Not Implemented',
198 502: 'Bad Gateway',
199 503: 'Service Unavailable',
200 504: 'Gateway Timeout',
201 505: 'HTTP Version Not Supported',
202}
203
Georg Brandl80ba8e82005-09-29 20:16:07 +0000204# maximal amount of data to read at one time in _safe_read
205MAXAMOUNT = 1048576
206
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000207# maximal line length when calling readline().
208_MAXLINE = 65536
209
Barry Warsaw820c1202008-06-12 04:06:45 +0000210class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000211 # XXX The only usage of this method is in
212 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
213 # that it doesn't need to be part of the public API. The API has
214 # never been defined so this could cause backwards compatibility
215 # issues.
216
Barry Warsaw820c1202008-06-12 04:06:45 +0000217 def getallmatchingheaders(self, name):
218 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000219
Barry Warsaw820c1202008-06-12 04:06:45 +0000220 Look through the list of headers and find all lines matching a given
221 header name (and their continuation lines). A list of the lines is
222 returned, without interpretation. If the header does not occur, an
223 empty list is returned. If the header occurs multiple times, all
224 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000225
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000226 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000227 name = name.lower() + ':'
228 n = len(name)
229 lst = []
230 hit = 0
231 for line in self.keys():
232 if line[:n].lower() == name:
233 hit = 1
234 elif not line[:1].isspace():
235 hit = 0
236 if hit:
237 lst.append(line)
238 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000239
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000240def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000241 """Parses only RFC2822 headers from a file pointer.
242
243 email Parser wants to see strings rather than bytes.
244 But a TextIOWrapper around self.rfile would buffer too many bytes
245 from the stream, bytes which we later need to read as bytes.
246 So we read the correct bytes here, as bytes, for email Parser
247 to parse.
248
249 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000250 headers = []
251 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000252 line = fp.readline(_MAXLINE + 1)
253 if len(line) > _MAXLINE:
254 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000255 headers.append(line)
256 if line in (b'\r\n', b'\n', b''):
257 break
258 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000259 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000260
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000261
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000262_strict_sentinel = object()
263
264class HTTPResponse(io.RawIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000265
266 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
267
Jeremy Hylton811fc142007-08-03 13:30:02 +0000268 # The bytes from the socket object are iso-8859-1 strings.
269 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
270 # text following RFC 2047. The basic status line parsing only
271 # accepts iso-8859-1.
272
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000273 def __init__(self, sock, debuglevel=0, strict=_strict_sentinel, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000274 # If the response includes a content-length header, we need to
275 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000276 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000277 # the server times out and closes the connection. This will
278 # happen if a self.fp.read() is done (without a size) whether
279 # self.fp is buffered or not. So, no self.fp.read() by
280 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000281 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000282 self.debuglevel = debuglevel
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000283 if strict is not _strict_sentinel:
284 warnings.warn("the 'strict' argument isn't supported anymore; "
285 "http.client now always assumes HTTP/1.x compliant servers.",
286 DeprecationWarning, 2)
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000287 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000288
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000289 # The HTTPResponse object is returned via urllib. The clients
290 # of http and urllib expect different attributes for the
291 # headers. headers is used here and supports urllib. msg is
292 # provided as a backwards compatibility layer for http
293 # clients.
294
295 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000296
Greg Steindd6eefb2000-07-18 09:09:48 +0000297 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000298 self.version = _UNKNOWN # HTTP-Version
299 self.status = _UNKNOWN # Status-Code
300 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000301
Tim Peters07e99cb2001-01-14 23:47:14 +0000302 self.chunked = _UNKNOWN # is "chunked" being used?
303 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
304 self.length = _UNKNOWN # number of bytes left in response
305 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000306
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000307 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000308 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
309 if len(line) > _MAXLINE:
310 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000311 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000312 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000313 if not line:
314 # Presumably, the server closed the connection before
315 # sending a valid response.
316 raise BadStatusLine(line)
Greg Steindd6eefb2000-07-18 09:09:48 +0000317 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000318 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000319 except ValueError:
320 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000321 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000322 reason = ""
323 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000324 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000325 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000326 if not version.startswith("HTTP/"):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000327 self.close()
328 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000329
Jeremy Hylton23d40472001-04-13 14:57:08 +0000330 # The status code is a three-digit number
331 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000332 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000333 if status < 100 or status > 999:
334 raise BadStatusLine(line)
335 except ValueError:
336 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000337 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000338
Jeremy Hylton39c03802002-07-12 14:04:09 +0000339 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000340 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000341 # we've already started reading the response
342 return
343
344 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000345 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000346 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000347 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000348 break
349 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000350 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000351 skip = self.fp.readline(_MAXLINE + 1)
352 if len(skip) > _MAXLINE:
353 raise LineTooLong("header line")
354 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000355 if not skip:
356 break
357 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000358 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000359
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000360 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000361 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000362 if version in ("HTTP/1.0", "HTTP/0.9"):
363 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000364 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000365 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000366 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000367 else:
368 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000369
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000370 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000371
Jeremy Hylton30f86742000-09-18 22:50:38 +0000372 if self.debuglevel > 0:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000373 for hdr in self.headers:
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000374 print("header:", hdr, end=" ")
Greg Stein5e0fa402000-06-26 08:28:01 +0000375
Greg Steindd6eefb2000-07-18 09:09:48 +0000376 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000377 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000378 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000379 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000380 self.chunk_left = None
381 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000382 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000383
Greg Steindd6eefb2000-07-18 09:09:48 +0000384 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000385 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000386
Greg Steindd6eefb2000-07-18 09:09:48 +0000387 # do we have a Content-Length?
388 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000389 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000390 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000391
392 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000393 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000394 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000395 try:
396 self.length = int(length)
397 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000398 self.length = None
399 else:
400 if self.length < 0: # ignore nonsensical negative lengths
401 self.length = None
402 else:
403 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000404
Greg Steindd6eefb2000-07-18 09:09:48 +0000405 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000406 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000407 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000408 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000409 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000410
Greg Steindd6eefb2000-07-18 09:09:48 +0000411 # if the connection remains open, and we aren't using chunked, and
412 # a content-length was not provided, then assume that the connection
413 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000414 if (not self.will_close and
415 not self.chunked and
416 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000417 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000418
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000419 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000420 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000421 if self.version == 11:
422 # An HTTP/1.1 proxy is assumed to stay open unless
423 # explicitly closed.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000424 conn = self.headers.get("connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000425 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000426 return True
427 return False
428
Jeremy Hylton2c178252004-08-07 16:28:14 +0000429 # Some HTTP/1.0 implementations have support for persistent
430 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000431
Christian Heimes895627f2007-12-08 17:28:33 +0000432 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000433 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000434 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000435
Jeremy Hylton2c178252004-08-07 16:28:14 +0000436 # At least Akamai returns a "Connection: Keep-Alive" header,
437 # which was supposed to be sent by the client.
438 if conn and "keep-alive" in conn.lower():
439 return False
440
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000441 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000442 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000443 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000444 return False
445
446 # otherwise, assume it will close
447 return True
448
Greg Steindd6eefb2000-07-18 09:09:48 +0000449 def close(self):
450 if self.fp:
451 self.fp.close()
452 self.fp = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000453
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000454 # These implementations are for the benefit of io.BufferedReader.
455
456 # XXX This class should probably be revised to act more like
457 # the "raw stream" that BufferedReader expects.
458
459 @property
460 def closed(self):
461 return self.isclosed()
462
463 def flush(self):
464 self.fp.flush()
465
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000466 def readable(self):
467 return True
468
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000469 # End of "raw stream" methods
470
Greg Steindd6eefb2000-07-18 09:09:48 +0000471 def isclosed(self):
472 # NOTE: it is possible that we will not ever call self.close(). This
473 # case occurs when will_close is TRUE, length is None, and we
474 # read up to the last byte, but NOT past it.
475 #
476 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
477 # called, meaning self.isclosed() is meaningful.
478 return self.fp is None
479
480 def read(self, amt=None):
481 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000482 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000483
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000484 if self._method == "HEAD":
Senthil Kumaran447f82f2010-06-04 16:38:00 +0000485 self.close()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000486 return b""
487
Greg Steindd6eefb2000-07-18 09:09:48 +0000488 if self.chunked:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000489 return self._read_chunked(amt)
Tim Peters230a60c2002-11-09 05:08:07 +0000490
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000491 if amt is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000492 # unbounded read
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000493 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000494 s = self.fp.read()
495 else:
496 s = self._safe_read(self.length)
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000497 self.length = 0
Tim Peters07e99cb2001-01-14 23:47:14 +0000498 self.close() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000499 return s
500
501 if self.length is not None:
502 if amt > self.length:
503 # clip the read to the "end of response"
504 amt = self.length
Greg Steindd6eefb2000-07-18 09:09:48 +0000505
506 # we do not use _safe_read() here because this may be a .will_close
507 # connection, and the user is reading more bytes than will be provided
508 # (for example, reading in 1k chunks)
509 s = self.fp.read(amt)
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000510 if self.length is not None:
511 self.length -= len(s)
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000512 if not self.length:
513 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +0000514 return s
515
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000516 def _read_chunked(self, amt):
517 assert self.chunked != _UNKNOWN
518 chunk_left = self.chunk_left
Chris Withers9c33cbf2009-09-04 17:04:16 +0000519 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000520 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000521 if chunk_left is None:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000522 line = self.fp.readline(_MAXLINE + 1)
523 if len(line) > _MAXLINE:
524 raise LineTooLong("chunk size")
Georg Brandl95ba4692008-01-26 09:45:58 +0000525 i = line.find(b";")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000526 if i >= 0:
527 line = line[:i] # strip chunk-extensions
Christian Heimesa612dc02008-02-24 13:08:18 +0000528 try:
529 chunk_left = int(line, 16)
530 except ValueError:
531 # close the connection as protocol synchronisation is
532 # probably lost
533 self.close()
Chris Withers9c33cbf2009-09-04 17:04:16 +0000534 raise IncompleteRead(b''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000535 if chunk_left == 0:
536 break
537 if amt is None:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000538 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000539 elif amt < chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000540 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000541 self.chunk_left = chunk_left - amt
Chris Withers9c33cbf2009-09-04 17:04:16 +0000542 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000543 elif amt == chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000544 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000545 self._safe_read(2) # toss the CRLF at the end of the chunk
546 self.chunk_left = None
Chris Withers9c33cbf2009-09-04 17:04:16 +0000547 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000548 else:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000549 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000550 amt -= chunk_left
551
552 # we read the whole chunk, get another
553 self._safe_read(2) # toss the CRLF at the end of the chunk
554 chunk_left = None
555
556 # read and discard trailer up to the CRLF terminator
557 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000558 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000559 line = self.fp.readline(_MAXLINE + 1)
560 if len(line) > _MAXLINE:
561 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000562 if not line:
563 # a vanishingly small number of sites EOF without
564 # sending the trailer
565 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800566 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000567 break
568
569 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000570 self.close()
571
Chris Withers9c33cbf2009-09-04 17:04:16 +0000572 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000573
Greg Steindd6eefb2000-07-18 09:09:48 +0000574 def _safe_read(self, amt):
575 """Read the number of bytes requested, compensating for partial reads.
576
577 Normally, we have a blocking socket, but a read() can be interrupted
578 by a signal (resulting in a partial read).
579
580 Note that we cannot distinguish between EOF and an interrupt when zero
581 bytes have been read. IncompleteRead() will be raised in this
582 situation.
583
584 This function should be used when <amt> bytes "should" be present for
585 reading. If the bytes are truly not available (due to EOF), then the
586 IncompleteRead exception can be used to detect the problem.
587 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000588 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000589 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000590 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000591 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000592 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000593 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000594 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000595 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000596
Antoine Pitroub353c122009-02-11 00:39:14 +0000597 def fileno(self):
598 return self.fp.fileno()
599
Greg Steindd6eefb2000-07-18 09:09:48 +0000600 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000601 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000602 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000603 headers = self.headers.get_all(name) or default
604 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
605 return headers
606 else:
607 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000608
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000609 def getheaders(self):
610 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000611 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000612 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000613 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000614
Antoine Pitroub353c122009-02-11 00:39:14 +0000615 # We override IOBase.__iter__ so that it doesn't check for closed-ness
616
617 def __iter__(self):
618 return self
619
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000620 # For compatibility with old-style urllib responses.
621
622 def info(self):
623 return self.headers
624
625 def geturl(self):
626 return self.url
627
628 def getcode(self):
629 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000630
631class HTTPConnection:
632
Greg Steindd6eefb2000-07-18 09:09:48 +0000633 _http_vsn = 11
634 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000635
Greg Steindd6eefb2000-07-18 09:09:48 +0000636 response_class = HTTPResponse
637 default_port = HTTP_PORT
638 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000639 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000640
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000641 def __init__(self, host, port=None, strict=_strict_sentinel,
Gregory P. Smithb4066372010-01-03 03:28:29 +0000642 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000643 if strict is not _strict_sentinel:
644 warnings.warn("the 'strict' argument isn't supported anymore; "
645 "http.client now always assumes HTTP/1.x compliant servers.",
646 DeprecationWarning, 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000647 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000648 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000649 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000650 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000651 self.__response = None
652 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000653 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000654 self._tunnel_host = None
655 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000656 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000657
Greg Steindd6eefb2000-07-18 09:09:48 +0000658 self._set_hostport(host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000659
Senthil Kumaran47fff872009-12-20 07:10:31 +0000660 def set_tunnel(self, host, port=None, headers=None):
661 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
662
663 The headers argument should be a mapping of extra HTTP headers
664 to send with the CONNECT request.
665 """
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000666 self._tunnel_host = host
667 self._tunnel_port = port
Senthil Kumaran47fff872009-12-20 07:10:31 +0000668 if headers:
669 self._tunnel_headers = headers
670 else:
671 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000672
Greg Steindd6eefb2000-07-18 09:09:48 +0000673 def _set_hostport(self, host, port):
674 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000675 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000676 j = host.rfind(']') # ipv6 addresses have [...]
677 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000678 try:
679 port = int(host[i+1:])
680 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200681 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
682 port = self.default_port
683 else:
684 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000685 host = host[:i]
686 else:
687 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000688 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000689 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000690 self.host = host
691 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000692
Jeremy Hylton30f86742000-09-18 22:50:38 +0000693 def set_debuglevel(self, level):
694 self.debuglevel = level
695
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000696 def _tunnel(self):
697 self._set_hostport(self._tunnel_host, self._tunnel_port)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000698 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self.host, self.port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000699 connect_bytes = connect_str.encode("ascii")
700 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000701 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000702 header_str = "%s: %s\r\n" % (header, value)
Armin Ronacher59531282011-01-22 13:44:22 +0000703 header_bytes = header_str.encode("latin1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000704 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000705 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000706
Georg Brandlc7c199f2011-01-22 22:06:24 +0000707 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000708 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000709
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000710 if code != 200:
711 self.close()
712 raise socket.error("Tunnel connection failed: %d %s" % (code,
713 message.strip()))
714 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000715 line = response.fp.readline(_MAXLINE + 1)
716 if len(line) > _MAXLINE:
717 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800718 if not line:
719 # for sites which EOF without sending a trailer
720 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800721 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000722 break
723
Greg Steindd6eefb2000-07-18 09:09:48 +0000724 def connect(self):
725 """Connect to the host and port specified in __init__."""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000726 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +0000727 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000728 if self._tunnel_host:
729 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000730
Greg Steindd6eefb2000-07-18 09:09:48 +0000731 def close(self):
732 """Close the connection to the HTTP server."""
733 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000734 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000735 self.sock = None
736 if self.__response:
737 self.__response.close()
738 self.__response = None
739 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000740
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000741 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000742 """Send `data' to the server.
743 ``data`` can be a string object, a bytes object, an array object, a
744 file-like object that supports a .read() method, or an iterable object.
745 """
746
Greg Steindd6eefb2000-07-18 09:09:48 +0000747 if self.sock is None:
748 if self.auto_open:
749 self.connect()
750 else:
751 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000752
Jeremy Hylton30f86742000-09-18 22:50:38 +0000753 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000754 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000755 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000756 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000757 if self.debuglevel > 0:
758 print("sendIng a read()able")
759 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000760 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000761 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000762 except AttributeError:
763 # io.BytesIO and other file-like objects don't have a `mode`
764 # attribute.
765 pass
766 else:
767 if "b" not in mode:
768 encode = True
769 if self.debuglevel > 0:
770 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000771 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000772 datablock = data.read(blocksize)
773 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000774 break
775 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000776 datablock = datablock.encode("iso-8859-1")
777 self.sock.sendall(datablock)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000778
779 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000780 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000781 except TypeError:
782 if isinstance(data, collections.Iterable):
783 for d in data:
784 self.sock.sendall(d)
785 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200786 raise TypeError("data should be a bytes-like object "
787 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000788
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000789 def _output(self, s):
790 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000791
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000792 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000793 """
794 self._buffer.append(s)
795
Benjamin Peterson1742e402008-11-30 22:15:29 +0000796 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000797 """Send the currently buffered request and clear the buffer.
798
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000799 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000800 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000801 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000802 self._buffer.extend((b"", b""))
803 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000804 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000805 # If msg and message_body are sent in a single send() call,
806 # it will avoid performance problems caused by the interaction
Ezio Melotti42da6632011-03-15 05:18:48 +0200807 # between delayed ack and the Nagle algorithm.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000808 if isinstance(message_body, bytes):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000809 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000810 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000811 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000812 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000813 # message_body was not a string (i.e. it is a file), and
814 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000815 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000816
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000817 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000818 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000819
Greg Steindd6eefb2000-07-18 09:09:48 +0000820 `method' specifies an HTTP request method, e.g. 'GET'.
821 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000822 `skip_host' if True does not add automatically a 'Host:' header
823 `skip_accept_encoding' if True does not add automatically an
824 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000825 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000826
Greg Stein616a58d2003-06-24 06:35:19 +0000827 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000828 if self.__response and self.__response.isclosed():
829 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000830
Tim Peters58eb11c2004-01-18 20:29:55 +0000831
Greg Steindd6eefb2000-07-18 09:09:48 +0000832 # in certain cases, we cannot issue another request on this connection.
833 # this occurs when:
834 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
835 # 2) a response to a previous request has signalled that it is going
836 # to close the connection upon completion.
837 # 3) the headers for the previous response have not been read, thus
838 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
839 #
840 # if there is no prior response, then we can request at will.
841 #
842 # if point (2) is true, then we will have passed the socket to the
843 # response (effectively meaning, "there is no prior response"), and
844 # will open a new one when a new request is made.
845 #
846 # Note: if a prior response exists, then we *can* start a new request.
847 # We are not allowed to begin fetching the response to this new
848 # request, however, until that prior response is complete.
849 #
850 if self.__state == _CS_IDLE:
851 self.__state = _CS_REQ_STARTED
852 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000853 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +0000854
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000855 # Save the method we use, we need it later in the response phase
856 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000857 if not url:
858 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000859 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000860
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000861 # Non-ASCII characters should have been eliminated earlier
862 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +0000863
Greg Steindd6eefb2000-07-18 09:09:48 +0000864 if self._http_vsn == 11:
865 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000866
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000867 if not skip_host:
868 # this header is issued *only* for HTTP/1.1
869 # connections. more specifically, this means it is
870 # only issued when the client uses the new
871 # HTTPConnection() class. backwards-compat clients
872 # will be using HTTP/1.0 and those clients may be
873 # issuing this header themselves. we should NOT issue
874 # it twice; some web servers (such as Apache) barf
875 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000876
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000877 # If we need a non-standard port,include it in the
878 # header. If the request is going through a proxy,
879 # but the host of the actual URL, not the host of the
880 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000881
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000882 netloc = ''
883 if url.startswith('http'):
884 nil, netloc, nil, nil, nil = urlsplit(url)
885
886 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000887 try:
888 netloc_enc = netloc.encode("ascii")
889 except UnicodeEncodeError:
890 netloc_enc = netloc.encode("idna")
891 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000892 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000893 try:
894 host_enc = self.host.encode("ascii")
895 except UnicodeEncodeError:
896 host_enc = self.host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +0000897
898 # As per RFC 273, IPv6 address should be wrapped with []
899 # when used as Host header
900
901 if self.host.find(':') >= 0:
902 host_enc = b'[' + host_enc + b']'
903
Georg Brandl86b2fb92008-07-16 03:43:04 +0000904 if self.port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000905 self.putheader('Host', host_enc)
906 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000907 host_enc = host_enc.decode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000908 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000909
Greg Steindd6eefb2000-07-18 09:09:48 +0000910 # note: we are assuming that clients will not attempt to set these
911 # headers since *this* library must deal with the
912 # consequences. this also means that when the supporting
913 # libraries are updated to recognize other forms, then this
914 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000915
Greg Steindd6eefb2000-07-18 09:09:48 +0000916 # we only want a Content-Encoding of "identity" since we don't
917 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000918 if not skip_accept_encoding:
919 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000920
Greg Steindd6eefb2000-07-18 09:09:48 +0000921 # we can accept "chunked" Transfer-Encodings, but no others
922 # NOTE: no TE header implies *only* "chunked"
923 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000924
Greg Steindd6eefb2000-07-18 09:09:48 +0000925 # if TE is supplied in the header, then it must appear in a
926 # Connection header.
927 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000928
Greg Steindd6eefb2000-07-18 09:09:48 +0000929 else:
930 # For HTTP/1.0, the server will assume "not chunked"
931 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000932
Benjamin Petersonf608c612008-11-16 18:33:53 +0000933 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000934 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000935
Greg Steindd6eefb2000-07-18 09:09:48 +0000936 For example: h.putheader('Accept', 'text/html')
937 """
938 if self.__state != _CS_REQ_STARTED:
939 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000940
Guido van Rossum98297ee2007-11-06 21:34:58 +0000941 if hasattr(header, 'encode'):
942 header = header.encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000943 values = list(values)
944 for i, one_value in enumerate(values):
945 if hasattr(one_value, 'encode'):
Armin Ronacher59531282011-01-22 13:44:22 +0000946 values[i] = one_value.encode('latin1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +0000947 elif isinstance(one_value, int):
948 values[i] = str(one_value).encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000949 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000950 header = header + b': ' + value
951 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +0000952
Benjamin Peterson1742e402008-11-30 22:15:29 +0000953 def endheaders(self, message_body=None):
954 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000955
Senthil Kumaranad87fa62011-10-05 23:26:49 +0800956 This method sends the request to the server. The optional message_body
957 argument can be used to pass a message body associated with the
958 request. The message body will be sent in the same packet as the
959 message headers if it is a string, otherwise it is sent as a separate
960 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +0000961 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000962 if self.__state == _CS_REQ_STARTED:
963 self.__state = _CS_REQ_SENT
964 else:
965 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +0000966 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000967
Greg Steindd6eefb2000-07-18 09:09:48 +0000968 def request(self, method, url, body=None, headers={}):
969 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +0000970 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000971
Benjamin Peterson1742e402008-11-30 22:15:29 +0000972 def _set_content_length(self, body):
973 # Set the content-length based on the body.
974 thelen = None
975 try:
976 thelen = str(len(body))
977 except TypeError as te:
978 # If this is a file-like object, try to
979 # fstat its file descriptor
Benjamin Peterson1742e402008-11-30 22:15:29 +0000980 try:
981 thelen = str(os.fstat(body.fileno()).st_size)
982 except (AttributeError, OSError):
983 # Don't send a length if this failed
984 if self.debuglevel > 0: print("Cannot stat!!")
985
986 if thelen is not None:
987 self.putheader('Content-Length', thelen)
988
Greg Steindd6eefb2000-07-18 09:09:48 +0000989 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +0000990 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +0000991 header_names = dict.fromkeys([k.lower() for k in headers])
992 skips = {}
993 if 'host' in header_names:
994 skips['skip_host'] = 1
995 if 'accept-encoding' in header_names:
996 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +0000997
Jeremy Hylton2c178252004-08-07 16:28:14 +0000998 self.putrequest(method, url, **skips)
999
1000 if body and ('content-length' not in header_names):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001001 self._set_content_length(body)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001002 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001003 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001004 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001005 # RFC 2616 Section 3.7.1 says that text default has a
1006 # default charset of iso-8859-1.
1007 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001008 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001009
Greg Steindd6eefb2000-07-18 09:09:48 +00001010 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001011 """Get the response from the server.
1012
1013 If the HTTPConnection is in the correct state, returns an
1014 instance of HTTPResponse or of whatever object is returned by
1015 class the response_class variable.
1016
1017 If a request has not been sent or if a previous response has
1018 not be handled, ResponseNotReady is raised. If the HTTP
1019 response indicates that the connection should be closed, then
1020 it will be closed before the response is returned. When the
1021 connection is closed, the underlying socket is closed.
1022 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001023
Greg Stein616a58d2003-06-24 06:35:19 +00001024 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001025 if self.__response and self.__response.isclosed():
1026 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001027
Greg Steindd6eefb2000-07-18 09:09:48 +00001028 # if a prior response exists, then it must be completed (otherwise, we
1029 # cannot read this response's header to determine the connection-close
1030 # behavior)
1031 #
1032 # note: if a prior response existed, but was connection-close, then the
1033 # socket and response were made independent of this HTTPConnection
1034 # object since a new request requires that we open a whole new
1035 # connection
1036 #
1037 # this means the prior response had one of two states:
1038 # 1) will_close: this connection was reset and the prior socket and
1039 # response operate independently
1040 # 2) persistent: the response was retained and we await its
1041 # isclosed() status to become true.
1042 #
1043 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001044 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001045
Jeremy Hylton30f86742000-09-18 22:50:38 +00001046 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001047 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001048 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001049 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001050 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001051
Jeremy Hylton39c03802002-07-12 14:04:09 +00001052 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001053 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001054 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001055
Greg Steindd6eefb2000-07-18 09:09:48 +00001056 if response.will_close:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001057 # this effectively passes the connection to the response
1058 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001059 else:
1060 # remember this, so we can tell when it is complete
1061 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001062
Greg Steindd6eefb2000-07-18 09:09:48 +00001063 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001064
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001065try:
1066 import ssl
1067except ImportError:
1068 pass
1069else:
1070 class HTTPSConnection(HTTPConnection):
1071 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001072
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001073 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001074
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001075 # XXX Should key_file and cert_file be deprecated in favour of context?
1076
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001077 def __init__(self, host, port=None, key_file=None, cert_file=None,
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001078 strict=_strict_sentinel, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001079 source_address=None, *, context=None, check_hostname=None):
Gregory P. Smithb4066372010-01-03 03:28:29 +00001080 super(HTTPSConnection, self).__init__(host, port, strict, timeout,
1081 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001082 self.key_file = key_file
1083 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001084 if context is None:
1085 # Some reasonable defaults
1086 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
1087 context.options |= ssl.OP_NO_SSLv2
1088 will_verify = context.verify_mode != ssl.CERT_NONE
1089 if check_hostname is None:
1090 check_hostname = will_verify
1091 elif check_hostname and not will_verify:
1092 raise ValueError("check_hostname needs a SSL context with "
1093 "either CERT_OPTIONAL or CERT_REQUIRED")
1094 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001095 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001096 self._context = context
1097 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001098
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001099 def connect(self):
1100 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001101
Jeremy Hylton636950f2009-03-28 04:34:21 +00001102 sock = socket.create_connection((self.host, self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +00001103 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001104
1105 if self._tunnel_host:
1106 self.sock = sock
1107 self._tunnel()
1108
Antoine Pitroud5323212010-10-22 18:19:07 +00001109 server_hostname = self.host if ssl.HAS_SNI else None
1110 self.sock = self._context.wrap_socket(sock,
1111 server_hostname=server_hostname)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001112 try:
1113 if self._check_hostname:
1114 ssl.match_hostname(self.sock.getpeercert(), self.host)
1115 except Exception:
1116 self.sock.shutdown(socket.SHUT_RDWR)
1117 self.sock.close()
1118 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001119
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001120 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001121
Greg Stein5e0fa402000-06-26 08:28:01 +00001122class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001123 # Subclasses that define an __init__ must call Exception.__init__
1124 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001125 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001126
1127class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001128 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001129
Skip Montanaro9d389972002-03-24 16:53:50 +00001130class InvalidURL(HTTPException):
1131 pass
1132
Greg Stein5e0fa402000-06-26 08:28:01 +00001133class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001134 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001135 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001136 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001137
1138class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001139 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001140
Greg Stein5e0fa402000-06-26 08:28:01 +00001141class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001142 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001143
1144class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001145 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001146 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001147 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001148 self.expected = expected
1149 def __repr__(self):
1150 if self.expected is not None:
1151 e = ', %i more expected' % self.expected
1152 else:
1153 e = ''
1154 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1155 def __str__(self):
1156 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001157
1158class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001159 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001160
1161class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001162 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001163
1164class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001165 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001166
1167class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001168 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001169
1170class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001171 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001172 if not line:
1173 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001174 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001175 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001176
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001177class LineTooLong(HTTPException):
1178 def __init__(self, line_type):
1179 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1180 % (_MAXLINE, line_type))
1181
Greg Stein5e0fa402000-06-26 08:28:01 +00001182# for backwards compatibility
1183error = HTTPException