def make_ssl_aware(self, sock, protocol):
"""
Make the socket SSL aware
"""
try:
ssl_sock = wrap_socket(sock,
keyfile=self.key_file,
certfile=self.cert_file,
ssl_version=protocol,
server_hostname=self.host,
timeout=cf.get('timeout'))
except ssl.SSLError, ssl_exc:
msg = "SSL connection error occurred with protocol %s: '%s'"
debug(msg % (protocol, ssl_exc))
# Always close the tcp/ip connection on error
sock.close()
def _raw_read(self, amt=None):
"""
This is the original read function from httplib with a minor
modification that allows me to check the size of the file being
fetched, and throw an exception in case it is too big.
"""
if self.fp is None:
return ''
max_file_size = cf.get('max_file_size') or None
if max_file_size:
if self.length > max_file_size:
self.status = NO_CONTENT
self.reason = 'No Content' # Reason-Phrase
self.close()
return ''
if self.chunked:
return self._read_chunked(amt)
if amt is None:
# unbounded read
if self.length is None:
s = self.fp.read()
else:
s = self._safe_read(self.length)
self.length = 0
self.close() # we read everything
return s
if self.length is not None:
if amt > self.length:
# clip the read to the "end of response"
amt = self.length
# we do not use _safe_read() here because this may be a .will_close
# connection, and the user is reading more bytes than will be provided
# (for example, reading in 1k chunks)
s = self.fp.read(amt)
if self.length is not None:
self.length -= len(s)
return s
def _raw_read(self, amt=None):
"""
This is the original read function from httplib with a minor
modification that allows me to check the size of the file being
fetched, and throw an exception in case it is too big.
"""
if self.fp is None:
return ''
max_file_size = cf.get('max_file_size') or None
if max_file_size:
if self.length > max_file_size:
self.status = NO_CONTENT
self.reason = 'No Content' # Reason-Phrase
self.close()
return ''
if self.chunked:
return self._read_chunked(amt)
if amt is None:
# unbounded read
if self.length is None:
s = self.fp.read()
else:
s = self._safe_read(self.length)
self.length = 0
self.close() # we read everything
return s
if self.length is not None:
if amt > self.length:
# clip the read to the "end of response"
amt = self.length
# we do not use _safe_read() here because this may be a .will_close
# connection, and the user is reading more bytes than will be provided
# (for example, reading in 1k chunks)
s = self.fp.read(amt)
if self.length is not None:
self.length -= len(s)
return s
def __init__(self, host, port=None, strict=None):
_HTTPConnection.__init__(self, host,
port=port,
strict=strict,
timeout=cf.get('timeout'))