Web: get real ip (on proxy use)

circuits/web/tools.py

@@ -14,6 +14,9 @@
 from datetime import datetime, timedelta
 from email.generator import _make_boundary
 
+from circuits import BaseComponent, handler
+from circuits.web.wrappers import Host
+
 mimetypes.init()
 mimetypes.add_type("image/x-dwg", ".dwg")
 mimetypes.add_type("image/x-icon", ".ico")
@@ -449,3 +452,24 @@ def gzip(response, level=4, mime_types=("text/html", "text/plain",)):
     return httperror(
         response.request, response, 406, description="identity, gzip"
     )
+
+
+class ReverseProxy(BaseComponent):
+    headers = ('X-Real-IP', 'X-Forwarded-For')
+
+    def __init__(self, headers=None):
+        """ Component for have the original client IP when a reverse proxy is used
+
+        :param headers: List of HTTP headers to read the original client IP
+        """
+        super(ReverseProxy, self).__init__()
+
+        if headers:
+            self.headers = headers
+
+    @handler('request', priority=1)
+    def _on_request(self, req, *_):
+        for index in self.headers:
+            real_ip = req.headers.get(index)
+            req.remote = Host(real_ip, '', real_ip)
+            return