скрученные методы модификации запроса прокси

Question

Как просмотреть запросы, отправленные витым прокси-сервером, на какую часть файла twisted.web.proxy я должен смотреть. Цель состоит в том, чтобы добавить дополнительные заголовки, прежде чем dataReceived будет отправлен на целевой сервер. Например, добавив дополнительный заголовок, как «Xzone: FOO.COM \ г» для всех запросов прокси-сервера, так что окончательный запрос выглядит следующим образом:

GET/HTTP/1.1\r 
Xzone: foo.com\r 
Host: bbb.com\r 
User-Agent: mozilla\r 
Accept: */*\r 

Заранее спасибо за любую информацию.

Answer 1

К сожалению, twisted.web.proxy довольно крут и не был написан с учетом такого рода настроек. Тем не менее, можно вводить новые заголовки в прокси-запрос несколькими способами. Общая идея состоит в том, чтобы подключиться к одному из этапов обработки запроса и изменить заголовки запросов, чтобы они включали ваш новый заголовок.

Например, вы можете подключить и ProxyRequest.process:

from twisted.web.proxy import ProxyRequest 

class ChangeSomeHeadersProxyRequest(ProxyRequest): 
    def process(self): 
     self.changeHeaders() 
     return ProxyRequest.process(self) 

    def changeHeaders(self): 
     # Inject an extra request header. 
     self.requestHeaders.addRawHeader(b"Xzone", b"foo.com") 

Затем подключить это к остальной части вашего прокси-сервера:

from twisted.web.proxy import Proxy 
from twisted.web.server import Site 

class ChangeSomeHeadersSite(Site): 
    protocol = Proxy 
    requestFactory = ChangeSomeHeadersProxyRequest 

Теперь настроить ChangeSomeHeadersSite, как вы бы обычный Site ,