allow limiting servers with a white list

[punjab-krb5-preauth] / punjab / httpb.py

diff --git a/punjab/httpb.py b/punjab/httpb.py
index e0b081f..35debf7 100644 (file)
--- a/punjab/httpb.py
+++ b/punjab/httpb.py
@@ -532,8 +532,12 @@ class HttpbService(punjab.Service):
 
     implements(IHttpbService)
 
-    def __init__(self, verbose = 0, polling = 15, 
-                 use_raw = False, bindAddress=("0.0.0.0", 0), session_creator = None):
+    white_list = []
+
+    def __init__(self, 
+                 verbose = 0, polling = 15, 
+                 use_raw = False, bindAddress=("0.0.0.0", 0), 
+                 session_creator = None):
         if session_creator is not None:
             self.make_session = session_creator
         else:
@@ -577,6 +581,9 @@ class HttpbService(punjab.Service):
         if not body.hasAttribute('to') or body['to']=='':
             return None, defer.fail(error.BadRequest)
         
+        if self.white_list and body['to'] not in self.white_list:
+            return None, defer.fail(error.BadRequest)
+
         # look for wait
         if not body.hasAttribute('wait') or body['wait']=='':
             body['wait'] = 3