source4/lib/ldb/ldb_map/ldb_map_inbound.c

   1 /*
   2    ldb database mapping module
   3
   4    Copyright (C) Jelmer Vernooij 2005
   5    Copyright (C) Martin Kuehl <mkhl@samba.org> 2006
   6    Copyright (C) Simo Sorce <idra@samba.org> 2008
   7
   8      ** NOTE! The following LGPL license applies to the ldb
   9      ** library. This does NOT imply that all of Samba is released
  10      ** under the LGPL
  11
  12    This library is free software; you can redistribute it and/or
  13    modify it under the terms of the GNU Lesser General Public
  14    License as published by the Free Software Foundation; either
  15    version 3 of the License, or (at your option) any later version.
  16
  17    This library is distributed in the hope that it will be useful,
  18    but WITHOUT ANY WARRANTY; without even the implied warranty of
  19    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  20    Lesser General Public License for more details.
  21
  22    You should have received a copy of the GNU Lesser General Public
  23    License along with this library; if not, see <http://www.gnu.org/licenses/>.
  24
  25 */
  26
  27 #include "ldb_includes.h"
  28 #include "ldb_map.h"
  29 #include "ldb_map_private.h"
  30
  31
  32 /* Mapping message elements
  33  * ======================== */
  34
  35 /* Map a message element into the remote partition. */
  36 static struct ldb_message_element *ldb_msg_el_map_local(struct ldb_module *module, void *mem_ctx, const struct ldb_map_attribute *map, const struct ldb_message_element *old)
  37 {
  38         struct ldb_message_element *el;
  39         unsigned int i;
  40
  41         el = talloc_zero(mem_ctx, struct ldb_message_element);
  42         if (el == NULL) {
  43                 map_oom(module);
  44                 return NULL;
  45         }
  46
  47         el->num_values = old->num_values;
  48         el->values = talloc_array(el, struct ldb_val, el->num_values);
  49         if (el->values == NULL) {
  50                 talloc_free(el);
  51                 map_oom(module);
  52                 return NULL;
  53         }
  54
  55         el->name = map_attr_map_local(el, map, old->name);
  56
  57         for (i = 0; i < el->num_values; i++) {
  58                 el->values[i] = ldb_val_map_local(module, el->values, map, &old->values[i]);
  59         }
  60
  61         return el;
  62 }
  63
  64 /* Add a message element either to a local or to a remote message,
  65  * depending on whether it goes into the local or remote partition. */
  66 static int ldb_msg_el_partition(struct ldb_module *module, struct ldb_message *local, struct ldb_message *remote, const struct ldb_message *msg, const char *attr_name, /* const char * const names[], */ const struct ldb_message_element *old)
  67 {
  68         const struct ldb_map_context *data = map_get_context(module);
  69         const struct ldb_map_attribute *map = map_attr_find_local(data, attr_name);
  70         struct ldb_message_element *el=NULL;
  71         struct ldb_context *ldb = ldb_module_get_ctx(module);
  72
  73         /* Unknown attribute: ignore */
  74         if (map == NULL) {
  75                 ldb_debug(ldb, LDB_DEBUG_WARNING, "ldb_map: "
  76                           "Not mapping attribute '%s': no mapping found",
  77                           old->name);
  78                 goto local;
  79         }
  80
  81         switch (map->type) {
  82         case LDB_MAP_IGNORE:
  83                 goto local;
  84
  85         case LDB_MAP_CONVERT:
  86                 if (map->u.convert.convert_local == NULL) {
  87                         ldb_debug(ldb, LDB_DEBUG_WARNING, "ldb_map: "
  88                                   "Not mapping attribute '%s': "
  89                                   "'convert_local' not set",
  90                                   map->local_name);
  91                         goto local;
  92                 }
  93                 /* fall through */
  94         case LDB_MAP_KEEP:
  95         case LDB_MAP_RENAME:
  96                 el = ldb_msg_el_map_local(module, remote, map, old);
  97                 break;
  98
  99         case LDB_MAP_GENERATE:
 100                 if (map->u.generate.generate_remote == NULL) {
 101                         ldb_debug(ldb, LDB_DEBUG_WARNING, "ldb_map: "
 102                                   "Not mapping attribute '%s': "
 103                                   "'generate_remote' not set",
 104                                   map->local_name);
 105                         goto local;
 106                 }
 107
 108                 /* TODO: if this attr requires context:
 109                  *       make sure all context attrs are mappable (in 'names')
 110                  *       make sure all context attrs have already been mapped?
 111                  *       maybe postpone generation until they have been mapped?
 112                  */
 113
 114                 map->u.generate.generate_remote(module, map->local_name, msg, remote, local);
 115                 return 0;
 116         }
 117
 118         if (el == NULL) {
 119                 return -1;
 120         }
 121
 122         return ldb_msg_add(remote, el, old->flags);
 123
 124 local:
 125         el = talloc(local, struct ldb_message_element);
 126         if (el == NULL) {
 127                 map_oom(module);
 128                 return -1;
 129         }
 130
 131         *el = *old;                     /* copy the old element */
 132
 133         return ldb_msg_add(local, el, old->flags);
 134 }
 135
 136 /* Mapping messages
 137  * ================ */
 138
 139 /* Check whether a message will be (partially) mapped into the remote partition. */
 140 static bool ldb_msg_check_remote(struct ldb_module *module, const struct ldb_message *msg)
 141 {
 142         const struct ldb_map_context *data = map_get_context(module);
 143         bool ret;
 144         unsigned int i;
 145
 146         for (i = 0; i < msg->num_elements; i++) {
 147                 ret = map_attr_check_remote(data, msg->elements[i].name);
 148                 if (ret) {
 149                         return ret;
 150                 }
 151         }
 152
 153         return false;
 154 }
 155
 156 /* Split message elements that stay in the local partition from those
 157  * that are mapped into the remote partition. */
 158 static int ldb_msg_partition(struct ldb_module *module, struct ldb_message *local, struct ldb_message *remote, const struct ldb_message *msg)
 159 {
 160         /* const char * const names[]; */
 161         struct ldb_context *ldb;
 162         unsigned int i;
 163         int ret;
 164
 165         ldb = ldb_module_get_ctx(module);
 166
 167         for (i = 0; i < msg->num_elements; i++) {
 168                 /* Skip 'IS_MAPPED' */
 169                 if (ldb_attr_cmp(msg->elements[i].name, IS_MAPPED) == 0) {
 170                         ldb_debug(ldb, LDB_DEBUG_WARNING, "ldb_map: "
 171                                   "Skipping attribute '%s'",
 172                                   msg->elements[i].name);
 173                         continue;
 174                 }
 175
 176                 ret = ldb_msg_el_partition(module, local, remote, msg, msg->elements[i].name, &msg->elements[i]);
 177                 if (ret) {
 178                         return ret;
 179                 }
 180         }
 181
 182         return 0;
 183 }
 184
 185
 186 static int map_add_do_local(struct map_context *ac);
 187 static int map_modify_do_local(struct map_context *ac);
 188 static int map_delete_do_local(struct map_context *ac);
 189 static int map_rename_do_local(struct map_context *ac);
 190 static int map_rename_do_fixup(struct map_context *ac);
 191 static int map_rename_local_callback(struct ldb_request *req,
 192                                      struct ldb_reply *ares);
 193
 194
 195 /*****************************************************************************
 196  * COMMON INBOUND functions
 197 *****************************************************************************/
 198
 199 /* Store the DN of a single search result in context. */
 200 static int map_search_self_callback(struct ldb_request *req, struct ldb_reply *ares)
 201 {
 202         struct ldb_context *ldb;
 203         struct map_context *ac;
 204         int ret;
 205
 206         ac = talloc_get_type(req->context, struct map_context);
 207         ldb = ldb_module_get_ctx(ac->module);
 208
 209         if (!ares) {
 210                 return ldb_module_done(ac->req, NULL, NULL,
 211                                         LDB_ERR_OPERATIONS_ERROR);
 212         }
 213         if (ares->error != LDB_SUCCESS) {
 214                 return ldb_module_done(ac->req, ares->controls,
 215                                         ares->response, ares->error);
 216         }
 217
 218         /* We are interested only in the single reply */
 219         switch(ares->type) {
 220         case LDB_REPLY_ENTRY:
 221                 /* We have already found a remote DN */
 222                 if (ac->local_dn) {
 223                         ldb_set_errstring(ldb,
 224                                           "Too many results!");
 225                         return ldb_module_done(ac->req, NULL, NULL,
 226                                                 LDB_ERR_OPERATIONS_ERROR);
 227                 }
 228
 229                 /* Store local DN */
 230                 ac->local_dn = talloc_steal(ac, ares->message->dn);
 231                 break;
 232
 233         case LDB_REPLY_DONE:
 234
 235                 switch (ac->req->operation) {
 236                 case LDB_MODIFY:
 237                         ret = map_modify_do_local(ac);
 238                         break;
 239                 case LDB_DELETE:
 240                         ret = map_delete_do_local(ac);
 241                         break;
 242                 case LDB_RENAME:
 243                         ret = map_rename_do_local(ac);
 244                         break;
 245                 default:
 246                         /* if we get here we have definitely a problem */
 247                         ret = LDB_ERR_OPERATIONS_ERROR;
 248                 }
 249                 if (ret != LDB_SUCCESS) {
 250                         return ldb_module_done(ac->req, NULL, NULL,
 251                                                 LDB_ERR_OPERATIONS_ERROR);
 252                 }
 253
 254         default:
 255                 /* ignore referrals */
 256                 break;
 257         }
 258
 259         talloc_free(ares);
 260         return LDB_SUCCESS;
 261 }
 262
 263 /* Build a request to search the local record by its DN. */
 264 static int map_search_self_req(struct ldb_request **req,
 265                                 struct map_context *ac,
 266                                 struct ldb_dn *dn)
 267 {
 268         /* attrs[] is returned from this function in
 269          * ac->search_req->op.search.attrs, so it must be static, as
 270          * otherwise the compiler can put it on the stack */
 271         static const char * const attrs[] = { IS_MAPPED, NULL };
 272         struct ldb_parse_tree *tree;
 273
 274         /* Limit search to records with 'IS_MAPPED' present */
 275         tree = ldb_parse_tree(ac, "(" IS_MAPPED "=*)");
 276         if (tree == NULL) {
 277                 map_oom(ac->module);
 278                 return LDB_ERR_OPERATIONS_ERROR;
 279         }
 280
 281         *req = map_search_base_req(ac, dn, attrs, tree,
 282                                    ac, map_search_self_callback);
 283         if (*req == NULL) {
 284                 return LDB_ERR_OPERATIONS_ERROR;
 285         }
 286
 287         return LDB_SUCCESS;
 288 }
 289
 290 static int map_op_local_callback(struct ldb_request *req,
 291                                  struct ldb_reply *ares)
 292 {
 293         struct ldb_context *ldb;
 294         struct map_context *ac;
 295         int ret;
 296
 297         ac = talloc_get_type(req->context, struct map_context);
 298         ldb = ldb_module_get_ctx(ac->module);
 299
 300         if (!ares) {
 301                 return ldb_module_done(ac->req, NULL, NULL,
 302                                         LDB_ERR_OPERATIONS_ERROR);
 303         }
 304         if (ares->error != LDB_SUCCESS) {
 305                 return ldb_module_done(ac->req, ares->controls,
 306                                         ares->response, ares->error);
 307         }
 308
 309         if (ares->type != LDB_REPLY_DONE) {
 310                 ldb_set_errstring(ldb, "Invalid reply type!");
 311                 return ldb_module_done(ac->req, NULL, NULL,
 312                                         LDB_ERR_OPERATIONS_ERROR);
 313         }
 314
 315         /* Do the remote request. */
 316         ret = ldb_next_remote_request(ac->module, ac->remote_req);
 317         if (ret != LDB_SUCCESS) {
 318                 return ldb_module_done(ac->req, NULL, NULL,
 319                                         LDB_ERR_OPERATIONS_ERROR);
 320         }
 321
 322         return LDB_SUCCESS;
 323 }
 324
 325 static int map_op_remote_callback(struct ldb_request *req,
 326                                   struct ldb_reply *ares)
 327 {
 328         struct ldb_context *ldb;
 329         struct map_context *ac;
 330
 331         ac = talloc_get_type(req->context, struct map_context);
 332         ldb = ldb_module_get_ctx(ac->module);
 333
 334         if (!ares) {
 335                 return ldb_module_done(ac->req, NULL, NULL,
 336                                         LDB_ERR_OPERATIONS_ERROR);
 337         }
 338         if (ares->error != LDB_SUCCESS) {
 339                 return ldb_module_done(ac->req, ares->controls,
 340                                         ares->response, ares->error);
 341         }
 342
 343         if (ares->type != LDB_REPLY_DONE) {
 344                 ldb_set_errstring(ldb, "Invalid reply type!");
 345                 return ldb_module_done(ac->req, NULL, NULL,
 346                                         LDB_ERR_OPERATIONS_ERROR);
 347         }
 348
 349         return ldb_module_done(ac->req, ares->controls,
 350                                         ares->response, ares->error);
 351 }
 352
 353
 354 /*****************************************************************************
 355  * ADD operations
 356 *****************************************************************************/
 357
 358
 359 /* Add a record. */
 360 int map_add(struct ldb_module *module, struct ldb_request *req)
 361 {
 362         const struct ldb_message *msg = req->op.add.message;
 363         struct ldb_context *ldb;
 364         struct map_context *ac;
 365         struct ldb_message *remote_msg;
 366         int ret;
 367
 368         ldb = ldb_module_get_ctx(module);
 369
 370         /* Do not manipulate our control entries */
 371         if (ldb_dn_is_special(msg->dn)) {
 372                 return ldb_next_request(module, req);
 373         }
 374
 375         /* No mapping requested (perhaps no DN mapping specified), skip to next module */
 376         if (!ldb_dn_check_local(module, msg->dn)) {
 377                 return ldb_next_request(module, req);
 378         }
 379
 380         /* No mapping needed, fail */
 381         if (!ldb_msg_check_remote(module, msg)) {
 382                 return LDB_ERR_OPERATIONS_ERROR;
 383         }
 384
 385         /* Prepare context and handle */
 386         ac = map_init_context(module, req);
 387         if (ac == NULL) {
 388                 return LDB_ERR_OPERATIONS_ERROR;
 389         }
 390
 391
 392         /* Prepare the local message */
 393         ac->local_msg = ldb_msg_new(ac);
 394         if (ac->local_msg == NULL) {
 395                 map_oom(module);
 396                 return LDB_ERR_OPERATIONS_ERROR;
 397         }
 398         ac->local_msg->dn = msg->dn;
 399
 400         /* Prepare the remote message */
 401         remote_msg = ldb_msg_new(ac);
 402         if (remote_msg == NULL) {
 403                 map_oom(module);
 404                 return LDB_ERR_OPERATIONS_ERROR;
 405         }
 406         remote_msg->dn = ldb_dn_map_local(ac->module, remote_msg, msg->dn);
 407
 408         /* Split local from remote message */
 409         ldb_msg_partition(module, ac->local_msg, remote_msg, msg);
 410
 411         /* Prepare the remote operation */
 412         ret = ldb_build_add_req(&ac->remote_req, ldb,
 413                                 ac, remote_msg,
 414                                 req->controls,
 415                                 ac, map_op_remote_callback,
 416                                 req);
 417         if (ret != LDB_SUCCESS) {
 418                 return LDB_ERR_OPERATIONS_ERROR;
 419         }
 420
 421         if ((ac->local_msg->num_elements == 0) ||
 422             ( ! map_check_local_db(ac->module))) {
 423                 /* No local data or db, just run the remote request */
 424                 return ldb_next_remote_request(ac->module, ac->remote_req);
 425         }
 426
 427         /* Store remote DN in 'IS_MAPPED' */
 428         /* TODO: use GUIDs here instead */
 429         ret = ldb_msg_add_linearized_dn(ac->local_msg, IS_MAPPED,
 430                                         remote_msg->dn);
 431         if (ret != LDB_SUCCESS) {
 432                 return LDB_ERR_OPERATIONS_ERROR;
 433         }
 434
 435         return map_add_do_local(ac);
 436 }
 437
 438 /* Add the local record. */
 439 static int map_add_do_local(struct map_context *ac)
 440 {
 441         struct ldb_request *local_req;
 442         struct ldb_context *ldb;
 443         int ret;
 444
 445         ldb = ldb_module_get_ctx(ac->module);
 446
 447         /* Prepare the local operation */
 448         ret = ldb_build_add_req(&local_req, ldb, ac,
 449                                 ac->local_msg,
 450                                 ac->req->controls,
 451                                 ac,
 452                                 map_op_local_callback,
 453                                 ac->req);
 454         if (ret != LDB_SUCCESS) {
 455                 return LDB_ERR_OPERATIONS_ERROR;
 456         }
 457         return ldb_next_request(ac->module, local_req);
 458 }
 459
 460 /*****************************************************************************
 461  * MODIFY operations
 462 *****************************************************************************/
 463
 464 /* Modify a record. */
 465 int map_modify(struct ldb_module *module, struct ldb_request *req)
 466 {
 467         const struct ldb_message *msg = req->op.mod.message;
 468         struct ldb_request *search_req;
 469         struct ldb_message *remote_msg;
 470         struct ldb_context *ldb;
 471         struct map_context *ac;
 472         int ret;
 473
 474         ldb = ldb_module_get_ctx(module);
 475
 476         /* Do not manipulate our control entries */
 477         if (ldb_dn_is_special(msg->dn)) {
 478                 return ldb_next_request(module, req);
 479         }
 480
 481         /* No mapping requested (perhaps no DN mapping specified), skip to next module */
 482         if (!ldb_dn_check_local(module, msg->dn)) {
 483                 return ldb_next_request(module, req);
 484         }
 485
 486         /* No mapping needed, skip to next module */
 487         /* TODO: What if the remote part exists, the local doesn't,
 488          *       and this request wants to modify local data and thus
 489          *       add the local record? */
 490         if (!ldb_msg_check_remote(module, msg)) {
 491                 return LDB_ERR_OPERATIONS_ERROR;
 492         }
 493
 494         /* Prepare context and handle */
 495         ac = map_init_context(module, req);
 496         if (ac == NULL) {
 497                 return LDB_ERR_OPERATIONS_ERROR;
 498         }
 499
 500         /* Prepare the local message */
 501         ac->local_msg = ldb_msg_new(ac);
 502         if (ac->local_msg == NULL) {
 503                 map_oom(module);
 504                 return LDB_ERR_OPERATIONS_ERROR;
 505         }
 506         ac->local_msg->dn = msg->dn;
 507
 508         /* Prepare the remote message */
 509         remote_msg = ldb_msg_new(ac->remote_req);
 510         if (remote_msg == NULL) {
 511                 map_oom(module);
 512                 return LDB_ERR_OPERATIONS_ERROR;
 513         }
 514         remote_msg->dn = ldb_dn_map_local(ac->module, remote_msg, msg->dn);
 515
 516         /* Split local from remote message */
 517         ldb_msg_partition(module, ac->local_msg, remote_msg, msg);
 518
 519         /* Prepare the remote operation */
 520         ret = ldb_build_mod_req(&ac->remote_req, ldb,
 521                                 ac, remote_msg,
 522                                 req->controls,
 523                                 ac, map_op_remote_callback,
 524                                 req);
 525         if (ret != LDB_SUCCESS) {
 526                 return LDB_ERR_OPERATIONS_ERROR;
 527         }
 528
 529         if ((ac->local_msg->num_elements == 0) ||
 530             ( ! map_check_local_db(ac->module))) {
 531                 /* No local data or db, just run the remote request */
 532                 return ldb_next_remote_request(ac->module, ac->remote_req);
 533         }
 534
 535         /* prepare the search operation */
 536         ret = map_search_self_req(&search_req, ac, msg->dn);
 537         if (ret != LDB_SUCCESS) {
 538                 return LDB_ERR_OPERATIONS_ERROR;
 539         }
 540
 541         return ldb_next_request(module, search_req);
 542 }
 543
 544 /* Modify the local record. */
 545 static int map_modify_do_local(struct map_context *ac)
 546 {
 547         struct ldb_request *local_req;
 548         struct ldb_context *ldb;
 549         int ret;
 550
 551         ldb = ldb_module_get_ctx(ac->module);
 552
 553         if (ac->local_dn == NULL) {
 554                 /* No local record present, add it instead */
 555                 /* Add local 'IS_MAPPED' */
 556                 /* TODO: use GUIDs here instead */
 557                 if (ldb_msg_add_empty(ac->local_msg, IS_MAPPED,
 558                                         LDB_FLAG_MOD_ADD, NULL) != 0) {
 559                         return LDB_ERR_OPERATIONS_ERROR;
 560                 }
 561                 ret = ldb_msg_add_linearized_dn(ac->local_msg, IS_MAPPED,
 562                                                 ac->remote_req->op.mod.message->dn);
 563                 if (ret != 0) {
 564                         return LDB_ERR_OPERATIONS_ERROR;
 565                 }
 566
 567                 /* Prepare the local operation */
 568                 ret = ldb_build_add_req(&local_req, ldb, ac,
 569                                         ac->local_msg,
 570                                         ac->req->controls,
 571                                         ac,
 572                                         map_op_local_callback,
 573                                         ac->req);
 574                 if (ret != LDB_SUCCESS) {
 575                         return LDB_ERR_OPERATIONS_ERROR;
 576                 }
 577         } else {
 578                 /* Prepare the local operation */
 579                 ret = ldb_build_mod_req(&local_req, ldb, ac,
 580                                         ac->local_msg,
 581                                         ac->req->controls,
 582                                         ac,
 583                                         map_op_local_callback,
 584                                         ac->req);
 585                 if (ret != LDB_SUCCESS) {
 586                         return LDB_ERR_OPERATIONS_ERROR;
 587                 }
 588         }
 589
 590         return ldb_next_request(ac->module, local_req);
 591 }
 592
 593 /*****************************************************************************
 594  * DELETE operations
 595 *****************************************************************************/
 596
 597 /* Delete a record. */
 598 int map_delete(struct ldb_module *module, struct ldb_request *req)
 599 {
 600         struct ldb_request *search_req;
 601         struct ldb_context *ldb;
 602         struct map_context *ac;
 603         int ret;
 604
 605         ldb = ldb_module_get_ctx(module);
 606
 607         /* Do not manipulate our control entries */
 608         if (ldb_dn_is_special(req->op.del.dn)) {
 609                 return ldb_next_request(module, req);
 610         }
 611
 612         /* No mapping requested (perhaps no DN mapping specified).
 613          * Skip to next module */
 614         if (!ldb_dn_check_local(module, req->op.del.dn)) {
 615                 return ldb_next_request(module, req);
 616         }
 617
 618         /* Prepare context and handle */
 619         ac = map_init_context(module, req);
 620         if (ac == NULL) {
 621                 return LDB_ERR_OPERATIONS_ERROR;
 622         }
 623
 624         /* Prepare the remote operation */
 625         ret = ldb_build_del_req(&ac->remote_req, ldb, ac,
 626                                    ldb_dn_map_local(module, ac, req->op.del.dn),
 627                                    req->controls,
 628                                    ac,
 629                                    map_op_remote_callback,
 630                                    req);
 631         if (ret != LDB_SUCCESS) {
 632                 return LDB_ERR_OPERATIONS_ERROR;
 633         }
 634
 635         /* No local db, just run the remote request */
 636         if (!map_check_local_db(ac->module)) {
 637                 /* Do the remote request. */
 638                 return ldb_next_remote_request(ac->module, ac->remote_req);
 639         }
 640
 641         /* Prepare the search operation */
 642         ret = map_search_self_req(&search_req, ac, req->op.del.dn);
 643         if (ret != LDB_SUCCESS) {
 644                 map_oom(module);
 645                 return LDB_ERR_OPERATIONS_ERROR;
 646         }
 647
 648         return ldb_next_request(module, search_req);
 649 }
 650
 651 /* Delete the local record. */
 652 static int map_delete_do_local(struct map_context *ac)
 653 {
 654         struct ldb_request *local_req;
 655         struct ldb_context *ldb;
 656         int ret;
 657
 658         ldb = ldb_module_get_ctx(ac->module);
 659
 660         /* No local record, continue remotely */
 661         if (ac->local_dn == NULL) {
 662                 /* Do the remote request. */
 663                 return ldb_next_remote_request(ac->module, ac->remote_req);
 664         }
 665
 666         /* Prepare the local operation */
 667         ret = ldb_build_del_req(&local_req, ldb, ac,
 668                                    ac->req->op.del.dn,
 669                                    ac->req->controls,
 670                                    ac,
 671                                    map_op_local_callback,
 672                                    ac->req);
 673         if (ret != LDB_SUCCESS) {
 674                 return LDB_ERR_OPERATIONS_ERROR;
 675         }
 676         return ldb_next_request(ac->module, local_req);
 677 }
 678
 679 /*****************************************************************************
 680  * RENAME operations
 681 *****************************************************************************/
 682
 683 /* Rename a record. */
 684 int map_rename(struct ldb_module *module, struct ldb_request *req)
 685 {
 686         struct ldb_request *search_req;
 687         struct ldb_context *ldb;
 688         struct map_context *ac;
 689         int ret;
 690
 691         ldb = ldb_module_get_ctx(module);
 692
 693         /* Do not manipulate our control entries */
 694         if (ldb_dn_is_special(req->op.rename.olddn)) {
 695                 return ldb_next_request(module, req);
 696         }
 697
 698         /* No mapping requested (perhaps no DN mapping specified).
 699          * Skip to next module */
 700         if ((!ldb_dn_check_local(module, req->op.rename.olddn)) &&
 701             (!ldb_dn_check_local(module, req->op.rename.newdn))) {
 702                 return ldb_next_request(module, req);
 703         }
 704
 705         /* Rename into/out of the mapped partition requested, bail out */
 706         if (!ldb_dn_check_local(module, req->op.rename.olddn) ||
 707             !ldb_dn_check_local(module, req->op.rename.newdn)) {
 708                 return LDB_ERR_AFFECTS_MULTIPLE_DSAS;
 709         }
 710
 711         /* Prepare context and handle */
 712         ac = map_init_context(module, req);
 713         if (ac == NULL) {
 714                 return LDB_ERR_OPERATIONS_ERROR;
 715         }
 716
 717         /* Prepare the remote operation */
 718         ret = ldb_build_rename_req(&ac->remote_req, ldb, ac,
 719                                    ldb_dn_map_local(module, ac, req->op.rename.olddn),
 720                                    ldb_dn_map_local(module, ac, req->op.rename.newdn),
 721                                    req->controls,
 722                                    ac, map_op_remote_callback,
 723                                    req);
 724         if (ret != LDB_SUCCESS) {
 725                 return LDB_ERR_OPERATIONS_ERROR;
 726         }
 727
 728         /* No local db, just run the remote request */
 729         if (!map_check_local_db(ac->module)) {
 730                 /* Do the remote request. */
 731                 return ldb_next_remote_request(ac->module, ac->remote_req);
 732         }
 733
 734         /* Prepare the search operation */
 735         ret = map_search_self_req(&search_req, ac, req->op.rename.olddn);
 736         if (ret != LDB_SUCCESS) {
 737                 map_oom(module);
 738                 return LDB_ERR_OPERATIONS_ERROR;
 739         }
 740
 741         return ldb_next_request(module, search_req);
 742 }
 743
 744 /* Rename the local record. */
 745 static int map_rename_do_local(struct map_context *ac)
 746 {
 747         struct ldb_request *local_req;
 748         struct ldb_context *ldb;
 749         int ret;
 750
 751         ldb = ldb_module_get_ctx(ac->module);
 752
 753         /* No local record, continue remotely */
 754         if (ac->local_dn == NULL) {
 755                 /* Do the remote request. */
 756                 return ldb_next_remote_request(ac->module, ac->remote_req);
 757         }
 758
 759         /* Prepare the local operation */
 760         ret = ldb_build_rename_req(&local_req, ldb, ac,
 761                                    ac->req->op.rename.olddn,
 762                                    ac->req->op.rename.newdn,
 763                                    ac->req->controls,
 764                                    ac,
 765                                    map_rename_local_callback,
 766                                    ac->req);
 767         if (ret != LDB_SUCCESS) {
 768                 return LDB_ERR_OPERATIONS_ERROR;
 769         }
 770
 771         return ldb_next_request(ac->module, local_req);
 772 }
 773
 774 static int map_rename_local_callback(struct ldb_request *req,
 775                                      struct ldb_reply *ares)
 776 {
 777         struct ldb_context *ldb;
 778         struct map_context *ac;
 779         int ret;
 780
 781         ac = talloc_get_type(req->context, struct map_context);
 782         ldb = ldb_module_get_ctx(ac->module);
 783
 784         if (!ares) {
 785                 return ldb_module_done(ac->req, NULL, NULL,
 786                                         LDB_ERR_OPERATIONS_ERROR);
 787         }
 788         if (ares->error != LDB_SUCCESS) {
 789                 return ldb_module_done(ac->req, ares->controls,
 790                                         ares->response, ares->error);
 791         }
 792
 793         if (ares->type != LDB_REPLY_DONE) {
 794                 ldb_set_errstring(ldb, "Invalid reply type!");
 795                 return ldb_module_done(ac->req, NULL, NULL,
 796                                         LDB_ERR_OPERATIONS_ERROR);
 797         }
 798
 799         /* proceed with next step */
 800         ret = map_rename_do_fixup(ac);
 801         if (ret != LDB_SUCCESS) {
 802                 return ldb_module_done(ac->req, NULL, NULL,
 803                                         LDB_ERR_OPERATIONS_ERROR);
 804         }
 805
 806         return LDB_SUCCESS;
 807 }
 808
 809 /* Update the local 'IS_MAPPED' attribute. */
 810 static int map_rename_do_fixup(struct map_context *ac)
 811 {
 812         struct ldb_request *local_req;
 813
 814         /* Prepare the fixup operation */
 815         /* TODO: use GUIDs here instead -- or skip it when GUIDs are used. */
 816         local_req = map_build_fixup_req(ac,
 817                                         ac->req->op.rename.newdn,
 818                                         ac->remote_req->op.rename.newdn,
 819                                         ac,
 820                                         map_op_local_callback);
 821         if (local_req == NULL) {
 822                 return LDB_ERR_OPERATIONS_ERROR;
 823         }
 824
 825         return ldb_next_request(ac->module, local_req);
 826 }