1 | 1 | | #!/usr/bin/env python |
2 | 2 | | # -*- coding: utf-8 -*- |
3 | 3 | | |
4 | | - | from burp import IBurpExtender |
5 | | - | from burp import IHttpListener |
6 | | - | |
| 4 | + | from burp import IBurpExtender, IHttpListener, IProxyListener |
7 | 5 | | from authorization.authorization import handle_message |
8 | | - | |
9 | 6 | | from helpers.initiator import Initiator |
| 7 | + | from helpers.filters import handle_proxy_message |
10 | 8 | | |
11 | | - | class BurpExtender(IBurpExtender, IHttpListener): |
| 9 | + | class BurpExtender(IBurpExtender, IHttpListener, IProxyListener): |
12 | 10 | | |
13 | 11 | | def registerExtenderCallbacks(self, callbacks): |
14 | 12 | | self._callbacks = callbacks |
| skipped 21 lines |
36 | 34 | | def processHttpMessage(self, toolFlag, messageIsRequest, messageInfo): |
37 | 35 | | handle_message(self, toolFlag, messageIsRequest, messageInfo) |
38 | 36 | | |
| 37 | + | # |
| 38 | + | # implement IProxyListener |
| 39 | + | # |
| 40 | + | def processProxyMessage(self, messageIsRequest, message): |
| 41 | + | handle_proxy_message(self,message) |
| 42 | + | |
| 43 | + | |