implement IProxyListener

P1a0z1 · Apr 11, 2023 · 6366b8f · 6366b8f
1 parent cd74b0d
commit 6366b8f
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/Autorize.py b/Autorize.py
@@ -1,14 +1,12 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*- 
 
-from burp import IBurpExtender
-from burp import IHttpListener
-
+from burp import IBurpExtender, IHttpListener, IProxyListener
 from authorization.authorization import handle_message
-
 from helpers.initiator import Initiator
+from helpers.filters import handle_proxy_message
 
-class BurpExtender(IBurpExtender, IHttpListener):
+class BurpExtender(IBurpExtender, IHttpListener, IProxyListener):
 
     def registerExtenderCallbacks(self, callbacks):
         self._callbacks = callbacks
@@ -35,3 +33,10 @@ def registerExtenderCallbacks(self, callbacks):
     #
     def processHttpMessage(self, toolFlag, messageIsRequest, messageInfo):      
         handle_message(self, toolFlag, messageIsRequest, messageInfo)
+
+    #
+    # implement IProxyListener
+    #
+    def processProxyMessage(self, messageIsRequest, message):
+        handle_proxy_message(self,message)
+