python实现代理服务功能 实例

代理服务原理很简单就拿浏览器与web服务器来说。无非是A浏览器
发request给B代理B代理再把request把送给C web服务cookies是什么意思B代理再把request把送给C web服务然后C的reponse-B-A。
要写web代理服务就要先了解下http协议当然并不要多深入除非要实现强大的功能：修改XX信息、
负载均衡等。http请求由三部分组成：请求行、消息报头、请求正文；
详细的网上有想了解可以看看。下面是一个正常的GET请求头（Cookie部分本人没截屏使用的系统w7）：

可以看到首行：GET是请求方法 /是路径在后面是协议版本；第二行以后是请求报头都是键值对形式；
GET方法没有正文。post有正文除此之外请求方法头部基本一致每一行结尾都是\r\n。
基本的请求方法如下：

与第一张图片对比一下有什么不同……第一行的资源路径不对。当浏览器上设置代理请求时把整个url都作为资源路径了所以我们要把域名删掉然后代理服务器在把修改后的请求发送给目标
web服务器。就这么简单当然CONNECT方法特别要特别对待所以先说其他方法。
基本的思路：
1、代理服务器运行监听当有客户端浏览器请求到来时通过accept()获得client句柄（或者叫描述符）；
2、利用client描述符接收浏览器发来的request分离出第一行为了修改第一行和获得method
要去掉的的部分除去的部分用targetHost表示吧。
3、通过第2步能够获得方法method、request和targetHost这一步可以根据不同的method做不同的处理
由于GET、POET、PUT、DELETE等除了CONNECT处理基本一致所以处理首行比如：
代码如下:
GET HTTP/1.1
替换为
GET / HTTP/1.1

此时targetHost也就是红色的部分默认的请求80端口此时port为80;如果targetHost中有端口（比如：8081）
就要分理处端口此时port为8081。然后根据targetHost和port连接到目标服务器target了实现代码如下：
代码如下:
def getTargetInfo(self,host): #处理targetHost获得网址和端口作为返回值。
port=0
site=None
if ‘:’ in host:
tmp=host.split(‘:’)
site=tmp[0]
port=int(tmp[1])
else:
site=host
port=80
return site,port
def commonMethod(self,request): #处理除CONNECT以外的方法
tmp=self.targetHost.split(‘/’)
net=tmp[0]+’//’+tmp[2]
request=request.replace(net,”) #替换掉首行不必要的部分
targetAddr=self.getTargetInfo(tmp[2]) #调用上面的函数
try:
(fam,_,_,_,addr)=socket.getaddrinfo(targetAddr[0],targetAddr[1])[0]
except Exception as e:
print e
return
self.target=socket.socket(fam)
self.target.connect(addr) #连接到目标web服务

4、这一步就好办了根据第三步处理后的request就可以self.target.send(request)发送给web服务器了。
5、这一步web服务器的reponse反响通过代理服务直接转发给客户端就行了本人用了非阻塞select可以试试epoll。
基本步骤就是这样使用的方法函数可以改进比如主函数部分使用的多线程或者多进程怎样选择……
但是思路差不多都是这样啦。想测试的话chrome安装SwitchySharp插件设置一下代理端口8083；
firefox插件autoproxy。
对于connect的处理还在解决中（如果有博友帮助就更好了）所以现在这个代理程序不支持https协议。
代理服务可以获得http协议的所有信息想了解学习http利用代理服务器是个不错的方法。
下面附上代码
代码如下:
#-*- coding: UTF-8 -*-
import socket,select
import sys
import thread
from multiprocessing import Process
class Proxy:
def __init__(self,soc):
self.client,_=soc.accept()
self.target=None
self.request_url=None
self.BUFSIZE=4096
self.method=None
self.targetHost=None
def getClientRequest(self):
request=self.client.recv(self.BUFSIZE)
if not request:
return None
cn=request.find(‘\n’)
firstLine=request[:cn]
print firstLine[:len(firstLine)-9]
line=firstLine.split()
self.method=line[0]
self.targetHost=line[1]
return request
def commonMethod(self,request):
tmp=self.targetHost.split(‘/’)
net=tmp[0]+’//’+tmp[2]
request=request.replace(net,”)
targetAddr=self.getTargetInfo(tmp[2])
try:
(fam,_,_,_,addr)=socket.getaddrinfo(targetAddr[0],targetAddr[1])[0]
except Exception as e:
print e
return
self.target=socket.socket(fam)
self.target.connect(addr)
self.target.send(request)
self.nonblocking()
def connectMethod(self,request): #对于CONNECT处理可以添加在这里
pass
def run(self):
request=self.getClientRequest()
if request:
if self.method in [‘GET’,’POST’,’PUT’,”DELETE”,’HAVE’]:
self.commonMethod(request)
elif self.method==’CONNECT’:
self.connectMethod(request)
def nonblocking(self):
inputs=[self.client,self.target]
while True:
readable,writeable,errs=select.select(inputs,[],inputs,3)
if errs:
break
for soc in readable:
data=soc.recv(self.BUFSIZE)
if data:
if soc is self.client:
self.target.send(data)
elif soc is self.target:
self.client.send(data)
else:
break
self.client.close()
self.target.close()
def getTargetInfo(self,host):
port=0
site=None
if ‘:’ in host:
tmp=host.split(‘:’)
site=tmp[0]
port=int(tmp[1])
else:
site=host
port=80
return site,port
if __name__==’__main__’:
host = ‘127.0.0.1’
port = 8083
backlog = 5
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)
server.bind((host,port))
server.listen(5)
while True:
thread.start_new_thread(Proxy(server).run,())
# p=Process(target=Proxy(server).run, args=()) #多进程
# p.start()