s4-dsdb: added new control DSDB_MODIFY_PARTIAL_REPLICA

[samba.git] / source4 / dsdb / samdb / ldb_modules / partition_init.c

diff --git a/source4/dsdb/samdb/ldb_modules/partition_init.c b/source4/dsdb/samdb/ldb_modules/partition_init.c
index 0dd3a472c2a39bf66885736f347e834b459afcf1..030f592d3a365b91ce2355eb4f272bd5c7ce156e 100644 (file)
--- a/source4/dsdb/samdb/ldb_modules/partition_init.c
+++ b/source4/dsdb/samdb/ldb_modules/partition_init.c
@@ -770,6 +770,18 @@ int partition_create(struct ldb_module *module, struct ldb_request *req)
                if (ret != LDB_SUCCESS) {
                        return ret;
                }
+
+               if (ldb_request_get_control(req, DSDB_CONTROL_PARTIAL_REPLICA)) {
+                       /* this new partition is a partial replica */
+                       ret = ldb_msg_add_empty(mod_msg, "partialReplica", LDB_FLAG_MOD_ADD, NULL);
+                       if (ret != LDB_SUCCESS) {
+                               return ret;
+                       }
+                       ret = ldb_msg_add_fmt(mod_msg, "partialReplica", "%s", ldb_dn_get_linearized(dn));
+                       if (ret != LDB_SUCCESS) {
+                               return ret;
+                       }
+               }
                
                /* Perform modify on @PARTITION record */
                ret = ldb_build_mod_req(&mod_req, ldb, req, mod_msg, NULL, NULL,