Rewrite HTTP_REFERRER check using BeforeTraverse API rather than monkey paching

everybody's REQUEST.
Note: the new way to enable it is by calling 'enableRefererCheck' method on the
portal.



git-svn-id: https://svn.erp5.org/repos/public/erp5/trunk@8709 20353a03-c40f-0410-a6d1-a30d3c3de9de

product/ERP5/ERP5Site.py

@@ -15,8 +15,10 @@
 """
 
 import Globals
+import AccessControl
 from Globals import package_home
 
+from ZPublisher import BeforeTraverse
 from AccessControl import ClassSecurityInfo
 from Products.CMFDefault.Portal import CMFSite, PortalGenerator
 from Products.CMFCore.utils import getToolByName, _getAuthenticatedUser
@@ -30,9 +32,8 @@ from Products.ERP5Type.ERP5Type import ERP5TypeInformation
 from Products.ERP5.Document.BusinessTemplate import BusinessTemplate
 
 import ERP5Defaults
-from os import path
 
-from zLOG import LOG
+from zLOG import LOG, INFO
 from string import join
 import os
 MARKER = []
@@ -89,6 +90,49 @@ def manage_addERP5Site(self,
   if RESPONSE is not None:
     RESPONSE.redirect(p.absolute_url())
 
+
+class ReferCheckerBeforeTraverseHook:
+  """This before traverse hook checks the HTTP_REFERER argument in the request
+  and refuses access to anything else that portal_url.
+
+  This is enabled by calling the method enableRefererCheck on the portal.
+  """
+  handle = '_erp5_referer_check'
+
+  def __call__(self, container, request):
+    """Checks the request contains a valid referrer.
+    """
+    response = request.RESPONSE
+    http_url = request.get('ACTUAL_URL', '').strip()
+    http_referer = request.get('HTTP_REFERER', '').strip()
+
+    security_manager = AccessControl.getSecurityManager()
+    user = security_manager.getUser()
+    user_roles = user.getRolesInContext(object)
+
+    # Manager can do anything
+    if 'Manager' in user_roles:
+      return
+    
+    portal_url = container.portal_url.getPortalObject().absolute_url()
+    if http_referer != '':
+      # if HTTP_REFERER is set, user can acces the object if referer is ok
+      if http_referer.startswith(portal_url):
+        return
+      LOG('HTTP_REFERER_CHECK : BAD REFERER !', INFO,
+          'request : "%s", referer : "%s"' % (http_url, http_referer))
+      response.unauthorized()
+    else:
+      # no HTTP_REFERER, we only allow to reach portal_url
+      for i in ('/', '/index_html', '/login_form', '/view'):
+        if http_url.endswith(i):
+          http_url = http_url[:-len(i)]
+          break
+      if len(http_url) == 0 or not portal_url.startswith(http_url):
+        LOG('HTTP_REFERER_CHECK : NO REFERER !', INFO,
+            'request : "%s"' % http_url)
+        response.unauthorized()
+
 class ERP5Site(FolderMixIn, CMFSite):
   """
   The *only* function this class should have is to help in the setup
@@ -121,6 +165,20 @@ class ERP5Site(FolderMixIn, CMFSite):
     """
     return self.index_html()
   
+  security.declareProtected(Permissions.ManagePortal, 'enableRefererCheck')
+  def enableRefererCheck(self):
+    """Enable a ReferCheckerBeforeTraverseHook to check users have valid
+    HTTP_REFERER
+    """
+    BeforeTraverse.registerBeforeTraverse(self,
+                                        ReferCheckerBeforeTraverseHook(),
+                                        ReferCheckerBeforeTraverseHook.handle)
+  
+  def _disableRefererCheck(self):
+    """Disable the HTTP_REFERER check."""
+    BeforeTraverse.unregisterBeforeTraverse(self,
+                                        ReferCheckerBeforeTraverseHook.handle)
+
   def hasObject(self, id):
     """
     Check if the portal has an id.

product/ERP5Type/ZopePatch.py

@@ -33,7 +33,6 @@ from Products.ERP5Type.patches import ppml
 from Products.ERP5Type.patches import Expression
 from Products.ERP5Type.patches import sqlvar
 from Products.ERP5Type.patches import CMFCatalogAware
-from Products.ERP5Type.patches import BaseRequest
 from Products.ERP5Type.patches import ProductContext
 from Products.ERP5Type.patches import PropertiedUser
 from Products.ERP5Type.patches import States

product/ERP5Type/patches/BaseRequest.py

-##############################################################################
-#
-# Copyright (c) 2001 Zope Corporation and Contributors. All Rights Reserved.
-# Copyright (c) 2002,2005 Nexedi SARL and Contributors. All Rights Reserved.
-#
-# This software is subject to the provisions of the Zope Public License,
-# Version 2.0 (ZPL).  A copy of the ZPL should accompany this distribution.
-# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
-# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
-# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
-# FOR A PARTICULAR PURPOSE
-#
-##############################################################################
-
-# ZPublisher should drop requests without a good http referer
-
-from ZPublisher.BaseRequest import BaseRequest
-from zLOG import LOG
-import AccessControl
-
-BaseRequest.erp5_old_traverse = BaseRequest.traverse
-
-def erp5_new_traverse(request, path, response=None, validated_hook=None):
-
-  if response is None: response=request.response
-  object = BaseRequest.erp5_old_traverse(request, path, response=response, validated_hook=validated_hook)
-  http_url = request.get('ACTUAL_URL', '').strip()
-  http_referer = request.get('HTTP_REFERER', '').strip()
-
-  security_manager = AccessControl.getSecurityManager()
-  user = security_manager.getUser()
-  user_roles = user.getRolesInContext(object)
-
-  # Manager can do anything
-  if 'Manager' in user_roles:
-    return object
-
-  # are we within a portal ?
-  try:
-    context = getattr(object, 'im_self', None)
-    if context is not None:
-      try:
-        portal_object = context.getPortalObject()
-      except AttributeError:
-        portal_object = object.getPortalObject()
-    else :
-      portal_object = object.getPortalObject()
-  except AttributeError:
-    pass
-  else:
-    if not getattr(portal_object, 'require_referer', 0):
-      return object
-    portal_url = portal_object.absolute_url()
-    if http_referer != '':
-      # if HTTP_REFERER is set, user can acces the object if referer is ok
-      if http_referer.startswith(portal_url):
-        return object
-      else:
-        LOG('HTTP_REFERER_CHECK : BAD REFERER !', 0, 'request : "%s", referer : "%s"' % (http_url, referer))
-        response.unauthorized()
-    else:
-      # no HTTP_REFERER, we only allow to reach portal_url
-      for i in ('/', '/index_html', '/login_form', '/view'):
-        if http_url.endswith(i):
-          http_url = http_url[:-len(i)]
-          break
-      if len(http_url) == 0 or not portal_url.startswith(http_url):
-        LOG('HTTP_REFERER_CHECK : NO REFERER !', 0, 'request : "%s"' % http_url)
-        response.unauthorized()
-
-  return object
-
-BaseRequest.traverse = erp5_new_traverse