blob: 4574eedaf2a8e1adfa8be2fda0fa7e773b65b006 [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()
Antoine Pitroude595652013-02-02 23:08:51 +0100518 elif self.length is not None:
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000519 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
Greg Steindd6eefb2000-07-18 09:09:48 +0000523 return s
524
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000525 def _read_chunked(self, amt):
526 assert self.chunked != _UNKNOWN
527 chunk_left = self.chunk_left
Chris Withers9c33cbf2009-09-04 17:04:16 +0000528 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000529 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000530 if chunk_left is None:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000531 line = self.fp.readline(_MAXLINE + 1)
532 if len(line) > _MAXLINE:
533 raise LineTooLong("chunk size")
Georg Brandl95ba4692008-01-26 09:45:58 +0000534 i = line.find(b";")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000535 if i >= 0:
536 line = line[:i] # strip chunk-extensions
Christian Heimesa612dc02008-02-24 13:08:18 +0000537 try:
538 chunk_left = int(line, 16)
539 except ValueError:
540 # close the connection as protocol synchronisation is
541 # probably lost
542 self.close()
Chris Withers9c33cbf2009-09-04 17:04:16 +0000543 raise IncompleteRead(b''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000544 if chunk_left == 0:
545 break
546 if amt is None:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000547 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000548 elif amt < chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000549 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000550 self.chunk_left = chunk_left - amt
Chris Withers9c33cbf2009-09-04 17:04:16 +0000551 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000552 elif amt == chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000553 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000554 self._safe_read(2) # toss the CRLF at the end of the chunk
555 self.chunk_left = None
Chris Withers9c33cbf2009-09-04 17:04:16 +0000556 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000557 else:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000558 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000559 amt -= chunk_left
560
561 # we read the whole chunk, get another
562 self._safe_read(2) # toss the CRLF at the end of the chunk
563 chunk_left = None
564
565 # read and discard trailer up to the CRLF terminator
566 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000567 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000568 line = self.fp.readline(_MAXLINE + 1)
569 if len(line) > _MAXLINE:
570 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000571 if not line:
572 # a vanishingly small number of sites EOF without
573 # sending the trailer
574 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800575 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000576 break
577
578 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000579 self.close()
580
Chris Withers9c33cbf2009-09-04 17:04:16 +0000581 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000582
Greg Steindd6eefb2000-07-18 09:09:48 +0000583 def _safe_read(self, amt):
584 """Read the number of bytes requested, compensating for partial reads.
585
586 Normally, we have a blocking socket, but a read() can be interrupted
587 by a signal (resulting in a partial read).
588
589 Note that we cannot distinguish between EOF and an interrupt when zero
590 bytes have been read. IncompleteRead() will be raised in this
591 situation.
592
593 This function should be used when <amt> bytes "should" be present for
594 reading. If the bytes are truly not available (due to EOF), then the
595 IncompleteRead exception can be used to detect the problem.
596 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000597 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000598 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000599 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000600 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000601 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000602 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000603 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000604 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000605
Antoine Pitroub353c122009-02-11 00:39:14 +0000606 def fileno(self):
607 return self.fp.fileno()
608
Greg Steindd6eefb2000-07-18 09:09:48 +0000609 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000610 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000611 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000612 headers = self.headers.get_all(name) or default
613 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
614 return headers
615 else:
616 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000617
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000618 def getheaders(self):
619 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000620 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000621 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000622 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000623
Antoine Pitroub353c122009-02-11 00:39:14 +0000624 # We override IOBase.__iter__ so that it doesn't check for closed-ness
625
626 def __iter__(self):
627 return self
628
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000629 # For compatibility with old-style urllib responses.
630
631 def info(self):
632 return self.headers
633
634 def geturl(self):
635 return self.url
636
637 def getcode(self):
638 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000639
640class HTTPConnection:
641
Greg Steindd6eefb2000-07-18 09:09:48 +0000642 _http_vsn = 11
643 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000644
Greg Steindd6eefb2000-07-18 09:09:48 +0000645 response_class = HTTPResponse
646 default_port = HTTP_PORT
647 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000648 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000649
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000650 def __init__(self, host, port=None, strict=_strict_sentinel,
Gregory P. Smithb4066372010-01-03 03:28:29 +0000651 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000652 if strict is not _strict_sentinel:
653 warnings.warn("the 'strict' argument isn't supported anymore; "
654 "http.client now always assumes HTTP/1.x compliant servers.",
655 DeprecationWarning, 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000656 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000657 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000658 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000659 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000660 self.__response = None
661 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000662 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000663 self._tunnel_host = None
664 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000665 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000666
Greg Steindd6eefb2000-07-18 09:09:48 +0000667 self._set_hostport(host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000668
Senthil Kumaran47fff872009-12-20 07:10:31 +0000669 def set_tunnel(self, host, port=None, headers=None):
670 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
671
672 The headers argument should be a mapping of extra HTTP headers
673 to send with the CONNECT request.
674 """
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000675 self._tunnel_host = host
676 self._tunnel_port = port
Senthil Kumaran47fff872009-12-20 07:10:31 +0000677 if headers:
678 self._tunnel_headers = headers
679 else:
680 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000681
Greg Steindd6eefb2000-07-18 09:09:48 +0000682 def _set_hostport(self, host, port):
683 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000684 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000685 j = host.rfind(']') # ipv6 addresses have [...]
686 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000687 try:
688 port = int(host[i+1:])
689 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200690 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
691 port = self.default_port
692 else:
693 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000694 host = host[:i]
695 else:
696 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000697 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000698 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000699 self.host = host
700 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000701
Jeremy Hylton30f86742000-09-18 22:50:38 +0000702 def set_debuglevel(self, level):
703 self.debuglevel = level
704
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000705 def _tunnel(self):
706 self._set_hostport(self._tunnel_host, self._tunnel_port)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000707 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self.host, self.port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000708 connect_bytes = connect_str.encode("ascii")
709 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000710 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000711 header_str = "%s: %s\r\n" % (header, value)
Armin Ronacher59531282011-01-22 13:44:22 +0000712 header_bytes = header_str.encode("latin1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000713 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000714 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000715
Georg Brandlc7c199f2011-01-22 22:06:24 +0000716 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000717 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000718
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000719 if code != 200:
720 self.close()
721 raise socket.error("Tunnel connection failed: %d %s" % (code,
722 message.strip()))
723 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000724 line = response.fp.readline(_MAXLINE + 1)
725 if len(line) > _MAXLINE:
726 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800727 if not line:
728 # for sites which EOF without sending a trailer
729 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800730 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000731 break
732
Greg Steindd6eefb2000-07-18 09:09:48 +0000733 def connect(self):
734 """Connect to the host and port specified in __init__."""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000735 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +0000736 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000737 if self._tunnel_host:
738 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000739
Greg Steindd6eefb2000-07-18 09:09:48 +0000740 def close(self):
741 """Close the connection to the HTTP server."""
742 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000743 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000744 self.sock = None
745 if self.__response:
746 self.__response.close()
747 self.__response = None
748 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000749
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000750 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000751 """Send `data' to the server.
752 ``data`` can be a string object, a bytes object, an array object, a
753 file-like object that supports a .read() method, or an iterable object.
754 """
755
Greg Steindd6eefb2000-07-18 09:09:48 +0000756 if self.sock is None:
757 if self.auto_open:
758 self.connect()
759 else:
760 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000761
Jeremy Hylton30f86742000-09-18 22:50:38 +0000762 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000763 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000764 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000765 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000766 if self.debuglevel > 0:
767 print("sendIng a read()able")
768 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000769 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000770 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000771 except AttributeError:
772 # io.BytesIO and other file-like objects don't have a `mode`
773 # attribute.
774 pass
775 else:
776 if "b" not in mode:
777 encode = True
778 if self.debuglevel > 0:
779 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000780 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000781 datablock = data.read(blocksize)
782 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000783 break
784 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000785 datablock = datablock.encode("iso-8859-1")
786 self.sock.sendall(datablock)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000787
788 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000789 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000790 except TypeError:
791 if isinstance(data, collections.Iterable):
792 for d in data:
793 self.sock.sendall(d)
794 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200795 raise TypeError("data should be a bytes-like object "
796 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000797
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000798 def _output(self, s):
799 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000800
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000801 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000802 """
803 self._buffer.append(s)
804
Benjamin Peterson1742e402008-11-30 22:15:29 +0000805 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000806 """Send the currently buffered request and clear the buffer.
807
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000808 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000809 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000810 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000811 self._buffer.extend((b"", b""))
812 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000813 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000814 # If msg and message_body are sent in a single send() call,
815 # it will avoid performance problems caused by the interaction
Ezio Melotti42da6632011-03-15 05:18:48 +0200816 # between delayed ack and the Nagle algorithm.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000817 if isinstance(message_body, bytes):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000818 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000819 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000820 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000821 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000822 # message_body was not a string (i.e. it is a file), and
823 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000824 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000825
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000826 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000827 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000828
Greg Steindd6eefb2000-07-18 09:09:48 +0000829 `method' specifies an HTTP request method, e.g. 'GET'.
830 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000831 `skip_host' if True does not add automatically a 'Host:' header
832 `skip_accept_encoding' if True does not add automatically an
833 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000834 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000835
Greg Stein616a58d2003-06-24 06:35:19 +0000836 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000837 if self.__response and self.__response.isclosed():
838 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000839
Tim Peters58eb11c2004-01-18 20:29:55 +0000840
Greg Steindd6eefb2000-07-18 09:09:48 +0000841 # in certain cases, we cannot issue another request on this connection.
842 # this occurs when:
843 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
844 # 2) a response to a previous request has signalled that it is going
845 # to close the connection upon completion.
846 # 3) the headers for the previous response have not been read, thus
847 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
848 #
849 # if there is no prior response, then we can request at will.
850 #
851 # if point (2) is true, then we will have passed the socket to the
852 # response (effectively meaning, "there is no prior response"), and
853 # will open a new one when a new request is made.
854 #
855 # Note: if a prior response exists, then we *can* start a new request.
856 # We are not allowed to begin fetching the response to this new
857 # request, however, until that prior response is complete.
858 #
859 if self.__state == _CS_IDLE:
860 self.__state = _CS_REQ_STARTED
861 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000862 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +0000863
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000864 # Save the method we use, we need it later in the response phase
865 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000866 if not url:
867 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000868 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000869
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000870 # Non-ASCII characters should have been eliminated earlier
871 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +0000872
Greg Steindd6eefb2000-07-18 09:09:48 +0000873 if self._http_vsn == 11:
874 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000875
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000876 if not skip_host:
877 # this header is issued *only* for HTTP/1.1
878 # connections. more specifically, this means it is
879 # only issued when the client uses the new
880 # HTTPConnection() class. backwards-compat clients
881 # will be using HTTP/1.0 and those clients may be
882 # issuing this header themselves. we should NOT issue
883 # it twice; some web servers (such as Apache) barf
884 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000885
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000886 # If we need a non-standard port,include it in the
887 # header. If the request is going through a proxy,
888 # but the host of the actual URL, not the host of the
889 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000890
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000891 netloc = ''
892 if url.startswith('http'):
893 nil, netloc, nil, nil, nil = urlsplit(url)
894
895 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000896 try:
897 netloc_enc = netloc.encode("ascii")
898 except UnicodeEncodeError:
899 netloc_enc = netloc.encode("idna")
900 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000901 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000902 try:
903 host_enc = self.host.encode("ascii")
904 except UnicodeEncodeError:
905 host_enc = self.host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +0000906
907 # As per RFC 273, IPv6 address should be wrapped with []
908 # when used as Host header
909
910 if self.host.find(':') >= 0:
911 host_enc = b'[' + host_enc + b']'
912
Georg Brandl86b2fb92008-07-16 03:43:04 +0000913 if self.port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000914 self.putheader('Host', host_enc)
915 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000916 host_enc = host_enc.decode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000917 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000918
Greg Steindd6eefb2000-07-18 09:09:48 +0000919 # note: we are assuming that clients will not attempt to set these
920 # headers since *this* library must deal with the
921 # consequences. this also means that when the supporting
922 # libraries are updated to recognize other forms, then this
923 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000924
Greg Steindd6eefb2000-07-18 09:09:48 +0000925 # we only want a Content-Encoding of "identity" since we don't
926 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000927 if not skip_accept_encoding:
928 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000929
Greg Steindd6eefb2000-07-18 09:09:48 +0000930 # we can accept "chunked" Transfer-Encodings, but no others
931 # NOTE: no TE header implies *only* "chunked"
932 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000933
Greg Steindd6eefb2000-07-18 09:09:48 +0000934 # if TE is supplied in the header, then it must appear in a
935 # Connection header.
936 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000937
Greg Steindd6eefb2000-07-18 09:09:48 +0000938 else:
939 # For HTTP/1.0, the server will assume "not chunked"
940 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000941
Benjamin Petersonf608c612008-11-16 18:33:53 +0000942 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000943 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000944
Greg Steindd6eefb2000-07-18 09:09:48 +0000945 For example: h.putheader('Accept', 'text/html')
946 """
947 if self.__state != _CS_REQ_STARTED:
948 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000949
Guido van Rossum98297ee2007-11-06 21:34:58 +0000950 if hasattr(header, 'encode'):
951 header = header.encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000952 values = list(values)
953 for i, one_value in enumerate(values):
954 if hasattr(one_value, 'encode'):
Armin Ronacher59531282011-01-22 13:44:22 +0000955 values[i] = one_value.encode('latin1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +0000956 elif isinstance(one_value, int):
957 values[i] = str(one_value).encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000958 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000959 header = header + b': ' + value
960 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +0000961
Benjamin Peterson1742e402008-11-30 22:15:29 +0000962 def endheaders(self, message_body=None):
963 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000964
Senthil Kumaranad87fa62011-10-05 23:26:49 +0800965 This method sends the request to the server. The optional message_body
966 argument can be used to pass a message body associated with the
967 request. The message body will be sent in the same packet as the
968 message headers if it is a string, otherwise it is sent as a separate
969 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +0000970 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000971 if self.__state == _CS_REQ_STARTED:
972 self.__state = _CS_REQ_SENT
973 else:
974 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +0000975 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000976
Greg Steindd6eefb2000-07-18 09:09:48 +0000977 def request(self, method, url, body=None, headers={}):
978 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +0000979 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000980
Benjamin Peterson1742e402008-11-30 22:15:29 +0000981 def _set_content_length(self, body):
982 # Set the content-length based on the body.
983 thelen = None
984 try:
985 thelen = str(len(body))
986 except TypeError as te:
987 # If this is a file-like object, try to
988 # fstat its file descriptor
Benjamin Peterson1742e402008-11-30 22:15:29 +0000989 try:
990 thelen = str(os.fstat(body.fileno()).st_size)
991 except (AttributeError, OSError):
992 # Don't send a length if this failed
993 if self.debuglevel > 0: print("Cannot stat!!")
994
995 if thelen is not None:
996 self.putheader('Content-Length', thelen)
997
Greg Steindd6eefb2000-07-18 09:09:48 +0000998 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +0000999 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +00001000 header_names = dict.fromkeys([k.lower() for k in headers])
1001 skips = {}
1002 if 'host' in header_names:
1003 skips['skip_host'] = 1
1004 if 'accept-encoding' in header_names:
1005 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001006
Jeremy Hylton2c178252004-08-07 16:28:14 +00001007 self.putrequest(method, url, **skips)
1008
Senthil Kumaran5fa4a892012-05-19 16:58:09 +08001009 if body is not None and ('content-length' not in header_names):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001010 self._set_content_length(body)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001011 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001012 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001013 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001014 # RFC 2616 Section 3.7.1 says that text default has a
1015 # default charset of iso-8859-1.
1016 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001017 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001018
Greg Steindd6eefb2000-07-18 09:09:48 +00001019 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001020 """Get the response from the server.
1021
1022 If the HTTPConnection is in the correct state, returns an
1023 instance of HTTPResponse or of whatever object is returned by
1024 class the response_class variable.
1025
1026 If a request has not been sent or if a previous response has
1027 not be handled, ResponseNotReady is raised. If the HTTP
1028 response indicates that the connection should be closed, then
1029 it will be closed before the response is returned. When the
1030 connection is closed, the underlying socket is closed.
1031 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001032
Greg Stein616a58d2003-06-24 06:35:19 +00001033 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001034 if self.__response and self.__response.isclosed():
1035 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001036
Greg Steindd6eefb2000-07-18 09:09:48 +00001037 # if a prior response exists, then it must be completed (otherwise, we
1038 # cannot read this response's header to determine the connection-close
1039 # behavior)
1040 #
1041 # note: if a prior response existed, but was connection-close, then the
1042 # socket and response were made independent of this HTTPConnection
1043 # object since a new request requires that we open a whole new
1044 # connection
1045 #
1046 # this means the prior response had one of two states:
1047 # 1) will_close: this connection was reset and the prior socket and
1048 # response operate independently
1049 # 2) persistent: the response was retained and we await its
1050 # isclosed() status to become true.
1051 #
1052 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001053 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001054
Jeremy Hylton30f86742000-09-18 22:50:38 +00001055 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001056 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001057 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001058 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001059 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001060
Jeremy Hylton39c03802002-07-12 14:04:09 +00001061 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001062 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001063 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001064
Greg Steindd6eefb2000-07-18 09:09:48 +00001065 if response.will_close:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001066 # this effectively passes the connection to the response
1067 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001068 else:
1069 # remember this, so we can tell when it is complete
1070 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001071
Greg Steindd6eefb2000-07-18 09:09:48 +00001072 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001073
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001074try:
1075 import ssl
1076except ImportError:
1077 pass
1078else:
1079 class HTTPSConnection(HTTPConnection):
1080 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001081
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001082 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001083
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001084 # XXX Should key_file and cert_file be deprecated in favour of context?
1085
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001086 def __init__(self, host, port=None, key_file=None, cert_file=None,
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001087 strict=_strict_sentinel, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001088 source_address=None, *, context=None, check_hostname=None):
Gregory P. Smithb4066372010-01-03 03:28:29 +00001089 super(HTTPSConnection, self).__init__(host, port, strict, timeout,
1090 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001091 self.key_file = key_file
1092 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001093 if context is None:
1094 # Some reasonable defaults
1095 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
1096 context.options |= ssl.OP_NO_SSLv2
1097 will_verify = context.verify_mode != ssl.CERT_NONE
1098 if check_hostname is None:
1099 check_hostname = will_verify
1100 elif check_hostname and not will_verify:
1101 raise ValueError("check_hostname needs a SSL context with "
1102 "either CERT_OPTIONAL or CERT_REQUIRED")
1103 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001104 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001105 self._context = context
1106 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001107
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001108 def connect(self):
1109 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001110
Jeremy Hylton636950f2009-03-28 04:34:21 +00001111 sock = socket.create_connection((self.host, self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +00001112 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001113
1114 if self._tunnel_host:
1115 self.sock = sock
1116 self._tunnel()
1117
Antoine Pitroud5323212010-10-22 18:19:07 +00001118 server_hostname = self.host if ssl.HAS_SNI else None
1119 self.sock = self._context.wrap_socket(sock,
1120 server_hostname=server_hostname)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001121 try:
1122 if self._check_hostname:
1123 ssl.match_hostname(self.sock.getpeercert(), self.host)
1124 except Exception:
1125 self.sock.shutdown(socket.SHUT_RDWR)
1126 self.sock.close()
1127 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001128
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001129 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001130
Greg Stein5e0fa402000-06-26 08:28:01 +00001131class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001132 # Subclasses that define an __init__ must call Exception.__init__
1133 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001134 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001135
1136class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001137 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001138
Skip Montanaro9d389972002-03-24 16:53:50 +00001139class InvalidURL(HTTPException):
1140 pass
1141
Greg Stein5e0fa402000-06-26 08:28:01 +00001142class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001143 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001144 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001145 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001146
1147class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001148 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001149
Greg Stein5e0fa402000-06-26 08:28:01 +00001150class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001151 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001152
1153class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001154 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001155 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001156 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001157 self.expected = expected
1158 def __repr__(self):
1159 if self.expected is not None:
1160 e = ', %i more expected' % self.expected
1161 else:
1162 e = ''
1163 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1164 def __str__(self):
1165 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001166
1167class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001168 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001169
1170class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001171 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001172
1173class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001174 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001175
1176class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001177 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001178
1179class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001180 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001181 if not line:
1182 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001183 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001184 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001185
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001186class LineTooLong(HTTPException):
1187 def __init__(self, line_type):
1188 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1189 % (_MAXLINE, line_type))
1190
Greg Stein5e0fa402000-06-26 08:28:01 +00001191# for backwards compatibility
1192error = HTTPException