blob: b339f20fc957a08432c3e5720c5e0889d835e05a [file] [log] [blame]
R David Murray44b548d2016-09-08 13:59:53 -04001r"""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
R David Murraycae7bdb2015-04-05 19:26:29 -040023 |\_____________________________
24 | | getresponse() raises
25 | response = getresponse() | ConnectionError
26 v v
27 Unread-response Idle
28 [Response-headers-read]
Greg Stein5e0fa402000-06-26 08:28:01 +000029 |\____________________
Tim Peters5ceadc82001-01-13 19:16:21 +000030 | |
31 | response.read() | putrequest()
32 v v
33 Idle Req-started-unread-response
34 ______/|
35 / |
36 response.read() | | ( putheader() )* endheaders()
37 v v
38 Request-started Req-sent-unread-response
39 |
40 | response.read()
41 v
42 Request-sent
Greg Stein5e0fa402000-06-26 08:28:01 +000043
44This diagram presents the following rules:
45 -- a second request may not be started until {response-headers-read}
46 -- a response [object] cannot be retrieved until {request-sent}
47 -- there is no differentiation between an unread response body and a
48 partially read response body
49
50Note: this enforcement is applied by the HTTPConnection class. The
51 HTTPResponse class does not enforce this state machine, which
52 implies sophisticated clients may accelerate the request/response
53 pipeline. Caution should be taken, though: accelerating the states
54 beyond the above pattern may imply knowledge of the server's
55 connection-close behavior for certain requests. For example, it
56 is impossible to tell whether the server will close the connection
57 UNTIL the response headers have been read; this means that further
58 requests cannot be placed into the pipeline until it is known that
59 the server will NOT be closing the connection.
60
61Logical State __state __response
62------------- ------- ----------
63Idle _CS_IDLE None
64Request-started _CS_REQ_STARTED None
65Request-sent _CS_REQ_SENT None
66Unread-response _CS_IDLE <response_class>
67Req-started-unread-response _CS_REQ_STARTED <response_class>
68Req-sent-unread-response _CS_REQ_SENT <response_class>
Guido van Rossum41999c11997-12-09 00:12:23 +000069"""
Guido van Rossum23acc951994-02-21 16:36:04 +000070
Barry Warsaw820c1202008-06-12 04:06:45 +000071import email.parser
72import email.message
Serhiy Storchakae4db7692014-12-23 16:28:28 +020073import http
Jeremy Hylton636950f2009-03-28 04:34:21 +000074import io
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +020075import re
Jeremy Hylton636950f2009-03-28 04:34:21 +000076import socket
Serhiy Storchaka2e576f52017-04-24 09:05:00 +030077import collections.abc
Jeremy Hylton1afc1692008-06-18 20:49:58 +000078from urllib.parse import urlsplit
Guido van Rossum23acc951994-02-21 16:36:04 +000079
Berker Peksagbabc6882015-02-20 09:39:38 +020080# HTTPMessage, parse_headers(), and the HTTP status code constants are
81# intentionally omitted for simplicity
Thomas Wouters47b49bf2007-08-30 22:15:33 +000082__all__ = ["HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000083 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000084 "UnknownTransferEncoding", "UnimplementedFileMode",
85 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
86 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
R David Murraycae7bdb2015-04-05 19:26:29 -040087 "BadStatusLine", "LineTooLong", "RemoteDisconnected", "error",
88 "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000089
Guido van Rossum23acc951994-02-21 16:36:04 +000090HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000091HTTPS_PORT = 443
92
Greg Stein5e0fa402000-06-26 08:28:01 +000093_UNKNOWN = 'UNKNOWN'
94
95# connection states
96_CS_IDLE = 'Idle'
97_CS_REQ_STARTED = 'Request-started'
98_CS_REQ_SENT = 'Request-sent'
99
Martin v. Löwis39a31782004-09-18 09:03:49 +0000100
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200101# hack to maintain backwards compatibility
102globals().update(http.HTTPStatus.__members__)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000103
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200104# another hack to maintain backwards compatibility
Georg Brandl6aab16e2006-02-17 19:17:25 +0000105# Mapping status codes to official W3C names
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200106responses = {v: v.phrase for v in http.HTTPStatus.__members__.values()}
Georg Brandl6aab16e2006-02-17 19:17:25 +0000107
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000108# maximal line length when calling readline().
109_MAXLINE = 65536
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100110_MAXHEADERS = 100
111
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200112# Header name/value ABNF (http://tools.ietf.org/html/rfc7230#section-3.2)
113#
114# VCHAR = %x21-7E
115# obs-text = %x80-FF
116# header-field = field-name ":" OWS field-value OWS
117# field-name = token
118# field-value = *( field-content / obs-fold )
119# field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
120# field-vchar = VCHAR / obs-text
121#
122# obs-fold = CRLF 1*( SP / HTAB )
123# ; obsolete line folding
124# ; see Section 3.2.4
125
126# token = 1*tchar
127#
128# tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*"
129# / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~"
130# / DIGIT / ALPHA
131# ; any VCHAR, except delimiters
132#
133# VCHAR defined in http://tools.ietf.org/html/rfc5234#appendix-B.1
134
Raymond Hettinger15f44ab2016-08-30 10:47:49 -0700135# the patterns for both name and value are more lenient than RFC
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200136# definitions to allow for backwards compatibility
137_is_legal_header_name = re.compile(rb'[^:\s][^:\r\n]*').fullmatch
138_is_illegal_header_value = re.compile(rb'\n(?![ \t])|\r(?![ \t\n])').search
139
Gregory P. Smithc4e671e2019-04-30 19:12:21 -0700140# These characters are not allowed within HTTP URL paths.
141# See https://tools.ietf.org/html/rfc3986#section-3.3 and the
142# https://tools.ietf.org/html/rfc3986#appendix-A pchar definition.
143# Prevents CVE-2019-9740. Includes control characters such as \r\n.
144# We don't restrict chars above \x7f as putrequest() limits us to ASCII.
145_contains_disallowed_url_pchar_re = re.compile('[\x00-\x20\x7f]')
146# Arguably only these _should_ allowed:
147# _is_allowed_url_pchars_re = re.compile(r"^[/!$&'()*+,;=:@%a-zA-Z0-9._~-]+$")
148# We are more lenient for assumed real world compatibility purposes.
149
AMIR8ca8a2e2020-07-19 00:46:10 +0430150# These characters are not allowed within HTTP method names
151# to prevent http header injection.
152_contains_disallowed_method_pchar_re = re.compile('[\x00-\x1f]')
153
R David Murraybeed8402015-03-22 15:18:23 -0400154# We always set the Content-Length header for these methods because some
155# servers will otherwise respond with a 411
156_METHODS_EXPECTING_BODY = {'PATCH', 'POST', 'PUT'}
157
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000158
Martin Panter44391482016-02-09 10:20:52 +0000159def _encode(data, name='data'):
160 """Call data.encode("latin-1") but show a better error message."""
161 try:
162 return data.encode("latin-1")
163 except UnicodeEncodeError as err:
164 raise UnicodeEncodeError(
165 err.encoding,
166 err.object,
167 err.start,
168 err.end,
169 "%s (%.20r) is not valid Latin-1. Use %s.encode('utf-8') "
170 "if you want to send it encoded in UTF-8." %
171 (name.title(), data[err.start:err.end], name)) from None
172
173
Barry Warsaw820c1202008-06-12 04:06:45 +0000174class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000175 # XXX The only usage of this method is in
176 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
177 # that it doesn't need to be part of the public API. The API has
178 # never been defined so this could cause backwards compatibility
179 # issues.
180
Barry Warsaw820c1202008-06-12 04:06:45 +0000181 def getallmatchingheaders(self, name):
182 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000183
Barry Warsaw820c1202008-06-12 04:06:45 +0000184 Look through the list of headers and find all lines matching a given
185 header name (and their continuation lines). A list of the lines is
186 returned, without interpretation. If the header does not occur, an
187 empty list is returned. If the header occurs multiple times, all
188 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000189
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000190 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000191 name = name.lower() + ':'
192 n = len(name)
193 lst = []
194 hit = 0
195 for line in self.keys():
196 if line[:n].lower() == name:
197 hit = 1
198 elif not line[:1].isspace():
199 hit = 0
200 if hit:
201 lst.append(line)
202 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000203
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000204def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000205 """Parses only RFC2822 headers from a file pointer.
206
207 email Parser wants to see strings rather than bytes.
208 But a TextIOWrapper around self.rfile would buffer too many bytes
209 from the stream, bytes which we later need to read as bytes.
210 So we read the correct bytes here, as bytes, for email Parser
211 to parse.
212
213 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000214 headers = []
215 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000216 line = fp.readline(_MAXLINE + 1)
217 if len(line) > _MAXLINE:
218 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000219 headers.append(line)
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100220 if len(headers) > _MAXHEADERS:
221 raise HTTPException("got more than %d headers" % _MAXHEADERS)
Barry Warsaw820c1202008-06-12 04:06:45 +0000222 if line in (b'\r\n', b'\n', b''):
223 break
224 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000225 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000226
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000227
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000228class HTTPResponse(io.BufferedIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000229
230 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
231
Jeremy Hylton811fc142007-08-03 13:30:02 +0000232 # The bytes from the socket object are iso-8859-1 strings.
233 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
234 # text following RFC 2047. The basic status line parsing only
235 # accepts iso-8859-1.
236
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700237 def __init__(self, sock, debuglevel=0, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000238 # If the response includes a content-length header, we need to
239 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000240 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000241 # the server times out and closes the connection. This will
242 # happen if a self.fp.read() is done (without a size) whether
243 # self.fp is buffered or not. So, no self.fp.read() by
244 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000245 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000246 self.debuglevel = debuglevel
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000247 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000248
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000249 # The HTTPResponse object is returned via urllib. The clients
250 # of http and urllib expect different attributes for the
251 # headers. headers is used here and supports urllib. msg is
252 # provided as a backwards compatibility layer for http
253 # clients.
254
255 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000256
Greg Steindd6eefb2000-07-18 09:09:48 +0000257 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000258 self.version = _UNKNOWN # HTTP-Version
259 self.status = _UNKNOWN # Status-Code
260 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000261
Tim Peters07e99cb2001-01-14 23:47:14 +0000262 self.chunked = _UNKNOWN # is "chunked" being used?
263 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
264 self.length = _UNKNOWN # number of bytes left in response
265 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000266
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000267 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000268 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
269 if len(line) > _MAXLINE:
270 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000271 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000272 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000273 if not line:
274 # Presumably, the server closed the connection before
275 # sending a valid response.
R David Murraycae7bdb2015-04-05 19:26:29 -0400276 raise RemoteDisconnected("Remote end closed connection without"
277 " response")
Greg Steindd6eefb2000-07-18 09:09:48 +0000278 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000279 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000280 except ValueError:
281 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000282 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000283 reason = ""
284 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000285 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000286 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000287 if not version.startswith("HTTP/"):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200288 self._close_conn()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000289 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000290
Jeremy Hylton23d40472001-04-13 14:57:08 +0000291 # The status code is a three-digit number
292 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000293 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000294 if status < 100 or status > 999:
295 raise BadStatusLine(line)
296 except ValueError:
297 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000298 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000299
Jeremy Hylton39c03802002-07-12 14:04:09 +0000300 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000301 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000302 # we've already started reading the response
303 return
304
305 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000306 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000307 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000308 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000309 break
310 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000311 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000312 skip = self.fp.readline(_MAXLINE + 1)
313 if len(skip) > _MAXLINE:
314 raise LineTooLong("header line")
315 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000316 if not skip:
317 break
318 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000319 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000320
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000321 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000322 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000323 if version in ("HTTP/1.0", "HTTP/0.9"):
324 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000325 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000326 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000327 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000328 else:
329 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000330
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000331 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000332
Jeremy Hylton30f86742000-09-18 22:50:38 +0000333 if self.debuglevel > 0:
Matt Houglum461c4162019-04-03 21:36:47 -0700334 for hdr, val in self.headers.items():
335 print("header:", hdr + ":", val)
Greg Stein5e0fa402000-06-26 08:28:01 +0000336
Greg Steindd6eefb2000-07-18 09:09:48 +0000337 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000338 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000339 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000340 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000341 self.chunk_left = None
342 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000343 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000344
Greg Steindd6eefb2000-07-18 09:09:48 +0000345 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000346 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000347
Greg Steindd6eefb2000-07-18 09:09:48 +0000348 # do we have a Content-Length?
349 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000350 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000351 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000352
353 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000354 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000355 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000356 try:
357 self.length = int(length)
358 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000359 self.length = None
360 else:
361 if self.length < 0: # ignore nonsensical negative lengths
362 self.length = None
363 else:
364 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000365
Greg Steindd6eefb2000-07-18 09:09:48 +0000366 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000367 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000368 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000369 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000370 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000371
Greg Steindd6eefb2000-07-18 09:09:48 +0000372 # if the connection remains open, and we aren't using chunked, and
373 # a content-length was not provided, then assume that the connection
374 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000375 if (not self.will_close and
376 not self.chunked and
377 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000378 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000379
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000380 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000381 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000382 if self.version == 11:
383 # An HTTP/1.1 proxy is assumed to stay open unless
384 # explicitly closed.
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000385 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000386 return True
387 return False
388
Jeremy Hylton2c178252004-08-07 16:28:14 +0000389 # Some HTTP/1.0 implementations have support for persistent
390 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000391
Christian Heimes895627f2007-12-08 17:28:33 +0000392 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000393 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000394 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000395
Jeremy Hylton2c178252004-08-07 16:28:14 +0000396 # At least Akamai returns a "Connection: Keep-Alive" header,
397 # which was supposed to be sent by the client.
398 if conn and "keep-alive" in conn.lower():
399 return False
400
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000401 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000402 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000403 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000404 return False
405
406 # otherwise, assume it will close
407 return True
408
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200409 def _close_conn(self):
410 fp = self.fp
411 self.fp = None
412 fp.close()
413
Greg Steindd6eefb2000-07-18 09:09:48 +0000414 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300415 try:
416 super().close() # set "closed" flag
417 finally:
418 if self.fp:
419 self._close_conn()
Greg Stein5e0fa402000-06-26 08:28:01 +0000420
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000421 # These implementations are for the benefit of io.BufferedReader.
422
423 # XXX This class should probably be revised to act more like
424 # the "raw stream" that BufferedReader expects.
425
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000426 def flush(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200427 super().flush()
428 if self.fp:
429 self.fp.flush()
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000430
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000431 def readable(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700432 """Always returns True"""
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000433 return True
434
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000435 # End of "raw stream" methods
436
Greg Steindd6eefb2000-07-18 09:09:48 +0000437 def isclosed(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200438 """True if the connection is closed."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000439 # NOTE: it is possible that we will not ever call self.close(). This
440 # case occurs when will_close is TRUE, length is None, and we
441 # read up to the last byte, but NOT past it.
442 #
443 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
444 # called, meaning self.isclosed() is meaningful.
445 return self.fp is None
446
447 def read(self, amt=None):
448 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000449 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000450
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000451 if self._method == "HEAD":
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200452 self._close_conn()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000453 return b""
454
Bruce Merry152f0b82020-06-25 08:30:21 +0200455 if self.chunked:
456 return self._read_chunked(amt)
457
Antoine Pitrou38d96432011-12-06 22:33:57 +0100458 if amt is not None:
Bruce Merry152f0b82020-06-25 08:30:21 +0200459 if self.length is not None and amt > self.length:
460 # clip the read to the "end of response"
461 amt = self.length
462 s = self.fp.read(amt)
463 if not s and amt:
464 # Ideally, we would raise IncompleteRead if the content-length
465 # wasn't satisfied, but it might break compatibility.
466 self._close_conn()
467 elif self.length is not None:
468 self.length -= len(s)
469 if not self.length:
470 self._close_conn()
471 return s
Antoine Pitrou38d96432011-12-06 22:33:57 +0100472 else:
473 # Amount is not given (unbounded read) so we must check self.length
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000474 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000475 s = self.fp.read()
476 else:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100477 try:
478 s = self._safe_read(self.length)
479 except IncompleteRead:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200480 self._close_conn()
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100481 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000482 self.length = 0
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200483 self._close_conn() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000484 return s
485
Antoine Pitrou38d96432011-12-06 22:33:57 +0100486 def readinto(self, b):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700487 """Read up to len(b) bytes into bytearray b and return the number
488 of bytes read.
489 """
490
Antoine Pitrou38d96432011-12-06 22:33:57 +0100491 if self.fp is None:
492 return 0
493
494 if self._method == "HEAD":
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200495 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100496 return 0
497
498 if self.chunked:
499 return self._readinto_chunked(b)
500
Greg Steindd6eefb2000-07-18 09:09:48 +0000501 if self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100502 if len(b) > self.length:
Greg Steindd6eefb2000-07-18 09:09:48 +0000503 # clip the read to the "end of response"
Antoine Pitrou38d96432011-12-06 22:33:57 +0100504 b = memoryview(b)[0: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)
Antoine Pitrou38d96432011-12-06 22:33:57 +0100509 n = self.fp.readinto(b)
Serhiy Storchaka1c84ac12013-12-17 21:50:02 +0200510 if not n and b:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100511 # Ideally, we would raise IncompleteRead if the content-length
512 # wasn't satisfied, but it might break compatibility.
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200513 self._close_conn()
Antoine Pitrou6a35e182013-02-02 23:04:56 +0100514 elif self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100515 self.length -= n
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000516 if not self.length:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200517 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100518 return n
Greg Steindd6eefb2000-07-18 09:09:48 +0000519
Antoine Pitrou38d96432011-12-06 22:33:57 +0100520 def _read_next_chunk_size(self):
521 # Read the next chunk size from the file
522 line = self.fp.readline(_MAXLINE + 1)
523 if len(line) > _MAXLINE:
524 raise LineTooLong("chunk size")
525 i = line.find(b";")
526 if i >= 0:
527 line = line[:i] # strip chunk-extensions
528 try:
529 return int(line, 16)
530 except ValueError:
531 # close the connection as protocol synchronisation is
532 # probably lost
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200533 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100534 raise
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000535
Antoine Pitrou38d96432011-12-06 22:33:57 +0100536 def _read_and_discard_trailer(self):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000537 # read and discard trailer up to the CRLF terminator
538 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000539 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000540 line = self.fp.readline(_MAXLINE + 1)
541 if len(line) > _MAXLINE:
542 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000543 if not line:
544 # a vanishingly small number of sites EOF without
545 # sending the trailer
546 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800547 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000548 break
549
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000550 def _get_chunk_left(self):
551 # return self.chunk_left, reading a new chunk if necessary.
552 # chunk_left == 0: at the end of the current chunk, need to close it
553 # chunk_left == None: No current chunk, should read next.
554 # This function returns non-zero or None if the last chunk has
555 # been read.
556 chunk_left = self.chunk_left
557 if not chunk_left: # Can be 0 or None
558 if chunk_left is not None:
Mike53f7a7c2017-12-14 14:04:53 +0300559 # We are at the end of chunk, discard chunk end
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000560 self._safe_read(2) # toss the CRLF at the end of the chunk
561 try:
562 chunk_left = self._read_next_chunk_size()
563 except ValueError:
564 raise IncompleteRead(b'')
565 if chunk_left == 0:
566 # last chunk: 1*("0") [ chunk-extension ] CRLF
567 self._read_and_discard_trailer()
568 # we read everything; close the "file"
569 self._close_conn()
570 chunk_left = None
571 self.chunk_left = chunk_left
572 return chunk_left
573
Bruce Merry152f0b82020-06-25 08:30:21 +0200574 def _read_chunked(self, amt=None):
Antoine Pitrou38d96432011-12-06 22:33:57 +0100575 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100576 value = []
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000577 try:
578 while True:
579 chunk_left = self._get_chunk_left()
580 if chunk_left is None:
581 break
Bruce Merry152f0b82020-06-25 08:30:21 +0200582
583 if amt is not None and amt <= chunk_left:
584 value.append(self._safe_read(amt))
585 self.chunk_left = chunk_left - amt
586 break
587
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000588 value.append(self._safe_read(chunk_left))
Bruce Merry152f0b82020-06-25 08:30:21 +0200589 if amt is not None:
590 amt -= chunk_left
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000591 self.chunk_left = 0
592 return b''.join(value)
593 except IncompleteRead:
594 raise IncompleteRead(b''.join(value))
Tim Peters230a60c2002-11-09 05:08:07 +0000595
Antoine Pitrou38d96432011-12-06 22:33:57 +0100596 def _readinto_chunked(self, b):
597 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100598 total_bytes = 0
599 mvb = memoryview(b)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000600 try:
601 while True:
602 chunk_left = self._get_chunk_left()
603 if chunk_left is None:
604 return total_bytes
Antoine Pitrou4ce6aa42011-12-06 22:34:36 +0100605
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000606 if len(mvb) <= chunk_left:
607 n = self._safe_readinto(mvb)
608 self.chunk_left = chunk_left - n
609 return total_bytes + n
610
611 temp_mvb = mvb[:chunk_left]
Antoine Pitrou38d96432011-12-06 22:33:57 +0100612 n = self._safe_readinto(temp_mvb)
613 mvb = mvb[n:]
614 total_bytes += n
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000615 self.chunk_left = 0
Antoine Pitrou38d96432011-12-06 22:33:57 +0100616
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000617 except IncompleteRead:
618 raise IncompleteRead(bytes(b[0:total_bytes]))
Antoine Pitrou38d96432011-12-06 22:33:57 +0100619
Greg Steindd6eefb2000-07-18 09:09:48 +0000620 def _safe_read(self, amt):
Inada Naokid6bf6f22019-04-06 18:06:19 +0900621 """Read the number of bytes requested.
Greg Steindd6eefb2000-07-18 09:09:48 +0000622
623 This function should be used when <amt> bytes "should" be present for
624 reading. If the bytes are truly not available (due to EOF), then the
625 IncompleteRead exception can be used to detect the problem.
626 """
Inada Naokid6bf6f22019-04-06 18:06:19 +0900627 data = self.fp.read(amt)
628 if len(data) < amt:
629 raise IncompleteRead(data, amt-len(data))
630 return data
Greg Steindd6eefb2000-07-18 09:09:48 +0000631
Antoine Pitrou38d96432011-12-06 22:33:57 +0100632 def _safe_readinto(self, b):
633 """Same as _safe_read, but for reading into a buffer."""
Inada Naokid6bf6f22019-04-06 18:06:19 +0900634 amt = len(b)
635 n = self.fp.readinto(b)
636 if n < amt:
637 raise IncompleteRead(bytes(b[:n]), amt-n)
638 return n
Antoine Pitrou38d96432011-12-06 22:33:57 +0100639
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000640 def read1(self, n=-1):
641 """Read with at most one underlying system call. If at least one
642 byte is buffered, return that instead.
643 """
644 if self.fp is None or self._method == "HEAD":
645 return b""
646 if self.chunked:
647 return self._read1_chunked(n)
Martin Panterce911c32016-03-17 06:42:48 +0000648 if self.length is not None and (n < 0 or n > self.length):
649 n = self.length
Martin Panter741d4942016-10-21 00:52:04 +0000650 result = self.fp.read1(n)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000651 if not result and n:
652 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000653 elif self.length is not None:
654 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000655 return result
656
657 def peek(self, n=-1):
658 # Having this enables IOBase.readline() to read more than one
659 # byte at a time
660 if self.fp is None or self._method == "HEAD":
661 return b""
662 if self.chunked:
663 return self._peek_chunked(n)
664 return self.fp.peek(n)
665
666 def readline(self, limit=-1):
667 if self.fp is None or self._method == "HEAD":
668 return b""
669 if self.chunked:
670 # Fallback to IOBase readline which uses peek() and read()
671 return super().readline(limit)
Martin Panterce911c32016-03-17 06:42:48 +0000672 if self.length is not None and (limit < 0 or limit > self.length):
673 limit = self.length
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000674 result = self.fp.readline(limit)
675 if not result and limit:
676 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000677 elif self.length is not None:
678 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000679 return result
680
681 def _read1_chunked(self, n):
682 # Strictly speaking, _get_chunk_left() may cause more than one read,
683 # but that is ok, since that is to satisfy the chunked protocol.
684 chunk_left = self._get_chunk_left()
685 if chunk_left is None or n == 0:
686 return b''
687 if not (0 <= n <= chunk_left):
688 n = chunk_left # if n is negative or larger than chunk_left
689 read = self.fp.read1(n)
690 self.chunk_left -= len(read)
691 if not read:
692 raise IncompleteRead(b"")
693 return read
694
695 def _peek_chunked(self, n):
696 # Strictly speaking, _get_chunk_left() may cause more than one read,
697 # but that is ok, since that is to satisfy the chunked protocol.
698 try:
699 chunk_left = self._get_chunk_left()
700 except IncompleteRead:
701 return b'' # peek doesn't worry about protocol
702 if chunk_left is None:
703 return b'' # eof
704 # peek is allowed to return more than requested. Just request the
705 # entire chunk, and truncate what we get.
706 return self.fp.peek(chunk_left)[:chunk_left]
707
Antoine Pitroub353c122009-02-11 00:39:14 +0000708 def fileno(self):
709 return self.fp.fileno()
710
Greg Steindd6eefb2000-07-18 09:09:48 +0000711 def getheader(self, name, default=None):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700712 '''Returns the value of the header matching *name*.
713
714 If there are multiple matching headers, the values are
715 combined into a single string separated by commas and spaces.
716
717 If no matching header is found, returns *default* or None if
718 the *default* is not specified.
719
720 If the headers are unknown, raises http.client.ResponseNotReady.
721
722 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000723 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000724 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000725 headers = self.headers.get_all(name) or default
726 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
727 return headers
728 else:
729 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000730
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000731 def getheaders(self):
732 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000733 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000734 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000735 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000736
Antoine Pitroub353c122009-02-11 00:39:14 +0000737 # We override IOBase.__iter__ so that it doesn't check for closed-ness
738
739 def __iter__(self):
740 return self
741
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000742 # For compatibility with old-style urllib responses.
743
744 def info(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700745 '''Returns an instance of the class mimetools.Message containing
746 meta-information associated with the URL.
747
748 When the method is HTTP, these headers are those returned by
749 the server at the head of the retrieved HTML page (including
750 Content-Length and Content-Type).
751
752 When the method is FTP, a Content-Length header will be
753 present if (as is now usual) the server passed back a file
754 length in response to the FTP retrieval request. A
755 Content-Type header will be present if the MIME type can be
756 guessed.
757
758 When the method is local-file, returned headers will include
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200759 a Date representing the file's last-modified time, a
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700760 Content-Length giving file size, and a Content-Type
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200761 containing a guess at the file's type. See also the
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700762 description of the mimetools module.
763
764 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000765 return self.headers
766
767 def geturl(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700768 '''Return the real URL of the page.
769
770 In some cases, the HTTP server redirects a client to another
771 URL. The urlopen() function handles this transparently, but in
772 some cases the caller needs to know which URL the client was
773 redirected to. The geturl() method can be used to get at this
774 redirected URL.
775
776 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000777 return self.url
778
779 def getcode(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700780 '''Return the HTTP status code that was sent with the response,
781 or None if the URL is not an HTTP URL.
782
783 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000784 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000785
786class HTTPConnection:
787
Greg Steindd6eefb2000-07-18 09:09:48 +0000788 _http_vsn = 11
789 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000790
Greg Steindd6eefb2000-07-18 09:09:48 +0000791 response_class = HTTPResponse
792 default_port = HTTP_PORT
793 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000794 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000795
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000796 @staticmethod
797 def _is_textIO(stream):
798 """Test whether a file-like object is a text or a binary stream.
799 """
800 return isinstance(stream, io.TextIOBase)
801
802 @staticmethod
803 def _get_content_length(body, method):
804 """Get the content-length based on the body.
805
Martin Panteref91bb22016-08-27 01:39:26 +0000806 If the body is None, we set Content-Length: 0 for methods that expect
807 a body (RFC 7230, Section 3.3.2). We also set the Content-Length for
808 any method if the body is a str or bytes-like object and not a file.
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000809 """
Martin Panteref91bb22016-08-27 01:39:26 +0000810 if body is None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000811 # do an explicit check for not None here to distinguish
812 # between unset and set but empty
Martin Panteref91bb22016-08-27 01:39:26 +0000813 if method.upper() in _METHODS_EXPECTING_BODY:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000814 return 0
815 else:
816 return None
817
818 if hasattr(body, 'read'):
819 # file-like object.
Martin Panteref91bb22016-08-27 01:39:26 +0000820 return None
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000821
822 try:
823 # does it implement the buffer protocol (bytes, bytearray, array)?
824 mv = memoryview(body)
825 return mv.nbytes
826 except TypeError:
827 pass
828
829 if isinstance(body, str):
830 return len(body)
831
832 return None
833
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700834 def __init__(self, host, port=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Nir Sofferad455cd2017-11-06 23:16:37 +0200835 source_address=None, blocksize=8192):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000836 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000837 self.source_address = source_address
Nir Sofferad455cd2017-11-06 23:16:37 +0200838 self.blocksize = blocksize
Greg Steindd6eefb2000-07-18 09:09:48 +0000839 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000840 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000841 self.__response = None
842 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000843 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000844 self._tunnel_host = None
845 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000846 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000847
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400848 (self.host, self.port) = self._get_hostport(host, port)
849
Ashwin Ramaswami9165add2020-03-14 14:56:06 -0400850 self._validate_host(self.host)
851
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400852 # This is stored as an instance variable to allow unit
853 # tests to replace it with a suitable mockup
854 self._create_connection = socket.create_connection
Greg Stein5e0fa402000-06-26 08:28:01 +0000855
Senthil Kumaran47fff872009-12-20 07:10:31 +0000856 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400857 """Set up host and port for HTTP CONNECT tunnelling.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000858
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400859 In a connection that uses HTTP CONNECT tunneling, the host passed to the
860 constructor is used as a proxy server that relays all communication to
861 the endpoint passed to `set_tunnel`. This done by sending an HTTP
862 CONNECT request to the proxy server when the connection is established.
863
Fernando Toledo711381d2020-12-21 11:06:31 -0300864 This method must be called before the HTTP connection has been
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400865 established.
866
867 The headers argument should be a mapping of extra HTTP headers to send
868 with the CONNECT request.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000869 """
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400870
871 if self.sock:
872 raise RuntimeError("Can't set up tunnel for established connection")
873
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200874 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000875 if headers:
876 self._tunnel_headers = headers
877 else:
878 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000879
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400880 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000881 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000882 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000883 j = host.rfind(']') # ipv6 addresses have [...]
884 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000885 try:
886 port = int(host[i+1:])
887 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200888 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
889 port = self.default_port
890 else:
891 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000892 host = host[:i]
893 else:
894 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000895 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000896 host = host[1:-1]
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400897
898 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000899
Jeremy Hylton30f86742000-09-18 22:50:38 +0000900 def set_debuglevel(self, level):
901 self.debuglevel = level
902
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000903 def _tunnel(self):
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800904 connect = b"CONNECT %s:%d HTTP/1.0\r\n" % (
905 self._tunnel_host.encode("ascii"), self._tunnel_port)
906 headers = [connect]
Georg Brandlc7c199f2011-01-22 22:06:24 +0000907 for header, value in self._tunnel_headers.items():
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800908 headers.append(f"{header}: {value}\r\n".encode("latin-1"))
909 headers.append(b"\r\n")
910 # Making a single send() call instead of one per line encourages
911 # the host OS to use a more optimal packet size instead of
912 # potentially emitting a series of small packets.
913 self.send(b"".join(headers))
914 del headers
Senthil Kumaran47fff872009-12-20 07:10:31 +0000915
Georg Brandlc7c199f2011-01-22 22:06:24 +0000916 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000917 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000918
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200919 if code != http.HTTPStatus.OK:
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000920 self.close()
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800921 raise OSError(f"Tunnel connection failed: {code} {message.strip()}")
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000922 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000923 line = response.fp.readline(_MAXLINE + 1)
924 if len(line) > _MAXLINE:
925 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800926 if not line:
927 # for sites which EOF without sending a trailer
928 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800929 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000930 break
931
Berker Peksagab53ab02015-02-03 12:22:11 +0200932 if self.debuglevel > 0:
933 print('header:', line.decode())
934
Greg Steindd6eefb2000-07-18 09:09:48 +0000935 def connect(self):
936 """Connect to the host and port specified in __init__."""
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500937 self.sock = self._create_connection(
938 (self.host,self.port), self.timeout, self.source_address)
939 self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400940
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000941 if self._tunnel_host:
942 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000943
Greg Steindd6eefb2000-07-18 09:09:48 +0000944 def close(self):
945 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000946 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300947 try:
948 sock = self.sock
949 if sock:
950 self.sock = None
951 sock.close() # close it manually... there may be other refs
952 finally:
953 response = self.__response
954 if response:
955 self.__response = None
956 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000957
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000958 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000959 """Send `data' to the server.
960 ``data`` can be a string object, a bytes object, an array object, a
961 file-like object that supports a .read() method, or an iterable object.
962 """
963
Greg Steindd6eefb2000-07-18 09:09:48 +0000964 if self.sock is None:
965 if self.auto_open:
966 self.connect()
967 else:
968 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000969
Jeremy Hylton30f86742000-09-18 22:50:38 +0000970 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000971 print("send:", repr(data))
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000972 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000973 if self.debuglevel > 0:
974 print("sendIng a read()able")
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000975 encode = self._is_textIO(data)
976 if encode and self.debuglevel > 0:
977 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000978 while 1:
Nir Sofferad455cd2017-11-06 23:16:37 +0200979 datablock = data.read(self.blocksize)
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000980 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000981 break
982 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000983 datablock = datablock.encode("iso-8859-1")
984 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300985 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000986 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000987 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000988 except TypeError:
Serhiy Storchaka2e576f52017-04-24 09:05:00 +0300989 if isinstance(data, collections.abc.Iterable):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000990 for d in data:
991 self.sock.sendall(d)
992 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200993 raise TypeError("data should be a bytes-like object "
994 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000995
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000996 def _output(self, s):
997 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000998
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000999 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001000 """
1001 self._buffer.append(s)
1002
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001003 def _read_readable(self, readable):
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001004 if self.debuglevel > 0:
1005 print("sendIng a read()able")
1006 encode = self._is_textIO(readable)
1007 if encode and self.debuglevel > 0:
1008 print("encoding file using iso-8859-1")
1009 while True:
Nir Sofferad455cd2017-11-06 23:16:37 +02001010 datablock = readable.read(self.blocksize)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001011 if not datablock:
1012 break
1013 if encode:
1014 datablock = datablock.encode("iso-8859-1")
1015 yield datablock
1016
1017 def _send_output(self, message_body=None, encode_chunked=False):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001018 """Send the currently buffered request and clear the buffer.
1019
Jeremy Hyltone3252ec2002-07-16 21:41:43 +00001020 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001021 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001022 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001023 self._buffer.extend((b"", b""))
1024 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001025 del self._buffer[:]
1026 self.send(msg)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001027
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001028 if message_body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001029
1030 # create a consistent interface to message_body
1031 if hasattr(message_body, 'read'):
1032 # Let file-like take precedence over byte-like. This
1033 # is needed to allow the current position of mmap'ed
1034 # files to be taken into account.
1035 chunks = self._read_readable(message_body)
1036 else:
1037 try:
1038 # this is solely to check to see if message_body
1039 # implements the buffer API. it /would/ be easier
1040 # to capture if PyObject_CheckBuffer was exposed
1041 # to Python.
1042 memoryview(message_body)
1043 except TypeError:
1044 try:
1045 chunks = iter(message_body)
1046 except TypeError:
1047 raise TypeError("message_body should be a bytes-like "
1048 "object or an iterable, got %r"
1049 % type(message_body))
1050 else:
1051 # the object implements the buffer interface and
1052 # can be passed directly into socket methods
1053 chunks = (message_body,)
1054
1055 for chunk in chunks:
1056 if not chunk:
1057 if self.debuglevel > 0:
1058 print('Zero length chunk ignored')
1059 continue
1060
1061 if encode_chunked and self._http_vsn == 11:
1062 # chunked encoding
Eric V. Smith451d0e32016-09-09 21:56:20 -04001063 chunk = f'{len(chunk):X}\r\n'.encode('ascii') + chunk \
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001064 + b'\r\n'
1065 self.send(chunk)
1066
1067 if encode_chunked and self._http_vsn == 11:
1068 # end chunked transfer
1069 self.send(b'0\r\n\r\n')
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001070
Senthil Kumaran5dc504c2016-09-08 14:28:01 -07001071 def putrequest(self, method, url, skip_host=False,
1072 skip_accept_encoding=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001073 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001074
Greg Steindd6eefb2000-07-18 09:09:48 +00001075 `method' specifies an HTTP request method, e.g. 'GET'.
1076 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001077 `skip_host' if True does not add automatically a 'Host:' header
1078 `skip_accept_encoding' if True does not add automatically an
1079 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +00001080 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001081
Greg Stein616a58d2003-06-24 06:35:19 +00001082 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001083 if self.__response and self.__response.isclosed():
1084 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001085
Tim Peters58eb11c2004-01-18 20:29:55 +00001086
Greg Steindd6eefb2000-07-18 09:09:48 +00001087 # in certain cases, we cannot issue another request on this connection.
1088 # this occurs when:
1089 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
1090 # 2) a response to a previous request has signalled that it is going
1091 # to close the connection upon completion.
1092 # 3) the headers for the previous response have not been read, thus
1093 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
1094 #
1095 # if there is no prior response, then we can request at will.
1096 #
1097 # if point (2) is true, then we will have passed the socket to the
1098 # response (effectively meaning, "there is no prior response"), and
1099 # will open a new one when a new request is made.
1100 #
1101 # Note: if a prior response exists, then we *can* start a new request.
1102 # We are not allowed to begin fetching the response to this new
1103 # request, however, until that prior response is complete.
1104 #
1105 if self.__state == _CS_IDLE:
1106 self.__state = _CS_REQ_STARTED
1107 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001108 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001109
AMIR8ca8a2e2020-07-19 00:46:10 +04301110 self._validate_method(method)
1111
Jason R. Coombs7774d782019-09-28 08:32:01 -04001112 # Save the method for use later in the response phase
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001113 self._method = method
Jason R. Coombs7774d782019-09-28 08:32:01 -04001114
1115 url = url or '/'
1116 self._validate_path(url)
1117
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001118 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001119
Jason R. Coombs7774d782019-09-28 08:32:01 -04001120 self._output(self._encode_request(request))
Greg Stein5e0fa402000-06-26 08:28:01 +00001121
Greg Steindd6eefb2000-07-18 09:09:48 +00001122 if self._http_vsn == 11:
1123 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001124
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001125 if not skip_host:
1126 # this header is issued *only* for HTTP/1.1
1127 # connections. more specifically, this means it is
1128 # only issued when the client uses the new
1129 # HTTPConnection() class. backwards-compat clients
1130 # will be using HTTP/1.0 and those clients may be
1131 # issuing this header themselves. we should NOT issue
1132 # it twice; some web servers (such as Apache) barf
1133 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001134
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001135 # If we need a non-standard port,include it in the
1136 # header. If the request is going through a proxy,
1137 # but the host of the actual URL, not the host of the
1138 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001139
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001140 netloc = ''
1141 if url.startswith('http'):
1142 nil, netloc, nil, nil, nil = urlsplit(url)
1143
1144 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001145 try:
1146 netloc_enc = netloc.encode("ascii")
1147 except UnicodeEncodeError:
1148 netloc_enc = netloc.encode("idna")
1149 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001150 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001151 if self._tunnel_host:
1152 host = self._tunnel_host
1153 port = self._tunnel_port
1154 else:
1155 host = self.host
1156 port = self.port
1157
Thomas Wouters477c8d52006-05-27 19:21:47 +00001158 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001159 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001160 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001161 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001162
1163 # As per RFC 273, IPv6 address should be wrapped with []
1164 # when used as Host header
1165
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001166 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001167 host_enc = b'[' + host_enc + b']'
1168
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001169 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001170 self.putheader('Host', host_enc)
1171 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001172 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001173 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001174
Greg Steindd6eefb2000-07-18 09:09:48 +00001175 # note: we are assuming that clients will not attempt to set these
1176 # headers since *this* library must deal with the
1177 # consequences. this also means that when the supporting
1178 # libraries are updated to recognize other forms, then this
1179 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001180
Greg Steindd6eefb2000-07-18 09:09:48 +00001181 # we only want a Content-Encoding of "identity" since we don't
1182 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001183 if not skip_accept_encoding:
1184 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001185
Greg Steindd6eefb2000-07-18 09:09:48 +00001186 # we can accept "chunked" Transfer-Encodings, but no others
1187 # NOTE: no TE header implies *only* "chunked"
1188 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001189
Greg Steindd6eefb2000-07-18 09:09:48 +00001190 # if TE is supplied in the header, then it must appear in a
1191 # Connection header.
1192 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001193
Greg Steindd6eefb2000-07-18 09:09:48 +00001194 else:
1195 # For HTTP/1.0, the server will assume "not chunked"
1196 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001197
Jason R. Coombs7774d782019-09-28 08:32:01 -04001198 def _encode_request(self, request):
1199 # ASCII also helps prevent CVE-2019-9740.
1200 return request.encode('ascii')
1201
AMIR8ca8a2e2020-07-19 00:46:10 +04301202 def _validate_method(self, method):
1203 """Validate a method name for putrequest."""
1204 # prevent http header injection
1205 match = _contains_disallowed_method_pchar_re.search(method)
1206 if match:
1207 raise ValueError(
1208 f"method can't contain control characters. {method!r} "
1209 f"(found at least {match.group()!r})")
1210
Jason R. Coombs7774d782019-09-28 08:32:01 -04001211 def _validate_path(self, url):
1212 """Validate a url for putrequest."""
1213 # Prevent CVE-2019-9740.
1214 match = _contains_disallowed_url_pchar_re.search(url)
1215 if match:
1216 raise InvalidURL(f"URL can't contain control characters. {url!r} "
1217 f"(found at least {match.group()!r})")
1218
Ashwin Ramaswami9165add2020-03-14 14:56:06 -04001219 def _validate_host(self, host):
1220 """Validate a host so it doesn't contain control characters."""
1221 # Prevent CVE-2019-18348.
1222 match = _contains_disallowed_url_pchar_re.search(host)
1223 if match:
1224 raise InvalidURL(f"URL can't contain control characters. {host!r} "
1225 f"(found at least {match.group()!r})")
1226
Benjamin Petersonf608c612008-11-16 18:33:53 +00001227 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001228 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001229
Greg Steindd6eefb2000-07-18 09:09:48 +00001230 For example: h.putheader('Accept', 'text/html')
1231 """
1232 if self.__state != _CS_REQ_STARTED:
1233 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001234
Guido van Rossum98297ee2007-11-06 21:34:58 +00001235 if hasattr(header, 'encode'):
1236 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001237
1238 if not _is_legal_header_name(header):
1239 raise ValueError('Invalid header name %r' % (header,))
1240
Benjamin Petersonf608c612008-11-16 18:33:53 +00001241 values = list(values)
1242 for i, one_value in enumerate(values):
1243 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001244 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001245 elif isinstance(one_value, int):
1246 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001247
1248 if _is_illegal_header_value(values[i]):
1249 raise ValueError('Invalid header value %r' % (values[i],))
1250
Benjamin Petersonf608c612008-11-16 18:33:53 +00001251 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001252 header = header + b': ' + value
1253 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001254
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001255 def endheaders(self, message_body=None, *, encode_chunked=False):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001256 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001257
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001258 This method sends the request to the server. The optional message_body
1259 argument can be used to pass a message body associated with the
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001260 request.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001261 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001262 if self.__state == _CS_REQ_STARTED:
1263 self.__state = _CS_REQ_SENT
1264 else:
1265 raise CannotSendHeader()
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001266 self._send_output(message_body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001267
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001268 def request(self, method, url, body=None, headers={}, *,
1269 encode_chunked=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001270 """Send a complete request to the server."""
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001271 self._send_request(method, url, body, headers, encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001272
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001273 def _send_request(self, method, url, body, headers, encode_chunked):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001274 # Honor explicitly requested Host: and Accept-Encoding: headers.
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001275 header_names = frozenset(k.lower() for k in headers)
Jeremy Hylton2c178252004-08-07 16:28:14 +00001276 skips = {}
1277 if 'host' in header_names:
1278 skips['skip_host'] = 1
1279 if 'accept-encoding' in header_names:
1280 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001281
Jeremy Hylton2c178252004-08-07 16:28:14 +00001282 self.putrequest(method, url, **skips)
1283
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001284 # chunked encoding will happen if HTTP/1.1 is used and either
1285 # the caller passes encode_chunked=True or the following
1286 # conditions hold:
1287 # 1. content-length has not been explicitly set
Martin Panteref91bb22016-08-27 01:39:26 +00001288 # 2. the body is a file or iterable, but not a str or bytes-like
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001289 # 3. Transfer-Encoding has NOT been explicitly set by the caller
1290
R David Murraybeed8402015-03-22 15:18:23 -04001291 if 'content-length' not in header_names:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001292 # only chunk body if not explicitly set for backwards
1293 # compatibility, assuming the client code is already handling the
1294 # chunking
1295 if 'transfer-encoding' not in header_names:
1296 # if content-length cannot be automatically determined, fall
1297 # back to chunked encoding
1298 encode_chunked = False
1299 content_length = self._get_content_length(body, method)
1300 if content_length is None:
Martin Panteref91bb22016-08-27 01:39:26 +00001301 if body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001302 if self.debuglevel > 0:
1303 print('Unable to determine size of %r' % body)
1304 encode_chunked = True
1305 self.putheader('Transfer-Encoding', 'chunked')
1306 else:
1307 self.putheader('Content-Length', str(content_length))
1308 else:
1309 encode_chunked = False
1310
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001311 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001312 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001313 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001314 # RFC 2616 Section 3.7.1 says that text default has a
1315 # default charset of iso-8859-1.
Martin Panter44391482016-02-09 10:20:52 +00001316 body = _encode(body, 'body')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001317 self.endheaders(body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001318
Greg Steindd6eefb2000-07-18 09:09:48 +00001319 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001320 """Get the response from the server.
1321
1322 If the HTTPConnection is in the correct state, returns an
1323 instance of HTTPResponse or of whatever object is returned by
Martin Pantercc71a792016-04-05 06:19:42 +00001324 the response_class variable.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001325
1326 If a request has not been sent or if a previous response has
1327 not be handled, ResponseNotReady is raised. If the HTTP
1328 response indicates that the connection should be closed, then
1329 it will be closed before the response is returned. When the
1330 connection is closed, the underlying socket is closed.
1331 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001332
Greg Stein616a58d2003-06-24 06:35:19 +00001333 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001334 if self.__response and self.__response.isclosed():
1335 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001336
Greg Steindd6eefb2000-07-18 09:09:48 +00001337 # if a prior response exists, then it must be completed (otherwise, we
1338 # cannot read this response's header to determine the connection-close
1339 # behavior)
1340 #
1341 # note: if a prior response existed, but was connection-close, then the
1342 # socket and response were made independent of this HTTPConnection
1343 # object since a new request requires that we open a whole new
1344 # connection
1345 #
1346 # this means the prior response had one of two states:
1347 # 1) will_close: this connection was reset and the prior socket and
1348 # response operate independently
1349 # 2) persistent: the response was retained and we await its
1350 # isclosed() status to become true.
1351 #
1352 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001353 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001354
Jeremy Hylton30f86742000-09-18 22:50:38 +00001355 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001356 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001357 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001358 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001359 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001360
Serhiy Storchakab491e052014-12-01 13:07:45 +02001361 try:
R David Murraycae7bdb2015-04-05 19:26:29 -04001362 try:
1363 response.begin()
1364 except ConnectionError:
1365 self.close()
1366 raise
Serhiy Storchakab491e052014-12-01 13:07:45 +02001367 assert response.will_close != _UNKNOWN
1368 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001369
Serhiy Storchakab491e052014-12-01 13:07:45 +02001370 if response.will_close:
1371 # this effectively passes the connection to the response
1372 self.close()
1373 else:
1374 # remember this, so we can tell when it is complete
1375 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001376
Serhiy Storchakab491e052014-12-01 13:07:45 +02001377 return response
1378 except:
1379 response.close()
1380 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001381
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001382try:
1383 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001384except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001385 pass
1386else:
1387 class HTTPSConnection(HTTPConnection):
1388 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001389
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001390 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001391
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001392 # XXX Should key_file and cert_file be deprecated in favour of context?
1393
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001394 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001395 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1396 source_address=None, *, context=None,
Nir Sofferad455cd2017-11-06 23:16:37 +02001397 check_hostname=None, blocksize=8192):
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001398 super(HTTPSConnection, self).__init__(host, port, timeout,
Nir Sofferad455cd2017-11-06 23:16:37 +02001399 source_address,
1400 blocksize=blocksize)
Christian Heimesd0486372016-09-10 23:23:33 +02001401 if (key_file is not None or cert_file is not None or
1402 check_hostname is not None):
1403 import warnings
1404 warnings.warn("key_file, cert_file and check_hostname are "
1405 "deprecated, use a custom context instead.",
1406 DeprecationWarning, 2)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001407 self.key_file = key_file
1408 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001409 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001410 context = ssl._create_default_https_context()
Christian Heimesf97406b2020-11-13 16:37:52 +01001411 # send ALPN extension to indicate HTTP/1.1 protocol
1412 if self._http_vsn == 11:
1413 context.set_alpn_protocols(['http/1.1'])
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001414 # enable PHA for TLS 1.3 connections if available
1415 if context.post_handshake_auth is not None:
1416 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001417 will_verify = context.verify_mode != ssl.CERT_NONE
1418 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001419 check_hostname = context.check_hostname
1420 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001421 raise ValueError("check_hostname needs a SSL context with "
1422 "either CERT_OPTIONAL or CERT_REQUIRED")
1423 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001424 context.load_cert_chain(cert_file, key_file)
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001425 # cert and key file means the user wants to authenticate.
1426 # enable TLS 1.3 PHA implicitly even for custom contexts.
1427 if context.post_handshake_auth is not None:
1428 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001429 self._context = context
Christian Heimes61d478c2018-01-27 15:51:38 +01001430 if check_hostname is not None:
1431 self._context.check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001432
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001433 def connect(self):
1434 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001435
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001436 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001437
1438 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001439 server_hostname = self._tunnel_host
1440 else:
1441 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001442
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001443 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001444 server_hostname=server_hostname)
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001445
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001446 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001447
Greg Stein5e0fa402000-06-26 08:28:01 +00001448class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001449 # Subclasses that define an __init__ must call Exception.__init__
1450 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001451 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001452
1453class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001454 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001455
Skip Montanaro9d389972002-03-24 16:53:50 +00001456class InvalidURL(HTTPException):
1457 pass
1458
Greg Stein5e0fa402000-06-26 08:28:01 +00001459class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001460 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001461 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001462 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001463
1464class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001465 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001466
Greg Stein5e0fa402000-06-26 08:28:01 +00001467class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001468 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001469
1470class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001471 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001472 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001473 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001474 self.expected = expected
1475 def __repr__(self):
1476 if self.expected is not None:
1477 e = ', %i more expected' % self.expected
1478 else:
1479 e = ''
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001480 return '%s(%i bytes read%s)' % (self.__class__.__name__,
1481 len(self.partial), e)
Serhiy Storchaka96aeaec2019-05-06 22:29:40 +03001482 __str__ = object.__str__
Greg Stein5e0fa402000-06-26 08:28:01 +00001483
1484class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001485 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001486
1487class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001488 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001489
1490class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001491 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001492
1493class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001494 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001495
1496class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001497 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001498 if not line:
1499 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001500 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001501 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001502
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001503class LineTooLong(HTTPException):
1504 def __init__(self, line_type):
1505 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1506 % (_MAXLINE, line_type))
1507
R David Murraycae7bdb2015-04-05 19:26:29 -04001508class RemoteDisconnected(ConnectionResetError, BadStatusLine):
1509 def __init__(self, *pos, **kw):
1510 BadStatusLine.__init__(self, "")
1511 ConnectionResetError.__init__(self, *pos, **kw)
1512
Greg Stein5e0fa402000-06-26 08:28:01 +00001513# for backwards compatibility
1514error = HTTPException