blob: 155c2e3ec4fe9796aae43c3906c936b00f0f21cf [file] [log] [blame]
Greg Stein5e0fa402000-06-26 08:28:01 +00001"""HTTP/1.1 client library
Guido van Rossum41999c11997-12-09 00:12:23 +00002
Greg Stein5e0fa402000-06-26 08:28:01 +00003<intro stuff goes here>
4<other stuff, too>
Guido van Rossum41999c11997-12-09 00:12:23 +00005
Thomas Wouters0e3f5912006-08-11 14:57:12 +00006HTTPConnection goes through a number of "states", which define when a client
Greg Stein5e0fa402000-06-26 08:28:01 +00007may legally make another request or fetch the response for a particular
8request. This diagram details these state transitions:
Guido van Rossum41999c11997-12-09 00:12:23 +00009
Greg Stein5e0fa402000-06-26 08:28:01 +000010 (null)
11 |
12 | HTTPConnection()
13 v
14 Idle
15 |
16 | putrequest()
17 v
18 Request-started
19 |
20 | ( putheader() )* endheaders()
21 v
22 Request-sent
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
75import os
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +020076import re
Jeremy Hylton636950f2009-03-28 04:34:21 +000077import socket
Senthil Kumaran7bc0d872010-12-19 10:49:52 +000078import collections
Jeremy Hylton1afc1692008-06-18 20:49:58 +000079from urllib.parse import urlsplit
Guido van Rossum23acc951994-02-21 16:36:04 +000080
Berker Peksagbabc6882015-02-20 09:39:38 +020081# HTTPMessage, parse_headers(), and the HTTP status code constants are
82# intentionally omitted for simplicity
Thomas Wouters47b49bf2007-08-30 22:15:33 +000083__all__ = ["HTTPResponse", "HTTPConnection",
Skip Montanaro951a8842001-06-01 16:25:38 +000084 "HTTPException", "NotConnected", "UnknownProtocol",
Jeremy Hylton7c75c992002-06-28 23:38:14 +000085 "UnknownTransferEncoding", "UnimplementedFileMode",
86 "IncompleteRead", "InvalidURL", "ImproperConnectionState",
87 "CannotSendRequest", "CannotSendHeader", "ResponseNotReady",
R David Murraycae7bdb2015-04-05 19:26:29 -040088 "BadStatusLine", "LineTooLong", "RemoteDisconnected", "error",
89 "responses"]
Skip Montanaro2dd42762001-01-23 15:35:05 +000090
Guido van Rossum23acc951994-02-21 16:36:04 +000091HTTP_PORT = 80
Guido van Rossum09c8b6c1999-12-07 21:37:17 +000092HTTPS_PORT = 443
93
Greg Stein5e0fa402000-06-26 08:28:01 +000094_UNKNOWN = 'UNKNOWN'
95
96# connection states
97_CS_IDLE = 'Idle'
98_CS_REQ_STARTED = 'Request-started'
99_CS_REQ_SENT = 'Request-sent'
100
Martin v. Löwis39a31782004-09-18 09:03:49 +0000101
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200102# hack to maintain backwards compatibility
103globals().update(http.HTTPStatus.__members__)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000104
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200105# another hack to maintain backwards compatibility
Georg Brandl6aab16e2006-02-17 19:17:25 +0000106# Mapping status codes to official W3C names
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200107responses = {v: v.phrase for v in http.HTTPStatus.__members__.values()}
Georg Brandl6aab16e2006-02-17 19:17:25 +0000108
Georg Brandl80ba8e82005-09-29 20:16:07 +0000109# maximal amount of data to read at one time in _safe_read
110MAXAMOUNT = 1048576
111
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000112# maximal line length when calling readline().
113_MAXLINE = 65536
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100114_MAXHEADERS = 100
115
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +0200116# Header name/value ABNF (http://tools.ietf.org/html/rfc7230#section-3.2)
117#
118# VCHAR = %x21-7E
119# obs-text = %x80-FF
120# header-field = field-name ":" OWS field-value OWS
121# field-name = token
122# field-value = *( field-content / obs-fold )
123# field-content = field-vchar [ 1*( SP / HTAB ) field-vchar ]
124# field-vchar = VCHAR / obs-text
125#
126# obs-fold = CRLF 1*( SP / HTAB )
127# ; obsolete line folding
128# ; see Section 3.2.4
129
130# token = 1*tchar
131#
132# tchar = "!" / "#" / "$" / "%" / "&" / "'" / "*"
133# / "+" / "-" / "." / "^" / "_" / "`" / "|" / "~"
134# / DIGIT / ALPHA
135# ; any VCHAR, except delimiters
136#
137# VCHAR defined in http://tools.ietf.org/html/rfc5234#appendix-B.1
138
139# the patterns for both name and value are more leniant than RFC
140# definitions to allow for backwards compatibility
141_is_legal_header_name = re.compile(rb'[^:\s][^:\r\n]*').fullmatch
142_is_illegal_header_value = re.compile(rb'\n(?![ \t])|\r(?![ \t\n])').search
143
R David Murraybeed8402015-03-22 15:18:23 -0400144# We always set the Content-Length header for these methods because some
145# servers will otherwise respond with a 411
146_METHODS_EXPECTING_BODY = {'PATCH', 'POST', 'PUT'}
147
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000148
Barry Warsaw820c1202008-06-12 04:06:45 +0000149class HTTPMessage(email.message.Message):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000150 # XXX The only usage of this method is in
151 # http.server.CGIHTTPRequestHandler. Maybe move the code there so
152 # that it doesn't need to be part of the public API. The API has
153 # never been defined so this could cause backwards compatibility
154 # issues.
155
Barry Warsaw820c1202008-06-12 04:06:45 +0000156 def getallmatchingheaders(self, name):
157 """Find all header lines matching a given header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000158
Barry Warsaw820c1202008-06-12 04:06:45 +0000159 Look through the list of headers and find all lines matching a given
160 header name (and their continuation lines). A list of the lines is
161 returned, without interpretation. If the header does not occur, an
162 empty list is returned. If the header occurs multiple times, all
163 occurrences are returned. Case is not important in the header name.
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000164
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000165 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000166 name = name.lower() + ':'
167 n = len(name)
168 lst = []
169 hit = 0
170 for line in self.keys():
171 if line[:n].lower() == name:
172 hit = 1
173 elif not line[:1].isspace():
174 hit = 0
175 if hit:
176 lst.append(line)
177 return lst
Jeremy Hylton6d0a4c72002-07-07 16:51:37 +0000178
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000179def parse_headers(fp, _class=HTTPMessage):
Barry Warsaw820c1202008-06-12 04:06:45 +0000180 """Parses only RFC2822 headers from a file pointer.
181
182 email Parser wants to see strings rather than bytes.
183 But a TextIOWrapper around self.rfile would buffer too many bytes
184 from the stream, bytes which we later need to read as bytes.
185 So we read the correct bytes here, as bytes, for email Parser
186 to parse.
187
188 """
Barry Warsaw820c1202008-06-12 04:06:45 +0000189 headers = []
190 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000191 line = fp.readline(_MAXLINE + 1)
192 if len(line) > _MAXLINE:
193 raise LineTooLong("header line")
Barry Warsaw820c1202008-06-12 04:06:45 +0000194 headers.append(line)
Georg Brandlbf3f8eb2013-10-27 07:34:48 +0100195 if len(headers) > _MAXHEADERS:
196 raise HTTPException("got more than %d headers" % _MAXHEADERS)
Barry Warsaw820c1202008-06-12 04:06:45 +0000197 if line in (b'\r\n', b'\n', b''):
198 break
199 hstring = b''.join(headers).decode('iso-8859-1')
Jeremy Hylton98eb6c22009-03-27 18:31:36 +0000200 return email.parser.Parser(_class=_class).parsestr(hstring)
Greg Stein5e0fa402000-06-26 08:28:01 +0000201
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000202
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000203class HTTPResponse(io.BufferedIOBase):
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000204
205 # See RFC 2616 sec 19.6 and RFC 1945 sec 6 for details.
206
Jeremy Hylton811fc142007-08-03 13:30:02 +0000207 # The bytes from the socket object are iso-8859-1 strings.
208 # See RFC 2616 sec 2.2 which notes an exception for MIME-encoded
209 # text following RFC 2047. The basic status line parsing only
210 # accepts iso-8859-1.
211
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700212 def __init__(self, sock, debuglevel=0, method=None, url=None):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000213 # If the response includes a content-length header, we need to
214 # make sure that the client doesn't read more than the
Jeremy Hylton39b198d2007-08-04 19:22:00 +0000215 # specified number of bytes. If it does, it will block until
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000216 # the server times out and closes the connection. This will
217 # happen if a self.fp.read() is done (without a size) whether
218 # self.fp is buffered or not. So, no self.fp.read() by
219 # clients unless they know what they are doing.
Benjamin Petersonf72d9fb2009-02-08 00:29:20 +0000220 self.fp = sock.makefile("rb")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000221 self.debuglevel = debuglevel
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000222 self._method = method
Greg Stein5e0fa402000-06-26 08:28:01 +0000223
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000224 # The HTTPResponse object is returned via urllib. The clients
225 # of http and urllib expect different attributes for the
226 # headers. headers is used here and supports urllib. msg is
227 # provided as a backwards compatibility layer for http
228 # clients.
229
230 self.headers = self.msg = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000231
Greg Steindd6eefb2000-07-18 09:09:48 +0000232 # from the Status-Line of the response
Tim Peters07e99cb2001-01-14 23:47:14 +0000233 self.version = _UNKNOWN # HTTP-Version
234 self.status = _UNKNOWN # Status-Code
235 self.reason = _UNKNOWN # Reason-Phrase
Greg Stein5e0fa402000-06-26 08:28:01 +0000236
Tim Peters07e99cb2001-01-14 23:47:14 +0000237 self.chunked = _UNKNOWN # is "chunked" being used?
238 self.chunk_left = _UNKNOWN # bytes left to read in current chunk
239 self.length = _UNKNOWN # number of bytes left in response
240 self.will_close = _UNKNOWN # conn will close at end of response
Greg Stein5e0fa402000-06-26 08:28:01 +0000241
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000242 def _read_status(self):
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000243 line = str(self.fp.readline(_MAXLINE + 1), "iso-8859-1")
244 if len(line) > _MAXLINE:
245 raise LineTooLong("status line")
Jeremy Hylton30f86742000-09-18 22:50:38 +0000246 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000247 print("reply:", repr(line))
Jeremy Hyltonb6769522003-06-29 17:55:05 +0000248 if not line:
249 # Presumably, the server closed the connection before
250 # sending a valid response.
R David Murraycae7bdb2015-04-05 19:26:29 -0400251 raise RemoteDisconnected("Remote end closed connection without"
252 " response")
Greg Steindd6eefb2000-07-18 09:09:48 +0000253 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000254 version, status, reason = line.split(None, 2)
Greg Steindd6eefb2000-07-18 09:09:48 +0000255 except ValueError:
256 try:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000257 version, status = line.split(None, 1)
Greg Steindd6eefb2000-07-18 09:09:48 +0000258 reason = ""
259 except ValueError:
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000260 # empty version will cause next test to fail.
Jeremy Hyltond46aa372002-07-06 18:48:07 +0000261 version = ""
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000262 if not version.startswith("HTTP/"):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200263 self._close_conn()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000264 raise BadStatusLine(line)
Greg Stein5e0fa402000-06-26 08:28:01 +0000265
Jeremy Hylton23d40472001-04-13 14:57:08 +0000266 # The status code is a three-digit number
267 try:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000268 status = int(status)
Jeremy Hylton23d40472001-04-13 14:57:08 +0000269 if status < 100 or status > 999:
270 raise BadStatusLine(line)
271 except ValueError:
272 raise BadStatusLine(line)
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000273 return version, status, reason
Greg Stein5e0fa402000-06-26 08:28:01 +0000274
Jeremy Hylton39c03802002-07-12 14:04:09 +0000275 def begin(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000276 if self.headers is not None:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000277 # we've already started reading the response
278 return
279
280 # read until we get a non-100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000281 while True:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000282 version, status, reason = self._read_status()
Martin v. Löwis39a31782004-09-18 09:03:49 +0000283 if status != CONTINUE:
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000284 break
285 # skip the header from the 100 response
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000286 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000287 skip = self.fp.readline(_MAXLINE + 1)
288 if len(skip) > _MAXLINE:
289 raise LineTooLong("header line")
290 skip = skip.strip()
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000291 if not skip:
292 break
293 if self.debuglevel > 0:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000294 print("header:", skip)
Tim Petersc411dba2002-07-16 21:35:23 +0000295
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000296 self.code = self.status = status
Jeremy Hyltonbe4fcf12002-06-28 22:38:01 +0000297 self.reason = reason.strip()
Antoine Pitrou988dbd72010-12-17 17:35:56 +0000298 if version in ("HTTP/1.0", "HTTP/0.9"):
299 # Some servers might still return "0.9", treat it as 1.0 anyway
Greg Steindd6eefb2000-07-18 09:09:48 +0000300 self.version = 10
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000301 elif version.startswith("HTTP/1."):
Tim Peters07e99cb2001-01-14 23:47:14 +0000302 self.version = 11 # use HTTP/1.1 code for HTTP/1.x where x>=1
Greg Steindd6eefb2000-07-18 09:09:48 +0000303 else:
304 raise UnknownProtocol(version)
Greg Stein5e0fa402000-06-26 08:28:01 +0000305
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000306 self.headers = self.msg = parse_headers(self.fp)
Barry Warsaw820c1202008-06-12 04:06:45 +0000307
Jeremy Hylton30f86742000-09-18 22:50:38 +0000308 if self.debuglevel > 0:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000309 for hdr in self.headers:
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000310 print("header:", hdr, end=" ")
Greg Stein5e0fa402000-06-26 08:28:01 +0000311
Greg Steindd6eefb2000-07-18 09:09:48 +0000312 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000313 tr_enc = self.headers.get("transfer-encoding")
Jeremy Hyltond229b3a2002-09-03 19:24:24 +0000314 if tr_enc and tr_enc.lower() == "chunked":
Jeremy Hylton236156f2008-12-15 03:00:50 +0000315 self.chunked = True
Greg Steindd6eefb2000-07-18 09:09:48 +0000316 self.chunk_left = None
317 else:
Jeremy Hylton236156f2008-12-15 03:00:50 +0000318 self.chunked = False
Greg Stein5e0fa402000-06-26 08:28:01 +0000319
Greg Steindd6eefb2000-07-18 09:09:48 +0000320 # will the connection close at the end of the response?
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000321 self.will_close = self._check_close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000322
Greg Steindd6eefb2000-07-18 09:09:48 +0000323 # do we have a Content-Length?
324 # NOTE: RFC 2616, S4.4, #3 says we ignore this if tr_enc is "chunked"
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000325 self.length = None
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000326 length = self.headers.get("content-length")
Barry Warsaw820c1202008-06-12 04:06:45 +0000327
328 # are we using the chunked-style of transfer encoding?
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000329 tr_enc = self.headers.get("transfer-encoding")
Greg Steindd6eefb2000-07-18 09:09:48 +0000330 if length and not self.chunked:
Jeremy Hylton30a81812000-09-14 20:34:27 +0000331 try:
332 self.length = int(length)
333 except ValueError:
Christian Heimesa612dc02008-02-24 13:08:18 +0000334 self.length = None
335 else:
336 if self.length < 0: # ignore nonsensical negative lengths
337 self.length = None
338 else:
339 self.length = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000340
Greg Steindd6eefb2000-07-18 09:09:48 +0000341 # does the body have a fixed length? (of zero)
Martin v. Löwis39a31782004-09-18 09:03:49 +0000342 if (status == NO_CONTENT or status == NOT_MODIFIED or
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000343 100 <= status < 200 or # 1xx codes
Jeremy Hyltone5d0e842007-08-03 13:45:24 +0000344 self._method == "HEAD"):
Greg Steindd6eefb2000-07-18 09:09:48 +0000345 self.length = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000346
Greg Steindd6eefb2000-07-18 09:09:48 +0000347 # if the connection remains open, and we aren't using chunked, and
348 # a content-length was not provided, then assume that the connection
349 # WILL close.
Jeremy Hylton0ee5eeb2007-08-04 03:25:17 +0000350 if (not self.will_close and
351 not self.chunked and
352 self.length is None):
Jeremy Hylton236156f2008-12-15 03:00:50 +0000353 self.will_close = True
Greg Stein5e0fa402000-06-26 08:28:01 +0000354
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000355 def _check_close(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000356 conn = self.headers.get("connection")
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000357 if self.version == 11:
358 # An HTTP/1.1 proxy is assumed to stay open unless
359 # explicitly closed.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000360 conn = self.headers.get("connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000361 if conn and "close" in conn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000362 return True
363 return False
364
Jeremy Hylton2c178252004-08-07 16:28:14 +0000365 # Some HTTP/1.0 implementations have support for persistent
366 # connections, using rules different than HTTP/1.1.
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000367
Christian Heimes895627f2007-12-08 17:28:33 +0000368 # For older HTTP, Keep-Alive indicates persistent connection.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000369 if self.headers.get("keep-alive"):
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000370 return False
Tim Peters77c06fb2002-11-24 02:35:35 +0000371
Jeremy Hylton2c178252004-08-07 16:28:14 +0000372 # At least Akamai returns a "Connection: Keep-Alive" header,
373 # which was supposed to be sent by the client.
374 if conn and "keep-alive" in conn.lower():
375 return False
376
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000377 # Proxy-Connection is a netscape hack.
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000378 pconn = self.headers.get("proxy-connection")
Raymond Hettingerbac788a2004-05-04 09:21:43 +0000379 if pconn and "keep-alive" in pconn.lower():
Jeremy Hylton22b3a492002-11-13 17:27:43 +0000380 return False
381
382 # otherwise, assume it will close
383 return True
384
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200385 def _close_conn(self):
386 fp = self.fp
387 self.fp = None
388 fp.close()
389
Greg Steindd6eefb2000-07-18 09:09:48 +0000390 def close(self):
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300391 try:
392 super().close() # set "closed" flag
393 finally:
394 if self.fp:
395 self._close_conn()
Greg Stein5e0fa402000-06-26 08:28:01 +0000396
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000397 # These implementations are for the benefit of io.BufferedReader.
398
399 # XXX This class should probably be revised to act more like
400 # the "raw stream" that BufferedReader expects.
401
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000402 def flush(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200403 super().flush()
404 if self.fp:
405 self.fp.flush()
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000406
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000407 def readable(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700408 """Always returns True"""
Jeremy Hyltona7cff022009-04-01 02:35:56 +0000409 return True
410
Jeremy Hyltondf5f6b52007-08-08 17:36:33 +0000411 # End of "raw stream" methods
412
Greg Steindd6eefb2000-07-18 09:09:48 +0000413 def isclosed(self):
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200414 """True if the connection is closed."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000415 # NOTE: it is possible that we will not ever call self.close(). This
416 # case occurs when will_close is TRUE, length is None, and we
417 # read up to the last byte, but NOT past it.
418 #
419 # IMPLIES: if will_close is FALSE, then self.close() will ALWAYS be
420 # called, meaning self.isclosed() is meaningful.
421 return self.fp is None
422
423 def read(self, amt=None):
424 if self.fp is None:
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000425 return b""
Greg Steindd6eefb2000-07-18 09:09:48 +0000426
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000427 if self._method == "HEAD":
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200428 self._close_conn()
Senthil Kumaran71fb6c82010-04-28 17:39:48 +0000429 return b""
430
Antoine Pitrou38d96432011-12-06 22:33:57 +0100431 if amt is not None:
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000432 # Amount is given, implement using readinto
433 b = bytearray(amt)
434 n = self.readinto(b)
435 return memoryview(b)[:n].tobytes()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100436 else:
437 # Amount is not given (unbounded read) so we must check self.length
438 # and self.chunked
Tim Peters230a60c2002-11-09 05:08:07 +0000439
Antoine Pitrou38d96432011-12-06 22:33:57 +0100440 if self.chunked:
441 return self._readall_chunked()
442
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000443 if self.length is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000444 s = self.fp.read()
445 else:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100446 try:
447 s = self._safe_read(self.length)
448 except IncompleteRead:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200449 self._close_conn()
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100450 raise
Jeremy Hyltondef9d2a2004-11-07 16:13:49 +0000451 self.length = 0
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200452 self._close_conn() # we read everything
Greg Steindd6eefb2000-07-18 09:09:48 +0000453 return s
454
Antoine Pitrou38d96432011-12-06 22:33:57 +0100455 def readinto(self, b):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700456 """Read up to len(b) bytes into bytearray b and return the number
457 of bytes read.
458 """
459
Antoine Pitrou38d96432011-12-06 22:33:57 +0100460 if self.fp is None:
461 return 0
462
463 if self._method == "HEAD":
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200464 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100465 return 0
466
467 if self.chunked:
468 return self._readinto_chunked(b)
469
Greg Steindd6eefb2000-07-18 09:09:48 +0000470 if self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100471 if len(b) > self.length:
Greg Steindd6eefb2000-07-18 09:09:48 +0000472 # clip the read to the "end of response"
Antoine Pitrou38d96432011-12-06 22:33:57 +0100473 b = memoryview(b)[0:self.length]
Greg Steindd6eefb2000-07-18 09:09:48 +0000474
475 # we do not use _safe_read() here because this may be a .will_close
476 # connection, and the user is reading more bytes than will be provided
477 # (for example, reading in 1k chunks)
Antoine Pitrou38d96432011-12-06 22:33:57 +0100478 n = self.fp.readinto(b)
Serhiy Storchaka1c84ac12013-12-17 21:50:02 +0200479 if not n and b:
Antoine Pitroubeec61a2013-02-02 22:49:34 +0100480 # Ideally, we would raise IncompleteRead if the content-length
481 # wasn't satisfied, but it might break compatibility.
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200482 self._close_conn()
Antoine Pitrou6a35e182013-02-02 23:04:56 +0100483 elif self.length is not None:
Antoine Pitrou38d96432011-12-06 22:33:57 +0100484 self.length -= n
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000485 if not self.length:
Serhiy Storchakab5b9c8c2013-02-06 10:31:57 +0200486 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100487 return n
Greg Steindd6eefb2000-07-18 09:09:48 +0000488
Antoine Pitrou38d96432011-12-06 22:33:57 +0100489 def _read_next_chunk_size(self):
490 # Read the next chunk size from the file
491 line = self.fp.readline(_MAXLINE + 1)
492 if len(line) > _MAXLINE:
493 raise LineTooLong("chunk size")
494 i = line.find(b";")
495 if i >= 0:
496 line = line[:i] # strip chunk-extensions
497 try:
498 return int(line, 16)
499 except ValueError:
500 # close the connection as protocol synchronisation is
501 # probably lost
Serhiy Storchakab6c86fd2013-02-06 10:35:40 +0200502 self._close_conn()
Antoine Pitrou38d96432011-12-06 22:33:57 +0100503 raise
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000504
Antoine Pitrou38d96432011-12-06 22:33:57 +0100505 def _read_and_discard_trailer(self):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000506 # read and discard trailer up to the CRLF terminator
507 ### note: we shouldn't have any trailers!
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000508 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000509 line = self.fp.readline(_MAXLINE + 1)
510 if len(line) > _MAXLINE:
511 raise LineTooLong("trailer line")
Christian Heimes0bd4e112008-02-12 22:59:25 +0000512 if not line:
513 # a vanishingly small number of sites EOF without
514 # sending the trailer
515 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800516 if line in (b'\r\n', b'\n', b''):
Jeremy Hyltond4c472c2002-09-03 20:49:06 +0000517 break
518
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000519 def _get_chunk_left(self):
520 # return self.chunk_left, reading a new chunk if necessary.
521 # chunk_left == 0: at the end of the current chunk, need to close it
522 # chunk_left == None: No current chunk, should read next.
523 # This function returns non-zero or None if the last chunk has
524 # been read.
525 chunk_left = self.chunk_left
526 if not chunk_left: # Can be 0 or None
527 if chunk_left is not None:
528 # We are at the end of chunk. dicard chunk end
529 self._safe_read(2) # toss the CRLF at the end of the chunk
530 try:
531 chunk_left = self._read_next_chunk_size()
532 except ValueError:
533 raise IncompleteRead(b'')
534 if chunk_left == 0:
535 # last chunk: 1*("0") [ chunk-extension ] CRLF
536 self._read_and_discard_trailer()
537 # we read everything; close the "file"
538 self._close_conn()
539 chunk_left = None
540 self.chunk_left = chunk_left
541 return chunk_left
542
Antoine Pitrou38d96432011-12-06 22:33:57 +0100543 def _readall_chunked(self):
544 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100545 value = []
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000546 try:
547 while True:
548 chunk_left = self._get_chunk_left()
549 if chunk_left is None:
550 break
551 value.append(self._safe_read(chunk_left))
552 self.chunk_left = 0
553 return b''.join(value)
554 except IncompleteRead:
555 raise IncompleteRead(b''.join(value))
Tim Peters230a60c2002-11-09 05:08:07 +0000556
Antoine Pitrou38d96432011-12-06 22:33:57 +0100557 def _readinto_chunked(self, b):
558 assert self.chunked != _UNKNOWN
Antoine Pitrou38d96432011-12-06 22:33:57 +0100559 total_bytes = 0
560 mvb = memoryview(b)
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000561 try:
562 while True:
563 chunk_left = self._get_chunk_left()
564 if chunk_left is None:
565 return total_bytes
Antoine Pitrou4ce6aa42011-12-06 22:34:36 +0100566
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000567 if len(mvb) <= chunk_left:
568 n = self._safe_readinto(mvb)
569 self.chunk_left = chunk_left - n
570 return total_bytes + n
571
572 temp_mvb = mvb[:chunk_left]
Antoine Pitrou38d96432011-12-06 22:33:57 +0100573 n = self._safe_readinto(temp_mvb)
574 mvb = mvb[n:]
575 total_bytes += n
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000576 self.chunk_left = 0
Antoine Pitrou38d96432011-12-06 22:33:57 +0100577
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000578 except IncompleteRead:
579 raise IncompleteRead(bytes(b[0:total_bytes]))
Antoine Pitrou38d96432011-12-06 22:33:57 +0100580
Greg Steindd6eefb2000-07-18 09:09:48 +0000581 def _safe_read(self, amt):
582 """Read the number of bytes requested, compensating for partial reads.
583
584 Normally, we have a blocking socket, but a read() can be interrupted
585 by a signal (resulting in a partial read).
586
587 Note that we cannot distinguish between EOF and an interrupt when zero
588 bytes have been read. IncompleteRead() will be raised in this
589 situation.
590
591 This function should be used when <amt> bytes "should" be present for
592 reading. If the bytes are truly not available (due to EOF), then the
593 IncompleteRead exception can be used to detect the problem.
594 """
Georg Brandl80ba8e82005-09-29 20:16:07 +0000595 s = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000596 while amt > 0:
Georg Brandl80ba8e82005-09-29 20:16:07 +0000597 chunk = self.fp.read(min(amt, MAXAMOUNT))
Greg Steindd6eefb2000-07-18 09:09:48 +0000598 if not chunk:
Benjamin Peterson6accb982009-03-02 22:50:25 +0000599 raise IncompleteRead(b''.join(s), amt)
Georg Brandl80ba8e82005-09-29 20:16:07 +0000600 s.append(chunk)
Raymond Hettingerb2e0b922003-02-26 22:45:18 +0000601 amt -= len(chunk)
Guido van Rossuma00f1232007-09-12 19:43:09 +0000602 return b"".join(s)
Greg Steindd6eefb2000-07-18 09:09:48 +0000603
Antoine Pitrou38d96432011-12-06 22:33:57 +0100604 def _safe_readinto(self, b):
605 """Same as _safe_read, but for reading into a buffer."""
606 total_bytes = 0
607 mvb = memoryview(b)
608 while total_bytes < len(b):
609 if MAXAMOUNT < len(mvb):
610 temp_mvb = mvb[0:MAXAMOUNT]
611 n = self.fp.readinto(temp_mvb)
612 else:
613 n = self.fp.readinto(mvb)
614 if not n:
615 raise IncompleteRead(bytes(mvb[0:total_bytes]), len(b))
616 mvb = mvb[n:]
617 total_bytes += n
618 return total_bytes
619
Kristján Valur Jónsson8e5d0ca2014-03-19 10:07:26 +0000620 def read1(self, n=-1):
621 """Read with at most one underlying system call. If at least one
622 byte is buffered, return that instead.
623 """
624 if self.fp is None or self._method == "HEAD":
625 return b""
626 if self.chunked:
627 return self._read1_chunked(n)
628 try:
629 result = self.fp.read1(n)
630 except ValueError:
631 if n >= 0:
632 raise
633 # some implementations, like BufferedReader, don't support -1
634 # Read an arbitrarily selected largeish chunk.
635 result = self.fp.read1(16*1024)
636 if not result and n:
637 self._close_conn()
638 return result
639
640 def peek(self, n=-1):
641 # Having this enables IOBase.readline() to read more than one
642 # byte at a time
643 if self.fp is None or self._method == "HEAD":
644 return b""
645 if self.chunked:
646 return self._peek_chunked(n)
647 return self.fp.peek(n)
648
649 def readline(self, limit=-1):
650 if self.fp is None or self._method == "HEAD":
651 return b""
652 if self.chunked:
653 # Fallback to IOBase readline which uses peek() and read()
654 return super().readline(limit)
655 result = self.fp.readline(limit)
656 if not result and limit:
657 self._close_conn()
658 return result
659
660 def _read1_chunked(self, n):
661 # Strictly speaking, _get_chunk_left() may cause more than one read,
662 # but that is ok, since that is to satisfy the chunked protocol.
663 chunk_left = self._get_chunk_left()
664 if chunk_left is None or n == 0:
665 return b''
666 if not (0 <= n <= chunk_left):
667 n = chunk_left # if n is negative or larger than chunk_left
668 read = self.fp.read1(n)
669 self.chunk_left -= len(read)
670 if not read:
671 raise IncompleteRead(b"")
672 return read
673
674 def _peek_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 try:
678 chunk_left = self._get_chunk_left()
679 except IncompleteRead:
680 return b'' # peek doesn't worry about protocol
681 if chunk_left is None:
682 return b'' # eof
683 # peek is allowed to return more than requested. Just request the
684 # entire chunk, and truncate what we get.
685 return self.fp.peek(chunk_left)[:chunk_left]
686
Antoine Pitroub353c122009-02-11 00:39:14 +0000687 def fileno(self):
688 return self.fp.fileno()
689
Greg Steindd6eefb2000-07-18 09:09:48 +0000690 def getheader(self, name, default=None):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700691 '''Returns the value of the header matching *name*.
692
693 If there are multiple matching headers, the values are
694 combined into a single string separated by commas and spaces.
695
696 If no matching header is found, returns *default* or None if
697 the *default* is not specified.
698
699 If the headers are unknown, raises http.client.ResponseNotReady.
700
701 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000702 if self.headers is None:
Greg Steindd6eefb2000-07-18 09:09:48 +0000703 raise ResponseNotReady()
Senthil Kumaran9f8dc442010-08-02 11:04:58 +0000704 headers = self.headers.get_all(name) or default
705 if isinstance(headers, str) or not hasattr(headers, '__iter__'):
706 return headers
707 else:
708 return ', '.join(headers)
Greg Stein5e0fa402000-06-26 08:28:01 +0000709
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000710 def getheaders(self):
711 """Return list of (header, value) tuples."""
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000712 if self.headers is None:
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000713 raise ResponseNotReady()
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000714 return list(self.headers.items())
Martin v. Löwisdeacce22004-08-18 12:46:26 +0000715
Antoine Pitroub353c122009-02-11 00:39:14 +0000716 # We override IOBase.__iter__ so that it doesn't check for closed-ness
717
718 def __iter__(self):
719 return self
720
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000721 # For compatibility with old-style urllib responses.
722
723 def info(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700724 '''Returns an instance of the class mimetools.Message containing
725 meta-information associated with the URL.
726
727 When the method is HTTP, these headers are those returned by
728 the server at the head of the retrieved HTML page (including
729 Content-Length and Content-Type).
730
731 When the method is FTP, a Content-Length header will be
732 present if (as is now usual) the server passed back a file
733 length in response to the FTP retrieval request. A
734 Content-Type header will be present if the MIME type can be
735 guessed.
736
737 When the method is local-file, returned headers will include
738 a Date representing the file’s last-modified time, a
739 Content-Length giving file size, and a Content-Type
740 containing a guess at the file’s type. See also the
741 description of the mimetools module.
742
743 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000744 return self.headers
745
746 def geturl(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700747 '''Return the real URL of the page.
748
749 In some cases, the HTTP server redirects a client to another
750 URL. The urlopen() function handles this transparently, but in
751 some cases the caller needs to know which URL the client was
752 redirected to. The geturl() method can be used to get at this
753 redirected URL.
754
755 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000756 return self.url
757
758 def getcode(self):
Raymond Hettinger15b87bf2015-08-18 22:03:08 -0700759 '''Return the HTTP status code that was sent with the response,
760 or None if the URL is not an HTTP URL.
761
762 '''
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +0000763 return self.status
Greg Stein5e0fa402000-06-26 08:28:01 +0000764
765class HTTPConnection:
766
Greg Steindd6eefb2000-07-18 09:09:48 +0000767 _http_vsn = 11
768 _http_vsn_str = 'HTTP/1.1'
Greg Stein5e0fa402000-06-26 08:28:01 +0000769
Greg Steindd6eefb2000-07-18 09:09:48 +0000770 response_class = HTTPResponse
771 default_port = HTTP_PORT
772 auto_open = 1
Jeremy Hylton30f86742000-09-18 22:50:38 +0000773 debuglevel = 0
Greg Stein5e0fa402000-06-26 08:28:01 +0000774
Senthil Kumaran052ddb02013-03-18 14:11:41 -0700775 def __init__(self, host, port=None, timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
776 source_address=None):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000777 self.timeout = timeout
Gregory P. Smithb4066372010-01-03 03:28:29 +0000778 self.source_address = source_address
Greg Steindd6eefb2000-07-18 09:09:48 +0000779 self.sock = None
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000780 self._buffer = []
Greg Steindd6eefb2000-07-18 09:09:48 +0000781 self.__response = None
782 self.__state = _CS_IDLE
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +0000783 self._method = None
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000784 self._tunnel_host = None
785 self._tunnel_port = None
Georg Brandlc7c199f2011-01-22 22:06:24 +0000786 self._tunnel_headers = {}
Tim Petersc411dba2002-07-16 21:35:23 +0000787
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400788 (self.host, self.port) = self._get_hostport(host, port)
789
790 # This is stored as an instance variable to allow unit
791 # tests to replace it with a suitable mockup
792 self._create_connection = socket.create_connection
Greg Stein5e0fa402000-06-26 08:28:01 +0000793
Senthil Kumaran47fff872009-12-20 07:10:31 +0000794 def set_tunnel(self, host, port=None, headers=None):
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400795 """Set up host and port for HTTP CONNECT tunnelling.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000796
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400797 In a connection that uses HTTP CONNECT tunneling, the host passed to the
798 constructor is used as a proxy server that relays all communication to
799 the endpoint passed to `set_tunnel`. This done by sending an HTTP
800 CONNECT request to the proxy server when the connection is established.
801
802 This method must be called before the HTML connection has been
803 established.
804
805 The headers argument should be a mapping of extra HTTP headers to send
806 with the CONNECT request.
Senthil Kumaran47fff872009-12-20 07:10:31 +0000807 """
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400808
809 if self.sock:
810 raise RuntimeError("Can't set up tunnel for established connection")
811
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200812 self._tunnel_host, self._tunnel_port = self._get_hostport(host, port)
Senthil Kumaran47fff872009-12-20 07:10:31 +0000813 if headers:
814 self._tunnel_headers = headers
815 else:
816 self._tunnel_headers.clear()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000817
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400818 def _get_hostport(self, host, port):
Greg Steindd6eefb2000-07-18 09:09:48 +0000819 if port is None:
Skip Montanaro10e6e0e2004-09-14 16:32:02 +0000820 i = host.rfind(':')
Skip Montanarocae14d22004-09-14 17:55:21 +0000821 j = host.rfind(']') # ipv6 addresses have [...]
822 if i > j:
Skip Montanaro9d389972002-03-24 16:53:50 +0000823 try:
824 port = int(host[i+1:])
825 except ValueError:
Łukasz Langaa5a9a9c2011-10-18 21:17:39 +0200826 if host[i+1:] == "": # http://foo.com:/ == http://foo.com/
827 port = self.default_port
828 else:
829 raise InvalidURL("nonnumeric port: '%s'" % host[i+1:])
Greg Steindd6eefb2000-07-18 09:09:48 +0000830 host = host[:i]
831 else:
832 port = self.default_port
Raymond Hettinger4d037912004-10-14 15:23:38 +0000833 if host and host[0] == '[' and host[-1] == ']':
Brett Cannon0a1af4a2004-09-15 23:26:23 +0000834 host = host[1:-1]
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400835
836 return (host, port)
Greg Stein5e0fa402000-06-26 08:28:01 +0000837
Jeremy Hylton30f86742000-09-18 22:50:38 +0000838 def set_debuglevel(self, level):
839 self.debuglevel = level
840
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000841 def _tunnel(self):
Serhiy Storchaka4ac7ed92014-12-12 09:29:15 +0200842 connect_str = "CONNECT %s:%d HTTP/1.0\r\n" % (self._tunnel_host,
843 self._tunnel_port)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000844 connect_bytes = connect_str.encode("ascii")
845 self.send(connect_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000846 for header, value in self._tunnel_headers.items():
Senthil Kumaran47fff872009-12-20 07:10:31 +0000847 header_str = "%s: %s\r\n" % (header, value)
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000848 header_bytes = header_str.encode("latin-1")
Senthil Kumaran47fff872009-12-20 07:10:31 +0000849 self.send(header_bytes)
Georg Brandlc7c199f2011-01-22 22:06:24 +0000850 self.send(b'\r\n')
Senthil Kumaran47fff872009-12-20 07:10:31 +0000851
Georg Brandlc7c199f2011-01-22 22:06:24 +0000852 response = self.response_class(self.sock, method=self._method)
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000853 (version, code, message) = response._read_status()
Senthil Kumaran47fff872009-12-20 07:10:31 +0000854
Serhiy Storchakae4db7692014-12-23 16:28:28 +0200855 if code != http.HTTPStatus.OK:
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000856 self.close()
Andrew Svetlov0832af62012-12-18 23:10:48 +0200857 raise OSError("Tunnel connection failed: %d %s" % (code,
858 message.strip()))
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000859 while True:
Senthil Kumaran5466bf12010-12-18 16:55:23 +0000860 line = response.fp.readline(_MAXLINE + 1)
861 if len(line) > _MAXLINE:
862 raise LineTooLong("header line")
Senthil Kumaranb12771a2012-04-23 23:50:07 +0800863 if not line:
864 # for sites which EOF without sending a trailer
865 break
Senthil Kumaran7e70a5c2012-04-29 10:39:49 +0800866 if line in (b'\r\n', b'\n', b''):
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000867 break
868
Berker Peksagab53ab02015-02-03 12:22:11 +0200869 if self.debuglevel > 0:
870 print('header:', line.decode())
871
Greg Steindd6eefb2000-07-18 09:09:48 +0000872 def connect(self):
873 """Connect to the host and port specified in __init__."""
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500874 self.sock = self._create_connection(
875 (self.host,self.port), self.timeout, self.source_address)
876 self.sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
Senthil Kumaran9da047b2014-04-14 13:07:56 -0400877
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +0000878 if self._tunnel_host:
879 self._tunnel()
Greg Stein5e0fa402000-06-26 08:28:01 +0000880
Greg Steindd6eefb2000-07-18 09:09:48 +0000881 def close(self):
882 """Close the connection to the HTTP server."""
Greg Steindd6eefb2000-07-18 09:09:48 +0000883 self.__state = _CS_IDLE
Serhiy Storchaka7e7a3db2015-04-10 13:24:41 +0300884 try:
885 sock = self.sock
886 if sock:
887 self.sock = None
888 sock.close() # close it manually... there may be other refs
889 finally:
890 response = self.__response
891 if response:
892 self.__response = None
893 response.close()
Greg Stein5e0fa402000-06-26 08:28:01 +0000894
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000895 def send(self, data):
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000896 """Send `data' to the server.
897 ``data`` can be a string object, a bytes object, an array object, a
898 file-like object that supports a .read() method, or an iterable object.
899 """
900
Greg Steindd6eefb2000-07-18 09:09:48 +0000901 if self.sock is None:
902 if self.auto_open:
903 self.connect()
904 else:
905 raise NotConnected()
Greg Stein5e0fa402000-06-26 08:28:01 +0000906
Jeremy Hylton30f86742000-09-18 22:50:38 +0000907 if self.debuglevel > 0:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000908 print("send:", repr(data))
Jeremy Hylton636950f2009-03-28 04:34:21 +0000909 blocksize = 8192
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000910 if hasattr(data, "read") :
Jeremy Hylton636950f2009-03-28 04:34:21 +0000911 if self.debuglevel > 0:
912 print("sendIng a read()able")
913 encode = False
Antoine Pitrouead1d622009-09-29 18:44:53 +0000914 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000915 mode = data.mode
Antoine Pitrouead1d622009-09-29 18:44:53 +0000916 except AttributeError:
917 # io.BytesIO and other file-like objects don't have a `mode`
918 # attribute.
919 pass
920 else:
921 if "b" not in mode:
922 encode = True
923 if self.debuglevel > 0:
924 print("encoding file using iso-8859-1")
Jeremy Hylton636950f2009-03-28 04:34:21 +0000925 while 1:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000926 datablock = data.read(blocksize)
927 if not datablock:
Jeremy Hylton636950f2009-03-28 04:34:21 +0000928 break
929 if encode:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000930 datablock = datablock.encode("iso-8859-1")
931 self.sock.sendall(datablock)
Andrew Svetlov7b2c8bb2013-04-12 22:49:19 +0300932 return
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000933 try:
Senthil Kumaran5a2da3b2010-10-02 10:33:13 +0000934 self.sock.sendall(data)
Senthil Kumaran7bc0d872010-12-19 10:49:52 +0000935 except TypeError:
936 if isinstance(data, collections.Iterable):
937 for d in data:
938 self.sock.sendall(d)
939 else:
Georg Brandl25e2cd12011-08-03 08:27:00 +0200940 raise TypeError("data should be a bytes-like object "
941 "or an iterable, got %r" % type(data))
Greg Stein5e0fa402000-06-26 08:28:01 +0000942
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000943 def _output(self, s):
944 """Add a line of output to the current request buffer.
Tim Peters469cdad2002-08-08 20:19:19 +0000945
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000946 Assumes that the line does *not* end with \\r\\n.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000947 """
948 self._buffer.append(s)
949
Benjamin Peterson1742e402008-11-30 22:15:29 +0000950 def _send_output(self, message_body=None):
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000951 """Send the currently buffered request and clear the buffer.
952
Jeremy Hyltone3252ec2002-07-16 21:41:43 +0000953 Appends an extra \\r\\n to the buffer.
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000954 A message_body may be specified, to be appended to the request.
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000955 """
Martin v. Löwisdd5a8602007-06-30 09:22:09 +0000956 self._buffer.extend((b"", b""))
957 msg = b"\r\n".join(self._buffer)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000958 del self._buffer[:]
Benjamin Peterson9d8a3ad2015-01-23 11:02:57 -0500959
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000960 self.send(msg)
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000961 if message_body is not None:
Benjamin Peterson822b21c2009-01-18 00:04:57 +0000962 self.send(message_body)
Jeremy Hylton8531b1b2002-07-16 21:21:11 +0000963
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000964 def putrequest(self, method, url, skip_host=0, skip_accept_encoding=0):
Greg Steindd6eefb2000-07-18 09:09:48 +0000965 """Send a request to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +0000966
Greg Steindd6eefb2000-07-18 09:09:48 +0000967 `method' specifies an HTTP request method, e.g. 'GET'.
968 `url' specifies the object being requested, e.g. '/index.html'.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +0000969 `skip_host' if True does not add automatically a 'Host:' header
970 `skip_accept_encoding' if True does not add automatically an
971 'Accept-Encoding:' header
Greg Steindd6eefb2000-07-18 09:09:48 +0000972 """
Greg Stein5e0fa402000-06-26 08:28:01 +0000973
Greg Stein616a58d2003-06-24 06:35:19 +0000974 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +0000975 if self.__response and self.__response.isclosed():
976 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +0000977
Tim Peters58eb11c2004-01-18 20:29:55 +0000978
Greg Steindd6eefb2000-07-18 09:09:48 +0000979 # in certain cases, we cannot issue another request on this connection.
980 # this occurs when:
981 # 1) we are in the process of sending a request. (_CS_REQ_STARTED)
982 # 2) a response to a previous request has signalled that it is going
983 # to close the connection upon completion.
984 # 3) the headers for the previous response have not been read, thus
985 # we cannot determine whether point (2) is true. (_CS_REQ_SENT)
986 #
987 # if there is no prior response, then we can request at will.
988 #
989 # if point (2) is true, then we will have passed the socket to the
990 # response (effectively meaning, "there is no prior response"), and
991 # will open a new one when a new request is made.
992 #
993 # Note: if a prior response exists, then we *can* start a new request.
994 # We are not allowed to begin fetching the response to this new
995 # request, however, until that prior response is complete.
996 #
997 if self.__state == _CS_IDLE:
998 self.__state = _CS_REQ_STARTED
999 else:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001000 raise CannotSendRequest(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001001
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001002 # Save the method we use, we need it later in the response phase
1003 self._method = method
Greg Steindd6eefb2000-07-18 09:09:48 +00001004 if not url:
1005 url = '/'
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001006 request = '%s %s %s' % (method, url, self._http_vsn_str)
Greg Stein5e0fa402000-06-26 08:28:01 +00001007
Martin v. Löwisdd5a8602007-06-30 09:22:09 +00001008 # Non-ASCII characters should have been eliminated earlier
1009 self._output(request.encode('ascii'))
Greg Stein5e0fa402000-06-26 08:28:01 +00001010
Greg Steindd6eefb2000-07-18 09:09:48 +00001011 if self._http_vsn == 11:
1012 # Issue some standard headers for better HTTP/1.1 compliance
Greg Stein5e0fa402000-06-26 08:28:01 +00001013
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001014 if not skip_host:
1015 # this header is issued *only* for HTTP/1.1
1016 # connections. more specifically, this means it is
1017 # only issued when the client uses the new
1018 # HTTPConnection() class. backwards-compat clients
1019 # will be using HTTP/1.0 and those clients may be
1020 # issuing this header themselves. we should NOT issue
1021 # it twice; some web servers (such as Apache) barf
1022 # when they see two Host: headers
Guido van Rossumf6922aa2001-01-14 21:03:01 +00001023
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001024 # If we need a non-standard port,include it in the
1025 # header. If the request is going through a proxy,
1026 # but the host of the actual URL, not the host of the
1027 # proxy.
Jeremy Hylton8acf1e02002-03-08 19:35:51 +00001028
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001029 netloc = ''
1030 if url.startswith('http'):
1031 nil, netloc, nil, nil, nil = urlsplit(url)
1032
1033 if netloc:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001034 try:
1035 netloc_enc = netloc.encode("ascii")
1036 except UnicodeEncodeError:
1037 netloc_enc = netloc.encode("idna")
1038 self.putheader('Host', netloc_enc)
Jeremy Hylton3921ff62002-03-09 06:07:23 +00001039 else:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001040 if self._tunnel_host:
1041 host = self._tunnel_host
1042 port = self._tunnel_port
1043 else:
1044 host = self.host
1045 port = self.port
1046
Thomas Wouters477c8d52006-05-27 19:21:47 +00001047 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001048 host_enc = host.encode("ascii")
Thomas Wouters477c8d52006-05-27 19:21:47 +00001049 except UnicodeEncodeError:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001050 host_enc = host.encode("idna")
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001051
1052 # As per RFC 273, IPv6 address should be wrapped with []
1053 # when used as Host header
1054
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001055 if host.find(':') >= 0:
Senthil Kumaran74ebd9e2010-11-13 12:27:49 +00001056 host_enc = b'[' + host_enc + b']'
1057
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001058 if port == self.default_port:
Thomas Wouters477c8d52006-05-27 19:21:47 +00001059 self.putheader('Host', host_enc)
1060 else:
Guido van Rossum98297ee2007-11-06 21:34:58 +00001061 host_enc = host_enc.decode("ascii")
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001062 self.putheader('Host', "%s:%s" % (host_enc, port))
Greg Stein5e0fa402000-06-26 08:28:01 +00001063
Greg Steindd6eefb2000-07-18 09:09:48 +00001064 # note: we are assuming that clients will not attempt to set these
1065 # headers since *this* library must deal with the
1066 # consequences. this also means that when the supporting
1067 # libraries are updated to recognize other forms, then this
1068 # code should be changed (removed or updated).
Greg Stein5e0fa402000-06-26 08:28:01 +00001069
Greg Steindd6eefb2000-07-18 09:09:48 +00001070 # we only want a Content-Encoding of "identity" since we don't
1071 # support encodings such as x-gzip or x-deflate.
Martin v. Löwisaf7dc8d2003-11-19 19:51:55 +00001072 if not skip_accept_encoding:
1073 self.putheader('Accept-Encoding', 'identity')
Greg Stein5e0fa402000-06-26 08:28:01 +00001074
Greg Steindd6eefb2000-07-18 09:09:48 +00001075 # we can accept "chunked" Transfer-Encodings, but no others
1076 # NOTE: no TE header implies *only* "chunked"
1077 #self.putheader('TE', 'chunked')
Greg Stein5e0fa402000-06-26 08:28:01 +00001078
Greg Steindd6eefb2000-07-18 09:09:48 +00001079 # if TE is supplied in the header, then it must appear in a
1080 # Connection header.
1081 #self.putheader('Connection', 'TE')
Greg Stein5e0fa402000-06-26 08:28:01 +00001082
Greg Steindd6eefb2000-07-18 09:09:48 +00001083 else:
1084 # For HTTP/1.0, the server will assume "not chunked"
1085 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001086
Benjamin Petersonf608c612008-11-16 18:33:53 +00001087 def putheader(self, header, *values):
Greg Steindd6eefb2000-07-18 09:09:48 +00001088 """Send a request header line to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001089
Greg Steindd6eefb2000-07-18 09:09:48 +00001090 For example: h.putheader('Accept', 'text/html')
1091 """
1092 if self.__state != _CS_REQ_STARTED:
1093 raise CannotSendHeader()
Greg Stein5e0fa402000-06-26 08:28:01 +00001094
Guido van Rossum98297ee2007-11-06 21:34:58 +00001095 if hasattr(header, 'encode'):
1096 header = header.encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001097
1098 if not _is_legal_header_name(header):
1099 raise ValueError('Invalid header name %r' % (header,))
1100
Benjamin Petersonf608c612008-11-16 18:33:53 +00001101 values = list(values)
1102 for i, one_value in enumerate(values):
1103 if hasattr(one_value, 'encode'):
Marc-André Lemburg8f36af72011-02-25 15:42:01 +00001104 values[i] = one_value.encode('latin-1')
Senthil Kumaran58d5dbf2010-10-03 18:22:42 +00001105 elif isinstance(one_value, int):
1106 values[i] = str(one_value).encode('ascii')
Serhiy Storchakaa112a8a2015-03-12 11:13:36 +02001107
1108 if _is_illegal_header_value(values[i]):
1109 raise ValueError('Invalid header value %r' % (values[i],))
1110
Benjamin Petersonf608c612008-11-16 18:33:53 +00001111 value = b'\r\n\t'.join(values)
Guido van Rossum98297ee2007-11-06 21:34:58 +00001112 header = header + b': ' + value
1113 self._output(header)
Greg Stein5e0fa402000-06-26 08:28:01 +00001114
Benjamin Peterson1742e402008-11-30 22:15:29 +00001115 def endheaders(self, message_body=None):
1116 """Indicate that the last header line has been sent to the server.
Greg Stein5e0fa402000-06-26 08:28:01 +00001117
Senthil Kumaranad87fa62011-10-05 23:26:49 +08001118 This method sends the request to the server. The optional message_body
1119 argument can be used to pass a message body associated with the
1120 request. The message body will be sent in the same packet as the
1121 message headers if it is a string, otherwise it is sent as a separate
1122 packet.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001123 """
Greg Steindd6eefb2000-07-18 09:09:48 +00001124 if self.__state == _CS_REQ_STARTED:
1125 self.__state = _CS_REQ_SENT
1126 else:
1127 raise CannotSendHeader()
Benjamin Peterson1742e402008-11-30 22:15:29 +00001128 self._send_output(message_body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001129
Greg Steindd6eefb2000-07-18 09:09:48 +00001130 def request(self, method, url, body=None, headers={}):
1131 """Send a complete request to the server."""
Jeremy Hylton636950f2009-03-28 04:34:21 +00001132 self._send_request(method, url, body, headers)
Greg Stein5e0fa402000-06-26 08:28:01 +00001133
R David Murraybeed8402015-03-22 15:18:23 -04001134 def _set_content_length(self, body, method):
1135 # Set the content-length based on the body. If the body is "empty", we
1136 # set Content-Length: 0 for methods that expect a body (RFC 7230,
1137 # Section 3.3.2). If the body is set for other methods, we set the
1138 # header provided we can figure out what the length is.
Benjamin Peterson1742e402008-11-30 22:15:29 +00001139 thelen = None
R David Murraybeed8402015-03-22 15:18:23 -04001140 method_expects_body = method.upper() in _METHODS_EXPECTING_BODY
1141 if body is None and method_expects_body:
1142 thelen = '0'
1143 elif body is not None:
Benjamin Peterson1742e402008-11-30 22:15:29 +00001144 try:
R David Murraybeed8402015-03-22 15:18:23 -04001145 thelen = str(len(body))
1146 except TypeError:
1147 # If this is a file-like object, try to
1148 # fstat its file descriptor
1149 try:
1150 thelen = str(os.fstat(body.fileno()).st_size)
1151 except (AttributeError, OSError):
1152 # Don't send a length if this failed
1153 if self.debuglevel > 0: print("Cannot stat!!")
Benjamin Peterson1742e402008-11-30 22:15:29 +00001154
1155 if thelen is not None:
1156 self.putheader('Content-Length', thelen)
1157
Greg Steindd6eefb2000-07-18 09:09:48 +00001158 def _send_request(self, method, url, body, headers):
Jeremy Hylton636950f2009-03-28 04:34:21 +00001159 # Honor explicitly requested Host: and Accept-Encoding: headers.
Jeremy Hylton2c178252004-08-07 16:28:14 +00001160 header_names = dict.fromkeys([k.lower() for k in headers])
1161 skips = {}
1162 if 'host' in header_names:
1163 skips['skip_host'] = 1
1164 if 'accept-encoding' in header_names:
1165 skips['skip_accept_encoding'] = 1
Greg Stein5e0fa402000-06-26 08:28:01 +00001166
Jeremy Hylton2c178252004-08-07 16:28:14 +00001167 self.putrequest(method, url, **skips)
1168
R David Murraybeed8402015-03-22 15:18:23 -04001169 if 'content-length' not in header_names:
1170 self._set_content_length(body, method)
Guido van Rossumcc2b0162007-02-11 06:12:03 +00001171 for hdr, value in headers.items():
Greg Steindd6eefb2000-07-18 09:09:48 +00001172 self.putheader(hdr, value)
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001173 if isinstance(body, str):
Jeremy Hylton236654b2009-03-27 20:24:34 +00001174 # RFC 2616 Section 3.7.1 says that text default has a
1175 # default charset of iso-8859-1.
1176 body = body.encode('iso-8859-1')
Jeremy Hyltonef9f48e2009-03-26 22:04:05 +00001177 self.endheaders(body)
Greg Stein5e0fa402000-06-26 08:28:01 +00001178
Greg Steindd6eefb2000-07-18 09:09:48 +00001179 def getresponse(self):
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001180 """Get the response from the server.
1181
1182 If the HTTPConnection is in the correct state, returns an
1183 instance of HTTPResponse or of whatever object is returned by
1184 class the response_class variable.
1185
1186 If a request has not been sent or if a previous response has
1187 not be handled, ResponseNotReady is raised. If the HTTP
1188 response indicates that the connection should be closed, then
1189 it will be closed before the response is returned. When the
1190 connection is closed, the underlying socket is closed.
1191 """
Greg Stein5e0fa402000-06-26 08:28:01 +00001192
Greg Stein616a58d2003-06-24 06:35:19 +00001193 # if a prior response has been completed, then forget about it.
Greg Steindd6eefb2000-07-18 09:09:48 +00001194 if self.__response and self.__response.isclosed():
1195 self.__response = None
Greg Stein5e0fa402000-06-26 08:28:01 +00001196
Greg Steindd6eefb2000-07-18 09:09:48 +00001197 # if a prior response exists, then it must be completed (otherwise, we
1198 # cannot read this response's header to determine the connection-close
1199 # behavior)
1200 #
1201 # note: if a prior response existed, but was connection-close, then the
1202 # socket and response were made independent of this HTTPConnection
1203 # object since a new request requires that we open a whole new
1204 # connection
1205 #
1206 # this means the prior response had one of two states:
1207 # 1) will_close: this connection was reset and the prior socket and
1208 # response operate independently
1209 # 2) persistent: the response was retained and we await its
1210 # isclosed() status to become true.
1211 #
1212 if self.__state != _CS_REQ_SENT or self.__response:
Jeremy Hylton6c5e28c2009-03-31 14:35:53 +00001213 raise ResponseNotReady(self.__state)
Greg Stein5e0fa402000-06-26 08:28:01 +00001214
Jeremy Hylton30f86742000-09-18 22:50:38 +00001215 if self.debuglevel > 0:
Jeremy Hyltond46aa372002-07-06 18:48:07 +00001216 response = self.response_class(self.sock, self.debuglevel,
Jeremy Hyltonc1b2cb92003-05-05 16:13:58 +00001217 method=self._method)
Jeremy Hylton30f86742000-09-18 22:50:38 +00001218 else:
Antoine Pitrou988dbd72010-12-17 17:35:56 +00001219 response = self.response_class(self.sock, method=self._method)
Greg Stein5e0fa402000-06-26 08:28:01 +00001220
Serhiy Storchakab491e052014-12-01 13:07:45 +02001221 try:
R David Murraycae7bdb2015-04-05 19:26:29 -04001222 try:
1223 response.begin()
1224 except ConnectionError:
1225 self.close()
1226 raise
Serhiy Storchakab491e052014-12-01 13:07:45 +02001227 assert response.will_close != _UNKNOWN
1228 self.__state = _CS_IDLE
Greg Stein5e0fa402000-06-26 08:28:01 +00001229
Serhiy Storchakab491e052014-12-01 13:07:45 +02001230 if response.will_close:
1231 # this effectively passes the connection to the response
1232 self.close()
1233 else:
1234 # remember this, so we can tell when it is complete
1235 self.__response = response
Greg Stein5e0fa402000-06-26 08:28:01 +00001236
Serhiy Storchakab491e052014-12-01 13:07:45 +02001237 return response
1238 except:
1239 response.close()
1240 raise
Greg Stein5e0fa402000-06-26 08:28:01 +00001241
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001242try:
1243 import ssl
Brett Cannoncd171c82013-07-04 17:43:24 -04001244except ImportError:
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001245 pass
1246else:
1247 class HTTPSConnection(HTTPConnection):
1248 "This class allows communication via SSL."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001249
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001250 default_port = HTTPS_PORT
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001251
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001252 # XXX Should key_file and cert_file be deprecated in favour of context?
1253
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001254 def __init__(self, host, port=None, key_file=None, cert_file=None,
Senthil Kumaran052ddb02013-03-18 14:11:41 -07001255 timeout=socket._GLOBAL_DEFAULT_TIMEOUT,
1256 source_address=None, *, context=None,
1257 check_hostname=None):
1258 super(HTTPSConnection, self).__init__(host, port, timeout,
Gregory P. Smithb4066372010-01-03 03:28:29 +00001259 source_address)
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001260 self.key_file = key_file
1261 self.cert_file = cert_file
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001262 if context is None:
Benjamin Peterson4ffb0752014-11-03 14:29:33 -05001263 context = ssl._create_default_https_context()
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001264 will_verify = context.verify_mode != ssl.CERT_NONE
1265 if check_hostname is None:
Benjamin Petersona090f012014-12-07 13:18:25 -05001266 check_hostname = context.check_hostname
1267 if check_hostname and not will_verify:
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001268 raise ValueError("check_hostname needs a SSL context with "
1269 "either CERT_OPTIONAL or CERT_REQUIRED")
1270 if key_file or cert_file:
Antoine Pitrou80603992010-10-13 10:45:16 +00001271 context.load_cert_chain(cert_file, key_file)
Antoine Pitrou803e6d62010-10-13 10:36:15 +00001272 self._context = context
1273 self._check_hostname = check_hostname
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001274
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001275 def connect(self):
1276 "Connect to a host on a given (SSL) port."
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001277
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001278 super().connect()
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001279
1280 if self._tunnel_host:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001281 server_hostname = self._tunnel_host
1282 else:
1283 server_hostname = self.host
Senthil Kumaran97f0c6b2009-07-25 04:24:38 +00001284
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001285 self.sock = self._context.wrap_socket(self.sock,
Benjamin Peterson7243b572014-11-23 17:04:34 -06001286 server_hostname=server_hostname)
Christian Heimesf723c712013-12-05 07:51:17 +01001287 if not self._context.check_hostname and self._check_hostname:
1288 try:
Senthil Kumaran9da047b2014-04-14 13:07:56 -04001289 ssl.match_hostname(self.sock.getpeercert(), server_hostname)
Christian Heimesf723c712013-12-05 07:51:17 +01001290 except Exception:
1291 self.sock.shutdown(socket.SHUT_RDWR)
1292 self.sock.close()
1293 raise
Jeremy Hylton29d27ac2002-07-09 21:22:36 +00001294
Thomas Wouters47b49bf2007-08-30 22:15:33 +00001295 __all__.append("HTTPSConnection")
Greg Stein5e0fa402000-06-26 08:28:01 +00001296
Greg Stein5e0fa402000-06-26 08:28:01 +00001297class HTTPException(Exception):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001298 # Subclasses that define an __init__ must call Exception.__init__
1299 # or define self.args. Otherwise, str() will fail.
Greg Steindd6eefb2000-07-18 09:09:48 +00001300 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001301
1302class NotConnected(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001303 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001304
Skip Montanaro9d389972002-03-24 16:53:50 +00001305class InvalidURL(HTTPException):
1306 pass
1307
Greg Stein5e0fa402000-06-26 08:28:01 +00001308class UnknownProtocol(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001309 def __init__(self, version):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001310 self.args = version,
Greg Steindd6eefb2000-07-18 09:09:48 +00001311 self.version = version
Greg Stein5e0fa402000-06-26 08:28:01 +00001312
1313class UnknownTransferEncoding(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001314 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001315
Greg Stein5e0fa402000-06-26 08:28:01 +00001316class UnimplementedFileMode(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001317 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001318
1319class IncompleteRead(HTTPException):
Benjamin Peterson6accb982009-03-02 22:50:25 +00001320 def __init__(self, partial, expected=None):
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001321 self.args = partial,
Greg Steindd6eefb2000-07-18 09:09:48 +00001322 self.partial = partial
Benjamin Peterson6accb982009-03-02 22:50:25 +00001323 self.expected = expected
1324 def __repr__(self):
1325 if self.expected is not None:
1326 e = ', %i more expected' % self.expected
1327 else:
1328 e = ''
Serhiy Storchaka465e60e2014-07-25 23:36:00 +03001329 return '%s(%i bytes read%s)' % (self.__class__.__name__,
1330 len(self.partial), e)
Benjamin Peterson6accb982009-03-02 22:50:25 +00001331 def __str__(self):
1332 return repr(self)
Greg Stein5e0fa402000-06-26 08:28:01 +00001333
1334class ImproperConnectionState(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001335 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001336
1337class CannotSendRequest(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001338 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001339
1340class CannotSendHeader(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001341 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001342
1343class ResponseNotReady(ImproperConnectionState):
Greg Steindd6eefb2000-07-18 09:09:48 +00001344 pass
Greg Stein5e0fa402000-06-26 08:28:01 +00001345
1346class BadStatusLine(HTTPException):
Greg Steindd6eefb2000-07-18 09:09:48 +00001347 def __init__(self, line):
Benjamin Peterson11dbfd42010-03-21 22:50:04 +00001348 if not line:
1349 line = repr(line)
Jeremy Hylton12f4f352002-07-06 18:55:01 +00001350 self.args = line,
Greg Steindd6eefb2000-07-18 09:09:48 +00001351 self.line = line
Greg Stein5e0fa402000-06-26 08:28:01 +00001352
Senthil Kumaran5466bf12010-12-18 16:55:23 +00001353class LineTooLong(HTTPException):
1354 def __init__(self, line_type):
1355 HTTPException.__init__(self, "got more than %d bytes when reading %s"
1356 % (_MAXLINE, line_type))
1357
R David Murraycae7bdb2015-04-05 19:26:29 -04001358class RemoteDisconnected(ConnectionResetError, BadStatusLine):
1359 def __init__(self, *pos, **kw):
1360 BadStatusLine.__init__(self, "")
1361 ConnectionResetError.__init__(self, *pos, **kw)
1362
Greg Stein5e0fa402000-06-26 08:28:01 +00001363# for backwards compatibility
1364error = HTTPException