python中socket实现代理功能_python实现代理服务功能实例
代理服務(wù)原理很簡單，就拿瀏覽器與web服務(wù)器來說。無非是A瀏覽器
發(fā)request給B代理，B代理再把request把送給C web服務(wù)，然后C的reponse->B->A。
要寫web代理服務(wù)就要先了解下http協(xié)議，當然并不要多深入，除非要實現(xiàn)強大的功能：修改XX信息、
負載均衡等。http請求由三部分組成：請求行、消息報頭、請求正文；
詳細的網(wǎng)上有，想了解可以看看。下面是一個正常的GET請求頭(Cookie部分本人沒截屏，使用的系統(tǒng)w7)：
可以看到首行：GET是請求方法， /是路徑，在后面是協(xié)議版本；第二行以后是請求報頭，都是鍵值對形式；
GET方法沒有正文。post有正文，除此之外，請求方法頭部基本一致，每一行結(jié)尾都是\r\n。
基本的請求方法，如下：
GET??????? 請求獲取Request-URI所標識的資源
POST????? 在Request-URI所標識的資源后附加新的數(shù)據(jù)
HEAD????? 請求獲取由Request-URI所標識的資源的響應(yīng)消息報頭
PUT???????? 請求服務(wù)器存儲一個資源，并用Request-URI作為其標識
DELETE?? 請求服務(wù)器刪除Request-URI所標識的資源
TRACE???? 請求服務(wù)器回送收到的請求信息，主要用于測試或診斷
CONNECT 保留將來使用
OPTIONS? 請求查詢服務(wù)器的性能，或者查詢與資源相關(guān)的選項和需求
但是使用代理后，從代理服務(wù)上得到的請求如下：
與第一張圖片對比一下，有什么不同......第一行的資源路徑不對。當瀏覽器上設(shè)置代理請求時把整個url都作為資源路徑了，所以我們要把域名刪掉，然后代理服務(wù)器在把修改后的請求發(fā)送給目標
web服務(wù)器。就這么簡單，當然CONNECT方法特別，要特別對待，所以先說其他方法。
基本的思路：
1、代理服務(wù)器運行監(jiān)聽，當有客戶端瀏覽器請求到來時通過accept()獲得client句柄(或者叫描述符)；
2、利用client描述符接收瀏覽器發(fā)來的request，分離出第一行為了修改第一行和獲得method，
要去掉的的部分，除去http://的部分用targetHost表示吧。
3、通過第2步能夠獲得方法method、request和targetHost，這一步可以根據(jù)不同的method做不同的處理，
由于GET、POET、PUT、DELETE等除了CONNECT處理基本一致，所以處理首行，比如：
GET http://www.a.com/ HTTP/1.1
替換為
GET / HTTP/1.1
此時targetHost也就是紅色的部分，默認的請求80端口，此時port為80;如果targetHost中有端口(比如www.a.com：8081)，
就要分理處端口，此時port為8081。然后根據(jù)targetHost和port連接到目標服務(wù)器target了，實現(xiàn)代碼如下：
def getTargetInfo(self,host): #處理targetHost獲得網(wǎng)址和端口，作為返回值。
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]) #調(diào)用上面的函數(shù)
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服務(wù)
4、這一步就好辦了，根據(jù)第三步處理后的request就可以self.target.send(request)發(fā)送給web服務(wù)器了。
5、這一步web服務(wù)器的reponse反響通過代理服務(wù)直接轉(zhuǎn)發(fā)給客戶端就行了，本人用了非阻塞select，可以試試epoll。
基本步驟就是這樣，使用的方法函數(shù)可以改進，比如主函數(shù)部分使用的多線程或者多進程，怎樣選擇......
但是思路差不多都是這樣啦。想測試的話，chrome安裝SwitchySharp插件，設(shè)置一下，代理端口8083；
firefox插件autoproxy。
對于connect的處理還在解決中(如果有博友幫助就更好了)，所以現(xiàn)在這個代理程序不支持https協(xié)議。
代理服務(wù)可以獲得http協(xié)議的所有信息，想了解學(xué)習(xí)http，利用代理服務(wù)器是個不錯的方法。
下面附上代碼
#-*- 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()
總結(jié)
以上是生活随笔為你收集整理的python中socket实现代理功能_python实现代理服务功能实例的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： textarea 换行_textarea
- 下一篇： python的盈利模式_八大盈利模式是什