blob: 36528dacba7abad17b35315b71d5d3b8ff3f988f [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:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100496 try:
497 s = self._safe_read(self.length)
498 except IncompleteRead:
499 self.close()
500 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000501 self.length = 0
Tim Peters07e99cb2001-01-14 23:47:14 +0000502 self.close() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000503 return s
504
505 if self.length is not None:
506 if amt > self.length:
507 # clip the read to the "end of response"
508 amt = self.length
Greg Steindd6eefb2000-07-18 09:09:48 +0000509
510 # we do not use _safe_read() here because this may be a .will_close
511 # connection, and the user is reading more bytes than will be provided
512 # (for example, reading in 1k chunks)
513 s = self.fp.read(amt)
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100514 if not s:
515 # Ideally, we would raise IncompleteRead if the content-length
516 # wasn't satisfied, but it might break compatibility.
517 self.close()
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000518 if self.length is not None:
519 self.length -= len(s)
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000520 if not self.length:
521 self.close()
Antoine Pitrou084daa22012-12-15 19:11:54 +0100522 else:
523 if not s:
524 self.close()
525
Greg Steindd6eefb2000-07-18 09:09:48 +0000526 return s
527
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000528 def _read_chunked(self, amt):
529 assert self.chunked != _UNKNOWN
530 chunk_left = self.chunk_left
Chris Withers9c33cbf2009-09-04 17:04:16 +0000531 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000532 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000533 if chunk_left is None:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000534 line = self.fp.readline(_MAXLINE + 1)
535 if len(line) > _MAXLINE:
536 raise LineTooLong("chunk size")
Georg Brandl95ba4692008-01-26 09:45:58 +0000537 i = line.find(b";")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000538 if i >= 0:
539 line = line[:i] # strip chunk-extensions
Christian Heimesa612dc02008-02-24 13:08:18 +0000540 try:
541 chunk_left = int(line, 16)
542 except ValueError:
543 # close the connection as protocol synchronisation is
544 # probably lost
545 self.close()
Chris Withers9c33cbf2009-09-04 17:04:16 +0000546 raise IncompleteRead(b''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000547 if chunk_left == 0:
548 break
549 if amt is None:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000550 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000551 elif amt < chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000552 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000553 self.chunk_left = chunk_left - amt
Chris Withers9c33cbf2009-09-04 17:04:16 +0000554 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000555 elif amt == chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000556 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000557 self._safe_read(2) # toss the CRLF at the end of the chunk
558 self.chunk_left = None
Chris Withers9c33cbf2009-09-04 17:04:16 +0000559 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000560 else:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000561 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000562 amt -= chunk_left
563
564 # we read the whole chunk, get another
565 self._safe_read(2) # toss the CRLF at the end of the chunk
566 chunk_left = None
567
568 # read and discard trailer up to the CRLF terminator
569 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000570 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000571 line = self.fp.readline(_MAXLINE + 1)
572 if len(line) > _MAXLINE:
573 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000574 if not line:
575 # a vanishingly small number of sites EOF without
576 # sending the trailer
577 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800578 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000579 break
580
581 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000582 self.close()
583
Chris Withers9c33cbf2009-09-04 17:04:16 +0000584 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000585
Greg Steindd6eefb2000-07-18 09:09:48 +0000586 def _safe_read(self, amt):
587 """Read the number of bytes requested, compensating for partial reads.
588
589 Normally, we have a blocking socket, but a read() can be interrupted
590 by a signal (resulting in a partial read).
591
592 Note that we cannot distinguish between EOF and an interrupt when zero
593 bytes have been read. IncompleteRead() will be raised in this
594 situation.
595
596 This function should be used when <amt> bytes "should" be present for
597 reading. If the bytes are truly not available (due to EOF), then the
598 IncompleteRead exception can be used to detect the problem.
599 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000600 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000601 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000602 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000603 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000604 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000605 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000606 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000607 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000608
Antoine Pitroub353c122009-02-11 00:39:14 +0000609 def fileno(self):
610 return self.fp.fileno()
611
Greg Steindd6eefb2000-07-18 09:09:48 +0000612 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000613 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000614 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000615 headers = self.headers.get_all(name) or default
616 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
617 return headers
618 else:
619 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000620
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000621 def getheaders(self):
622 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000623 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000624 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000625 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000626
Antoine Pitroub353c122009-02-11 00:39:14 +0000627 # We override IOBase.__iter__ so that it doesn't check for closed-ness
628
629 def __iter__(self):
630 return self
631
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000632 # For compatibility with old-style urllib responses.
633
634 def info(self):
635 return self.headers
636
637 def geturl(self):
638 return self.url
639
640 def getcode(self):
641 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000642
643class HTTPConnection:
644
Greg Steindd6eefb2000-07-18 09:09:48 +0000645 _http_vsn = 11
646 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000647
Greg Steindd6eefb2000-07-18 09:09:48 +0000648 response_class = HTTPResponse
649 default_port = HTTP_PORT
650 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000651 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000652
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000653 def __init__(self, host, port=None, strict=_strict_sentinel,
Gregory P. Smithb4066372010-01-03 03:28:29 +0000654 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000655 if strict is not _strict_sentinel:
656 warnings.warn("the 'strict' argument isn't supported anymore; "
657 "http.client now always assumes HTTP/1.x compliant servers.",
658 DeprecationWarning, 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000659 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000660 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000661 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000662 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000663 self.__response = None
664 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000665 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000666 self._tunnel_host = None
667 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000668 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000669
Greg Steindd6eefb2000-07-18 09:09:48 +0000670 self._set_hostport(host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000671
Senthil Kumaran47fff872009-12-20 07:10:31 +0000672 def set_tunnel(self, host, port=None, headers=None):
673 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
674
675 The headers argument should be a mapping of extra HTTP headers
676 to send with the CONNECT request.
677 """
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000678 self._tunnel_host = host
679 self._tunnel_port = port
Senthil Kumaran47fff872009-12-20 07:10:31 +0000680 if headers:
681 self._tunnel_headers = headers
682 else:
683 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000684
Greg Steindd6eefb2000-07-18 09:09:48 +0000685 def _set_hostport(self, host, port):
686 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000687 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000688 j = host.rfind(']') # ipv6 addresses have [...]
689 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000690 try:
691 port = int(host[i+1:])
692 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200693 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
694 port = self.default_port
695 else:
696 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000697 host = host[:i]
698 else:
699 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000700 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000701 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000702 self.host = host
703 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000704
Jeremy Hylton30f86742000-09-18 22:50:38 +0000705 def set_debuglevel(self, level):
706 self.debuglevel = level
707
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000708 def _tunnel(self):
709 self._set_hostport(self._tunnel_host, self._tunnel_port)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000710 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self.host, self.port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000711 connect_bytes = connect_str.encode("ascii")
712 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000713 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000714 header_str = "%s: %s\r\n" % (header, value)
Armin Ronacher59531282011-01-22 13:44:22 +0000715 header_bytes = header_str.encode("latin1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000716 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000717 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000718
Georg Brandlc7c199f2011-01-22 22:06:24 +0000719 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000720 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000721
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000722 if code != 200:
723 self.close()
724 raise socket.error("Tunnel connection failed: %d %s" % (code,
725 message.strip()))
726 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000727 line = response.fp.readline(_MAXLINE + 1)
728 if len(line) > _MAXLINE:
729 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800730 if not line:
731 # for sites which EOF without sending a trailer
732 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800733 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000734 break
735
Greg Steindd6eefb2000-07-18 09:09:48 +0000736 def connect(self):
737 """Connect to the host and port specified in __init__."""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000738 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +0000739 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000740 if self._tunnel_host:
741 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000742
Greg Steindd6eefb2000-07-18 09:09:48 +0000743 def close(self):
744 """Close the connection to the HTTP server."""
745 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000746 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000747 self.sock = None
748 if self.__response:
749 self.__response.close()
750 self.__response = None
751 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000752
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000753 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000754 """Send `data' to the server.
755 ``data`` can be a string object, a bytes object, an array object, a
756 file-like object that supports a .read() method, or an iterable object.
757 """
758
Greg Steindd6eefb2000-07-18 09:09:48 +0000759 if self.sock is None:
760 if self.auto_open:
761 self.connect()
762 else:
763 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000764
Jeremy Hylton30f86742000-09-18 22:50:38 +0000765 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000766 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000767 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000768 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000769 if self.debuglevel > 0:
770 print("sendIng a read()able")
771 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000772 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000773 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000774 except AttributeError:
775 # io.BytesIO and other file-like objects don't have a `mode`
776 # attribute.
777 pass
778 else:
779 if "b" not in mode:
780 encode = True
781 if self.debuglevel > 0:
782 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000783 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000784 datablock = data.read(blocksize)
785 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000786 break
787 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000788 datablock = datablock.encode("iso-8859-1")
789 self.sock.sendall(datablock)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000790
791 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000792 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000793 except TypeError:
794 if isinstance(data, collections.Iterable):
795 for d in data:
796 self.sock.sendall(d)
797 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200798 raise TypeError("data should be a bytes-like object "
799 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000800
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000801 def _output(self, s):
802 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000803
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000804 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000805 """
806 self._buffer.append(s)
807
Benjamin Peterson1742e402008-11-30 22:15:29 +0000808 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000809 """Send the currently buffered request and clear the buffer.
810
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000811 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000812 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000813 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000814 self._buffer.extend((b"", b""))
815 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000816 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000817 # If msg and message_body are sent in a single send() call,
818 # it will avoid performance problems caused by the interaction
Ezio Melotti42da6632011-03-15 05:18:48 +0200819 # between delayed ack and the Nagle algorithm.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000820 if isinstance(message_body, bytes):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000821 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000822 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000823 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000824 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000825 # message_body was not a string (i.e. it is a file), and
826 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000827 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000828
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000829 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000830 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000831
Greg Steindd6eefb2000-07-18 09:09:48 +0000832 `method' specifies an HTTP request method, e.g. 'GET'.
833 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000834 `skip_host' if True does not add automatically a 'Host:' header
835 `skip_accept_encoding' if True does not add automatically an
836 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000837 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000838
Greg Stein616a58d2003-06-24 06:35:19 +0000839 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000840 if self.__response and self.__response.isclosed():
841 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000842
Tim Peters58eb11c2004-01-18 20:29:55 +0000843
Greg Steindd6eefb2000-07-18 09:09:48 +0000844 # in certain cases, we cannot issue another request on this connection.
845 # this occurs when:
846 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
847 # 2) a response to a previous request has signalled that it is going
848 # to close the connection upon completion.
849 # 3) the headers for the previous response have not been read, thus
850 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
851 #
852 # if there is no prior response, then we can request at will.
853 #
854 # if point (2) is true, then we will have passed the socket to the
855 # response (effectively meaning, "there is no prior response"), and
856 # will open a new one when a new request is made.
857 #
858 # Note: if a prior response exists, then we *can* start a new request.
859 # We are not allowed to begin fetching the response to this new
860 # request, however, until that prior response is complete.
861 #
862 if self.__state == _CS_IDLE:
863 self.__state = _CS_REQ_STARTED
864 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000865 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +0000866
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000867 # Save the method we use, we need it later in the response phase
868 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000869 if not url:
870 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000871 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000872
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000873 # Non-ASCII characters should have been eliminated earlier
874 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +0000875
Greg Steindd6eefb2000-07-18 09:09:48 +0000876 if self._http_vsn == 11:
877 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000878
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000879 if not skip_host:
880 # this header is issued *only* for HTTP/1.1
881 # connections. more specifically, this means it is
882 # only issued when the client uses the new
883 # HTTPConnection() class. backwards-compat clients
884 # will be using HTTP/1.0 and those clients may be
885 # issuing this header themselves. we should NOT issue
886 # it twice; some web servers (such as Apache) barf
887 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000888
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000889 # If we need a non-standard port,include it in the
890 # header. If the request is going through a proxy,
891 # but the host of the actual URL, not the host of the
892 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000893
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000894 netloc = ''
895 if url.startswith('http'):
896 nil, netloc, nil, nil, nil = urlsplit(url)
897
898 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000899 try:
900 netloc_enc = netloc.encode("ascii")
901 except UnicodeEncodeError:
902 netloc_enc = netloc.encode("idna")
903 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000904 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000905 try:
906 host_enc = self.host.encode("ascii")
907 except UnicodeEncodeError:
908 host_enc = self.host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +0000909
910 # As per RFC 273, IPv6 address should be wrapped with []
911 # when used as Host header
912
913 if self.host.find(':') >= 0:
914 host_enc = b'[' + host_enc + b']'
915
Georg Brandl86b2fb92008-07-16 03:43:04 +0000916 if self.port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000917 self.putheader('Host', host_enc)
918 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000919 host_enc = host_enc.decode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000920 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000921
Greg Steindd6eefb2000-07-18 09:09:48 +0000922 # note: we are assuming that clients will not attempt to set these
923 # headers since *this* library must deal with the
924 # consequences. this also means that when the supporting
925 # libraries are updated to recognize other forms, then this
926 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000927
Greg Steindd6eefb2000-07-18 09:09:48 +0000928 # we only want a Content-Encoding of "identity" since we don't
929 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000930 if not skip_accept_encoding:
931 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000932
Greg Steindd6eefb2000-07-18 09:09:48 +0000933 # we can accept "chunked" Transfer-Encodings, but no others
934 # NOTE: no TE header implies *only* "chunked"
935 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000936
Greg Steindd6eefb2000-07-18 09:09:48 +0000937 # if TE is supplied in the header, then it must appear in a
938 # Connection header.
939 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000940
Greg Steindd6eefb2000-07-18 09:09:48 +0000941 else:
942 # For HTTP/1.0, the server will assume "not chunked"
943 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000944
Benjamin Petersonf608c612008-11-16 18:33:53 +0000945 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000946 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000947
Greg Steindd6eefb2000-07-18 09:09:48 +0000948 For example: h.putheader('Accept', 'text/html')
949 """
950 if self.__state != _CS_REQ_STARTED:
951 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000952
Guido van Rossum98297ee2007-11-06 21:34:58 +0000953 if hasattr(header, 'encode'):
954 header = header.encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000955 values = list(values)
956 for i, one_value in enumerate(values):
957 if hasattr(one_value, 'encode'):
Armin Ronacher59531282011-01-22 13:44:22 +0000958 values[i] = one_value.encode('latin1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +0000959 elif isinstance(one_value, int):
960 values[i] = str(one_value).encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000961 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000962 header = header + b': ' + value
963 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +0000964
Benjamin Peterson1742e402008-11-30 22:15:29 +0000965 def endheaders(self, message_body=None):
966 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000967
Senthil Kumaranad87fa62011-10-05 23:26:49 +0800968 This method sends the request to the server. The optional message_body
969 argument can be used to pass a message body associated with the
970 request. The message body will be sent in the same packet as the
971 message headers if it is a string, otherwise it is sent as a separate
972 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +0000973 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000974 if self.__state == _CS_REQ_STARTED:
975 self.__state = _CS_REQ_SENT
976 else:
977 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +0000978 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000979
Greg Steindd6eefb2000-07-18 09:09:48 +0000980 def request(self, method, url, body=None, headers={}):
981 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +0000982 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000983
Benjamin Peterson1742e402008-11-30 22:15:29 +0000984 def _set_content_length(self, body):
985 # Set the content-length based on the body.
986 thelen = None
987 try:
988 thelen = str(len(body))
989 except TypeError as te:
990 # If this is a file-like object, try to
991 # fstat its file descriptor
Benjamin Peterson1742e402008-11-30 22:15:29 +0000992 try:
993 thelen = str(os.fstat(body.fileno()).st_size)
994 except (AttributeError, OSError):
995 # Don't send a length if this failed
996 if self.debuglevel > 0: print("Cannot stat!!")
997
998 if thelen is not None:
999 self.putheader('Content-Length', thelen)
1000
Greg Steindd6eefb2000-07-18 09:09:48 +00001001 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001002 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +00001003 header_names = dict.fromkeys([k.lower() for k in headers])
1004 skips = {}
1005 if 'host' in header_names:
1006 skips['skip_host'] = 1
1007 if 'accept-encoding' in header_names:
1008 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001009
Jeremy Hylton2c178252004-08-07 16:28:14 +00001010 self.putrequest(method, url, **skips)
1011
Senthil Kumaran5fa4a892012-05-19 16:58:09 +08001012 if body is not None and ('content-length' not in header_names):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001013 self._set_content_length(body)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001014 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001015 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001016 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001017 # RFC 2616 Section 3.7.1 says that text default has a
1018 # default charset of iso-8859-1.
1019 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001020 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001021
Greg Steindd6eefb2000-07-18 09:09:48 +00001022 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001023 """Get the response from the server.
1024
1025 If the HTTPConnection is in the correct state, returns an
1026 instance of HTTPResponse or of whatever object is returned by
1027 class the response_class variable.
1028
1029 If a request has not been sent or if a previous response has
1030 not be handled, ResponseNotReady is raised. If the HTTP
1031 response indicates that the connection should be closed, then
1032 it will be closed before the response is returned. When the
1033 connection is closed, the underlying socket is closed.
1034 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001035
Greg Stein616a58d2003-06-24 06:35:19 +00001036 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001037 if self.__response and self.__response.isclosed():
1038 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001039
Greg Steindd6eefb2000-07-18 09:09:48 +00001040 # if a prior response exists, then it must be completed (otherwise, we
1041 # cannot read this response's header to determine the connection-close
1042 # behavior)
1043 #
1044 # note: if a prior response existed, but was connection-close, then the
1045 # socket and response were made independent of this HTTPConnection
1046 # object since a new request requires that we open a whole new
1047 # connection
1048 #
1049 # this means the prior response had one of two states:
1050 # 1) will_close: this connection was reset and the prior socket and
1051 # response operate independently
1052 # 2) persistent: the response was retained and we await its
1053 # isclosed() status to become true.
1054 #
1055 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001056 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001057
Jeremy Hylton30f86742000-09-18 22:50:38 +00001058 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001059 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001060 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001061 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001062 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001063
Jeremy Hylton39c03802002-07-12 14:04:09 +00001064 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001065 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001066 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001067
Greg Steindd6eefb2000-07-18 09:09:48 +00001068 if response.will_close:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001069 # this effectively passes the connection to the response
1070 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001071 else:
1072 # remember this, so we can tell when it is complete
1073 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001074
Greg Steindd6eefb2000-07-18 09:09:48 +00001075 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001076
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001077try:
1078 import ssl
1079except ImportError:
1080 pass
1081else:
1082 class HTTPSConnection(HTTPConnection):
1083 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001084
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001085 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001086
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001087 # XXX Should key_file and cert_file be deprecated in favour of context?
1088
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001089 def __init__(self, host, port=None, key_file=None, cert_file=None,
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001090 strict=_strict_sentinel, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001091 source_address=None, *, context=None, check_hostname=None):
Gregory P. Smithb4066372010-01-03 03:28:29 +00001092 super(HTTPSConnection, self).__init__(host, port, strict, timeout,
1093 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001094 self.key_file = key_file
1095 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001096 if context is None:
1097 # Some reasonable defaults
1098 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
1099 context.options |= ssl.OP_NO_SSLv2
1100 will_verify = context.verify_mode != ssl.CERT_NONE
1101 if check_hostname is None:
1102 check_hostname = will_verify
1103 elif check_hostname and not will_verify:
1104 raise ValueError("check_hostname needs a SSL context with "
1105 "either CERT_OPTIONAL or CERT_REQUIRED")
1106 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001107 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001108 self._context = context
1109 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001110
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001111 def connect(self):
1112 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001113
Jeremy Hylton636950f2009-03-28 04:34:21 +00001114 sock = socket.create_connection((self.host, self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +00001115 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001116
1117 if self._tunnel_host:
1118 self.sock = sock
1119 self._tunnel()
1120
Antoine Pitroud5323212010-10-22 18:19:07 +00001121 server_hostname = self.host if ssl.HAS_SNI else None
1122 self.sock = self._context.wrap_socket(sock,
1123 server_hostname=server_hostname)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001124 try:
1125 if self._check_hostname:
1126 ssl.match_hostname(self.sock.getpeercert(), self.host)
1127 except Exception:
1128 self.sock.shutdown(socket.SHUT_RDWR)
1129 self.sock.close()
1130 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001131
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001132 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001133
Greg Stein5e0fa402000-06-26 08:28:01 +00001134class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001135 # Subclasses that define an __init__ must call Exception.__init__
1136 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001137 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001138
1139class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001140 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001141
Skip Montanaro9d389972002-03-24 16:53:50 +00001142class InvalidURL(HTTPException):
1143 pass
1144
Greg Stein5e0fa402000-06-26 08:28:01 +00001145class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001146 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001147 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001148 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001149
1150class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001151 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001152
Greg Stein5e0fa402000-06-26 08:28:01 +00001153class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001154 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001155
1156class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001157 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001158 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001159 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001160 self.expected = expected
1161 def __repr__(self):
1162 if self.expected is not None:
1163 e = ', %i more expected' % self.expected
1164 else:
1165 e = ''
1166 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1167 def __str__(self):
1168 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001169
1170class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001171 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001172
1173class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001174 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001175
1176class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001177 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001178
1179class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001180 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001181
1182class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001183 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001184 if not line:
1185 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001186 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001187 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001188
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001189class LineTooLong(HTTPException):
1190 def __init__(self, line_type):
1191 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1192 % (_MAXLINE, line_type))
1193
Greg Stein5e0fa402000-06-26 08:28:01 +00001194# for backwards compatibility
1195error = HTTPException