blob: a54679cf84d18a3f5db07b1c99722663e23ec78c [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
864 This method must be called before the HTML connection has been
865 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):
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200904 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host,
905 self._tunnel_port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000906 connect_bytes = connect_str.encode("ascii")
907 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000908 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000909 header_str = "%s: %s\r\n" % (header, value)
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000910 header_bytes = header_str.encode("latin-1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000911 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000912 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000913
Georg Brandlc7c199f2011-01-22 22:06:24 +0000914 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000915 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000916
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200917 if code != http.HTTPStatus.OK:
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000918 self.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +0200919 raise OSError("Tunnel connection failed: %d %s" % (code,
920 message.strip()))
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000921 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000922 line = response.fp.readline(_MAXLINE + 1)
923 if len(line) > _MAXLINE:
924 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800925 if not line:
926 # for sites which EOF without sending a trailer
927 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800928 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000929 break
930
Berker Peksagab53ab02015-02-03 12:22:11 +0200931 if self.debuglevel > 0:
932 print('header:', line.decode())
933
Greg Steindd6eefb2000-07-18 09:09:48 +0000934 def connect(self):
935 """Connect to the host and port specified in __init__."""
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500936 self.sock = self._create_connection(
937 (self.host,self.port), self.timeout, self.source_address)
938 self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400939
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000940 if self._tunnel_host:
941 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000942
Greg Steindd6eefb2000-07-18 09:09:48 +0000943 def close(self):
944 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000945 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300946 try:
947 sock = self.sock
948 if sock:
949 self.sock = None
950 sock.close() # close it manually... there may be other refs
951 finally:
952 response = self.__response
953 if response:
954 self.__response = None
955 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000956
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000957 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000958 """Send `data' to the server.
959 ``data`` can be a string object, a bytes object, an array object, a
960 file-like object that supports a .read() method, or an iterable object.
961 """
962
Greg Steindd6eefb2000-07-18 09:09:48 +0000963 if self.sock is None:
964 if self.auto_open:
965 self.connect()
966 else:
967 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000968
Jeremy Hylton30f86742000-09-18 22:50:38 +0000969 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000970 print("send:", repr(data))
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000971 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000972 if self.debuglevel > 0:
973 print("sendIng a read()able")
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000974 encode = self._is_textIO(data)
975 if encode and self.debuglevel > 0:
976 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000977 while 1:
Nir Sofferad455cd2017-11-06 23:16:37 +0200978 datablock = data.read(self.blocksize)
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000979 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000980 break
981 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000982 datablock = datablock.encode("iso-8859-1")
983 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300984 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000985 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000986 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000987 except TypeError:
Serhiy Storchaka2e576f52017-04-24 09:05:00 +0300988 if isinstance(data, collections.abc.Iterable):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000989 for d in data:
990 self.sock.sendall(d)
991 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200992 raise TypeError("data should be a bytes-like object "
993 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000994
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000995 def _output(self, s):
996 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000997
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000998 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000999 """
1000 self._buffer.append(s)
1001
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001002 def _read_readable(self, readable):
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001003 if self.debuglevel > 0:
1004 print("sendIng a read()able")
1005 encode = self._is_textIO(readable)
1006 if encode and self.debuglevel > 0:
1007 print("encoding file using iso-8859-1")
1008 while True:
Nir Sofferad455cd2017-11-06 23:16:37 +02001009 datablock = readable.read(self.blocksize)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001010 if not datablock:
1011 break
1012 if encode:
1013 datablock = datablock.encode("iso-8859-1")
1014 yield datablock
1015
1016 def _send_output(self, message_body=None, encode_chunked=False):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001017 """Send the currently buffered request and clear the buffer.
1018
Jeremy Hyltone3252ec2002-07-16 21:41:43 +00001019 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001020 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001021 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001022 self._buffer.extend((b"", b""))
1023 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001024 del self._buffer[:]
1025 self.send(msg)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001026
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001027 if message_body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001028
1029 # create a consistent interface to message_body
1030 if hasattr(message_body, 'read'):
1031 # Let file-like take precedence over byte-like. This
1032 # is needed to allow the current position of mmap'ed
1033 # files to be taken into account.
1034 chunks = self._read_readable(message_body)
1035 else:
1036 try:
1037 # this is solely to check to see if message_body
1038 # implements the buffer API. it /would/ be easier
1039 # to capture if PyObject_CheckBuffer was exposed
1040 # to Python.
1041 memoryview(message_body)
1042 except TypeError:
1043 try:
1044 chunks = iter(message_body)
1045 except TypeError:
1046 raise TypeError("message_body should be a bytes-like "
1047 "object or an iterable, got %r"
1048 % type(message_body))
1049 else:
1050 # the object implements the buffer interface and
1051 # can be passed directly into socket methods
1052 chunks = (message_body,)
1053
1054 for chunk in chunks:
1055 if not chunk:
1056 if self.debuglevel > 0:
1057 print('Zero length chunk ignored')
1058 continue
1059
1060 if encode_chunked and self._http_vsn == 11:
1061 # chunked encoding
Eric V. Smith451d0e32016-09-09 21:56:20 -04001062 chunk = f'{len(chunk):X}\r\n'.encode('ascii') + chunk \
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001063 + b'\r\n'
1064 self.send(chunk)
1065
1066 if encode_chunked and self._http_vsn == 11:
1067 # end chunked transfer
1068 self.send(b'0\r\n\r\n')
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001069
Senthil Kumaran5dc504c2016-09-08 14:28:01 -07001070 def putrequest(self, method, url, skip_host=False,
1071 skip_accept_encoding=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001072 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001073
Greg Steindd6eefb2000-07-18 09:09:48 +00001074 `method' specifies an HTTP request method, e.g. 'GET'.
1075 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001076 `skip_host' if True does not add automatically a 'Host:' header
1077 `skip_accept_encoding' if True does not add automatically an
1078 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +00001079 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001080
Greg Stein616a58d2003-06-24 06:35:19 +00001081 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001082 if self.__response and self.__response.isclosed():
1083 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001084
Tim Peters58eb11c2004-01-18 20:29:55 +00001085
Greg Steindd6eefb2000-07-18 09:09:48 +00001086 # in certain cases, we cannot issue another request on this connection.
1087 # this occurs when:
1088 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
1089 # 2) a response to a previous request has signalled that it is going
1090 # to close the connection upon completion.
1091 # 3) the headers for the previous response have not been read, thus
1092 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
1093 #
1094 # if there is no prior response, then we can request at will.
1095 #
1096 # if point (2) is true, then we will have passed the socket to the
1097 # response (effectively meaning, "there is no prior response"), and
1098 # will open a new one when a new request is made.
1099 #
1100 # Note: if a prior response exists, then we *can* start a new request.
1101 # We are not allowed to begin fetching the response to this new
1102 # request, however, until that prior response is complete.
1103 #
1104 if self.__state == _CS_IDLE:
1105 self.__state = _CS_REQ_STARTED
1106 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001107 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001108
AMIR8ca8a2e2020-07-19 00:46:10 +04301109 self._validate_method(method)
1110
Jason R. Coombs7774d782019-09-28 08:32:01 -04001111 # Save the method for use later in the response phase
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001112 self._method = method
Jason R. Coombs7774d782019-09-28 08:32:01 -04001113
1114 url = url or '/'
1115 self._validate_path(url)
1116
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001117 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001118
Jason R. Coombs7774d782019-09-28 08:32:01 -04001119 self._output(self._encode_request(request))
Greg Stein5e0fa402000-06-26 08:28:01 +00001120
Greg Steindd6eefb2000-07-18 09:09:48 +00001121 if self._http_vsn == 11:
1122 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001123
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001124 if not skip_host:
1125 # this header is issued *only* for HTTP/1.1
1126 # connections. more specifically, this means it is
1127 # only issued when the client uses the new
1128 # HTTPConnection() class. backwards-compat clients
1129 # will be using HTTP/1.0 and those clients may be
1130 # issuing this header themselves. we should NOT issue
1131 # it twice; some web servers (such as Apache) barf
1132 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001133
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001134 # If we need a non-standard port,include it in the
1135 # header. If the request is going through a proxy,
1136 # but the host of the actual URL, not the host of the
1137 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001138
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001139 netloc = ''
1140 if url.startswith('http'):
1141 nil, netloc, nil, nil, nil = urlsplit(url)
1142
1143 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001144 try:
1145 netloc_enc = netloc.encode("ascii")
1146 except UnicodeEncodeError:
1147 netloc_enc = netloc.encode("idna")
1148 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001149 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001150 if self._tunnel_host:
1151 host = self._tunnel_host
1152 port = self._tunnel_port
1153 else:
1154 host = self.host
1155 port = self.port
1156
Thomas Wouters477c8d52006-05-27 19:21:47 +00001157 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001158 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001159 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001160 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001161
1162 # As per RFC 273, IPv6 address should be wrapped with []
1163 # when used as Host header
1164
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001165 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001166 host_enc = b'[' + host_enc + b']'
1167
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001168 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001169 self.putheader('Host', host_enc)
1170 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001171 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001172 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001173
Greg Steindd6eefb2000-07-18 09:09:48 +00001174 # note: we are assuming that clients will not attempt to set these
1175 # headers since *this* library must deal with the
1176 # consequences. this also means that when the supporting
1177 # libraries are updated to recognize other forms, then this
1178 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001179
Greg Steindd6eefb2000-07-18 09:09:48 +00001180 # we only want a Content-Encoding of "identity" since we don't
1181 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001182 if not skip_accept_encoding:
1183 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001184
Greg Steindd6eefb2000-07-18 09:09:48 +00001185 # we can accept "chunked" Transfer-Encodings, but no others
1186 # NOTE: no TE header implies *only* "chunked"
1187 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001188
Greg Steindd6eefb2000-07-18 09:09:48 +00001189 # if TE is supplied in the header, then it must appear in a
1190 # Connection header.
1191 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001192
Greg Steindd6eefb2000-07-18 09:09:48 +00001193 else:
1194 # For HTTP/1.0, the server will assume "not chunked"
1195 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001196
Jason R. Coombs7774d782019-09-28 08:32:01 -04001197 def _encode_request(self, request):
1198 # ASCII also helps prevent CVE-2019-9740.
1199 return request.encode('ascii')
1200
AMIR8ca8a2e2020-07-19 00:46:10 +04301201 def _validate_method(self, method):
1202 """Validate a method name for putrequest."""
1203 # prevent http header injection
1204 match = _contains_disallowed_method_pchar_re.search(method)
1205 if match:
1206 raise ValueError(
1207 f"method can't contain control characters. {method!r} "
1208 f"(found at least {match.group()!r})")
1209
Jason R. Coombs7774d782019-09-28 08:32:01 -04001210 def _validate_path(self, url):
1211 """Validate a url for putrequest."""
1212 # Prevent CVE-2019-9740.
1213 match = _contains_disallowed_url_pchar_re.search(url)
1214 if match:
1215 raise InvalidURL(f"URL can't contain control characters. {url!r} "
1216 f"(found at least {match.group()!r})")
1217
Ashwin Ramaswami9165add2020-03-14 14:56:06 -04001218 def _validate_host(self, host):
1219 """Validate a host so it doesn't contain control characters."""
1220 # Prevent CVE-2019-18348.
1221 match = _contains_disallowed_url_pchar_re.search(host)
1222 if match:
1223 raise InvalidURL(f"URL can't contain control characters. {host!r} "
1224 f"(found at least {match.group()!r})")
1225
Benjamin Petersonf608c612008-11-16 18:33:53 +00001226 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001227 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001228
Greg Steindd6eefb2000-07-18 09:09:48 +00001229 For example: h.putheader('Accept', 'text/html')
1230 """
1231 if self.__state != _CS_REQ_STARTED:
1232 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001233
Guido van Rossum98297ee2007-11-06 21:34:58 +00001234 if hasattr(header, 'encode'):
1235 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001236
1237 if not _is_legal_header_name(header):
1238 raise ValueError('Invalid header name %r' % (header,))
1239
Benjamin Petersonf608c612008-11-16 18:33:53 +00001240 values = list(values)
1241 for i, one_value in enumerate(values):
1242 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001243 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001244 elif isinstance(one_value, int):
1245 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001246
1247 if _is_illegal_header_value(values[i]):
1248 raise ValueError('Invalid header value %r' % (values[i],))
1249
Benjamin Petersonf608c612008-11-16 18:33:53 +00001250 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001251 header = header + b': ' + value
1252 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001253
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001254 def endheaders(self, message_body=None, *, encode_chunked=False):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001255 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001256
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001257 This method sends the request to the server. The optional message_body
1258 argument can be used to pass a message body associated with the
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001259 request.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001260 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001261 if self.__state == _CS_REQ_STARTED:
1262 self.__state = _CS_REQ_SENT
1263 else:
1264 raise CannotSendHeader()
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001265 self._send_output(message_body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001266
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001267 def request(self, method, url, body=None, headers={}, *,
1268 encode_chunked=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001269 """Send a complete request to the server."""
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001270 self._send_request(method, url, body, headers, encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001271
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001272 def _send_request(self, method, url, body, headers, encode_chunked):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001273 # Honor explicitly requested Host: and Accept-Encoding: headers.
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001274 header_names = frozenset(k.lower() for k in headers)
Jeremy Hylton2c178252004-08-07 16:28:14 +00001275 skips = {}
1276 if 'host' in header_names:
1277 skips['skip_host'] = 1
1278 if 'accept-encoding' in header_names:
1279 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001280
Jeremy Hylton2c178252004-08-07 16:28:14 +00001281 self.putrequest(method, url, **skips)
1282
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001283 # chunked encoding will happen if HTTP/1.1 is used and either
1284 # the caller passes encode_chunked=True or the following
1285 # conditions hold:
1286 # 1. content-length has not been explicitly set
Martin Panteref91bb22016-08-27 01:39:26 +00001287 # 2. the body is a file or iterable, but not a str or bytes-like
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001288 # 3. Transfer-Encoding has NOT been explicitly set by the caller
1289
R David Murraybeed8402015-03-22 15:18:23 -04001290 if 'content-length' not in header_names:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001291 # only chunk body if not explicitly set for backwards
1292 # compatibility, assuming the client code is already handling the
1293 # chunking
1294 if 'transfer-encoding' not in header_names:
1295 # if content-length cannot be automatically determined, fall
1296 # back to chunked encoding
1297 encode_chunked = False
1298 content_length = self._get_content_length(body, method)
1299 if content_length is None:
Martin Panteref91bb22016-08-27 01:39:26 +00001300 if body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001301 if self.debuglevel > 0:
1302 print('Unable to determine size of %r' % body)
1303 encode_chunked = True
1304 self.putheader('Transfer-Encoding', 'chunked')
1305 else:
1306 self.putheader('Content-Length', str(content_length))
1307 else:
1308 encode_chunked = False
1309
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001310 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001311 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001312 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001313 # RFC 2616 Section 3.7.1 says that text default has a
1314 # default charset of iso-8859-1.
Martin Panter44391482016-02-09 10:20:52 +00001315 body = _encode(body, 'body')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001316 self.endheaders(body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001317
Greg Steindd6eefb2000-07-18 09:09:48 +00001318 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001319 """Get the response from the server.
1320
1321 If the HTTPConnection is in the correct state, returns an
1322 instance of HTTPResponse or of whatever object is returned by
Martin Pantercc71a792016-04-05 06:19:42 +00001323 the response_class variable.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001324
1325 If a request has not been sent or if a previous response has
1326 not be handled, ResponseNotReady is raised. If the HTTP
1327 response indicates that the connection should be closed, then
1328 it will be closed before the response is returned. When the
1329 connection is closed, the underlying socket is closed.
1330 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001331
Greg Stein616a58d2003-06-24 06:35:19 +00001332 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001333 if self.__response and self.__response.isclosed():
1334 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001335
Greg Steindd6eefb2000-07-18 09:09:48 +00001336 # if a prior response exists, then it must be completed (otherwise, we
1337 # cannot read this response's header to determine the connection-close
1338 # behavior)
1339 #
1340 # note: if a prior response existed, but was connection-close, then the
1341 # socket and response were made independent of this HTTPConnection
1342 # object since a new request requires that we open a whole new
1343 # connection
1344 #
1345 # this means the prior response had one of two states:
1346 # 1) will_close: this connection was reset and the prior socket and
1347 # response operate independently
1348 # 2) persistent: the response was retained and we await its
1349 # isclosed() status to become true.
1350 #
1351 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001352 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001353
Jeremy Hylton30f86742000-09-18 22:50:38 +00001354 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001355 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001356 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001357 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001358 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001359
Serhiy Storchakab491e052014-12-01 13:07:45 +02001360 try:
R David Murraycae7bdb2015-04-05 19:26:29 -04001361 try:
1362 response.begin()
1363 except ConnectionError:
1364 self.close()
1365 raise
Serhiy Storchakab491e052014-12-01 13:07:45 +02001366 assert response.will_close != _UNKNOWN
1367 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001368
Serhiy Storchakab491e052014-12-01 13:07:45 +02001369 if response.will_close:
1370 # this effectively passes the connection to the response
1371 self.close()
1372 else:
1373 # remember this, so we can tell when it is complete
1374 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001375
Serhiy Storchakab491e052014-12-01 13:07:45 +02001376 return response
1377 except:
1378 response.close()
1379 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001380
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001381try:
1382 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001383except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001384 pass
1385else:
1386 class HTTPSConnection(HTTPConnection):
1387 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001388
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001389 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001390
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001391 # XXX Should key_file and cert_file be deprecated in favour of context?
1392
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001393 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001394 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1395 source_address=None, *, context=None,
Nir Sofferad455cd2017-11-06 23:16:37 +02001396 check_hostname=None, blocksize=8192):
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001397 super(HTTPSConnection, self).__init__(host, port, timeout,
Nir Sofferad455cd2017-11-06 23:16:37 +02001398 source_address,
1399 blocksize=blocksize)
Christian Heimesd0486372016-09-10 23:23:33 +02001400 if (key_file is not None or cert_file is not None or
1401 check_hostname is not None):
1402 import warnings
1403 warnings.warn("key_file, cert_file and check_hostname are "
1404 "deprecated, use a custom context instead.",
1405 DeprecationWarning, 2)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001406 self.key_file = key_file
1407 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001408 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001409 context = ssl._create_default_https_context()
Christian Heimesf97406b2020-11-13 16:37:52 +01001410 # send ALPN extension to indicate HTTP/1.1 protocol
1411 if self._http_vsn == 11:
1412 context.set_alpn_protocols(['http/1.1'])
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001413 # enable PHA for TLS 1.3 connections if available
1414 if context.post_handshake_auth is not None:
1415 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001416 will_verify = context.verify_mode != ssl.CERT_NONE
1417 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001418 check_hostname = context.check_hostname
1419 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001420 raise ValueError("check_hostname needs a SSL context with "
1421 "either CERT_OPTIONAL or CERT_REQUIRED")
1422 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001423 context.load_cert_chain(cert_file, key_file)
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001424 # cert and key file means the user wants to authenticate.
1425 # enable TLS 1.3 PHA implicitly even for custom contexts.
1426 if context.post_handshake_auth is not None:
1427 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001428 self._context = context
Christian Heimes61d478c2018-01-27 15:51:38 +01001429 if check_hostname is not None:
1430 self._context.check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001431
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001432 def connect(self):
1433 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001434
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001435 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001436
1437 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001438 server_hostname = self._tunnel_host
1439 else:
1440 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001441
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001442 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001443 server_hostname=server_hostname)
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001444
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001445 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001446
Greg Stein5e0fa402000-06-26 08:28:01 +00001447class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001448 # Subclasses that define an __init__ must call Exception.__init__
1449 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001450 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001451
1452class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001453 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001454
Skip Montanaro9d389972002-03-24 16:53:50 +00001455class InvalidURL(HTTPException):
1456 pass
1457
Greg Stein5e0fa402000-06-26 08:28:01 +00001458class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001459 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001460 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001461 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001462
1463class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001464 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001465
Greg Stein5e0fa402000-06-26 08:28:01 +00001466class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001467 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001468
1469class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001470 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001471 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001472 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001473 self.expected = expected
1474 def __repr__(self):
1475 if self.expected is not None:
1476 e = ', %i more expected' % self.expected
1477 else:
1478 e = ''
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001479 return '%s(%i bytes read%s)' % (self.__class__.__name__,
1480 len(self.partial), e)
Serhiy Storchaka96aeaec2019-05-06 22:29:40 +03001481 __str__ = object.__str__
Greg Stein5e0fa402000-06-26 08:28:01 +00001482
1483class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001484 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001485
1486class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001487 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001488
1489class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001490 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001491
1492class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001493 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001494
1495class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001496 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001497 if not line:
1498 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001499 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001500 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001501
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001502class LineTooLong(HTTPException):
1503 def __init__(self, line_type):
1504 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1505 % (_MAXLINE, line_type))
1506
R David Murraycae7bdb2015-04-05 19:26:29 -04001507class RemoteDisconnected(ConnectionResetError, BadStatusLine):
1508 def __init__(self, *pos, **kw):
1509 BadStatusLine.__init__(self, "")
1510 ConnectionResetError.__init__(self, *pos, **kw)
1511
Greg Stein5e0fa402000-06-26 08:28:01 +00001512# for backwards compatibility
1513error = HTTPException