2 # -*- coding: utf-8 -*-
5 sys
.path
.insert(0, 'bin/python')
9 import samba
.getopt
as options
16 from samba
.netcmd
.main
import cmd_sambatool
18 # We try to use the test infrastructure of Samba 4.3+, but if it
19 # doesn't work, we are probably in a back-ported patch and trying to
20 # run on 4.1 or something.
22 # Don't copy this horror into ordinary tests -- it is special for
23 # performance tests that want to apply to old versions.
25 from samba
.tests
.subunitrun
import SubunitOptions
, TestProgram
29 samba
.ensure_external_module("testtools", "testtools")
30 samba
.ensure_external_module("subunit", "subunit/python")
31 from subunit
.run
import SubunitTestRunner
34 from samba
.samdb
import SamDB
35 from samba
.auth
import system_session
37 from samba
.ndr
import ndr_pack
, ndr_unpack
38 from samba
.dcerpc
import security
39 from samba
.dcerpc
import drsuapi
41 parser
= optparse
.OptionParser("ndr_pack_performance.py [options] <host>")
42 sambaopts
= options
.SambaOptions(parser
)
43 parser
.add_option_group(sambaopts
)
44 parser
.add_option_group(options
.VersionOptions(parser
))
47 subunitopts
= SubunitOptions(parser
)
48 parser
.add_option_group(subunitopts
)
50 # use command line creds if available
51 credopts
= options
.CredentialsOptions(parser
)
52 parser
.add_option_group(credopts
)
53 opts
, args
= parser
.parse_args()
61 lp
= sambaopts
.get_loadparm()
62 creds
= credopts
.get_credentials(lp
)
67 BIG_SD_SDDL
= ''.join(
68 """O:S-1-5-21-3328325300-3937145445-4190589019-512G:S-1-5-2
69 1-3328325300-3937145445-4190589019-512D:AI(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;S-
70 1-5-21-3328325300-3937145445-4190589019-512)(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;
71 SY)(A;;RPLCLORC;;;AU)(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;AO)(A;;RPLCLORC;;;PS)(O
72 A;;CR;ab721a55-1e2f-11d0-9819-00aa0040529b;;AU)(OA;;RP;46a9b11d-60ae-405a-b7e
73 8-ff8a58d456d2;;S-1-5-32-560)(OA;CIIOID;RP;4c164200-20c0-11d0-a768-00aa006e05
74 29;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;4c164200-20c0-11d0-a
75 768-00aa006e0529;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;5f2020
76 10-79a5-11d0-9020-00c04fc2d4cf;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CI
77 IOID;RP;5f202010-79a5-11d0-9020-00c04fc2d4cf;bf967aba-0de6-11d0-a285-00aa0030
78 49e2;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c04fc2d4cf;4828cc14-1437-45bc
79 -9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c04fc2d4cf;bf96
80 7aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;59ba2f42-79a2-11d0-9020-00c
81 04fc2d3cf;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;59ba2f42-79a2
82 -11d0-9020-00c04fc2d3cf;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP
83 ;037088f8-0ae1-11d2-b422-00a0c968f939;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU
84 )(OA;CIIOID;RP;037088f8-0ae1-11d2-b422-00a0c968f939;bf967aba-0de6-11d0-a285-0
85 0aa003049e2;RU)(OA;CIIOID;RP;b7c69e6d-2cc7-11d2-854e-00a0c983f608;bf967a86-0d
86 e6-11d0-a285-00aa003049e2;ED)(OA;CIID;RP;b7c69e6d-2cc7-11d2-854e-00a0c983f608
87 ;bf967a9c-0de6-11d0-a285-00aa003049e2;ED)(OA;CIIOID;RP;b7c69e6d-2cc7-11d2-854
88 e-00a0c983f608;bf967aba-0de6-11d0-a285-00aa003049e2;ED)(OA;CIIOID;RPLCLORC;;4
89 828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIID;RPLCLORC;;bf967a9c-0de6-11d0-
90 a285-00aa003049e2;RU)(OA;CIIOID;RPLCLORC;;bf967aba-0de6-11d0-a285-00aa003049e
91 2;RU)(OA;CIID;RPWPCR;91e647de-d96f-4b70-9557-d63ff4f3ccd8;;PS)(A;CIID;RPWPCRC
92 CDCLCLORCWOWDSDDTSW;;;S-1-5-21-3328325300-3937145445-4190589019-519)(A;CIID;L
93 C;;;RU)(A;CIID;RPWPCRCCLCLORCWOWDSDSW;;;BA)(OA;CIIOID;RP;4c164200-20c0-11d0-a
94 768-00aa006e0529;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;4c1642
95 00-20c0-11d0-a768-00aa006e0529;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CI
96 IOID;RP;5f202010-79a5-11d0-9020-00c04fc2d4cf;4828cc14-1437-45bc-9b07-ad6f015e
97 5f28;RU)(OA;CIIOID;RP;5f202010-79a5-11d0-9020-00c04fc2d4cf;bf967aba-0de6-11d0
98 -a285-00aa003049e2;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c04fc2d4cf;4828
99 cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c
100 04fc2d4cf;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;59ba2f42-79a2
101 -11d0-9020-00c04fc2d3cf;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP
102 ;59ba2f42-79a2-11d0-9020-00c04fc2d3cf;bf967aba-0de6-11d0-a285-00aa003049e2;RU
103 )(OA;CIIOID;RP;037088f8-0ae1-11d2-b422-00a0c968f939;4828cc14-1437-45bc-9b07-a
104 d6f015e5f28;RU)(OA;CIIOID;RP;037088f8-0ae1-11d2-b422-00a0c968f939;bf967aba-0d
105 e6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;b7c69e6d-2cc7-11d2-854e-00a0c983f6
106 08;bf967a86-0de6-11d0-a285-00aa003049e2;ED)(OA;CIID;RP;b7c69e6d-2cc7-11d2-854
107 e-00a0c983f608;bf967a9c-0de6-11d0-a285-00aa003049e2;ED)(OA;CIIOID;RP;b7c69e6d
108 -2cc7-11d2-854e-00a0c983f608;bf967aba-0de6-11d0-a285-00aa003049e2;ED)(OA;CIIO
109 ID;RPLCLORC;;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIID;RPLCLORC;;bf967
110 a9c-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RPLCLORC;;bf967aba-0de6-11d0-a2
111 85-00aa003049e2;RU)(OA;CIID;RPWPCR;91e647de-d96f-4b70-9557-d63ff4f3ccd8;;PS)(
112 A;CIID;RPWPCRCCDCLCLORCWOWDSDDTSW;;;S-1-5-21-3328325300-3937145445-4190589019
113 -519)(A;CIID;LC;;;RU)(A;CIID;RPWPCRCCLCLORCWOWDSDSW;;;BA)S:AI(OU;CIIOIDSA;WP;
114 f30e3bbe-9ff0-11d1-b603-0000f80367c1;bf967aa5-0de6-11d0-a285-00aa003049e2;WD)
115 (OU;CIIOIDSA;WP;f30e3bbf-9ff0-11d1-b603-0000f80367c1;bf967aa5-0de6-11d0-a285-
116 00aa003049e2;WD)(OU;CIIOIDSA;WP;f30e3bbe-9ff0-11d1-b603-0000f80367c1;bf967aa5
117 -0de6-11d0-a285-00aa003049e2;WD)(OU;CIIOIDSA;WP;f30e3bbf-9ff0-11d1-b603-0000f
118 80367c1;bf967aa5-0de6-11d0-a285-00aa003049e2;WD)""".split())
120 LITTLE_SD_SDDL
= ''.join(
121 """O:S-1-5-21-3328325300-3937145445-4190589019-512G:S-1-5-2
122 1-3328325300-3937145445-4190589019-512D:AI(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;S-
123 1-5-21-3328325300-3937145445-4190589019-512)(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;
124 SY)(A;;RPLCLORC;;;AU)(A;;RPWPCRCCDCLCLORCWOWDSDDTSW;;;AO)(A;;RPLCLORC;;;PS)(O
125 A;;CR;ab721a55-1e2f-11d0-9819-00aa0040529b;;AU)(OA;;RP;46a9b11d-60ae-405a-b7e
126 8-ff8a58d456d2;;S-1-5-32-560)(OA;CIIOID;RP;4c164200-20c0-11d0-a768-00aa006e05
127 29;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;4c164200-20c0-11d0-a
128 768-00aa006e0529;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;5f2020
129 10-79a5-11d0-9020-00c04fc2d4cf;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CI
130 IOID;RP;5f202010-79a5-11d0-9020-00c04fc2d4cf;bf967aba-0de6-11d0-a285-00aa0030
131 49e2;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c04fc2d4cf;4828cc14-1437-45bc
132 -9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;bc0ac240-79a9-11d0-9020-00c04fc2d4cf;bf96
133 7aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP;59ba2f42-79a2-11d0-9020-00c
134 04fc2d3cf;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU)(OA;CIIOID;RP;59ba2f42-79a2
135 -11d0-9020-00c04fc2d3cf;bf967aba-0de6-11d0-a285-00aa003049e2;RU)(OA;CIIOID;RP
136 ;037088f8-0ae1-11d2-b422-00a0c968f939;4828cc14-1437-45bc-9b07-ad6f015e5f28;RU
137 )(OA;CIIOID;RP;037088f8-0ae1-11d2-b422-00a0c968f939;bf967aba-0de6-11d0-a285-0
138 0aa003049e2;RU)(OA;CIIOID;RP;b7c69e6d-2cc7-11d2-854e-00a0c983f608;bf967a86-0d
139 e6-11d0-a285-00aa003049e2;ED)""".split())
141 # set SCALE = 100 for normal test, or 1 for testing the test.
144 class UserTests(samba
.tests
.TestCase
):
146 def get_file_blob(self
, filename
):
147 if filename
.endswith('.gz'):
148 f
= gzip
.open(filename
)
153 def get_desc(self
, sddl
):
154 dummy_sid
= security
.dom_sid("S-2-0-0")
155 return security
.descriptor
.from_sddl(sddl
, dummy_sid
)
157 def get_blob(self
, sddl
):
158 return ndr_pack(self
.get_desc(sddl
))
160 def test_00_00_do_nothing(self
):
161 # this gives us an idea of the overhead
164 def _test_pack(self
, unpacked
, cycles
=10000):
165 for i
in range(SCALE
* cycles
):
168 def _test_unpack(self
, blob
, cycles
=10000, cls
=security
.descriptor
):
169 for i
in range(SCALE
* cycles
):
170 ndr_unpack(cls
, blob
)
172 def _test_pack_unpack(self
, desc
, cycles
=5000, cls
=security
.descriptor
):
173 blob2
= ndr_pack(desc
)
175 for i
in range(SCALE
* cycles
):
176 blob
= ndr_pack(desc
)
177 desc
= ndr_unpack(cls
, blob
)
179 self
.assertEqual(blob
, blob2
)
181 def test_pack_big_sd(self
):
182 unpacked
= self
.get_desc(BIG_SD_SDDL
)
183 self
._test
_pack
(unpacked
)
185 def test_unpack_big_sd(self
):
186 blob
= self
.get_blob(BIG_SD_SDDL
)
187 self
._test
_unpack
(blob
)
189 def test_pack_unpack_big_sd(self
):
190 unpacked
= self
.get_desc(BIG_SD_SDDL
)
191 self
._test
_pack
_unpack
(unpacked
)
193 def test_pack_little_sd(self
):
194 unpacked
= self
.get_desc(LITTLE_SD_SDDL
)
195 self
._test
_pack
(unpacked
)
197 def test_unpack_little_sd(self
):
198 blob
= self
.get_blob(LITTLE_SD_SDDL
)
199 self
._test
_unpack
(blob
)
201 def test_pack_unpack_little_sd(self
):
202 unpacked
= self
.get_desc(LITTLE_SD_SDDL
)
203 self
._test
_pack
_unpack
(unpacked
)
205 def test_unpack_repl_sample(self
):
206 blob
= self
.get_file_blob('testdata/replication-ndrpack-example.gz')
207 self
._test
_unpack
(blob
, cycles
=20, cls
=drsuapi
.DsGetNCChangesCtr6
)
209 def test_pack_repl_sample(self
):
210 blob
= self
.get_file_blob('testdata/replication-ndrpack-example.gz')
211 desc
= ndr_unpack(drsuapi
.DsGetNCChangesCtr6
, blob
)
212 self
._test
_pack
(desc
, cycles
=20)
214 if "://" not in host
:
215 if os
.path
.isfile(host
):
216 host
= "tdb://%s" % host
218 host
= "ldap://%s" % host
222 runner
= SubunitTestRunner()
223 if not runner
.run(unittest
.makeSuite(UserTests
)).wasSuccessful():
227 TestProgram(module
=__name__
, opts
=subunitopts
)