python编写https代理服务_Http/Https代理Python实现

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

对Python感兴趣，工作之余，参照部分文档初步实现了HTTP/HTTPs代理。(代码比较凌乱，各位留情。

)

使用Fiddler调试。

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~import socket

import time

import threading

BUFSIZE=1024

class Access_to_Host(object):

def handler(self,conn,addr):

self.conn=conn

self.addr=addr

all_src_data,hostname,port,ssl_flag=self.get_dst_host_from_header(self.conn,self.addr)

all_dst_data=self.get_data_from_host(hostname,port,all_src_data,ssl_flag)

if all_dst_data and not ssl_flag:

#self.send_data_to_client(self.conn,all_dst_data)

self.ssl_client_server_client(self.conn,self.conn_dst,all_dst_data)

elif ssl_flag:

sample_data_to_client=b"HTTP/1.0 200 Connection Established\r\n\r\n"

# print("\nSSL_Flag-1")

#self.send_data_to_client(self.conn,all_dst_data)

#print("SSL_Flag-2")

self.ssl_client_server_client(self.conn,self.conn_dst,sample_data_to_client)

#print("\nSSL_Flag-3")

else:

print('pls check network. cannot get hostname:'+hostname)

#self.conn.close()

def ssl_client_server(self,src_conn,dst_conn):

self.src_conn=src_conn

self.dst_conn=dst_conn

while True:

###get data from client

try:

ssl_client_data=self.src_conn.recv(BUFSIZE)

except Exception as e:

print("client disconnct ")

print(e)

self.src_conn.close()

#self.dst_conn.close()

return False

if ssl_client_data:

#####send data to server

try:

self.dst_conn.sendall(ssl_client_data)

except Exception as e:

print("server disconnct Err")

self.dst_conn.close()

return False

else:

self.src_conn.close()

return False

def ssl_server_client(self,src_conn,dst_conn):

self.src_conn=src_conn

self.dst_conn=dst_conn

while True:

###get data from server

try:

ssl_server_data=self.dst_conn.recv(BUFSIZE)

except Exception as e:

print("server disconnct ")

self.dst_conn.close()

return False

if ssl_server_data:

#####send data to client

try:

self.src_conn.sendall(ssl_server_data)

except Exception as e:

print("Client disconnct Err")

self.src_conn.close()

return False

else:

self.dst_conn.close()

return False

def ssl_client_server_client(self,src_conn,dst_conn,all_dst_data):

self.src_conn=src_conn

self.dst_conn=dst_conn

try:

#print(all_dst_data)

self.src_conn.sendall(all_dst_data)

except Exception as e:

print(e)

print("cannot sent data(HTTP/1.0 200) to SSL client")

return False

threadlist=[]

t1=threading.Thread(target=self.ssl_client_server,args=(self.src_conn,self.dst_conn))

t2=threading.Thread(target=self.ssl_server_client,args=(self.src_conn,self.dst_conn))

threadlist.append(t1)

threadlist.append(t2)

for t in threadlist:

t.start()

#t.join()

######线程控制,等待线程结束后,远程主机关闭socket后，客户端到主机的socket也不需要再做任何操作了。

while not self.dst_conn._closed:

time.sleep(1)

self.src_conn.close()

def get_src_client(self):

self.src_ip=self.s_src.getpeername()

return self.src_ip

def send_data_to_client(self,conn_src,data):

self.conn_src=conn_src

try:

self.conn_src.sendall(data)

except Exception as e:

print(e)

print("cannot sent data to client")

return False

#self.conn_dst.close()

def get_dst_host_from_header(self,conn_sock,addr):

self.s_src=conn_sock

self.addr=addr

header=""

ssl_flag=False

while True:

#print("Loop Loop Loop")

header=self.s_src.recv(BUFSIZE)

if header :

#####header的一行含有CONNECT，即为SSL(HTTPS)

indexssl=header.split(b"\n")[0].find(b"CONNECT")

# print("indexsll:"+str(indexssl))

if indexssl>-1:

#####CONNECT===7 +8 前面一个空格

hostname=str(header.split(b"\n")[0].split(b":")[0].decode())

hostname=hostname[indexssl+8:]

port=443

ssl_flag=True

return header,hostname,port,ssl_flag

index1=header.find(b"Host:")

index2=header.find(b"GET http")

index3=header.find(b"POST http")

if index1>-1:

indexofn=header.find(b"\n",index1)

##host:===5

host=header[index1+5:indexofn]

elif index2>-1 or index3>-1:

###no host sample :'GET http://saxn.sina.com.cn/mfp/view?......

host=header.split(b"/")[2]

else:

print("src socket host:")

print(self.s_src.getpeername())

print("cannot find out host!!:"+repr(header))

return

break

host=str(host.decode().strip("\r").lstrip())

if len(host.split(":"))==2:

port=host.split(":")[1]

hostname=host.split(":")[0].strip("")

else:

port=80

hostname=host.split(":")[0].strip("")

ssl_flag=False

return header,hostname,int(port),ssl_flag

def get_data_from_host(self,host,port,sdata,ssl_flag):

self.conn_dst=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

all_dst_data=""

try:

self.conn_dst.connect((str(host),port))

except Exception as e:

print(e)

print("get_data_from_host: cannot get host:"+host)

self.conn_dst.close()

return False

#con_string="("+server+","+port+")"

############https只建立链接

try:

if ssl_flag:

return all_dst_data

else:

self.conn_dst.sendall(sdata)

except Exception as e:

print(e)

print("cannot send data to host:"+host)

self.conn_dst.close()

return False

# buffer=[]

rc_data=self.conn_dst.recv(BUFSIZE)

#####剩下的data交给线程去获取

return rc_data

class Server(object):

def Handle_Rec(conn_socket,addr):

print("This is Handler Fun")

pass

def __init__(self,host,port):

print("Server starting......")

self.host=host

self.port=port

self.s_s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)

self.s_s.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,1)

self.s_s.bind((host,port))

self.s_s.listen(20)

def start(self):

while True:

try:

conn,addr=self.s_s.accept()

threading.Thread(target=Access_to_Host().handler,args=(conn,addr)).start()

except Exception as e:

print(str(e))

print("\nExcept happend")

if __name__=="__main__":

svr=Server("127.0.0.1",8080)

svr.start()