s4:schema Add some error checking to the schema load
[ira/wip.git] / source4 / dsdb / schema / schema_inferiors.c
1 /* 
2    Unix SMB/CIFS mplementation.
3
4    implement possibleInferiors calculation
5    
6    Copyright (C) Andrew Tridgell 2009
7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2009
8
9    This program is free software; you can redistribute it and/or modify
10    it under the terms of the GNU General Public License as published by
11    the Free Software Foundation; either version 3 of the License, or
12    (at your option) any later version.
13    
14    This program is distributed in the hope that it will be useful,
15    but WITHOUT ANY WARRANTY; without even the implied warranty of
16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17    GNU General Public License for more details.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
21    
22 */
23 /*
24   This module is a C implementation of the logic in the
25   dsdb/samdb/ldb_modules/tests/possibleInferiors.py code
26
27   To understand the C code, please see the python code first
28  */
29
30 #include "includes.h"
31 #include "dsdb/samdb/samdb.h"
32
33
34 /*
35   create the SUPCLASSES() list
36  */
37 static char **schema_supclasses(struct dsdb_schema *schema, struct dsdb_class *schema_class)
38 {
39         char **list;
40
41         if (schema_class->supclasses) {
42                 return schema_class->supclasses;
43         }
44
45         list = str_list_make_empty(schema_class);
46         if (list == NULL) {
47                 DEBUG(0,(__location__ " out of memory\n"));
48                 return NULL;
49         }
50
51         /* Cope with 'top SUP top', ie top is subClassOf top */ 
52         if (schema_class->subClassOf &&
53             strcmp(schema_class->lDAPDisplayName, schema_class->subClassOf) == 0) {
54                 schema_class->supclasses = list;
55                 return list;
56         }
57
58         if (schema_class->subClassOf) {
59                 const char **list2;
60                 list = str_list_add_const(list, schema_class->subClassOf);
61
62                 list2 = schema_supclasses(schema,  
63                                           discard_const_p(struct dsdb_class, 
64                                                           dsdb_class_by_lDAPDisplayName(schema, 
65                                                                                         schema_class->subClassOf)));
66                 list = str_list_append_const(list, list2);
67         }
68
69         schema_class->supclasses = str_list_unique(list);
70         
71         return list;
72 }
73
74 /*
75   this one is used internally
76   matches SUBCLASSES() python function
77  */
78 static char **schema_subclasses(struct dsdb_schema *schema, TALLOC_CTX *mem_ctx, char **oclist)
79 {
80         char **list = str_list_make_empty(mem_ctx);
81         int i;
82
83         for (i=0; oclist && oclist[i]; i++) {
84                 struct dsdb_class *schema_class = dsdb_class_by_lDAPDisplayName(schema, oclist[i]);
85                 if (!schema_class) {
86                         DEBUG(0, ("ERROR: Unable to locate subClass: '%s'\n", oclist[i]));
87                         continue;
88                 }
89                 list = str_list_append_const(list, schema_class->subclasses);
90         }
91         return list;
92 }
93
94
95 /* 
96    equivalent of the POSSSUPERIORS() python function
97  */
98 static char **schema_posssuperiors(struct dsdb_schema *schema, 
99                                    struct dsdb_class *schema_class)
100 {
101         if (schema_class->posssuperiors == NULL) {
102                 char **list2 = str_list_make_empty(schema_class);
103                 char **list3;
104                 int i;
105
106                 list2 = str_list_append_const(list2, schema_class->systemPossSuperiors);
107                 list2 = str_list_append_const(list2, schema_class->possSuperiors);
108                 list3 = schema_supclasses(schema, schema_class);
109                 for (i=0; list3 && list3[i]; i++) {
110                         struct dsdb_class *class2 = dsdb_class_by_lDAPDisplayName(schema, list3[i]);
111                         if (!class2) {
112                                 DEBUG(0, ("ERROR: Unable to locate supClass: '%s'\n", list3[i]));
113                                 continue;
114                         }
115                         list2 = str_list_append_const(list2, schema_posssuperiors(schema, class2));
116                 }
117                 list2 = str_list_append_const(list2, schema_subclasses(schema, list2, list2));
118
119                 schema_class->posssuperiors = str_list_unique(list2);
120         }
121
122         return schema_class->posssuperiors;
123 }
124
125 static char **schema_subclasses_recurse(struct dsdb_schema *schema, struct dsdb_class *schema_class)
126 {
127         char **list = str_list_copy_const(schema_class, schema_class->subclasses_direct);
128         int i;
129         for (i=0;list && list[i]; i++) {
130                 struct dsdb_class *schema_class2 = dsdb_class_by_lDAPDisplayName(schema, list[i]);
131                 if (schema_class != schema_class2) {
132                         list = str_list_append_const(list, schema_subclasses_recurse(schema, schema_class2));
133                 }
134         }
135         return list;
136 }
137
138 /* Walk down the subClass tree, setting a higher index as we go down
139  * each level.  top is 1, subclasses of top are 2, etc */
140 void schema_subclasses_order_recurse(struct dsdb_schema *schema, struct dsdb_class *schema_class, int order)
141 {
142         const char **list = schema_class->subclasses_direct;
143         int i;
144         schema_class->subClass_order = order;
145         for (i=0;list && list[i]; i++) {
146                 struct dsdb_class *schema_class2 = dsdb_class_by_lDAPDisplayName(schema, list[i]);
147                 schema_subclasses_order_recurse(schema, schema_class2, order+1);
148         }
149         return;
150 }
151
152 static int schema_create_subclasses(struct dsdb_schema *schema)
153 {
154         struct dsdb_class *schema_class, *top;
155
156         for (schema_class=schema->classes; schema_class; schema_class=schema_class->next) {
157                 struct dsdb_class *schema_class2 = dsdb_class_by_lDAPDisplayName(schema, schema_class->subClassOf);
158                 if (schema_class2 == NULL) {
159                         DEBUG(0,("ERROR: no subClassOf for '%s'\n", schema_class->lDAPDisplayName));
160                         return LDB_ERR_OPERATIONS_ERROR;
161                 }
162                 if (schema_class2 && schema_class != schema_class2) {
163                         if (schema_class2->subclasses_direct == NULL) {
164                                 schema_class2->subclasses_direct = str_list_make_empty(schema_class2);
165                                 if (!schema_class2->subclasses_direct) {
166                                         return LDB_ERR_OPERATIONS_ERROR;
167                                 }
168                         }
169                         schema_class2->subclasses_direct = str_list_add_const(schema_class2->subclasses_direct, 
170                                                                         schema_class->lDAPDisplayName);
171                 }
172         }
173
174         for (schema_class=schema->classes; schema_class; schema_class=schema_class->next) {
175                 schema_class->subclasses = str_list_unique(schema_subclasses_recurse(schema, schema_class));
176
177                 /* Initilise the subClass order, to ensure we can't have uninitilised sort on the subClass hirarchy */
178                 schema_class->subClass_order = 0;
179         }
180
181         top = dsdb_class_by_lDAPDisplayName(schema, "top");
182         if (!top) {
183                 DEBUG(0,("ERROR: no 'top' class in loaded schema\n"));
184                 return LDB_ERR_OPERATIONS_ERROR;
185         }
186
187         schema_subclasses_order_recurse(schema, top, 1);
188         return LDB_SUCCESS;
189 }
190
191 static void schema_fill_possible_inferiors(struct dsdb_schema *schema, struct dsdb_class *schema_class)
192 {
193         struct dsdb_class *c2;
194
195         for (c2=schema->classes; c2; c2=c2->next) {
196                 char **superiors = schema_posssuperiors(schema, c2);
197                 if (c2->systemOnly == false 
198                     && c2->objectClassCategory != 2 
199                     && c2->objectClassCategory != 3
200                     && str_list_check(superiors, schema_class->lDAPDisplayName)) {
201                         if (schema_class->possibleInferiors == NULL) {
202                                 schema_class->possibleInferiors = str_list_make_empty(schema_class);
203                         }
204                         schema_class->possibleInferiors = str_list_add_const(schema_class->possibleInferiors,
205                                                                        c2->lDAPDisplayName);
206                 }
207         }
208         schema_class->possibleInferiors = str_list_unique(schema_class->possibleInferiors);
209 }
210
211 static void schema_fill_system_possible_inferiors(struct dsdb_schema *schema, struct dsdb_class *schema_class)
212 {
213         struct dsdb_class *c2;
214
215         for (c2=schema->classes; c2; c2=c2->next) {
216                 char **superiors = schema_posssuperiors(schema, c2);
217                 if (c2->objectClassCategory != 2
218                     && c2->objectClassCategory != 3
219                     && str_list_check(superiors, schema_class->lDAPDisplayName)) {
220                         if (schema_class->systemPossibleInferiors == NULL) {
221                                 schema_class->systemPossibleInferiors = str_list_make_empty(schema_class);
222                         }
223                         schema_class->systemPossibleInferiors = str_list_add_const(schema_class->systemPossibleInferiors,
224                                                                                    c2->lDAPDisplayName);
225                 }
226         }
227         schema_class->systemPossibleInferiors = str_list_unique(schema_class->systemPossibleInferiors);
228 }
229
230 /*
231   fill in a string class name from a governs_ID
232  */
233 static void schema_fill_from_class_one(struct dsdb_schema *schema, struct dsdb_class *c, 
234                                     const char **s, uint32_t id)
235 {
236         if (*s == NULL && id != 0) {
237                 struct dsdb_class *c2 = dsdb_class_by_governsID_id(schema, id);
238                 if (c2) {
239                         *s = c2->lDAPDisplayName;
240                 }
241         }
242 }
243
244 /*
245   fill in a list of string class names from a governs_ID list
246  */
247 static void schema_fill_from_class_list(struct dsdb_schema *schema, struct dsdb_class *c, 
248                                      const char ***s, uint32_t *ids)
249 {
250         if (*s == NULL && ids != NULL) {
251                 int i;
252                 for (i=0;ids[i];i++) ;
253                 *s = talloc_array(c, const char *, i+1);
254                 for (i=0;ids[i];i++) {
255                         struct dsdb_class *c2 = dsdb_class_by_governsID_id(schema, ids[i]);
256                         if (c2) {
257                                 (*s)[i] = c2->lDAPDisplayName;
258                         } else {
259                                 (*s)[i] = NULL;                         
260                         }
261                 }
262                 (*s)[i] = NULL;                         
263         }
264 }
265
266 /*
267   fill in a list of string attribute names from a attributeID list
268  */
269 static void schema_fill_from_attribute_list(struct dsdb_schema *schema, struct dsdb_class *c, 
270                                             const char ***s, uint32_t *ids)
271 {
272         if (*s == NULL && ids != NULL) {
273                 int i;
274                 for (i=0;ids[i];i++) ;
275                 *s = talloc_array(c, const char *, i+1);
276                 for (i=0;ids[i];i++) {
277                         struct dsdb_attribute *a = dsdb_attribute_by_attributeID_id(schema, ids[i]);
278                         if (a) {
279                                 (*s)[i] = a->lDAPDisplayName;
280                         } else {
281                                 (*s)[i] = NULL;                         
282                         }
283                 }
284                 (*s)[i] = NULL;                         
285         }
286 }
287
288 /*
289   if the schema came from DRS then some attributes will be setup as IDs
290  */
291 static void schema_fill_from_ids(struct dsdb_schema *schema)
292 {
293         struct dsdb_class *c;
294         for (c=schema->classes; c; c=c->next) {
295                 schema_fill_from_class_one(schema, c, &c->subClassOf, c->subClassOf_id);
296                 schema_fill_from_attribute_list(schema, c, &c->systemMayContain, c->systemMayContain_ids);
297                 schema_fill_from_attribute_list(schema, c, &c->systemMustContain, c->systemMustContain_ids);
298                 schema_fill_from_attribute_list(schema, c, &c->mustContain, c->mustContain_ids);
299                 schema_fill_from_attribute_list(schema, c, &c->mayContain, c->mayContain_ids);
300                 schema_fill_from_class_list(schema, c, &c->possSuperiors, c->possSuperiors_ids);
301                 schema_fill_from_class_list(schema, c, &c->systemPossSuperiors, c->systemPossSuperiors_ids);
302                 schema_fill_from_class_list(schema, c, &c->systemAuxiliaryClass, c->systemAuxiliaryClass_ids);
303                 schema_fill_from_class_list(schema, c, &c->auxiliaryClass, c->auxiliaryClass_ids);
304         }
305 }
306
307 int schema_fill_constructed(struct dsdb_schema *schema) 
308 {
309         int ret;
310         struct dsdb_class *schema_class;
311
312         schema_fill_from_ids(schema);
313
314         ret = schema_create_subclasses(schema);
315         if (ret != LDB_SUCCESS) {
316                 return ret;
317         }
318
319         for (schema_class=schema->classes; schema_class; schema_class=schema_class->next) {
320                 schema_fill_possible_inferiors(schema, schema_class);
321                 schema_fill_system_possible_inferiors(schema, schema_class);
322         }
323
324         /* free up our internal cache elements */
325         for (schema_class=schema->classes; schema_class; schema_class=schema_class->next) {
326                 talloc_free(schema_class->supclasses);
327                 talloc_free(schema_class->subclasses_direct);
328                 talloc_free(schema_class->subclasses);
329                 talloc_free(schema_class->posssuperiors);
330                 schema_class->supclasses = NULL;
331                 schema_class->subclasses_direct = NULL;
332                 schema_class->subclasses = NULL;
333                 schema_class->posssuperiors = NULL;
334         }
335         return LDB_SUCCESS;
336 }