1
0
mirror of https://github.com/samba-team/samba.git synced 2025-03-11 16:58:40 +03:00

RawDCERPCTest: add some more auth_length related asserts

BUG: https://bugzilla.samba.org/show_bug.cgi?id=14356

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>
(cherry picked from commit bb8ad1f22924b581bfb66555713e98efa91372b2)
This commit is contained in:
Stefan Metzmacher 2020-11-11 16:49:25 +01:00 committed by Jule Anger
parent fcbb5243d5
commit 465bcb6055

View File

@ -889,6 +889,7 @@ class RawDCERPCTest(TestCase):
if len(ai) > samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH:
p.auth_length = len(ai) - samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH
else:
self.assertEqual(len(ai), 0)
p.auth_length = 0
p.call_id = call_id
p.u = payload
@ -1001,10 +1002,10 @@ class RawDCERPCTest(TestCase):
if len(ai) > samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH:
self.assertEqual(p.auth_length,
len(ai) - samba.dcerpc.dcerpc.DCERPC_AUTH_TRAILER_LENGTH)
elif auth_length is not None:
self.assertEqual(p.auth_length, auth_length)
else:
elif auth_length is None:
self.assertEqual(p.auth_length, 0)
if auth_length is not None:
self.assertEqual(p.auth_length, auth_length)
self.assertEqual(p.call_id, call_id)
return