sendback an empty packet if auth success

This commit is contained in:
破娃酱 2016-09-30 10:12:24 +08:00
parent e7db6d4248
commit ada4eab628

View file

@ -544,6 +544,8 @@ class auth_sha1_v2(verify_base):
return (buf, False)
self.recv_buf += buf
out_buf = b''
sendback = False
if not self.has_recv_header:
if len(self.recv_buf) < 6:
return (b'', False)
@ -580,8 +582,8 @@ class auth_sha1_v2(verify_base):
return self.not_match_return(self.recv_buf)
self.recv_buf = self.recv_buf[length:]
self.has_recv_header = True
sendback = True
sendback = False
while len(self.recv_buf) > 2:
length = struct.unpack('>H', self.recv_buf[:2])[0]
if length >= 8192 or length < 7:
@ -965,6 +967,8 @@ class auth_sha1_v4(verify_base):
return (buf, False)
self.recv_buf += buf
out_buf = b''
sendback = False
if not self.has_recv_header:
if len(self.recv_buf) <= 6:
return (b'', False)
@ -1004,8 +1008,8 @@ class auth_sha1_v4(verify_base):
return self.not_match_return(self.recv_buf)
self.recv_buf = self.recv_buf[length:]
self.has_recv_header = True
sendback = True
sendback = False
while len(self.recv_buf) > 4:
crc = struct.pack('<H', binascii.crc32(self.recv_buf[:2]) & 0xFFFF)
if crc != self.recv_buf[2:4]:
@ -1201,6 +1205,8 @@ class auth_aes128(verify_base):
return (buf, False)
self.recv_buf += buf
out_buf = b''
sendback = False
if not self.has_recv_header:
if len(self.recv_buf) < 30:
return (b'', False)
@ -1239,8 +1245,8 @@ class auth_aes128(verify_base):
return self.not_match_return(self.recv_buf)
self.recv_buf = self.recv_buf[length:]
self.has_recv_header = True
sendback = True
sendback = False
while len(self.recv_buf) > 4:
crc = struct.pack('<H', binascii.crc32(self.recv_buf[:2]) & 0xFFFF)
if crc != self.recv_buf[2:4]: