Skip to content
This repository
Fetching contributors…

Cannot retrieve contributors at this time

file 45 lines (34 sloc) 1.177 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
from base import *

DIR = "x_real_ip_cgi_1"
REMOTE_IP = "1.2.3.4"

CONF = """
vserver!1!rule!2320!match = directory
vserver!1!rule!2320!match!directory = /%s
vserver!1!rule!2320!handler = cgi
vserver!1!rule!2320!handler!x_real_ip_enabled = 1
vserver!1!rule!2320!handler!x_real_ip_access_all = 0
vserver!1!rule!2320!handler!x_real_ip_access = ::1,127.0.0.1
""" % (DIR)

CGI_CODE = """#!/bin/sh

echo "Content-Type: text/plain"
echo

echo "REMOTE_ADDR ->${REMOTE_ADDR}<-"
echo
"""

class Test (TestBase):
    def __init__ (self):
        TestBase.__init__ (self, __file__)
        self.name = "X-Real-IP to REMOTE_ADDR: CGI"

        self.request = "GET /%s/test HTTP/1.0\r\n" % (DIR) + \
                                 "X-Real-IP: %s\r\n" % (REMOTE_IP)
        self.expected_error = 200
        self.conf = CONF
        self.proxy_suitable = False

    def CustomTest (self):
        body = self.reply.split ("\r\n\r\n")[1]

        if "REMOTE_ADDR ->%s<-" %(REMOTE_IP) in body:
            return 0

        return -1

    def Prepare (self, www):
        d = self.Mkdir (www, DIR, 0777)
        self.WriteFile (d, "test", 0555, CGI_CODE)
Something went wrong with that request. Please try again.