From 08ca878020ff1203e21399772118447b2c8f0f13 Mon Sep 17 00:00:00 2001 From: Christopher tofu Zorn Date: Mon, 8 Feb 2010 09:07:29 -0500 Subject: [PATCH] allow limiting servers with a white list --- punjab/httpb.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/punjab/httpb.py b/punjab/httpb.py index e0b081f..35debf7 100644 --- 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 -- 1.7.10.4