blob: 8d62aa57a7538b0ca38f0fd191c41d0f59666199 [file] [log] [blame]
Greg Stein5e0fa402000-06-26 08:28:01 +00001"""HTTP/1.1 client library
Guido van Rossum41999c11997-12-09 00:12:23 +00002
Greg Stein5e0fa402000-06-26 08:28:01 +00003<intro stuff goes here>
4<other stuff, too>
Guido van Rossum41999c11997-12-09 00:12:23 +00005
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006HTTPConnection goes through a number of "states", which define when a client
Greg Stein5e0fa402000-06-26 08:28:01 +00007may legally make another request or fetch the response for a particular
8request. This diagram details these state transitions:
Guido van Rossum41999c11997-12-09 00:12:23 +00009
Greg Stein5e0fa402000-06-26 08:28:01 +000010 (null)
11 |
12 | HTTPConnection()
13 v
14 Idle
15 |
16 | putrequest()
17 v
18 Request-started
19 |
20 | ( putheader() )* endheaders()
21 v
22 Request-sent
23 |
24 | response = getresponse()
25 v
26 Unread-response [Response-headers-read]
27 |\____________________
Tim Peters5ceadc82001-01-13 19:16:21 +000028 | |
29 | response.read() | putrequest()
30 v v
31 Idle Req-started-unread-response
32 ______/|
33 / |
34 response.read() | | ( putheader() )* endheaders()
35 v v
36 Request-started Req-sent-unread-response
37 |
38 | response.read()
39 v
40 Request-sent
Greg Stein5e0fa402000-06-26 08:28:01 +000041
42This diagram presents the following rules:
43 -- a second request may not be started until {response-headers-read}
44 -- a response [object] cannot be retrieved until {request-sent}
45 -- there is no differentiation between an unread response body and a
46 partially read response body
47
48Note: this enforcement is applied by the HTTPConnection class. The
49 HTTPResponse class does not enforce this state machine, which
50 implies sophisticated clients may accelerate the request/response
51 pipeline. Caution should be taken, though: accelerating the states
52 beyond the above pattern may imply knowledge of the server's
53 connection-close behavior for certain requests. For example, it
54 is impossible to tell whether the server will close the connection
55 UNTIL the response headers have been read; this means that further
56 requests cannot be placed into the pipeline until it is known that
57 the server will NOT be closing the connection.
58
59Logical State __state __response
60------------- ------- ----------
61Idle _CS_IDLE None
62Request-started _CS_REQ_STARTED None
63Request-sent _CS_REQ_SENT None
64Unread-response _CS_IDLE <response_class>
65Req-started-unread-response _CS_REQ_STARTED <response_class>
66Req-sent-unread-response _CS_REQ_SENT <response_class>
Guido van Rossum41999c11997-12-09 00:12:23 +000067"""
Guido van Rossum23acc951994-02-21 16:36:04 +000068
Barry Warsaw820c1202008-06-12 04:06:45 +000069import email.parser
70import email.message
Jeremy Hylton636950f2009-03-28 04:34:21 +000071import io
72import os
73import socket
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000074import collections
Jeremy Hylton1afc1692008-06-18 20:49:58 +000075from urllib.parse import urlsplit
Thomas Wouters89d996e2007-09-08 17:39:28 +000076import warnings
Guido van Rossum23acc951994-02-21 16:36:04 +000077
Thomas Wouters47b49bf2007-08-30 22:15:33 +000078__all__ = ["HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000079 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000080 "UnknownTransferEncoding", "UnimplementedFileMode",
81 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
82 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
Georg Brandl6aab16e2006-02-17 19:17:25 +000083 "BadStatusLine", "error", "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000084
Guido van Rossum23acc951994-02-21 16:36:04 +000085HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000086HTTPS_PORT = 443
87
Greg Stein5e0fa402000-06-26 08:28:01 +000088_UNKNOWN = 'UNKNOWN'
89
90# connection states
91_CS_IDLE = 'Idle'
92_CS_REQ_STARTED = 'Request-started'
93_CS_REQ_SENT = 'Request-sent'
94
Martin v. Löwis39a31782004-09-18 09:03:49 +000095# status codes
96# informational
97CONTINUE = 100
98SWITCHING_PROTOCOLS = 101
99PROCESSING = 102
100
101# successful
102OK = 200
103CREATED = 201
104ACCEPTED = 202
105NON_AUTHORITATIVE_INFORMATION = 203
106NO_CONTENT = 204
107RESET_CONTENT = 205
108PARTIAL_CONTENT = 206
109MULTI_STATUS = 207
110IM_USED = 226
111
112# redirection
113MULTIPLE_CHOICES = 300
114MOVED_PERMANENTLY = 301
115FOUND = 302
116SEE_OTHER = 303
117NOT_MODIFIED = 304
118USE_PROXY = 305
119TEMPORARY_REDIRECT = 307
120
121# client error
122BAD_REQUEST = 400
123UNAUTHORIZED = 401
124PAYMENT_REQUIRED = 402
125FORBIDDEN = 403
126NOT_FOUND = 404
127METHOD_NOT_ALLOWED = 405
128NOT_ACCEPTABLE = 406
129PROXY_AUTHENTICATION_REQUIRED = 407
130REQUEST_TIMEOUT = 408
131CONFLICT = 409
132GONE = 410
133LENGTH_REQUIRED = 411
134PRECONDITION_FAILED = 412
135REQUEST_ENTITY_TOO_LARGE = 413
136REQUEST_URI_TOO_LONG = 414
137UNSUPPORTED_MEDIA_TYPE = 415
138REQUESTED_RANGE_NOT_SATISFIABLE = 416
139EXPECTATION_FAILED = 417
140UNPROCESSABLE_ENTITY = 422
141LOCKED = 423
142FAILED_DEPENDENCY = 424
143UPGRADE_REQUIRED = 426
144
145# server error
146INTERNAL_SERVER_ERROR = 500
147NOT_IMPLEMENTED = 501
148BAD_GATEWAY = 502
149SERVICE_UNAVAILABLE = 503
150GATEWAY_TIMEOUT = 504
151HTTP_VERSION_NOT_SUPPORTED = 505
152INSUFFICIENT_STORAGE = 507
153NOT_EXTENDED = 510
154
Georg Brandl6aab16e2006-02-17 19:17:25 +0000155# Mapping status codes to official W3C names
156responses = {
157 100: 'Continue',
158 101: 'Switching Protocols',
159
160 200: 'OK',
161 201: 'Created',
162 202: 'Accepted',
163 203: 'Non-Authoritative Information',
164 204: 'No Content',
165 205: 'Reset Content',
166 206: 'Partial Content',
167
168 300: 'Multiple Choices',
169 301: 'Moved Permanently',
170 302: 'Found',
171 303: 'See Other',
172 304: 'Not Modified',
173 305: 'Use Proxy',
174 306: '(Unused)',
175 307: 'Temporary Redirect',
176
177 400: 'Bad Request',
178 401: 'Unauthorized',
179 402: 'Payment Required',
180 403: 'Forbidden',
181 404: 'Not Found',
182 405: 'Method Not Allowed',
183 406: 'Not Acceptable',
184 407: 'Proxy Authentication Required',
185 408: 'Request Timeout',
186 409: 'Conflict',
187 410: 'Gone',
188 411: 'Length Required',
189 412: 'Precondition Failed',
190 413: 'Request Entity Too Large',
191 414: 'Request-URI Too Long',
192 415: 'Unsupported Media Type',
193 416: 'Requested Range Not Satisfiable',
194 417: 'Expectation Failed',
195
196 500: 'Internal Server Error',
197 501: 'Not Implemented',
198 502: 'Bad Gateway',
199 503: 'Service Unavailable',
200 504: 'Gateway Timeout',
201 505: 'HTTP Version Not Supported',
202}
203
Georg Brandl80ba8e82005-09-29 20:16:07 +0000204# maximal amount of data to read at one time in _safe_read
205MAXAMOUNT = 1048576
206
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000207# maximal line length when calling readline().
208_MAXLINE = 65536
209
Barry Warsaw820c1202008-06-12 04:06:45 +0000210class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000211 # XXX The only usage of this method is in
212 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
213 # that it doesn't need to be part of the public API. The API has
214 # never been defined so this could cause backwards compatibility
215 # issues.
216
Barry Warsaw820c1202008-06-12 04:06:45 +0000217 def getallmatchingheaders(self, name):
218 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000219
Barry Warsaw820c1202008-06-12 04:06:45 +0000220 Look through the list of headers and find all lines matching a given
221 header name (and their continuation lines). A list of the lines is
222 returned, without interpretation. If the header does not occur, an
223 empty list is returned. If the header occurs multiple times, all
224 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000225
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000226 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000227 name = name.lower() + ':'
228 n = len(name)
229 lst = []
230 hit = 0
231 for line in self.keys():
232 if line[:n].lower() == name:
233 hit = 1
234 elif not line[:1].isspace():
235 hit = 0
236 if hit:
237 lst.append(line)
238 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000239
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000240def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000241 """Parses only RFC2822 headers from a file pointer.
242
243 email Parser wants to see strings rather than bytes.
244 But a TextIOWrapper around self.rfile would buffer too many bytes
245 from the stream, bytes which we later need to read as bytes.
246 So we read the correct bytes here, as bytes, for email Parser
247 to parse.
248
249 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000250 headers = []
251 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000252 line = fp.readline(_MAXLINE + 1)
253 if len(line) > _MAXLINE:
254 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000255 headers.append(line)
256 if line in (b'\r\n', b'\n', b''):
257 break
258 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000259 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000260
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000261
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000262_strict_sentinel = object()
263
264class HTTPResponse(io.RawIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000265
266 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
267
Jeremy Hylton811fc142007-08-03 13:30:02 +0000268 # The bytes from the socket object are iso-8859-1 strings.
269 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
270 # text following RFC 2047. The basic status line parsing only
271 # accepts iso-8859-1.
272
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000273 def __init__(self, sock, debuglevel=0, strict=_strict_sentinel, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000274 # If the response includes a content-length header, we need to
275 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000276 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000277 # the server times out and closes the connection. This will
278 # happen if a self.fp.read() is done (without a size) whether
279 # self.fp is buffered or not. So, no self.fp.read() by
280 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000281 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000282 self.debuglevel = debuglevel
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000283 if strict is not _strict_sentinel:
284 warnings.warn("the 'strict' argument isn't supported anymore; "
285 "http.client now always assumes HTTP/1.x compliant servers.",
286 DeprecationWarning, 2)
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000287 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000288
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000289 # The HTTPResponse object is returned via urllib. The clients
290 # of http and urllib expect different attributes for the
291 # headers. headers is used here and supports urllib. msg is
292 # provided as a backwards compatibility layer for http
293 # clients.
294
295 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000296
Greg Steindd6eefb2000-07-18 09:09:48 +0000297 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000298 self.version = _UNKNOWN # HTTP-Version
299 self.status = _UNKNOWN # Status-Code
300 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000301
Tim Peters07e99cb2001-01-14 23:47:14 +0000302 self.chunked = _UNKNOWN # is "chunked" being used?
303 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
304 self.length = _UNKNOWN # number of bytes left in response
305 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000306
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000307 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000308 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
309 if len(line) > _MAXLINE:
310 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000311 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000312 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000313 if not line:
314 # Presumably, the server closed the connection before
315 # sending a valid response.
316 raise BadStatusLine(line)
Greg Steindd6eefb2000-07-18 09:09:48 +0000317 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000318 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000319 except ValueError:
320 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000321 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000322 reason = ""
323 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000324 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000325 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000326 if not version.startswith("HTTP/"):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000327 self.close()
328 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000329
Jeremy Hylton23d40472001-04-13 14:57:08 +0000330 # The status code is a three-digit number
331 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000332 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000333 if status < 100 or status > 999:
334 raise BadStatusLine(line)
335 except ValueError:
336 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000337 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000338
Jeremy Hylton39c03802002-07-12 14:04:09 +0000339 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000340 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000341 # we've already started reading the response
342 return
343
344 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000345 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000346 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000347 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000348 break
349 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000350 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000351 skip = self.fp.readline(_MAXLINE + 1)
352 if len(skip) > _MAXLINE:
353 raise LineTooLong("header line")
354 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000355 if not skip:
356 break
357 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000358 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000359
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000360 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000361 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000362 if version in ("HTTP/1.0", "HTTP/0.9"):
363 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000364 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000365 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000366 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000367 else:
368 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000369
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000370 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000371
Jeremy Hylton30f86742000-09-18 22:50:38 +0000372 if self.debuglevel > 0:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000373 for hdr in self.headers:
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000374 print("header:", hdr, end=" ")
Greg Stein5e0fa402000-06-26 08:28:01 +0000375
Greg Steindd6eefb2000-07-18 09:09:48 +0000376 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000377 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000378 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000379 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000380 self.chunk_left = None
381 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000382 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000383
Greg Steindd6eefb2000-07-18 09:09:48 +0000384 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000385 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000386
Greg Steindd6eefb2000-07-18 09:09:48 +0000387 # do we have a Content-Length?
388 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000389 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000390 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000391
392 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000393 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000394 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000395 try:
396 self.length = int(length)
397 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000398 self.length = None
399 else:
400 if self.length < 0: # ignore nonsensical negative lengths
401 self.length = None
402 else:
403 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000404
Greg Steindd6eefb2000-07-18 09:09:48 +0000405 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000406 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000407 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000408 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000409 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000410
Greg Steindd6eefb2000-07-18 09:09:48 +0000411 # if the connection remains open, and we aren't using chunked, and
412 # a content-length was not provided, then assume that the connection
413 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000414 if (not self.will_close and
415 not self.chunked and
416 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000417 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000418
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000419 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000420 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000421 if self.version == 11:
422 # An HTTP/1.1 proxy is assumed to stay open unless
423 # explicitly closed.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000424 conn = self.headers.get("connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000425 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000426 return True
427 return False
428
Jeremy Hylton2c178252004-08-07 16:28:14 +0000429 # Some HTTP/1.0 implementations have support for persistent
430 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000431
Christian Heimes895627f2007-12-08 17:28:33 +0000432 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000433 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000434 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000435
Jeremy Hylton2c178252004-08-07 16:28:14 +0000436 # At least Akamai returns a "Connection: Keep-Alive" header,
437 # which was supposed to be sent by the client.
438 if conn and "keep-alive" in conn.lower():
439 return False
440
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000441 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000442 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000443 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000444 return False
445
446 # otherwise, assume it will close
447 return True
448
Greg Steindd6eefb2000-07-18 09:09:48 +0000449 def close(self):
450 if self.fp:
451 self.fp.close()
452 self.fp = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000453
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000454 # These implementations are for the benefit of io.BufferedReader.
455
456 # XXX This class should probably be revised to act more like
457 # the "raw stream" that BufferedReader expects.
458
459 @property
460 def closed(self):
461 return self.isclosed()
462
463 def flush(self):
464 self.fp.flush()
465
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000466 def readable(self):
467 return True
468
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000469 # End of "raw stream" methods
470
Greg Steindd6eefb2000-07-18 09:09:48 +0000471 def isclosed(self):
472 # NOTE: it is possible that we will not ever call self.close(). This
473 # case occurs when will_close is TRUE, length is None, and we
474 # read up to the last byte, but NOT past it.
475 #
476 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
477 # called, meaning self.isclosed() is meaningful.
478 return self.fp is None
479
480 def read(self, amt=None):
481 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000482 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000483
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000484 if self._method == "HEAD":
Senthil Kumaran447f82f2010-06-04 16:38:00 +0000485 self.close()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000486 return b""
487
Greg Steindd6eefb2000-07-18 09:09:48 +0000488 if self.chunked:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000489 return self._read_chunked(amt)
Tim Peters230a60c2002-11-09 05:08:07 +0000490
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000491 if amt is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000492 # unbounded read
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000493 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000494 s = self.fp.read()
495 else:
496 s = self._safe_read(self.length)
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000497 self.length = 0
Tim Peters07e99cb2001-01-14 23:47:14 +0000498 self.close() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000499 return s
500
501 if self.length is not None:
502 if amt > self.length:
503 # clip the read to the "end of response"
504 amt = self.length
Greg Steindd6eefb2000-07-18 09:09:48 +0000505
506 # we do not use _safe_read() here because this may be a .will_close
507 # connection, and the user is reading more bytes than will be provided
508 # (for example, reading in 1k chunks)
509 s = self.fp.read(amt)
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000510 if self.length is not None:
511 self.length -= len(s)
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000512 if not self.length:
513 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +0000514 return s
515
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000516 def _read_chunked(self, amt):
517 assert self.chunked != _UNKNOWN
518 chunk_left = self.chunk_left
Chris Withers9c33cbf2009-09-04 17:04:16 +0000519 value = []
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000520 while True:
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000521 if chunk_left is None:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000522 line = self.fp.readline(_MAXLINE + 1)
523 if len(line) > _MAXLINE:
524 raise LineTooLong("chunk size")
Georg Brandl95ba4692008-01-26 09:45:58 +0000525 i = line.find(b";")
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000526 if i >= 0:
527 line = line[:i] # strip chunk-extensions
Christian Heimesa612dc02008-02-24 13:08:18 +0000528 try:
529 chunk_left = int(line, 16)
530 except ValueError:
531 # close the connection as protocol synchronisation is
532 # probably lost
533 self.close()
Chris Withers9c33cbf2009-09-04 17:04:16 +0000534 raise IncompleteRead(b''.join(value))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000535 if chunk_left == 0:
536 break
537 if amt is None:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000538 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000539 elif amt < chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000540 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000541 self.chunk_left = chunk_left - amt
Chris Withers9c33cbf2009-09-04 17:04:16 +0000542 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000543 elif amt == chunk_left:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000544 value.append(self._safe_read(amt))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000545 self._safe_read(2) # toss the CRLF at the end of the chunk
546 self.chunk_left = None
Chris Withers9c33cbf2009-09-04 17:04:16 +0000547 return b''.join(value)
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000548 else:
Chris Withers9c33cbf2009-09-04 17:04:16 +0000549 value.append(self._safe_read(chunk_left))
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000550 amt -= chunk_left
551
552 # we read the whole chunk, get another
553 self._safe_read(2) # toss the CRLF at the end of the chunk
554 chunk_left = None
555
556 # read and discard trailer up to the CRLF terminator
557 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000558 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000559 line = self.fp.readline(_MAXLINE + 1)
560 if len(line) > _MAXLINE:
561 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000562 if not line:
563 # a vanishingly small number of sites EOF without
564 # sending the trailer
565 break
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000566 if line == b"\r\n":
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000567 break
568
569 # we read everything; close the "file"
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000570 self.close()
571
Chris Withers9c33cbf2009-09-04 17:04:16 +0000572 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000573
Greg Steindd6eefb2000-07-18 09:09:48 +0000574 def _safe_read(self, amt):
575 """Read the number of bytes requested, compensating for partial reads.
576
577 Normally, we have a blocking socket, but a read() can be interrupted
578 by a signal (resulting in a partial read).
579
580 Note that we cannot distinguish between EOF and an interrupt when zero
581 bytes have been read. IncompleteRead() will be raised in this
582 situation.
583
584 This function should be used when <amt> bytes "should" be present for
585 reading. If the bytes are truly not available (due to EOF), then the
586 IncompleteRead exception can be used to detect the problem.
587 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000588 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000589 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000590 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000591 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000592 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000593 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000594 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000595 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000596
Antoine Pitroub353c122009-02-11 00:39:14 +0000597 def fileno(self):
598 return self.fp.fileno()
599
Greg Steindd6eefb2000-07-18 09:09:48 +0000600 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000601 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000602 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000603 headers = self.headers.get_all(name) or default
604 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
605 return headers
606 else:
607 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000608
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000609 def getheaders(self):
610 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000611 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000612 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000613 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000614
Antoine Pitroub353c122009-02-11 00:39:14 +0000615 # We override IOBase.__iter__ so that it doesn't check for closed-ness
616
617 def __iter__(self):
618 return self
619
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000620 # For compatibility with old-style urllib responses.
621
622 def info(self):
623 return self.headers
624
625 def geturl(self):
626 return self.url
627
628 def getcode(self):
629 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000630
631class HTTPConnection:
632
Greg Steindd6eefb2000-07-18 09:09:48 +0000633 _http_vsn = 11
634 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000635
Greg Steindd6eefb2000-07-18 09:09:48 +0000636 response_class = HTTPResponse
637 default_port = HTTP_PORT
638 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000639 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000640
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000641 def __init__(self, host, port=None, strict=_strict_sentinel,
Gregory P. Smithb4066372010-01-03 03:28:29 +0000642 timeout=socket._GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000643 if strict is not _strict_sentinel:
644 warnings.warn("the 'strict' argument isn't supported anymore; "
645 "http.client now always assumes HTTP/1.x compliant servers.",
646 DeprecationWarning, 2)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000647 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000648 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000649 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000650 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000651 self.__response = None
652 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000653 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000654 self._tunnel_host = None
655 self._tunnel_port = None
Tim Petersc411dba2002-07-16 21:35:23 +0000656
Greg Steindd6eefb2000-07-18 09:09:48 +0000657 self._set_hostport(host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000658
Senthil Kumaran47fff872009-12-20 07:10:31 +0000659 def set_tunnel(self, host, port=None, headers=None):
660 """ Sets up the host and the port for the HTTP CONNECT Tunnelling.
661
662 The headers argument should be a mapping of extra HTTP headers
663 to send with the CONNECT request.
664 """
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000665 self._tunnel_host = host
666 self._tunnel_port = port
Senthil Kumaran47fff872009-12-20 07:10:31 +0000667 if headers:
668 self._tunnel_headers = headers
669 else:
670 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000671
Greg Steindd6eefb2000-07-18 09:09:48 +0000672 def _set_hostport(self, host, port):
673 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000674 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000675 j = host.rfind(']') # ipv6 addresses have [...]
676 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000677 try:
678 port = int(host[i+1:])
679 except ValueError:
Jeremy Hyltonfbd79942002-07-02 20:19:08 +0000680 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000681 host = host[:i]
682 else:
683 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000684 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000685 host = host[1:-1]
Greg Steindd6eefb2000-07-18 09:09:48 +0000686 self.host = host
687 self.port = port
Greg Stein5e0fa402000-06-26 08:28:01 +0000688
Jeremy Hylton30f86742000-09-18 22:50:38 +0000689 def set_debuglevel(self, level):
690 self.debuglevel = level
691
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000692 def _tunnel(self):
693 self._set_hostport(self._tunnel_host, self._tunnel_port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000694 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" %(self.host, self.port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000695 connect_bytes = connect_str.encode("ascii")
696 self.send(connect_bytes)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000697 for header, value in self._tunnel_headers.iteritems():
698 header_str = "%s: %s\r\n" % (header, value)
699 header_bytes = header_str.encode("ascii")
700 self.send(header_bytes)
701
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000702 response = self.response_class(self.sock, method = self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000703 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000704
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000705 if code != 200:
706 self.close()
707 raise socket.error("Tunnel connection failed: %d %s" % (code,
708 message.strip()))
709 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000710 line = response.fp.readline(_MAXLINE + 1)
711 if len(line) > _MAXLINE:
712 raise LineTooLong("header line")
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000713 if line == b'\r\n':
714 break
715
Greg Steindd6eefb2000-07-18 09:09:48 +0000716 def connect(self):
717 """Connect to the host and port specified in __init__."""
Guido van Rossumd8faa362007-04-27 19:54:29 +0000718 self.sock = socket.create_connection((self.host,self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +0000719 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000720 if self._tunnel_host:
721 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000722
Greg Steindd6eefb2000-07-18 09:09:48 +0000723 def close(self):
724 """Close the connection to the HTTP server."""
725 if self.sock:
Tim Peters07e99cb2001-01-14 23:47:14 +0000726 self.sock.close() # close it manually... there may be other refs
Greg Steindd6eefb2000-07-18 09:09:48 +0000727 self.sock = None
728 if self.__response:
729 self.__response.close()
730 self.__response = None
731 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +0000732
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000733 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000734 """Send `data' to the server.
735 ``data`` can be a string object, a bytes object, an array object, a
736 file-like object that supports a .read() method, or an iterable object.
737 """
738
Greg Steindd6eefb2000-07-18 09:09:48 +0000739 if self.sock is None:
740 if self.auto_open:
741 self.connect()
742 else:
743 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000744
Jeremy Hylton30f86742000-09-18 22:50:38 +0000745 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000746 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000747 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000748 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000749 if self.debuglevel > 0:
750 print("sendIng a read()able")
751 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000752 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000753 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000754 except AttributeError:
755 # io.BytesIO and other file-like objects don't have a `mode`
756 # attribute.
757 pass
758 else:
759 if "b" not in mode:
760 encode = True
761 if self.debuglevel > 0:
762 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000763 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000764 datablock = data.read(blocksize)
765 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000766 break
767 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000768 datablock = datablock.encode("iso-8859-1")
769 self.sock.sendall(datablock)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000770
771 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000772 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000773 except TypeError:
774 if isinstance(data, collections.Iterable):
775 for d in data:
776 self.sock.sendall(d)
777 else:
778 raise TypeError("data should be byte-like object\
779 or an iterable, got %r " % type(it))
Greg Stein5e0fa402000-06-26 08:28:01 +0000780
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000781 def _output(self, s):
782 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000783
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000784 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000785 """
786 self._buffer.append(s)
787
Benjamin Peterson1742e402008-11-30 22:15:29 +0000788 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000789 """Send the currently buffered request and clear the buffer.
790
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000791 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000792 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000793 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000794 self._buffer.extend((b"", b""))
795 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000796 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000797 # If msg and message_body are sent in a single send() call,
798 # it will avoid performance problems caused by the interaction
799 # between delayed ack and the Nagle algorithim.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000800 if isinstance(message_body, bytes):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000801 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000802 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000803 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000804 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000805 # message_body was not a string (i.e. it is a file), and
806 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000807 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000808
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000809 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000810 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000811
Greg Steindd6eefb2000-07-18 09:09:48 +0000812 `method' specifies an HTTP request method, e.g. 'GET'.
813 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000814 `skip_host' if True does not add automatically a 'Host:' header
815 `skip_accept_encoding' if True does not add automatically an
816 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000817 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000818
Greg Stein616a58d2003-06-24 06:35:19 +0000819 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000820 if self.__response and self.__response.isclosed():
821 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000822
Tim Peters58eb11c2004-01-18 20:29:55 +0000823
Greg Steindd6eefb2000-07-18 09:09:48 +0000824 # in certain cases, we cannot issue another request on this connection.
825 # this occurs when:
826 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
827 # 2) a response to a previous request has signalled that it is going
828 # to close the connection upon completion.
829 # 3) the headers for the previous response have not been read, thus
830 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
831 #
832 # if there is no prior response, then we can request at will.
833 #
834 # if point (2) is true, then we will have passed the socket to the
835 # response (effectively meaning, "there is no prior response"), and
836 # will open a new one when a new request is made.
837 #
838 # Note: if a prior response exists, then we *can* start a new request.
839 # We are not allowed to begin fetching the response to this new
840 # request, however, until that prior response is complete.
841 #
842 if self.__state == _CS_IDLE:
843 self.__state = _CS_REQ_STARTED
844 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000845 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +0000846
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000847 # Save the method we use, we need it later in the response phase
848 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +0000849 if not url:
850 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000851 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +0000852
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000853 # Non-ASCII characters should have been eliminated earlier
854 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +0000855
Greg Steindd6eefb2000-07-18 09:09:48 +0000856 if self._http_vsn == 11:
857 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +0000858
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000859 if not skip_host:
860 # this header is issued *only* for HTTP/1.1
861 # connections. more specifically, this means it is
862 # only issued when the client uses the new
863 # HTTPConnection() class. backwards-compat clients
864 # will be using HTTP/1.0 and those clients may be
865 # issuing this header themselves. we should NOT issue
866 # it twice; some web servers (such as Apache) barf
867 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +0000868
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000869 # If we need a non-standard port,include it in the
870 # header. If the request is going through a proxy,
871 # but the host of the actual URL, not the host of the
872 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +0000873
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000874 netloc = ''
875 if url.startswith('http'):
876 nil, netloc, nil, nil, nil = urlsplit(url)
877
878 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000879 try:
880 netloc_enc = netloc.encode("ascii")
881 except UnicodeEncodeError:
882 netloc_enc = netloc.encode("idna")
883 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +0000884 else:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000885 try:
886 host_enc = self.host.encode("ascii")
887 except UnicodeEncodeError:
888 host_enc = self.host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +0000889
890 # As per RFC 273, IPv6 address should be wrapped with []
891 # when used as Host header
892
893 if self.host.find(':') >= 0:
894 host_enc = b'[' + host_enc + b']'
895
Georg Brandl86b2fb92008-07-16 03:43:04 +0000896 if self.port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +0000897 self.putheader('Host', host_enc)
898 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +0000899 host_enc = host_enc.decode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +0000900 self.putheader('Host', "%s:%s" % (host_enc, self.port))
Greg Stein5e0fa402000-06-26 08:28:01 +0000901
Greg Steindd6eefb2000-07-18 09:09:48 +0000902 # note: we are assuming that clients will not attempt to set these
903 # headers since *this* library must deal with the
904 # consequences. this also means that when the supporting
905 # libraries are updated to recognize other forms, then this
906 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +0000907
Greg Steindd6eefb2000-07-18 09:09:48 +0000908 # we only want a Content-Encoding of "identity" since we don't
909 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000910 if not skip_accept_encoding:
911 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +0000912
Greg Steindd6eefb2000-07-18 09:09:48 +0000913 # we can accept "chunked" Transfer-Encodings, but no others
914 # NOTE: no TE header implies *only* "chunked"
915 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +0000916
Greg Steindd6eefb2000-07-18 09:09:48 +0000917 # if TE is supplied in the header, then it must appear in a
918 # Connection header.
919 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +0000920
Greg Steindd6eefb2000-07-18 09:09:48 +0000921 else:
922 # For HTTP/1.0, the server will assume "not chunked"
923 pass
Greg Stein5e0fa402000-06-26 08:28:01 +0000924
Benjamin Petersonf608c612008-11-16 18:33:53 +0000925 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +0000926 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000927
Greg Steindd6eefb2000-07-18 09:09:48 +0000928 For example: h.putheader('Accept', 'text/html')
929 """
930 if self.__state != _CS_REQ_STARTED:
931 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +0000932
Guido van Rossum98297ee2007-11-06 21:34:58 +0000933 if hasattr(header, 'encode'):
934 header = header.encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000935 values = list(values)
936 for i, one_value in enumerate(values):
937 if hasattr(one_value, 'encode'):
938 values[i] = one_value.encode('ascii')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +0000939 elif isinstance(one_value, int):
940 values[i] = str(one_value).encode('ascii')
Benjamin Petersonf608c612008-11-16 18:33:53 +0000941 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +0000942 header = header + b': ' + value
943 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +0000944
Benjamin Peterson1742e402008-11-30 22:15:29 +0000945 def endheaders(self, message_body=None):
946 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000947
Benjamin Peterson1742e402008-11-30 22:15:29 +0000948 This method sends the request to the server. The optional
949 message_body argument can be used to pass message body
950 associated with the request. The message body will be sent in
951 the same packet as the message headers if possible. The
952 message_body should be a string.
953 """
Greg Steindd6eefb2000-07-18 09:09:48 +0000954 if self.__state == _CS_REQ_STARTED:
955 self.__state = _CS_REQ_SENT
956 else:
957 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +0000958 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +0000959
Greg Steindd6eefb2000-07-18 09:09:48 +0000960 def request(self, method, url, body=None, headers={}):
961 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +0000962 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000963
Benjamin Peterson1742e402008-11-30 22:15:29 +0000964 def _set_content_length(self, body):
965 # Set the content-length based on the body.
966 thelen = None
967 try:
968 thelen = str(len(body))
969 except TypeError as te:
970 # If this is a file-like object, try to
971 # fstat its file descriptor
Benjamin Peterson1742e402008-11-30 22:15:29 +0000972 try:
973 thelen = str(os.fstat(body.fileno()).st_size)
974 except (AttributeError, OSError):
975 # Don't send a length if this failed
976 if self.debuglevel > 0: print("Cannot stat!!")
977
978 if thelen is not None:
979 self.putheader('Content-Length', thelen)
980
Greg Steindd6eefb2000-07-18 09:09:48 +0000981 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +0000982 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +0000983 header_names = dict.fromkeys([k.lower() for k in headers])
984 skips = {}
985 if 'host' in header_names:
986 skips['skip_host'] = 1
987 if 'accept-encoding' in header_names:
988 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +0000989
Jeremy Hylton2c178252004-08-07 16:28:14 +0000990 self.putrequest(method, url, **skips)
991
992 if body and ('content-length' not in header_names):
Benjamin Peterson1742e402008-11-30 22:15:29 +0000993 self._set_content_length(body)
Guido van Rossumcc2b0162007-02-11 06:12:03 +0000994 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +0000995 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +0000996 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +0000997 # RFC 2616 Section 3.7.1 says that text default has a
998 # default charset of iso-8859-1.
999 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001000 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001001
Greg Steindd6eefb2000-07-18 09:09:48 +00001002 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001003 """Get the response from the server.
1004
1005 If the HTTPConnection is in the correct state, returns an
1006 instance of HTTPResponse or of whatever object is returned by
1007 class the response_class variable.
1008
1009 If a request has not been sent or if a previous response has
1010 not be handled, ResponseNotReady is raised. If the HTTP
1011 response indicates that the connection should be closed, then
1012 it will be closed before the response is returned. When the
1013 connection is closed, the underlying socket is closed.
1014 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001015
Greg Stein616a58d2003-06-24 06:35:19 +00001016 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001017 if self.__response and self.__response.isclosed():
1018 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001019
Greg Steindd6eefb2000-07-18 09:09:48 +00001020 # if a prior response exists, then it must be completed (otherwise, we
1021 # cannot read this response's header to determine the connection-close
1022 # behavior)
1023 #
1024 # note: if a prior response existed, but was connection-close, then the
1025 # socket and response were made independent of this HTTPConnection
1026 # object since a new request requires that we open a whole new
1027 # connection
1028 #
1029 # this means the prior response had one of two states:
1030 # 1) will_close: this connection was reset and the prior socket and
1031 # response operate independently
1032 # 2) persistent: the response was retained and we await its
1033 # isclosed() status to become true.
1034 #
1035 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001036 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001037
Jeremy Hylton30f86742000-09-18 22:50:38 +00001038 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001039 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001040 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001041 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001042 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001043
Jeremy Hylton39c03802002-07-12 14:04:09 +00001044 response.begin()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +00001045 assert response.will_close != _UNKNOWN
Greg Steindd6eefb2000-07-18 09:09:48 +00001046 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001047
Greg Steindd6eefb2000-07-18 09:09:48 +00001048 if response.will_close:
Guido van Rossumd8faa362007-04-27 19:54:29 +00001049 # this effectively passes the connection to the response
1050 self.close()
Greg Steindd6eefb2000-07-18 09:09:48 +00001051 else:
1052 # remember this, so we can tell when it is complete
1053 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001054
Greg Steindd6eefb2000-07-18 09:09:48 +00001055 return response
Greg Stein5e0fa402000-06-26 08:28:01 +00001056
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001057try:
1058 import ssl
1059except ImportError:
1060 pass
1061else:
1062 class HTTPSConnection(HTTPConnection):
1063 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001064
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001065 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001066
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001067 # XXX Should key_file and cert_file be deprecated in favour of context?
1068
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001069 def __init__(self, host, port=None, key_file=None, cert_file=None,
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001070 strict=_strict_sentinel, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001071 source_address=None, *, context=None, check_hostname=None):
Gregory P. Smithb4066372010-01-03 03:28:29 +00001072 super(HTTPSConnection, self).__init__(host, port, strict, timeout,
1073 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001074 self.key_file = key_file
1075 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001076 if context is None:
1077 # Some reasonable defaults
1078 context = ssl.SSLContext(ssl.PROTOCOL_SSLv23)
1079 context.options |= ssl.OP_NO_SSLv2
1080 will_verify = context.verify_mode != ssl.CERT_NONE
1081 if check_hostname is None:
1082 check_hostname = will_verify
1083 elif check_hostname and not will_verify:
1084 raise ValueError("check_hostname needs a SSL context with "
1085 "either CERT_OPTIONAL or CERT_REQUIRED")
1086 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001087 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001088 self._context = context
1089 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001090
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001091 def connect(self):
1092 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001093
Jeremy Hylton636950f2009-03-28 04:34:21 +00001094 sock = socket.create_connection((self.host, self.port),
Gregory P. Smithb4066372010-01-03 03:28:29 +00001095 self.timeout, self.source_address)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001096
1097 if self._tunnel_host:
1098 self.sock = sock
1099 self._tunnel()
1100
Antoine Pitroud5323212010-10-22 18:19:07 +00001101 server_hostname = self.host if ssl.HAS_SNI else None
1102 self.sock = self._context.wrap_socket(sock,
1103 server_hostname=server_hostname)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001104 try:
1105 if self._check_hostname:
1106 ssl.match_hostname(self.sock.getpeercert(), self.host)
1107 except Exception:
1108 self.sock.shutdown(socket.SHUT_RDWR)
1109 self.sock.close()
1110 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001111
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001112 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001113
Greg Stein5e0fa402000-06-26 08:28:01 +00001114class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001115 # Subclasses that define an __init__ must call Exception.__init__
1116 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001117 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001118
1119class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001120 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001121
Skip Montanaro9d389972002-03-24 16:53:50 +00001122class InvalidURL(HTTPException):
1123 pass
1124
Greg Stein5e0fa402000-06-26 08:28:01 +00001125class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001126 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001127 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001128 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001129
1130class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001131 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001132
Greg Stein5e0fa402000-06-26 08:28:01 +00001133class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001134 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001135
1136class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001137 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001138 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001139 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001140 self.expected = expected
1141 def __repr__(self):
1142 if self.expected is not None:
1143 e = ', %i more expected' % self.expected
1144 else:
1145 e = ''
1146 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1147 def __str__(self):
1148 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001149
1150class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001151 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001152
1153class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001154 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001155
1156class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001157 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001158
1159class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001160 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001161
1162class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001163 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001164 if not line:
1165 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001166 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001167 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001168
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001169class LineTooLong(HTTPException):
1170 def __init__(self, line_type):
1171 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1172 % (_MAXLINE, line_type))
1173
Greg Stein5e0fa402000-06-26 08:28:01 +00001174# for backwards compatibility
1175error = HTTPException