Merge branch 'master' into issue_341

2 files changed, 20 insertions, 8 deletions

diff --git a/libmproxy/cmdline.py b/libmproxy/cmdline.py
index b9e62b71..28156a59 100644
--- a/libmproxy/cmdline.py
+++ b/libmproxy/cmdline.py
@@ -421,8 +421,7 @@ def common_options(parser):
         "Proxy Authentication",
         """
             Specify which users are allowed to access the proxy and the method
-            used for authenticating them. These options are ignored if the
-            proxy is in transparent or reverse proxy mode.
+            used for authenticating them.
         """
     )
     user_specification_group = group.add_mutually_exclusive_group()
diff --git a/libmproxy/script.py b/libmproxy/script.py
index 706d84d5..f5fb6b41 100644
--- a/libmproxy/script.py
+++ b/libmproxy/script.py
@@ -108,15 +108,28 @@ class Script:
             return (False, None)
 
 
+class ReplyProxy(object):
+    def __init__(self, original_reply):
+        self._ignore_calls = 1
+        self.lock = threading.Lock()
+        self.original_reply = original_reply
+
+    def __call__(self, *args, **kwargs):
+        with self.lock:
+            if self._ignore_calls > 0:
+                self._ignore_calls -= 1
+                return
+        self.original_reply(*args, **kwargs)
+
+    def __getattr__ (self, k):
+        return getattr(self.original_reply, k)
+
+
 def _handle_concurrent_reply(fn, o, *args, **kwargs):
     # Make first call to o.reply a no op
-    original_reply = o.reply
 
-    def restore_original_reply():
-        o.reply = original_reply
-    if hasattr(original_reply, "q"):
-        restore_original_reply.q = original_reply.q
-    o.reply = restore_original_reply
+    reply_proxy = ReplyProxy(o.reply)
+    o.reply = reply_proxy
 
     def run():
         fn(*args, **kwargs)