blob: 4d93b93ff012d460b9ebab87ad13d17e23dbb1fe [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()
Antoine Pitrou084daa22012-12-15 19:11:54 +0100514 else:
515 if not s:
516 self.close()
517
Greg Steindd6eefb2000-07-18 09:09:48 +0000518 return s
519
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000520 def _read_chunked(self, amt):
521 assert self.chunked != _UNKNOWN
522 chunk_left = self.chunk_left
Chris Withers9c33cbf2009-09-04 17:04:16 +0000523 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000524 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000525 if chunk_left is None:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000526 line = self.fp.readline(_MAXLINE + 1)
527 if len(line) > _MAXLINE:
528 raise LineTooLong("chunk size")
Georg Brandl95ba4692008-01-26 09:45:58 +0000529 i = line.find(b";")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000530 if i >= 0:
531 line = line[:i] # strip chunk-extensions
Christian Heimesa612dc02008-02-24 13:08:18 +0000532 try:
533 chunk_left = int(line, 16)
534 except ValueError:
535 # close the connection as protocol synchronisation is
536 # probably lost
537 self.close()
Chris Withers9c33cbf2009-09-04 17:04:16 +0000538 raise IncompleteRead(b''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000539 if chunk_left == 0:
540 break
541 if amt is None:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000542 value.append(self._safe_read(chunk_left))
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.chunk_left = chunk_left - amt
Chris Withers9c33cbf2009-09-04 17:04:16 +0000546 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000547 elif amt == chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000548 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000549 self._safe_read(2) # toss the CRLF at the end of the chunk
550 self.chunk_left = None
Chris Withers9c33cbf2009-09-04 17:04:16 +0000551 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000552 else:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000553 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000554 amt -= chunk_left
555
556 # we read the whole chunk, get another
557 self._safe_read(2) # toss the CRLF at the end of the chunk
558 chunk_left = None
559
560 # read and discard trailer up to the CRLF terminator
561 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000562 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000563 line = self.fp.readline(_MAXLINE + 1)
564 if len(line) > _MAXLINE:
565 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000566 if not line:
567 # a vanishingly small number of sites EOF without
568 # sending the trailer
569 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800570 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000571 break
572
573 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000574 self.close()
575
Chris Withers9c33cbf2009-09-04 17:04:16 +0000576 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000577
Greg Steindd6eefb2000-07-18 09:09:48 +0000578 def _safe_read(self, amt):
579 """Read the number of bytes requested, compensating for partial reads.
580
581 Normally, we have a blocking socket, but a read() can be interrupted
582 by a signal (resulting in a partial read).
583
584 Note that we cannot distinguish between EOF and an interrupt when zero
585 bytes have been read. IncompleteRead() will be raised in this
586 situation.
587
588 This function should be used when <amt> bytes "should" be present for
589 reading. If the bytes are truly not available (due to EOF), then the
590 IncompleteRead exception can be used to detect the problem.
591 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000592 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000593 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000594 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000595 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000596 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000597 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000598 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000599 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000600
Antoine Pitroub353c122009-02-11 00:39:14 +0000601 def fileno(self):
602 return self.fp.fileno()
603
Greg Steindd6eefb2000-07-18 09:09:48 +0000604 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000605 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000606 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000607 headers = self.headers.get_all(name) or default
608 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
609 return headers
610 else:
611 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000612
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000613 def getheaders(self):
614 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000615 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000616 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000617 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000618
Antoine Pitroub353c122009-02-11 00:39:14 +0000619 # We override IOBase.__iter__ so that it doesn't check for closed-ness
620
621 def __iter__(self):
622 return self
623
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000624 # For compatibility with old-style urllib responses.
625
626 def info(self):
627 return self.headers
628
629 def geturl(self):
630 return self.url
631
632 def getcode(self):
633 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000634
635class HTTPConnection:
636
Greg Steindd6eefb2000-07-18 09:09:48 +0000637 _http_vsn = 11
638 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000639
Greg Steindd6eefb2000-07-18 09:09:48 +0000640 response_class = HTTPResponse
641 default_port = HTTP_PORT
642 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000643 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000644
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000645 def __init__(self, host, port=None, strict=_strict_sentinel,
Gregory P. Smithb4066372010-01-03 03:28:29 +0000646 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000647 if strict is not _strict_sentinel:
648 warnings.warn("the 'strict' argument isn't supported anymore; "
649 "http.client now always assumes HTTP/1.x compliant servers.",
650 DeprecationWarning, 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000651 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000652 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000653 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000654 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000655 self.__response = None
656 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000657 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000658 self._tunnel_host = None
659 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000660 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000661
Greg Steindd6eefb2000-07-18 09:09:48 +0000662 self._set_hostport(host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000663
Senthil Kumaran47fff872009-12-20 07:10:31 +0000664 def set_tunnel(self, host, port=None, headers=None):
665 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
666
667 The headers argument should be a mapping of extra HTTP headers
668 to send with the CONNECT request.
669 """
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000670 self._tunnel_host = host
671 self._tunnel_port = port
Senthil Kumaran47fff872009-12-20 07:10:31 +0000672 if headers:
673 self._tunnel_headers = headers
674 else:
675 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000676
Greg Steindd6eefb2000-07-18 09:09:48 +0000677 def _set_hostport(self, host, port):
678 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000679 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000680 j = host.rfind(']') # ipv6 addresses have [...]
681 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000682 try:
683 port = int(host[i+1:])
684 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200685 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
686 port = self.default_port
687 else:
688 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000689 host = host[:i]
690 else:
691 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000692 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000693 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000694 self.host = host
695 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000696
Jeremy Hylton30f86742000-09-18 22:50:38 +0000697 def set_debuglevel(self, level):
698 self.debuglevel = level
699
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000700 def _tunnel(self):
701 self._set_hostport(self._tunnel_host, self._tunnel_port)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000702 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self.host, self.port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000703 connect_bytes = connect_str.encode("ascii")
704 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000705 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000706 header_str = "%s: %s\r\n" % (header, value)
Armin Ronacher59531282011-01-22 13:44:22 +0000707 header_bytes = header_str.encode("latin1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000708 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000709 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000710
Georg Brandlc7c199f2011-01-22 22:06:24 +0000711 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000712 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000713
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000714 if code != 200:
715 self.close()
716 raise socket.error("Tunnel connection failed: %d %s" % (code,
717 message.strip()))
718 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000719 line = response.fp.readline(_MAXLINE + 1)
720 if len(line) > _MAXLINE:
721 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800722 if not line:
723 # for sites which EOF without sending a trailer
724 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800725 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000726 break
727
Greg Steindd6eefb2000-07-18 09:09:48 +0000728 def connect(self):
729 """Connect to the host and port specified in __init__."""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000730 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +0000731 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000732 if self._tunnel_host:
733 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000734
Greg Steindd6eefb2000-07-18 09:09:48 +0000735 def close(self):
736 """Close the connection to the HTTP server."""
737 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000738 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000739 self.sock = None
740 if self.__response:
741 self.__response.close()
742 self.__response = None
743 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000744
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000745 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000746 """Send `data' to the server.
747 ``data`` can be a string object, a bytes object, an array object, a
748 file-like object that supports a .read() method, or an iterable object.
749 """
750
Greg Steindd6eefb2000-07-18 09:09:48 +0000751 if self.sock is None:
752 if self.auto_open:
753 self.connect()
754 else:
755 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000756
Jeremy Hylton30f86742000-09-18 22:50:38 +0000757 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000758 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000759 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000760 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000761 if self.debuglevel > 0:
762 print("sendIng a read()able")
763 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000764 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000765 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000766 except AttributeError:
767 # io.BytesIO and other file-like objects don't have a `mode`
768 # attribute.
769 pass
770 else:
771 if "b" not in mode:
772 encode = True
773 if self.debuglevel > 0:
774 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000775 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000776 datablock = data.read(blocksize)
777 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000778 break
779 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000780 datablock = datablock.encode("iso-8859-1")
781 self.sock.sendall(datablock)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000782
783 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000784 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000785 except TypeError:
786 if isinstance(data, collections.Iterable):
787 for d in data:
788 self.sock.sendall(d)
789 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200790 raise TypeError("data should be a bytes-like object "
791 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000792
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000793 def _output(self, s):
794 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000795
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000796 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000797 """
798 self._buffer.append(s)
799
Benjamin Peterson1742e402008-11-30 22:15:29 +0000800 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000801 """Send the currently buffered request and clear the buffer.
802
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000803 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000804 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000805 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000806 self._buffer.extend((b"", b""))
807 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000808 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000809 # If msg and message_body are sent in a single send() call,
810 # it will avoid performance problems caused by the interaction
Ezio Melotti42da6632011-03-15 05:18:48 +0200811 # between delayed ack and the Nagle algorithm.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000812 if isinstance(message_body, bytes):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000813 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000814 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000815 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000816 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000817 # message_body was not a string (i.e. it is a file), and
818 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000819 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000820
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000821 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000822 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000823
Greg Steindd6eefb2000-07-18 09:09:48 +0000824 `method' specifies an HTTP request method, e.g. 'GET'.
825 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000826 `skip_host' if True does not add automatically a 'Host:' header
827 `skip_accept_encoding' if True does not add automatically an
828 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000829 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000830
Greg Stein616a58d2003-06-24 06:35:19 +0000831 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000832 if self.__response and self.__response.isclosed():
833 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000834
Tim Peters58eb11c2004-01-18 20:29:55 +0000835
Greg Steindd6eefb2000-07-18 09:09:48 +0000836 # in certain cases, we cannot issue another request on this connection.
837 # this occurs when:
838 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
839 # 2) a response to a previous request has signalled that it is going
840 # to close the connection upon completion.
841 # 3) the headers for the previous response have not been read, thus
842 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
843 #
844 # if there is no prior response, then we can request at will.
845 #
846 # if point (2) is true, then we will have passed the socket to the
847 # response (effectively meaning, "there is no prior response"), and
848 # will open a new one when a new request is made.
849 #
850 # Note: if a prior response exists, then we *can* start a new request.
851 # We are not allowed to begin fetching the response to this new
852 # request, however, until that prior response is complete.
853 #
854 if self.__state == _CS_IDLE:
855 self.__state = _CS_REQ_STARTED
856 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000857 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +0000858
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000859 # Save the method we use, we need it later in the response phase
860 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000861 if not url:
862 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000863 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000864
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000865 # Non-ASCII characters should have been eliminated earlier
866 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +0000867
Greg Steindd6eefb2000-07-18 09:09:48 +0000868 if self._http_vsn == 11:
869 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000870
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000871 if not skip_host:
872 # this header is issued *only* for HTTP/1.1
873 # connections. more specifically, this means it is
874 # only issued when the client uses the new
875 # HTTPConnection() class. backwards-compat clients
876 # will be using HTTP/1.0 and those clients may be
877 # issuing this header themselves. we should NOT issue
878 # it twice; some web servers (such as Apache) barf
879 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000880
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000881 # If we need a non-standard port,include it in the
882 # header. If the request is going through a proxy,
883 # but the host of the actual URL, not the host of the
884 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000885
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000886 netloc = ''
887 if url.startswith('http'):
888 nil, netloc, nil, nil, nil = urlsplit(url)
889
890 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000891 try:
892 netloc_enc = netloc.encode("ascii")
893 except UnicodeEncodeError:
894 netloc_enc = netloc.encode("idna")
895 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000896 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000897 try:
898 host_enc = self.host.encode("ascii")
899 except UnicodeEncodeError:
900 host_enc = self.host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +0000901
902 # As per RFC 273, IPv6 address should be wrapped with []
903 # when used as Host header
904
905 if self.host.find(':') >= 0:
906 host_enc = b'[' + host_enc + b']'
907
Georg Brandl86b2fb92008-07-16 03:43:04 +0000908 if self.port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000909 self.putheader('Host', host_enc)
910 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000911 host_enc = host_enc.decode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000912 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000913
Greg Steindd6eefb2000-07-18 09:09:48 +0000914 # note: we are assuming that clients will not attempt to set these
915 # headers since *this* library must deal with the
916 # consequences. this also means that when the supporting
917 # libraries are updated to recognize other forms, then this
918 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000919
Greg Steindd6eefb2000-07-18 09:09:48 +0000920 # we only want a Content-Encoding of "identity" since we don't
921 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000922 if not skip_accept_encoding:
923 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000924
Greg Steindd6eefb2000-07-18 09:09:48 +0000925 # we can accept "chunked" Transfer-Encodings, but no others
926 # NOTE: no TE header implies *only* "chunked"
927 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000928
Greg Steindd6eefb2000-07-18 09:09:48 +0000929 # if TE is supplied in the header, then it must appear in a
930 # Connection header.
931 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000932
Greg Steindd6eefb2000-07-18 09:09:48 +0000933 else:
934 # For HTTP/1.0, the server will assume "not chunked"
935 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000936
Benjamin Petersonf608c612008-11-16 18:33:53 +0000937 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000938 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000939
Greg Steindd6eefb2000-07-18 09:09:48 +0000940 For example: h.putheader('Accept', 'text/html')
941 """
942 if self.__state != _CS_REQ_STARTED:
943 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000944
Guido van Rossum98297ee2007-11-06 21:34:58 +0000945 if hasattr(header, 'encode'):
946 header = header.encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000947 values = list(values)
948 for i, one_value in enumerate(values):
949 if hasattr(one_value, 'encode'):
Armin Ronacher59531282011-01-22 13:44:22 +0000950 values[i] = one_value.encode('latin1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +0000951 elif isinstance(one_value, int):
952 values[i] = str(one_value).encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000953 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000954 header = header + b': ' + value
955 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +0000956
Benjamin Peterson1742e402008-11-30 22:15:29 +0000957 def endheaders(self, message_body=None):
958 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000959
Senthil Kumaranad87fa62011-10-05 23:26:49 +0800960 This method sends the request to the server. The optional message_body
961 argument can be used to pass a message body associated with the
962 request. The message body will be sent in the same packet as the
963 message headers if it is a string, otherwise it is sent as a separate
964 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +0000965 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000966 if self.__state == _CS_REQ_STARTED:
967 self.__state = _CS_REQ_SENT
968 else:
969 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +0000970 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000971
Greg Steindd6eefb2000-07-18 09:09:48 +0000972 def request(self, method, url, body=None, headers={}):
973 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +0000974 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000975
Benjamin Peterson1742e402008-11-30 22:15:29 +0000976 def _set_content_length(self, body):
977 # Set the content-length based on the body.
978 thelen = None
979 try:
980 thelen = str(len(body))
981 except TypeError as te:
982 # If this is a file-like object, try to
983 # fstat its file descriptor
Benjamin Peterson1742e402008-11-30 22:15:29 +0000984 try:
985 thelen = str(os.fstat(body.fileno()).st_size)
986 except (AttributeError, OSError):
987 # Don't send a length if this failed
988 if self.debuglevel > 0: print("Cannot stat!!")
989
990 if thelen is not None:
991 self.putheader('Content-Length', thelen)
992
Greg Steindd6eefb2000-07-18 09:09:48 +0000993 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +0000994 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +0000995 header_names = dict.fromkeys([k.lower() for k in headers])
996 skips = {}
997 if 'host' in header_names:
998 skips['skip_host'] = 1
999 if 'accept-encoding' in header_names:
1000 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001001
Jeremy Hylton2c178252004-08-07 16:28:14 +00001002 self.putrequest(method, url, **skips)
1003
Senthil Kumaran5fa4a892012-05-19 16:58:09 +08001004 if body is not None and ('content-length' not in header_names):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001005 self._set_content_length(body)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001006 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001007 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001008 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001009 # RFC 2616 Section 3.7.1 says that text default has a
1010 # default charset of iso-8859-1.
1011 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001012 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001013
Greg Steindd6eefb2000-07-18 09:09:48 +00001014 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001015 """Get the response from the server.
1016
1017 If the HTTPConnection is in the correct state, returns an
1018 instance of HTTPResponse or of whatever object is returned by
1019 class the response_class variable.
1020
1021 If a request has not been sent or if a previous response has
1022 not be handled, ResponseNotReady is raised. If the HTTP
1023 response indicates that the connection should be closed, then
1024 it will be closed before the response is returned. When the
1025 connection is closed, the underlying socket is closed.
1026 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001027
Greg Stein616a58d2003-06-24 06:35:19 +00001028 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001029 if self.__response and self.__response.isclosed():
1030 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001031
Greg Steindd6eefb2000-07-18 09:09:48 +00001032 # if a prior response exists, then it must be completed (otherwise, we
1033 # cannot read this response's header to determine the connection-close
1034 # behavior)
1035 #
1036 # note: if a prior response existed, but was connection-close, then the
1037 # socket and response were made independent of this HTTPConnection
1038 # object since a new request requires that we open a whole new
1039 # connection
1040 #
1041 # this means the prior response had one of two states:
1042 # 1) will_close: this connection was reset and the prior socket and
1043 # response operate independently
1044 # 2) persistent: the response was retained and we await its
1045 # isclosed() status to become true.
1046 #
1047 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001048 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001049
Jeremy Hylton30f86742000-09-18 22:50:38 +00001050 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001051 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001052 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001053 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001054 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001055
Jeremy Hylton39c03802002-07-12 14:04:09 +00001056 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001057 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001058 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001059
Greg Steindd6eefb2000-07-18 09:09:48 +00001060 if response.will_close:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001061 # this effectively passes the connection to the response
1062 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001063 else:
1064 # remember this, so we can tell when it is complete
1065 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001066
Greg Steindd6eefb2000-07-18 09:09:48 +00001067 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001068
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001069try:
1070 import ssl
1071except ImportError:
1072 pass
1073else:
1074 class HTTPSConnection(HTTPConnection):
1075 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001076
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001077 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001078
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001079 # XXX Should key_file and cert_file be deprecated in favour of context?
1080
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001081 def __init__(self, host, port=None, key_file=None, cert_file=None,
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001082 strict=_strict_sentinel, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001083 source_address=None, *, context=None, check_hostname=None):
Gregory P. Smithb4066372010-01-03 03:28:29 +00001084 super(HTTPSConnection, self).__init__(host, port, strict, timeout,
1085 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001086 self.key_file = key_file
1087 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001088 if context is None:
1089 # Some reasonable defaults
1090 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
1091 context.options |= ssl.OP_NO_SSLv2
1092 will_verify = context.verify_mode != ssl.CERT_NONE
1093 if check_hostname is None:
1094 check_hostname = will_verify
1095 elif check_hostname and not will_verify:
1096 raise ValueError("check_hostname needs a SSL context with "
1097 "either CERT_OPTIONAL or CERT_REQUIRED")
1098 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001099 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001100 self._context = context
1101 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001102
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001103 def connect(self):
1104 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001105
Jeremy Hylton636950f2009-03-28 04:34:21 +00001106 sock = socket.create_connection((self.host, self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +00001107 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001108
1109 if self._tunnel_host:
1110 self.sock = sock
1111 self._tunnel()
1112
Antoine Pitroud5323212010-10-22 18:19:07 +00001113 server_hostname = self.host if ssl.HAS_SNI else None
1114 self.sock = self._context.wrap_socket(sock,
1115 server_hostname=server_hostname)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001116 try:
1117 if self._check_hostname:
1118 ssl.match_hostname(self.sock.getpeercert(), self.host)
1119 except Exception:
1120 self.sock.shutdown(socket.SHUT_RDWR)
1121 self.sock.close()
1122 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001123
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001124 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001125
Greg Stein5e0fa402000-06-26 08:28:01 +00001126class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001127 # Subclasses that define an __init__ must call Exception.__init__
1128 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001129 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001130
1131class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001132 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001133
Skip Montanaro9d389972002-03-24 16:53:50 +00001134class InvalidURL(HTTPException):
1135 pass
1136
Greg Stein5e0fa402000-06-26 08:28:01 +00001137class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001138 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001139 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001140 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001141
1142class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001143 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001144
Greg Stein5e0fa402000-06-26 08:28:01 +00001145class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001146 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001147
1148class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001149 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001150 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001151 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001152 self.expected = expected
1153 def __repr__(self):
1154 if self.expected is not None:
1155 e = ', %i more expected' % self.expected
1156 else:
1157 e = ''
1158 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1159 def __str__(self):
1160 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001161
1162class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001163 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001164
1165class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001166 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001167
1168class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001169 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001170
1171class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001172 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001173
1174class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001175 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001176 if not line:
1177 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001178 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001179 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001180
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001181class LineTooLong(HTTPException):
1182 def __init__(self, line_type):
1183 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1184 % (_MAXLINE, line_type))
1185
Greg Stein5e0fa402000-06-26 08:28:01 +00001186# for backwards compatibility
1187error = HTTPException