?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
s4/drs(tort): Unit test for prefixMap implementation.
[Samba/fernandojvsilva.git] / source4 / torture / drs / internal / prefixmap_tests.c
blob54bd6612f75b50dc04d3a63f920ba3ad8a64acff
1 /*
2 Unix SMB/CIFS implementation.
3
4 DRSUAPI prefixMap unit tests
5
6 Copyright (C) Kamen Mazdrashki <kamen.mazdrashki@postpath.com> 2009
7
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
12
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
17
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "torture/smbtorture.h"
24 #include "torture/rpc/drsuapi.h"
25 #include "dsdb/samdb/samdb.h"
26
27
28 /**
29 * Private data to be shared among all test in Test case
30 */
31 struct drsut_prefixmap_data {
32 struct dsdb_schema_prefixmap *pfm_new;
33 struct dsdb_schema_prefixmap *pfm_full;
34 };
35
36 /**
37 * Test-oid data structure
38 */
39 struct drsut_pfm_oid_data {
40 uint32_t id;
41 const char *bin_oid;
42 const char *oid_prefix;
43 };
44
45 /**
46 * Default prefixMap initialization data
47 */
48 static const struct drsut_pfm_oid_data _prefixmap_test_new_data[] = {
49 {.id=0x00000000, .bin_oid="5504", .oid_prefix="2.5.4"},
50 {.id=0x00000001, .bin_oid="5506", .oid_prefix="2.5.6"},
51 {.id=0x00000002, .bin_oid="2A864886F7140102", .oid_prefix="1.2.840.113556.1.2"},
52 {.id=0x00000003, .bin_oid="2A864886F7140103", .oid_prefix="1.2.840.113556.1.3"},
53 {.id=0x00000004, .bin_oid="6086480165020201", .oid_prefix="2.16.840.1.101.2.2.1"},
54 {.id=0x00000005, .bin_oid="6086480165020203", .oid_prefix="2.16.840.1.101.2.2.3"},
55 {.id=0x00000006, .bin_oid="6086480165020105", .oid_prefix="2.16.840.1.101.2.1.5"},
56 {.id=0x00000007, .bin_oid="6086480165020104", .oid_prefix="2.16.840.1.101.2.1.4"},
57 {.id=0x00000008, .bin_oid="5505", .oid_prefix="2.5.5"},
58 {.id=0x00000009, .bin_oid="2A864886F7140104", .oid_prefix="1.2.840.113556.1.4"},
59 {.id=0x0000000A, .bin_oid="2A864886F7140105", .oid_prefix="1.2.840.113556.1.5"},
60 {.id=0x00000013, .bin_oid="0992268993F22C64", .oid_prefix="0.9.2342.19200300.100"},
61 {.id=0x00000014, .bin_oid="6086480186F84203", .oid_prefix="2.16.840.1.113730.3"},
62 {.id=0x00000015, .bin_oid="0992268993F22C6401", .oid_prefix="0.9.2342.19200300.100.1"},
63 {.id=0x00000016, .bin_oid="6086480186F8420301", .oid_prefix="2.16.840.1.113730.3.1"},
64 {.id=0x00000017, .bin_oid="2A864886F7140105B658", .oid_prefix="1.2.840.113556.1.5.7000"},
65 {.id=0x00000018, .bin_oid="5515", .oid_prefix="2.5.21"},
66 {.id=0x00000019, .bin_oid="5512", .oid_prefix="2.5.18"},
67 {.id=0x0000001A, .bin_oid="5514", .oid_prefix="2.5.20"},
68 };
69
70 /**
71 * Data to be used for creating full prefix map for testing
72 */
73 static const struct drsut_pfm_oid_data _prefixmap_full_map_data[] = {
74 {.id=0x00000000, .bin_oid="0x5504", .oid_prefix="2.5.4"},
75 {.id=0x00000001, .bin_oid="0x5506", .oid_prefix="2.5.6"},
76 {.id=0x00000002, .bin_oid="0x2A864886F7140102", .oid_prefix="1.2.840.113556.1.2"},
77 {.id=0x00000003, .bin_oid="0x2A864886F7140103", .oid_prefix="1.2.840.113556.1.3"},
78 {.id=0x00000004, .bin_oid="0x6086480165020201", .oid_prefix="2.16.840.1.101.2.2.1"},
79 {.id=0x00000005, .bin_oid="0x6086480165020203", .oid_prefix="2.16.840.1.101.2.2.3"},
80 {.id=0x00000006, .bin_oid="0x6086480165020105", .oid_prefix="2.16.840.1.101.2.1.5"},
81 {.id=0x00000007, .bin_oid="0x6086480165020104", .oid_prefix="2.16.840.1.101.2.1.4"},
82 {.id=0x00000008, .bin_oid="0x5505", .oid_prefix="2.5.5"},
83 {.id=0x00000009, .bin_oid="0x2A864886F7140104", .oid_prefix="1.2.840.113556.1.4"},
84 {.id=0x0000000a, .bin_oid="0x2A864886F7140105", .oid_prefix="1.2.840.113556.1.5"},
85 {.id=0x00000013, .bin_oid="0x0992268993F22C64", .oid_prefix="0.9.2342.19200300.100"},
86 {.id=0x00000014, .bin_oid="0x6086480186F84203", .oid_prefix="2.16.840.1.113730.3"},
87 {.id=0x00000015, .bin_oid="0x0992268993F22C6401", .oid_prefix="0.9.2342.19200300.100.1"},
88 {.id=0x00000016, .bin_oid="0x6086480186F8420301", .oid_prefix="2.16.840.1.113730.3.1"},
89 {.id=0x00000017, .bin_oid="0x2A864886F7140105B658", .oid_prefix="1.2.840.113556.1.5.7000"},
90 {.id=0x00000018, .bin_oid="0x5515", .oid_prefix="2.5.21"},
91 {.id=0x00000019, .bin_oid="0x5512", .oid_prefix="2.5.18"},
92 {.id=0x0000001a, .bin_oid="0x5514", .oid_prefix="2.5.20"},
93 {.id=0x0000000b, .bin_oid="0x2A864886F71401048204", .oid_prefix="1.2.840.113556.1.4.260"},
94 {.id=0x0000000c, .bin_oid="0x2A864886F714010538", .oid_prefix="1.2.840.113556.1.5.56"},
95 {.id=0x0000000d, .bin_oid="0x2A864886F71401048206", .oid_prefix="1.2.840.113556.1.4.262"},
96 {.id=0x0000000e, .bin_oid="0x2A864886F714010539", .oid_prefix="1.2.840.113556.1.5.57"},
97 {.id=0x0000000f, .bin_oid="0x2A864886F71401048207", .oid_prefix="1.2.840.113556.1.4.263"},
98 {.id=0x00000010, .bin_oid="0x2A864886F71401053A", .oid_prefix="1.2.840.113556.1.5.58"},
99 {.id=0x00000011, .bin_oid="0x2A864886F714010549", .oid_prefix="1.2.840.113556.1.5.73"},
100 {.id=0x00000012, .bin_oid="0x2A864886F71401048231", .oid_prefix="1.2.840.113556.1.4.305"},
101 {.id=0x0000001b, .bin_oid="0x2B060104018B3A6577", .oid_prefix="1.3.6.1.4.1.1466.101.119"},
102 {.id=0x0000001c, .bin_oid="0x6086480186F8420302", .oid_prefix="2.16.840.1.113730.3.2"},
103 {.id=0x0000001d, .bin_oid="0x2B06010401817A01", .oid_prefix="1.3.6.1.4.1.250.1"},
104 {.id=0x0000001e, .bin_oid="0x2A864886F70D0109", .oid_prefix="1.2.840.113549.1.9"},
105 {.id=0x0000001f, .bin_oid="0x0992268993F22C6404", .oid_prefix="0.9.2342.19200300.100.4"},
106 {.id=0x00000020, .bin_oid="0x2A864886F714010617", .oid_prefix="1.2.840.113556.1.6.23"},
107 {.id=0x00000021, .bin_oid="0x2A864886F71401061201", .oid_prefix="1.2.840.113556.1.6.18.1"},
108 {.id=0x00000022, .bin_oid="0x2A864886F71401061202", .oid_prefix="1.2.840.113556.1.6.18.2"},
109 {.id=0x00000023, .bin_oid="0x2A864886F71401060D03", .oid_prefix="1.2.840.113556.1.6.13.3"},
110 {.id=0x00000024, .bin_oid="0x2A864886F71401060D04", .oid_prefix="1.2.840.113556.1.6.13.4"},
111 {.id=0x00000025, .bin_oid="0x2B0601010101", .oid_prefix="1.3.6.1.1.1.1"},
112 {.id=0x00000026, .bin_oid="0x2B0601010102", .oid_prefix="1.3.6.1.1.1.2"},
113 {.id=0x000003ed, .bin_oid="0x2A864886F7140104B65866", .oid_prefix="1.2.840.113556.1.4.7000.102"},
114 {.id=0x00000428, .bin_oid="0x2A864886F7140105B6583E", .oid_prefix="1.2.840.113556.1.5.7000.62"},
115 {.id=0x0000044c, .bin_oid="0x2A864886F7140104B6586683", .oid_prefix="1.2.840.113556.1.4.7000.102:0x83"},
116 {.id=0x0000044f, .bin_oid="0x2A864886F7140104B6586681", .oid_prefix="1.2.840.113556.1.4.7000.102:0x81"},
117 {.id=0x0000047d, .bin_oid="0x2A864886F7140105B6583E81", .oid_prefix="1.2.840.113556.1.5.7000.62:0x81"},
118 {.id=0x00000561, .bin_oid="0x2A864886F7140105B6583E83", .oid_prefix="1.2.840.113556.1.5.7000.62:0x83"},
119 {.id=0x000007d1, .bin_oid="0x2A864886F71401061401", .oid_prefix="1.2.840.113556.1.6.20.1"},
120 {.id=0x000007e1, .bin_oid="0x2A864886F71401061402", .oid_prefix="1.2.840.113556.1.6.20.2"},
121 {.id=0x00001b86, .bin_oid="0x2A817A", .oid_prefix="1.2.250"},
122 {.id=0x00001c78, .bin_oid="0x2A817A81", .oid_prefix="1.2.250:0x81"},
123 {.id=0x00001c7b, .bin_oid="0x2A817A8180", .oid_prefix="1.2.250:0x8180"},
124 };
125
126
127 /**
128 * OID-to-ATTID mappings to be used for testing
129 */
130 static const struct {
131 const char *oid;
132 uint32_t id;
133 uint32_t attid;
134 } _prefixmap_test_data[] = {
135 {.oid="2.5.4.0", .id=0x00000000, .attid=0x000000},
136 {.oid="2.5.4.42", .id=0x00000000, .attid=0x00002a},
137 {.oid="1.2.840.113556.1.2.1", .id=0x00000002, .attid=0x020001},
138 {.oid="1.2.840.113556.1.2.13", .id=0x00000002, .attid=0x02000d},
139 {.oid="1.2.840.113556.1.2.281", .id=0x00000002, .attid=0x020119},
140 {.oid="1.2.840.113556.1.4.125", .id=0x00000009, .attid=0x09007d},
141 {.oid="1.2.840.113556.1.4.146", .id=0x00000009, .attid=0x090092},
142 {.oid="1.2.250.1", .id=0x00001b86, .attid=0x1b860001},
143 {.oid="1.2.250.130", .id=0x00001b86, .attid=0x1b860082},
144 {.oid="1.2.250.16386", .id=0x00001c78, .attid=0x1c788002},
145 {.oid="1.2.250.2097154", .id=0x00001c7b, .attid=0x1c7b8002},
146 };
147
148
149 /**
150 * Creates dsdb_schema_prefixmap based on predefined data
151 *
152 */
153 static WERROR _drsut_prefixmap_new(const struct drsut_pfm_oid_data *_pfm_init_data, uint32_t count,
154 TALLOC_CTX *mem_ctx, struct dsdb_schema_prefixmap **_pfm)
155 {
156 uint32_t i;
157 struct dsdb_schema_prefixmap *pfm;
158
159 pfm = talloc(mem_ctx, struct dsdb_schema_prefixmap);
160 W_ERROR_HAVE_NO_MEMORY(pfm);
161
162 pfm->length = count;
163 pfm->prefixes = talloc_array(pfm, struct dsdb_schema_prefixmap_oid, pfm->length);
164 if (!pfm->prefixes) {
165 talloc_free(pfm);
166 return WERR_NOMEM;
167 }
168
169 for (i = 0; i < pfm->length; i++) {
170 pfm->prefixes[i].id = _pfm_init_data[i].id;
171 pfm->prefixes[i].bin_oid = strhex_to_data_blob(pfm, _pfm_init_data[i].bin_oid);
172 if (!pfm->prefixes[i].bin_oid.data) {
173 talloc_free(pfm);
174 return WERR_NOMEM;
175 }
176 }
177
178 *_pfm = pfm;
179
180 return WERR_OK;
181 }
182
183
184 /*
185 * Tests dsdb_schema_pfm_new()
186 */
187 static bool torture_drs_unit_pfm_new(struct torture_context *tctx, struct drsut_prefixmap_data *priv)
188 {
189 int i;
190 WERROR werr;
191 TALLOC_CTX *mem_ctx;
192 struct dsdb_schema_prefixmap *pfm = NULL;
193
194 mem_ctx = talloc_new(priv);
195
196 /* create new prefix map */
197 werr = dsdb_schema_pfm_new(mem_ctx, &pfm);
198 torture_assert_werr_ok(tctx, werr, "dsdb_schema_pfm_new() failed!");
199 torture_assert(tctx, pfm != NULL, "NULL prefixMap created!");
200 torture_assert(tctx, pfm->length > 0, "Empty prefixMap created!");
201 torture_assert(tctx, pfm->prefixes != NULL, "No prefixes for newly created prefixMap!");
202
203 /* compare newly created prefixMap with template one */
204 torture_assert(tctx, priv->pfm_new->length == pfm->length,
205 "Newly created prefixMap differs in length with expected one");
206 for (i = 0; i < pfm->length; i++) {
207 struct dsdb_schema_prefixmap_oid *pfm_oid = &pfm->prefixes[i];
208 struct dsdb_schema_prefixmap_oid *pfm_oid_exp = &priv->pfm_new->prefixes[i];
209
210 torture_assert(tctx, pfm_oid->id == pfm_oid_exp->id,
211 talloc_asprintf(mem_ctx, "Different IDs for index=%d", i));
212 torture_assert_data_blob_equal(tctx, pfm_oid->bin_oid, pfm_oid_exp->bin_oid,
213 talloc_asprintf(mem_ctx, "Different bin_oid for index=%d", i));
214 }
215
216 talloc_free(mem_ctx);
217
218 return true;
219 }
220
221 /**
222 * Tests dsdb_schema_pfm_make_attid() using full prefixMap.
223 * In this test we know exactly which ATTID and prefixMap->ID
224 * should be returned, i.e. no prefixMap entries should be added.
225 */
226 static bool torture_drs_unit_pfm_make_attid_full_map(struct torture_context *tctx, struct drsut_prefixmap_data *priv)
227 {
228 WERROR werr;
229 uint32_t i, count;
230 uint32_t attid;
231 char *err_msg;
232
233 count = ARRAY_SIZE(_prefixmap_test_data);
234 for (i = 0; i < count; i++) {
235 werr = dsdb_schema_pfm_make_attid(priv->pfm_full, _prefixmap_test_data[i].oid, &attid);
236 /* prepare error message */
237 err_msg = talloc_asprintf(priv, "dsdb_schema_pfm_make_attid() failed with %s",
238 _prefixmap_test_data[i].oid);
239 torture_assert(tctx, err_msg, "Unexpected: Have no memory!");
240 /* verify result and returned ATTID */
241 torture_assert_werr_ok(tctx, werr, err_msg);
242 torture_assert_int_equal(tctx, attid, _prefixmap_test_data[i].attid, err_msg);
243 /* reclaim memory for prepared error message */
244 talloc_free(err_msg);
245 }
246
247 return true;
248 }
249
250 /**
251 * Tests dsdb_schema_pfm_make_attid() using initially small prefixMap.
252 * In this test we don't know exactly which ATTID and prefixMap->ID
253 * should be returned, but we can verify lo-word of ATTID.
254 * This test verifies implementation branch when a new
255 * prefix should be added into prefixMap.
256 */
257 static bool torture_drs_unit_pfm_make_attid_small_map(struct torture_context *tctx, struct drsut_prefixmap_data *priv)
258 {
259 WERROR werr;
260 uint32_t i, j;
261 uint32_t idx;
262 uint32_t attid, attid_2;
263 char *err_msg;
264 struct dsdb_schema_prefixmap *pfm = NULL;
265 TALLOC_CTX *mem_ctx;
266
267 mem_ctx = talloc_new(priv);
268
269 /* create new prefix map */
270 werr = dsdb_schema_pfm_new(mem_ctx, &pfm);
271 torture_assert_werr_ok(tctx, werr, "dsdb_schema_pfm_new() failed!");
272 torture_assert(tctx, pfm != NULL, "NULL prefixMap created!");
273 torture_assert(tctx, pfm->length > 0, "Empty prefixMap created!");
274 torture_assert(tctx, pfm->prefixes != NULL, "No prefixes for newly created prefixMap!");
275
276 /* make some ATTIDs and check result */
277 for (i = 0; i < ARRAY_SIZE(_prefixmap_test_data); i++) {
278 werr = dsdb_schema_pfm_make_attid(pfm, _prefixmap_test_data[i].oid, &attid);
279
280 /* prepare error message */
281 err_msg = talloc_asprintf(mem_ctx, "dsdb_schema_pfm_make_attid() failed with %s",
282 _prefixmap_test_data[i].oid);
283 torture_assert(tctx, err_msg, "Unexpected: Have no memory!");
284
285 /* verify result and returned ATTID */
286 torture_assert_werr_ok(tctx, werr, err_msg);
287 /* verify ATTID lo-word */
288 torture_assert_int_equal(tctx, attid & 0xFFFF, _prefixmap_test_data[i].attid & 0xFFFF, err_msg);
289
290 /* try again, this time verify for whole ATTID */
291 werr = dsdb_schema_pfm_make_attid(pfm, _prefixmap_test_data[i].oid, &attid_2);
292 torture_assert_werr_ok(tctx, werr, err_msg);
293 torture_assert_int_equal(tctx, attid_2, attid, err_msg);
294
295 /* reclaim memory for prepared error message */
296 talloc_free(err_msg);
297
298 /* check there is such an index in modified prefixMap */
299 idx = (attid >> 16);
300 for (j = 0; j < pfm->length; j++) {
301 if (pfm->prefixes[j].id == idx)
302 break;
303 }
304 if (j >= pfm->length) {
305 torture_result(tctx, TORTURE_FAIL, __location__": No prefix for ATTID=0x%08X", attid);
306 return false;
307 }
308
309 }
310
311 talloc_free(mem_ctx);
312
313 return true;
314 }
315
316 /**
317 * Tests dsdb_schema_pfm_oid_from_attid() using full prefixMap.
318 */
319 static bool torture_drs_unit_pfm_oid_from_attid(struct torture_context *tctx, struct drsut_prefixmap_data *priv)
320 {
321 WERROR werr;
322 uint32_t i, count;
323 char *err_msg;
324 const char *oid;
325
326 count = ARRAY_SIZE(_prefixmap_test_data);
327 for (i = 0; i < count; i++) {
328 oid = NULL;
329 werr = dsdb_schema_pfm_oid_from_attid(priv->pfm_full, _prefixmap_test_data[i].attid,
330 priv, &oid);
331 /* prepare error message */
332 err_msg = talloc_asprintf(priv, "dsdb_schema_pfm_oid_from_attid() failed with 0x%08X",
333 _prefixmap_test_data[i].attid);
334 torture_assert(tctx, err_msg, "Unexpected: Have no memory!");
335 /* verify result and returned ATTID */
336 torture_assert_werr_ok(tctx, werr, err_msg);
337 torture_assert(tctx, oid, "dsdb_schema_pfm_oid_from_attid() returned NULL OID!!!");
338 torture_assert_str_equal(tctx, oid, _prefixmap_test_data[i].oid, err_msg);
339 /* reclaim memory for prepared error message */
340 talloc_free(err_msg);
341 /* free memory for OID */
342 talloc_free(discard_const(oid));
343 }
344
345 return true;
346 }
347
348
349 /*
350 * Setup/Teardown for test case
351 */
352 static bool torture_drs_unit_prefixmap_setup(struct torture_context *tctx, struct drsut_prefixmap_data **priv)
353 {
354 WERROR werr;
355
356 *priv = talloc_zero(tctx, struct drsut_prefixmap_data);
357 torture_assert(tctx, *priv != NULL, "Not enough memory");
358
359 werr = _drsut_prefixmap_new(_prefixmap_test_new_data, ARRAY_SIZE(_prefixmap_test_new_data), tctx, &(*priv)->pfm_new);
360 torture_assert_werr_ok(tctx, werr, "failed to create pfm_new");
361
362 werr = _drsut_prefixmap_new(_prefixmap_full_map_data, ARRAY_SIZE(_prefixmap_full_map_data), tctx, &(*priv)->pfm_full);
363 torture_assert_werr_ok(tctx, werr, "failed to create pfm_test");
364
365 return true;
366 }
367
368 static bool torture_drs_unit_prefixmap_teardown(struct torture_context *tctx, struct drsut_prefixmap_data *priv)
369 {
370 talloc_free(priv);
371
372 return true;
373 }
374
375 /**
376 * Test case initialization for
377 * DRS-UNIT.prefixMap
378 */
379 struct torture_tcase * torture_drs_unit_prefixmap(struct torture_suite *suite)
380 {
381 typedef bool (*pfn_setup)(struct torture_context *, void **);
382 typedef bool (*pfn_teardown)(struct torture_context *, void *);
383 typedef bool (*pfn_run)(struct torture_context *, void *);
384
385 struct torture_tcase * tc = torture_suite_add_tcase(suite, "prefixMap");
386
387 torture_tcase_set_fixture(tc,
388 (pfn_setup)torture_drs_unit_prefixmap_setup,
389 (pfn_teardown)torture_drs_unit_prefixmap_teardown);
390
391 tc->description = talloc_strdup(tc, "Unit tests for DRSUAPI::prefixMap implementation");
392
393 torture_tcase_add_simple_test(tc, "new", (pfn_run)torture_drs_unit_pfm_new);
394
395 torture_tcase_add_simple_test(tc, "make_attid_full_map", (pfn_run)torture_drs_unit_pfm_make_attid_full_map);
396 torture_tcase_add_simple_test(tc, "make_attid_small_map", (pfn_run)torture_drs_unit_pfm_make_attid_small_map);
397 torture_tcase_add_simple_test(tc, "oid_from_attid_full_map", (pfn_run)torture_drs_unit_pfm_oid_from_attid);
398
399 return tc;
400 }
fernandojvsilva's samba4 branch