Initial Implementation of the DS objects access checks.
[sfrench/samba-autobuild/.git] / source4 / dsdb / samdb / ldb_modules / rootdse.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    rootDSE ldb module
5
6    Copyright (C) Andrew Tridgell 2005
7    Copyright (C) Simo Sorce 2005-2008
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 #include "includes.h"
24 #include "lib/ldb/include/ldb.h"
25 #include "lib/ldb/include/ldb_module.h"
26 #include "system/time.h"
27 #include "dsdb/samdb/samdb.h"
28 #include "version.h"
29
30 struct private_data {
31         int num_controls;
32         char **controls;
33         int num_partitions;
34         struct ldb_dn **partitions;
35 };
36
37 /*
38   return 1 if a specific attribute has been requested
39 */
40 static int do_attribute(const char * const *attrs, const char *name)
41 {
42         return attrs == NULL ||
43                 ldb_attr_in_list(attrs, name) ||
44                 ldb_attr_in_list(attrs, "*");
45 }
46
47 static int do_attribute_explicit(const char * const *attrs, const char *name)
48 {
49         return attrs != NULL && ldb_attr_in_list(attrs, name);
50 }
51
52
53 /*
54   add dynamically generated attributes to rootDSE result
55 */
56 static int rootdse_add_dynamic(struct ldb_module *module, struct ldb_message *msg, const char * const *attrs)
57 {
58         struct ldb_context *ldb;
59         struct private_data *priv = talloc_get_type(ldb_module_get_private(module), struct private_data);
60         char **server_sasl;
61         const struct dsdb_schema *schema;
62         int *val;
63
64         ldb = ldb_module_get_ctx(module);
65         schema = dsdb_get_schema(ldb);
66
67         msg->dn = ldb_dn_new(msg, ldb, NULL);
68
69         /* don't return the distinduishedName, cn and name attributes */
70         ldb_msg_remove_attr(msg, "distinguishedName");
71         ldb_msg_remove_attr(msg, "cn");
72         ldb_msg_remove_attr(msg, "name");
73
74         if (do_attribute(attrs, "currentTime")) {
75                 if (ldb_msg_add_steal_string(msg, "currentTime", 
76                                              ldb_timestring(msg, time(NULL))) != 0) {
77                         goto failed;
78                 }
79         }
80
81         if (priv && do_attribute(attrs, "supportedControl")) {
82                 int i;
83                 for (i = 0; i < priv->num_controls; i++) {
84                         char *control = talloc_strdup(msg, priv->controls[i]);
85                         if (!control) {
86                                 goto failed;
87                         }
88                         if (ldb_msg_add_steal_string(msg, "supportedControl",
89                                                      control) != 0) {
90                                 goto failed;
91                         }
92                 }
93         }
94
95         if (priv && do_attribute(attrs, "namingContexts")) {
96                 int i;
97                 for (i = 0; i < priv->num_partitions; i++) {
98                         struct ldb_dn *dn = priv->partitions[i];
99                         if (ldb_msg_add_steal_string(msg, "namingContexts",
100                                                      ldb_dn_alloc_linearized(msg, dn)) != 0) {
101                                 goto failed;
102                         }
103                 }
104         }
105
106         server_sasl = talloc_get_type(ldb_get_opaque(ldb, "supportedSASLMechanims"), 
107                                        char *);
108         if (server_sasl && do_attribute(attrs, "supportedSASLMechanisms")) {
109                 int i;
110                 for (i = 0; server_sasl && server_sasl[i]; i++) {
111                         char *sasl_name = talloc_strdup(msg, server_sasl[i]);
112                         if (!sasl_name) {
113                                 goto failed;
114                         }
115                         if (ldb_msg_add_steal_string(msg, "supportedSASLMechanisms",
116                                                      sasl_name) != 0) {
117                                 goto failed;
118                         }
119                 }
120         }
121
122         if (do_attribute(attrs, "highestCommittedUSN")) {
123                 uint64_t seq_num;
124                 int ret = ldb_sequence_number(ldb, LDB_SEQ_HIGHEST_SEQ, &seq_num);
125                 if (ret == LDB_SUCCESS) {
126                         if (ldb_msg_add_fmt(msg, "highestCommittedUSN", 
127                                             "%llu", (unsigned long long)seq_num) != 0) {
128                                 goto failed;
129                         }
130                 }
131         }
132
133         if (schema && do_attribute_explicit(attrs, "dsSchemaAttrCount")) {
134                 struct dsdb_attribute *cur;
135                 uint32_t n = 0;
136
137                 for (cur = schema->attributes; cur; cur = cur->next) {
138                         n++;
139                 }
140
141                 if (ldb_msg_add_fmt(msg, "dsSchemaAttrCount", 
142                                     "%u", n) != 0) {
143                         goto failed;
144                 }
145         }
146
147         if (schema && do_attribute_explicit(attrs, "dsSchemaClassCount")) {
148                 struct dsdb_class *cur;
149                 uint32_t n = 0;
150
151                 for (cur = schema->classes; cur; cur = cur->next) {
152                         n++;
153                 }
154
155                 if (ldb_msg_add_fmt(msg, "dsSchemaClassCount", 
156                                     "%u", n) != 0) {
157                         goto failed;
158                 }
159         }
160
161         if (schema && do_attribute_explicit(attrs, "dsSchemaPrefixCount")) {
162                 if (ldb_msg_add_fmt(msg, "dsSchemaPrefixCount", 
163                                     "%u", schema->num_prefixes) != 0) {
164                         goto failed;
165                 }
166         }
167
168         if (do_attribute_explicit(attrs, "validFSMOs")) {
169                 const struct dsdb_naming_fsmo *naming_fsmo;
170                 const struct dsdb_pdc_fsmo *pdc_fsmo;
171                 const char *dn_str;
172
173                 if (schema && schema->fsmo.we_are_master) {
174                         dn_str = ldb_dn_get_linearized(samdb_schema_dn(ldb));
175                         if (dn_str && dn_str[0]) {
176                                 if (ldb_msg_add_fmt(msg, "validFSMOs", "%s", dn_str) != 0) {
177                                         goto failed;
178                                 }
179                         }
180                 }
181
182                 naming_fsmo = talloc_get_type(ldb_get_opaque(ldb, "dsdb_naming_fsmo"),
183                                               struct dsdb_naming_fsmo);
184                 if (naming_fsmo && naming_fsmo->we_are_master) {
185                         dn_str = ldb_dn_get_linearized(samdb_partitions_dn(ldb, msg));
186                         if (dn_str && dn_str[0]) {
187                                 if (ldb_msg_add_fmt(msg, "validFSMOs", "%s", dn_str) != 0) {
188                                         goto failed;
189                                 }
190                         }
191                 }
192
193                 pdc_fsmo = talloc_get_type(ldb_get_opaque(ldb, "dsdb_pdc_fsmo"),
194                                            struct dsdb_pdc_fsmo);
195                 if (pdc_fsmo && pdc_fsmo->we_are_master) {
196                         dn_str = ldb_dn_get_linearized(samdb_base_dn(ldb));
197                         if (dn_str && dn_str[0]) {
198                                 if (ldb_msg_add_fmt(msg, "validFSMOs", "%s", dn_str) != 0) {
199                                         goto failed;
200                                 }
201                         }
202                 }
203         }
204
205         if (do_attribute_explicit(attrs, "vendorVersion")) {
206                 if (ldb_msg_add_fmt(msg, "vendorVersion", 
207                                     "%s", SAMBA_VERSION_STRING) != 0) {
208                         goto failed;
209                 }
210         }
211
212         if (priv && do_attribute(attrs, "domainFunctionality")
213             && (val = talloc_get_type(ldb_get_opaque(ldb, "domainFunctionality"), int))) {
214                 if (ldb_msg_add_fmt(msg, "domainFunctionality", 
215                                     "%d", *val) != 0) {
216                         goto failed;
217                 }
218         }
219
220         if (priv && do_attribute(attrs, "forestFunctionality")
221             && (val = talloc_get_type(ldb_get_opaque(ldb, "forestFunctionality"), int))) {
222                 if (ldb_msg_add_fmt(msg, "forestFunctionality", 
223                                     "%d", *val) != 0) {
224                         goto failed;
225                 }
226         }
227
228         if (priv && do_attribute(attrs, "domainControllerFunctionality")
229             && (val = talloc_get_type(ldb_get_opaque(ldb, "domainControllerFunctionality"), int))) {
230                 if (ldb_msg_add_fmt(msg, "domainControllerFunctionality", 
231                                     "%d", *val) != 0) {
232                         goto failed;
233                 }
234         }
235
236         /* TODO: lots more dynamic attributes should be added here */
237
238         return LDB_SUCCESS;
239
240 failed:
241         return LDB_ERR_OPERATIONS_ERROR;
242 }
243
244 /*
245   handle search requests
246 */
247
248 struct rootdse_context {
249         struct ldb_module *module;
250         struct ldb_request *req;
251 };
252
253 static struct rootdse_context *rootdse_init_context(struct ldb_module *module,
254                                                     struct ldb_request *req)
255 {
256         struct ldb_context *ldb;
257         struct rootdse_context *ac;
258
259         ldb = ldb_module_get_ctx(module);
260
261         ac = talloc_zero(req, struct rootdse_context);
262         if (ac == NULL) {
263                 ldb_set_errstring(ldb, "Out of Memory");
264                 return NULL;
265         }
266
267         ac->module = module;
268         ac->req = req;
269
270         return ac;
271 }
272
273 static int rootdse_callback(struct ldb_request *req, struct ldb_reply *ares)
274 {
275         struct rootdse_context *ac;
276         int ret;
277
278         ac = talloc_get_type(req->context, struct rootdse_context);
279
280         if (!ares) {
281                 return ldb_module_done(ac->req, NULL, NULL,
282                                         LDB_ERR_OPERATIONS_ERROR);
283         }
284         if (ares->error != LDB_SUCCESS) {
285                 return ldb_module_done(ac->req, ares->controls,
286                                         ares->response, ares->error);
287         }
288
289         switch (ares->type) {
290         case LDB_REPLY_ENTRY:
291                 /*
292                  * if the client explicit asks for the 'netlogon' attribute
293                  * the reply_entry needs to be skipped
294                  */
295                 if (ac->req->op.search.attrs &&
296                     ldb_attr_in_list(ac->req->op.search.attrs, "netlogon")) {
297                         talloc_free(ares);
298                         return LDB_SUCCESS;
299                 }
300
301                 /* for each record returned post-process to add any dynamic
302                    attributes that have been asked for */
303                 ret = rootdse_add_dynamic(ac->module, ares->message,
304                                           ac->req->op.search.attrs);
305                 if (ret != LDB_SUCCESS) {
306                         talloc_free(ares);
307                         return ldb_module_done(ac->req, NULL, NULL, ret);
308                 }
309
310                 return ldb_module_send_entry(ac->req, ares->message, ares->controls);
311
312         case LDB_REPLY_REFERRAL:
313                 /* should we allow the backend to return referrals in this case
314                  * ?? */
315                 break;
316
317         case LDB_REPLY_DONE:
318                 return ldb_module_done(ac->req, ares->controls,
319                                         ares->response, ares->error);
320         }
321
322         talloc_free(ares);
323         return LDB_SUCCESS;
324 }
325
326 static int rootdse_search(struct ldb_module *module, struct ldb_request *req)
327 {
328         struct ldb_context *ldb;
329         struct rootdse_context *ac;
330         struct ldb_request *down_req;
331         int ret;
332
333         ldb = ldb_module_get_ctx(module);
334
335         /* see if its for the rootDSE - only a base search on the "" DN qualifies */
336         if (!(req->op.search.scope == LDB_SCOPE_BASE && ldb_dn_is_null(req->op.search.base))) {
337                 /* Otherwise, pass down to the rest of the stack */
338                 return ldb_next_request(module, req);
339         }
340
341         ac = rootdse_init_context(module, req);
342         if (ac == NULL) {
343                 return LDB_ERR_OPERATIONS_ERROR;
344         }
345
346         /* in our db we store the rootDSE with a DN of @ROOTDSE */
347         ret = ldb_build_search_req(&down_req, ldb, ac,
348                                         ldb_dn_new(ac, ldb, "@ROOTDSE"),
349                                         LDB_SCOPE_BASE,
350                                         NULL,
351                                         req->op.search.attrs,
352                                         NULL,/* for now skip the controls from the client */
353                                         ac, rootdse_callback,
354                                         req);
355         if (ret != LDB_SUCCESS) {
356                 return ret;
357         }
358
359         return ldb_next_request(module, down_req);
360 }
361
362 static int rootdse_register_control(struct ldb_module *module, struct ldb_request *req)
363 {
364         struct private_data *priv = talloc_get_type(ldb_module_get_private(module), struct private_data);
365         char **list;
366
367         list = talloc_realloc(priv, priv->controls, char *, priv->num_controls + 1);
368         if (!list) {
369                 return LDB_ERR_OPERATIONS_ERROR;
370         }
371
372         list[priv->num_controls] = talloc_strdup(list, req->op.reg_control.oid);
373         if (!list[priv->num_controls]) {
374                 return LDB_ERR_OPERATIONS_ERROR;
375         }
376
377         priv->num_controls += 1;
378         priv->controls = list;
379
380         return ldb_module_done(req, NULL, NULL, LDB_SUCCESS);
381 }
382
383 static int rootdse_register_partition(struct ldb_module *module, struct ldb_request *req)
384 {
385         struct private_data *priv = talloc_get_type(ldb_module_get_private(module), struct private_data);
386         struct ldb_dn **list;
387
388         list = talloc_realloc(priv, priv->partitions, struct ldb_dn *, priv->num_partitions + 1);
389         if (!list) {
390                 return LDB_ERR_OPERATIONS_ERROR;
391         }
392
393         list[priv->num_partitions] = ldb_dn_copy(list, req->op.reg_partition.dn);
394         if (!list[priv->num_partitions]) {
395                 return LDB_ERR_OPERATIONS_ERROR;
396         }
397
398         priv->num_partitions += 1;
399         priv->partitions = list;
400
401         return ldb_module_done(req, NULL, NULL, LDB_SUCCESS);
402 }
403
404
405 static int rootdse_request(struct ldb_module *module, struct ldb_request *req)
406 {
407         switch (req->operation) {
408
409         case LDB_REQ_REGISTER_CONTROL:
410                 return rootdse_register_control(module, req);
411         case LDB_REQ_REGISTER_PARTITION:
412                 return rootdse_register_partition(module, req);
413
414         default:
415                 break;
416         }
417         return ldb_next_request(module, req);
418 }
419
420 static int rootdse_init(struct ldb_module *module)
421 {
422         int ret;
423         struct ldb_context *ldb;
424         struct ldb_result *res;
425         struct private_data *data;
426         const char *attrs[] = { "msDS-Behavior-Version", NULL };
427         const char *ds_attrs[] = { "dsServiceName", NULL };
428         TALLOC_CTX *mem_ctx;
429
430         ldb = ldb_module_get_ctx(module);
431
432         data = talloc_zero(module, struct private_data);
433         if (data == NULL) {
434                 return -1;
435         }
436
437         data->num_controls = 0;
438         data->controls = NULL;
439         data->num_partitions = 0;
440         data->partitions = NULL;
441         ldb_module_set_private(module, data);
442
443         ldb_set_default_dns(ldb);
444
445         ret = ldb_next_init(module);
446
447         if (ret) {
448                 return ret;
449         }
450
451         mem_ctx = talloc_new(data);
452         if (!mem_ctx) {
453                 ldb_oom(ldb);
454                 return LDB_ERR_OPERATIONS_ERROR;
455         }
456
457         /* Now that the partitions are set up, do a search for:
458            - domainControllerFunctionality
459            - domainFunctionality
460            - forestFunctionality
461
462            Then stuff these values into an opaque
463         */
464         ret = ldb_search(ldb, mem_ctx, &res,
465                          ldb_get_default_basedn(ldb),
466                          LDB_SCOPE_BASE, attrs, NULL);
467         if (ret == LDB_SUCCESS && res->count == 1) {
468                 int domain_behaviour_version
469                         = ldb_msg_find_attr_as_int(res->msgs[0], 
470                                                    "msDS-Behavior-Version", -1);
471                 if (domain_behaviour_version != -1) {
472                         int *val = talloc(ldb, int);
473                         if (!val) {
474                                 ldb_oom(ldb);
475                                 talloc_free(mem_ctx);
476                                 return LDB_ERR_OPERATIONS_ERROR;
477                         }
478                         *val = domain_behaviour_version;
479                         ret = ldb_set_opaque(ldb, "domainFunctionality", val);
480                         if (ret != LDB_SUCCESS) {
481                                 talloc_free(mem_ctx);
482                                 return ret;
483                         }
484                 }
485         }
486
487         ret = ldb_search(ldb, mem_ctx, &res,
488                          samdb_partitions_dn(ldb, mem_ctx),
489                          LDB_SCOPE_BASE, attrs, NULL);
490         if (ret == LDB_SUCCESS && res->count == 1) {
491                 int forest_behaviour_version
492                         = ldb_msg_find_attr_as_int(res->msgs[0], 
493                                                    "msDS-Behavior-Version", -1);
494                 if (forest_behaviour_version != -1) {
495                         int *val = talloc(ldb, int);
496                         if (!val) {
497                                 ldb_oom(ldb);
498                                 talloc_free(mem_ctx);
499                                 return LDB_ERR_OPERATIONS_ERROR;
500                         }
501                         *val = forest_behaviour_version;
502                         ret = ldb_set_opaque(ldb, "forestFunctionality", val);
503                         if (ret != LDB_SUCCESS) {
504                                 talloc_free(mem_ctx);
505                                 return ret;
506                         }
507                 }
508         }
509
510         ret = ldb_search(ldb, mem_ctx, &res,
511                          ldb_dn_new(mem_ctx, ldb, ""),
512                          LDB_SCOPE_BASE, ds_attrs, NULL);
513         if (ret == LDB_SUCCESS && res->count == 1) {
514                 struct ldb_dn *ds_dn
515                         = ldb_msg_find_attr_as_dn(ldb, mem_ctx, res->msgs[0], 
516                                                   "dsServiceName");
517                 if (ds_dn) {
518                         ret = ldb_search(ldb, mem_ctx, &res, ds_dn, 
519                                          LDB_SCOPE_BASE, attrs, NULL);
520                         if (ret == LDB_SUCCESS && res->count == 1) {
521                                 int domain_controller_behaviour_version
522                                         = ldb_msg_find_attr_as_int(res->msgs[0], 
523                                                                    "msDS-Behavior-Version", -1);
524                                 if (domain_controller_behaviour_version != -1) {
525                                         int *val = talloc(ldb, int);
526                                         if (!val) {
527                                                 ldb_oom(ldb);
528                                                 talloc_free(mem_ctx);
529                                         return LDB_ERR_OPERATIONS_ERROR;
530                                         }
531                                         *val = domain_controller_behaviour_version;
532                                         ret = ldb_set_opaque(ldb, 
533                                                              "domainControllerFunctionality", val);
534                                         if (ret != LDB_SUCCESS) {
535                                                 talloc_free(mem_ctx);
536                                                 return ret;
537                                         }
538                                 }
539                         }
540                 }
541         }
542
543         talloc_free(mem_ctx);
544         
545         return LDB_SUCCESS;
546 }
547
548 static int rootdse_modify(struct ldb_module *module, struct ldb_request *req)
549 {
550         struct ldb_context *ldb;
551         struct ldb_result *ext_res;
552         int ret;
553         struct ldb_dn *schema_dn;
554         struct ldb_message_element *schemaUpdateNowAttr;
555         
556         /*
557                 If dn is not "" we should let it pass through
558         */
559         if (!ldb_dn_is_null(req->op.mod.message->dn)) {
560                 return ldb_next_request(module, req);
561         }
562
563         ldb = ldb_module_get_ctx(module);
564
565         /*
566                 dn is empty so check for schemaUpdateNow attribute
567                 "The type of modification and values specified in the LDAP modify operation do not matter." MSDN
568         */
569         schemaUpdateNowAttr = ldb_msg_find_element(req->op.mod.message, "schemaUpdateNow");
570         if (!schemaUpdateNowAttr) {
571                 return LDB_ERR_OPERATIONS_ERROR;
572         }
573
574         schema_dn = samdb_schema_dn(ldb);
575         if (!schema_dn) {
576                 ldb_reset_err_string(ldb);
577                 ldb_debug(ldb, LDB_DEBUG_WARNING,
578                           "rootdse_modify: no schema dn present: (skip ldb_extended call)\n");
579                 return ldb_next_request(module, req);
580         }
581
582         ret = ldb_extended(ldb, DSDB_EXTENDED_SCHEMA_UPDATE_NOW_OID, schema_dn, &ext_res);
583         if (ret != LDB_SUCCESS) {
584                 return LDB_ERR_OPERATIONS_ERROR;
585         }
586         
587         talloc_free(ext_res);
588         return ldb_request_done(req, ret);
589 }
590
591 _PUBLIC_ const struct ldb_module_ops ldb_rootdse_module_ops = {
592         .name                   = "rootdse",
593         .init_context   = rootdse_init,
594         .search         = rootdse_search,
595         .request                = rootdse_request,
596         .modify         = rootdse_modify
597 };