def read_header(self, data):
data = data.strip()
if data == '':
# check for Accept-Encoding: header, pick a
if self.headers.has_key('accept-encoding'):
ae = self.headers['accept-encoding']
log.debug("Got Accept-Encoding: " + ae + "\n")
else:
#identity assumed if no header
ae = 'identity'
# this eventually needs to support multple acceptable types
# q-values and all that fancy HTTP crap
# for now assume we're only communicating with our own client
if ae.find('gzip') != -1:
self.encoding = 'gzip'
else:
#default to identity.
self.encoding = 'identity'
r = self.getfunc(self, self.path, self.headers)
if r is not None:
self.answer(r)
return None
try:
i = data.index(':')
except ValueError:
return None
self.headers[data[:i].strip().lower()] = data[i+1:].strip()
log.debug(data[:i].strip() + ": " + data[i+1:].strip())
return self.read_header
class HTTPSConnection(Dispatcher):
def __init__(self, socket, getfunc):
Dispatcher.__init__(self, socket)
self.req = ''
self.set_terminator('\n')
self.getfunc = getfunc
self.next_func = self.read_type
## HTTP handling methods ##
def read_type(self, data):
self.header = data.strip()
words = data.split()
if len(words) == 3:
self.command, self.path, garbage = words
self.pre1 = False
elif len(words) == 2:
self.command, self.path = words
self.pre1 = True
if self.command != 'GET':
return None
else:
return None
if self.command not in ('HEAD', 'GET'):
return None
self.headers = {}
return self.read_header
def read_header(self, data):
data = data.strip()
if data == '':
# check for Accept-Encoding: header, pick a
if self.headers.has_key('accept-encoding'):
ae = self.headers['accept-encoding']
log.debug("Got Accept-Encoding: " + ae + "\n")
else:
#identity assumed if no header
ae = 'identity'
# this eventually needs to support multple acceptable types
# q-values and all that fancy HTTP crap
# for now assume we're only communicating with our own client
if ae.find('gzip') != -1:
self.encoding = 'gzip'
else:
#default to identity.
self.encoding = 'identity'
r = self.getfunc(self, self.path, self.headers)
if r is not None:
self.answer(r)
return None
try:
i = data.index(':')
except ValueError:
return None
self.headers[data[:i].strip().lower()] = data[i+1:].strip()
log.debug(data[:i].strip() + ": " + data[i+1:].strip())
return self.read_header
def answer(self, (responsecode, responsestring, headers, data)):
if self.encoding == 'gzip':
#transform data using gzip compression
#this is nasty but i'm unsure of a better way at the moment
compressed = StringIO()
gz = GzipFile(fileobj = compressed, mode = 'wb', compresslevel = 9)
gz.write(data)
gz.close()
compressed.seek(0,0)
cdata = compressed.read()
compressed.close()
if len(cdata) >= len(data):
self.encoding = 'identity'
else:
log.debug("Compressed: %i Uncompressed: %i\n" % (len(cdata),len(data)))
data = cdata
headers['Content-Encoding'] = 'gzip'
# i'm abusing the identd field here, but this should be ok
if self.encoding == 'identity':
ident = '-'
else:
ident = self.encoding
username = '******'
referer = self.headers.get('referer','-')
useragent = self.headers.get('user-agent','-')
timestamp = strftime("%d/%b/%Y:%H:%I:%S")
log.info('%s %s %s [%s] "%s" %i %i "%s" "%s"' % (
self.socket.addr[0], ident, username, timestamp, self.header,
responsecode, len(data), referer, useragent))
r = StringIO()
r.write('HTTP/1.0 ' + str(responsecode) + ' ' + responsestring + '\r\n')
if not self.pre1:
headers['Content-Length'] = len(data)
for key, value in headers.items():
r.write(key + ': ' + str(value) + '\r\n')
r.write('\r\n')
if self.command != 'HEAD':
r.write(data)
self.push(r.getvalue())
self.close_when_done()
