blob: 1c69dcb5c734af7a3e51e0206d85de500727800f [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
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +020073import re
Jeremy Hylton636950f2009-03-28 04:34:21 +000074import socket
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000075import collections
Jeremy Hylton1afc1692008-06-18 20:49:58 +000076from urllib.parse import urlsplit
Guido van Rossum23acc951994-02-21 16:36:04 +000077
Berker Peksagbabc6882015-02-20 09:39:38 +020078# HTTPMessage, parse_headers(), and the HTTP status code constants are
79# intentionally omitted for simplicity
Thomas Wouters47b49bf2007-08-30 22:15:33 +000080__all__ = ["HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000081 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000082 "UnknownTransferEncoding", "UnimplementedFileMode",
83 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
84 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
Berker Peksagbabc6882015-02-20 09:39:38 +020085 "BadStatusLine", "LineTooLong", "error", "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000086
Guido van Rossum23acc951994-02-21 16:36:04 +000087HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000088HTTPS_PORT = 443
89
Greg Stein5e0fa402000-06-26 08:28:01 +000090_UNKNOWN = 'UNKNOWN'
91
92# connection states
93_CS_IDLE = 'Idle'
94_CS_REQ_STARTED = 'Request-started'
95_CS_REQ_SENT = 'Request-sent'
96
Martin v. Löwis39a31782004-09-18 09:03:49 +000097# status codes
98# informational
99CONTINUE = 100
100SWITCHING_PROTOCOLS = 101
101PROCESSING = 102
102
103# successful
104OK = 200
105CREATED = 201
106ACCEPTED = 202
107NON_AUTHORITATIVE_INFORMATION = 203
108NO_CONTENT = 204
109RESET_CONTENT = 205
110PARTIAL_CONTENT = 206
111MULTI_STATUS = 207
112IM_USED = 226
113
114# redirection
115MULTIPLE_CHOICES = 300
116MOVED_PERMANENTLY = 301
117FOUND = 302
118SEE_OTHER = 303
119NOT_MODIFIED = 304
120USE_PROXY = 305
121TEMPORARY_REDIRECT = 307
122
123# client error
124BAD_REQUEST = 400
125UNAUTHORIZED = 401
126PAYMENT_REQUIRED = 402
127FORBIDDEN = 403
128NOT_FOUND = 404
129METHOD_NOT_ALLOWED = 405
130NOT_ACCEPTABLE = 406
131PROXY_AUTHENTICATION_REQUIRED = 407
132REQUEST_TIMEOUT = 408
133CONFLICT = 409
134GONE = 410
135LENGTH_REQUIRED = 411
136PRECONDITION_FAILED = 412
137REQUEST_ENTITY_TOO_LARGE = 413
138REQUEST_URI_TOO_LONG = 414
139UNSUPPORTED_MEDIA_TYPE = 415
140REQUESTED_RANGE_NOT_SATISFIABLE = 416
141EXPECTATION_FAILED = 417
142UNPROCESSABLE_ENTITY = 422
143LOCKED = 423
144FAILED_DEPENDENCY = 424
145UPGRADE_REQUIRED = 426
Hynek Schlawack51b2ed52012-05-16 09:51:07 +0200146PRECONDITION_REQUIRED = 428
147TOO_MANY_REQUESTS = 429
148REQUEST_HEADER_FIELDS_TOO_LARGE = 431
Martin v. Löwis39a31782004-09-18 09:03:49 +0000149
150# server error
151INTERNAL_SERVER_ERROR = 500
152NOT_IMPLEMENTED = 501
153BAD_GATEWAY = 502
154SERVICE_UNAVAILABLE = 503
155GATEWAY_TIMEOUT = 504
156HTTP_VERSION_NOT_SUPPORTED = 505
157INSUFFICIENT_STORAGE = 507
158NOT_EXTENDED = 510
Hynek Schlawack51b2ed52012-05-16 09:51:07 +0200159NETWORK_AUTHENTICATION_REQUIRED = 511
Martin v. Löwis39a31782004-09-18 09:03:49 +0000160
Georg Brandl6aab16e2006-02-17 19:17:25 +0000161# Mapping status codes to official W3C names
162responses = {
163 100: 'Continue',
164 101: 'Switching Protocols',
165
166 200: 'OK',
167 201: 'Created',
168 202: 'Accepted',
169 203: 'Non-Authoritative Information',
170 204: 'No Content',
171 205: 'Reset Content',
172 206: 'Partial Content',
173
174 300: 'Multiple Choices',
175 301: 'Moved Permanently',
176 302: 'Found',
177 303: 'See Other',
178 304: 'Not Modified',
179 305: 'Use Proxy',
180 306: '(Unused)',
181 307: 'Temporary Redirect',
182
183 400: 'Bad Request',
184 401: 'Unauthorized',
185 402: 'Payment Required',
186 403: 'Forbidden',
187 404: 'Not Found',
188 405: 'Method Not Allowed',
189 406: 'Not Acceptable',
190 407: 'Proxy Authentication Required',
191 408: 'Request Timeout',
192 409: 'Conflict',
193 410: 'Gone',
194 411: 'Length Required',
195 412: 'Precondition Failed',
196 413: 'Request Entity Too Large',
197 414: 'Request-URI Too Long',
198 415: 'Unsupported Media Type',
199 416: 'Requested Range Not Satisfiable',
200 417: 'Expectation Failed',
Hynek Schlawack51b2ed52012-05-16 09:51:07 +0200201 428: 'Precondition Required',
202 429: 'Too Many Requests',
203 431: 'Request Header Fields Too Large',
Georg Brandl6aab16e2006-02-17 19:17:25 +0000204
205 500: 'Internal Server Error',
206 501: 'Not Implemented',
207 502: 'Bad Gateway',
208 503: 'Service Unavailable',
209 504: 'Gateway Timeout',
210 505: 'HTTP Version Not Supported',
Hynek Schlawack51b2ed52012-05-16 09:51:07 +0200211 511: 'Network Authentication Required',
Georg Brandl6aab16e2006-02-17 19:17:25 +0000212}
213
Georg Brandl80ba8e82005-09-29 20:16:07 +0000214# maximal amount of data to read at one time in _safe_read
215MAXAMOUNT = 1048576
216
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000217# maximal line length when calling readline().
218_MAXLINE = 65536
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100219_MAXHEADERS = 100
220
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200221# Header name/value ABNF (http://tools.ietf.org/html/rfc7230#section-3.2)
222#
223# VCHAR = %x21-7E
224# obs-text = %x80-FF
225# header-field = field-name ":" OWS field-value OWS
226# field-name = token
227# field-value = *( field-content / obs-fold )
228# field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
229# field-vchar = VCHAR / obs-text
230#
231# obs-fold = CRLF 1*( SP / HTAB )
232# ; obsolete line folding
233# ; see Section 3.2.4
234
235# token = 1*tchar
236#
237# tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*"
238# / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~"
239# / DIGIT / ALPHA
240# ; any VCHAR, except delimiters
241#
242# VCHAR defined in http://tools.ietf.org/html/rfc5234#appendix-B.1
243
244# the patterns for both name and value are more leniant than RFC
245# definitions to allow for backwards compatibility
246_is_legal_header_name = re.compile(rb'[^:\s][^:\r\n]*').fullmatch
247_is_illegal_header_value = re.compile(rb'\n(?![ \t])|\r(?![ \t\n])').search
248
R David Murraybeed8402015-03-22 15:18:23 -0400249# We always set the Content-Length header for these methods because some
250# servers will otherwise respond with a 411
251_METHODS_EXPECTING_BODY = {'PATCH', 'POST', 'PUT'}
252
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000253
Barry Warsaw820c1202008-06-12 04:06:45 +0000254class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000255 # XXX The only usage of this method is in
256 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
257 # that it doesn't need to be part of the public API. The API has
258 # never been defined so this could cause backwards compatibility
259 # issues.
260
Barry Warsaw820c1202008-06-12 04:06:45 +0000261 def getallmatchingheaders(self, name):
262 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000263
Barry Warsaw820c1202008-06-12 04:06:45 +0000264 Look through the list of headers and find all lines matching a given
265 header name (and their continuation lines). A list of the lines is
266 returned, without interpretation. If the header does not occur, an
267 empty list is returned. If the header occurs multiple times, all
268 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000269
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000270 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000271 name = name.lower() + ':'
272 n = len(name)
273 lst = []
274 hit = 0
275 for line in self.keys():
276 if line[:n].lower() == name:
277 hit = 1
278 elif not line[:1].isspace():
279 hit = 0
280 if hit:
281 lst.append(line)
282 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000283
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000284def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000285 """Parses only RFC2822 headers from a file pointer.
286
287 email Parser wants to see strings rather than bytes.
288 But a TextIOWrapper around self.rfile would buffer too many bytes
289 from the stream, bytes which we later need to read as bytes.
290 So we read the correct bytes here, as bytes, for email Parser
291 to parse.
292
293 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000294 headers = []
295 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000296 line = fp.readline(_MAXLINE + 1)
297 if len(line) > _MAXLINE:
298 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000299 headers.append(line)
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100300 if len(headers) > _MAXHEADERS:
301 raise HTTPException("got more than %d headers" % _MAXHEADERS)
Barry Warsaw820c1202008-06-12 04:06:45 +0000302 if line in (b'\r\n', b'\n', b''):
303 break
304 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000305 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000306
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000307
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000308class HTTPResponse(io.RawIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000309
310 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
311
Jeremy Hylton811fc142007-08-03 13:30:02 +0000312 # The bytes from the socket object are iso-8859-1 strings.
313 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
314 # text following RFC 2047. The basic status line parsing only
315 # accepts iso-8859-1.
316
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700317 def __init__(self, sock, debuglevel=0, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000318 # If the response includes a content-length header, we need to
319 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000320 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000321 # the server times out and closes the connection. This will
322 # happen if a self.fp.read() is done (without a size) whether
323 # self.fp is buffered or not. So, no self.fp.read() by
324 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000325 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000326 self.debuglevel = debuglevel
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000327 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000328
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000329 # The HTTPResponse object is returned via urllib. The clients
330 # of http and urllib expect different attributes for the
331 # headers. headers is used here and supports urllib. msg is
332 # provided as a backwards compatibility layer for http
333 # clients.
334
335 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000336
Greg Steindd6eefb2000-07-18 09:09:48 +0000337 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000338 self.version = _UNKNOWN # HTTP-Version
339 self.status = _UNKNOWN # Status-Code
340 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000341
Tim Peters07e99cb2001-01-14 23:47:14 +0000342 self.chunked = _UNKNOWN # is "chunked" being used?
343 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
344 self.length = _UNKNOWN # number of bytes left in response
345 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000346
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000347 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000348 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
349 if len(line) > _MAXLINE:
350 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000351 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000352 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000353 if not line:
354 # Presumably, the server closed the connection before
355 # sending a valid response.
356 raise BadStatusLine(line)
Greg Steindd6eefb2000-07-18 09:09:48 +0000357 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000358 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000359 except ValueError:
360 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000361 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000362 reason = ""
363 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000364 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000365 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000366 if not version.startswith("HTTP/"):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200367 self._close_conn()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000368 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000369
Jeremy Hylton23d40472001-04-13 14:57:08 +0000370 # The status code is a three-digit number
371 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000372 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000373 if status < 100 or status > 999:
374 raise BadStatusLine(line)
375 except ValueError:
376 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000377 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000378
Jeremy Hylton39c03802002-07-12 14:04:09 +0000379 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000380 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000381 # we've already started reading the response
382 return
383
384 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000385 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000386 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000387 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000388 break
389 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000390 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000391 skip = self.fp.readline(_MAXLINE + 1)
392 if len(skip) > _MAXLINE:
393 raise LineTooLong("header line")
394 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000395 if not skip:
396 break
397 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000398 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000399
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000400 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000401 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000402 if version in ("HTTP/1.0", "HTTP/0.9"):
403 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000404 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000405 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000406 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000407 else:
408 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000409
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000410 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000411
Jeremy Hylton30f86742000-09-18 22:50:38 +0000412 if self.debuglevel > 0:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000413 for hdr in self.headers:
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000414 print("header:", hdr, end=" ")
Greg Stein5e0fa402000-06-26 08:28:01 +0000415
Greg Steindd6eefb2000-07-18 09:09:48 +0000416 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000417 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000418 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000419 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000420 self.chunk_left = None
421 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000422 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000423
Greg Steindd6eefb2000-07-18 09:09:48 +0000424 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000425 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000426
Greg Steindd6eefb2000-07-18 09:09:48 +0000427 # do we have a Content-Length?
428 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000429 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000430 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000431
432 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000433 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000434 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000435 try:
436 self.length = int(length)
437 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000438 self.length = None
439 else:
440 if self.length < 0: # ignore nonsensical negative lengths
441 self.length = None
442 else:
443 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000444
Greg Steindd6eefb2000-07-18 09:09:48 +0000445 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000446 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000447 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000448 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000449 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000450
Greg Steindd6eefb2000-07-18 09:09:48 +0000451 # if the connection remains open, and we aren't using chunked, and
452 # a content-length was not provided, then assume that the connection
453 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000454 if (not self.will_close and
455 not self.chunked and
456 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000457 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000458
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000459 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000460 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000461 if self.version == 11:
462 # An HTTP/1.1 proxy is assumed to stay open unless
463 # explicitly closed.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000464 conn = self.headers.get("connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000465 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000466 return True
467 return False
468
Jeremy Hylton2c178252004-08-07 16:28:14 +0000469 # Some HTTP/1.0 implementations have support for persistent
470 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000471
Christian Heimes895627f2007-12-08 17:28:33 +0000472 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000473 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000474 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000475
Jeremy Hylton2c178252004-08-07 16:28:14 +0000476 # At least Akamai returns a "Connection: Keep-Alive" header,
477 # which was supposed to be sent by the client.
478 if conn and "keep-alive" in conn.lower():
479 return False
480
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000481 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000482 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000483 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000484 return False
485
486 # otherwise, assume it will close
487 return True
488
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200489 def _close_conn(self):
490 fp = self.fp
491 self.fp = None
492 fp.close()
493
Greg Steindd6eefb2000-07-18 09:09:48 +0000494 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300495 try:
496 super().close() # set "closed" flag
497 finally:
498 if self.fp:
499 self._close_conn()
Greg Stein5e0fa402000-06-26 08:28:01 +0000500
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000501 # These implementations are for the benefit of io.BufferedReader.
502
503 # XXX This class should probably be revised to act more like
504 # the "raw stream" that BufferedReader expects.
505
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000506 def flush(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200507 super().flush()
508 if self.fp:
509 self.fp.flush()
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000510
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000511 def readable(self):
512 return True
513
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000514 # End of "raw stream" methods
515
Greg Steindd6eefb2000-07-18 09:09:48 +0000516 def isclosed(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200517 """True if the connection is closed."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000518 # NOTE: it is possible that we will not ever call self.close(). This
519 # case occurs when will_close is TRUE, length is None, and we
520 # read up to the last byte, but NOT past it.
521 #
522 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
523 # called, meaning self.isclosed() is meaningful.
524 return self.fp is None
525
526 def read(self, amt=None):
527 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000528 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000529
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000530 if self._method == "HEAD":
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200531 self._close_conn()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000532 return b""
533
Antoine Pitrou38d96432011-12-06 22:33:57 +0100534 if amt is not None:
535 # Amount is given, so call base class version
536 # (which is implemented in terms of self.readinto)
537 return super(HTTPResponse, self).read(amt)
538 else:
539 # Amount is not given (unbounded read) so we must check self.length
540 # and self.chunked
Tim Peters230a60c2002-11-09 05:08:07 +0000541
Antoine Pitrou38d96432011-12-06 22:33:57 +0100542 if self.chunked:
543 return self._readall_chunked()
544
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000545 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000546 s = self.fp.read()
547 else:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100548 try:
549 s = self._safe_read(self.length)
550 except IncompleteRead:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200551 self._close_conn()
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100552 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000553 self.length = 0
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200554 self._close_conn() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000555 return s
556
Antoine Pitrou38d96432011-12-06 22:33:57 +0100557 def readinto(self, b):
558 if self.fp is None:
559 return 0
560
561 if self._method == "HEAD":
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200562 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100563 return 0
564
565 if self.chunked:
566 return self._readinto_chunked(b)
567
Greg Steindd6eefb2000-07-18 09:09:48 +0000568 if self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100569 if len(b) > self.length:
Greg Steindd6eefb2000-07-18 09:09:48 +0000570 # clip the read to the "end of response"
Antoine Pitrou38d96432011-12-06 22:33:57 +0100571 b = memoryview(b)[0:self.length]
Greg Steindd6eefb2000-07-18 09:09:48 +0000572
573 # we do not use _safe_read() here because this may be a .will_close
574 # connection, and the user is reading more bytes than will be provided
575 # (for example, reading in 1k chunks)
Antoine Pitrou38d96432011-12-06 22:33:57 +0100576 n = self.fp.readinto(b)
Serhiy Storchaka1c84ac12013-12-17 21:50:02 +0200577 if not n and b:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100578 # Ideally, we would raise IncompleteRead if the content-length
579 # wasn't satisfied, but it might break compatibility.
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200580 self._close_conn()
Antoine Pitrou6a35e182013-02-02 23:04:56 +0100581 elif self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100582 self.length -= n
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000583 if not self.length:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200584 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100585 return n
Greg Steindd6eefb2000-07-18 09:09:48 +0000586
Antoine Pitrou38d96432011-12-06 22:33:57 +0100587 def _read_next_chunk_size(self):
588 # Read the next chunk size from the file
589 line = self.fp.readline(_MAXLINE + 1)
590 if len(line) > _MAXLINE:
591 raise LineTooLong("chunk size")
592 i = line.find(b";")
593 if i >= 0:
594 line = line[:i] # strip chunk-extensions
595 try:
596 return int(line, 16)
597 except ValueError:
598 # close the connection as protocol synchronisation is
599 # probably lost
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200600 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100601 raise
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000602
Antoine Pitrou38d96432011-12-06 22:33:57 +0100603 def _read_and_discard_trailer(self):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000604 # read and discard trailer up to the CRLF terminator
605 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000606 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000607 line = self.fp.readline(_MAXLINE + 1)
608 if len(line) > _MAXLINE:
609 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000610 if not line:
611 # a vanishingly small number of sites EOF without
612 # sending the trailer
613 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800614 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000615 break
616
Antoine Pitrou38d96432011-12-06 22:33:57 +0100617 def _readall_chunked(self):
618 assert self.chunked != _UNKNOWN
619 chunk_left = self.chunk_left
620 value = []
621 while True:
622 if chunk_left is None:
623 try:
624 chunk_left = self._read_next_chunk_size()
625 if chunk_left == 0:
626 break
627 except ValueError:
628 raise IncompleteRead(b''.join(value))
629 value.append(self._safe_read(chunk_left))
630
631 # we read the whole chunk, get another
632 self._safe_read(2) # toss the CRLF at the end of the chunk
633 chunk_left = None
634
635 self._read_and_discard_trailer()
636
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000637 # we read everything; close the "file"
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200638 self._close_conn()
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000639
Chris Withers9c33cbf2009-09-04 17:04:16 +0000640 return b''.join(value)
Tim Peters230a60c2002-11-09 05:08:07 +0000641
Antoine Pitrou38d96432011-12-06 22:33:57 +0100642 def _readinto_chunked(self, b):
643 assert self.chunked != _UNKNOWN
644 chunk_left = self.chunk_left
645
646 total_bytes = 0
647 mvb = memoryview(b)
648 while True:
649 if chunk_left is None:
650 try:
651 chunk_left = self._read_next_chunk_size()
652 if chunk_left == 0:
653 break
654 except ValueError:
655 raise IncompleteRead(bytes(b[0:total_bytes]))
Antoine Pitrou4ce6aa42011-12-06 22:34:36 +0100656
Antoine Pitrou38d96432011-12-06 22:33:57 +0100657 if len(mvb) < chunk_left:
658 n = self._safe_readinto(mvb)
659 self.chunk_left = chunk_left - n
Antoine Pitrouf7e78182012-01-04 18:57:22 +0100660 return total_bytes + n
Antoine Pitrou38d96432011-12-06 22:33:57 +0100661 elif len(mvb) == chunk_left:
662 n = self._safe_readinto(mvb)
663 self._safe_read(2) # toss the CRLF at the end of the chunk
664 self.chunk_left = None
Antoine Pitrouf7e78182012-01-04 18:57:22 +0100665 return total_bytes + n
Antoine Pitrou38d96432011-12-06 22:33:57 +0100666 else:
667 temp_mvb = mvb[0:chunk_left]
668 n = self._safe_readinto(temp_mvb)
669 mvb = mvb[n:]
670 total_bytes += n
671
672 # we read the whole chunk, get another
673 self._safe_read(2) # toss the CRLF at the end of the chunk
674 chunk_left = None
675
676 self._read_and_discard_trailer()
677
678 # we read everything; close the "file"
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200679 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100680
681 return total_bytes
682
Greg Steindd6eefb2000-07-18 09:09:48 +0000683 def _safe_read(self, amt):
684 """Read the number of bytes requested, compensating for partial reads.
685
686 Normally, we have a blocking socket, but a read() can be interrupted
687 by a signal (resulting in a partial read).
688
689 Note that we cannot distinguish between EOF and an interrupt when zero
690 bytes have been read. IncompleteRead() will be raised in this
691 situation.
692
693 This function should be used when <amt> bytes "should" be present for
694 reading. If the bytes are truly not available (due to EOF), then the
695 IncompleteRead exception can be used to detect the problem.
696 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000697 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000698 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000699 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000700 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000701 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000702 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000703 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000704 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000705
Antoine Pitrou38d96432011-12-06 22:33:57 +0100706 def _safe_readinto(self, b):
707 """Same as _safe_read, but for reading into a buffer."""
708 total_bytes = 0
709 mvb = memoryview(b)
710 while total_bytes < len(b):
711 if MAXAMOUNT < len(mvb):
712 temp_mvb = mvb[0:MAXAMOUNT]
713 n = self.fp.readinto(temp_mvb)
714 else:
715 n = self.fp.readinto(mvb)
716 if not n:
717 raise IncompleteRead(bytes(mvb[0:total_bytes]), len(b))
718 mvb = mvb[n:]
719 total_bytes += n
720 return total_bytes
721
Antoine Pitroub353c122009-02-11 00:39:14 +0000722 def fileno(self):
723 return self.fp.fileno()
724
Greg Steindd6eefb2000-07-18 09:09:48 +0000725 def getheader(self, name, default=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000726 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000727 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000728 headers = self.headers.get_all(name) or default
729 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
730 return headers
731 else:
732 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000733
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000734 def getheaders(self):
735 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000736 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000737 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000738 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000739
Antoine Pitroub353c122009-02-11 00:39:14 +0000740 # We override IOBase.__iter__ so that it doesn't check for closed-ness
741
742 def __iter__(self):
743 return self
744
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000745 # For compatibility with old-style urllib responses.
746
747 def info(self):
748 return self.headers
749
750 def geturl(self):
751 return self.url
752
753 def getcode(self):
754 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000755
756class HTTPConnection:
757
Greg Steindd6eefb2000-07-18 09:09:48 +0000758 _http_vsn = 11
759 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000760
Greg Steindd6eefb2000-07-18 09:09:48 +0000761 response_class = HTTPResponse
762 default_port = HTTP_PORT
763 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000764 debuglevel = 0
Antoine Pitrou90e47742013-01-02 22:10:47 +0100765 # TCP Maximum Segment Size (MSS) is determined by the TCP stack on
766 # a per-connection basis. There is no simple and efficient
767 # platform independent mechanism for determining the MSS, so
768 # instead a reasonable estimate is chosen. The getsockopt()
769 # interface using the TCP_MAXSEG parameter may be a suitable
770 # approach on some operating systems. A value of 16KiB is chosen
771 # as a reasonable estimate of the maximum MSS.
772 mss = 16384
Greg Stein5e0fa402000-06-26 08:28:01 +0000773
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700774 def __init__(self, host, port=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
775 source_address=None):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000776 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000777 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000778 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000779 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000780 self.__response = None
781 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000782 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000783 self._tunnel_host = None
784 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000785 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000786
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400787 (self.host, self.port) = self._get_hostport(host, port)
788
789 # This is stored as an instance variable to allow unit
790 # tests to replace it with a suitable mockup
791 self._create_connection = socket.create_connection
Greg Stein5e0fa402000-06-26 08:28:01 +0000792
Senthil Kumaran47fff872009-12-20 07:10:31 +0000793 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400794 """Set up host and port for HTTP CONNECT tunnelling.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000795
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400796 In a connection that uses HTTP CONNECT tunneling, the host passed to the
797 constructor is used as a proxy server that relays all communication to
798 the endpoint passed to `set_tunnel`. This done by sending an HTTP
799 CONNECT request to the proxy server when the connection is established.
800
801 This method must be called before the HTML connection has been
802 established.
803
804 The headers argument should be a mapping of extra HTTP headers to send
805 with the CONNECT request.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000806 """
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400807
808 if self.sock:
809 raise RuntimeError("Can't set up tunnel for established connection")
810
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200811 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000812 if headers:
813 self._tunnel_headers = headers
814 else:
815 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000816
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400817 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000818 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000819 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000820 j = host.rfind(']') # ipv6 addresses have [...]
821 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000822 try:
823 port = int(host[i+1:])
824 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200825 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
826 port = self.default_port
827 else:
828 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000829 host = host[:i]
830 else:
831 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000832 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000833 host = host[1:-1]
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400834
835 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000836
Jeremy Hylton30f86742000-09-18 22:50:38 +0000837 def set_debuglevel(self, level):
838 self.debuglevel = level
839
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000840 def _tunnel(self):
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200841 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host,
842 self._tunnel_port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000843 connect_bytes = connect_str.encode("ascii")
844 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000845 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000846 header_str = "%s: %s\r\n" % (header, value)
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000847 header_bytes = header_str.encode("latin-1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000848 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000849 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000850
Georg Brandlc7c199f2011-01-22 22:06:24 +0000851 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000852 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000853
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000854 if code != 200:
855 self.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +0200856 raise OSError("Tunnel connection failed: %d %s" % (code,
857 message.strip()))
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000858 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000859 line = response.fp.readline(_MAXLINE + 1)
860 if len(line) > _MAXLINE:
861 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800862 if not line:
863 # for sites which EOF without sending a trailer
864 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800865 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000866 break
867
Greg Steindd6eefb2000-07-18 09:09:48 +0000868 def connect(self):
869 """Connect to the host and port specified in __init__."""
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400870 self.sock = self._create_connection((self.host,self.port),
871 self.timeout, self.source_address)
872
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000873 if self._tunnel_host:
874 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000875
Greg Steindd6eefb2000-07-18 09:09:48 +0000876 def close(self):
877 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000878 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300879 try:
880 sock = self.sock
881 if sock:
882 self.sock = None
883 sock.close() # close it manually... there may be other refs
884 finally:
885 response = self.__response
886 if response:
887 self.__response = None
888 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000889
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000890 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000891 """Send `data' to the server.
892 ``data`` can be a string object, a bytes object, an array object, a
893 file-like object that supports a .read() method, or an iterable object.
894 """
895
Greg Steindd6eefb2000-07-18 09:09:48 +0000896 if self.sock is None:
897 if self.auto_open:
898 self.connect()
899 else:
900 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000901
Jeremy Hylton30f86742000-09-18 22:50:38 +0000902 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000903 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000904 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000905 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000906 if self.debuglevel > 0:
907 print("sendIng a read()able")
908 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000909 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000910 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000911 except AttributeError:
912 # io.BytesIO and other file-like objects don't have a `mode`
913 # attribute.
914 pass
915 else:
916 if "b" not in mode:
917 encode = True
918 if self.debuglevel > 0:
919 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000920 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000921 datablock = data.read(blocksize)
922 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000923 break
924 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000925 datablock = datablock.encode("iso-8859-1")
926 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300927 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000928 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000929 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000930 except TypeError:
931 if isinstance(data, collections.Iterable):
932 for d in data:
933 self.sock.sendall(d)
934 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200935 raise TypeError("data should be a bytes-like object "
936 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000937
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000938 def _output(self, s):
939 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000940
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000941 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000942 """
943 self._buffer.append(s)
944
Benjamin Peterson1742e402008-11-30 22:15:29 +0000945 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000946 """Send the currently buffered request and clear the buffer.
947
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000948 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000949 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000950 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000951 self._buffer.extend((b"", b""))
952 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000953 del self._buffer[:]
Benjamin Peterson1742e402008-11-30 22:15:29 +0000954 # If msg and message_body are sent in a single send() call,
955 # it will avoid performance problems caused by the interaction
Antoine Pitrou90e47742013-01-02 22:10:47 +0100956 # between delayed ack and the Nagle algorithm. However,
957 # there is no performance gain if the message is larger
958 # than MSS (and there is a memory penalty for the message
959 # copy).
960 if isinstance(message_body, bytes) and len(message_body) < self.mss:
Benjamin Peterson1742e402008-11-30 22:15:29 +0000961 msg += message_body
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000962 message_body = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000963 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000964 if message_body is not None:
Jeremy Hylton236654b2009-03-27 20:24:34 +0000965 # message_body was not a string (i.e. it is a file), and
966 # we must run the risk of Nagle.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000967 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000968
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000969 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000970 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000971
Greg Steindd6eefb2000-07-18 09:09:48 +0000972 `method' specifies an HTTP request method, e.g. 'GET'.
973 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000974 `skip_host' if True does not add automatically a 'Host:' header
975 `skip_accept_encoding' if True does not add automatically an
976 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000977 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000978
Greg Stein616a58d2003-06-24 06:35:19 +0000979 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000980 if self.__response and self.__response.isclosed():
981 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000982
Tim Peters58eb11c2004-01-18 20:29:55 +0000983
Greg Steindd6eefb2000-07-18 09:09:48 +0000984 # in certain cases, we cannot issue another request on this connection.
985 # this occurs when:
986 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
987 # 2) a response to a previous request has signalled that it is going
988 # to close the connection upon completion.
989 # 3) the headers for the previous response have not been read, thus
990 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
991 #
992 # if there is no prior response, then we can request at will.
993 #
994 # if point (2) is true, then we will have passed the socket to the
995 # response (effectively meaning, "there is no prior response"), and
996 # will open a new one when a new request is made.
997 #
998 # Note: if a prior response exists, then we *can* start a new request.
999 # We are not allowed to begin fetching the response to this new
1000 # request, however, until that prior response is complete.
1001 #
1002 if self.__state == _CS_IDLE:
1003 self.__state = _CS_REQ_STARTED
1004 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001005 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001006
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001007 # Save the method we use, we need it later in the response phase
1008 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +00001009 if not url:
1010 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001011 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001012
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001013 # Non-ASCII characters should have been eliminated earlier
1014 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +00001015
Greg Steindd6eefb2000-07-18 09:09:48 +00001016 if self._http_vsn == 11:
1017 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001018
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001019 if not skip_host:
1020 # this header is issued *only* for HTTP/1.1
1021 # connections. more specifically, this means it is
1022 # only issued when the client uses the new
1023 # HTTPConnection() class. backwards-compat clients
1024 # will be using HTTP/1.0 and those clients may be
1025 # issuing this header themselves. we should NOT issue
1026 # it twice; some web servers (such as Apache) barf
1027 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001028
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001029 # If we need a non-standard port,include it in the
1030 # header. If the request is going through a proxy,
1031 # but the host of the actual URL, not the host of the
1032 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001033
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001034 netloc = ''
1035 if url.startswith('http'):
1036 nil, netloc, nil, nil, nil = urlsplit(url)
1037
1038 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001039 try:
1040 netloc_enc = netloc.encode("ascii")
1041 except UnicodeEncodeError:
1042 netloc_enc = netloc.encode("idna")
1043 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001044 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001045 if self._tunnel_host:
1046 host = self._tunnel_host
1047 port = self._tunnel_port
1048 else:
1049 host = self.host
1050 port = self.port
1051
Thomas Wouters477c8d52006-05-27 19:21:47 +00001052 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001053 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001054 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001055 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001056
1057 # As per RFC 273, IPv6 address should be wrapped with []
1058 # when used as Host header
1059
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001060 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001061 host_enc = b'[' + host_enc + b']'
1062
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001063 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001064 self.putheader('Host', host_enc)
1065 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001066 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001067 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001068
Greg Steindd6eefb2000-07-18 09:09:48 +00001069 # note: we are assuming that clients will not attempt to set these
1070 # headers since *this* library must deal with the
1071 # consequences. this also means that when the supporting
1072 # libraries are updated to recognize other forms, then this
1073 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001074
Greg Steindd6eefb2000-07-18 09:09:48 +00001075 # we only want a Content-Encoding of "identity" since we don't
1076 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001077 if not skip_accept_encoding:
1078 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001079
Greg Steindd6eefb2000-07-18 09:09:48 +00001080 # we can accept "chunked" Transfer-Encodings, but no others
1081 # NOTE: no TE header implies *only* "chunked"
1082 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001083
Greg Steindd6eefb2000-07-18 09:09:48 +00001084 # if TE is supplied in the header, then it must appear in a
1085 # Connection header.
1086 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001087
Greg Steindd6eefb2000-07-18 09:09:48 +00001088 else:
1089 # For HTTP/1.0, the server will assume "not chunked"
1090 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001091
Benjamin Petersonf608c612008-11-16 18:33:53 +00001092 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001093 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001094
Greg Steindd6eefb2000-07-18 09:09:48 +00001095 For example: h.putheader('Accept', 'text/html')
1096 """
1097 if self.__state != _CS_REQ_STARTED:
1098 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001099
Guido van Rossum98297ee2007-11-06 21:34:58 +00001100 if hasattr(header, 'encode'):
1101 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001102
1103 if not _is_legal_header_name(header):
1104 raise ValueError('Invalid header name %r' % (header,))
1105
Benjamin Petersonf608c612008-11-16 18:33:53 +00001106 values = list(values)
1107 for i, one_value in enumerate(values):
1108 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001109 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001110 elif isinstance(one_value, int):
1111 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001112
1113 if _is_illegal_header_value(values[i]):
1114 raise ValueError('Invalid header value %r' % (values[i],))
1115
Benjamin Petersonf608c612008-11-16 18:33:53 +00001116 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001117 header = header + b': ' + value
1118 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001119
Benjamin Peterson1742e402008-11-30 22:15:29 +00001120 def endheaders(self, message_body=None):
1121 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001122
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001123 This method sends the request to the server. The optional message_body
1124 argument can be used to pass a message body associated with the
1125 request. The message body will be sent in the same packet as the
1126 message headers if it is a string, otherwise it is sent as a separate
1127 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001128 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001129 if self.__state == _CS_REQ_STARTED:
1130 self.__state = _CS_REQ_SENT
1131 else:
1132 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +00001133 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001134
Greg Steindd6eefb2000-07-18 09:09:48 +00001135 def request(self, method, url, body=None, headers={}):
1136 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +00001137 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +00001138
R David Murraybeed8402015-03-22 15:18:23 -04001139 def _set_content_length(self, body, method):
1140 # Set the content-length based on the body. If the body is "empty", we
1141 # set Content-Length: 0 for methods that expect a body (RFC 7230,
1142 # Section 3.3.2). If the body is set for other methods, we set the
1143 # header provided we can figure out what the length is.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001144 thelen = None
R David Murraybeed8402015-03-22 15:18:23 -04001145 method_expects_body = method.upper() in _METHODS_EXPECTING_BODY
1146 if body is None and method_expects_body:
1147 thelen = '0'
1148 elif body is not None:
Benjamin Peterson1742e402008-11-30 22:15:29 +00001149 try:
R David Murraybeed8402015-03-22 15:18:23 -04001150 thelen = str(len(body))
1151 except TypeError:
1152 # If this is a file-like object, try to
1153 # fstat its file descriptor
1154 try:
1155 thelen = str(os.fstat(body.fileno()).st_size)
1156 except (AttributeError, OSError):
1157 # Don't send a length if this failed
1158 if self.debuglevel > 0: print("Cannot stat!!")
Benjamin Peterson1742e402008-11-30 22:15:29 +00001159
1160 if thelen is not None:
1161 self.putheader('Content-Length', thelen)
1162
Greg Steindd6eefb2000-07-18 09:09:48 +00001163 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001164 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +00001165 header_names = dict.fromkeys([k.lower() for k in headers])
1166 skips = {}
1167 if 'host' in header_names:
1168 skips['skip_host'] = 1
1169 if 'accept-encoding' in header_names:
1170 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001171
Jeremy Hylton2c178252004-08-07 16:28:14 +00001172 self.putrequest(method, url, **skips)
1173
R David Murraybeed8402015-03-22 15:18:23 -04001174 if 'content-length' not in header_names:
1175 self._set_content_length(body, method)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001176 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001177 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001178 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001179 # RFC 2616 Section 3.7.1 says that text default has a
1180 # default charset of iso-8859-1.
1181 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001182 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001183
Greg Steindd6eefb2000-07-18 09:09:48 +00001184 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001185 """Get the response from the server.
1186
1187 If the HTTPConnection is in the correct state, returns an
1188 instance of HTTPResponse or of whatever object is returned by
1189 class the response_class variable.
1190
1191 If a request has not been sent or if a previous response has
1192 not be handled, ResponseNotReady is raised. If the HTTP
1193 response indicates that the connection should be closed, then
1194 it will be closed before the response is returned. When the
1195 connection is closed, the underlying socket is closed.
1196 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001197
Greg Stein616a58d2003-06-24 06:35:19 +00001198 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001199 if self.__response and self.__response.isclosed():
1200 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001201
Greg Steindd6eefb2000-07-18 09:09:48 +00001202 # if a prior response exists, then it must be completed (otherwise, we
1203 # cannot read this response's header to determine the connection-close
1204 # behavior)
1205 #
1206 # note: if a prior response existed, but was connection-close, then the
1207 # socket and response were made independent of this HTTPConnection
1208 # object since a new request requires that we open a whole new
1209 # connection
1210 #
1211 # this means the prior response had one of two states:
1212 # 1) will_close: this connection was reset and the prior socket and
1213 # response operate independently
1214 # 2) persistent: the response was retained and we await its
1215 # isclosed() status to become true.
1216 #
1217 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001218 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001219
Jeremy Hylton30f86742000-09-18 22:50:38 +00001220 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001221 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001222 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001223 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001224 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001225
Serhiy Storchakab491e052014-12-01 13:07:45 +02001226 try:
1227 response.begin()
1228 assert response.will_close != _UNKNOWN
1229 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001230
Serhiy Storchakab491e052014-12-01 13:07:45 +02001231 if response.will_close:
1232 # this effectively passes the connection to the response
1233 self.close()
1234 else:
1235 # remember this, so we can tell when it is complete
1236 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001237
Serhiy Storchakab491e052014-12-01 13:07:45 +02001238 return response
1239 except:
1240 response.close()
1241 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001242
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001243try:
1244 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001245except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001246 pass
1247else:
1248 class HTTPSConnection(HTTPConnection):
1249 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001250
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001251 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001252
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001253 # XXX Should key_file and cert_file be deprecated in favour of context?
1254
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001255 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001256 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1257 source_address=None, *, context=None,
1258 check_hostname=None):
1259 super(HTTPSConnection, self).__init__(host, port, timeout,
Gregory P. Smithb4066372010-01-03 03:28:29 +00001260 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001261 self.key_file = key_file
1262 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001263 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001264 context = ssl._create_default_https_context()
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001265 will_verify = context.verify_mode != ssl.CERT_NONE
1266 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001267 check_hostname = context.check_hostname
1268 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001269 raise ValueError("check_hostname needs a SSL context with "
1270 "either CERT_OPTIONAL or CERT_REQUIRED")
1271 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001272 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001273 self._context = context
1274 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001275
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001276 def connect(self):
1277 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001278
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001279 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001280
1281 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001282 server_hostname = self._tunnel_host
1283 else:
1284 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001285
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001286 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001287 server_hostname=server_hostname)
Christian Heimesf723c712013-12-05 07:51:17 +01001288 if not self._context.check_hostname and self._check_hostname:
1289 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001290 ssl.match_hostname(self.sock.getpeercert(), server_hostname)
Christian Heimesf723c712013-12-05 07:51:17 +01001291 except Exception:
1292 self.sock.shutdown(socket.SHUT_RDWR)
1293 self.sock.close()
1294 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001295
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001296 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001297
Greg Stein5e0fa402000-06-26 08:28:01 +00001298class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001299 # Subclasses that define an __init__ must call Exception.__init__
1300 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001301 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001302
1303class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001304 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001305
Skip Montanaro9d389972002-03-24 16:53:50 +00001306class InvalidURL(HTTPException):
1307 pass
1308
Greg Stein5e0fa402000-06-26 08:28:01 +00001309class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001310 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001311 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001312 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001313
1314class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001315 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001316
Greg Stein5e0fa402000-06-26 08:28:01 +00001317class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001318 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001319
1320class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001321 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001322 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001323 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001324 self.expected = expected
1325 def __repr__(self):
1326 if self.expected is not None:
1327 e = ', %i more expected' % self.expected
1328 else:
1329 e = ''
1330 return 'IncompleteRead(%i bytes read%s)' % (len(self.partial), e)
1331 def __str__(self):
1332 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001333
1334class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001335 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001336
1337class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001338 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001339
1340class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001341 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001342
1343class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001344 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001345
1346class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001347 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001348 if not line:
1349 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001350 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001351 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001352
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001353class LineTooLong(HTTPException):
1354 def __init__(self, line_type):
1355 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1356 % (_MAXLINE, line_type))
1357
Greg Stein5e0fa402000-06-26 08:28:01 +00001358# for backwards compatibility
1359error = HTTPException