vbatts/shadowsocks
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

server can limit the host setting

Browse source
This commit is contained in:
破娃酱 2016-10-04 20:01:07 +08:00
parent 293a937501
commit 1183d567c9
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
shadowsocks/obfsplugin/http_simple.py
View file

@ -153,6 +153,14 @@ class http_simple(plain.plain):
return ret_buf return ret_buf
return b'' return b''
def get_host_from_http_header(self, buf):
ret_buf = b''
lines = buf.split(b'\r\n')
if lines and len(lines) > 4:
for line in lines:
if match_begin(line, b"Host: "):
return line[6:]
def not_match_return(self, buf): def not_match_return(self, buf):
self.has_sent_header = True self.has_sent_header = True
self.has_recv_header = True self.has_recv_header = True
@ -182,6 +190,14 @@ class http_simple(plain.plain):
if b'\r\n\r\n' in buf: if b'\r\n\r\n' in buf:
datas = buf.split(b'\r\n\r\n', 1) datas = buf.split(b'\r\n\r\n', 1)
ret_buf = self.get_data_from_http_header(buf) ret_buf = self.get_data_from_http_header(buf)
host = self.get_host_from_http_header(buf)
if host and self.server_info.obfs_param:
pos = host.find(":")
if pos >= 0:
host = host[:pos]
hosts = self.server_info.obfs_param.split(',')
if host not in hosts:
return self.not_match_return(buf)
if len(datas) > 1: if len(datas) > 1:
ret_buf += datas[1] ret_buf += datas[1]
if len(ret_buf) >= 7: if len(ret_buf) >= 7: