sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-02-02 09:47:23 +03:00
Code

selftest: Add test for structure with NDR_BUFFERS only in a union

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

Signed-off-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Gary Lockyer <gary@catalyst.net.nz>
This commit is contained in:
Andrew Bartlett 2019-12-04 14:46:32 +13:00
parent 3f69c6b132
commit cf83eec565
2 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
python/samba/tests/blackbox/ndrdump.py
View File

@ -312,3 +312,41 @@ dump OK
# check_output will return bytes
# convert expected to bytes for python 3
self.assertEqual(actual, expected.encode('utf-8'))
# This is a good example of a union with an empty default
# and no buffers to parse.
def test_ndrdump_fuzzed_spoolss_EnumForms(self):
expected_head = b'''pull returned Success
WARNING! 2 unread bytes
[0000] 00 00 .. ''' b'''
spoolss_EnumForms: struct spoolss_EnumForms
out: struct spoolss_EnumForms
count : *
count : 0x00000100 (256)
info : *
info : *
info: ARRAY(256)
info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
'''
expected_tail = b'''info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
info : union spoolss_FormInfo(case 0)
needed : *
needed : 0x00000000 (0)
result : DOS code 0xa9a9a900
dump OK
'''
try:
actual = self.check_output(
"ndrdump spoolss spoolss_EnumForms out --base64-input " +\
"--input AAAAAQAAAAAAAAAAAAEAAACpqakAAA="
)
except BlackboxProcessError as e:
self.fail(e)
self.assertEqual(actual[:len(expected_head)],
expected_head)
self.assertTrue(actual.endswith(expected_tail))

1
selftest/knownfail.d/ndrdump-spoolss_EnumForms Normal file
View File

@ -0,0 +1 @@
^samba.tests.blackbox.ndrdump.samba.tests.blackbox.ndrdump.NdrDumpTests.test_ndrdump_fuzzed_spoolss_EnumForms