Fix the mess with ldb includes.
[metze/samba/wip.git] / source4 / lib / ldb / ldb_tdb / ldb_cache.c
index 5634e9ad16328f14ba486bcac33e91ebb7ffe07c..43b965f239a0c419b8d5475267d12a41ae4c40bc 100644 (file)
@@ -10,7 +10,7 @@
    This library is free software; you can redistribute it and/or
    modify it under the terms of the GNU Lesser General Public
    License as published by the Free Software Foundation; either
-   version 2 of the License, or (at your option) any later version.
+   version 3 of the License, or (at your option) any later version.
 
    This library is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ -18,8 +18,7 @@
    Lesser General Public License for more details.
 
    You should have received a copy of the GNU Lesser General Public
-   License along with this library; if not, write to the Free Software
-   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+   License along with this library; if not, see <http://www.gnu.org/licenses/>.
 */
 
 /*
  *  Author: Andrew Tridgell
  */
 
-#include "includes.h"
-#include "ldb/include/includes.h"
-
-#include "ldb/ldb_tdb/ldb_tdb.h"
+#include "ldb_tdb.h"
 
 #define LTDB_FLAG_CASE_INSENSITIVE (1<<0)
 #define LTDB_FLAG_INTEGER          (1<<1)
 #define LTDB_FLAG_HIDDEN           (1<<2)
-#define LTDB_FLAG_OBJECTCLASS      (1<<3)
 
 /* valid attribute flags */
 static const struct {
@@ -60,10 +55,14 @@ static const struct {
 */
 static void ltdb_attributes_unload(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
+       struct ldb_context *ldb;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
        struct ldb_message *msg;
        int i;
 
+       ldb = ldb_module_get_ctx(module);
+
        if (ltdb->cache->attributes == NULL) {
                /* no previously loaded attributes */
                return;
@@ -71,13 +70,7 @@ static void ltdb_attributes_unload(struct ldb_module *module)
 
        msg = ltdb->cache->attributes;
        for (i=0;i<msg->num_elements;i++) {
-               const struct ldb_attrib_handler *h;
-               /* this is rather ugly - a consequence of const handling */
-               h = ldb_attrib_handler(module->ldb, msg->elements[i].name);
-               ldb_remove_attrib_handler(module->ldb, msg->elements[i].name);
-               if (strcmp(h->attr, msg->elements[i].name) == 0) {
-                       talloc_steal(msg, h->attr);
-               }
+               ldb_schema_attribute_remove(ldb, msg->elements[i].name);
        }
 
        talloc_free(ltdb->cache->attributes);
@@ -113,29 +106,35 @@ static int ltdb_attributes_flags(struct ldb_message_element *el, unsigned *v)
 */
 static int ltdb_attributes_load(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
+       struct ldb_context *ldb;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
        struct ldb_message *msg = ltdb->cache->attributes;
        struct ldb_dn *dn;
-       int i;
+       int i, r;
+
+       ldb = ldb_module_get_ctx(module);
 
-       dn = ldb_dn_explode(module->ldb, LTDB_ATTRIBUTES);
+       dn = ldb_dn_new(module, ldb, LTDB_ATTRIBUTES);
        if (dn == NULL) goto failed;
 
-       if (ltdb_search_dn1(module, dn, msg) == -1) {
-               talloc_free(dn);
+       r = ltdb_search_dn1(module, dn, msg);
+       talloc_free(dn);
+       if (r != LDB_SUCCESS && r != LDB_ERR_NO_SUCH_OBJECT) {
                goto failed;
        }
-       talloc_free(dn);
+       if (r == LDB_ERR_NO_SUCH_OBJECT) {
+               return 0;
+       }
        /* mapping these flags onto ldap 'syntaxes' isn't strictly correct,
           but its close enough for now */
        for (i=0;i<msg->num_elements;i++) {
                unsigned flags;
                const char *syntax;
-               const struct ldb_attrib_handler *h;
-               struct ldb_attrib_handler h2;
+               const struct ldb_schema_syntax *s;
 
                if (ltdb_attributes_flags(&msg->elements[i], &flags) != 0) {
-                       ldb_debug(module->ldb, LDB_DEBUG_ERROR, "Invalid @ATTRIBUTES element for '%s'\n", msg->elements[i].name);
+                       ldb_debug(ldb, LDB_DEBUG_ERROR, "Invalid @ATTRIBUTES element for '%s'\n", msg->elements[i].name);
                        goto failed;
                }
                switch (flags & ~LTDB_FLAG_HIDDEN) {
@@ -149,58 +148,23 @@ static int ltdb_attributes_load(struct ldb_module *module)
                        syntax = LDB_SYNTAX_INTEGER;
                        break;
                default:
-                       ldb_debug(module->ldb, LDB_DEBUG_ERROR, 
+                       ldb_debug(ldb, LDB_DEBUG_ERROR, 
                                  "Invalid flag combination 0x%x for '%s' in @ATTRIBUTES\n",
                                  flags, msg->elements[i].name);
                        goto failed;
                }
 
-               h = ldb_attrib_handler_syntax(module->ldb, syntax);
-               if (h == NULL) {
-                       ldb_debug(module->ldb, LDB_DEBUG_ERROR, 
+               s = ldb_standard_syntax_by_name(ldb, syntax);
+               if (s == NULL) {
+                       ldb_debug(ldb, LDB_DEBUG_ERROR, 
                                  "Invalid attribute syntax '%s' for '%s' in @ATTRIBUTES\n",
                                  syntax, msg->elements[i].name);
                        goto failed;
                }
-               h2 = *h;
-               h2.attr = talloc_strdup(module, msg->elements[i].name);
-               if (ldb_set_attrib_handlers(module->ldb, &h2, 1) != 0) {
-                       goto failed;
-               }
-       }
-
-       return 0;
-failed:
-       return -1;
-}
-
-
-/*
-  register any subclasses from @SUBCLASSES
-*/
-static int ltdb_subclasses_load(struct ldb_module *module)
-{
-       struct ltdb_private *ltdb = module->private_data;
-       struct ldb_message *msg = ltdb->cache->subclasses;
-       struct ldb_dn *dn;
-       int i, j;
-
-       dn = ldb_dn_explode(module->ldb, LTDB_SUBCLASSES);
-       if (dn == NULL) goto failed;
 
-       if (ltdb_search_dn1(module, dn, msg) == -1) {
-               talloc_free(dn);
-               goto failed;
-       }
-       talloc_free(dn);
-
-       for (i=0;i<msg->num_elements;i++) {
-               struct ldb_message_element *el = &msg->elements[i];
-               for (j=0;j<el->num_values;j++) {
-                       if (ldb_subclass_add(module->ldb, el->name, 
-                                            (char *)el->values[j].data) != 0) {
-                               goto failed;
-                       }
+               flags |= LDB_ATTR_FLAG_ALLOCATED;
+               if (ldb_schema_attribute_add_with_syntax(ldb, msg->elements[i].name, flags, s) != 0) {
+                       goto failed;
                }
        }
 
@@ -210,36 +174,14 @@ failed:
 }
 
 
-/*
-  de-register any @SUBCLASSES
-*/
-static void ltdb_subclasses_unload(struct ldb_module *module)
-{
-       struct ltdb_private *ltdb = module->private_data;
-       struct ldb_message *msg;
-       int i;
-
-       if (ltdb->cache->subclasses == NULL) {
-               /* no previously loaded subclasses */
-               return;
-       }
-
-       msg = ltdb->cache->subclasses;
-       for (i=0;i<msg->num_elements;i++) {
-               ldb_subclass_remove(module->ldb, msg->elements[i].name);
-       }
-
-       talloc_free(ltdb->cache->subclasses);
-       ltdb->cache->subclasses = NULL;
-}
-
-
 /*
   initialise the baseinfo record
 */
 static int ltdb_baseinfo_init(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
+       struct ldb_context *ldb;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
        struct ldb_message *msg;
        struct ldb_message_element el;
        struct ldb_val val;
@@ -249,6 +191,8 @@ static int ltdb_baseinfo_init(struct ldb_module *module)
           out. */
        const char *initial_sequence_number = "1";
 
+       ldb = ldb_module_get_ctx(module);
+
        ltdb->sequence_number = atof(initial_sequence_number);
 
        msg = talloc(ltdb, struct ldb_message);
@@ -258,7 +202,7 @@ static int ltdb_baseinfo_init(struct ldb_module *module)
 
        msg->num_elements = 1;
        msg->elements = &el;
-       msg->dn = ldb_dn_explode(msg, LTDB_BASEINFO);
+       msg->dn = ldb_dn_new(msg, ldb, LTDB_BASEINFO);
        if (!msg->dn) {
                goto failed;
        }
@@ -284,7 +228,7 @@ static int ltdb_baseinfo_init(struct ldb_module *module)
 failed:
        talloc_free(msg);
        errno = ENOMEM;
-       return -1;
+       return LDB_ERR_OPERATIONS_ERROR;
 }
 
 /*
@@ -292,7 +236,8 @@ failed:
  */
 static void ltdb_cache_free(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
 
        ltdb->sequence_number = 0;
        talloc_free(ltdb->cache);
@@ -305,7 +250,6 @@ static void ltdb_cache_free(struct ldb_module *module)
 int ltdb_cache_reload(struct ldb_module *module)
 {
        ltdb_attributes_unload(module);
-       ltdb_subclasses_unload(module);
        ltdb_cache_free(module);
        return ltdb_cache_load(module);
 }
@@ -315,91 +259,120 @@ int ltdb_cache_reload(struct ldb_module *module)
 */
 int ltdb_cache_load(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
-       struct ldb_dn *baseinfo_dn = NULL;
+       struct ldb_context *ldb;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
+       struct ldb_dn *baseinfo_dn = NULL, *options_dn = NULL;
        struct ldb_dn *indexlist_dn = NULL;
-       double seq;
+       uint64_t seq;
+       struct ldb_message *baseinfo = NULL, *options = NULL;
+       int r;
+
+       ldb = ldb_module_get_ctx(module);
+
+       /* a very fast check to avoid extra database reads */
+       if (ltdb->cache != NULL && 
+           tdb_get_seqnum(ltdb->tdb) == ltdb->tdb_seqnum) {
+               return 0;
+       }
 
        if (ltdb->cache == NULL) {
                ltdb->cache = talloc_zero(ltdb, struct ltdb_cache);
                if (ltdb->cache == NULL) goto failed;
                ltdb->cache->indexlist = talloc_zero(ltdb->cache, struct ldb_message);
-               ltdb->cache->subclasses = talloc_zero(ltdb->cache, struct ldb_message);
                ltdb->cache->attributes = talloc_zero(ltdb->cache, struct ldb_message);
                if (ltdb->cache->indexlist == NULL ||
-                   ltdb->cache->subclasses == NULL ||
                    ltdb->cache->attributes == NULL) {
                        goto failed;
                }
        }
 
-       talloc_free(ltdb->cache->baseinfo);
-       ltdb->cache->baseinfo = talloc(ltdb->cache, struct ldb_message);
-       if (ltdb->cache->baseinfo == NULL) goto failed;
+       baseinfo = talloc(ltdb->cache, struct ldb_message);
+       if (baseinfo == NULL) goto failed;
 
-       baseinfo_dn = ldb_dn_explode(module->ldb, LTDB_BASEINFO);
+       baseinfo_dn = ldb_dn_new(module, ldb, LTDB_BASEINFO);
        if (baseinfo_dn == NULL) goto failed;
 
-       if (ltdb_search_dn1(module, baseinfo_dn, ltdb->cache->baseinfo) == -1) {
+       r= ltdb_search_dn1(module, baseinfo_dn, baseinfo);
+       if (r != LDB_SUCCESS && r != LDB_ERR_NO_SUCH_OBJECT) {
                goto failed;
        }
        
        /* possibly initialise the baseinfo */
-       if (!ltdb->cache->baseinfo->dn) {
-               if (ltdb_baseinfo_init(module) != 0) {
+       if (r == LDB_ERR_NO_SUCH_OBJECT) {
+               if (ltdb_baseinfo_init(module) != LDB_SUCCESS) {
                        goto failed;
                }
-               if (ltdb_search_dn1(module, baseinfo_dn, ltdb->cache->baseinfo) != 1) {
+               if (ltdb_search_dn1(module, baseinfo_dn, baseinfo) != LDB_SUCCESS) {
                        goto failed;
                }
        }
 
+       ltdb->tdb_seqnum = tdb_get_seqnum(ltdb->tdb);
+
        /* if the current internal sequence number is the same as the one
           in the database then assume the rest of the cache is OK */
-       seq = ldb_msg_find_attr_as_double(ltdb->cache->baseinfo, LTDB_SEQUENCE_NUMBER, 0);
+       seq = ldb_msg_find_attr_as_uint64(baseinfo, LTDB_SEQUENCE_NUMBER, 0);
        if (seq == ltdb->sequence_number) {
                goto done;
        }
        ltdb->sequence_number = seq;
 
+       /* Read an interpret database options */
+       options = talloc(ltdb->cache, struct ldb_message);
+       if (options == NULL) goto failed;
+
+       options_dn = ldb_dn_new(options, ldb, LTDB_OPTIONS);
+       if (options_dn == NULL) goto failed;
+
+       r= ltdb_search_dn1(module, options_dn, options);
+       if (r != LDB_SUCCESS && r != LDB_ERR_NO_SUCH_OBJECT) {
+               goto failed;
+       }
+       
+       /* set flag for checking base DN on searches */
+       if (r == LDB_SUCCESS) {
+               ltdb->check_base = ldb_msg_find_attr_as_bool(options, LTDB_CHECK_BASE, false);
+       } else {
+               ltdb->check_base = false;
+       }
+
        talloc_free(ltdb->cache->last_attribute.name);
        memset(&ltdb->cache->last_attribute, 0, sizeof(ltdb->cache->last_attribute));
 
        ltdb_attributes_unload(module);
-       ltdb_subclasses_unload(module);
 
        talloc_free(ltdb->cache->indexlist);
-       talloc_free(ltdb->cache->subclasses);
 
        ltdb->cache->indexlist = talloc_zero(ltdb->cache, struct ldb_message);
-       ltdb->cache->subclasses = talloc_zero(ltdb->cache, struct ldb_message);
        ltdb->cache->attributes = talloc_zero(ltdb->cache, struct ldb_message);
        if (ltdb->cache->indexlist == NULL ||
-           ltdb->cache->subclasses == NULL ||
            ltdb->cache->attributes == NULL) {
                goto failed;
        }
            
-       indexlist_dn = ldb_dn_explode(module->ldb, LTDB_INDEXLIST);
+       indexlist_dn = ldb_dn_new(module, ldb, LTDB_INDEXLIST);
        if (indexlist_dn == NULL) goto failed;
 
-       if (ltdb_search_dn1(module, indexlist_dn, ltdb->cache->indexlist) == -1) {
+       r = ltdb_search_dn1(module, indexlist_dn, ltdb->cache->indexlist);
+       if (r != LDB_SUCCESS && r != LDB_ERR_NO_SUCH_OBJECT) {
                goto failed;
        }
 
        if (ltdb_attributes_load(module) == -1) {
                goto failed;
        }
-       if (ltdb_subclasses_load(module) == -1) {
-               goto failed;
-       }
 
 done:
+       talloc_free(options);
+       talloc_free(baseinfo);
        talloc_free(baseinfo_dn);
        talloc_free(indexlist_dn);
        return 0;
 
 failed:
+       talloc_free(options);
+       talloc_free(baseinfo);
        talloc_free(baseinfo_dn);
        talloc_free(indexlist_dn);
        return -1;
@@ -411,101 +384,80 @@ failed:
 */
 int ltdb_increase_sequence_number(struct ldb_module *module)
 {
-       struct ltdb_private *ltdb = module->private_data;
+       struct ldb_context *ldb;
+       void *data = ldb_module_get_private(module);
+       struct ltdb_private *ltdb = talloc_get_type(data, struct ltdb_private);
        struct ldb_message *msg;
-       struct ldb_message_element el;
+       struct ldb_message_element el[2];
        struct ldb_val val;
+       struct ldb_val val_time;
+       time_t t = time(NULL);
        char *s = NULL;
        int ret;
 
+       ldb = ldb_module_get_ctx(module);
+
        msg = talloc(ltdb, struct ldb_message);
        if (msg == NULL) {
                errno = ENOMEM;
-               return -1;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       s = talloc_asprintf(msg, "%.0f", ltdb->sequence_number+1);
+       s = talloc_asprintf(msg, "%llu", ltdb->sequence_number+1);
        if (!s) {
                errno = ENOMEM;
-               return -1;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       msg->num_elements = 1;
-       msg->elements = &el;
-       msg->dn = ldb_dn_explode(msg, LTDB_BASEINFO);
+       msg->num_elements = ARRAY_SIZE(el);
+       msg->elements = el;
+       msg->dn = ldb_dn_new(msg, ldb, LTDB_BASEINFO);
        if (msg->dn == NULL) {
                talloc_free(msg);
                errno = ENOMEM;
-               return -1;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
-       el.name = talloc_strdup(msg, LTDB_SEQUENCE_NUMBER);
-       if (el.name == NULL) {
+       el[0].name = talloc_strdup(msg, LTDB_SEQUENCE_NUMBER);
+       if (el[0].name == NULL) {
                talloc_free(msg);
                errno = ENOMEM;
-               return -1;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
-       el.values = &val;
-       el.num_values = 1;
-       el.flags = LDB_FLAG_MOD_REPLACE;
+       el[0].values = &val;
+       el[0].num_values = 1;
+       el[0].flags = LDB_FLAG_MOD_REPLACE;
        val.data = (uint8_t *)s;
        val.length = strlen(s);
 
-       ret = ltdb_modify_internal(module, msg);
-
-       talloc_free(msg);
-
-       if (ret == 0) {
-               ltdb->sequence_number += 1;
-       }
-
-       return ret;
-}
-
-
-/*
-  return the attribute flags from the @ATTRIBUTES record 
-  for the given attribute
-*/
-int ltdb_attribute_flags(struct ldb_module *module, const char *attr_name)
-{
-       struct ltdb_private *ltdb = module->private_data;
-       const struct ldb_message_element *attr_el;
-       int i, j, ret=0;
-
-       if (ltdb->cache->last_attribute.name &&
-           ldb_attr_cmp(ltdb->cache->last_attribute.name, attr_name) == 0) {
-               return ltdb->cache->last_attribute.flags;
+       el[1].name = talloc_strdup(msg, LTDB_MOD_TIMESTAMP);
+       if (el[1].name == NULL) {
+               talloc_free(msg);
+               errno = ENOMEM;
+               return LDB_ERR_OPERATIONS_ERROR;
        }
+       el[1].values = &val_time;
+       el[1].num_values = 1;
+       el[1].flags = LDB_FLAG_MOD_REPLACE;
 
-       /* objectclass is a special default case */
-       if (ldb_attr_cmp(attr_name, LTDB_OBJECTCLASS) == 0) {
-               ret = LTDB_FLAG_OBJECTCLASS | LTDB_FLAG_CASE_INSENSITIVE;
+       s = ldb_timestring(msg, t);
+       if (s == NULL) {
+               return LDB_ERR_OPERATIONS_ERROR;
        }
 
-       attr_el = ldb_msg_find_element(ltdb->cache->attributes, attr_name);
+       val_time.data = (uint8_t *)s;
+       val_time.length = strlen(s);
 
-       if (!attr_el) {
-               /* check if theres a wildcard attribute */
-               attr_el = ldb_msg_find_element(ltdb->cache->attributes, "*");
+       ret = ltdb_modify_internal(module, msg);
 
-               if (!attr_el) {
-                       return ret;
-               }
-       }
+       talloc_free(msg);
 
-       for (i = 0; i < attr_el->num_values; i++) {
-               for (j=0; ltdb_valid_attr_flags[j].name; j++) {
-                       if (strcmp(ltdb_valid_attr_flags[j].name, 
-                                  (char *)attr_el->values[i].data) == 0) {
-                               ret |= ltdb_valid_attr_flags[j].value;
-                       }
-               }
+       if (ret == LDB_SUCCESS) {
+               ltdb->sequence_number += 1;
        }
 
-       talloc_free(ltdb->cache->last_attribute.name);
-
-       ltdb->cache->last_attribute.name = talloc_strdup(ltdb->cache, attr_name);
-       ltdb->cache->last_attribute.flags = ret;
+       /* updating the tdb_seqnum here avoids us reloading the cache
+          records due to our own modification */
+       ltdb->tdb_seqnum = tdb_get_seqnum(ltdb->tdb);
 
        return ret;
 }