Skip to content

Commit c7c8830

Browse files
committed
Add tests for NdrCString & NdrWString packing
1 parent c46f086 commit c7c8830

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/test_ndr.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,11 @@ class ComplexAlignementStructure(ndr.NdrStructure):
5858
NDR_PACK_TEST_CASE = [
5959
# Simple case
6060
(ndr.make_structure([ndr.NdrLong, ndr.NdrLong]), (2, 2), b"\x02\x00\x00\x00\x02\x00\x00\x00"),
61+
# String case, test packing works + \x00 is added if not present in string
62+
(ndr.NdrCString, "Hello", b"\x06\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00Hello\x00PP"),
63+
(ndr.NdrCString, "Hello\x00", b"\x06\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00Hello\x00PP"),
64+
(ndr.NdrWString, "Hello", b"\x06\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00H\x00e\x00l\x00l\x00o\x00\x00\x00"),
65+
(ndr.NdrWString, "Hello\x00", b"\x06\x00\x00\x00\x00\x00\x00\x00\x06\x00\x00\x00H\x00e\x00l\x00l\x00o\x00\x00\x00"),
6166
# Test GUID packing
6267
(ndr.NdrGuid, gdef.GUID.from_string("42424242-42424242-4242-4242-424242424242"), b"BBBBBBBBBBBBBBBB"),
6368
# Test CtxHandle packing

0 commit comments

Comments
 (0)