Mirror
/
zulip
mirror of https://github.com/zulip/zulip.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
zulip/tools/PythonProxy.py

244 lines
8.0 KiB
Python
Raw Normal View History

Add PythonProxy. This program can help us test error situations like CSS not loading. (imported from commit 4c3ed0825401741fe4aa826513d4eb43afd59ea3)
2013-12-10 21:19:37 +01:00
# -*- coding: cp1252 -*-
# <PythonProxy.py>
#
#Copyright (c) <2009> <F<>bio Domingues - fnds3000 in gmail.com>
#
#Permission is hereby granted, free of charge, to any person
#obtaining a copy of this software and associated documentation
#files (the "Software"), to deal in the Software without
#restriction, including without limitation the rights to use,
#copy, modify, merge, publish, distribute, sublicense, and/or sell
#copies of the Software, and to permit persons to whom the
#Software is furnished to do so, subject to the following
#conditions:
#
#The above copyright notice and this permission notice shall be
#included in all copies or substantial portions of the Software.
#
#THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
#EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
#OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
#NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
#HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
#WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
#FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
#OTHER DEALINGS IN THE SOFTWARE.
"""\
Copyright (c) <2009> <F<EFBFBD>bio Domingues - fnds3000 in gmail.com> <MIT Licence>
**************************************
*** Python Proxy - A Fast HTTP proxy ***
**************************************
Neste momento este proxy <EFBFBD> um Elie Proxy.
Suporta os m<EFBFBD>todos HTTP:
- OPTIONS;
- GET;
- HEAD;
- POST;
- PUT;
- DELETE;
- TRACE;
- CONENCT.
Suporta:
- Conex<EFBFBD>es dos cliente em IPv4 ou IPv6;
- Conex<EFBFBD>es ao alvo em IPv4 e IPv6;
- Conex<EFBFBD>es todo o tipo de transmiss<EFBFBD>o de dados TCP (CONNECT tunneling),
p.e. liga<EFBFBD><EFBFBD>es SSL, como <EFBFBD> o caso do HTTPS.
A fazer:
- Verificar se o input vindo do cliente est<EFBFBD> correcto;
- Enviar os devidos HTTP erros se n<EFBFBD>o, ou simplesmente quebrar a liga<EFBFBD><EFBFBD>o;
- Criar um gestor de erros;
- Criar ficheiro log de erros;
- Colocar excep<EFBFBD><EFBFBD>es nos s<EFBFBD>tios onde <EFBFBD> previs<EFBFBD>vel a ocorr<EFBFBD>ncia de erros,
p.e.sockets e ficheiros;
- Rever tudo e melhorar a estrutura do programar e colocar nomes adequados nas
vari<EFBFBD>veis e m<EFBFBD>todos;
- Comentar o programa decentemente;
- Doc Strings.
Funcionalidades futuras:
- Adiconar a funcionalidade de proxy an<EFBFBD>nimo e transparente;
- Suportar FTP?.
(!) Aten<EFBFBD><EFBFBD>o o que se segue s<EFBFBD> tem efeito em conex<EFBFBD>es n<EFBFBD>o CONNECT, para estas o
proxy <EFBFBD> sempre Elite.
Qual a diferen<EFBFBD>a entre um proxy Elite, An<EFBFBD>nimo e Transparente?
- Um proxy elite <EFBFBD> totalmente an<EFBFBD>nimo, o servidor que o recebe n<EFBFBD>o consegue ter
conhecimento da exist<EFBFBD>ncia do proxy e n<EFBFBD>o recebe o endere<EFBFBD>o IP do cliente;
- Quando <EFBFBD> usado um proxy an<EFBFBD>nimo o servidor sabe que o cliente est<EFBFBD> a usar um
proxy mas n<EFBFBD>o sabe o endere<EFBFBD>o IP do cliente;
<EFBFBD> enviado o cabe<EFBFBD>alho HTTP "Proxy-agent".
- Um proxy transparente fornece ao servidor o IP do cliente e um informa<EFBFBD><EFBFBD>o que
se est<EFBFBD> a usar um proxy.
S<EFBFBD>o enviados os cabe<EFBFBD>alhos HTTP "Proxy-agent" e "HTTP_X_FORWARDED_FOR".
"""
import socket, thread, select
PORT = 8085
__version__ = '0.1.0 Draft 1'
BUFLEN = 8192
VERSION = 'Python Proxy/'+__version__
HTTPVER = 'HTTP/1.1'
class ConnectionHandler:
def __init__(self, connection, address, timeout):
self.client = connection
self.client_buffer = ''
self.timeout = timeout
self.method, self.path, self.protocol = self.get_base_header()
if self.method=='CONNECT':
self.method_CONNECT()
elif self.method in ('OPTIONS', 'GET', 'HEAD', 'POST', 'PUT',
'DELETE', 'TRACE'):
self.method_others()
self.client.close()
self.target.close()
def get_base_header(self):
while 1:
self.client_buffer += self.client.recv(BUFLEN)
end = self.client_buffer.find('\n')
if end!=-1:
break
data = (self.client_buffer[:end+1]).split()
self.client_buffer = self.client_buffer[end+1:]
return data
def method_CONNECT(self):
print 'self.path', self.path
self._connect_target(self.path)
self.client.send(HTTPVER+' 200 Connection established\n'+
'Proxy-agent: %s\n\n'%VERSION)
self.client_buffer = ''
self._read_write()
def method_others(self):
print
print '====================================='
print 'method', self.method
print 'protocol', self.protocol
print 'self.path', self.path
print
horking = False
if horking:
if self.path.endswith('.js'):
print 'HORKING JS!!!!!!!'
print
return
if self.path.endswith('.png'):
print 'HORKING PNG!!!!!!!'
print
return
if self.path.endswith('.css'):
print 'HORKING CSS!!!!!!!'
print
return
self._connect_target()
payload = '%s %s %s\n'%(self.method, self.path, self.protocol)
buf = payload
try:
headers, rest = self.client_buffer.split('\r\n\r\n')
except:
headers = self.client_buffer
rest = None
for line in headers.split('\n'):
line = line.strip()
if not line:
continue
print repr(line)
field, value = line.split(':', 1)
if line.startswith('Host:'):
line = line.replace(str(PORT), '9991')
if field in ['Connection']:
line = 'Connection: close'
if field in ['If-Modified-Since']:
continue
buf += line + '\n'
buf += '\n'
buf = buf.replace('\n', '\r\n')
self.target.send(buf)
if rest:
print 'REST'
print repr(self.client_buffer)
print repr(buf)
print repr(rest)
self.target.send(rest)
self.client_buffer = ''
self._read_write()
def _connect_target(self):
host = '127.0.0.1'
port = 9991
(soc_family, _, _, _, address) = socket.getaddrinfo(host, port)[0]
self.target = socket.socket(soc_family)
print 'Connecting...', host, port
self.target.connect(address)
print 'Connected'
def _read_write(self):
time_out_max = self.timeout/3
socs = [self.client, self.target]
count = 0
while 1:
count += 1
(recv, _, error) = select.select(socs, [], socs, 3)
if error:
break
if recv:
for in_ in recv:
data = in_.recv(BUFLEN)
if in_ is self.client:
out = self.target
else:
if data:
print
print 'OUT'
print 'path =', self.path
print len(data)
print repr(data)
print
out = self.client
# super hacky and fragile
data = data.replace('9991', str(PORT))
if data:
if '400 Bad Request' in data:
print 'DONE!'
import sys; sys.exit(1)
out.send(data)
count = 0
if count == time_out_max:
break
def start_server(host='localhost', port=PORT, IPv6=False, timeout=60,
handler=ConnectionHandler):
if IPv6==True:
soc_type=socket.AF_INET6
else:
soc_type=socket.AF_INET
soc = socket.socket(soc_type)
soc.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
soc.bind((host, port))
print "Serving on %s:%d."%(host, port)#debug
soc.listen(0)
while 1:
thread.start_new_thread(handler, soc.accept()+(timeout,))
if __name__ == '__main__':
start_server()