s4/schema(tort): Refactor DSDB-SYNTAX test for better performance
[gd/samba/.git] / source4 / dsdb / schema / tests / schema_syntax.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    Test DSDB syntax functions
5
6    Copyright (C) Andrew Bartlet <abartlet@samba.org> 2008
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 "lib/events/events.h"
24 #include "lib/ldb/include/ldb.h"
25 #include "lib/ldb/include/ldb_errors.h"
26 #include "lib/ldb-samba/ldif_handlers.h"
27 #include "ldb_wrap.h"
28 #include "dsdb/samdb/samdb.h"
29 #include "param/param.h"
30 #include "torture/smbtorture.h"
31 #include "torture/local/proto.h"
32 #include "param/provision.h"
33
34
35 struct torture_dsdb_syntax {
36         struct ldb_context *ldb;
37         struct dsdb_schema *schema;
38 };
39
40 DATA_BLOB hexstr_to_data_blob(TALLOC_CTX *mem_ctx, const char *string) 
41 {
42         DATA_BLOB binary = data_blob_talloc(mem_ctx, NULL, strlen(string)/2);
43         binary.length = strhex_to_str((char *)binary.data, binary.length, string, strlen(string));
44         return binary;
45 }
46
47 static bool torture_syntax_add_OR_Name(struct torture_context *tctx,
48                                        struct ldb_context *ldb,
49                                        struct dsdb_schema *schema)
50 {
51         WERROR werr;
52         int ldb_res;
53         struct ldb_ldif *ldif;
54         const char *ldif_str =  "dn: CN=ms-Exch-Auth-Orig,CN=Schema,CN=Configuration,DC=kma-exch,DC=devel\n"
55                                 "changetype: add\n"
56                                 "cn: ms-Exch-Auth-Orig\n"
57                                 "attributeID: 1.2.840.113556.1.2.129\n"
58                                 "attributeSyntax: 2.5.5.7\n"
59                                 "isSingleValued: FALSE\n"
60                                 "linkID: 110\n"
61                                 "showInAdvancedViewOnly: TRUE\n"
62                                 "adminDisplayName: ms-Exch-Auth-Orig\n"
63                                 "oMObjectClass:: VgYBAgULHQ==\n"
64                                 "adminDescription: ms-Exch-Auth-Orig\n"
65                                 "oMSyntax: 127\n"
66                                 "searchFlags: 16\n"
67                                 "lDAPDisplayName: authOrig\n"
68                                 "name: ms-Exch-Auth-Orig\n"
69                                 "objectGUID:: 7tqEWktjAUqsZXqsFPQpRg==\n"
70                                 "schemaIDGUID:: l3PfqOrF0RG7ywCAx2ZwwA==\n"
71                                 "attributeSecurityGUID:: VAGN5Pi80RGHAgDAT7lgUA==\n"
72                                 "isMemberOfPartialAttributeSet: TRUE\n";
73
74         ldif = ldb_ldif_read_string(ldb, &ldif_str);
75         torture_assert(tctx, ldif, "Failed to parse LDIF for authOrig");
76
77         werr = dsdb_attribute_from_ldb(ldb, schema, ldif->msg);
78         ldb_ldif_read_free(ldb, ldif);
79         torture_assert_werr_ok(tctx, werr, "dsdb_attribute_from_ldb() failed!");
80
81         ldb_res = dsdb_set_schema(ldb, schema);
82         torture_assert_int_equal(tctx, ldb_res, LDB_SUCCESS, "dsdb_set_schema() failed");
83
84         return true;
85 };
86
87 static bool torture_test_syntax(struct torture_context *torture,
88                                 struct torture_dsdb_syntax *priv,
89                                 const char *oid,
90                                 const char *attr_string,
91                                 const char *ldb_str,
92                                 const char *drs_str)
93 {
94         TALLOC_CTX *tmp_ctx = talloc_new(torture);
95         DATA_BLOB drs_binary = hexstr_to_data_blob(tmp_ctx, drs_str);
96         DATA_BLOB ldb_blob = data_blob_string_const(ldb_str);
97         struct drsuapi_DsReplicaAttribute drs, drs2;
98         struct drsuapi_DsAttributeValue val;
99         const struct dsdb_syntax *syntax;
100         const struct dsdb_attribute *attr;
101         struct ldb_message_element el;
102         struct ldb_context *ldb = priv->ldb;
103         struct dsdb_schema *schema = priv->schema;
104
105         drs.value_ctr.num_values = 1;
106         drs.value_ctr.values = &val;
107         val.blob = &drs_binary;
108
109         torture_assert(torture, syntax = find_syntax_map_by_standard_oid(oid), "Failed to find syntax handler");
110         torture_assert(torture, attr = dsdb_attribute_by_lDAPDisplayName(schema, attr_string), "Failed to find attribute handler");
111         torture_assert_str_equal(torture, attr->syntax->name, syntax->name, "Syntax from schema not as expected");
112         
113
114         torture_assert_werr_ok(torture, syntax->drsuapi_to_ldb(ldb, schema, attr, &drs, tmp_ctx, &el), "Failed to convert from DRS to ldb format");
115
116         torture_assert_data_blob_equal(torture, el.values[0], ldb_blob, "Incorrect conversion from DRS to ldb format");
117
118         torture_assert_werr_ok(torture, syntax->ldb_to_drsuapi(ldb, schema, attr, &el, tmp_ctx, &drs2), "Failed to convert from ldb to DRS format");
119         
120         torture_assert(torture, drs2.value_ctr.values[0].blob, "No blob returned from conversion");
121
122         torture_assert_data_blob_equal(torture, *drs2.value_ctr.values[0].blob, drs_binary, "Incorrect conversion from ldb to DRS format");
123         return true;
124 }
125
126 static bool torture_dsdb_drs_DN_BINARY(struct torture_context *torture, struct torture_dsdb_syntax *priv)
127 {
128         bool ret;
129         const char *ldb_str = "B:32:A9D1CA15768811D1ADED00C04FD8D5CD:<GUID=a8378c29-6319-45b3-b216-6a3108452d6c>;CN=Users,DC=ad,DC=ruth,DC=abartlet,DC=net";
130         const char *drs_str = "8C00000000000000298C37A81963B345B2166A3108452D6C000000000000000000000000000000000000000000000000000000002900000043004E003D00550073006500720073002C00440043003D00610064002C00440043003D0072007500740068002C00440043003D00610062006100720074006C00650074002C00440043003D006E0065007400000014000000A9D1CA15768811D1ADED00C04FD8D5CD";
131         const char *ldb_str2 = "B:8:00000002:<GUID=2b475208-3180-4ad4-b6bb-a26cfb44ac50>;<SID=S-1-5-21-3686369990-3108025515-1819299124>;DC=ad,DC=ruth,DC=abartlet,DC=net";
132         const char *drs_str2 = "7A000000180000000852472B8031D44AB6BBA26CFB44AC50010400000000000515000000C68AB9DBABB440B9344D706C0000000020000000440043003D00610064002C00440043003D0072007500740068002C00440043003D00610062006100720074006C00650074002C00440043003D006E0065007400000000000800000000000002";
133         ret = torture_test_syntax(torture, priv, DSDB_SYNTAX_BINARY_DN, "wellKnownObjects", ldb_str, drs_str);
134         if (!ret) return false;
135         return torture_test_syntax(torture, priv, DSDB_SYNTAX_BINARY_DN, "msDS-HasInstantiatedNCs", ldb_str2, drs_str2);
136 }
137
138 static bool torture_dsdb_drs_DN(struct torture_context *torture, struct torture_dsdb_syntax *priv)
139 {
140         const char *ldb_str = "<GUID=fbee08fd-6f75-4bd4-af3f-e4f063a6379e>;OU=Domain Controllers,DC=ad,DC=naomi,DC=abartlet,DC=net";
141         const char *drs_str = "A800000000000000FD08EEFB756FD44BAF3FE4F063A6379E00000000000000000000000000000000000000000000000000000000370000004F0055003D0044006F006D00610069006E00200043006F006E00740072006F006C006C006500720073002C00440043003D00610064002C00440043003D006E0061006F006D0069002C00440043003D00610062006100720074006C00650074002C00440043003D006E00650074000000";
142         if (!torture_test_syntax(torture, priv, LDB_SYNTAX_DN, "lastKnownParent", ldb_str, drs_str)) {
143                 return false;
144         }
145
146         /* extended_dn with GUID and SID in it */
147         ldb_str = "<GUID=23cc7d16-3da0-4f3a-9921-0ad60a99230f>;<SID=S-1-5-21-3427639452-1671929926-2759570404-500>;CN=Administrator,CN=Users,DC=kma-exch,DC=devel";
148         drs_str = "960000001C000000167DCC23A03D3A4F99210AD60A99230F0105000000000005150000009CA04DCC46A0A763E4B37BA4F40100002E00000043004E003D00410064006D0069006E006900730074007200610074006F0072002C0043004E003D00550073006500720073002C00440043003D006B006D0061002D0065007800630068002C00440043003D0064006500760065006C000000";
149         return torture_test_syntax(torture, priv, LDB_SYNTAX_DN, "lastKnownParent", ldb_str, drs_str);
150 }
151
152 static bool torture_dsdb_drs_OR_Name(struct torture_context *torture, struct torture_dsdb_syntax *priv)
153 {
154         const char *ldb_str = "<GUID=23cc7d16-3da0-4f3a-9921-0ad60a99230f>;<SID=S-1-5-21-3427639452-1671929926-2759570404-500>;CN=Administrator,CN=Users,DC=kma-exch,DC=devel";
155         const char *drs_str = "960000001C000000167DCC23A03D3A4F99210AD60A99230F0105000000000005150000009CA04DCC46A0A763E4B37BA4F40100002E00000043004E003D00410064006D0069006E006900730074007200610074006F0072002C0043004E003D00550073006500720073002C00440043003D006B006D0061002D0065007800630068002C00440043003D0064006500760065006C000000000004000000";
156         return torture_test_syntax(torture, priv, DSDB_SYNTAX_OR_NAME, "authOrig", ldb_str, drs_str);
157 }
158
159 static bool torture_dsdb_drs_INT32(struct torture_context *torture, struct torture_dsdb_syntax *priv)
160 {
161         const char *ldb_str = "532480";
162         const char *drs_str = "00200800";
163         return torture_test_syntax(torture, priv, LDB_SYNTAX_INTEGER, "userAccountControl", ldb_str, drs_str);
164 }
165
166 static bool torture_dsdb_drs_INT64(struct torture_context *torture, struct torture_dsdb_syntax *priv)
167 {
168         const char *ldb_str = "129022979538281250";
169         const char *drs_str = "22E33D5FB761CA01";
170         return torture_test_syntax(torture, priv, "1.2.840.113556.1.4.906", "pwdLastSet", ldb_str, drs_str);
171 }
172
173 static bool torture_dsdb_drs_NTTIME(struct torture_context *torture, struct torture_dsdb_syntax *priv)
174 {
175         const char *ldb_str = "20091109003446.0Z";
176         const char *drs_str = "A6F4070103000000";
177         return torture_test_syntax(torture, priv, "1.3.6.1.4.1.1466.115.121.1.24", "whenCreated", ldb_str, drs_str);
178 }
179
180 static bool torture_dsdb_drs_BOOL(struct torture_context *torture, struct torture_dsdb_syntax *priv)
181 {
182         const char *ldb_str = "TRUE";
183         const char *drs_str = "01000000";
184         return torture_test_syntax(torture, priv, LDB_SYNTAX_BOOLEAN, "isDeleted", ldb_str, drs_str);
185 }
186
187 static bool torture_dsdb_drs_UNICODE(struct torture_context *torture, struct torture_dsdb_syntax *priv)
188 {
189         const char *ldb_str = "primaryTelexNumber,Numéro de télex";
190         const char *drs_str = "7000720069006D00610072007900540065006C00650078004E0075006D006200650072002C004E0075006D00E90072006F0020006400650020007400E9006C0065007800";
191         return torture_test_syntax(torture, priv, LDB_SYNTAX_DIRECTORY_STRING, "attributeDisplayNames", ldb_str, drs_str);
192 }
193
194 /*
195  * DSDB-SYNTAX fixture setup/teardown handlers implementation
196  */
197 static bool torture_dsdb_syntax_tcase_setup(struct torture_context *tctx, void **data)
198 {
199         struct torture_dsdb_syntax *priv;
200
201         priv = talloc_zero(tctx, struct torture_dsdb_syntax);
202         torture_assert(tctx, priv, "No memory");
203
204         priv->ldb = provision_get_schema(priv, tctx->lp_ctx);
205         torture_assert(tctx, priv->ldb, "Failed to load schema from disk");
206
207         priv->schema = dsdb_get_schema(priv->ldb);
208         torture_assert(tctx, priv->schema, "Failed to fetch schema");
209
210         /* add 'authOrig' attribute with OR-Name syntax to schema */
211         if (!torture_syntax_add_OR_Name(tctx, priv->ldb, priv->schema)) {
212                 return false;
213         }
214
215         *data = priv;
216         return true;
217 }
218
219 static bool torture_dsdb_syntax_tcase_teardown(struct torture_context *tctx, void *data)
220 {
221         struct torture_dsdb_syntax *priv;
222
223         priv = talloc_get_type_abort(data, struct torture_dsdb_syntax);
224         talloc_free(priv);
225
226         return true;
227 }
228
229 /**
230  * DSDB-SYNTAX test suite creation
231  */
232 struct torture_suite *torture_dsdb_syntax(TALLOC_CTX *mem_ctx)
233 {
234         typedef bool (*pfn_run)(struct torture_context *, void *);
235
236         struct torture_tcase *tc;
237         struct torture_suite *suite = torture_suite_create(mem_ctx, "DSDB-SYNTAX");
238
239         if (suite == NULL) {
240                 return NULL;
241         }
242
243         tc = torture_suite_add_tcase(suite, "tc");
244         if (!tc) {
245                 return NULL;
246         }
247
248         torture_tcase_set_fixture(tc,
249                                   torture_dsdb_syntax_tcase_setup,
250                                   torture_dsdb_syntax_tcase_teardown);
251
252         torture_tcase_add_simple_test(tc, "DN-BINARY", (pfn_run)torture_dsdb_drs_DN_BINARY);
253         torture_tcase_add_simple_test(tc, "DN", (pfn_run)torture_dsdb_drs_DN);
254         torture_tcase_add_simple_test(tc, "OR-Name", (pfn_run)torture_dsdb_drs_OR_Name);
255         torture_tcase_add_simple_test(tc, "INT32", (pfn_run)torture_dsdb_drs_INT32);
256         torture_tcase_add_simple_test(tc, "INT64", (pfn_run)torture_dsdb_drs_INT64);
257         torture_tcase_add_simple_test(tc, "NTTIME", (pfn_run)torture_dsdb_drs_NTTIME);
258         torture_tcase_add_simple_test(tc, "BOOL", (pfn_run)torture_dsdb_drs_BOOL);
259         torture_tcase_add_simple_test(tc, "UNICODE", (pfn_run)torture_dsdb_drs_UNICODE);
260
261         suite->description = talloc_strdup(suite, "DSDB syntax tests");
262
263         return suite;
264 }