r7709: - convert ldb to use popt, so that it can interact with the samba
[abartlet/samba.git/.git] / source4 / lib / ldb / ldb_tdb / ldb_tdb.c
index eb2decfe310a863e547548aed391c5e5e97f24cd..596ede8a4e152a4bcf568d76deeecea28392469d 100644 (file)
@@ -2,6 +2,8 @@
    ldb database library
 
    Copyright (C) Andrew Tridgell  2004
+   Copyright (C) Stefan Metzmacher  2004
+   
 
      ** NOTE! The following LGPL license applies to the ldb
      ** library. This does NOT imply that all of Samba is released
  *  Description: core functions for tdb backend
  *
  *  Author: Andrew Tridgell
+ *  Author: Stefan Metzmacher
  */
 
 #include "includes.h"
+#include "ldb/include/ldb.h"
+#include "ldb/include/ldb_private.h"
 #include "ldb/ldb_tdb/ldb_tdb.h"
 
+#define LDBLOCK        "INT_LDBLOCK"
+
+
+/*
+  callback function used in call to ldb_dn_fold() for determining whether an
+  attribute type requires case folding.
+*/
+static int ltdb_case_fold_attr_required(void * user_data, char *attr)
+{
+       struct ldb_module *module = talloc_get_type(user_data, struct ldb_module);
+
+       return ltdb_attribute_flags(module, attr) & LTDB_FLAG_CASE_INSENSITIVE;
+}
+
 /*
   form a TDB_DATA for a record key
   caller frees
@@ -42,8 +61,9 @@
   note that the key for a record can depend on whether the 
   dn refers to a case sensitive index record or not
 */
-struct TDB_DATA ltdb_key(struct ldb_context *ldb, const char *dn)
+struct TDB_DATA ltdb_key(struct ldb_module *module, const char *dn)
 {
+       struct ldb_context *ldb = module->ldb;
        TDB_DATA key;
        char *key_str = NULL;
        char *dn_folded = NULL;
@@ -54,39 +74,50 @@ struct TDB_DATA ltdb_key(struct ldb_context *ldb, const char *dn)
        /*
          most DNs are case insensitive. The exception is index DNs for
          case sensitive attributes
+
+         there are 3 cases dealt with in this code:
+
+         1) if the dn doesn't start with @INDEX: then uppercase the attribute
+             names and the attributes values of case insensitive attributes
+         2) if the dn starts with @INDEX:attr and 'attr' is a case insensitive
+            attribute then uppercase whole dn
+         3) if the dn starts with @INDEX:attr and 'attr' is a case sensitive
+            attribute then uppercase up to the value of the attribute, but 
+            not the value itself
        */
        if (strncmp(dn, prefix, strlen(prefix)) == 0 &&
            (s = strchr(dn+strlen(prefix), ':'))) {
                char *attr_name, *attr_name_folded;
-               attr_name = strndup(dn+strlen(prefix), (s-(dn+strlen(prefix))));
+               attr_name = talloc_strndup(ldb, dn+strlen(prefix), (s-(dn+strlen(prefix))));
                if (!attr_name) {
                        goto failed;
                }
-               flags = ltdb_attribute_flags(ldb, attr_name);
+               flags = ltdb_attribute_flags(module, attr_name);
                
                if (flags & LTDB_FLAG_CASE_INSENSITIVE) {
-                       dn_folded = ldb_casefold(dn);
+                       dn_folded = ldb_casefold(ldb, dn);
                } else {
-                       attr_name_folded = ldb_casefold(attr_name);
+                       attr_name_folded = ldb_casefold(ldb, attr_name);
                        if (!attr_name_folded) {
                                goto failed;
                        }
-                       asprintf(&dn_folded, "%s:%s:%s",
-                                prefix, attr_name_folded,
-                                s+1);
-                       free(attr_name_folded);
+                       dn_folded = talloc_asprintf(ldb, "%s:%s:%s",
+                                                   prefix, attr_name_folded,
+                                                   s+1);
+                       talloc_free(attr_name_folded);
                }
-               free(attr_name);
+               talloc_free(attr_name);
        } else {
-               dn_folded = ldb_casefold(dn);
+                dn_folded = ldb_dn_fold(module->ldb, dn,
+                                        module, ltdb_case_fold_attr_required);
        }
 
        if (!dn_folded) {
                goto failed;
        }
 
-       asprintf(&key_str, "DN=%s", dn_folded);
-       free(dn_folded);
+       key_str = talloc_asprintf(ldb, "DN=%s", dn_folded);
+       talloc_free(dn_folded);
 
        if (!key_str) {
                goto failed;
@@ -107,20 +138,24 @@ failed:
 /*
   lock the database for write - currently a single lock is used
 */
-static int ltdb_lock(struct ldb_context *ldb)
+static int ltdb_lock(struct ldb_module *module, const char *lockname)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA key;
        int ret;
 
-       key = ltdb_key(ldb, "LDBLOCK");
+       if (lockname == NULL) {
+               return -1;
+       }
+
+       key = ltdb_key(module, lockname);
        if (!key.dptr) {
                return -1;
        }
 
        ret = tdb_chainlock(ltdb->tdb, key);
 
-       free(key.dptr);
+       talloc_free(key.dptr);
 
        return ret;
 }
@@ -128,19 +163,86 @@ static int ltdb_lock(struct ldb_context *ldb)
 /*
   unlock the database after a ltdb_lock()
 */
-static void ltdb_unlock(struct ldb_context *ldb)
+static int ltdb_unlock(struct ldb_module *module, const char *lockname)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA key;
 
-       key = ltdb_key(ldb, "LDBLOCK");
+       if (lockname == NULL) {
+               return -1;
+       }
+
+       key = ltdb_key(module, lockname);
        if (!key.dptr) {
-               return;
+               return -1;
        }
 
        tdb_chainunlock(ltdb->tdb, key);
 
-       free(key.dptr);
+       talloc_free(key.dptr);
+
+       return 0;
+}
+
+
+/*
+  lock the database for read - use by ltdb_search
+*/
+int ltdb_lock_read(struct ldb_module *module)
+{
+       struct ltdb_private *ltdb = module->private_data;
+       TDB_DATA key;
+       int ret;
+       key = ltdb_key(module, LDBLOCK);
+       if (!key.dptr) {
+               return -1;
+       }
+       ret = tdb_chainlock_read(ltdb->tdb, key);
+       talloc_free(key.dptr);
+       return ret;
+}
+
+/*
+  unlock the database after a ltdb_lock_read()
+*/
+int ltdb_unlock_read(struct ldb_module *module)
+{
+       struct ltdb_private *ltdb = module->private_data;
+       TDB_DATA key;
+       key = ltdb_key(module, LDBLOCK);
+       if (!key.dptr) {
+               return -1;
+       }
+       tdb_chainunlock_read(ltdb->tdb, key);
+       talloc_free(key.dptr);
+       return 0;
+}
+
+/*
+  check special dn's have valid attributes
+  currently only @ATTRIBUTES is checked
+*/
+int ltdb_check_special_dn(struct ldb_module *module, const struct ldb_message *msg)
+{
+       struct ltdb_private *ltdb = module->private_data;
+       int i, j;
+
+       if (strcmp(msg->dn, LTDB_ATTRIBUTES) != 0) {
+               return 0;
+       }
+
+       /* we have @ATTRIBUTES, let's check attributes are fine */
+       /* should we check that we deny multivalued attributes ? */
+       for (i = 0; i < msg->num_elements; i++) {
+               for (j = 0; j < msg->elements[i].num_values; j++) {
+                       if (ltdb_check_at_attributes_values(&msg->elements[i].values[j]) != 0) {
+                               ltdb->last_err_string = "Invalid attribute value in an @ATTRIBUTES entry";
+                               return -1;
+                       }
+               }
+       }
+
+       return 0;
 }
 
 
@@ -148,18 +250,18 @@ static void ltdb_unlock(struct ldb_context *ldb)
   we've made a modification to a dn - possibly reindex and 
   update sequence number
 */
-static int ltdb_modified(struct ldb_context *ldb, const char *dn)
+static int ltdb_modified(struct ldb_module *module, const char *dn)
 {
        int ret = 0;
 
        if (strcmp(dn, LTDB_INDEXLIST) == 0 ||
            strcmp(dn, LTDB_ATTRIBUTES) == 0) {
-               ret = ltdb_reindex(ldb);
+               ret = ltdb_reindex(module);
        }
 
        if (ret == 0 &&
            strcmp(dn, LTDB_BASEINFO) != 0) {
-               ret = ltdb_increase_sequence_number(ldb);
+               ret = ltdb_increase_sequence_number(module);
        }
 
        return ret;
@@ -168,20 +270,20 @@ static int ltdb_modified(struct ldb_context *ldb, const char *dn)
 /*
   store a record into the db
 */
-int ltdb_store(struct ldb_context *ldb, const struct ldb_message *msg, int flgs)
+int ltdb_store(struct ldb_module *module, const struct ldb_message *msg, int flgs)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA tdb_key, tdb_data;
        int ret;
 
-       tdb_key = ltdb_key(ldb, msg->dn);
+       tdb_key = ltdb_key(module, msg->dn);
        if (!tdb_key.dptr) {
                return -1;
        }
 
-       ret = ltdb_pack_data(ldb, msg, &tdb_data);
+       ret = ltdb_pack_data(module, msg, &tdb_data);
        if (ret == -1) {
-               free(tdb_key.dptr);
+               talloc_free(tdb_key.dptr);
                return -1;
        }
 
@@ -190,14 +292,14 @@ int ltdb_store(struct ldb_context *ldb, const struct ldb_message *msg, int flgs)
                goto done;
        }
        
-       ret = ltdb_index_add(ldb, msg);
+       ret = ltdb_index_add(module, msg);
        if (ret == -1) {
                tdb_delete(ltdb->tdb, tdb_key);
        }
 
 done:
-       free(tdb_key.dptr);
-       free(tdb_data.dptr);
+       talloc_free(tdb_key.dptr);
+       talloc_free(tdb_data.dptr);
 
        return ret;
 }
@@ -206,26 +308,34 @@ done:
 /*
   add a record to the database
 */
-static int ltdb_add(struct ldb_context *ldb, const struct ldb_message *msg)
+static int ltdb_add(struct ldb_module *module, const struct ldb_message *msg)
 {
+       struct ltdb_private *ltdb = module->private_data;
        int ret;
 
-       if (ltdb_lock(ldb) != 0) {
+       ltdb->last_err_string = NULL;
+
+       ret = ltdb_check_special_dn(module, msg);
+       if (ret != 0) {
+               return ret;
+       }
+       
+       if (ltdb_lock(module, LDBLOCK) != 0) {
                return -1;
        }
 
-       if (ltdb_cache_load(ldb) != 0) {
-               ltdb_unlock(ldb);
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module, LDBLOCK);
                return -1;
        }
-       
-       ret = ltdb_store(ldb, msg, TDB_INSERT);
+
+       ret = ltdb_store(module, msg, TDB_INSERT);
 
        if (ret == 0) {
-               ltdb_modified(ldb, msg->dn);
+               ltdb_modified(module, msg->dn);
        }
 
-       ltdb_unlock(ldb);
+       ltdb_unlock(module, LDBLOCK);
        return ret;
 }
 
@@ -234,19 +344,19 @@ static int ltdb_add(struct ldb_context *ldb, const struct ldb_message *msg)
   delete a record from the database, not updating indexes (used for deleting
   index records)
 */
-int ltdb_delete_noindex(struct ldb_context *ldb, const char *dn)
+int ltdb_delete_noindex(struct ldb_module *module, const char *dn)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA tdb_key;
        int ret;
 
-       tdb_key = ltdb_key(ldb, dn);
+       tdb_key = ltdb_key(module, dn);
        if (!tdb_key.dptr) {
                return -1;
        }
 
        ret = tdb_delete(ltdb->tdb, tdb_key);
-       free(tdb_key.dptr);
+       talloc_free(tdb_key.dptr);
 
        return ret;
 }
@@ -254,48 +364,54 @@ int ltdb_delete_noindex(struct ldb_context *ldb, const char *dn)
 /*
   delete a record from the database
 */
-static int ltdb_delete(struct ldb_context *ldb, const char *dn)
+static int ltdb_delete(struct ldb_module *module, const char *dn)
 {
+       struct ltdb_private *ltdb = module->private_data;
        int ret;
-       struct ldb_message msg;
+       struct ldb_message *msg = NULL;
+
+       ltdb->last_err_string = NULL;
 
-       if (ltdb_lock(ldb) != 0) {
+       if (ltdb_lock(module, LDBLOCK) != 0) {
                return -1;
        }
 
-       if (ltdb_cache_load(ldb) != 0) {
-               ltdb_unlock(ldb);
-               return -1;
+       if (ltdb_cache_load(module) != 0) {
+               goto failed;
+       }
+
+       msg = talloc(module, struct ldb_message);
+       if (msg == NULL) {
+               goto failed;
        }
 
        /* in case any attribute of the message was indexed, we need
           to fetch the old record */
-       ret = ltdb_search_dn1(ldb, dn, &msg);
+       ret = ltdb_search_dn1(module, dn, msg);
        if (ret != 1) {
                /* not finding the old record is an error */
                goto failed;
        }
 
-       ret = ltdb_delete_noindex(ldb, dn);
+       ret = ltdb_delete_noindex(module, dn);
        if (ret == -1) {
-               ltdb_search_dn1_free(ldb, &msg);
                goto failed;
        }
 
        /* remove any indexed attributes */
-       ret = ltdb_index_del(ldb, &msg);
-
-       ltdb_search_dn1_free(ldb, &msg);
+       ret = ltdb_index_del(module, msg);
 
        if (ret == 0) {
-               ltdb_modified(ldb, dn);
+               ltdb_modified(module, dn);
        }
 
-       ltdb_unlock(ldb);
+       talloc_free(msg);
+       ltdb_unlock(module, LDBLOCK);
        return ret;
 
 failed:
-       ltdb_unlock(ldb);
+       talloc_free(msg);
+       ltdb_unlock(module, LDBLOCK);
        return -1;
 }
 
@@ -309,7 +425,7 @@ failed:
 */
 static int find_element(const struct ldb_message *msg, const char *name)
 {
-       int i;
+       unsigned int i;
        for (i=0;i<msg->num_elements;i++) {
                if (ldb_attr_cmp(msg->elements[i].name, name) == 0) {
                        return i;
@@ -326,13 +442,14 @@ static int find_element(const struct ldb_message *msg, const char *name)
 
   returns 0 on success, -1 on failure (and sets errno)
 */
-static int msg_add_element(struct ldb_message *msg, struct ldb_message_element *el)
+static int msg_add_element(struct ldb_context *ldb,
+                          struct ldb_message *msg, struct ldb_message_element *el)
 {
        struct ldb_message_element *e2;
-       int i;
+       unsigned int i;
 
-       e2 = realloc_p(msg->elements, struct ldb_message_element, 
-                      msg->num_elements+1);
+       e2 = talloc_realloc(msg, msg->elements, struct ldb_message_element, 
+                             msg->num_elements+1);
        if (!e2) {
                errno = ENOMEM;
                return -1;
@@ -346,9 +463,8 @@ static int msg_add_element(struct ldb_message *msg, struct ldb_message_element *
        e2->flags = el->flags;
        e2->values = NULL;
        if (el->num_values != 0) {
-               e2->values = malloc_array_p(struct ldb_val, el->num_values);
+               e2->values = talloc_array(msg->elements, struct ldb_val, el->num_values);
                if (!e2->values) {
-                       free(e2->name);
                        errno = ENOMEM;
                        return -1;
                }
@@ -366,29 +482,32 @@ static int msg_add_element(struct ldb_message *msg, struct ldb_message_element *
 /*
   delete all elements having a specified attribute name
 */
-static int msg_delete_attribute(struct ldb_message *msg, const char *name)
+static int msg_delete_attribute(struct ldb_module *module,
+                               struct ldb_context *ldb,
+                               struct ldb_message *msg, const char *name)
 {
-       int i, count=0;
-       struct ldb_message_element *el2;
-
-       el2 = malloc_array_p(struct ldb_message_element, msg->num_elements);
-       if (!el2) {
-               errno = ENOMEM;
-               return -1;
-       }
+       unsigned int i, j;
 
        for (i=0;i<msg->num_elements;i++) {
-               if (ldb_attr_cmp(msg->elements[i].name, name) != 0) {
-                       el2[count++] = msg->elements[i];
-               } else {
-                       if (msg->elements[i].values) free(msg->elements[i].values);
+               if (ldb_attr_cmp(msg->elements[i].name, name) == 0) {
+                       for (j=0;j<msg->elements[i].num_values;j++) {
+                               ltdb_index_del_value(module, msg->dn, &msg->elements[i], j);
+                       }
+                       talloc_free(msg->elements[i].values);
+                       if (msg->num_elements > (i+1)) {
+                               memmove(&msg->elements[i], 
+                                       &msg->elements[i+1], 
+                                       sizeof(struct ldb_message_element)*
+                                       (msg->num_elements - (i+1)));
+                       }
+                       msg->num_elements--;
+                       i--;
+                       msg->elements = talloc_realloc(msg, msg->elements, 
+                                                        struct ldb_message_element, 
+                                                        msg->num_elements);
                }
        }
 
-       msg->num_elements = count;
-       if (msg->elements) free(msg->elements);
-       msg->elements = el2;
-
        return 0;
 }
 
@@ -397,32 +516,37 @@ static int msg_delete_attribute(struct ldb_message *msg, const char *name)
 
   return 0 on success, -1 on failure
 */
-static int msg_delete_element(struct ldb_context *ldb,
+static int msg_delete_element(struct ldb_module *module,
                              struct ldb_message *msg, 
                              const char *name,
                              const struct ldb_val *val)
 {
-       int i;
+       struct ldb_context *ldb = module->ldb;
+       unsigned int i;
+       int found;
        struct ldb_message_element *el;
 
-       i = find_element(msg, name);
-       if (i == -1) {
+       found = find_element(msg, name);
+       if (found == -1) {
                return -1;
        }
 
-       el = &msg->elements[i];
+       el = &msg->elements[found];
 
        for (i=0;i<el->num_values;i++) {
-               if (ldb_val_equal(ldb, msg->elements[i].name, &el->values[i], val)) {
+               if (ltdb_val_equal(module, msg->elements[i].name, &el->values[i], val)) {
                        if (i<el->num_values-1) {
                                memmove(&el->values[i], &el->values[i+1],
-                                       sizeof(el->values[i])*el->num_values-(i+1));
+                                       sizeof(el->values[i])*(el->num_values-(i+1)));
                        }
                        el->num_values--;
+                       if (el->num_values == 0) {
+                               return msg_delete_attribute(module, ldb, msg, name);
+                       }
                        return 0;
                }
        }
-       
+
        return -1;
 }
 
@@ -434,59 +558,99 @@ static int msg_delete_element(struct ldb_context *ldb,
   get away with it, but if we ever have really large attribute lists 
   then we'll need to look at this again
 */
-int ltdb_modify_internal(struct ldb_context *ldb, const struct ldb_message *msg)
+int ltdb_modify_internal(struct ldb_module *module, const struct ldb_message *msg)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA tdb_key, tdb_data;
-       struct ldb_message msg2;
-       int ret, i, j;
+       struct ldb_message *msg2;
+       unsigned i, j;
+       int ret;
 
-       tdb_key = ltdb_key(ldb, msg->dn);
+       tdb_key = ltdb_key(module, msg->dn);
        if (!tdb_key.dptr) {
                return -1;
        }
 
        tdb_data = tdb_fetch(ltdb->tdb, tdb_key);
        if (!tdb_data.dptr) {
-               free(tdb_key.dptr);
+               talloc_free(tdb_key.dptr);
+               return -1;
+       }
+
+       msg2 = talloc(tdb_key.dptr, struct ldb_message);
+       if (msg2 == NULL) {
+               talloc_free(tdb_key.dptr);
                return -1;
        }
 
-       ret = ltdb_unpack_data(ldb, &tdb_data, &msg2);
+       ret = ltdb_unpack_data(module, &tdb_data, msg2);
        if (ret == -1) {
-               free(tdb_key.dptr);
+               talloc_free(tdb_key.dptr);
                free(tdb_data.dptr);
                return -1;
        }
 
-       if (!msg2.dn) {
-               msg2.dn = msg->dn;
+       if (!msg2->dn) {
+               msg2->dn = msg->dn;
        }
 
        for (i=0;i<msg->num_elements;i++) {
+               struct ldb_message_element *el = &msg->elements[i];
+               struct ldb_message_element *el2;
+               struct ldb_val *vals;
+
                switch (msg->elements[i].flags & LDB_FLAG_MOD_MASK) {
 
                case LDB_FLAG_MOD_ADD:
                        /* add this element to the message. fail if it
                           already exists */
-                       ret = find_element(&msg2, msg->elements[i].name);
-                       if (ret != -1) {
-                               errno = EEXIST;
-                               goto failed;
+                       ret = find_element(msg2, el->name);
+
+                       if (ret == -1) {
+                               if (msg_add_element(ldb, msg2, el) != 0) {
+                                       goto failed;
+                               }
+                               continue;
                        }
-                       if (msg_add_element(&msg2, &msg->elements[i]) != 0) {
+
+                       el2 = &msg2->elements[ret];
+
+                       /* An attribute with this name already exists, add all
+                        * values if they don't already exist. */
+
+                       for (j=0;j<el->num_values;j++) {
+                               if (ldb_msg_find_val(el2, &el->values[j])) {
+                                       ltdb->last_err_string =
+                                               "Type or value exists";
+                                       goto failed;
+                               }
+                       }
+
+                       vals = talloc_realloc(msg2->elements, el2->values, struct ldb_val,
+                                               el2->num_values + el->num_values);
+
+                       if (vals == NULL)
                                goto failed;
+
+                       for (j=0;j<el->num_values;j++) {
+                               vals[el2->num_values + j] =
+                                       ldb_val_dup(vals, &el->values[j]);
                        }
+
+                       el2->values = vals;
+                       el2->num_values += el->num_values;
+
                        break;
 
                case LDB_FLAG_MOD_REPLACE:
                        /* replace all elements of this attribute name with the elements
-                          listed */
-                       if (msg_delete_attribute(&msg2, msg->elements[i].name) != 0) {
-                               goto failed;
-                       }
-                       /* add the replacement element */
-                       if (msg_add_element(&msg2, &msg->elements[i]) != 0) {
+                          listed. The attribute not existing is not an error */
+                       msg_delete_attribute(module, ldb, msg2, msg->elements[i].name);
+
+                       /* add the replacement element, if not empty */
+                       if (msg->elements[i].num_values != 0 &&
+                           msg_add_element(ldb, msg2, &msg->elements[i]) != 0) {
                                goto failed;
                        }
                        break;
@@ -495,17 +659,22 @@ int ltdb_modify_internal(struct ldb_context *ldb, const struct ldb_message *msg)
                        /* we could be being asked to delete all
                           values or just some values */
                        if (msg->elements[i].num_values == 0) {
-                               if (msg_delete_attribute(&msg2, 
-                                                         msg->elements[i].name) != 0) {
+                               if (msg_delete_attribute(module, ldb, msg2, 
+                                                        msg->elements[i].name) != 0) {
+                                       ltdb->last_err_string = "No such attribute";
                                        goto failed;
                                }
                                break;
                        }
                        for (j=0;j<msg->elements[i].num_values;j++) {
-                               if (msg_delete_element(ldb,
-                                                      &msg2, 
+                               if (msg_delete_element(module,
+                                                      msg2, 
                                                       msg->elements[i].name,
                                                       &msg->elements[i].values[j]) != 0) {
+                                       ltdb->last_err_string = "No such attribute";
+                                       goto failed;
+                               }
+                               if (ltdb_index_del_value(module, msg->dn, &msg->elements[i], j) != 0) {
                                        goto failed;
                                }
                        }
@@ -514,106 +683,176 @@ int ltdb_modify_internal(struct ldb_context *ldb, const struct ldb_message *msg)
        }
 
        /* we've made all the mods - save the modified record back into the database */
-       ret = ltdb_store(ldb, &msg2, TDB_MODIFY);
+       ret = ltdb_store(module, msg2, TDB_MODIFY);
 
-       free(tdb_key.dptr);
+       talloc_free(tdb_key.dptr);
        free(tdb_data.dptr);
-       ltdb_unpack_data_free(&msg2);
        return ret;
 
 failed:
-       free(tdb_key.dptr);
+       talloc_free(tdb_key.dptr);
        free(tdb_data.dptr);
-       ltdb_unpack_data_free(&msg2);
        return -1;
 }
 
 /*
   modify a record
 */
-static int ltdb_modify(struct ldb_context *ldb, const struct ldb_message *msg)
+static int ltdb_modify(struct ldb_module *module, const struct ldb_message *msg)
 {
+       struct ltdb_private *ltdb = module->private_data;
        int ret;
 
-       if (ltdb_lock(ldb) != 0) {
+       ltdb->last_err_string = NULL;
+
+       ret = ltdb_check_special_dn(module, msg);
+       if (ret != 0) {
+               return ret;
+       }
+       
+       if (ltdb_lock(module, LDBLOCK) != 0) {
                return -1;
        }
 
-       if (ltdb_cache_load(ldb) != 0) {
-               ltdb_unlock(ldb);
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module, LDBLOCK);
                return -1;
        }
 
-       ret = ltdb_modify_internal(ldb, msg);
+       ret = ltdb_modify_internal(module, msg);
 
        if (ret == 0) {
-               ltdb_modified(ldb, msg->dn);
+               ltdb_modified(module, msg->dn);
        }
 
-       ltdb_unlock(ldb);
+       ltdb_unlock(module, LDBLOCK);
 
        return ret;
 }
 
 /*
-  close database
+  rename a record
 */
-static int ltdb_close(struct ldb_context *ldb)
+static int ltdb_rename(struct ldb_module *module, const char *olddn, const char *newdn)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
        int ret;
+       struct ldb_message *msg;
+       const char *error_str;
+
+       ltdb->last_err_string = NULL;
+
+       if (ltdb_lock(module, LDBLOCK) != 0) {
+               return -1;
+       }
+
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module, LDBLOCK);
+               return -1;
+       }
+
+       msg = talloc(module, struct ldb_message);
+       if (msg == NULL) {
+               goto failed;
+       }
+
+       /* in case any attribute of the message was indexed, we need
+          to fetch the old record */
+       ret = ltdb_search_dn1(module, olddn, msg);
+       if (ret != 1) {
+               /* not finding the old record is an error */
+               goto failed;
+       }
+
+       msg->dn = talloc_strdup(msg, newdn);
+       if (!msg->dn) {
+               goto failed;
+       }
+
+       ret = ltdb_add(module, msg);
+       if (ret == -1) {
+               goto failed;
+       }
+
+       ret = ltdb_delete(module, olddn);
+       error_str = ltdb->last_err_string;
+       if (ret == -1) {
+               ltdb_delete(module, newdn);
+       }
+
+       ltdb->last_err_string = error_str;
 
-       ltdb_cache_free(ldb);
+       talloc_free(msg);
+       ltdb_unlock(module, LDBLOCK);
 
-       ret = tdb_close(ltdb->tdb);
-       free(ltdb);
-       free(ldb);
        return ret;
+
+failed:
+       talloc_free(msg);
+       ltdb_unlock(module, LDBLOCK);
+       return -1;
 }
-                     
+
 
 /*
   return extended error information
 */
-static const char *ltdb_errstring(struct ldb_context *ldb)
+static const char *ltdb_errstring(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = ldb->private_data;
+       struct ltdb_private *ltdb = module->private_data;
+       if (ltdb->last_err_string) {
+               return ltdb->last_err_string;
+       }
        return tdb_errorstr(ltdb->tdb);
 }
 
 
-static const struct ldb_backend_ops ltdb_ops = {
-       ltdb_close, 
-       ltdb_search,
-       ltdb_search_free,
-       ltdb_add,
-       ltdb_modify,
-       ltdb_delete,
-       ltdb_errstring
+static const struct ldb_module_ops ltdb_ops = {
+       .name          = "tdb",
+       .search        = ltdb_search,
+       .search_bytree = ltdb_search_bytree,
+       .add_record    = ltdb_add,
+       .modify_record = ltdb_modify,
+       .delete_record = ltdb_delete,
+       .rename_record = ltdb_rename,
+       .named_lock    = ltdb_lock,
+       .named_unlock  = ltdb_unlock,
+       .errstring     = ltdb_errstring
 };
 
 
+/*
+  destroy the ltdb context
+*/
+static int ltdb_destructor(void *p)
+{
+       struct ltdb_private *ltdb = p;
+       tdb_close(ltdb->tdb);
+       return 0;
+}
+
 /*
   connect to the database
 */
-struct ldb_context *ltdb_connect(const char *url, 
-                                unsigned int flags, 
-                                const char *options[])
+int ltdb_connect(struct ldb_context *ldb, const char *url, 
+                unsigned int flags, const char *options[])
 {
        const char *path;
        int tdb_flags, open_flags;
        struct ltdb_private *ltdb;
        TDB_CONTEXT *tdb;
-       struct ldb_context *ldb;
 
        /* parse the url */
-       if (strncmp(url, "tdb://", 6) != 0) {
-               errno = EINVAL;
-               return NULL;
+       if (strchr(url, ':')) {
+               if (strncmp(url, "tdb://", 6) != 0) {
+                       ldb_debug(ldb, LDB_DEBUG_ERROR, "Invalid tdb URL '%s'", url);
+                       return -1;
+               }
+               path = url+6;
+       } else {
+               path = url;
        }
 
-       path = url+6;
-
        tdb_flags = TDB_DEFAULT;
 
        if (flags & LDB_FLG_RDONLY) {
@@ -625,31 +864,32 @@ struct ldb_context *ltdb_connect(const char *url,
        /* note that we use quite a large default hash size */
        tdb = tdb_open(path, 10000, tdb_flags, open_flags, 0666);
        if (!tdb) {
-               return NULL;
+               ldb_debug(ldb, LDB_DEBUG_ERROR, "Unable to open tdb '%s'", path);
+               return -1;
        }
 
-       ltdb = malloc_p(struct ltdb_private);
+       ltdb = talloc_zero(ldb, struct ltdb_private);
        if (!ltdb) {
                tdb_close(tdb);
-               errno = ENOMEM;
-               return NULL;
+               ldb_oom(ldb);
+               return -1;
        }
 
        ltdb->tdb = tdb;
        ltdb->sequence_number = 0;
 
-       memset(&ltdb->cache, 0, sizeof(ltdb->cache));
+       talloc_set_destructor(ltdb, ltdb_destructor);
 
-       ldb = malloc_p(struct ldb_context);
-       if (!ldb) {
-               tdb_close(tdb);
-               free(ltdb);
-               errno = ENOMEM;
-               return NULL;
+       ldb->modules = talloc(ldb, struct ldb_module);
+       if (!ldb->modules) {
+               ldb_oom(ldb);
+               talloc_free(ltdb);
+               return -1;
        }
+       ldb->modules->ldb = ldb;
+       ldb->modules->prev = ldb->modules->next = NULL;
+       ldb->modules->private_data = ltdb;
+       ldb->modules->ops = &ltdb_ops;
 
-       ldb->private_data = ltdb;
-       ldb->ops = &ltdb_ops;
-
-       return ldb;
+       return 0;
 }