proper fix

stamparm · stamparm · commit 1b2cd4425545 · 2012-04-04T10:35:52.000Z
diff --git a/lib/core/common.py b/lib/core/common.py
@@ -1107,6 +1107,18 @@ def parseTargetUrl():
     conf.url = "%s://%s:%d%s" % (conf.scheme, conf.hostname, conf.port, conf.path)
     conf.url = conf.url.replace(URI_QUESTION_MARKER, '?')
 
+    if not conf.referer and intersect(REFERER_ALIASES, conf.testParameter, True):
+        debugMsg = "setting the HTTP Referer header to the target url"
+        logger.debug(debugMsg)
+        conf.httpHeaders = filter(lambda (key, value): key != HTTPHEADER.REFERER, conf.httpHeaders)
+        conf.httpHeaders.append((HTTPHEADER.REFERER, conf.url))
+
+    if not conf.host and intersect(HOST_ALIASES, conf.testParameter, True):
+        debugMsg = "setting the HTTP Host header to the target url"
+        logger.debug(debugMsg)
+        conf.httpHeaders = filter(lambda (key, value): key != HTTPHEADER.HOST, conf.httpHeaders)
+        conf.httpHeaders.append((HTTPHEADER.HOST, getHostHeader(conf.url)))
+
 def expandAsteriskForColumns(expression):
     """
     If the user provided an asterisk rather than the column(s)
