Нужна помощь в написании скрученного прокси - PullRequest
Новая
       15

Нужна помощь в написании скрученного прокси

9 голосов
/ 27 июня 2011

Я хочу написать простой прокси, который перетасовывает текст в теле запрашиваемых страниц. Я прочитал части скрученной документации и некоторые другие подобные вопросы здесь по stackoverflow, но я немного новичок, поэтому я до сих пор не понимаю.

Это то, что у меня сейчас, я не знаю, как получить доступ и изменить страницу 

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

log.startLogging(sys.stdout)

class ProxyProtocol(http.HTTPChannel):
   requestFactory = PageHandler

class ProxyFactory(http.HTTPFactory):
   protocol = ProxyProtocol

if __name__ == '__main__':
   reactor.listenTCP(8080, ProxyFactory())
   reactor.run()

Не могли бы вы мне помочь? Я был бы признателен за простой пример (например, добавить что-то в тело и т. Д.).

1 Ответ

6 голосов
/ 29 июня 2011

Что я делаю, так это внедряю новый ProxyClient, где я изменяю данные после того, как я скачал их с веб-сервера, и перед тем, как отправить их в веб-браузер. 

from twisted.web import proxy, http
class MyProxyClient(proxy.ProxyClient):
 def __init__(self,*args,**kwargs):
  self.buffer = ""
  proxy.ProxyClient.__init__(self,*args,**kwargs)
 def handleResponsePart(self, buffer):
  # Here you will get the data retrieved from the web server
  # In this example, we will buffer the page while we shuffle it.
  self.buffer = buffer + self.buffer
 def handleResponseEnd(self):
  if not self._finished:
   # We might have increased or decreased the page size. Since we have not written
   # to the client yet, we can still modify the headers.
   self.father.responseHeaders.setRawHeaders("content-length", [len(self.buffer)])
   self.father.write(self.buffer)
  proxy.ProxyClient.handleResponseEnd(self)

class MyProxyClientFactory(proxy.ProxyClientFactory):
 protocol = MyProxyClient

class ProxyRequest(proxy.ProxyRequest):
 protocols = {'http': MyProxyClientFactory}
 ports = {'http': 80 }
 def process(self):
  proxy.ProxyRequest.process(self)

class MyProxy(http.HTTPChannel):
 requestFactory = ProxyRequest

class ProxyFactory(http.HTTPFactory):
 protocol = MyProxy

Надеюсь, этотакже работает для вас.

...