blob: 5aa178d7b127d90de276dbd627c48304f39eacbf [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
Georg Brandl80ba8e82005-09-29 20:16:07 +0000108# maximal amount of data to read at one time in _safe_read
109MAXAMOUNT = 1048576
110
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000111# maximal line length when calling readline().
112_MAXLINE = 65536
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100113_MAXHEADERS = 100
114
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200115# Header name/value ABNF (http://tools.ietf.org/html/rfc7230#section-3.2)
116#
117# VCHAR = %x21-7E
118# obs-text = %x80-FF
119# header-field = field-name ":" OWS field-value OWS
120# field-name = token
121# field-value = *( field-content / obs-fold )
122# field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
123# field-vchar = VCHAR / obs-text
124#
125# obs-fold = CRLF 1*( SP / HTAB )
126# ; obsolete line folding
127# ; see Section 3.2.4
128
129# token = 1*tchar
130#
131# tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*"
132# / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~"
133# / DIGIT / ALPHA
134# ; any VCHAR, except delimiters
135#
136# VCHAR defined in http://tools.ietf.org/html/rfc5234#appendix-B.1
137
Raymond Hettinger15f44ab2016-08-30 10:47:49 -0700138# the patterns for both name and value are more lenient than RFC
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200139# definitions to allow for backwards compatibility
140_is_legal_header_name = re.compile(rb'[^:\s][^:\r\n]*').fullmatch
141_is_illegal_header_value = re.compile(rb'\n(?![ \t])|\r(?![ \t\n])').search
142
R David Murraybeed8402015-03-22 15:18:23 -0400143# We always set the Content-Length header for these methods because some
144# servers will otherwise respond with a 411
145_METHODS_EXPECTING_BODY = {'PATCH', 'POST', 'PUT'}
146
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000147
Martin Panter44391482016-02-09 10:20:52 +0000148def _encode(data, name='data'):
149 """Call data.encode("latin-1") but show a better error message."""
150 try:
151 return data.encode("latin-1")
152 except UnicodeEncodeError as err:
153 raise UnicodeEncodeError(
154 err.encoding,
155 err.object,
156 err.start,
157 err.end,
158 "%s (%.20r) is not valid Latin-1. Use %s.encode('utf-8') "
159 "if you want to send it encoded in UTF-8." %
160 (name.title(), data[err.start:err.end], name)) from None
161
162
Barry Warsaw820c1202008-06-12 04:06:45 +0000163class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000164 # XXX The only usage of this method is in
165 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
166 # that it doesn't need to be part of the public API. The API has
167 # never been defined so this could cause backwards compatibility
168 # issues.
169
Barry Warsaw820c1202008-06-12 04:06:45 +0000170 def getallmatchingheaders(self, name):
171 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000172
Barry Warsaw820c1202008-06-12 04:06:45 +0000173 Look through the list of headers and find all lines matching a given
174 header name (and their continuation lines). A list of the lines is
175 returned, without interpretation. If the header does not occur, an
176 empty list is returned. If the header occurs multiple times, all
177 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000178
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000179 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000180 name = name.lower() + ':'
181 n = len(name)
182 lst = []
183 hit = 0
184 for line in self.keys():
185 if line[:n].lower() == name:
186 hit = 1
187 elif not line[:1].isspace():
188 hit = 0
189 if hit:
190 lst.append(line)
191 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000192
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000193def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000194 """Parses only RFC2822 headers from a file pointer.
195
196 email Parser wants to see strings rather than bytes.
197 But a TextIOWrapper around self.rfile would buffer too many bytes
198 from the stream, bytes which we later need to read as bytes.
199 So we read the correct bytes here, as bytes, for email Parser
200 to parse.
201
202 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000203 headers = []
204 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000205 line = fp.readline(_MAXLINE + 1)
206 if len(line) > _MAXLINE:
207 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000208 headers.append(line)
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100209 if len(headers) > _MAXHEADERS:
210 raise HTTPException("got more than %d headers" % _MAXHEADERS)
Barry Warsaw820c1202008-06-12 04:06:45 +0000211 if line in (b'\r\n', b'\n', b''):
212 break
213 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000214 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000215
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000216
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000217class HTTPResponse(io.BufferedIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000218
219 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
220
Jeremy Hylton811fc142007-08-03 13:30:02 +0000221 # The bytes from the socket object are iso-8859-1 strings.
222 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
223 # text following RFC 2047. The basic status line parsing only
224 # accepts iso-8859-1.
225
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700226 def __init__(self, sock, debuglevel=0, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000227 # If the response includes a content-length header, we need to
228 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000229 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000230 # the server times out and closes the connection. This will
231 # happen if a self.fp.read() is done (without a size) whether
232 # self.fp is buffered or not. So, no self.fp.read() by
233 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000234 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000235 self.debuglevel = debuglevel
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000236 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000237
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000238 # The HTTPResponse object is returned via urllib. The clients
239 # of http and urllib expect different attributes for the
240 # headers. headers is used here and supports urllib. msg is
241 # provided as a backwards compatibility layer for http
242 # clients.
243
244 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000245
Greg Steindd6eefb2000-07-18 09:09:48 +0000246 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000247 self.version = _UNKNOWN # HTTP-Version
248 self.status = _UNKNOWN # Status-Code
249 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000250
Tim Peters07e99cb2001-01-14 23:47:14 +0000251 self.chunked = _UNKNOWN # is "chunked" being used?
252 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
253 self.length = _UNKNOWN # number of bytes left in response
254 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000255
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000256 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000257 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
258 if len(line) > _MAXLINE:
259 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000260 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000261 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000262 if not line:
263 # Presumably, the server closed the connection before
264 # sending a valid response.
R David Murraycae7bdb2015-04-05 19:26:29 -0400265 raise RemoteDisconnected("Remote end closed connection without"
266 " response")
Greg Steindd6eefb2000-07-18 09:09:48 +0000267 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000268 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000269 except ValueError:
270 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000271 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000272 reason = ""
273 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000274 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000275 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000276 if not version.startswith("HTTP/"):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200277 self._close_conn()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000278 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000279
Jeremy Hylton23d40472001-04-13 14:57:08 +0000280 # The status code is a three-digit number
281 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000282 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000283 if status < 100 or status > 999:
284 raise BadStatusLine(line)
285 except ValueError:
286 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000287 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000288
Jeremy Hylton39c03802002-07-12 14:04:09 +0000289 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000290 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000291 # we've already started reading the response
292 return
293
294 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000295 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000296 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000297 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000298 break
299 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000300 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000301 skip = self.fp.readline(_MAXLINE + 1)
302 if len(skip) > _MAXLINE:
303 raise LineTooLong("header line")
304 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000305 if not skip:
306 break
307 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000308 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000309
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000310 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000311 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000312 if version in ("HTTP/1.0", "HTTP/0.9"):
313 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000314 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000315 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000316 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000317 else:
318 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000319
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000320 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000321
Jeremy Hylton30f86742000-09-18 22:50:38 +0000322 if self.debuglevel > 0:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000323 for hdr in self.headers:
Marco Strigl936f03e2018-06-19 15:20:58 +0200324 print("header:", hdr + ":", self.headers.get(hdr))
Greg Stein5e0fa402000-06-26 08:28:01 +0000325
Greg Steindd6eefb2000-07-18 09:09:48 +0000326 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000327 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000328 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000329 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000330 self.chunk_left = None
331 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000332 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000333
Greg Steindd6eefb2000-07-18 09:09:48 +0000334 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000335 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000336
Greg Steindd6eefb2000-07-18 09:09:48 +0000337 # do we have a Content-Length?
338 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000339 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000340 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000341
342 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000343 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000344 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000345 try:
346 self.length = int(length)
347 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000348 self.length = None
349 else:
350 if self.length < 0: # ignore nonsensical negative lengths
351 self.length = None
352 else:
353 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000354
Greg Steindd6eefb2000-07-18 09:09:48 +0000355 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000356 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000357 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000358 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000359 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000360
Greg Steindd6eefb2000-07-18 09:09:48 +0000361 # if the connection remains open, and we aren't using chunked, and
362 # a content-length was not provided, then assume that the connection
363 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000364 if (not self.will_close and
365 not self.chunked and
366 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000367 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000368
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000369 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000370 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000371 if self.version == 11:
372 # An HTTP/1.1 proxy is assumed to stay open unless
373 # explicitly closed.
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000374 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000375 return True
376 return False
377
Jeremy Hylton2c178252004-08-07 16:28:14 +0000378 # Some HTTP/1.0 implementations have support for persistent
379 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000380
Christian Heimes895627f2007-12-08 17:28:33 +0000381 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000382 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000383 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000384
Jeremy Hylton2c178252004-08-07 16:28:14 +0000385 # At least Akamai returns a "Connection: Keep-Alive" header,
386 # which was supposed to be sent by the client.
387 if conn and "keep-alive" in conn.lower():
388 return False
389
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000390 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000391 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000392 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000393 return False
394
395 # otherwise, assume it will close
396 return True
397
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200398 def _close_conn(self):
399 fp = self.fp
400 self.fp = None
401 fp.close()
402
Greg Steindd6eefb2000-07-18 09:09:48 +0000403 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300404 try:
405 super().close() # set "closed" flag
406 finally:
407 if self.fp:
408 self._close_conn()
Greg Stein5e0fa402000-06-26 08:28:01 +0000409
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000410 # These implementations are for the benefit of io.BufferedReader.
411
412 # XXX This class should probably be revised to act more like
413 # the "raw stream" that BufferedReader expects.
414
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000415 def flush(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200416 super().flush()
417 if self.fp:
418 self.fp.flush()
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000419
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000420 def readable(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700421 """Always returns True"""
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000422 return True
423
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000424 # End of "raw stream" methods
425
Greg Steindd6eefb2000-07-18 09:09:48 +0000426 def isclosed(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200427 """True if the connection is closed."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000428 # NOTE: it is possible that we will not ever call self.close(). This
429 # case occurs when will_close is TRUE, length is None, and we
430 # read up to the last byte, but NOT past it.
431 #
432 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
433 # called, meaning self.isclosed() is meaningful.
434 return self.fp is None
435
436 def read(self, amt=None):
437 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000438 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000439
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000440 if self._method == "HEAD":
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200441 self._close_conn()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000442 return b""
443
Antoine Pitrou38d96432011-12-06 22:33:57 +0100444 if amt is not None:
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000445 # Amount is given, implement using readinto
446 b = bytearray(amt)
447 n = self.readinto(b)
448 return memoryview(b)[:n].tobytes()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100449 else:
450 # Amount is not given (unbounded read) so we must check self.length
451 # and self.chunked
Tim Peters230a60c2002-11-09 05:08:07 +0000452
Antoine Pitrou38d96432011-12-06 22:33:57 +0100453 if self.chunked:
454 return self._readall_chunked()
455
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000456 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000457 s = self.fp.read()
458 else:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100459 try:
460 s = self._safe_read(self.length)
461 except IncompleteRead:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200462 self._close_conn()
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100463 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000464 self.length = 0
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200465 self._close_conn() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000466 return s
467
Antoine Pitrou38d96432011-12-06 22:33:57 +0100468 def readinto(self, b):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700469 """Read up to len(b) bytes into bytearray b and return the number
470 of bytes read.
471 """
472
Antoine Pitrou38d96432011-12-06 22:33:57 +0100473 if self.fp is None:
474 return 0
475
476 if self._method == "HEAD":
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200477 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100478 return 0
479
480 if self.chunked:
481 return self._readinto_chunked(b)
482
Greg Steindd6eefb2000-07-18 09:09:48 +0000483 if self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100484 if len(b) > self.length:
Greg Steindd6eefb2000-07-18 09:09:48 +0000485 # clip the read to the "end of response"
Antoine Pitrou38d96432011-12-06 22:33:57 +0100486 b = memoryview(b)[0:self.length]
Greg Steindd6eefb2000-07-18 09:09:48 +0000487
488 # we do not use _safe_read() here because this may be a .will_close
489 # connection, and the user is reading more bytes than will be provided
490 # (for example, reading in 1k chunks)
Antoine Pitrou38d96432011-12-06 22:33:57 +0100491 n = self.fp.readinto(b)
Serhiy Storchaka1c84ac12013-12-17 21:50:02 +0200492 if not n and b:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100493 # Ideally, we would raise IncompleteRead if the content-length
494 # wasn't satisfied, but it might break compatibility.
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200495 self._close_conn()
Antoine Pitrou6a35e182013-02-02 23:04:56 +0100496 elif self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100497 self.length -= n
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000498 if not self.length:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200499 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100500 return n
Greg Steindd6eefb2000-07-18 09:09:48 +0000501
Antoine Pitrou38d96432011-12-06 22:33:57 +0100502 def _read_next_chunk_size(self):
503 # Read the next chunk size from the file
504 line = self.fp.readline(_MAXLINE + 1)
505 if len(line) > _MAXLINE:
506 raise LineTooLong("chunk size")
507 i = line.find(b";")
508 if i >= 0:
509 line = line[:i] # strip chunk-extensions
510 try:
511 return int(line, 16)
512 except ValueError:
513 # close the connection as protocol synchronisation is
514 # probably lost
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200515 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100516 raise
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000517
Antoine Pitrou38d96432011-12-06 22:33:57 +0100518 def _read_and_discard_trailer(self):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000519 # read and discard trailer up to the CRLF terminator
520 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000521 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000522 line = self.fp.readline(_MAXLINE + 1)
523 if len(line) > _MAXLINE:
524 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000525 if not line:
526 # a vanishingly small number of sites EOF without
527 # sending the trailer
528 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800529 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000530 break
531
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000532 def _get_chunk_left(self):
533 # return self.chunk_left, reading a new chunk if necessary.
534 # chunk_left == 0: at the end of the current chunk, need to close it
535 # chunk_left == None: No current chunk, should read next.
536 # This function returns non-zero or None if the last chunk has
537 # been read.
538 chunk_left = self.chunk_left
539 if not chunk_left: # Can be 0 or None
540 if chunk_left is not None:
Mike53f7a7c2017-12-14 14:04:53 +0300541 # We are at the end of chunk, discard chunk end
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000542 self._safe_read(2) # toss the CRLF at the end of the chunk
543 try:
544 chunk_left = self._read_next_chunk_size()
545 except ValueError:
546 raise IncompleteRead(b'')
547 if chunk_left == 0:
548 # last chunk: 1*("0") [ chunk-extension ] CRLF
549 self._read_and_discard_trailer()
550 # we read everything; close the "file"
551 self._close_conn()
552 chunk_left = None
553 self.chunk_left = chunk_left
554 return chunk_left
555
Antoine Pitrou38d96432011-12-06 22:33:57 +0100556 def _readall_chunked(self):
557 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100558 value = []
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000559 try:
560 while True:
561 chunk_left = self._get_chunk_left()
562 if chunk_left is None:
563 break
564 value.append(self._safe_read(chunk_left))
565 self.chunk_left = 0
566 return b''.join(value)
567 except IncompleteRead:
568 raise IncompleteRead(b''.join(value))
Tim Peters230a60c2002-11-09 05:08:07 +0000569
Antoine Pitrou38d96432011-12-06 22:33:57 +0100570 def _readinto_chunked(self, b):
571 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100572 total_bytes = 0
573 mvb = memoryview(b)
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 return total_bytes
Antoine Pitrou4ce6aa42011-12-06 22:34:36 +0100579
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000580 if len(mvb) <= chunk_left:
581 n = self._safe_readinto(mvb)
582 self.chunk_left = chunk_left - n
583 return total_bytes + n
584
585 temp_mvb = mvb[:chunk_left]
Antoine Pitrou38d96432011-12-06 22:33:57 +0100586 n = self._safe_readinto(temp_mvb)
587 mvb = mvb[n:]
588 total_bytes += n
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000589 self.chunk_left = 0
Antoine Pitrou38d96432011-12-06 22:33:57 +0100590
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000591 except IncompleteRead:
592 raise IncompleteRead(bytes(b[0:total_bytes]))
Antoine Pitrou38d96432011-12-06 22:33:57 +0100593
Greg Steindd6eefb2000-07-18 09:09:48 +0000594 def _safe_read(self, amt):
595 """Read the number of bytes requested, compensating for partial reads.
596
597 Normally, we have a blocking socket, but a read() can be interrupted
598 by a signal (resulting in a partial read).
599
600 Note that we cannot distinguish between EOF and an interrupt when zero
601 bytes have been read. IncompleteRead() will be raised in this
602 situation.
603
604 This function should be used when <amt> bytes "should" be present for
605 reading. If the bytes are truly not available (due to EOF), then the
606 IncompleteRead exception can be used to detect the problem.
607 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000608 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000609 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000610 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000611 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000612 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000613 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000614 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000615 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000616
Antoine Pitrou38d96432011-12-06 22:33:57 +0100617 def _safe_readinto(self, b):
618 """Same as _safe_read, but for reading into a buffer."""
619 total_bytes = 0
620 mvb = memoryview(b)
621 while total_bytes < len(b):
622 if MAXAMOUNT < len(mvb):
623 temp_mvb = mvb[0:MAXAMOUNT]
624 n = self.fp.readinto(temp_mvb)
625 else:
626 n = self.fp.readinto(mvb)
627 if not n:
628 raise IncompleteRead(bytes(mvb[0:total_bytes]), len(b))
629 mvb = mvb[n:]
630 total_bytes += n
631 return total_bytes
632
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000633 def read1(self, n=-1):
634 """Read with at most one underlying system call. If at least one
635 byte is buffered, return that instead.
636 """
637 if self.fp is None or self._method == "HEAD":
638 return b""
639 if self.chunked:
640 return self._read1_chunked(n)
Martin Panterce911c32016-03-17 06:42:48 +0000641 if self.length is not None and (n < 0 or n > self.length):
642 n = self.length
Martin Panter741d4942016-10-21 00:52:04 +0000643 result = self.fp.read1(n)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000644 if not result and n:
645 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000646 elif self.length is not None:
647 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000648 return result
649
650 def peek(self, n=-1):
651 # Having this enables IOBase.readline() to read more than one
652 # byte at a time
653 if self.fp is None or self._method == "HEAD":
654 return b""
655 if self.chunked:
656 return self._peek_chunked(n)
657 return self.fp.peek(n)
658
659 def readline(self, limit=-1):
660 if self.fp is None or self._method == "HEAD":
661 return b""
662 if self.chunked:
663 # Fallback to IOBase readline which uses peek() and read()
664 return super().readline(limit)
Martin Panterce911c32016-03-17 06:42:48 +0000665 if self.length is not None and (limit < 0 or limit > self.length):
666 limit = self.length
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000667 result = self.fp.readline(limit)
668 if not result and limit:
669 self._close_conn()
Martin Panterce911c32016-03-17 06:42:48 +0000670 elif self.length is not None:
671 self.length -= len(result)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000672 return result
673
674 def _read1_chunked(self, n):
675 # Strictly speaking, _get_chunk_left() may cause more than one read,
676 # but that is ok, since that is to satisfy the chunked protocol.
677 chunk_left = self._get_chunk_left()
678 if chunk_left is None or n == 0:
679 return b''
680 if not (0 <= n <= chunk_left):
681 n = chunk_left # if n is negative or larger than chunk_left
682 read = self.fp.read1(n)
683 self.chunk_left -= len(read)
684 if not read:
685 raise IncompleteRead(b"")
686 return read
687
688 def _peek_chunked(self, n):
689 # Strictly speaking, _get_chunk_left() may cause more than one read,
690 # but that is ok, since that is to satisfy the chunked protocol.
691 try:
692 chunk_left = self._get_chunk_left()
693 except IncompleteRead:
694 return b'' # peek doesn't worry about protocol
695 if chunk_left is None:
696 return b'' # eof
697 # peek is allowed to return more than requested. Just request the
698 # entire chunk, and truncate what we get.
699 return self.fp.peek(chunk_left)[:chunk_left]
700
Antoine Pitroub353c122009-02-11 00:39:14 +0000701 def fileno(self):
702 return self.fp.fileno()
703
Greg Steindd6eefb2000-07-18 09:09:48 +0000704 def getheader(self, name, default=None):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700705 '''Returns the value of the header matching *name*.
706
707 If there are multiple matching headers, the values are
708 combined into a single string separated by commas and spaces.
709
710 If no matching header is found, returns *default* or None if
711 the *default* is not specified.
712
713 If the headers are unknown, raises http.client.ResponseNotReady.
714
715 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000716 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000717 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000718 headers = self.headers.get_all(name) or default
719 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
720 return headers
721 else:
722 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000723
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000724 def getheaders(self):
725 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000726 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000727 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000728 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000729
Antoine Pitroub353c122009-02-11 00:39:14 +0000730 # We override IOBase.__iter__ so that it doesn't check for closed-ness
731
732 def __iter__(self):
733 return self
734
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000735 # For compatibility with old-style urllib responses.
736
737 def info(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700738 '''Returns an instance of the class mimetools.Message containing
739 meta-information associated with the URL.
740
741 When the method is HTTP, these headers are those returned by
742 the server at the head of the retrieved HTML page (including
743 Content-Length and Content-Type).
744
745 When the method is FTP, a Content-Length header will be
746 present if (as is now usual) the server passed back a file
747 length in response to the FTP retrieval request. A
748 Content-Type header will be present if the MIME type can be
749 guessed.
750
751 When the method is local-file, returned headers will include
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200752 a Date representing the file's last-modified time, a
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700753 Content-Length giving file size, and a Content-Type
Serhiy Storchaka3fd4a732015-12-18 13:10:37 +0200754 containing a guess at the file's type. See also the
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700755 description of the mimetools module.
756
757 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000758 return self.headers
759
760 def geturl(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700761 '''Return the real URL of the page.
762
763 In some cases, the HTTP server redirects a client to another
764 URL. The urlopen() function handles this transparently, but in
765 some cases the caller needs to know which URL the client was
766 redirected to. The geturl() method can be used to get at this
767 redirected URL.
768
769 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000770 return self.url
771
772 def getcode(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700773 '''Return the HTTP status code that was sent with the response,
774 or None if the URL is not an HTTP URL.
775
776 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000777 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000778
779class HTTPConnection:
780
Greg Steindd6eefb2000-07-18 09:09:48 +0000781 _http_vsn = 11
782 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000783
Greg Steindd6eefb2000-07-18 09:09:48 +0000784 response_class = HTTPResponse
785 default_port = HTTP_PORT
786 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000787 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000788
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000789 @staticmethod
790 def _is_textIO(stream):
791 """Test whether a file-like object is a text or a binary stream.
792 """
793 return isinstance(stream, io.TextIOBase)
794
795 @staticmethod
796 def _get_content_length(body, method):
797 """Get the content-length based on the body.
798
Martin Panteref91bb22016-08-27 01:39:26 +0000799 If the body is None, we set Content-Length: 0 for methods that expect
800 a body (RFC 7230, Section 3.3.2). We also set the Content-Length for
801 any method if the body is a str or bytes-like object and not a file.
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000802 """
Martin Panteref91bb22016-08-27 01:39:26 +0000803 if body is None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000804 # do an explicit check for not None here to distinguish
805 # between unset and set but empty
Martin Panteref91bb22016-08-27 01:39:26 +0000806 if method.upper() in _METHODS_EXPECTING_BODY:
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000807 return 0
808 else:
809 return None
810
811 if hasattr(body, 'read'):
812 # file-like object.
Martin Panteref91bb22016-08-27 01:39:26 +0000813 return None
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000814
815 try:
816 # does it implement the buffer protocol (bytes, bytearray, array)?
817 mv = memoryview(body)
818 return mv.nbytes
819 except TypeError:
820 pass
821
822 if isinstance(body, str):
823 return len(body)
824
825 return None
826
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700827 def __init__(self, host, port=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
Nir Sofferad455cd2017-11-06 23:16:37 +0200828 source_address=None, blocksize=8192):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000829 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000830 self.source_address = source_address
Nir Sofferad455cd2017-11-06 23:16:37 +0200831 self.blocksize = blocksize
Greg Steindd6eefb2000-07-18 09:09:48 +0000832 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000833 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000834 self.__response = None
835 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000836 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000837 self._tunnel_host = None
838 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000839 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000840
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400841 (self.host, self.port) = self._get_hostport(host, port)
842
843 # This is stored as an instance variable to allow unit
844 # tests to replace it with a suitable mockup
845 self._create_connection = socket.create_connection
Greg Stein5e0fa402000-06-26 08:28:01 +0000846
Senthil Kumaran47fff872009-12-20 07:10:31 +0000847 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400848 """Set up host and port for HTTP CONNECT tunnelling.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000849
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400850 In a connection that uses HTTP CONNECT tunneling, the host passed to the
851 constructor is used as a proxy server that relays all communication to
852 the endpoint passed to `set_tunnel`. This done by sending an HTTP
853 CONNECT request to the proxy server when the connection is established.
854
855 This method must be called before the HTML connection has been
856 established.
857
858 The headers argument should be a mapping of extra HTTP headers to send
859 with the CONNECT request.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000860 """
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400861
862 if self.sock:
863 raise RuntimeError("Can't set up tunnel for established connection")
864
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200865 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000866 if headers:
867 self._tunnel_headers = headers
868 else:
869 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000870
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400871 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000872 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000873 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000874 j = host.rfind(']') # ipv6 addresses have [...]
875 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000876 try:
877 port = int(host[i+1:])
878 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200879 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
880 port = self.default_port
881 else:
882 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000883 host = host[:i]
884 else:
885 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000886 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000887 host = host[1:-1]
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400888
889 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000890
Jeremy Hylton30f86742000-09-18 22:50:38 +0000891 def set_debuglevel(self, level):
892 self.debuglevel = level
893
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000894 def _tunnel(self):
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200895 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host,
896 self._tunnel_port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000897 connect_bytes = connect_str.encode("ascii")
898 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000899 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000900 header_str = "%s: %s\r\n" % (header, value)
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000901 header_bytes = header_str.encode("latin-1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000902 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000903 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000904
Georg Brandlc7c199f2011-01-22 22:06:24 +0000905 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000906 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000907
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200908 if code != http.HTTPStatus.OK:
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000909 self.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +0200910 raise OSError("Tunnel connection failed: %d %s" % (code,
911 message.strip()))
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000912 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000913 line = response.fp.readline(_MAXLINE + 1)
914 if len(line) > _MAXLINE:
915 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800916 if not line:
917 # for sites which EOF without sending a trailer
918 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800919 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000920 break
921
Berker Peksagab53ab02015-02-03 12:22:11 +0200922 if self.debuglevel > 0:
923 print('header:', line.decode())
924
Greg Steindd6eefb2000-07-18 09:09:48 +0000925 def connect(self):
926 """Connect to the host and port specified in __init__."""
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500927 self.sock = self._create_connection(
928 (self.host,self.port), self.timeout, self.source_address)
929 self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400930
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000931 if self._tunnel_host:
932 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000933
Greg Steindd6eefb2000-07-18 09:09:48 +0000934 def close(self):
935 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000936 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300937 try:
938 sock = self.sock
939 if sock:
940 self.sock = None
941 sock.close() # close it manually... there may be other refs
942 finally:
943 response = self.__response
944 if response:
945 self.__response = None
946 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000947
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000948 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000949 """Send `data' to the server.
950 ``data`` can be a string object, a bytes object, an array object, a
951 file-like object that supports a .read() method, or an iterable object.
952 """
953
Greg Steindd6eefb2000-07-18 09:09:48 +0000954 if self.sock is None:
955 if self.auto_open:
956 self.connect()
957 else:
958 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000959
Jeremy Hylton30f86742000-09-18 22:50:38 +0000960 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000961 print("send:", repr(data))
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000962 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000963 if self.debuglevel > 0:
964 print("sendIng a read()able")
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000965 encode = self._is_textIO(data)
966 if encode and self.debuglevel > 0:
967 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000968 while 1:
Nir Sofferad455cd2017-11-06 23:16:37 +0200969 datablock = data.read(self.blocksize)
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000970 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000971 break
972 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000973 datablock = datablock.encode("iso-8859-1")
974 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300975 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000976 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000977 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000978 except TypeError:
Serhiy Storchaka2e576f52017-04-24 09:05:00 +0300979 if isinstance(data, collections.abc.Iterable):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000980 for d in data:
981 self.sock.sendall(d)
982 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200983 raise TypeError("data should be a bytes-like object "
984 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000985
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000986 def _output(self, s):
987 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000988
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000989 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000990 """
991 self._buffer.append(s)
992
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000993 def _read_readable(self, readable):
Martin Panter3c0d0ba2016-08-24 06:33:33 +0000994 if self.debuglevel > 0:
995 print("sendIng a read()able")
996 encode = self._is_textIO(readable)
997 if encode and self.debuglevel > 0:
998 print("encoding file using iso-8859-1")
999 while True:
Nir Sofferad455cd2017-11-06 23:16:37 +02001000 datablock = readable.read(self.blocksize)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001001 if not datablock:
1002 break
1003 if encode:
1004 datablock = datablock.encode("iso-8859-1")
1005 yield datablock
1006
1007 def _send_output(self, message_body=None, encode_chunked=False):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001008 """Send the currently buffered request and clear the buffer.
1009
Jeremy Hyltone3252ec2002-07-16 21:41:43 +00001010 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001011 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001012 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001013 self._buffer.extend((b"", b""))
1014 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001015 del self._buffer[:]
1016 self.send(msg)
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001017
Benjamin Peterson822b21c2009-01-18 00:04:57 +00001018 if message_body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001019
1020 # create a consistent interface to message_body
1021 if hasattr(message_body, 'read'):
1022 # Let file-like take precedence over byte-like. This
1023 # is needed to allow the current position of mmap'ed
1024 # files to be taken into account.
1025 chunks = self._read_readable(message_body)
1026 else:
1027 try:
1028 # this is solely to check to see if message_body
1029 # implements the buffer API. it /would/ be easier
1030 # to capture if PyObject_CheckBuffer was exposed
1031 # to Python.
1032 memoryview(message_body)
1033 except TypeError:
1034 try:
1035 chunks = iter(message_body)
1036 except TypeError:
1037 raise TypeError("message_body should be a bytes-like "
1038 "object or an iterable, got %r"
1039 % type(message_body))
1040 else:
1041 # the object implements the buffer interface and
1042 # can be passed directly into socket methods
1043 chunks = (message_body,)
1044
1045 for chunk in chunks:
1046 if not chunk:
1047 if self.debuglevel > 0:
1048 print('Zero length chunk ignored')
1049 continue
1050
1051 if encode_chunked and self._http_vsn == 11:
1052 # chunked encoding
Eric V. Smith451d0e32016-09-09 21:56:20 -04001053 chunk = f'{len(chunk):X}\r\n'.encode('ascii') + chunk \
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001054 + b'\r\n'
1055 self.send(chunk)
1056
1057 if encode_chunked and self._http_vsn == 11:
1058 # end chunked transfer
1059 self.send(b'0\r\n\r\n')
Jeremy Hylton8531b1b2002-07-16 21:21:11 +00001060
Senthil Kumaran5dc504c2016-09-08 14:28:01 -07001061 def putrequest(self, method, url, skip_host=False,
1062 skip_accept_encoding=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001063 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001064
Greg Steindd6eefb2000-07-18 09:09:48 +00001065 `method' specifies an HTTP request method, e.g. 'GET'.
1066 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001067 `skip_host' if True does not add automatically a 'Host:' header
1068 `skip_accept_encoding' if True does not add automatically an
1069 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +00001070 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001071
Greg Stein616a58d2003-06-24 06:35:19 +00001072 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001073 if self.__response and self.__response.isclosed():
1074 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001075
Tim Peters58eb11c2004-01-18 20:29:55 +00001076
Greg Steindd6eefb2000-07-18 09:09:48 +00001077 # in certain cases, we cannot issue another request on this connection.
1078 # this occurs when:
1079 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
1080 # 2) a response to a previous request has signalled that it is going
1081 # to close the connection upon completion.
1082 # 3) the headers for the previous response have not been read, thus
1083 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
1084 #
1085 # if there is no prior response, then we can request at will.
1086 #
1087 # if point (2) is true, then we will have passed the socket to the
1088 # response (effectively meaning, "there is no prior response"), and
1089 # will open a new one when a new request is made.
1090 #
1091 # Note: if a prior response exists, then we *can* start a new request.
1092 # We are not allowed to begin fetching the response to this new
1093 # request, however, until that prior response is complete.
1094 #
1095 if self.__state == _CS_IDLE:
1096 self.__state = _CS_REQ_STARTED
1097 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001098 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001099
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001100 # Save the method we use, we need it later in the response phase
1101 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +00001102 if not url:
1103 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001104 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001105
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001106 # Non-ASCII characters should have been eliminated earlier
1107 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +00001108
Greg Steindd6eefb2000-07-18 09:09:48 +00001109 if self._http_vsn == 11:
1110 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001111
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001112 if not skip_host:
1113 # this header is issued *only* for HTTP/1.1
1114 # connections. more specifically, this means it is
1115 # only issued when the client uses the new
1116 # HTTPConnection() class. backwards-compat clients
1117 # will be using HTTP/1.0 and those clients may be
1118 # issuing this header themselves. we should NOT issue
1119 # it twice; some web servers (such as Apache) barf
1120 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001121
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001122 # If we need a non-standard port,include it in the
1123 # header. If the request is going through a proxy,
1124 # but the host of the actual URL, not the host of the
1125 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001126
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001127 netloc = ''
1128 if url.startswith('http'):
1129 nil, netloc, nil, nil, nil = urlsplit(url)
1130
1131 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001132 try:
1133 netloc_enc = netloc.encode("ascii")
1134 except UnicodeEncodeError:
1135 netloc_enc = netloc.encode("idna")
1136 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001137 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001138 if self._tunnel_host:
1139 host = self._tunnel_host
1140 port = self._tunnel_port
1141 else:
1142 host = self.host
1143 port = self.port
1144
Thomas Wouters477c8d52006-05-27 19:21:47 +00001145 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001146 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001147 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001148 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001149
1150 # As per RFC 273, IPv6 address should be wrapped with []
1151 # when used as Host header
1152
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001153 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001154 host_enc = b'[' + host_enc + b']'
1155
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001156 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001157 self.putheader('Host', host_enc)
1158 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001159 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001160 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001161
Greg Steindd6eefb2000-07-18 09:09:48 +00001162 # note: we are assuming that clients will not attempt to set these
1163 # headers since *this* library must deal with the
1164 # consequences. this also means that when the supporting
1165 # libraries are updated to recognize other forms, then this
1166 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001167
Greg Steindd6eefb2000-07-18 09:09:48 +00001168 # we only want a Content-Encoding of "identity" since we don't
1169 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001170 if not skip_accept_encoding:
1171 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001172
Greg Steindd6eefb2000-07-18 09:09:48 +00001173 # we can accept "chunked" Transfer-Encodings, but no others
1174 # NOTE: no TE header implies *only* "chunked"
1175 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001176
Greg Steindd6eefb2000-07-18 09:09:48 +00001177 # if TE is supplied in the header, then it must appear in a
1178 # Connection header.
1179 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001180
Greg Steindd6eefb2000-07-18 09:09:48 +00001181 else:
1182 # For HTTP/1.0, the server will assume "not chunked"
1183 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001184
Benjamin Petersonf608c612008-11-16 18:33:53 +00001185 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001186 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001187
Greg Steindd6eefb2000-07-18 09:09:48 +00001188 For example: h.putheader('Accept', 'text/html')
1189 """
1190 if self.__state != _CS_REQ_STARTED:
1191 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001192
Guido van Rossum98297ee2007-11-06 21:34:58 +00001193 if hasattr(header, 'encode'):
1194 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001195
1196 if not _is_legal_header_name(header):
1197 raise ValueError('Invalid header name %r' % (header,))
1198
Benjamin Petersonf608c612008-11-16 18:33:53 +00001199 values = list(values)
1200 for i, one_value in enumerate(values):
1201 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001202 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001203 elif isinstance(one_value, int):
1204 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001205
1206 if _is_illegal_header_value(values[i]):
1207 raise ValueError('Invalid header value %r' % (values[i],))
1208
Benjamin Petersonf608c612008-11-16 18:33:53 +00001209 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001210 header = header + b': ' + value
1211 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001212
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001213 def endheaders(self, message_body=None, *, encode_chunked=False):
Benjamin Peterson1742e402008-11-30 22:15:29 +00001214 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001215
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001216 This method sends the request to the server. The optional message_body
1217 argument can be used to pass a message body associated with the
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001218 request.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001219 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001220 if self.__state == _CS_REQ_STARTED:
1221 self.__state = _CS_REQ_SENT
1222 else:
1223 raise CannotSendHeader()
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001224 self._send_output(message_body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001225
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001226 def request(self, method, url, body=None, headers={}, *,
1227 encode_chunked=False):
Greg Steindd6eefb2000-07-18 09:09:48 +00001228 """Send a complete request to the server."""
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001229 self._send_request(method, url, body, headers, encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001230
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001231 def _send_request(self, method, url, body, headers, encode_chunked):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001232 # Honor explicitly requested Host: and Accept-Encoding: headers.
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001233 header_names = frozenset(k.lower() for k in headers)
Jeremy Hylton2c178252004-08-07 16:28:14 +00001234 skips = {}
1235 if 'host' in header_names:
1236 skips['skip_host'] = 1
1237 if 'accept-encoding' in header_names:
1238 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001239
Jeremy Hylton2c178252004-08-07 16:28:14 +00001240 self.putrequest(method, url, **skips)
1241
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001242 # chunked encoding will happen if HTTP/1.1 is used and either
1243 # the caller passes encode_chunked=True or the following
1244 # conditions hold:
1245 # 1. content-length has not been explicitly set
Martin Panteref91bb22016-08-27 01:39:26 +00001246 # 2. the body is a file or iterable, but not a str or bytes-like
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001247 # 3. Transfer-Encoding has NOT been explicitly set by the caller
1248
R David Murraybeed8402015-03-22 15:18:23 -04001249 if 'content-length' not in header_names:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001250 # only chunk body if not explicitly set for backwards
1251 # compatibility, assuming the client code is already handling the
1252 # chunking
1253 if 'transfer-encoding' not in header_names:
1254 # if content-length cannot be automatically determined, fall
1255 # back to chunked encoding
1256 encode_chunked = False
1257 content_length = self._get_content_length(body, method)
1258 if content_length is None:
Martin Panteref91bb22016-08-27 01:39:26 +00001259 if body is not None:
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001260 if self.debuglevel > 0:
1261 print('Unable to determine size of %r' % body)
1262 encode_chunked = True
1263 self.putheader('Transfer-Encoding', 'chunked')
1264 else:
1265 self.putheader('Content-Length', str(content_length))
1266 else:
1267 encode_chunked = False
1268
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001269 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001270 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001271 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001272 # RFC 2616 Section 3.7.1 says that text default has a
1273 # default charset of iso-8859-1.
Martin Panter44391482016-02-09 10:20:52 +00001274 body = _encode(body, 'body')
Martin Panter3c0d0ba2016-08-24 06:33:33 +00001275 self.endheaders(body, encode_chunked=encode_chunked)
Greg Stein5e0fa402000-06-26 08:28:01 +00001276
Greg Steindd6eefb2000-07-18 09:09:48 +00001277 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001278 """Get the response from the server.
1279
1280 If the HTTPConnection is in the correct state, returns an
1281 instance of HTTPResponse or of whatever object is returned by
Martin Pantercc71a792016-04-05 06:19:42 +00001282 the response_class variable.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001283
1284 If a request has not been sent or if a previous response has
1285 not be handled, ResponseNotReady is raised. If the HTTP
1286 response indicates that the connection should be closed, then
1287 it will be closed before the response is returned. When the
1288 connection is closed, the underlying socket is closed.
1289 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001290
Greg Stein616a58d2003-06-24 06:35:19 +00001291 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001292 if self.__response and self.__response.isclosed():
1293 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001294
Greg Steindd6eefb2000-07-18 09:09:48 +00001295 # if a prior response exists, then it must be completed (otherwise, we
1296 # cannot read this response's header to determine the connection-close
1297 # behavior)
1298 #
1299 # note: if a prior response existed, but was connection-close, then the
1300 # socket and response were made independent of this HTTPConnection
1301 # object since a new request requires that we open a whole new
1302 # connection
1303 #
1304 # this means the prior response had one of two states:
1305 # 1) will_close: this connection was reset and the prior socket and
1306 # response operate independently
1307 # 2) persistent: the response was retained and we await its
1308 # isclosed() status to become true.
1309 #
1310 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001311 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001312
Jeremy Hylton30f86742000-09-18 22:50:38 +00001313 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001314 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001315 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001316 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001317 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001318
Serhiy Storchakab491e052014-12-01 13:07:45 +02001319 try:
R David Murraycae7bdb2015-04-05 19:26:29 -04001320 try:
1321 response.begin()
1322 except ConnectionError:
1323 self.close()
1324 raise
Serhiy Storchakab491e052014-12-01 13:07:45 +02001325 assert response.will_close != _UNKNOWN
1326 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001327
Serhiy Storchakab491e052014-12-01 13:07:45 +02001328 if response.will_close:
1329 # this effectively passes the connection to the response
1330 self.close()
1331 else:
1332 # remember this, so we can tell when it is complete
1333 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001334
Serhiy Storchakab491e052014-12-01 13:07:45 +02001335 return response
1336 except:
1337 response.close()
1338 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001339
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001340try:
1341 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001342except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001343 pass
1344else:
1345 class HTTPSConnection(HTTPConnection):
1346 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001347
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001348 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001349
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001350 # XXX Should key_file and cert_file be deprecated in favour of context?
1351
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001352 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001353 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1354 source_address=None, *, context=None,
Nir Sofferad455cd2017-11-06 23:16:37 +02001355 check_hostname=None, blocksize=8192):
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001356 super(HTTPSConnection, self).__init__(host, port, timeout,
Nir Sofferad455cd2017-11-06 23:16:37 +02001357 source_address,
1358 blocksize=blocksize)
Christian Heimesd0486372016-09-10 23:23:33 +02001359 if (key_file is not None or cert_file is not None or
1360 check_hostname is not None):
1361 import warnings
1362 warnings.warn("key_file, cert_file and check_hostname are "
1363 "deprecated, use a custom context instead.",
1364 DeprecationWarning, 2)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001365 self.key_file = key_file
1366 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001367 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001368 context = ssl._create_default_https_context()
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001369 will_verify = context.verify_mode != ssl.CERT_NONE
1370 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001371 check_hostname = context.check_hostname
1372 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001373 raise ValueError("check_hostname needs a SSL context with "
1374 "either CERT_OPTIONAL or CERT_REQUIRED")
1375 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001376 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001377 self._context = context
Christian Heimes61d478c2018-01-27 15:51:38 +01001378 if check_hostname is not None:
1379 self._context.check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001380
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001381 def connect(self):
1382 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001383
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001384 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001385
1386 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001387 server_hostname = self._tunnel_host
1388 else:
1389 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001390
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001391 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001392 server_hostname=server_hostname)
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001393
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001394 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001395
Greg Stein5e0fa402000-06-26 08:28:01 +00001396class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001397 # Subclasses that define an __init__ must call Exception.__init__
1398 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001399 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001400
1401class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001402 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001403
Skip Montanaro9d389972002-03-24 16:53:50 +00001404class InvalidURL(HTTPException):
1405 pass
1406
Greg Stein5e0fa402000-06-26 08:28:01 +00001407class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001408 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001409 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001410 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001411
1412class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001413 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001414
Greg Stein5e0fa402000-06-26 08:28:01 +00001415class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001416 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001417
1418class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001419 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001420 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001421 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001422 self.expected = expected
1423 def __repr__(self):
1424 if self.expected is not None:
1425 e = ', %i more expected' % self.expected
1426 else:
1427 e = ''
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001428 return '%s(%i bytes read%s)' % (self.__class__.__name__,
1429 len(self.partial), e)
Benjamin Peterson6accb982009-03-02 22:50:25 +00001430 def __str__(self):
1431 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001432
1433class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001434 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001435
1436class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001437 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001438
1439class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001440 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001441
1442class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001443 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001444
1445class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001446 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001447 if not line:
1448 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001449 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001450 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001451
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001452class LineTooLong(HTTPException):
1453 def __init__(self, line_type):
1454 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1455 % (_MAXLINE, line_type))
1456
R David Murraycae7bdb2015-04-05 19:26:29 -04001457class RemoteDisconnected(ConnectionResetError, BadStatusLine):
1458 def __init__(self, *pos, **kw):
1459 BadStatusLine.__init__(self, "")
1460 ConnectionResetError.__init__(self, *pos, **kw)
1461
Greg Stein5e0fa402000-06-26 08:28:01 +00001462# for backwards compatibility
1463error = HTTPException