4 Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
5 Copyright (C) Simo Sorce 2004-2008
6 Copyright (C) Matthias Dieter Wallnöfer 2009
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.
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.
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/>.
25 * Component: ldb samldb module
27 * Description: add embedded user/group creation functionality
33 #include "libcli/ldap/ldap_ndr.h"
34 #include "ldb_module.h"
35 #include "dsdb/samdb/samdb.h"
36 #include "dsdb/samdb/ldb_modules/util.h"
37 #include "libcli/security/security.h"
38 #include "librpc/gen_ndr/ndr_security.h"
39 #include "../lib/util/util_ldb.h"
41 #include "param/param.h"
45 typedef int (*samldb_step_fn_t)(struct samldb_ctx *);
48 struct samldb_step *next;
53 struct ldb_module *module;
54 struct ldb_request *req;
56 /* used for add operations */
59 /* the resulting message */
60 struct ldb_message *msg;
62 /* holds the entry SID */
65 /* holds a generic dn */
68 /* used in conjunction with "sid" in "samldb_dn_from_sid" */
69 struct ldb_dn *res_dn;
71 /* used in conjunction with "dn" in "samldb_sid_from_dn" */
72 struct dom_sid *res_sid;
74 /* used in "samldb_user_dn_to_prim_group_rid" */
75 uint32_t prim_group_rid;
77 /* used in conjunction with "prim_group_rid" in
78 * "samldb_prim_group_rid_to_users_cnt" */
79 unsigned int users_cnt;
81 /* used in "samldb_group_add_member" and "samldb_group_del_member" */
82 struct ldb_dn *group_dn;
83 struct ldb_dn *member_dn;
85 /* used in "samldb_primary_group_change" */
86 struct ldb_dn *user_dn;
87 struct ldb_dn *old_prim_group_dn, *new_prim_group_dn;
89 /* generic counter - used in "samldb_member_check" */
92 /* all the async steps necessary to complete the operation */
93 struct samldb_step *steps;
94 struct samldb_step *curstep;
96 /* If someone set an ares to forward controls and response back to the caller */
97 struct ldb_reply *ares;
100 static struct samldb_ctx *samldb_ctx_init(struct ldb_module *module,
101 struct ldb_request *req)
103 struct ldb_context *ldb;
104 struct samldb_ctx *ac;
106 ldb = ldb_module_get_ctx(module);
108 ac = talloc_zero(req, struct samldb_ctx);
120 static int samldb_add_step(struct samldb_ctx *ac, samldb_step_fn_t fn)
122 struct samldb_step *step, *stepper;
124 step = talloc_zero(ac, struct samldb_step);
126 return LDB_ERR_OPERATIONS_ERROR;
131 if (ac->steps == NULL) {
135 if (ac->curstep == NULL)
136 return LDB_ERR_OPERATIONS_ERROR;
137 for (stepper = ac->curstep; stepper->next != NULL;
138 stepper = stepper->next);
139 stepper->next = step;
145 static int samldb_first_step(struct samldb_ctx *ac)
147 if (ac->steps == NULL) {
148 return LDB_ERR_OPERATIONS_ERROR;
151 ac->curstep = ac->steps;
152 return ac->curstep->fn(ac);
155 static int samldb_next_step(struct samldb_ctx *ac)
157 if (ac->curstep->next) {
158 ac->curstep = ac->curstep->next;
159 return ac->curstep->fn(ac);
162 /* we exit the samldb module here */
163 /* If someone set an ares to forward controls and response back to the caller, use them */
165 return ldb_module_done(ac->req, ac->ares->controls,
166 ac->ares->response, LDB_SUCCESS);
168 return ldb_module_done(ac->req, NULL, NULL, LDB_SUCCESS);
172 static int samldb_generate_samAccountName(struct ldb_message *msg)
176 /* Format: $000000-000000000000 */
178 name = talloc_asprintf(msg, "$%.6X-%.6X%.6X",
179 (unsigned int)generate_random(),
180 (unsigned int)generate_random(),
181 (unsigned int)generate_random());
183 return LDB_ERR_OPERATIONS_ERROR;
185 return ldb_msg_add_steal_string(msg, "samAccountName", name);
189 * samldb_check_samAccountName (async)
192 static int samldb_check_samAccountName_callback(struct ldb_request *req,
193 struct ldb_reply *ares)
195 struct samldb_ctx *ac;
198 ac = talloc_get_type(req->context, struct samldb_ctx);
200 if (ares->error != LDB_SUCCESS) {
201 return ldb_module_done(ac->req, ares->controls,
202 ares->response, ares->error);
205 switch (ares->type) {
206 case LDB_REPLY_ENTRY:
207 /* if we get an entry it means this samAccountName
209 return ldb_module_done(ac->req, NULL, NULL,
210 LDB_ERR_ENTRY_ALREADY_EXISTS);
212 case LDB_REPLY_REFERRAL:
213 /* this should not happen */
214 return ldb_module_done(ac->req, NULL, NULL,
215 LDB_ERR_OPERATIONS_ERROR);
218 /* not found, go on */
220 ret = samldb_next_step(ac);
224 if (ret != LDB_SUCCESS) {
225 return ldb_module_done(ac->req, NULL, NULL, ret);
231 static int samldb_check_samAccountName(struct samldb_ctx *ac)
233 struct ldb_context *ldb;
234 struct ldb_request *req;
239 ldb = ldb_module_get_ctx(ac->module);
241 if (ldb_msg_find_element(ac->msg, "samAccountName") == NULL) {
242 ret = samldb_generate_samAccountName(ac->msg);
243 if (ret != LDB_SUCCESS) {
248 name = ldb_msg_find_attr_as_string(ac->msg, "samAccountName", NULL);
250 return LDB_ERR_OPERATIONS_ERROR;
252 filter = talloc_asprintf(ac, "samAccountName=%s",
253 ldb_binary_encode_string(ac, name));
254 if (filter == NULL) {
255 return LDB_ERR_OPERATIONS_ERROR;
258 ret = ldb_build_search_req(&req, ldb, ac,
259 samdb_base_dn(ldb), LDB_SCOPE_SUBTREE,
262 ac, samldb_check_samAccountName_callback,
265 if (ret != LDB_SUCCESS) {
268 return ldb_next_request(ac->module, req);
272 static int samldb_check_samAccountType(struct samldb_ctx *ac)
274 struct ldb_context *ldb;
275 unsigned int account_type;
276 unsigned int group_type;
280 ldb = ldb_module_get_ctx(ac->module);
282 /* make sure sAMAccountType is not specified */
283 if (ldb_msg_find_element(ac->msg, "sAMAccountType") != NULL) {
284 ldb_asprintf_errstring(ldb,
285 "sAMAccountType must not be specified!");
286 return LDB_ERR_UNWILLING_TO_PERFORM;
289 if (strcmp("user", ac->type) == 0) {
290 uac = samdb_result_uint(ac->msg, "userAccountControl", 0);
292 ldb_asprintf_errstring(ldb,
293 "userAccountControl invalid!");
294 return LDB_ERR_UNWILLING_TO_PERFORM;
296 account_type = ds_uf2atype(uac);
297 ret = samdb_msg_add_uint(ldb,
301 if (ret != LDB_SUCCESS) {
306 if (strcmp("group", ac->type) == 0) {
308 group_type = samdb_result_uint(ac->msg, "groupType", 0);
309 if (group_type == 0) {
310 ldb_asprintf_errstring(ldb,
311 "groupType invalid!\n");
312 return LDB_ERR_UNWILLING_TO_PERFORM;
314 account_type = ds_gtype2atype(group_type);
315 ret = samdb_msg_add_uint(ldb,
319 if (ret != LDB_SUCCESS) {
325 return samldb_next_step(ac);
328 static bool samldb_msg_add_sid(struct ldb_message *msg,
330 const struct dom_sid *sid)
333 enum ndr_err_code ndr_err;
335 ndr_err = ndr_push_struct_blob(&v, msg, NULL, sid,
336 (ndr_push_flags_fn_t)ndr_push_dom_sid);
337 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
340 return (ldb_msg_add_value(msg, name, &v, NULL) == 0);
344 /* allocate a SID using our RID Set */
345 static int samldb_allocate_sid(struct samldb_ctx *ac)
349 struct ldb_context *ldb = ldb_module_get_ctx(ac->module);
351 ret = ridalloc_allocate_rid(ac->module, &rid);
352 if (ret != LDB_SUCCESS) {
356 ac->sid = dom_sid_add_rid(ac, samdb_domain_sid(ldb), rid);
357 if (ac->sid == NULL) {
358 ldb_module_oom(ac->module);
359 return LDB_ERR_OPERATIONS_ERROR;
362 if ( ! samldb_msg_add_sid(ac->msg, "objectSid", ac->sid)) {
363 return LDB_ERR_OPERATIONS_ERROR;
366 return samldb_next_step(ac);
370 * samldb_dn_from_sid (async)
373 static int samldb_dn_from_sid(struct samldb_ctx *ac);
375 static int samldb_dn_from_sid_callback(struct ldb_request *req,
376 struct ldb_reply *ares)
378 struct ldb_context *ldb;
379 struct samldb_ctx *ac;
382 ac = talloc_get_type(req->context, struct samldb_ctx);
383 ldb = ldb_module_get_ctx(ac->module);
386 ret = LDB_ERR_OPERATIONS_ERROR;
389 if (ares->error != LDB_SUCCESS) {
390 return ldb_module_done(ac->req, ares->controls,
391 ares->response, ares->error);
394 switch (ares->type) {
395 case LDB_REPLY_ENTRY:
397 if (ac->res_dn != NULL) {
399 ldb_set_errstring(ldb,
400 "Invalid number of results while searching "
401 "for domain objects!");
402 ret = LDB_ERR_OPERATIONS_ERROR;
405 ac->res_dn = ldb_dn_copy(ac, ares->message->dn);
411 case LDB_REPLY_REFERRAL:
420 /* found or not found, go on */
421 ret = samldb_next_step(ac);
426 if (ret != LDB_SUCCESS) {
427 return ldb_module_done(ac->req, NULL, NULL, ret);
433 /* Finds the DN "res_dn" of an object with a given SID "sid" */
434 static int samldb_dn_from_sid(struct samldb_ctx *ac)
436 struct ldb_context *ldb;
437 static const char * const attrs[] = { NULL };
438 struct ldb_request *req;
442 ldb = ldb_module_get_ctx(ac->module);
445 return LDB_ERR_OPERATIONS_ERROR;
447 filter = talloc_asprintf(ac, "(objectSid=%s)",
448 ldap_encode_ndr_dom_sid(ac, ac->sid));
450 return LDB_ERR_OPERATIONS_ERROR;
452 ret = ldb_build_search_req(&req, ldb, ac,
453 ldb_get_default_basedn(ldb),
457 ac, samldb_dn_from_sid_callback,
459 if (ret != LDB_SUCCESS)
462 return ldb_next_request(ac->module, req);
466 static int samldb_check_primaryGroupID_1(struct samldb_ctx *ac)
468 struct ldb_context *ldb;
471 ldb = ldb_module_get_ctx(ac->module);
473 rid = samdb_result_uint(ac->msg, "primaryGroupID", ~0);
474 ac->sid = dom_sid_add_rid(ac, samdb_domain_sid(ldb), rid);
476 return LDB_ERR_OPERATIONS_ERROR;
479 return samldb_next_step(ac);
482 static int samldb_check_primaryGroupID_2(struct samldb_ctx *ac)
484 if (ac->res_dn == NULL) {
485 struct ldb_context *ldb;
486 ldb = ldb_module_get_ctx(ac->module);
487 ldb_asprintf_errstring(ldb,
488 "Failed to find group sid %s!",
489 dom_sid_string(ac->sid, ac->sid));
490 return LDB_ERR_UNWILLING_TO_PERFORM;
493 return samldb_next_step(ac);
498 * samldb_set_defaultObjectCategory_callback (async)
501 static int samldb_set_defaultObjectCategory_callback(struct ldb_request *req,
502 struct ldb_reply *ares)
504 struct ldb_context *ldb;
505 struct samldb_ctx *ac;
508 ac = talloc_get_type(req->context, struct samldb_ctx);
509 ldb = ldb_module_get_ctx(ac->module);
512 ret = LDB_ERR_OPERATIONS_ERROR;
515 if (ares->error != LDB_SUCCESS) {
516 return ldb_module_done(ac->req, ares->controls,
517 ares->response, ares->error);
519 if (ares->type != LDB_REPLY_DONE) {
520 ldb_set_errstring(ldb,
521 "Invalid reply type!");
522 ret = LDB_ERR_OPERATIONS_ERROR;
526 ret = samldb_next_step(ac);
529 if (ret != LDB_SUCCESS) {
530 return ldb_module_done(ac->req, NULL, NULL, ret);
536 static int samldb_set_defaultObjectCategory(struct samldb_ctx *ac)
538 struct ldb_context *ldb;
539 struct ldb_message *msg;
540 struct ldb_request *req;
543 ldb = ldb_module_get_ctx(ac->module);
545 /* (Re)set the default object category to have it set to the DN in the
547 msg = ldb_msg_new(ac);
548 msg->dn = ac->msg->dn;
549 ldb_msg_add_empty(msg, "defaultObjectCategory",
550 LDB_FLAG_MOD_REPLACE, NULL);
551 ldb_msg_add_steal_string(msg, "defaultObjectCategory",
552 ldb_dn_alloc_linearized(msg, ac->dn));
554 ret = ldb_build_mod_req(&req, ldb, ac,
557 samldb_set_defaultObjectCategory_callback,
559 if (ret != LDB_SUCCESS) {
564 return ldb_next_request(ac->module, req);
568 * samldb_find_for_defaultObjectCategory (async)
571 static int samldb_find_for_defaultObjectCategory_callback(struct ldb_request *req,
572 struct ldb_reply *ares)
574 struct ldb_context *ldb;
575 struct samldb_ctx *ac;
578 ac = talloc_get_type(req->context, struct samldb_ctx);
579 ldb = ldb_module_get_ctx(ac->module);
582 ret = LDB_ERR_OPERATIONS_ERROR;
585 if (ares->error != LDB_SUCCESS) {
586 if (ares->error == LDB_ERR_NO_SUCH_OBJECT) {
587 if (ldb_request_get_control(ac->req,
588 LDB_CONTROL_RELAX_OID) != NULL) {
589 /* Don't be pricky when the DN doesn't exist */
590 /* if we have the RELAX control specified */
591 ac->dn = req->op.search.base;
592 return samldb_next_step(ac);
594 ldb_set_errstring(ldb,
595 "samldb_find_defaultObjectCategory: "
596 "Invalid DN for 'defaultObjectCategory'!");
597 ares->error = LDB_ERR_CONSTRAINT_VIOLATION;
601 return ldb_module_done(ac->req, ares->controls,
602 ares->response, ares->error);
605 switch (ares->type) {
606 case LDB_REPLY_ENTRY:
607 ac->dn = talloc_steal(ac, ares->message->dn);
612 case LDB_REPLY_REFERRAL:
613 /* this should not happen */
615 ret = LDB_ERR_OPERATIONS_ERROR;
621 if (ac->dn != NULL) {
622 /* when found go on */
623 ret = samldb_next_step(ac);
625 ret = LDB_ERR_OPERATIONS_ERROR;
631 if (ret != LDB_SUCCESS) {
632 return ldb_module_done(ac->req, NULL, NULL, ret);
638 static int samldb_find_for_defaultObjectCategory(struct samldb_ctx *ac)
640 struct ldb_context *ldb;
641 struct ldb_request *req;
642 static const char *no_attrs[] = { NULL };
644 const struct ldb_val *val;
645 struct ldb_dn *def_obj_cat_dn;
647 ldb = ldb_module_get_ctx(ac->module);
651 val = ldb_msg_find_ldb_val(ac->msg, "defaultObjectCategory");
653 /* "defaultObjectCategory" has been set by the caller. Do some
654 * checks for consistency.
655 * NOTE: The real constraint check (that 'defaultObjectCategory'
656 * is the DN of the new objectclass or any parent of it) is
658 * For now we say that 'defaultObjectCategory' is valid if it
659 * exists and it is of objectclass "classSchema". */
660 def_obj_cat_dn = ldb_dn_from_ldb_val(ac, ldb, val);
661 if (def_obj_cat_dn == NULL) {
662 ldb_set_errstring(ldb,
663 "samldb_find_defaultObjectCategory: Invalid DN "
664 "for 'defaultObjectCategory'!");
665 return LDB_ERR_CONSTRAINT_VIOLATION;
668 /* "defaultObjectCategory" has not been set by the caller. Use
669 * the entry DN for it. */
670 def_obj_cat_dn = ac->msg->dn;
673 ret = ldb_build_search_req(&req, ldb, ac,
674 def_obj_cat_dn, LDB_SCOPE_BASE,
675 "objectClass=classSchema", no_attrs,
677 ac, samldb_find_for_defaultObjectCategory_callback,
679 if (ret != LDB_SUCCESS) {
683 ret = dsdb_request_add_controls(ac->module, req,
684 DSDB_SEARCH_SHOW_DN_IN_STORAGE_FORMAT);
685 if (ret != LDB_SUCCESS) {
689 return ldb_next_request(ac->module, req);
694 * samldb_add_entry (async)
697 static int samldb_add_entry_callback(struct ldb_request *req,
698 struct ldb_reply *ares)
700 struct ldb_context *ldb;
701 struct samldb_ctx *ac;
704 ac = talloc_get_type(req->context, struct samldb_ctx);
705 ldb = ldb_module_get_ctx(ac->module);
708 return ldb_module_done(ac->req, NULL, NULL,
709 LDB_ERR_OPERATIONS_ERROR);
711 if (ares->error != LDB_SUCCESS) {
712 return ldb_module_done(ac->req, ares->controls,
713 ares->response, ares->error);
715 if (ares->type != LDB_REPLY_DONE) {
716 ldb_set_errstring(ldb,
717 "Invalid reply type!\n");
718 return ldb_module_done(ac->req, NULL, NULL,
719 LDB_ERR_OPERATIONS_ERROR);
722 /* The caller may wish to get controls back from the add */
723 ac->ares = talloc_steal(ac, ares);
725 ret = samldb_next_step(ac);
726 if (ret != LDB_SUCCESS) {
727 return ldb_module_done(ac->req, NULL, NULL, ret);
732 static int samldb_add_entry(struct samldb_ctx *ac)
734 struct ldb_context *ldb;
735 struct ldb_request *req;
738 ldb = ldb_module_get_ctx(ac->module);
740 ret = ldb_build_add_req(&req, ldb, ac,
743 ac, samldb_add_entry_callback,
745 if (ret != LDB_SUCCESS) {
749 return ldb_next_request(ac->module, req);
753 static int samldb_fill_object(struct samldb_ctx *ac, const char *type)
755 struct ldb_context *ldb;
756 struct loadparm_context *lp_ctx;
757 enum sid_generator sid_generator;
760 ldb = ldb_module_get_ctx(ac->module);
762 /* Add informations for the different account types */
764 if (strcmp(ac->type, "user") == 0) {
765 ret = samdb_find_or_add_attribute(ldb, ac->msg,
766 "userAccountControl", "546");
767 if (ret != LDB_SUCCESS) return ret;
768 ret = samdb_find_or_add_attribute(ldb, ac->msg,
770 if (ret != LDB_SUCCESS) return ret;
771 ret = samdb_find_or_add_attribute(ldb, ac->msg,
773 if (ret != LDB_SUCCESS) return ret;
774 ret = samdb_find_or_add_attribute(ldb, ac->msg,
776 if (ret != LDB_SUCCESS) return ret;
777 ret = samdb_find_or_add_attribute(ldb, ac->msg,
778 "badPasswordTime", "0");
779 if (ret != LDB_SUCCESS) return ret;
780 ret = samdb_find_or_add_attribute(ldb, ac->msg,
782 if (ret != LDB_SUCCESS) return ret;
783 ret = samdb_find_or_add_attribute(ldb, ac->msg,
785 if (ret != LDB_SUCCESS) return ret;
786 ret = samdb_find_or_add_attribute(ldb, ac->msg,
788 if (ret != LDB_SUCCESS) return ret;
789 ret = samdb_find_or_add_attribute(ldb, ac->msg,
790 "primaryGroupID", "513");
791 if (ret != LDB_SUCCESS) return ret;
792 ret = samdb_find_or_add_attribute(ldb, ac->msg,
793 "accountExpires", "9223372036854775807");
794 if (ret != LDB_SUCCESS) return ret;
795 ret = samdb_find_or_add_attribute(ldb, ac->msg,
797 if (ret != LDB_SUCCESS) return ret;
798 } else if (strcmp(ac->type, "group") == 0) {
799 ret = samdb_find_or_add_attribute(ldb, ac->msg,
800 "groupType", "-2147483646");
801 if (ret != LDB_SUCCESS) return ret;
802 } else if (strcmp(ac->type, "classSchema") == 0) {
803 const struct ldb_val *rdn_value;
805 ret = samdb_find_or_add_attribute(ldb, ac->msg,
807 if (ret != LDB_SUCCESS) return ret;
809 rdn_value = ldb_dn_get_rdn_val(ac->msg->dn);
810 if (!ldb_msg_find_element(ac->msg, "lDAPDisplayName")) {
811 /* the RDN has prefix "CN" */
812 ret = ldb_msg_add_string(ac->msg, "lDAPDisplayName",
813 samdb_cn_to_lDAPDisplayName(ac,
814 (const char *) rdn_value->data));
815 if (ret != LDB_SUCCESS) {
821 if (!ldb_msg_find_element(ac->msg, "schemaIDGUID")) {
824 guid = GUID_random();
825 ret = dsdb_msg_add_guid(ac->msg, &guid, "schemaIDGUID");
826 if (ret != LDB_SUCCESS) {
832 ret = samldb_add_step(ac, samldb_add_entry);
833 if (ret != LDB_SUCCESS) return ret;
835 ret = samldb_add_step(ac, samldb_find_for_defaultObjectCategory);
836 if (ret != LDB_SUCCESS) return ret;
838 ret = samldb_add_step(ac, samldb_set_defaultObjectCategory);
839 if (ret != LDB_SUCCESS) return ret;
841 return samldb_first_step(ac);
842 } else if (strcmp(ac->type, "attributeSchema") == 0) {
843 const struct ldb_val *rdn_value;
844 rdn_value = ldb_dn_get_rdn_val(ac->msg->dn);
845 if (!ldb_msg_find_element(ac->msg, "lDAPDisplayName")) {
846 /* the RDN has prefix "CN" */
847 ret = ldb_msg_add_string(ac->msg, "lDAPDisplayName",
848 samdb_cn_to_lDAPDisplayName(ac,
849 (const char *) rdn_value->data));
850 if (ret != LDB_SUCCESS) {
856 ret = samdb_find_or_add_attribute(ldb, ac->msg,
857 "isSingleValued", "FALSE");
858 if (ret != LDB_SUCCESS) return ret;
860 if (!ldb_msg_find_element(ac->msg, "schemaIDGUID")) {
863 guid = GUID_random();
864 ret = dsdb_msg_add_guid(ac->msg, &guid, "schemaIDGUID");
865 if (ret != LDB_SUCCESS) {
871 ret = samldb_add_step(ac, samldb_add_entry);
872 if (ret != LDB_SUCCESS) return ret;
874 return samldb_first_step(ac);
876 ldb_asprintf_errstring(ldb,
877 "Invalid entry type!");
878 return LDB_ERR_OPERATIONS_ERROR;
881 /* check if we have a valid samAccountName */
882 ret = samldb_add_step(ac, samldb_check_samAccountName);
883 if (ret != LDB_SUCCESS) return ret;
885 /* check account_type/group_type */
886 ret = samldb_add_step(ac, samldb_check_samAccountType);
887 if (ret != LDB_SUCCESS) return ret;
889 /* check if we have a valid primary group ID */
890 if (strcmp(ac->type, "user") == 0) {
891 ret = samldb_add_step(ac, samldb_check_primaryGroupID_1);
892 if (ret != LDB_SUCCESS) return ret;
893 ret = samldb_add_step(ac, samldb_dn_from_sid);
894 if (ret != LDB_SUCCESS) return ret;
895 ret = samldb_add_step(ac, samldb_check_primaryGroupID_2);
896 if (ret != LDB_SUCCESS) return ret;
899 lp_ctx = talloc_get_type(ldb_get_opaque(ldb, "loadparm"),
900 struct loadparm_context);
902 /* don't allow objectSID to be specified without the RELAX control */
903 ac->sid = samdb_result_dom_sid(ac, ac->msg, "objectSid");
904 if (ac->sid && !ldb_request_get_control(ac->req, LDB_CONTROL_RELAX_OID) &&
905 !dsdb_module_am_system(ac->module)) {
906 ldb_asprintf_errstring(ldb, "No SID may be specified in user/group creation for %s",
907 ldb_dn_get_linearized(ac->msg->dn));
908 return LDB_ERR_UNWILLING_TO_PERFORM;
912 sid_generator = lp_sid_generator(lp_ctx);
913 if (sid_generator == SID_GENERATOR_INTERNAL) {
914 ret = samldb_add_step(ac, samldb_allocate_sid);
915 if (ret != LDB_SUCCESS) return ret;
919 /* finally proceed with adding the entry */
920 ret = samldb_add_step(ac, samldb_add_entry);
921 if (ret != LDB_SUCCESS) return ret;
923 return samldb_first_step(ac);
926 static int samldb_fill_foreignSecurityPrincipal_object(struct samldb_ctx *ac)
928 struct ldb_context *ldb;
931 ldb = ldb_module_get_ctx(ac->module);
933 ac->sid = samdb_result_dom_sid(ac->msg, ac->msg, "objectSid");
934 if (ac->sid == NULL) {
935 ac->sid = dom_sid_parse_talloc(ac->msg,
936 (const char *)ldb_dn_get_rdn_val(ac->msg->dn)->data);
938 ldb_set_errstring(ldb,
939 "No valid SID found in "
940 "ForeignSecurityPrincipal CN!");
942 return LDB_ERR_CONSTRAINT_VIOLATION;
944 if ( ! samldb_msg_add_sid(ac->msg, "objectSid", ac->sid)) {
946 return LDB_ERR_OPERATIONS_ERROR;
950 /* finally proceed with adding the entry */
951 ret = samldb_add_step(ac, samldb_add_entry);
952 if (ret != LDB_SUCCESS) return ret;
954 return samldb_first_step(ac);
957 static int samldb_check_rdn(struct ldb_module *module, struct ldb_dn *dn)
959 struct ldb_context *ldb;
960 const char *rdn_name;
962 ldb = ldb_module_get_ctx(module);
963 rdn_name = ldb_dn_get_rdn_name(dn);
965 if (strcasecmp(rdn_name, "cn") != 0) {
966 ldb_asprintf_errstring(ldb,
967 "Bad RDN (%s=) for samldb object, "
968 "should be CN=!", rdn_name);
969 return LDB_ERR_CONSTRAINT_VIOLATION;
976 * samldb_sid_from_dn (async)
979 static int samldb_sid_from_dn(struct samldb_ctx *ac);
981 static int samldb_sid_from_dn_callback(struct ldb_request *req,
982 struct ldb_reply *ares)
984 struct ldb_context *ldb;
985 struct samldb_ctx *ac;
988 ac = talloc_get_type(req->context, struct samldb_ctx);
989 ldb = ldb_module_get_ctx(ac->module);
992 ret = LDB_ERR_OPERATIONS_ERROR;
995 if (ares->error != LDB_SUCCESS) {
996 return ldb_module_done(ac->req, ares->controls,
997 ares->response, ares->error);
1000 switch (ares->type) {
1001 case LDB_REPLY_ENTRY:
1003 if (ac->res_sid != NULL) {
1005 ldb_set_errstring(ldb,
1006 "Invalid number of results while searching "
1007 "for domain objects!");
1008 ret = LDB_ERR_OPERATIONS_ERROR;
1011 ac->res_sid = samdb_result_dom_sid(ac, ares->message,
1018 case LDB_REPLY_REFERRAL:
1024 case LDB_REPLY_DONE:
1027 /* found or not found, go on */
1028 ret = samldb_next_step(ac);
1033 if (ret != LDB_SUCCESS) {
1034 return ldb_module_done(ac->req, NULL, NULL, ret);
1040 /* Finds the SID "res_sid" of an object with a given DN "dn" */
1041 static int samldb_sid_from_dn(struct samldb_ctx *ac)
1043 struct ldb_context *ldb;
1044 static const char * const attrs[] = { "objectSid", NULL };
1045 struct ldb_request *req;
1048 ldb = ldb_module_get_ctx(ac->module);
1051 return LDB_ERR_OPERATIONS_ERROR;
1053 ret = ldb_build_search_req(&req, ldb, ac,
1058 ac, samldb_sid_from_dn_callback,
1060 if (ret != LDB_SUCCESS)
1063 return ldb_next_request(ac->module, req);
1067 * samldb_user_dn_to_prim_group_rid (async)
1070 static int samldb_user_dn_to_prim_group_rid(struct samldb_ctx *ac);
1072 static int samldb_user_dn_to_prim_group_rid_callback(struct ldb_request *req,
1073 struct ldb_reply *ares)
1075 struct ldb_context *ldb;
1076 struct samldb_ctx *ac;
1079 ac = talloc_get_type(req->context, struct samldb_ctx);
1080 ldb = ldb_module_get_ctx(ac->module);
1083 ret = LDB_ERR_OPERATIONS_ERROR;
1086 if (ares->error != LDB_SUCCESS) {
1087 return ldb_module_done(ac->req, ares->controls,
1088 ares->response, ares->error);
1091 switch (ares->type) {
1092 case LDB_REPLY_ENTRY:
1094 if (ac->prim_group_rid != 0) {
1096 ldb_set_errstring(ldb,
1097 "Invalid number of results while searching "
1098 "for domain objects!");
1099 ret = LDB_ERR_OPERATIONS_ERROR;
1102 ac->prim_group_rid = samdb_result_uint(ares->message,
1103 "primaryGroupID", ~0);
1109 case LDB_REPLY_REFERRAL:
1115 case LDB_REPLY_DONE:
1117 if (ac->prim_group_rid == 0) {
1118 ldb_asprintf_errstring(ldb,
1119 "Unable to get the primary group RID!");
1120 ret = LDB_ERR_OPERATIONS_ERROR;
1125 ret = samldb_next_step(ac);
1130 if (ret != LDB_SUCCESS) {
1131 return ldb_module_done(ac->req, NULL, NULL, ret);
1137 /* Locates the "primaryGroupID" attribute from a certain user specified as
1138 * "user_dn". Saves the result in "prim_group_rid". */
1139 static int samldb_user_dn_to_prim_group_rid(struct samldb_ctx *ac)
1141 struct ldb_context *ldb;
1142 static const char * const attrs[] = { "primaryGroupID", NULL };
1143 struct ldb_request *req;
1146 ldb = ldb_module_get_ctx(ac->module);
1148 if (ac->user_dn == NULL)
1149 return LDB_ERR_OPERATIONS_ERROR;
1151 ret = ldb_build_search_req(&req, ldb, ac,
1156 ac, samldb_user_dn_to_prim_group_rid_callback,
1158 if (ret != LDB_SUCCESS)
1161 return ldb_next_request(ac->module, req);
1165 * samldb_prim_group_rid_to_users_cnt (async)
1168 static int samldb_prim_group_rid_to_users_cnt(struct samldb_ctx *ac);
1170 static int samldb_prim_group_rid_to_users_cnt_callback(struct ldb_request *req,
1171 struct ldb_reply *ares)
1173 struct ldb_context *ldb;
1174 struct samldb_ctx *ac;
1177 ac = talloc_get_type(req->context, struct samldb_ctx);
1178 ldb = ldb_module_get_ctx(ac->module);
1181 ret = LDB_ERR_OPERATIONS_ERROR;
1184 if (ares->error != LDB_SUCCESS) {
1185 return ldb_module_done(ac->req, ares->controls,
1186 ares->response, ares->error);
1189 switch (ares->type) {
1190 case LDB_REPLY_ENTRY:
1198 case LDB_REPLY_REFERRAL:
1204 case LDB_REPLY_DONE:
1207 /* found or not found, go on */
1208 ret = samldb_next_step(ac);
1213 if (ret != LDB_SUCCESS) {
1214 return ldb_module_done(ac->req, NULL, NULL, ret);
1220 /* Finds the amount of users which have the primary group "prim_group_rid" and
1221 * save the result in "users_cnt" */
1222 static int samldb_prim_group_rid_to_users_cnt(struct samldb_ctx *ac)
1224 struct ldb_context *ldb;
1225 static const char * const attrs[] = { NULL };
1226 struct ldb_request *req;
1230 ldb = ldb_module_get_ctx(ac->module);
1232 if ((ac->prim_group_rid == 0) || (ac->users_cnt != 0))
1233 return LDB_ERR_OPERATIONS_ERROR;
1235 filter = talloc_asprintf(ac, "(&(primaryGroupID=%u)(objectclass=user))",
1236 ac->prim_group_rid);
1238 return LDB_ERR_OPERATIONS_ERROR;
1240 ret = ldb_build_search_req(&req, ldb, ac,
1241 ldb_get_default_basedn(ldb),
1246 samldb_prim_group_rid_to_users_cnt_callback,
1248 if (ret != LDB_SUCCESS)
1251 return ldb_next_request(ac->module, req);
1255 * samldb_group_add_member (async)
1256 * samldb_group_del_member (async)
1259 static int samldb_group_add_del_member_callback(struct ldb_request *req,
1260 struct ldb_reply *ares)
1262 struct ldb_context *ldb;
1263 struct samldb_ctx *ac;
1266 ac = talloc_get_type(req->context, struct samldb_ctx);
1267 ldb = ldb_module_get_ctx(ac->module);
1270 ret = LDB_ERR_OPERATIONS_ERROR;
1273 if (ares->error != LDB_SUCCESS) {
1274 if (ares->error == LDB_ERR_NO_SUCH_ATTRIBUTE) {
1275 /* On error "NO_SUCH_ATTRIBUTE" (delete of an invalid
1276 * "member" attribute) return "UNWILLING_TO_PERFORM" */
1277 ares->error = LDB_ERR_UNWILLING_TO_PERFORM;
1279 return ldb_module_done(ac->req, ares->controls,
1280 ares->response, ares->error);
1282 if (ares->type != LDB_REPLY_DONE) {
1283 ldb_set_errstring(ldb,
1284 "Invalid reply type!");
1285 ret = LDB_ERR_OPERATIONS_ERROR;
1289 ret = samldb_next_step(ac);
1292 if (ret != LDB_SUCCESS) {
1293 return ldb_module_done(ac->req, NULL, NULL, ret);
1299 /* Adds a member with DN "member_dn" to a group with DN "group_dn" */
1300 static int samldb_group_add_member(struct samldb_ctx *ac)
1302 struct ldb_context *ldb;
1303 struct ldb_request *req;
1304 struct ldb_message *msg;
1307 ldb = ldb_module_get_ctx(ac->module);
1309 if ((ac->group_dn == NULL) || (ac->member_dn == NULL))
1310 return LDB_ERR_OPERATIONS_ERROR;
1312 msg = ldb_msg_new(ac);
1313 msg->dn = ac->group_dn;
1314 samdb_msg_add_addval(ldb, ac, msg, "member",
1315 ldb_dn_get_linearized(ac->member_dn));
1317 ret = ldb_build_mod_req(&req, ldb, ac,
1319 ac, samldb_group_add_del_member_callback,
1321 if (ret != LDB_SUCCESS)
1324 return ldb_next_request(ac->module, req);
1327 /* Removes a member with DN "member_dn" from a group with DN "group_dn" */
1328 static int samldb_group_del_member(struct samldb_ctx *ac)
1330 struct ldb_context *ldb;
1331 struct ldb_request *req;
1332 struct ldb_message *msg;
1335 ldb = ldb_module_get_ctx(ac->module);
1337 if ((ac->group_dn == NULL) || (ac->member_dn == NULL))
1338 return LDB_ERR_OPERATIONS_ERROR;
1340 msg = ldb_msg_new(ac);
1341 msg->dn = ac->group_dn;
1342 samdb_msg_add_delval(ldb, ac, msg, "member",
1343 ldb_dn_get_linearized(ac->member_dn));
1345 ret = ldb_build_mod_req(&req, ldb, ac,
1347 ac, samldb_group_add_del_member_callback,
1349 if (ret != LDB_SUCCESS)
1352 return ldb_next_request(ac->module, req);
1356 static int samldb_prim_group_change_1(struct samldb_ctx *ac)
1358 struct ldb_context *ldb;
1361 ldb = ldb_module_get_ctx(ac->module);
1363 ac->user_dn = ac->msg->dn;
1365 rid = samdb_result_uint(ac->msg, "primaryGroupID", ~0);
1366 ac->sid = dom_sid_add_rid(ac, samdb_domain_sid(ldb), rid);
1367 if (ac->sid == NULL)
1368 return LDB_ERR_OPERATIONS_ERROR;
1371 ac->prim_group_rid = 0;
1373 return samldb_next_step(ac);
1376 static int samldb_prim_group_change_2(struct samldb_ctx *ac)
1378 struct ldb_context *ldb;
1380 ldb = ldb_module_get_ctx(ac->module);
1382 if (ac->res_dn != NULL)
1383 ac->new_prim_group_dn = ac->res_dn;
1385 return LDB_ERR_UNWILLING_TO_PERFORM;
1387 ac->sid = dom_sid_add_rid(ac, samdb_domain_sid(ldb),
1388 ac->prim_group_rid);
1389 if (ac->sid == NULL)
1390 return LDB_ERR_OPERATIONS_ERROR;
1393 return samldb_next_step(ac);
1396 static int samldb_prim_group_change_4(struct samldb_ctx *ac);
1397 static int samldb_prim_group_change_5(struct samldb_ctx *ac);
1398 static int samldb_prim_group_change_6(struct samldb_ctx *ac);
1400 static int samldb_prim_group_change_3(struct samldb_ctx *ac)
1404 if (ac->res_dn != NULL)
1405 ac->old_prim_group_dn = ac->res_dn;
1407 return LDB_ERR_UNWILLING_TO_PERFORM;
1409 /* Only update when the primary group changed */
1410 if (ldb_dn_compare(ac->old_prim_group_dn, ac->new_prim_group_dn) != 0) {
1411 ac->member_dn = ac->user_dn;
1412 /* Remove the "member" attribute of the actual (new) primary
1415 ret = samldb_add_step(ac, samldb_prim_group_change_4);
1416 if (ret != LDB_SUCCESS) return ret;
1418 ret = samldb_add_step(ac, samldb_group_del_member);
1419 if (ret != LDB_SUCCESS) return ret;
1421 /* Add a "member" attribute for the previous primary group */
1423 ret = samldb_add_step(ac, samldb_prim_group_change_5);
1424 if (ret != LDB_SUCCESS) return ret;
1426 ret = samldb_add_step(ac, samldb_group_add_member);
1427 if (ret != LDB_SUCCESS) return ret;
1430 ret = samldb_add_step(ac, samldb_prim_group_change_6);
1431 if (ret != LDB_SUCCESS) return ret;
1433 return samldb_next_step(ac);
1436 static int samldb_prim_group_change_4(struct samldb_ctx *ac)
1438 ac->group_dn = ac->new_prim_group_dn;
1440 return samldb_next_step(ac);
1443 static int samldb_prim_group_change_5(struct samldb_ctx *ac)
1445 ac->group_dn = ac->old_prim_group_dn;
1447 return samldb_next_step(ac);
1450 static int samldb_prim_group_change_6(struct samldb_ctx *ac)
1452 return ldb_next_request(ac->module, ac->req);
1455 static int samldb_prim_group_change(struct samldb_ctx *ac)
1459 /* Finds out the DN of the new primary group */
1461 ret = samldb_add_step(ac, samldb_prim_group_change_1);
1462 if (ret != LDB_SUCCESS) return ret;
1464 ret = samldb_add_step(ac, samldb_dn_from_sid);
1465 if (ret != LDB_SUCCESS) return ret;
1467 ret = samldb_add_step(ac, samldb_user_dn_to_prim_group_rid);
1468 if (ret != LDB_SUCCESS) return ret;
1470 /* Finds out the DN of the old primary group */
1472 ret = samldb_add_step(ac, samldb_prim_group_change_2);
1473 if (ret != LDB_SUCCESS) return ret;
1475 ret = samldb_add_step(ac, samldb_dn_from_sid);
1476 if (ret != LDB_SUCCESS) return ret;
1478 ret = samldb_add_step(ac, samldb_prim_group_change_3);
1479 if (ret != LDB_SUCCESS) return ret;
1481 return samldb_first_step(ac);
1485 static int samldb_member_check_1(struct samldb_ctx *ac)
1487 struct ldb_context *ldb;
1488 struct ldb_message_element *el;
1490 ldb = ldb_module_get_ctx(ac->module);
1492 el = ldb_msg_find_element(ac->msg, "member");
1494 ac->user_dn = ldb_dn_from_ldb_val(ac, ldb, &el->values[ac->cnt]);
1495 if (!ldb_dn_validate(ac->user_dn))
1496 return LDB_ERR_OPERATIONS_ERROR;
1497 ac->prim_group_rid = 0;
1499 return samldb_next_step(ac);
1502 static int samldb_member_check_2(struct samldb_ctx *ac)
1504 struct ldb_context *ldb;
1506 ldb = ldb_module_get_ctx(ac->module);
1508 ac->sid = dom_sid_add_rid(ac, samdb_domain_sid(ldb),
1509 ac->prim_group_rid);
1510 if (ac->sid == NULL)
1511 return LDB_ERR_OPERATIONS_ERROR;
1514 return samldb_next_step(ac);
1517 static int samldb_member_check_3(struct samldb_ctx *ac)
1519 if (ldb_dn_compare(ac->res_dn, ac->msg->dn) == 0)
1520 return LDB_ERR_ENTRY_ALREADY_EXISTS;
1524 return samldb_next_step(ac);
1527 static int samldb_member_check_4(struct samldb_ctx *ac)
1529 return ldb_next_request(ac->module, ac->req);
1532 static int samldb_member_check(struct samldb_ctx *ac)
1534 struct ldb_message_element *el;
1537 el = ldb_msg_find_element(ac->msg, "member");
1539 for (i = 0; i < el->num_values; i++) {
1540 /* Denies to add "member"s to groups which are primary ones
1542 ret = samldb_add_step(ac, samldb_member_check_1);
1543 if (ret != LDB_SUCCESS) return ret;
1545 ret = samldb_add_step(ac, samldb_user_dn_to_prim_group_rid);
1546 if (ret != LDB_SUCCESS) return ret;
1548 ret = samldb_add_step(ac, samldb_member_check_2);
1549 if (ret != LDB_SUCCESS) return ret;
1551 ret = samldb_add_step(ac, samldb_dn_from_sid);
1552 if (ret != LDB_SUCCESS) return ret;
1554 ret = samldb_add_step(ac, samldb_member_check_3);
1555 if (ret != LDB_SUCCESS) return ret;
1558 ret = samldb_add_step(ac, samldb_member_check_4);
1559 if (ret != LDB_SUCCESS) return ret;
1561 return samldb_first_step(ac);
1565 static int samldb_prim_group_users_check_1(struct samldb_ctx *ac)
1567 ac->dn = ac->req->op.del.dn;
1570 return samldb_next_step(ac);
1573 static int samldb_prim_group_users_check_2(struct samldb_ctx *ac)
1578 if (ac->res_sid == NULL) {
1579 /* No SID - therefore ok here */
1580 return ldb_next_request(ac->module, ac->req);
1582 status = dom_sid_split_rid(ac, ac->res_sid, NULL, &rid);
1583 if (!NT_STATUS_IS_OK(status))
1584 return LDB_ERR_OPERATIONS_ERROR;
1587 /* Special object (security principal?) */
1588 return ldb_next_request(ac->module, ac->req);
1591 ac->prim_group_rid = rid;
1594 return samldb_next_step(ac);
1597 static int samldb_prim_group_users_check_3(struct samldb_ctx *ac)
1599 if (ac->users_cnt > 0)
1600 return LDB_ERR_ENTRY_ALREADY_EXISTS;
1602 return ldb_next_request(ac->module, ac->req);
1605 static int samldb_prim_group_users_check(struct samldb_ctx *ac)
1609 /* Finds out the SID/RID of the domain object */
1611 ret = samldb_add_step(ac, samldb_prim_group_users_check_1);
1612 if (ret != LDB_SUCCESS) return ret;
1614 ret = samldb_add_step(ac, samldb_sid_from_dn);
1615 if (ret != LDB_SUCCESS) return ret;
1617 /* Deny delete requests from groups which are primary ones */
1619 ret = samldb_add_step(ac, samldb_prim_group_users_check_2);
1620 if (ret != LDB_SUCCESS) return ret;
1622 ret = samldb_add_step(ac, samldb_prim_group_rid_to_users_cnt);
1623 if (ret != LDB_SUCCESS) return ret;
1625 ret = samldb_add_step(ac, samldb_prim_group_users_check_3);
1626 if (ret != LDB_SUCCESS) return ret;
1628 return samldb_first_step(ac);
1633 static int samldb_add(struct ldb_module *module, struct ldb_request *req)
1635 struct ldb_context *ldb;
1636 struct samldb_ctx *ac;
1639 ldb = ldb_module_get_ctx(module);
1640 ldb_debug(ldb, LDB_DEBUG_TRACE, "samldb_add\n");
1642 /* do not manipulate our control entries */
1643 if (ldb_dn_is_special(req->op.add.message->dn)) {
1644 return ldb_next_request(module, req);
1647 ac = samldb_ctx_init(module, req);
1649 return LDB_ERR_OPERATIONS_ERROR;
1652 /* build the new msg */
1653 ac->msg = ldb_msg_copy(ac, ac->req->op.add.message);
1656 ldb_debug(ldb, LDB_DEBUG_FATAL,
1657 "samldb_add: ldb_msg_copy failed!\n");
1658 return LDB_ERR_OPERATIONS_ERROR;
1661 if (samdb_find_attribute(ldb, ac->msg,
1662 "objectclass", "computer") != NULL) {
1664 /* make sure the computer object also has the 'user'
1665 * objectclass so it will be handled by the next call */
1666 ret = samdb_find_or_add_value(ldb, ac->msg,
1667 "objectclass", "user");
1668 if (ret != LDB_SUCCESS) {
1674 if (samdb_find_attribute(ldb, ac->msg,
1675 "objectclass", "user") != NULL) {
1677 ret = samldb_check_rdn(module, ac->req->op.add.message->dn);
1678 if (ret != LDB_SUCCESS) {
1683 return samldb_fill_object(ac, "user");
1686 if (samdb_find_attribute(ldb, ac->msg,
1687 "objectclass", "group") != NULL) {
1689 ret = samldb_check_rdn(module, ac->req->op.add.message->dn);
1690 if (ret != LDB_SUCCESS) {
1695 return samldb_fill_object(ac, "group");
1698 /* perhaps a foreignSecurityPrincipal? */
1699 if (samdb_find_attribute(ldb, ac->msg,
1701 "foreignSecurityPrincipal") != NULL) {
1703 ret = samldb_check_rdn(module, ac->req->op.add.message->dn);
1704 if (ret != LDB_SUCCESS) {
1709 return samldb_fill_foreignSecurityPrincipal_object(ac);
1712 if (samdb_find_attribute(ldb, ac->msg,
1713 "objectclass", "classSchema") != NULL) {
1715 ret = samldb_check_rdn(module, ac->req->op.add.message->dn);
1716 if (ret != LDB_SUCCESS) {
1721 return samldb_fill_object(ac, "classSchema");
1724 if (samdb_find_attribute(ldb, ac->msg,
1725 "objectclass", "attributeSchema") != NULL) {
1727 ret = samldb_check_rdn(module, ac->req->op.add.message->dn);
1728 if (ret != LDB_SUCCESS) {
1733 return samldb_fill_object(ac, "attributeSchema");
1738 /* nothing matched, go on */
1739 return ldb_next_request(module, req);
1743 static int samldb_modify(struct ldb_module *module, struct ldb_request *req)
1745 struct ldb_context *ldb;
1746 struct ldb_message *msg;
1747 struct ldb_message_element *el, *el2;
1749 uint32_t account_type;
1751 if (ldb_dn_is_special(req->op.mod.message->dn)) {
1752 /* do not manipulate our control entries */
1753 return ldb_next_request(module, req);
1756 ldb = ldb_module_get_ctx(module);
1758 if (ldb_msg_find_element(req->op.mod.message, "sAMAccountType") != NULL) {
1759 ldb_asprintf_errstring(ldb,
1760 "sAMAccountType must not be specified!");
1761 return LDB_ERR_UNWILLING_TO_PERFORM;
1764 /* TODO: do not modify original request, create a new one */
1766 el = ldb_msg_find_element(req->op.mod.message, "groupType");
1767 if (el && el->flags & (LDB_FLAG_MOD_ADD|LDB_FLAG_MOD_REPLACE) && el->num_values == 1) {
1768 uint32_t group_type;
1770 req->op.mod.message = msg = ldb_msg_copy_shallow(req,
1771 req->op.mod.message);
1773 group_type = strtoul((const char *)el->values[0].data, NULL, 0);
1774 account_type = ds_gtype2atype(group_type);
1775 ret = samdb_msg_add_uint(ldb, msg, msg,
1778 if (ret != LDB_SUCCESS) {
1781 el2 = ldb_msg_find_element(msg, "sAMAccountType");
1782 el2->flags = LDB_FLAG_MOD_REPLACE;
1785 el = ldb_msg_find_element(req->op.mod.message, "userAccountControl");
1786 if (el && el->flags & (LDB_FLAG_MOD_ADD|LDB_FLAG_MOD_REPLACE) && el->num_values == 1) {
1787 uint32_t user_account_control;
1789 req->op.mod.message = msg = ldb_msg_copy_shallow(req,
1790 req->op.mod.message);
1792 user_account_control = strtoul((const char *)el->values[0].data,
1794 account_type = ds_uf2atype(user_account_control);
1795 ret = samdb_msg_add_uint(ldb, msg, msg,
1798 if (ret != LDB_SUCCESS) {
1801 el2 = ldb_msg_find_element(msg, "sAMAccountType");
1802 el2->flags = LDB_FLAG_MOD_REPLACE;
1804 if (user_account_control & UF_SERVER_TRUST_ACCOUNT) {
1805 ret = samdb_msg_add_string(ldb, msg, msg,
1806 "isCriticalSystemObject", "TRUE");
1807 if (ret != LDB_SUCCESS) {
1810 el2 = ldb_msg_find_element(msg, "isCriticalSystemObject");
1811 el2->flags = LDB_FLAG_MOD_REPLACE;
1815 el = ldb_msg_find_element(req->op.mod.message, "primaryGroupID");
1816 if (el && el->flags & (LDB_FLAG_MOD_ADD|LDB_FLAG_MOD_REPLACE) && el->num_values == 1) {
1817 struct samldb_ctx *ac;
1819 ac = samldb_ctx_init(module, req);
1821 return LDB_ERR_OPERATIONS_ERROR;
1823 req->op.mod.message = ac->msg = ldb_msg_copy_shallow(req,
1824 req->op.mod.message);
1826 return samldb_prim_group_change(ac);
1829 el = ldb_msg_find_element(req->op.mod.message, "member");
1830 if (el && el->flags & (LDB_FLAG_MOD_ADD|LDB_FLAG_MOD_REPLACE) && el->num_values == 1) {
1831 struct samldb_ctx *ac;
1833 ac = samldb_ctx_init(module, req);
1835 return LDB_ERR_OPERATIONS_ERROR;
1837 req->op.mod.message = ac->msg = ldb_msg_copy_shallow(req,
1838 req->op.mod.message);
1840 return samldb_member_check(ac);
1843 /* nothing matched, go on */
1844 return ldb_next_request(module, req);
1848 static int samldb_delete(struct ldb_module *module, struct ldb_request *req)
1850 struct samldb_ctx *ac;
1852 if (ldb_dn_is_special(req->op.del.dn)) {
1853 /* do not manipulate our control entries */
1854 return ldb_next_request(module, req);
1857 ac = samldb_ctx_init(module, req);
1859 return LDB_ERR_OPERATIONS_ERROR;
1861 return samldb_prim_group_users_check(ac);
1864 static int samldb_extended_allocate_rid_pool(struct ldb_module *module, struct ldb_request *req)
1866 struct ldb_context *ldb = ldb_module_get_ctx(module);
1867 struct dsdb_fsmo_extended_op *exop;
1870 exop = talloc_get_type(req->op.extended.data, struct dsdb_fsmo_extended_op);
1872 ldb_debug(ldb, LDB_DEBUG_FATAL, "samldb_extended_allocate_rid_pool: invalid extended data\n");
1873 return LDB_ERR_PROTOCOL_ERROR;
1876 ret = ridalloc_allocate_rid_pool_fsmo(module, exop);
1877 if (ret != LDB_SUCCESS) {
1881 return ldb_module_done(req, NULL, NULL, LDB_SUCCESS);
1884 static int samldb_extended(struct ldb_module *module, struct ldb_request *req)
1886 if (strcmp(req->op.extended.oid, DSDB_EXTENDED_ALLOCATE_RID_POOL) == 0) {
1887 return samldb_extended_allocate_rid_pool(module, req);
1890 return ldb_next_request(module, req);
1894 _PUBLIC_ const struct ldb_module_ops ldb_samldb_module_ops = {
1897 .modify = samldb_modify,
1898 .del = samldb_delete,
1899 .extended = samldb_extended