blob: c526380dcabfc95bf8c6cd3f525d4c831dd6eb1f [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")
Greg Steindd6eefb2000-07-18 09:09:48 +0000352 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000353 try:
354 self.length = int(length)
355 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000356 self.length = None
357 else:
358 if self.length < 0: # ignore nonsensical negative lengths
359 self.length = None
360 else:
361 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000362
Greg Steindd6eefb2000-07-18 09:09:48 +0000363 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000364 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000365 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000366 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000367 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000368
Greg Steindd6eefb2000-07-18 09:09:48 +0000369 # if the connection remains open, and we aren't using chunked, and
370 # a content-length was not provided, then assume that the connection
371 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000372 if (not self.will_close and
373 not self.chunked and
374 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000375 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000376
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000377 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000378 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000379 if self.version == 11:
380 # An HTTP/1.1 proxy is assumed to stay open unless
381 # explicitly closed.
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000382 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000383 return True
384 return False
385
Jeremy Hylton2c178252004-08-07 16:28:14 +0000386 # Some HTTP/1.0 implementations have support for persistent
387 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000388
Christian Heimes895627f2007-12-08 17:28:33 +0000389 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000390 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000391 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000392
Jeremy Hylton2c178252004-08-07 16:28:14 +0000393 # At least Akamai returns a "Connection: Keep-Alive" header,
394 # which was supposed to be sent by the client.
395 if conn and "keep-alive" in conn.lower():
396 return False
397
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000398 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000399 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000400 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000401 return False
402
403 # otherwise, assume it will close
404 return True
405
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200406 def _close_conn(self):
407 fp = self.fp
408 self.fp = None
409 fp.close()
410
Greg Steindd6eefb2000-07-18 09:09:48 +0000411 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300412 try:
413 super().close() # set "closed" flag
414 finally:
415 if self.fp:
416 self._close_conn()
Greg Stein5e0fa402000-06-26 08:28:01 +0000417
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000418 # These implementations are for the benefit of io.BufferedReader.
419
420 # XXX This class should probably be revised to act more like
421 # the "raw stream" that BufferedReader expects.
422
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000423 def flush(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200424 super().flush()
425 if self.fp:
426 self.fp.flush()
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000427
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000428 def readable(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700429 """Always returns True"""
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000430 return True
431
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000432 # End of "raw stream" methods
433
Greg Steindd6eefb2000-07-18 09:09:48 +0000434 def isclosed(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200435 """True if the connection is closed."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000436 # NOTE: it is possible that we will not ever call self.close(). This
437 # case occurs when will_close is TRUE, length is None, and we
438 # read up to the last byte, but NOT past it.
439 #
440 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
441 # called, meaning self.isclosed() is meaningful.
442 return self.fp is None
443
444 def read(self, amt=None):
445 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000446 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000447
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000448 if self._method == "HEAD":
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200449 self._close_conn()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000450 return b""
451
Bruce Merry152f0b82020-06-25 08:30:21 +0200452 if self.chunked:
453 return self._read_chunked(amt)
454
Antoine Pitrou38d96432011-12-06 22:33:57 +0100455 if amt is not None:
Bruce Merry152f0b82020-06-25 08:30:21 +0200456 if self.length is not None and amt > self.length:
457 # clip the read to the "end of response"
458 amt = self.length
459 s = self.fp.read(amt)
460 if not s and amt:
461 # Ideally, we would raise IncompleteRead if the content-length
462 # wasn't satisfied, but it might break compatibility.
463 self._close_conn()
464 elif self.length is not None:
465 self.length -= len(s)
466 if not self.length:
467 self._close_conn()
468 return s
Antoine Pitrou38d96432011-12-06 22:33:57 +0100469 else:
470 # Amount is not given (unbounded read) so we must check self.length
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000471 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000472 s = self.fp.read()
473 else:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100474 try:
475 s = self._safe_read(self.length)
476 except IncompleteRead:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200477 self._close_conn()
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100478 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000479 self.length = 0
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200480 self._close_conn() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000481 return s
482
Antoine Pitrou38d96432011-12-06 22:33:57 +0100483 def readinto(self, b):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700484 """Read up to len(b) bytes into bytearray b and return the number
485 of bytes read.
486 """
487
Antoine Pitrou38d96432011-12-06 22:33:57 +0100488 if self.fp is None:
489 return 0
490
491 if self._method == "HEAD":
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200492 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100493 return 0
494
495 if self.chunked:
496 return self._readinto_chunked(b)
497
Greg Steindd6eefb2000-07-18 09:09:48 +0000498 if self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100499 if len(b) > self.length:
Greg Steindd6eefb2000-07-18 09:09:48 +0000500 # clip the read to the "end of response"
Antoine Pitrou38d96432011-12-06 22:33:57 +0100501 b = memoryview(b)[0:self.length]
Greg Steindd6eefb2000-07-18 09:09:48 +0000502
503 # we do not use _safe_read() here because this may be a .will_close
504 # connection, and the user is reading more bytes than will be provided
505 # (for example, reading in 1k chunks)
Antoine Pitrou38d96432011-12-06 22:33:57 +0100506 n = self.fp.readinto(b)
Serhiy Storchaka1c84ac12013-12-17 21:50:02 +0200507 if not n and b:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100508 # Ideally, we would raise IncompleteRead if the content-length
509 # wasn't satisfied, but it might break compatibility.
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200510 self._close_conn()
Antoine Pitrou6a35e182013-02-02 23:04:56 +0100511 elif self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100512 self.length -= n
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000513 if not self.length:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200514 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100515 return n
Greg Steindd6eefb2000-07-18 09:09:48 +0000516
Antoine Pitrou38d96432011-12-06 22:33:57 +0100517 def _read_next_chunk_size(self):
518 # Read the next chunk size from the file
519 line = self.fp.readline(_MAXLINE + 1)
520 if len(line) > _MAXLINE:
521 raise LineTooLong("chunk size")
522 i = line.find(b";")
523 if i >= 0:
524 line = line[:i] # strip chunk-extensions
525 try:
526 return int(line, 16)
527 except ValueError:
528 # close the connection as protocol synchronisation is
529 # probably lost
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200530 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100531 raise
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000532
Antoine Pitrou38d96432011-12-06 22:33:57 +0100533 def _read_and_discard_trailer(self):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000534 # read and discard trailer up to the CRLF terminator
535 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000536 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000537 line = self.fp.readline(_MAXLINE + 1)
538 if len(line) > _MAXLINE:
539 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000540 if not line:
541 # a vanishingly small number of sites EOF without
542 # sending the trailer
543 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800544 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000545 break
546
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000547 def _get_chunk_left(self):
548 # return self.chunk_left, reading a new chunk if necessary.
549 # chunk_left == 0: at the end of the current chunk, need to close it
550 # chunk_left == None: No current chunk, should read next.
551 # This function returns non-zero or None if the last chunk has
552 # been read.
553 chunk_left = self.chunk_left
554 if not chunk_left: # Can be 0 or None
555 if chunk_left is not None:
Mike53f7a7c2017-12-14 14:04:53 +0300556 # We are at the end of chunk, discard chunk end
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000557 self._safe_read(2) # toss the CRLF at the end of the chunk
558 try:
559 chunk_left = self._read_next_chunk_size()
560 except ValueError:
561 raise IncompleteRead(b'')
562 if chunk_left == 0:
563 # last chunk: 1*("0") [ chunk-extension ] CRLF
564 self._read_and_discard_trailer()
565 # we read everything; close the "file"
566 self._close_conn()
567 chunk_left = None
568 self.chunk_left = chunk_left
569 return chunk_left
570
Bruce Merry152f0b82020-06-25 08:30:21 +0200571 def _read_chunked(self, amt=None):
Antoine Pitrou38d96432011-12-06 22:33:57 +0100572 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100573 value = []
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000574 try:
575 while True:
576 chunk_left = self._get_chunk_left()
577 if chunk_left is None:
578 break
Bruce Merry152f0b82020-06-25 08:30:21 +0200579
580 if amt is not None and amt <= chunk_left:
581 value.append(self._safe_read(amt))
582 self.chunk_left = chunk_left - amt
583 break
584
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000585 value.append(self._safe_read(chunk_left))
Bruce Merry152f0b82020-06-25 08:30:21 +0200586 if amt is not None:
587 amt -= chunk_left
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000588 self.chunk_left = 0
589 return b''.join(value)
590 except IncompleteRead:
591 raise IncompleteRead(b''.join(value))
Tim Peters230a60c2002-11-09 05:08:07 +0000592
Antoine Pitrou38d96432011-12-06 22:33:57 +0100593 def _readinto_chunked(self, b):
594 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100595 total_bytes = 0
596 mvb = memoryview(b)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000597 try:
598 while True:
599 chunk_left = self._get_chunk_left()
600 if chunk_left is None:
601 return total_bytes
Antoine Pitrou4ce6aa42011-12-06 22:34:36 +0100602
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000603 if len(mvb) <= chunk_left:
604 n = self._safe_readinto(mvb)
605 self.chunk_left = chunk_left - n
606 return total_bytes + n
607
608 temp_mvb = mvb[:chunk_left]
Antoine Pitrou38d96432011-12-06 22:33:57 +0100609 n = self._safe_readinto(temp_mvb)
610 mvb = mvb[n:]
611 total_bytes += n
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000612 self.chunk_left = 0
Antoine Pitrou38d96432011-12-06 22:33:57 +0100613
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000614 except IncompleteRead:
615 raise IncompleteRead(bytes(b[0:total_bytes]))
Antoine Pitrou38d96432011-12-06 22:33:57 +0100616
Greg Steindd6eefb2000-07-18 09:09:48 +0000617 def _safe_read(self, amt):
Inada Naokid6bf6f22019-04-06 18:06:19 +0900618 """Read the number of bytes requested.
Greg Steindd6eefb2000-07-18 09:09:48 +0000619
620 This function should be used when <amt> bytes "should" be present for
621 reading. If the bytes are truly not available (due to EOF), then the
622 IncompleteRead exception can be used to detect the problem.
623 """
Inada Naokid6bf6f22019-04-06 18:06:19 +0900624 data = self.fp.read(amt)
625 if len(data) < amt:
626 raise IncompleteRead(data, amt-len(data))
627 return data
Greg Steindd6eefb2000-07-18 09:09:48 +0000628
Antoine Pitrou38d96432011-12-06 22:33:57 +0100629 def _safe_readinto(self, b):
630 """Same as _safe_read, but for reading into a buffer."""
Inada Naokid6bf6f22019-04-06 18:06:19 +0900631 amt = len(b)
632 n = self.fp.readinto(b)
633 if n < amt:
634 raise IncompleteRead(bytes(b[:n]), amt-n)
635 return n
Antoine Pitrou38d96432011-12-06 22:33:57 +0100636
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000637 def read1(self, n=-1):
638 """Read with at most one underlying system call. If at least one
639 byte is buffered, return that instead.
640 """
641 if self.fp is None or self._method == "HEAD":
642 return b""
643 if self.chunked:
644 return self._read1_chunked(n)
Martin Panterce911c32016-03-17 06:42:48 +0000645 if self.length is not None and (n < 0 or n > self.length):
646 n = self.length
Martin Panter741d4942016-10-21 00:52:04 +0000647 result = self.fp.read1(n)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000648 if not result and n:
649 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000650 elif self.length is not None:
651 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000652 return result
653
654 def peek(self, n=-1):
655 # Having this enables IOBase.readline() to read more than one
656 # byte at a time
657 if self.fp is None or self._method == "HEAD":
658 return b""
659 if self.chunked:
660 return self._peek_chunked(n)
661 return self.fp.peek(n)
662
663 def readline(self, limit=-1):
664 if self.fp is None or self._method == "HEAD":
665 return b""
666 if self.chunked:
667 # Fallback to IOBase readline which uses peek() and read()
668 return super().readline(limit)
Martin Panterce911c32016-03-17 06:42:48 +0000669 if self.length is not None and (limit < 0 or limit > self.length):
670 limit = self.length
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000671 result = self.fp.readline(limit)
672 if not result and limit:
673 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000674 elif self.length is not None:
675 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000676 return result
677
678 def _read1_chunked(self, n):
679 # Strictly speaking, _get_chunk_left() may cause more than one read,
680 # but that is ok, since that is to satisfy the chunked protocol.
681 chunk_left = self._get_chunk_left()
682 if chunk_left is None or n == 0:
683 return b''
684 if not (0 <= n <= chunk_left):
685 n = chunk_left # if n is negative or larger than chunk_left
686 read = self.fp.read1(n)
687 self.chunk_left -= len(read)
688 if not read:
689 raise IncompleteRead(b"")
690 return read
691
692 def _peek_chunked(self, n):
693 # Strictly speaking, _get_chunk_left() may cause more than one read,
694 # but that is ok, since that is to satisfy the chunked protocol.
695 try:
696 chunk_left = self._get_chunk_left()
697 except IncompleteRead:
698 return b'' # peek doesn't worry about protocol
699 if chunk_left is None:
700 return b'' # eof
701 # peek is allowed to return more than requested. Just request the
702 # entire chunk, and truncate what we get.
703 return self.fp.peek(chunk_left)[:chunk_left]
704
Antoine Pitroub353c122009-02-11 00:39:14 +0000705 def fileno(self):
706 return self.fp.fileno()
707
Greg Steindd6eefb2000-07-18 09:09:48 +0000708 def getheader(self, name, default=None):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700709 '''Returns the value of the header matching *name*.
710
711 If there are multiple matching headers, the values are
712 combined into a single string separated by commas and spaces.
713
714 If no matching header is found, returns *default* or None if
715 the *default* is not specified.
716
717 If the headers are unknown, raises http.client.ResponseNotReady.
718
719 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000720 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000721 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000722 headers = self.headers.get_all(name) or default
723 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
724 return headers
725 else:
726 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000727
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000728 def getheaders(self):
729 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000730 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000731 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000732 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000733
Antoine Pitroub353c122009-02-11 00:39:14 +0000734 # We override IOBase.__iter__ so that it doesn't check for closed-ness
735
736 def __iter__(self):
737 return self
738
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000739 # For compatibility with old-style urllib responses.
740
741 def info(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700742 '''Returns an instance of the class mimetools.Message containing
743 meta-information associated with the URL.
744
745 When the method is HTTP, these headers are those returned by
746 the server at the head of the retrieved HTML page (including
747 Content-Length and Content-Type).
748
749 When the method is FTP, a Content-Length header will be
750 present if (as is now usual) the server passed back a file
751 length in response to the FTP retrieval request. A
752 Content-Type header will be present if the MIME type can be
753 guessed.
754
755 When the method is local-file, returned headers will include
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200756 a Date representing the file's last-modified time, a
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700757 Content-Length giving file size, and a Content-Type
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200758 containing a guess at the file's type. See also the
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700759 description of the mimetools module.
760
761 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000762 return self.headers
763
764 def geturl(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700765 '''Return the real URL of the page.
766
767 In some cases, the HTTP server redirects a client to another
768 URL. The urlopen() function handles this transparently, but in
769 some cases the caller needs to know which URL the client was
770 redirected to. The geturl() method can be used to get at this
771 redirected URL.
772
773 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000774 return self.url
775
776 def getcode(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700777 '''Return the HTTP status code that was sent with the response,
778 or None if the URL is not an HTTP URL.
779
780 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000781 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000782
783class HTTPConnection:
784
Greg Steindd6eefb2000-07-18 09:09:48 +0000785 _http_vsn = 11
786 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000787
Greg Steindd6eefb2000-07-18 09:09:48 +0000788 response_class = HTTPResponse
789 default_port = HTTP_PORT
790 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000791 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000792
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000793 @staticmethod
794 def _is_textIO(stream):
795 """Test whether a file-like object is a text or a binary stream.
796 """
797 return isinstance(stream, io.TextIOBase)
798
799 @staticmethod
800 def _get_content_length(body, method):
801 """Get the content-length based on the body.
802
Martin Panteref91bb22016-08-27 01:39:26 +0000803 If the body is None, we set Content-Length: 0 for methods that expect
804 a body (RFC 7230, Section 3.3.2). We also set the Content-Length for
805 any method if the body is a str or bytes-like object and not a file.
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000806 """
Martin Panteref91bb22016-08-27 01:39:26 +0000807 if body is None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000808 # do an explicit check for not None here to distinguish
809 # between unset and set but empty
Martin Panteref91bb22016-08-27 01:39:26 +0000810 if method.upper() in _METHODS_EXPECTING_BODY:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000811 return 0
812 else:
813 return None
814
815 if hasattr(body, 'read'):
816 # file-like object.
Martin Panteref91bb22016-08-27 01:39:26 +0000817 return None
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000818
819 try:
820 # does it implement the buffer protocol (bytes, bytearray, array)?
821 mv = memoryview(body)
822 return mv.nbytes
823 except TypeError:
824 pass
825
826 if isinstance(body, str):
827 return len(body)
828
829 return None
830
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700831 def __init__(self, host, port=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Nir Sofferad455cd2017-11-06 23:16:37 +0200832 source_address=None, blocksize=8192):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000833 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000834 self.source_address = source_address
Nir Sofferad455cd2017-11-06 23:16:37 +0200835 self.blocksize = blocksize
Greg Steindd6eefb2000-07-18 09:09:48 +0000836 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000837 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000838 self.__response = None
839 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000840 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000841 self._tunnel_host = None
842 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000843 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000844
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400845 (self.host, self.port) = self._get_hostport(host, port)
846
Ashwin Ramaswami9165add2020-03-14 14:56:06 -0400847 self._validate_host(self.host)
848
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400849 # This is stored as an instance variable to allow unit
850 # tests to replace it with a suitable mockup
851 self._create_connection = socket.create_connection
Greg Stein5e0fa402000-06-26 08:28:01 +0000852
Senthil Kumaran47fff872009-12-20 07:10:31 +0000853 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400854 """Set up host and port for HTTP CONNECT tunnelling.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000855
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400856 In a connection that uses HTTP CONNECT tunneling, the host passed to the
857 constructor is used as a proxy server that relays all communication to
858 the endpoint passed to `set_tunnel`. This done by sending an HTTP
859 CONNECT request to the proxy server when the connection is established.
860
Fernando Toledo711381d2020-12-21 11:06:31 -0300861 This method must be called before the HTTP connection has been
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400862 established.
863
864 The headers argument should be a mapping of extra HTTP headers to send
865 with the CONNECT request.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000866 """
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400867
868 if self.sock:
869 raise RuntimeError("Can't set up tunnel for established connection")
870
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200871 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000872 if headers:
873 self._tunnel_headers = headers
874 else:
875 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000876
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400877 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000878 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000879 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000880 j = host.rfind(']') # ipv6 addresses have [...]
881 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000882 try:
883 port = int(host[i+1:])
884 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200885 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
886 port = self.default_port
887 else:
888 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000889 host = host[:i]
890 else:
891 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000892 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000893 host = host[1:-1]
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400894
895 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000896
Jeremy Hylton30f86742000-09-18 22:50:38 +0000897 def set_debuglevel(self, level):
898 self.debuglevel = level
899
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000900 def _tunnel(self):
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800901 connect = b"CONNECT %s:%d HTTP/1.0\r\n" % (
902 self._tunnel_host.encode("ascii"), self._tunnel_port)
903 headers = [connect]
Georg Brandlc7c199f2011-01-22 22:06:24 +0000904 for header, value in self._tunnel_headers.items():
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800905 headers.append(f"{header}: {value}\r\n".encode("latin-1"))
906 headers.append(b"\r\n")
907 # Making a single send() call instead of one per line encourages
908 # the host OS to use a more optimal packet size instead of
909 # potentially emitting a series of small packets.
910 self.send(b"".join(headers))
911 del headers
Senthil Kumaran47fff872009-12-20 07:10:31 +0000912
Georg Brandlc7c199f2011-01-22 22:06:24 +0000913 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000914 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000915
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200916 if code != http.HTTPStatus.OK:
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000917 self.close()
Gregory P. Smithc25910a2021-03-07 23:35:13 -0800918 raise OSError(f"Tunnel connection failed: {code} {message.strip()}")
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000919 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000920 line = response.fp.readline(_MAXLINE + 1)
921 if len(line) > _MAXLINE:
922 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800923 if not line:
924 # for sites which EOF without sending a trailer
925 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800926 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000927 break
928
Berker Peksagab53ab02015-02-03 12:22:11 +0200929 if self.debuglevel > 0:
930 print('header:', line.decode())
931
Greg Steindd6eefb2000-07-18 09:09:48 +0000932 def connect(self):
933 """Connect to the host and port specified in __init__."""
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500934 self.sock = self._create_connection(
935 (self.host,self.port), self.timeout, self.source_address)
936 self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400937
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000938 if self._tunnel_host:
939 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000940
Greg Steindd6eefb2000-07-18 09:09:48 +0000941 def close(self):
942 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000943 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300944 try:
945 sock = self.sock
946 if sock:
947 self.sock = None
948 sock.close() # close it manually... there may be other refs
949 finally:
950 response = self.__response
951 if response:
952 self.__response = None
953 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000954
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000955 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000956 """Send `data' to the server.
957 ``data`` can be a string object, a bytes object, an array object, a
958 file-like object that supports a .read() method, or an iterable object.
959 """
960
Greg Steindd6eefb2000-07-18 09:09:48 +0000961 if self.sock is None:
962 if self.auto_open:
963 self.connect()
964 else:
965 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000966
Jeremy Hylton30f86742000-09-18 22:50:38 +0000967 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000968 print("send:", repr(data))
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000969 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000970 if self.debuglevel > 0:
971 print("sendIng a read()able")
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000972 encode = self._is_textIO(data)
973 if encode and self.debuglevel > 0:
974 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000975 while 1:
Nir Sofferad455cd2017-11-06 23:16:37 +0200976 datablock = data.read(self.blocksize)
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000977 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000978 break
979 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000980 datablock = datablock.encode("iso-8859-1")
981 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300982 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000983 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000984 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000985 except TypeError:
Serhiy Storchaka2e576f52017-04-24 09:05:00 +0300986 if isinstance(data, collections.abc.Iterable):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000987 for d in data:
988 self.sock.sendall(d)
989 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200990 raise TypeError("data should be a bytes-like object "
991 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000992
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000993 def _output(self, s):
994 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000995
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000996 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000997 """
998 self._buffer.append(s)
999
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001000 def _read_readable(self, readable):
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001001 if self.debuglevel > 0:
1002 print("sendIng a read()able")
1003 encode = self._is_textIO(readable)
1004 if encode and self.debuglevel > 0:
1005 print("encoding file using iso-8859-1")
1006 while True:
Nir Sofferad455cd2017-11-06 23:16:37 +02001007 datablock = readable.read(self.blocksize)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001008 if not datablock:
1009 break
1010 if encode:
1011 datablock = datablock.encode("iso-8859-1")
1012 yield datablock
1013
1014 def _send_output(self, message_body=None, encode_chunked=False):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001015 """Send the currently buffered request and clear the buffer.
1016
Jeremy Hyltone3252ec2002-07-16 21:41:43 +00001017 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001018 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001019 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001020 self._buffer.extend((b"", b""))
1021 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001022 del self._buffer[:]
1023 self.send(msg)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001024
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001025 if message_body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001026
1027 # create a consistent interface to message_body
1028 if hasattr(message_body, 'read'):
1029 # Let file-like take precedence over byte-like. This
1030 # is needed to allow the current position of mmap'ed
1031 # files to be taken into account.
1032 chunks = self._read_readable(message_body)
1033 else:
1034 try:
1035 # this is solely to check to see if message_body
1036 # implements the buffer API. it /would/ be easier
1037 # to capture if PyObject_CheckBuffer was exposed
1038 # to Python.
1039 memoryview(message_body)
1040 except TypeError:
1041 try:
1042 chunks = iter(message_body)
1043 except TypeError:
1044 raise TypeError("message_body should be a bytes-like "
1045 "object or an iterable, got %r"
1046 % type(message_body))
1047 else:
1048 # the object implements the buffer interface and
1049 # can be passed directly into socket methods
1050 chunks = (message_body,)
1051
1052 for chunk in chunks:
1053 if not chunk:
1054 if self.debuglevel > 0:
1055 print('Zero length chunk ignored')
1056 continue
1057
1058 if encode_chunked and self._http_vsn == 11:
1059 # chunked encoding
Eric V. Smith451d0e32016-09-09 21:56:20 -04001060 chunk = f'{len(chunk):X}\r\n'.encode('ascii') + chunk \
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001061 + b'\r\n'
1062 self.send(chunk)
1063
1064 if encode_chunked and self._http_vsn == 11:
1065 # end chunked transfer
1066 self.send(b'0\r\n\r\n')
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001067
Senthil Kumaran5dc504c2016-09-08 14:28:01 -07001068 def putrequest(self, method, url, skip_host=False,
1069 skip_accept_encoding=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001070 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001071
Greg Steindd6eefb2000-07-18 09:09:48 +00001072 `method' specifies an HTTP request method, e.g. 'GET'.
1073 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001074 `skip_host' if True does not add automatically a 'Host:' header
1075 `skip_accept_encoding' if True does not add automatically an
1076 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +00001077 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001078
Greg Stein616a58d2003-06-24 06:35:19 +00001079 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001080 if self.__response and self.__response.isclosed():
1081 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001082
Tim Peters58eb11c2004-01-18 20:29:55 +00001083
Greg Steindd6eefb2000-07-18 09:09:48 +00001084 # in certain cases, we cannot issue another request on this connection.
1085 # this occurs when:
1086 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
1087 # 2) a response to a previous request has signalled that it is going
1088 # to close the connection upon completion.
1089 # 3) the headers for the previous response have not been read, thus
1090 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
1091 #
1092 # if there is no prior response, then we can request at will.
1093 #
1094 # if point (2) is true, then we will have passed the socket to the
1095 # response (effectively meaning, "there is no prior response"), and
1096 # will open a new one when a new request is made.
1097 #
1098 # Note: if a prior response exists, then we *can* start a new request.
1099 # We are not allowed to begin fetching the response to this new
1100 # request, however, until that prior response is complete.
1101 #
1102 if self.__state == _CS_IDLE:
1103 self.__state = _CS_REQ_STARTED
1104 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001105 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001106
AMIR8ca8a2e2020-07-19 00:46:10 +04301107 self._validate_method(method)
1108
Jason R. Coombs7774d782019-09-28 08:32:01 -04001109 # Save the method for use later in the response phase
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001110 self._method = method
Jason R. Coombs7774d782019-09-28 08:32:01 -04001111
1112 url = url or '/'
1113 self._validate_path(url)
1114
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001115 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001116
Jason R. Coombs7774d782019-09-28 08:32:01 -04001117 self._output(self._encode_request(request))
Greg Stein5e0fa402000-06-26 08:28:01 +00001118
Greg Steindd6eefb2000-07-18 09:09:48 +00001119 if self._http_vsn == 11:
1120 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001121
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001122 if not skip_host:
1123 # this header is issued *only* for HTTP/1.1
1124 # connections. more specifically, this means it is
1125 # only issued when the client uses the new
1126 # HTTPConnection() class. backwards-compat clients
1127 # will be using HTTP/1.0 and those clients may be
1128 # issuing this header themselves. we should NOT issue
1129 # it twice; some web servers (such as Apache) barf
1130 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001131
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001132 # If we need a non-standard port,include it in the
1133 # header. If the request is going through a proxy,
1134 # but the host of the actual URL, not the host of the
1135 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001136
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001137 netloc = ''
1138 if url.startswith('http'):
1139 nil, netloc, nil, nil, nil = urlsplit(url)
1140
1141 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001142 try:
1143 netloc_enc = netloc.encode("ascii")
1144 except UnicodeEncodeError:
1145 netloc_enc = netloc.encode("idna")
1146 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001147 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001148 if self._tunnel_host:
1149 host = self._tunnel_host
1150 port = self._tunnel_port
1151 else:
1152 host = self.host
1153 port = self.port
1154
Thomas Wouters477c8d52006-05-27 19:21:47 +00001155 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001156 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001157 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001158 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001159
1160 # As per RFC 273, IPv6 address should be wrapped with []
1161 # when used as Host header
1162
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001163 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001164 host_enc = b'[' + host_enc + b']'
1165
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001166 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001167 self.putheader('Host', host_enc)
1168 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001169 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001170 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001171
Greg Steindd6eefb2000-07-18 09:09:48 +00001172 # note: we are assuming that clients will not attempt to set these
1173 # headers since *this* library must deal with the
1174 # consequences. this also means that when the supporting
1175 # libraries are updated to recognize other forms, then this
1176 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001177
Greg Steindd6eefb2000-07-18 09:09:48 +00001178 # we only want a Content-Encoding of "identity" since we don't
1179 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001180 if not skip_accept_encoding:
1181 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001182
Greg Steindd6eefb2000-07-18 09:09:48 +00001183 # we can accept "chunked" Transfer-Encodings, but no others
1184 # NOTE: no TE header implies *only* "chunked"
1185 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001186
Greg Steindd6eefb2000-07-18 09:09:48 +00001187 # if TE is supplied in the header, then it must appear in a
1188 # Connection header.
1189 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001190
Greg Steindd6eefb2000-07-18 09:09:48 +00001191 else:
1192 # For HTTP/1.0, the server will assume "not chunked"
1193 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001194
Jason R. Coombs7774d782019-09-28 08:32:01 -04001195 def _encode_request(self, request):
1196 # ASCII also helps prevent CVE-2019-9740.
1197 return request.encode('ascii')
1198
AMIR8ca8a2e2020-07-19 00:46:10 +04301199 def _validate_method(self, method):
1200 """Validate a method name for putrequest."""
1201 # prevent http header injection
1202 match = _contains_disallowed_method_pchar_re.search(method)
1203 if match:
1204 raise ValueError(
1205 f"method can't contain control characters. {method!r} "
1206 f"(found at least {match.group()!r})")
1207
Jason R. Coombs7774d782019-09-28 08:32:01 -04001208 def _validate_path(self, url):
1209 """Validate a url for putrequest."""
1210 # Prevent CVE-2019-9740.
1211 match = _contains_disallowed_url_pchar_re.search(url)
1212 if match:
1213 raise InvalidURL(f"URL can't contain control characters. {url!r} "
1214 f"(found at least {match.group()!r})")
1215
Ashwin Ramaswami9165add2020-03-14 14:56:06 -04001216 def _validate_host(self, host):
1217 """Validate a host so it doesn't contain control characters."""
1218 # Prevent CVE-2019-18348.
1219 match = _contains_disallowed_url_pchar_re.search(host)
1220 if match:
1221 raise InvalidURL(f"URL can't contain control characters. {host!r} "
1222 f"(found at least {match.group()!r})")
1223
Benjamin Petersonf608c612008-11-16 18:33:53 +00001224 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001225 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001226
Greg Steindd6eefb2000-07-18 09:09:48 +00001227 For example: h.putheader('Accept', 'text/html')
1228 """
1229 if self.__state != _CS_REQ_STARTED:
1230 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001231
Guido van Rossum98297ee2007-11-06 21:34:58 +00001232 if hasattr(header, 'encode'):
1233 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001234
1235 if not _is_legal_header_name(header):
1236 raise ValueError('Invalid header name %r' % (header,))
1237
Benjamin Petersonf608c612008-11-16 18:33:53 +00001238 values = list(values)
1239 for i, one_value in enumerate(values):
1240 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001241 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001242 elif isinstance(one_value, int):
1243 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001244
1245 if _is_illegal_header_value(values[i]):
1246 raise ValueError('Invalid header value %r' % (values[i],))
1247
Benjamin Petersonf608c612008-11-16 18:33:53 +00001248 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001249 header = header + b': ' + value
1250 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001251
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001252 def endheaders(self, message_body=None, *, encode_chunked=False):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001253 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001254
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001255 This method sends the request to the server. The optional message_body
1256 argument can be used to pass a message body associated with the
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001257 request.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001258 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001259 if self.__state == _CS_REQ_STARTED:
1260 self.__state = _CS_REQ_SENT
1261 else:
1262 raise CannotSendHeader()
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001263 self._send_output(message_body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001264
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001265 def request(self, method, url, body=None, headers={}, *,
1266 encode_chunked=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001267 """Send a complete request to the server."""
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001268 self._send_request(method, url, body, headers, encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001269
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001270 def _send_request(self, method, url, body, headers, encode_chunked):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001271 # Honor explicitly requested Host: and Accept-Encoding: headers.
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001272 header_names = frozenset(k.lower() for k in headers)
Jeremy Hylton2c178252004-08-07 16:28:14 +00001273 skips = {}
1274 if 'host' in header_names:
1275 skips['skip_host'] = 1
1276 if 'accept-encoding' in header_names:
1277 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001278
Jeremy Hylton2c178252004-08-07 16:28:14 +00001279 self.putrequest(method, url, **skips)
1280
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001281 # chunked encoding will happen if HTTP/1.1 is used and either
1282 # the caller passes encode_chunked=True or the following
1283 # conditions hold:
1284 # 1. content-length has not been explicitly set
Martin Panteref91bb22016-08-27 01:39:26 +00001285 # 2. the body is a file or iterable, but not a str or bytes-like
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001286 # 3. Transfer-Encoding has NOT been explicitly set by the caller
1287
R David Murraybeed8402015-03-22 15:18:23 -04001288 if 'content-length' not in header_names:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001289 # only chunk body if not explicitly set for backwards
1290 # compatibility, assuming the client code is already handling the
1291 # chunking
1292 if 'transfer-encoding' not in header_names:
1293 # if content-length cannot be automatically determined, fall
1294 # back to chunked encoding
1295 encode_chunked = False
1296 content_length = self._get_content_length(body, method)
1297 if content_length is None:
Martin Panteref91bb22016-08-27 01:39:26 +00001298 if body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001299 if self.debuglevel > 0:
1300 print('Unable to determine size of %r' % body)
1301 encode_chunked = True
1302 self.putheader('Transfer-Encoding', 'chunked')
1303 else:
1304 self.putheader('Content-Length', str(content_length))
1305 else:
1306 encode_chunked = False
1307
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001308 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001309 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001310 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001311 # RFC 2616 Section 3.7.1 says that text default has a
1312 # default charset of iso-8859-1.
Martin Panter44391482016-02-09 10:20:52 +00001313 body = _encode(body, 'body')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001314 self.endheaders(body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001315
Greg Steindd6eefb2000-07-18 09:09:48 +00001316 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001317 """Get the response from the server.
1318
1319 If the HTTPConnection is in the correct state, returns an
1320 instance of HTTPResponse or of whatever object is returned by
Martin Pantercc71a792016-04-05 06:19:42 +00001321 the response_class variable.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001322
1323 If a request has not been sent or if a previous response has
1324 not be handled, ResponseNotReady is raised. If the HTTP
1325 response indicates that the connection should be closed, then
1326 it will be closed before the response is returned. When the
1327 connection is closed, the underlying socket is closed.
1328 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001329
Greg Stein616a58d2003-06-24 06:35:19 +00001330 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001331 if self.__response and self.__response.isclosed():
1332 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001333
Greg Steindd6eefb2000-07-18 09:09:48 +00001334 # if a prior response exists, then it must be completed (otherwise, we
1335 # cannot read this response's header to determine the connection-close
1336 # behavior)
1337 #
1338 # note: if a prior response existed, but was connection-close, then the
1339 # socket and response were made independent of this HTTPConnection
1340 # object since a new request requires that we open a whole new
1341 # connection
1342 #
1343 # this means the prior response had one of two states:
1344 # 1) will_close: this connection was reset and the prior socket and
1345 # response operate independently
1346 # 2) persistent: the response was retained and we await its
1347 # isclosed() status to become true.
1348 #
1349 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001350 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001351
Jeremy Hylton30f86742000-09-18 22:50:38 +00001352 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001353 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001354 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001355 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001356 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001357
Serhiy Storchakab491e052014-12-01 13:07:45 +02001358 try:
R David Murraycae7bdb2015-04-05 19:26:29 -04001359 try:
1360 response.begin()
1361 except ConnectionError:
1362 self.close()
1363 raise
Serhiy Storchakab491e052014-12-01 13:07:45 +02001364 assert response.will_close != _UNKNOWN
1365 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001366
Serhiy Storchakab491e052014-12-01 13:07:45 +02001367 if response.will_close:
1368 # this effectively passes the connection to the response
1369 self.close()
1370 else:
1371 # remember this, so we can tell when it is complete
1372 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001373
Serhiy Storchakab491e052014-12-01 13:07:45 +02001374 return response
1375 except:
1376 response.close()
1377 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001378
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001379try:
1380 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001381except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001382 pass
1383else:
1384 class HTTPSConnection(HTTPConnection):
1385 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001386
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001387 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001388
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001389 # XXX Should key_file and cert_file be deprecated in favour of context?
1390
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001391 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001392 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1393 source_address=None, *, context=None,
Nir Sofferad455cd2017-11-06 23:16:37 +02001394 check_hostname=None, blocksize=8192):
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001395 super(HTTPSConnection, self).__init__(host, port, timeout,
Nir Sofferad455cd2017-11-06 23:16:37 +02001396 source_address,
1397 blocksize=blocksize)
Christian Heimesd0486372016-09-10 23:23:33 +02001398 if (key_file is not None or cert_file is not None or
1399 check_hostname is not None):
1400 import warnings
1401 warnings.warn("key_file, cert_file and check_hostname are "
1402 "deprecated, use a custom context instead.",
1403 DeprecationWarning, 2)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001404 self.key_file = key_file
1405 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001406 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001407 context = ssl._create_default_https_context()
Christian Heimesf97406b2020-11-13 16:37:52 +01001408 # send ALPN extension to indicate HTTP/1.1 protocol
1409 if self._http_vsn == 11:
1410 context.set_alpn_protocols(['http/1.1'])
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001411 # enable PHA for TLS 1.3 connections if available
1412 if context.post_handshake_auth is not None:
1413 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001414 will_verify = context.verify_mode != ssl.CERT_NONE
1415 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001416 check_hostname = context.check_hostname
1417 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001418 raise ValueError("check_hostname needs a SSL context with "
1419 "either CERT_OPTIONAL or CERT_REQUIRED")
1420 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001421 context.load_cert_chain(cert_file, key_file)
Christian Heimesd1bd6e72019-07-01 08:32:24 +02001422 # cert and key file means the user wants to authenticate.
1423 # enable TLS 1.3 PHA implicitly even for custom contexts.
1424 if context.post_handshake_auth is not None:
1425 context.post_handshake_auth = True
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001426 self._context = context
Christian Heimes61d478c2018-01-27 15:51:38 +01001427 if check_hostname is not None:
1428 self._context.check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001429
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001430 def connect(self):
1431 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001432
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001433 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001434
1435 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001436 server_hostname = self._tunnel_host
1437 else:
1438 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001439
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001440 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001441 server_hostname=server_hostname)
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001442
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001443 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001444
Greg Stein5e0fa402000-06-26 08:28:01 +00001445class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001446 # Subclasses that define an __init__ must call Exception.__init__
1447 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001448 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001449
1450class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001451 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001452
Skip Montanaro9d389972002-03-24 16:53:50 +00001453class InvalidURL(HTTPException):
1454 pass
1455
Greg Stein5e0fa402000-06-26 08:28:01 +00001456class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001457 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001458 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001459 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001460
1461class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001462 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001463
Greg Stein5e0fa402000-06-26 08:28:01 +00001464class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001465 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001466
1467class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001468 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001469 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001470 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001471 self.expected = expected
1472 def __repr__(self):
1473 if self.expected is not None:
1474 e = ', %i more expected' % self.expected
1475 else:
1476 e = ''
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001477 return '%s(%i bytes read%s)' % (self.__class__.__name__,
1478 len(self.partial), e)
Serhiy Storchaka96aeaec2019-05-06 22:29:40 +03001479 __str__ = object.__str__
Greg Stein5e0fa402000-06-26 08:28:01 +00001480
1481class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001482 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001483
1484class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001485 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001486
1487class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001488 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001489
1490class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001491 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001492
1493class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001494 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001495 if not line:
1496 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001497 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001498 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001499
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001500class LineTooLong(HTTPException):
1501 def __init__(self, line_type):
1502 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1503 % (_MAXLINE, line_type))
1504
R David Murraycae7bdb2015-04-05 19:26:29 -04001505class RemoteDisconnected(ConnectionResetError, BadStatusLine):
1506 def __init__(self, *pos, **kw):
1507 BadStatusLine.__init__(self, "")
1508 ConnectionResetError.__init__(self, *pos, **kw)
1509
Greg Stein5e0fa402000-06-26 08:28:01 +00001510# for backwards compatibility
1511error = HTTPException