s4-dsdb: ensure that new partitions inherit any transaction
[ira/wip.git] / source4 / dsdb / samdb / ldb_modules / partition_init.c
1 /* 
2    Partitions ldb module
3
4    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
5    Copyright (C) Stefan Metzmacher <metze@samba.org> 2007
6
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 3 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 */
20
21 /*
22  *  Name: ldb
23  *
24  *  Component: ldb partitions module
25  *
26  *  Description: Implement LDAP partitions
27  *
28  *  Author: Andrew Bartlett
29  *  Author: Stefan Metzmacher
30  */
31
32 #include "dsdb/samdb/ldb_modules/partition.h"
33 static int partition_sort_compare(const void *v1, const void *v2)
34 {
35         const struct dsdb_partition *p1;
36         const struct dsdb_partition *p2;
37
38         p1 = *((struct dsdb_partition * const*)v1);
39         p2 = *((struct dsdb_partition * const*)v2);
40
41         return ldb_dn_compare(p1->ctrl->dn, p2->ctrl->dn);
42 }
43
44 /* Load the list of DNs that we must replicate to all partitions */
45 static int partition_load_replicate_dns(struct ldb_context *ldb, struct partition_private_data *data, struct ldb_message *msg) 
46 {
47         struct ldb_message_element *replicate_attributes = ldb_msg_find_element(msg, "replicateEntries");
48
49         talloc_free(data->replicate);
50         if (!replicate_attributes) {
51                 data->replicate = NULL;
52         } else {
53                 int i;
54                 data->replicate = talloc_array(data, struct ldb_dn *, replicate_attributes->num_values + 1);
55                 if (!data->replicate) {
56                         return LDB_ERR_OPERATIONS_ERROR;
57                 }
58
59                 for (i=0; i < replicate_attributes->num_values; i++) {
60                         data->replicate[i] = ldb_dn_from_ldb_val(data->replicate, ldb, &replicate_attributes->values[i]);
61                         if (!ldb_dn_validate(data->replicate[i])) {
62                                 ldb_asprintf_errstring(ldb,
63                                                         "partition_init: "
64                                                         "invalid DN in partition replicate record: %s", 
65                                                         replicate_attributes->values[i].data);
66                                 return LDB_ERR_CONSTRAINT_VIOLATION;
67                         }
68                 }
69                 data->replicate[i] = NULL;
70         }
71         return LDB_SUCCESS;
72 }
73
74 /* Load the list of modules for the partitions */
75 static int partition_load_modules(struct ldb_context *ldb, 
76                                   struct partition_private_data *data, struct ldb_message *msg) 
77 {
78         int i;
79         struct ldb_message_element *modules_attributes = ldb_msg_find_element(msg, "modules");
80         talloc_free(data->modules);
81         if (!modules_attributes) {
82                 return LDB_SUCCESS;
83         }
84         
85         data->modules = talloc_array(data, struct partition_module *, modules_attributes->num_values + 1);
86         if (!data->modules) {
87                 ldb_oom(ldb);
88                 return LDB_ERR_OPERATIONS_ERROR;
89         }
90         
91         for (i=0; i < modules_attributes->num_values; i++) {
92                 char *p;
93                 DATA_BLOB dn_blob;
94                 data->modules[i] = talloc(data->modules, struct partition_module);
95                 if (!data->modules[i]) {
96                         ldb_oom(ldb);
97                         return LDB_ERR_OPERATIONS_ERROR;
98                 }
99
100                 dn_blob = modules_attributes->values[i];
101
102                 p = strchr((const char *)dn_blob.data, ':');
103                 if (!p) {
104                         ldb_asprintf_errstring(ldb, 
105                                                "partition_load_modules: "
106                                                "invalid form for partition module record (missing ':'): %s", (const char *)dn_blob.data);
107                         return LDB_ERR_CONSTRAINT_VIOLATION;
108                 }
109                 /* Now trim off the filename */
110                 dn_blob.length = ((uint8_t *)p - dn_blob.data);
111
112                 p++;
113                 data->modules[i]->modules = ldb_modules_list_from_string(ldb, data->modules[i],
114                                                                          p);
115                 
116                 if (dn_blob.length == 1 && dn_blob.data[0] == '*') {
117                         data->modules[i]->dn = NULL;
118                 } else {
119                         data->modules[i]->dn = ldb_dn_from_ldb_val(data->modules[i], ldb, &dn_blob);
120                         if (!data->modules[i]->dn || !ldb_dn_validate(data->modules[i]->dn)) {
121                                 return LDB_ERR_OPERATIONS_ERROR;
122                         }
123                 }
124         }
125         data->modules[i] = NULL;
126         return LDB_SUCCESS;
127 }
128
129 static int partition_reload_metadata(struct ldb_module *module, struct partition_private_data *data, TALLOC_CTX *mem_ctx, struct ldb_message **_msg) 
130 {
131         int ret;
132         struct ldb_message *msg;
133         struct ldb_result *res;
134         struct ldb_context *ldb = ldb_module_get_ctx(module);
135         const char *attrs[] = { "partition", "replicateEntries", "modules", NULL };
136         /* perform search for @PARTITION, looking for module, replicateEntries and ldapBackend */
137         ret = dsdb_module_search_dn(module, mem_ctx, &res, 
138                                     ldb_dn_new(mem_ctx, ldb, DSDB_PARTITION_DN),
139                                     attrs);
140         if (ret != LDB_SUCCESS) {
141                 return ret;
142         }
143
144         msg = res->msgs[0];
145
146         ret = partition_load_replicate_dns(ldb, data, msg);
147         if (ret != LDB_SUCCESS) {
148                 return ret;
149         }
150
151         ret = partition_load_modules(ldb, data, msg);                   
152         if (ret != LDB_SUCCESS) {
153                 return ret;
154         }
155
156         data->ldapBackend = talloc_steal(data, ldb_msg_find_attr_as_string(msg, "ldapBackend", NULL));
157         if (_msg) {
158                 *_msg = msg;
159         } else {
160                 talloc_free(msg);
161         }
162
163         return LDB_SUCCESS;
164 }
165
166 static const char **find_modules_for_dn(struct partition_private_data *data, struct ldb_dn *dn) 
167 {
168         int i;
169         struct partition_module *default_mod = NULL;
170         for (i=0; data->modules && data->modules[i]; i++) {
171                 if (!data->modules[i]->dn) {
172                         default_mod = data->modules[i];
173                 } else if (ldb_dn_compare(dn, data->modules[i]->dn) == 0) {
174                         return data->modules[i]->modules;
175                 }
176         }
177         if (default_mod) {
178                 return default_mod->modules;
179         } else {
180                 return NULL;
181         }
182 }
183
184 static int new_partition_from_dn(struct ldb_context *ldb, struct partition_private_data *data, 
185                                  TALLOC_CTX *mem_ctx, 
186                                  struct ldb_dn *dn, const char *filename_base,
187                                  struct dsdb_partition **partition) {
188         const char *backend_url;
189         struct dsdb_control_current_partition *ctrl;
190         struct ldb_module *backend_module;
191         const char **modules;
192         int ret;
193
194         (*partition) = talloc(mem_ctx, struct dsdb_partition);
195         if (!*partition) {
196                 return LDB_ERR_OPERATIONS_ERROR;
197         }
198
199         (*partition)->ctrl = ctrl = talloc((*partition), struct dsdb_control_current_partition);
200         if (!ctrl) {
201                 talloc_free(*partition);
202                 ldb_oom(ldb);
203                 return LDB_ERR_OPERATIONS_ERROR;
204         }
205
206         /* See if an LDAP backend has been specified */
207         if (data->ldapBackend) {
208                 backend_url = data->ldapBackend;
209         } else {
210
211                 /* the backend LDB is the DN (base64 encoded if not 'plain') followed by .ldb */
212                 const char *p;
213                 char *backend_path;
214                 char *base64_dn = NULL;
215                 for (p = filename_base; *p; p++) {
216                         /* We have such a strict check because I don't want shell metacharacters in the file name, nor ../ */
217                         if (!(isalnum(*p) || *p == ' ' || *p == '=' || *p == ',')) {
218                                 break;
219                         }
220                 }
221                 if (*p) {
222                         filename_base = base64_dn = ldb_base64_encode(data, filename_base, strlen(filename_base));
223                 }
224                 
225                 backend_path = samdb_relative_path(ldb, 
226                                                    *partition, 
227                                                    filename_base);
228                 if (base64_dn) {
229                         talloc_free(base64_dn);
230                 }
231                 if (!backend_path) {
232                         ldb_asprintf_errstring(ldb, 
233                                                "partition_init: unable to determine an relative path for partition: %s", filename_base);
234                         talloc_free(*partition);
235                         return LDB_ERR_OPERATIONS_ERROR;                
236                 }
237                 backend_url = talloc_asprintf(*partition, "tdb://%s.ldb", 
238                                               backend_path); 
239                 talloc_free(backend_path);
240                 if (!backend_url) {
241                         ldb_oom(ldb);
242                         talloc_free(*partition);
243                         return LDB_ERR_OPERATIONS_ERROR;                
244                 }
245         }
246
247         (*partition)->backend_url = backend_url;
248         ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
249         ctrl->dn = talloc_steal(ctrl, dn);
250         
251         ret = ldb_connect_backend(ldb, backend_url, NULL, &backend_module);
252         if (ret != LDB_SUCCESS) {
253                 return ret;
254         }
255         talloc_steal((*partition), backend_module);
256
257         modules = find_modules_for_dn(data, dn);
258
259         if (!modules) {
260                 DEBUG(0, ("Unable to load partition modules for new DN %s, perhaps you need to reprovision?  See partition-upgrade.txt for instructions\n", ldb_dn_get_linearized(dn)));
261                 talloc_free(*partition);
262                 return LDB_ERR_CONSTRAINT_VIOLATION;
263         }
264         ret = ldb_load_modules_list(ldb, modules, backend_module, &(*partition)->module);
265         if (ret != LDB_SUCCESS) {
266                 ldb_asprintf_errstring(ldb, 
267                                        "partition_init: "
268                                        "loading backend for %s failed: %s", 
269                                        ldb_dn_get_linearized(dn), ldb_errstring(ldb));
270                 talloc_free(*partition);
271                 return ret;
272         }
273         ret = ldb_init_module_chain(ldb, (*partition)->module);
274         if (ret != LDB_SUCCESS) {
275                 ldb_asprintf_errstring(ldb,
276                                        "partition_init: "
277                                        "initialising backend for %s failed: %s", 
278                                        ldb_dn_get_linearized(dn), ldb_errstring(ldb));
279                 talloc_free(*partition);
280                 return ret;
281         }
282
283         talloc_steal((*partition), (*partition)->module);
284
285         /* if we were in a transaction then we need to start a
286            transaction on this new partition, otherwise we'll get a
287            transaction mismatch when we end the transaction */
288         if (data->in_transaction) {
289                 struct ldb_module *next = (*partition)->module;
290                 PARTITION_FIND_OP(next, start_transaction);
291
292                 ret = next->ops->start_transaction(next);
293         }
294
295         return ret;
296 }
297
298 /* Tell the rootDSE about the new partition */
299 static int partition_register(struct ldb_context *ldb, struct dsdb_control_current_partition *ctrl) 
300 {
301         struct ldb_request *req;
302         int ret;
303
304         req = talloc_zero(NULL, struct ldb_request);
305         if (req == NULL) {
306                 ldb_oom(ldb);
307                 return LDB_ERR_OPERATIONS_ERROR;
308         }
309                 
310         req->operation = LDB_REQ_REGISTER_PARTITION;
311         req->op.reg_partition.dn = ctrl->dn;
312         req->callback = ldb_op_default_callback;
313
314         ldb_set_timeout(ldb, req, 0);
315         
316         req->handle = ldb_handle_new(req, ldb);
317         if (req->handle == NULL) {
318                 talloc_free(req);
319                 return LDB_ERR_OPERATIONS_ERROR;
320         }
321         
322         ret = ldb_request(ldb, req);
323         if (ret == LDB_SUCCESS) {
324                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
325         }
326         if (ret != LDB_SUCCESS) {
327                 ldb_debug(ldb, LDB_DEBUG_ERROR, "partition: Unable to register partition with rootdse!\n");
328                 talloc_free(req);
329                 return LDB_ERR_OTHER;
330         }
331         talloc_free(req);
332
333         return LDB_SUCCESS;
334 }
335
336 /* Add a newly found partition to the global data */
337 static int add_partition_to_data(struct ldb_context *ldb, struct partition_private_data *data,
338                                  struct dsdb_partition *partition)
339 {
340         int i, ret;
341         /* Count the partitions */
342         for (i=0; data->partitions && data->partitions[i]; i++) { /* noop */};
343         
344         /* Add partition to list of partitions */
345         data->partitions = talloc_realloc(data, data->partitions, struct dsdb_partition *, i + 2);
346         if (!data->partitions) {
347                 ldb_oom(ldb);
348                 return LDB_ERR_OPERATIONS_ERROR;
349         }
350         data->partitions[i] = talloc_steal(data->partitions, partition);
351         data->partitions[i+1] = NULL;
352         
353         /* Sort again (should use binary insert) */
354         qsort(data->partitions, i+1,
355               sizeof(*data->partitions), partition_sort_compare);
356         
357         ret = partition_register(ldb, partition->ctrl);
358         if (ret != LDB_SUCCESS) {
359                 return ret;
360         }
361         return LDB_SUCCESS;
362 }
363
364 int partition_reload_if_required(struct ldb_module *module, 
365                                  struct partition_private_data *data)
366 {
367         uint64_t seq;
368         int ret, i;
369         struct ldb_context *ldb = ldb_module_get_ctx(module);
370         struct ldb_message *msg;
371         struct ldb_message_element *partition_attributes;
372         TALLOC_CTX *mem_ctx;
373
374         if (!data) {
375                 /* Not initilised yet */
376                 return LDB_SUCCESS;
377         }
378
379         mem_ctx = talloc_new(data);
380         if (!mem_ctx) {
381                 ldb_oom(ldb);
382                 return LDB_ERR_OPERATIONS_ERROR;
383         }
384
385         ret = partition_primary_sequence_number(module, mem_ctx, LDB_SEQ_HIGHEST_SEQ, &seq);
386         if (ret != LDB_SUCCESS) {
387                 talloc_free(mem_ctx);
388                 return ret;
389         }
390         if (seq == data->metadata_seq) {
391                 talloc_free(mem_ctx);
392                 return LDB_SUCCESS;
393         }
394
395         ret = partition_reload_metadata(module, data, mem_ctx, &msg);
396         if (ret != LDB_SUCCESS) {
397                 talloc_free(mem_ctx);
398                 return ret;
399         }
400
401         data->metadata_seq = seq;
402
403         partition_attributes = ldb_msg_find_element(msg, "partition");
404
405         for (i=0; partition_attributes && i < partition_attributes->num_values; i++) {
406                 int j;
407                 bool new_partition = true;
408                 const char *filename_base = NULL;
409                 DATA_BLOB dn_blob;
410                 struct ldb_dn *dn;
411                 struct dsdb_partition *partition;
412                 struct ldb_module tmp_module;
413                 struct ldb_result *dn_res;
414                 const char *no_attrs[] = { NULL };
415
416                 for (j=0; data->partitions && data->partitions[j]; j++) {
417                         DATA_BLOB casefold = data_blob_string_const(ldb_dn_get_casefold(data->partitions[j]->ctrl->dn));
418                         if (data_blob_cmp(&casefold, &partition_attributes->values[i]) == 0) {
419                                 new_partition = false;
420                                 break;
421                         }
422                 }
423                 if (new_partition == false) {
424                         continue;
425                 }
426
427                 dn_blob = partition_attributes->values[i];
428                 
429                 if (dn_blob.length > 4 && 
430                     (strncmp((const char *)&dn_blob.data[dn_blob.length-4], ".ldb", 4) == 0)) {
431
432                         /* Look for DN:filename.ldb */
433                         char *p = strchr((const char *)dn_blob.data, ':');
434                         if (!p) {
435                                 ldb_asprintf_errstring(ldb, 
436                                                        "partition_init: invalid DN in attempting to parse old-style partition record: %s", (const char *)dn_blob.data);
437                                 talloc_free(mem_ctx);
438                                 return LDB_ERR_CONSTRAINT_VIOLATION;
439                         }
440                         filename_base = p+1;
441                         
442                         /* Trim off the .ldb */
443                         dn_blob.data[dn_blob.length-4] = '\0';
444                         
445                         /* Now trim off the filename */
446                         dn_blob.length = ((uint8_t *)p - dn_blob.data);
447                 }
448
449                 dn = ldb_dn_from_ldb_val(mem_ctx, ldb, &dn_blob);
450                 if (!dn) {
451                         ldb_asprintf_errstring(ldb, 
452                                                "partition_init: invalid DN in partition record: %s", (const char *)dn_blob.data);
453                         talloc_free(mem_ctx);
454                         return LDB_ERR_CONSTRAINT_VIOLATION;
455                 }
456                 
457                 if (!filename_base) {
458                         filename_base = ldb_dn_get_linearized(dn);
459                 }
460                         
461                 /* We call ldb_dn_get_linearized() because the DN in
462                  * partition_attributes is already casefolded
463                  * correctly.  We don't want to mess that up as the
464                  * schema isn't loaded yet */
465                 ret = new_partition_from_dn(ldb, data, data->partitions, dn, 
466                                             filename_base,
467                                             &partition);
468                 if (ret != LDB_SUCCESS) {
469                         talloc_free(mem_ctx);
470                         return ret;
471                 }
472
473                 /* Hack to be able to re-use dsdb_module_search_dn, which calls ldb_next_request(), which needs ->next filled out */
474                 tmp_module = *partition->module;
475                 tmp_module.next = partition->module;
476                 
477                 /* Get the 'correct' case of the partition DNs from the database */
478                 ret = dsdb_module_search_dn(&tmp_module, data, &dn_res, 
479                                             dn, no_attrs);
480                 if (ret == LDB_SUCCESS) {
481                         talloc_free(partition->ctrl->dn);
482                         partition->ctrl->dn = talloc_steal(partition->ctrl, dn_res->msgs[0]->dn);
483                         talloc_free(dn_res);
484                 } else if (ret != LDB_ERR_NO_SUCH_OBJECT) {
485                         ldb_asprintf_errstring(ldb,
486                                                "Failed to search for partition base %s in new partition at %s: %s", 
487                                                ldb_dn_get_linearized(dn), 
488                                                partition->backend_url, 
489                                                ldb_errstring(ldb));
490                         talloc_free(mem_ctx);
491                         return ret;
492                 }
493
494                 ret = add_partition_to_data(ldb, data, partition);
495                 if (ret != LDB_SUCCESS) {
496                         talloc_free(mem_ctx);
497                         return ret;
498                 }
499         }
500
501         talloc_free(mem_ctx);
502         return LDB_SUCCESS;
503 }
504
505 /* Copy the metadata (@OPTIONS etc) for the new partition into the partition */
506
507 static int new_partition_set_replicated_metadata(struct ldb_context *ldb, 
508                                                  struct ldb_module *module, struct ldb_request *last_req, 
509                                                  struct partition_private_data *data, 
510                                                  struct dsdb_partition *partition)
511 {
512         int i, ret;
513         /* for each replicate, copy from main partition.  If we get an error, we report it up the chain */
514         for (i=0; data->replicate && data->replicate[i]; i++) {
515                 struct ldb_result *replicate_res;
516                 struct ldb_request *add_req;
517                 ret = dsdb_module_search_dn(module, last_req, &replicate_res, 
518                                             data->replicate[i],
519                                             NULL);
520                 if (ret == LDB_ERR_NO_SUCH_OBJECT) {
521                         continue;
522                 }
523                 if (ret != LDB_SUCCESS) {
524                         ldb_asprintf_errstring(ldb,
525                                                "Failed to search for %s from " DSDB_PARTITION_DN 
526                                                " replicateEntries for new partition at %s on %s: %s", 
527                                                ldb_dn_get_linearized(data->replicate[i]), 
528                                                partition->backend_url,
529                                                ldb_dn_get_linearized(partition->ctrl->dn), 
530                                                ldb_errstring(ldb));
531                         return ret;
532                 }
533
534                 /* Build add request */
535                 ret = ldb_build_add_req(&add_req, ldb, replicate_res, 
536                                         replicate_res->msgs[0], NULL, NULL, 
537                                         ldb_op_default_callback, last_req);
538                 last_req = add_req;
539                 if (ret != LDB_SUCCESS) {
540                         /* return directly, this is a very unlikely error */
541                         return ret;
542                 }
543                 /* do request */
544                 ret = partition_request(partition->module, add_req);
545                 /* wait */
546                 if (ret == LDB_SUCCESS) {
547                         ret = ldb_wait(add_req->handle, LDB_WAIT_ALL);
548                 }
549                 
550                 switch (ret) {
551                 case LDB_SUCCESS:
552                         break;
553
554                 case LDB_ERR_ENTRY_ALREADY_EXISTS:
555                         /* Handle this case specially - if the
556                          * metadata already exists, replace it */
557                 {
558                         struct ldb_request *del_req;
559                         
560                         /* Don't leave a confusing string in the ldb_errstring() */
561                         ldb_reset_err_string(ldb);
562                         /* Build del request */
563                         ret = ldb_build_del_req(&del_req, ldb, replicate_res, replicate_res->msgs[0]->dn, NULL, NULL, 
564                                                 ldb_op_default_callback, last_req);
565                         last_req = del_req;
566                         if (ret != LDB_SUCCESS) {
567                                 /* return directly, this is a very unlikely error */
568                                 return ret;
569                         }
570                         /* do request */
571                         ret = partition_request(partition->module, del_req);
572                         
573                         /* wait */
574                         if (ret == LDB_SUCCESS) {
575                                 ret = ldb_wait(del_req->handle, LDB_WAIT_ALL);
576                         }
577                         if (ret != LDB_SUCCESS) {
578                                 ldb_asprintf_errstring(ldb,
579                                                        "Failed to delete  (for re-add) %s from " DSDB_PARTITION_DN 
580                                                        " replicateEntries in new partition at %s on %s: %s", 
581                                                        ldb_dn_get_linearized(data->replicate[i]), 
582                                                        partition->backend_url,
583                                                        ldb_dn_get_linearized(partition->ctrl->dn), 
584                                                        ldb_errstring(ldb));
585                                 return ret;
586                         }
587                         
588                         /* Build add request */
589                         ret = ldb_build_add_req(&add_req, ldb, replicate_res, replicate_res->msgs[0], NULL, NULL, 
590                                                 ldb_op_default_callback, last_req);
591                         last_req = add_req;
592                         if (ret != LDB_SUCCESS) {
593                                 /* return directly, this is a very unlikely error */
594                                 return ret;
595                         }
596                         
597                         /* do the add again */
598                         ret = partition_request(partition->module, add_req);
599                         
600                         /* wait */
601                         if (ret == LDB_SUCCESS) {
602                                 ret = ldb_wait(add_req->handle, LDB_WAIT_ALL);
603                         }
604
605                         if (ret != LDB_SUCCESS) {
606                                 ldb_asprintf_errstring(ldb,
607                                                        "Failed to add (after delete) %s from " DSDB_PARTITION_DN 
608                                                        " replicateEntries to new partition at %s on %s: %s", 
609                                                        ldb_dn_get_linearized(data->replicate[i]), 
610                                                        partition->backend_url,
611                                                        ldb_dn_get_linearized(partition->ctrl->dn), 
612                                                        ldb_errstring(ldb));
613                                 return ret;
614                         }
615                         break;
616                 }
617                 default: 
618                 {
619                         ldb_asprintf_errstring(ldb,
620                                                "Failed to add %s from " DSDB_PARTITION_DN 
621                                                " replicateEntries to new partition at %s on %s: %s", 
622                                                ldb_dn_get_linearized(data->replicate[i]), 
623                                                partition->backend_url,
624                                                ldb_dn_get_linearized(partition->ctrl->dn), 
625                                                ldb_errstring(ldb));
626                         return ret;
627                 }
628                 }
629
630                 /* And around again, for the next thing we must merge */
631         }
632         return LDB_SUCCESS;
633 }
634
635 /* Extended operation to create a new partition, called when
636  * 'new_partition' detects that one is being added based on it's
637  * instanceType */
638 int partition_create(struct ldb_module *module, struct ldb_request *req)
639 {
640         int i, ret;
641         struct ldb_context *ldb = ldb_module_get_ctx(module);
642         struct ldb_request *mod_req, *last_req = req;
643         struct ldb_message *mod_msg;
644         struct partition_private_data *data;
645         struct dsdb_partition *partition = NULL;
646         const char *casefold_dn;
647         bool new_partition = false;
648
649         /* Check if this is already a partition */
650
651         struct dsdb_create_partition_exop *ex_op = talloc_get_type(req->op.extended.data, struct dsdb_create_partition_exop);
652         struct ldb_dn *dn = ex_op->new_dn;
653
654         data = talloc_get_type(module->private_data, struct partition_private_data);
655         if (!data) {
656                 /* We are not going to create a partition before we are even set up */
657                 return LDB_ERR_UNWILLING_TO_PERFORM;
658         }
659
660         for (i=0; data->partitions && data->partitions[i]; i++) {
661                 if (ldb_dn_compare(data->partitions[i]->ctrl->dn, dn) == 0) {
662                         partition = data->partitions[i];
663                 }
664         }
665
666         if (!partition) {
667                 new_partition = true;
668                 mod_msg = ldb_msg_new(req);
669                 if (!mod_msg) {
670                         ldb_oom(ldb);
671                         return LDB_ERR_OPERATIONS_ERROR;
672                 }
673                 
674                 mod_msg->dn = ldb_dn_new(mod_msg, ldb, DSDB_PARTITION_DN);
675                 ret = ldb_msg_add_empty(mod_msg, DSDB_PARTITION_ATTR, LDB_FLAG_MOD_ADD, NULL);
676                 if (ret != LDB_SUCCESS) {
677                         return ret;
678                 }
679                 
680                 casefold_dn = ldb_dn_get_casefold(dn);
681                 
682                 ret = ldb_msg_add_string(mod_msg, DSDB_PARTITION_ATTR, casefold_dn);
683                 if (ret != LDB_SUCCESS) {
684                         return ret;
685                 }
686                 
687                 /* Perform modify on @PARTITION record */
688                 ret = ldb_build_mod_req(&mod_req, ldb, req, mod_msg, NULL, NULL, 
689                                         ldb_op_default_callback, req);
690                 
691                 if (ret != LDB_SUCCESS) {
692                         return ret;
693                 }
694                 
695                 last_req = mod_req;
696
697                 ret = partition_request(module, mod_req);
698                 if (ret == LDB_SUCCESS) {
699                         ret = ldb_wait(mod_req->handle, LDB_WAIT_ALL);
700                 }
701                 
702                 if (ret != LDB_SUCCESS) {
703                         return ret;
704                 }
705                 
706                 /* Make a partition structure for this new partition, so we can copy in the template structure */ 
707                 ret = new_partition_from_dn(ldb, data, req, ldb_dn_copy(req, dn), casefold_dn, &partition);
708                 if (ret != LDB_SUCCESS) {
709                         return ret;
710                 }
711         }
712         
713         ret = new_partition_set_replicated_metadata(ldb, module, last_req, data, partition);
714         if (ret != LDB_SUCCESS) {
715                 return ret;
716         }
717         
718         if (new_partition) {
719                 ret = add_partition_to_data(ldb, data, partition);
720                 if (ret != LDB_SUCCESS) {
721                         return ret;
722                 }
723         }
724
725         /* send request done */
726         return ldb_module_done(req, NULL, NULL, LDB_SUCCESS);
727 }
728
729
730 int partition_init(struct ldb_module *module)
731 {
732         int ret;
733         TALLOC_CTX *mem_ctx = talloc_new(module);
734
735         struct partition_private_data *data;
736
737         if (!mem_ctx) {
738                 return LDB_ERR_OPERATIONS_ERROR;
739         }
740
741         data = talloc_zero(mem_ctx, struct partition_private_data);
742         if (data == NULL) {
743                 return LDB_ERR_OPERATIONS_ERROR;
744         }
745
746         /* This loads the partitions */
747         ret = partition_reload_if_required(module, data);
748         if (ret != LDB_SUCCESS) {
749                 return ret;
750         }
751
752         module->private_data = talloc_steal(module, data);
753         talloc_free(mem_ctx);
754
755         ret = ldb_mod_register_control(module, LDB_CONTROL_DOMAIN_SCOPE_OID);
756         if (ret != LDB_SUCCESS) {
757                 ldb_debug(ldb_module_get_ctx(module), LDB_DEBUG_ERROR,
758                         "partition: Unable to register control with rootdse!\n");
759                 return LDB_ERR_OPERATIONS_ERROR;
760         }
761
762         ret = ldb_mod_register_control(module, LDB_CONTROL_SEARCH_OPTIONS_OID);
763         if (ret != LDB_SUCCESS) {
764                 ldb_debug(ldb_module_get_ctx(module), LDB_DEBUG_ERROR,
765                         "partition: Unable to register control with rootdse!\n");
766                 return LDB_ERR_OPERATIONS_ERROR;
767         }
768
769         return ldb_next_init(module);
770 }