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

fix transfer upload size calculation. avoid flow attack

Browse source
This commit is contained in:
BreakWa11 2015-11-25 16:38:08 +08:00
parent 0504fb43cf
commit 57433bccb9
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
shadowsocks/tcprelay.py
View file

@ -275,7 +275,9 @@ class TCPRelayHandler(object):
if self._is_local: if self._is_local:
pass pass
else: else:
if sock == self._local_sock and self._encrypt_correct and (self._obfs is not None): if sock == self._remote_sock:
self._server.server_transfer_ul += len(data)
elif self._encrypt_correct and (self._obfs is not None):
obfs_encode = self._obfs.server_encode(data) obfs_encode = self._obfs.server_encode(data)
data = obfs_encode data = obfs_encode
if data: if data:
@ -614,7 +616,6 @@ class TCPRelayHandler(object):
return return
if not data: if not data:
return return
self._server.server_transfer_ul += len(data)
if self._stage == STAGE_STREAM: if self._stage == STAGE_STREAM:
if self._is_local: if self._is_local:
if self._encryptor is not None: if self._encryptor is not None: