210-Userdir_ScriptName.py

import os
import string
from base import *

DOMAIN      = "210domain"
DIR         = "210/directory"
FILE        = "file.cgi"
PUBLIC_HTML = "tmp"
MAGIC       = "alobbs.com"

CONF = """
vserver!2100!nick = %s
vserver!2100!document_root = %s
vserver!2100!user_dir = %s

vserver!2100!user_dir!rule!2!match!directory = /%s
vserver!2100!user_dir!rule!2!match = directory
vserver!2100!user_dir!rule!2!handler = cgi
vserver!2100!user_dir!rule!1!match = default
vserver!2100!user_dir!rule!1!handler = file
"""

CGI_BASE = """#!/bin/sh
echo "Content-type: text/html"
echo ""
echo "SCRIPT_NAME: -${SCRIPT_NAME}-"
"""

class Test (TestBase):
    def __init__ (self):
        TestBase.__init__ (self, __file__)
        self.name             = "Userdir: ScriptName"
        self.expected_error   = 200

    def Precondition (self):
        # ~/tmp/cherokee, alo
        tmp = self.Precondition_UserHome ("tmp")
        if not tmp:
            return False
        self.public_html, self.user = tmp

        self.request           = "GET /~%s/%s/%s HTTP/1.0\r\n" % (self.user, DIR, FILE) + \
                                 "Host: %s\r\n" % (DOMAIN)
        self.forbidden_content = ["/bin/sh", "echo"]
        self.expected_content  = ["SCRIPT_NAME: -/~%s/%s/%s-"%(self.user, DIR, FILE)]

        return True

    def Prepare (self, www):
        self.Remove (self.public_html, '210')
        droot = self.Mkdir (self.public_html, DIR)

        self.WriteFile (droot, FILE, 0555, CGI_BASE)
        self.conf = CONF % (DOMAIN, droot, PUBLIC_HTML, DIR)