Python Twisted 代理 - 如何拦截数据包

Question

我正在尝试使用 Python 打印出 HTTP 响应的主体。

到目前为止，这是我的代码:

from twisted.web import proxy, http
from twisted.internet import reactor
from twisted.python import log
import sys

log.startLogging(sys.stdout)

class ProxyFactory(http.HTTPFactory):
  protocol=proxy.Proxy

reactor.listenTCP(8080, ProxyFactory())
reactor.run()

当我将浏览器连接到 localhost:8080 时，我可以看到我的所有请求都通过本地运行的 Python 代理进行定向。但是我如何 1) 打印出响应主体和 2) 在将响应主体发送回浏览器之前编辑响应主体？

我希望有人能给我指出正确的方向——请记住我是 Python 的新手!

Answer 1

覆盖dataReceived协议(protocol)的方法(在您的情况下为 proxy.Proxy)并处理该方法中的数据修改:

from twisted.web import proxy, http
from twisted.internet import reactor
from twisted.python import log
import sys

log.startLogging(sys.stdout)

class MyProxy(proxy.Proxy):
    def dataReceived(self, data):

      # Modify the data here
      print data

      # perform the default functionality on modified data 
      return proxy.Proxy.dataReceived(self, data)

class ProxyFactory(http.HTTPFactory):
  protocol=MyProxy

factory = ProxyFactory()
reactor.listenTCP(8080, factory)
reactor.run()