sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-11 05:18:09 +03:00
Code

python:tests/dns_base: let dns_transaction_udp() take allow_{remaining,truncated}=True

Browse Source 
BUG: https://bugzilla.samba.org/show_bug.cgi?id=13019

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
This commit is contained in:
Stefan Metzmacher 2024-05-29 14:14:11 +02:00 committed by Andrew Bartlett
parent 27d92fa808
commit 1b1e7e06cf
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
python/samba/tests/dns_base.py
View File

@ -130,6 +130,8 @@ class DNSTest(TestCaseInTempDir):
return self.creds.get_realm().lower() return self.creds.get_realm().lower()
def dns_transaction_udp(self, packet, host, def dns_transaction_udp(self, packet, host,
allow_remaining=False,
allow_truncated=False,
dump=False, timeout=None): dump=False, timeout=None):
"send a DNS query and read the reply" "send a DNS query and read the reply"
s = None s = None
@ -146,7 +148,17 @@ class DNSTest(TestCaseInTempDir):
recv_packet = s.recv(2048, 0) recv_packet = s.recv(2048, 0)
if dump: if dump:
print(self.hexdump(recv_packet)) print(self.hexdump(recv_packet))
response = ndr.ndr_unpack(dns.name_packet, recv_packet) if allow_truncated:
# with allow_remaining
# we add some zero bytes
# in order to also parse truncated
# responses
recv_packet_p = recv_packet + 32*b"\x00"
allow_remaining = True
else:
recv_packet_p = recv_packet
response = ndr.ndr_unpack(dns.name_packet, recv_packet_p,
allow_remaining=allow_remaining)
return (response, recv_packet) return (response, recv_packet)
finally: finally:
if s is not None: if s is not None: