r3783: - don't use make proto for ldb anymore
[kamenim/samba.git] / source4 / lib / ldb / ldb_tdb / ldb_tdb.c
index 34cff417944ba69c3db5d7c615a8857102f3e6ec..5f6af43f94d3c96a9a90cc9b4b4ec6300c4fe013 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"
 
 /*
   form a TDB_DATA for a record key
   caller frees
+
+  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(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;
+       const char *prefix = LTDB_INDEX ":";
+       const char *s;
+       int flags;
+
+       /*
+         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 whole dn
+         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 = ldb_strndup(ldb, dn+strlen(prefix), (s-(dn+strlen(prefix))));
+               if (!attr_name) {
+                       goto failed;
+               }
+               flags = ltdb_attribute_flags(module, attr_name);
+               
+               if (flags & LTDB_FLAG_CASE_INSENSITIVE) {
+                       dn_folded = ldb_casefold(ldb, dn);
+               } else {
+                       attr_name_folded = ldb_casefold(ldb, attr_name);
+                       if (!attr_name_folded) {
+                               goto failed;
+                       }
+                       ldb_asprintf(ldb, &dn_folded, "%s:%s:%s",
+                                prefix, attr_name_folded,
+                                s+1);
+                       ldb_free(ldb, attr_name_folded);
+               }
+               ldb_free(ldb, attr_name);
+       } else {
+               dn_folded = ldb_casefold(ldb, dn);
+       }
+
+       if (!dn_folded) {
+               goto failed;
+       }
+
+       ldb_asprintf(ldb, &key_str, "DN=%s", dn_folded);
+       ldb_free(ldb, dn_folded);
 
-       asprintf(&key_str, "DN=%s", dn);
        if (!key_str) {
-               errno = ENOMEM;
-               key.dptr = NULL;
-               key.dsize = 0;
-               return key;
+               goto failed;
        }
 
        key.dptr = key_str;
        key.dsize = strlen(key_str)+1;
 
        return key;
+
+failed:
+       errno = ENOMEM;
+       key.dptr = NULL;
+       key.dsize = 0;
+       return key;
 }
 
 /*
   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)
 {
-       struct ltdb_private *ltdb = ldb->private;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA key;
        int ret;
 
-       key = ltdb_key("LDBLOCK");
+       key = ltdb_key(module, "LDBLOCK");
        if (!key.dptr) {
                return -1;
        }
 
        ret = tdb_chainlock(ltdb->tdb, key);
 
-       free(key.dptr);
+       ldb_free(ldb, key.dptr);
 
        return ret;
 }
@@ -82,38 +144,62 @@ static int ltdb_lock(struct ldb_context *ldb)
 /*
   unlock the database after a ltdb_lock()
 */
-static void ltdb_unlock(struct ldb_context *ldb)
+static void ltdb_unlock(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = ldb->private;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA key;
 
-       key = ltdb_key("LDBLOCK");
+       key = ltdb_key(module, "LDBLOCK");
        if (!key.dptr) {
                return;
        }
 
        tdb_chainunlock(ltdb->tdb, key);
 
-       free(key.dptr);
+       ldb_free(ldb, key.dptr);
+}
+
+
+/*
+  we've made a modification to a dn - possibly reindex and 
+  update sequence number
+*/
+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(module);
+       }
+
+       if (ret == 0 &&
+           strcmp(dn, LTDB_BASEINFO) != 0) {
+               ret = ltdb_increase_sequence_number(module);
+       }
+
+       return ret;
 }
 
 /*
   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;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA tdb_key, tdb_data;
        int ret;
 
-       tdb_key = ltdb_key(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);
+               ldb_free(ldb, tdb_key.dptr);
                return -1;
        }
 
@@ -122,14 +208,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);
+       ldb_free(ldb, tdb_key.dptr);
+       ldb_free(ldb, tdb_data.dptr);
 
        return ret;
 }
@@ -138,22 +224,29 @@ 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;
+
+       if (ltdb_lock(module) != 0) {
                return -1;
        }
-       
-       ret = ltdb_store(ldb, msg, TDB_INSERT);
 
-       if (strcmp(msg->dn, "@INDEXLIST") == 0) {
-               ltdb_reindex(ldb);
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module);
+               return -1;
        }
+       
+       ret = ltdb_store(module, msg, TDB_INSERT);
 
-       ltdb_unlock(ldb);
+       if (ret == 0) {
+               ltdb_modified(module, msg->dn);
+       }
 
+       ltdb_unlock(module);
        return ret;
 }
 
@@ -162,19 +255,20 @@ 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;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        TDB_DATA tdb_key;
        int ret;
 
-       tdb_key = ltdb_key(dn);
+       tdb_key = ltdb_key(module, dn);
        if (!tdb_key.dptr) {
                return -1;
        }
 
        ret = tdb_delete(ltdb->tdb, tdb_key);
-       free(tdb_key.dptr);
+       ldb_free(ldb, tdb_key.dptr);
 
        return ret;
 }
@@ -182,43 +276,51 @@ 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;
 
-       if (ltdb_lock(ldb) != 0) {
+       ltdb->last_err_string = NULL;
+
+       if (ltdb_lock(module) != 0) {
+               return -1;
+       }
+
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module);
                return -1;
        }
 
        /* 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);
+               ltdb_search_dn1_free(module, &msg);
                goto failed;
        }
 
        /* remove any indexed attributes */
-       ret = ltdb_index_del(ldb, &msg);
+       ret = ltdb_index_del(module, &msg);
 
-       ltdb_search_dn1_free(ldb, &msg);
+       ltdb_search_dn1_free(module, &msg);
 
-       if (strcmp(dn, "@INDEXLIST") == 0) {
-               ltdb_reindex(ldb);
+       if (ret == 0) {
+               ltdb_modified(module, dn);
        }
 
-       ltdb_unlock(ldb);
+       ltdb_unlock(module);
        return ret;
 
 failed:
-       ltdb_unlock(ldb);
+       ltdb_unlock(module);
        return -1;
 }
 
@@ -232,9 +334,9 @@ 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 (strcmp(msg->elements[i].name, name) == 0) {
+               if (ldb_attr_cmp(msg->elements[i].name, name) == 0) {
                        return i;
                }
        }
@@ -249,12 +351,13 @@ 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, 
+       e2 = ldb_realloc_p(ldb, msg->elements, struct ldb_message_element, 
                       msg->num_elements+1);
        if (!e2) {
                errno = ENOMEM;
@@ -269,9 +372,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 = ldb_malloc_array_p(ldb, struct ldb_val, el->num_values);
                if (!e2->values) {
-                       free(e2->name);
                        errno = ENOMEM;
                        return -1;
                }
@@ -289,27 +391,28 @@ 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_context *ldb,
+                               struct ldb_message *msg, const char *name)
 {
-       int i, count=0;
+       unsigned int i, count=0;
        struct ldb_message_element *el2;
 
-       el2 = malloc_array_p(struct ldb_message_element, msg->num_elements);
+       el2 = ldb_malloc_array_p(ldb, struct ldb_message_element, msg->num_elements);
        if (!el2) {
                errno = ENOMEM;
                return -1;
        }
 
        for (i=0;i<msg->num_elements;i++) {
-               if (strcmp(msg->elements[i].name, name) != 0) {
+               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);
+                       ldb_free(ldb, msg->elements[i].values);
                }
        }
 
        msg->num_elements = count;
-       if (msg->elements) free(msg->elements);
+       ldb_free(ldb, msg->elements);
        msg->elements = el2;
 
        return 0;
@@ -320,71 +423,78 @@ 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_message *msg, 
+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(&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(ldb, msg, name);
+                       }
                        return 0;
                }
        }
-       
+
        return -1;
 }
 
+
 /*
-  modify a record
+  modify a record - internal interface
 
   yuck - this is O(n^2). Luckily n is usually small so we probably
   get away with it, but if we ever have really large attribute lists 
   then we'll need to look at this again
 */
-static int ltdb_modify(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;
+       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;
-
-       if (ltdb_lock(ldb) != 0) {
-               return -1;
-       }
+       unsigned i, j;
+       int ret;
 
-       tdb_key = ltdb_key(msg->dn);
+       tdb_key = ltdb_key(module, msg->dn);
        if (!tdb_key.dptr) {
-               goto unlock_fail;
+               return -1;
        }
 
        tdb_data = tdb_fetch(ltdb->tdb, tdb_key);
        if (!tdb_data.dptr) {
-               free(tdb_key.dptr);
-               goto unlock_fail;
+               ldb_free(ldb, 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);
+               ldb_free(ldb, tdb_key.dptr);
                free(tdb_data.dptr);
-               goto unlock_fail;
+               return -1;
        }
 
-       msg2.dn = msg->dn;
+       if (!msg2.dn) {
+               msg2.dn = msg->dn;
+       }
 
        for (i=0;i<msg->num_elements;i++) {
                switch (msg->elements[i].flags & LDB_FLAG_MOD_MASK) {
@@ -394,22 +504,22 @@ static int ltdb_modify(struct ldb_context *ldb, const struct ldb_message *msg)
                           already exists */
                        ret = find_element(&msg2, msg->elements[i].name);
                        if (ret != -1) {
-                               errno = EEXIST;
+                               ltdb->last_err_string = "Attribute exists";
                                goto failed;
                        }
-                       if (msg_add_element(&msg2, &msg->elements[i]) != 0) {
+                       if (msg_add_element(ldb, &msg2, &msg->elements[i]) != 0) {
                                goto failed;
                        }
                        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(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;
@@ -418,16 +528,19 @@ static int ltdb_modify(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(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(&msg2, 
-                                                       msg->elements[i].name,
-                                                       &msg->elements[i].values[j]) != 0) {
+                               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;
                                }
                        }
@@ -436,39 +549,122 @@ static int ltdb_modify(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);
 
-       if (strcmp(msg2.dn, "@INDEXLIST") == 0) {
-               ltdb_reindex(ldb);
-       }
-
-       free(tdb_key.dptr);
+       ldb_free(ldb, tdb_key.dptr);
        free(tdb_data.dptr);
-       ltdb_unpack_data_free(&msg2);
-       ltdb_unlock(ldb);
-
+       ltdb_unpack_data_free(module, &msg2);
        return ret;
 
 failed:
-       free(tdb_key.dptr);
+       ldb_free(ldb, tdb_key.dptr);
        free(tdb_data.dptr);
-       ltdb_unpack_data_free(&msg2);
+       ltdb_unpack_data_free(module, &msg2);
+       return -1;
+}
 
-unlock_fail:
-       ltdb_unlock(ldb);
-       
+/*
+  modify a record
+*/
+static int ltdb_modify(struct ldb_module *module, const struct ldb_message *msg)
+{
+       struct ltdb_private *ltdb = module->private_data;
+       int ret;
+
+       ltdb->last_err_string = NULL;
+
+       if (ltdb_lock(module) != 0) {
+               return -1;
+       }
+
+       if (ltdb_cache_load(module) != 0) {
+               ltdb_unlock(module);
+               return -1;
+       }
+
+       ret = ltdb_modify_internal(module, msg);
+
+       if (ret == 0) {
+               ltdb_modified(module, msg->dn);
+       }
+
+       ltdb_unlock(module);
+
+       return ret;
+}
+
+/*
+  rename a record
+*/
+static int ltdb_rename(struct ldb_module *module, const char *olddn, const char *newdn)
+{
+       struct ldb_context *ldb = module->ldb;
+       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) != 0) {
+               return -1;
+       }
+
+       /* 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 = ldb_strdup(ldb,newdn);
+       if (!msg.dn) {
+               ltdb_search_dn1_free(module, &msg);
+               goto failed;
+       }
+
+       ret = ltdb_add(module, &msg);
+       if (ret == -1) {
+               ldb_free(ldb, msg.dn);
+               ltdb_search_dn1_free(module, &msg);
+               goto failed;
+       }
+       ldb_free(ldb, msg.dn);
+       ltdb_search_dn1_free(module, &msg);
+
+       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_unlock(module);
+
+       return ret;
+failed:
+       ltdb_unlock(module);
        return -1;
 }
 
 /*
   close database
 */
-static int ltdb_close(struct ldb_context *ldb)
+static int ltdb_close(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = ldb->private;
+       struct ldb_context *ldb = module->ldb;
+       struct ltdb_private *ltdb = module->private_data;
        int ret;
+
+       ltdb->last_err_string = NULL;
+
+       ltdb_cache_free(module);
+       ldb_set_alloc(ldb, NULL, NULL);
+
        ret = tdb_close(ltdb->tdb);
-       free(ltdb);
+       ldb_free(ldb, ltdb);
        free(ldb);
        return ret;
 }
@@ -477,21 +673,27 @@ static int ltdb_close(struct ldb_context *ldb)
 /*
   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;
+       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 = {
+static const struct ldb_module_ops ltdb_ops = {
+       "tdb",
        ltdb_close, 
        ltdb_search,
        ltdb_search_free,
        ltdb_add,
        ltdb_modify,
        ltdb_delete,
-       ltdb_errstring
+       ltdb_rename,
+       ltdb_errstring,
+       ltdb_cache_free
 };
 
 
@@ -508,13 +710,22 @@ struct ldb_context *ltdb_connect(const char *url,
        TDB_CONTEXT *tdb;
        struct ldb_context *ldb;
 
-       /* parse the url */
-       if (strncmp(url, "tdb://", 6) != 0) {
-               errno = EINVAL;
+       ldb = calloc(1, sizeof(struct ldb_context));
+       if (!ldb) {
+               errno = ENOMEM;
                return NULL;
        }
 
-       path = url+6;
+       /* parse the url */
+       if (strchr(url, ':')) {
+               if (strncmp(url, "tdb://", 6) != 0) {
+                       errno = EINVAL;
+                       return NULL;
+               }
+               path = url+6;
+       } else {
+               path = url;
+       }
 
        tdb_flags = TDB_DEFAULT;
 
@@ -524,31 +735,37 @@ struct ldb_context *ltdb_connect(const char *url,
                open_flags = O_CREAT | O_RDWR;
        }
 
-       tdb = tdb_open(path, 0, tdb_flags, open_flags, 0666);
+       /* note that we use quite a large default hash size */
+       tdb = tdb_open(path, 10000, tdb_flags, open_flags, 0666);
        if (!tdb) {
+               free(ldb);
                return NULL;
        }
 
-       ltdb = malloc_p(struct ltdb_private);
+       ltdb = ldb_malloc_p(ldb, struct ltdb_private);
        if (!ltdb) {
                tdb_close(tdb);
+               free(ldb);
                errno = ENOMEM;
                return NULL;
        }
 
        ltdb->tdb = tdb;
-       
+       ltdb->sequence_number = 0;
 
-       ldb = malloc_p(struct ldb_context);
-       if (!ldb) {
+       memset(&ltdb->cache, 0, sizeof(ltdb->cache));
+
+       ldb->modules = ldb_malloc_p(ldb, struct ldb_module);
+       if (!ldb->modules) {
                tdb_close(tdb);
-               free(ltdb);
+               free(ldb);
                errno = ENOMEM;
                return NULL;
        }
-
-       ldb->private = ltdb;
-       ldb->ops = &ltdb_ops;
+       ldb->modules->ldb = ldb;
+       ldb->modules->prev = ldb->modules->next = NULL;
+       ldb->modules->private_data = ltdb;
+       ldb->modules->ops = &ltdb_ops;
 
        return ldb;
 }