Produits Entr'ouvert » Larpe

import os

app_dir = '/var/lib/larpe'

    # Only filter html code

    if filter.req.content_type is not None:

        is_html = re.search('text/html', filter.req.content_type)

    if filter.req.content_type is not None and not is_html:

        filter.pass_on()

    else:

        if not hasattr(filter.req, 'temp_doc'): # the start

            filter.req.temp_doc = [] # create new attribute to hold document

            # If content-length ended up wrong, Gecko browsers truncated data, so

            if 'Content-Length' in filter.req.headers_out:

                del filter.req.headers_out['Content-Length']

#        filter.write(filter.req.headers_in['Cookie'])

#        delete_cookies(filter)

        #filter.req.headers_out['Set-Cookie'] = 'dc_admin="deleted"; max-age=0; expires=Thu, 01-Jan-1970 00:00:00 GMT; path=/'

        temp_doc = filter.req.temp_doc

        s = filter.read()

        while s: # could get '' at any point, but only get None at end

            temp_doc.append(s)

            s = filter.read()

        if s is None: # the end

            temp_doc = ''.join(temp_doc)

            #filter.req.set_content_length(len(temp_doc)) # this didn't seem to work

            filter.write(temp_doc)

            #filter.write(filter.req.uri)

            filter.close()

def get_abs_path(s):

    if not s:

        return s

    if s[0] == '/':

        return s

    return os.path.join(app_dir, s)

def get_proxied_site_path(filter):

    proxy_domain_name = filter.req.hostname

    proxied_site_dir = get_proxied_site_name(filter)

    return get_abs_path(os.path.join('sp', proxy_domain_name, proxied_site_dir))

def get_proxied_site_name(filter):

    uri_tokens = filter.req.uri.split('/')

    if uri_tokens[1] != 'liberty':

        return uri_tokens[1]

    return uri_tokens[2]

def delete_cookies(filter):

    success = False

    cookies_file_name = get_abs_path(os.path.join(get_proxied_site_path(filter), 'cookies_to_delete'))

    cookies_file = open(cookies_file_name, 'r')

    for cookie in cookies_file.read().split():

        if filter.req.headers_in.has_key('Cookie'):

            cookies_header = filter.req.headers_in['Cookie']

#            filter.req.temp_doc.append(filter.req.headers_in['Cookie'])

            #filter.req.temp_doc.append(cookie[len(cookie) -1:])

            cookies_match =  re.findall(cookie[:len(cookie) -1], cookies_header)

            if len(cookies_match) > 0:

                filter.req.temp_doc.append('User-Agent')

#                filter.req.temp_doc.append('%s="deleted"; max-age=0; expires=Thu, 01-Jan-1970 00:00:00 GMT; path=/' % cookie.split('=')[0])

                filter.req.headers_out['Set-Cookie'] = '%s="deleted"; max-age=0; expires=Thu, 01-Jan-1970 00:00:00 GMT; path=/' % cookie.split('=')[0]

#                cookies_file.close()

#                cookies_file = open(cookies_file_name, 'w')

                break

#            else:

#                filter.req.temp_doc.append('dommage')

    cookies_file.close()

#    if success:

#        cookies_file = open(cookies_file_name, 'w')

#        cookies_file.close()