diff options
Diffstat (limited to 'source4')
-rw-r--r-- | source4/lib/ldb/common/ldb_modules.c | 265 | ||||
-rw-r--r-- | source4/lib/ldb/include/dlinklist.h | 93 | ||||
-rw-r--r-- | source4/lib/ldb/modules/skel.c | 123 | ||||
-rw-r--r-- | source4/lib/ldb/modules/timestamps.c | 277 |
4 files changed, 758 insertions, 0 deletions
diff --git a/source4/lib/ldb/common/ldb_modules.c b/source4/lib/ldb/common/ldb_modules.c new file mode 100644 index 0000000000..a7cfd81d46 --- /dev/null +++ b/source4/lib/ldb/common/ldb_modules.c @@ -0,0 +1,265 @@ + +/* + ldb database library + + Copyright (C) Simo Sorce 2004 + + ** NOTE! The following LGPL license applies to the ldb + ** library. This does NOT imply that all of Samba is released + ** under the LGPL + + 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. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + 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 +*/ + +/* + * Name: ldb + * + * Component: ldb modules core + * + * Description: core modules routines + * + * Author: Simo Sorce + */ + +#include "includes.h" +#include "dlinklist.h" +#include <sys/types.h> +#include <sys/stat.h> +#include <unistd.h> + +#define LDB_MODULE_PREFIX "modules" +#define LDB_MODULE_PREFIX_LEN 7 +#define LDB_MODULE_SEP ':' + +int ldb_load_modules(struct ldb_context *ldb, const char *options[]) +{ + struct ldb_module *current; + char **modules; + char *p, *q; + int pn, i; + + /* find out which modules we are requested to activate */ + modules = NULL; + pn = 0; + + if (options) { + + for (i = 0; options[i] != NULL; i++) { + + if (strncmp(options[i], LDB_MODULE_PREFIX, LDB_MODULE_PREFIX_LEN) == 0) { + + p = q = ldb_strdup(ldb, &options[i][LDB_MODULE_PREFIX_LEN]); + if (*q != ':') { + ldb_free(ldb, q); + return -1; + } + do { + *p = '\0'; + q = p + 1; + pn++; + modules = ldb_realloc_array(ldb, modules, sizeof(char *), pn); + if (!modules) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in register_modules()\n"); + return -1; + } + modules[pn - 1] = q; + } while ((p = strchr(q, LDB_MODULE_SEP))); + } + } + } + + if (!modules) { /* no modules in the options, look for @MODULES in the db */ + int ret, j, k; + char * attrs[] = { "@MODULE" }; + struct ldb_message **msg; + + ret = ldb_search(ldb, "", LDB_SCOPE_BASE, "dn=@MODULES", (const char * const *)attrs, &msg); + if (ret == 0) { + ldb_debug(ldb, LDB_DEBUG_TRACE, "no modules required by the db\n"); + } else { + if (ret < 0) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "ldb error (%s) occurred searching for modules, bailing out\n", ldb_errstring(ldb)); + return -1; + } + if (ret > 1) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Too many records found, bailing out\n"); + return -1; + } + + for (j = 0; j < msg[0]->num_elements; j++) { + for (k = 0; k < msg[0]->elements[j].num_values; k++) { + pn++; + modules = ldb_realloc_array(ldb, modules, sizeof(char *), pn); + if (!modules) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in register_modules()\n"); + return -1; + } + modules[pn - 1] = ldb_strndup(ldb, msg[0]->elements[j].values[k].data, msg[0]->elements[j].values[k].length); + if (!modules[pn - 1]) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Out of Memory in register_modules()\n"); + return -1; + } + } + } + } + ldb_search_free(ldb, msg); + } + + if (modules) { + + for (i = 0; i < pn; i++) { + + if (strcmp(modules[i], "timestamps") == 0) { + current = timestamps_module_init(ldb, options); + if (!current) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "function 'init_module' in %s fails\n", modules[i]); + return -1; + } + DLIST_ADD(ldb->modules, current); + continue; + } +#if 0 + if (strcmp(modules[i], "schema") == 0) { + current = schema_module_init(ldb, options); + if (!current) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "function 'init_module' in %s fails\n", modules[i]); + return -1; + } + DLIST_ADD(ldb->modules, current); + continue; + } +#endif +#ifdef HAVE_DLOPEN_DISABLED + { + void *handle; + init_ldb_module_function init; + struct stat st; + const char *errstr; + + if (stat(modules[i], &st) < 0) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Required module [%s] not found, bailing out!\n", modules[i]); + return -1; + } + + handle = dlopen(modules[i], RTLD_LAZY); + + if (!handle) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Error loading module %s [%s]\n", modules[i], dlerror()); + return -1; + } + + init = (init_ldb_module_function)dlsym(handle, "init_module"); + + errstr = dlerror(); + if (errstr) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "Error trying to resolve symbol 'init_module' in %s [%s]\n", modules[i], errstr); + return -1; + } + + current = init(ldb, options); + if (!current) { + ldb_debug(ldb, LDB_DEBUG_FATAL, "function 'init_module' in %s fails\n", modules[i]); + return -1; + } + DLIST_ADD(ldb->modules, current); + } +#else + ldb_debug(ldb, LDB_DEBUG_FATAL, "Required module [%s] not found, bailing out!\n", modules[i]); + return -1; +#endif + } + } + + return 0; +} + +/* + helper functions to call the next module in chain +*/ +int ldb_next_close(struct ldb_module *module) +{ + if (!module->next) { + return -1; + } + return module->next->ops->close(module->next); +} + +int ldb_next_search(struct ldb_module *module, + const char *base, + enum ldb_scope scope, + const char *expression, + const char * const *attrs, struct ldb_message ***res) +{ + if (!module->next) { + return -1; + } + return module->next->ops->search(module->next, base, scope, expression, attrs, res); +} + +int ldb_next_search_free(struct ldb_module *module, struct ldb_message **msg) +{ + if (!module->next) { + return -1; + } + return module->next->ops->search_free(module->next, msg); +} + +int ldb_next_add_record(struct ldb_module *module, const struct ldb_message *message) +{ + if (!module->next) { + return -1; + } + return module->next->ops->add_record(module->next, message); +} + +int ldb_next_modify_record(struct ldb_module *module, const struct ldb_message *message) +{ + if (!module->next) { + return -1; + } + return module->next->ops->modify_record(module->next, message); +} + +int ldb_next_delete_record(struct ldb_module *module, const char *dn) +{ + if (!module->next) { + return -1; + } + return module->next->ops->delete_record(module->next, dn); +} + +int ldb_next_rename_record(struct ldb_module *module, const char *olddn, const char *newdn) +{ + if (!module->next) { + return -1; + } + return module->next->ops->rename_record(module->next, olddn, newdn); +} + +const char *ldb_next_errstring(struct ldb_module *module) +{ + if (!module->next) { + return NULL; + } + return module->next->ops->errstring(module->next); +} + +void ldb_next_cache_free(struct ldb_module *module) +{ + if (!module->next) { + return; + } + module->next->ops->cache_free(module->next); +} diff --git a/source4/lib/ldb/include/dlinklist.h b/source4/lib/ldb/include/dlinklist.h new file mode 100644 index 0000000000..40f7f0a0c7 --- /dev/null +++ b/source4/lib/ldb/include/dlinklist.h @@ -0,0 +1,93 @@ +/* + Unix SMB/CIFS implementation. + some simple double linked list macros + Copyright (C) Andrew Tridgell 1998 + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. +*/ + +/* To use these macros you must have a structure containing a next and + prev pointer */ + + +/* hook into the front of the list */ +#define DLIST_ADD(list, p) \ +do { \ + if (!(list)) { \ + (list) = (p); \ + (p)->next = (p)->prev = NULL; \ + } else { \ + (list)->prev = (p); \ + (p)->next = (list); \ + (p)->prev = NULL; \ + (list) = (p); \ + }\ +} while (0) + +/* remove an element from a list - element doesn't have to be in list. */ +#define DLIST_REMOVE(list, p) \ +do { \ + if ((p) == (list)) { \ + (list) = (p)->next; \ + if (list) (list)->prev = NULL; \ + } else { \ + if ((p)->prev) (p)->prev->next = (p)->next; \ + if ((p)->next) (p)->next->prev = (p)->prev; \ + } \ + if ((p) && ((p) != (list))) (p)->next = (p)->prev = NULL; \ +} while (0) + +/* promote an element to the top of the list */ +#define DLIST_PROMOTE(list, p) \ +do { \ + DLIST_REMOVE(list, p); \ + DLIST_ADD(list, p); \ +} while (0) + +/* hook into the end of the list - needs a tmp pointer */ +#define DLIST_ADD_END(list, p, type) \ +do { \ + if (!(list)) { \ + (list) = (p); \ + (p)->next = (p)->prev = NULL; \ + } else { \ + type tmp; \ + for (tmp = (list); tmp->next; tmp = tmp->next) ; \ + tmp->next = (p); \ + (p)->next = NULL; \ + (p)->prev = tmp; \ + } \ +} while (0) + +/* demote an element to the end of the list, needs a tmp pointer */ +#define DLIST_DEMOTE(list, p, tmp) \ +do { \ + DLIST_REMOVE(list, p); \ + DLIST_ADD_END(list, p, tmp); \ +} while (0) + +/* concatenate two lists - putting all elements of the 2nd list at the + end of the first list */ +#define DLIST_CONCATENATE(list1, list2, type) \ +do { \ + if (!(list1)) { \ + (list1) = (list2); \ + } else { \ + type tmp; \ + for (tmp = (list1); tmp->next; tmp = tmp->next) ; \ + tmp->next = (list2); \ + (list2)->prev = tmp; \ + } \ +} while (0) diff --git a/source4/lib/ldb/modules/skel.c b/source4/lib/ldb/modules/skel.c new file mode 100644 index 0000000000..1d6314ee74 --- /dev/null +++ b/source4/lib/ldb/modules/skel.c @@ -0,0 +1,123 @@ +/* + ldb database library + + Copyright (C) Simo Sorce 2004 + + ** NOTE! The following LGPL license applies to the ldb + ** library. This does NOT imply that all of Samba is released + ** under the LGPL + + 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. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + 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 +*/ + +/* + * Name: ldb + * + * Component: ldb skel module + * + * Description: example module + * + * Author: Simo Sorce + */ + +#include "includes.h" + +/* close */ +static int skel_close(struct ldb_module *module) +{ + return ldb_next_close(module); +} + +/* search */ +static int skel_search(struct ldb_module *module, const char *base, + enum ldb_scope scope, const char *expression, + const char * const *attrs, struct ldb_message ***res) +{ + return ldb_next_search(module, base, scope, expression, attrs, res); +} + +/* search_free */ +static int skel_search_free(struct ldb_module *module, struct ldb_message **res) +{ + return ldb_next_search_free(module, res); +} + +/* add_record */ +static int skel_add_record(struct ldb_module *module, const struct ldb_message *msg) +{ + return ldb_next_add_record(module, msg); +} + +/* modify_record */ +static int skel_modify_record(struct ldb_module *module, const struct ldb_message *msg) +{ + return ldb_next_modify_record(module, msg); +} + +/* delete_record */ +static int skel_delete_record(struct ldb_module *module, const char *dn) +{ + return ldb_next_delete_record(module, dn); +} + +/* rename_record */ +static int skel_rename_record(struct ldb_module *module, const char *olddn, const char *newdn) +{ + return ldb_next_rename_record(module, olddn, newdn); +} + +/* return extended error information */ +static const char *skel_errstring(struct ldb_module *module) +{ + return ldb_next_errstring(module); +} + +static void skel_cache_free(struct ldb_module *module) +{ + ldb_next_cache_free(module); +} + +static const struct ldb_module_ops skel_ops = { + "skel", + skel_close, + skel_search, + skel_search_free, + skel_add_record, + skel_modify_record, + skel_delete_record, + skel_rename_record, + skel_errstring, + skel_cache_free +}; + +#ifdef HAVE_DLOPEN + struct ldb_module *init_module(struct ldb_context *ldb, const char *options[]) +#else +struct ldb_module *skel_plugin_init(struct ldb_context *ldb, const char *options[]) +#endif +{ + struct ldb_module *ctx; + + ctx = (struct ldb_module *)malloc(sizeof(struct ldb_module)); + if (!ctx) + return NULL; + + ctx->ldb = ldb; + ctx->prev = ctx->next = NULL; + ctx->private_data = NULL; + ctx->ops = &skel_ops; + + return ctx; +} diff --git a/source4/lib/ldb/modules/timestamps.c b/source4/lib/ldb/modules/timestamps.c new file mode 100644 index 0000000000..64d7616fb1 --- /dev/null +++ b/source4/lib/ldb/modules/timestamps.c @@ -0,0 +1,277 @@ +/* + ldb database library + + Copyright (C) Simo Sorce 2004 + + ** NOTE! The following LGPL license applies to the ldb + ** library. This does NOT imply that all of Samba is released + ** under the LGPL + + 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. + + This library is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + 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 +*/ + +/* + * Name: ldb + * + * Component: ldb timestamps module + * + * Description: add object timestamping functionality + * + * Author: Simo Sorce + */ + +#include "includes.h" +#include <time.h> + +static int timestamps_close(struct ldb_module *module) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_close\n"); + return ldb_next_close(module); +} + +static int timestamps_search(struct ldb_module *module, const char *base, + enum ldb_scope scope, const char *expression, + const char * const *attrs, struct ldb_message ***res) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_search\n"); + return ldb_next_search(module, base, scope, expression, attrs, res); +} + +static int timestamps_search_free(struct ldb_module *module, struct ldb_message **res) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_search_free\n"); + return ldb_next_search_free(module, res); +} + +static int add_time_element(struct ldb_context *ldb, struct ldb_message *msg, const char *attr_name, const char *time_string, unsigned int flags) +{ + struct ldb_val *values; + char *name, *timestr; + int i; + + for (i = 0; i < msg->num_elements; i++) { + if (strcasecmp(msg->elements[i].name, attr_name) == 0) { + return 0; + } + } + + msg->elements = ldb_realloc_array(ldb, msg->elements, sizeof(struct ldb_message_element), msg->num_elements + 1); + name = ldb_strdup(ldb, attr_name); + timestr = ldb_strdup(ldb, time_string); + values = ldb_malloc(ldb, sizeof(struct ldb_val)); + if (!msg->elements || !name || !timestr || !values) { + return -1; + } + + msg->elements[msg->num_elements].name = name; + msg->elements[msg->num_elements].flags = flags; + msg->elements[msg->num_elements].num_values = 1; + msg->elements[msg->num_elements].values = values; + msg->elements[msg->num_elements].values[0].data = timestr; + msg->elements[msg->num_elements].values[0].length = strlen(timestr); + + msg->num_elements += 1; + + return 0; +} + +static void free_elements(struct ldb_context *ldb, struct ldb_message *msg, int real_el_num) +{ + int i; + + for (i = real_el_num; i < msg->num_elements; i++) { + ldb_free(ldb, msg->elements[i].name); + ldb_free(ldb, msg->elements[i].values[0].data); + ldb_free(ldb, msg->elements[i].values); + } + ldb_free(ldb, msg->elements); + ldb_free(ldb, msg); +} + +/* add_record: add crateTimestamp/modifyTimestamp attributes */ +static int timestamps_add_record(struct ldb_module *module, const struct ldb_message *msg) +{ + struct ldb_message *msg2 = NULL; + struct tm *tm; + char *timestr; + time_t timeval; + int ret, i; + + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_add_record\n"); + + if (msg->dn[0] != '@') { /* do not manipulate our control entries */ + timeval = time(NULL); + tm = gmtime(&timeval); + if (!tm) { + return -1; + } + + /* formatted like: 20040408072012.0Z */ + ldb_asprintf(module->ldb, ×tr, + "%04u%02u%02u%02u%02u%02u.0Z", + tm->tm_year+1900, tm->tm_mon+1, + tm->tm_mday, tm->tm_hour, tm->tm_min, + tm->tm_sec); + + if (!timestr) { + return -1; + } + + msg2 = ldb_malloc_p(module->ldb, struct ldb_message); + if (!msg2) { + return -1; + } + msg2->dn = msg->dn; + msg2->num_elements = msg->num_elements; + msg2->private_data = msg->private_data; + msg2->elements = ldb_malloc_array_p(module->ldb, struct ldb_message_element, msg2->num_elements); + for (i = 0; i < msg2->num_elements; i++) { + msg2->elements[i] = msg->elements[i]; + } + + add_time_element(module->ldb, msg2, "createTimestamp", timestr, LDB_FLAG_MOD_ADD); + add_time_element(module->ldb, msg2, "modifyTimestamp", timestr, LDB_FLAG_MOD_ADD); + add_time_element(module->ldb, msg2, "whenCreated", timestr, LDB_FLAG_MOD_ADD); + add_time_element(module->ldb, msg2, "whenChanged", timestr, LDB_FLAG_MOD_ADD); + + ldb_free(module->ldb, timestr); + } + + if (msg2) { + ret = ldb_next_add_record(module, msg2); + free_elements(module->ldb, msg2, msg->num_elements); + } else { + ret = ldb_next_add_record(module, msg); + } + + return ret; +} + +/* modify_record: change modifyTimestamp as well */ +static int timestamps_modify_record(struct ldb_module *module, const struct ldb_message *msg) +{ + struct ldb_message *msg2 = NULL; + struct tm *tm; + char *timestr; + time_t timeval; + int ret, i; + + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_modify_record\n"); + + if (msg->dn[0] != '@') { /* do not manipulate our control entries */ + timeval = time(NULL); + tm = gmtime(&timeval); + if (!tm) { + return -1; + } + + /* formatted like: 20040408072012.0Z */ + ldb_asprintf(module->ldb, ×tr, + "%04u%02u%02u%02u%02u%02u.0Z", + tm->tm_year+1900, tm->tm_mon+1, + tm->tm_mday, tm->tm_hour, tm->tm_min, + tm->tm_sec); + + if (!timestr) { + return -1; + } + + msg2 = ldb_malloc_p(module->ldb, struct ldb_message); + if (!msg2) { + return -1; + } + msg2->dn = msg->dn; + msg2->num_elements = msg->num_elements; + msg2->private_data = msg->private_data; + msg2->elements = ldb_malloc_array_p(module->ldb, struct ldb_message_element, msg2->num_elements); + for (i = 0; i < msg2->num_elements; i++) { + msg2->elements[i] = msg->elements[i]; + } + + add_time_element(module->ldb, msg2, "modifyTimestamp", timestr, LDB_FLAG_MOD_REPLACE); + add_time_element(module->ldb, msg2, "whenChanged", timestr, LDB_FLAG_MOD_REPLACE); + + ldb_free(module->ldb, timestr); + } + + if (msg2) { + ret = ldb_next_modify_record(module, msg2); + free_elements(module->ldb, msg2, msg->num_elements); + } else { + ret = ldb_next_modify_record(module, msg); + } + + return ret; +} + +static int timestamps_delete_record(struct ldb_module *module, const char *dn) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_delete_record\n"); + return ldb_next_delete_record(module, dn); +} + +static int timestamps_rename_record(struct ldb_module *module, const char *olddn, const char *newdn) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_rename_record\n"); + return ldb_next_rename_record(module, olddn, newdn); +} + +/* return extended error information */ +static const char *timestamps_errstring(struct ldb_module *module) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_errstring\n"); + return ldb_next_errstring(module); +} + +static void timestamps_cache_free(struct ldb_module *module) +{ + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "timestamps_cache_free\n"); + ldb_next_cache_free(module); +} + +static const struct ldb_module_ops timestamps_ops = { + "timestamps", + timestamps_close, + timestamps_search, + timestamps_search_free, + timestamps_add_record, + timestamps_modify_record, + timestamps_delete_record, + timestamps_rename_record, + timestamps_errstring, + timestamps_cache_free +}; + + +/* the init function */ +#ifdef HAVE_DLOPEN_DISABLED + struct ldb_module *init_module(struct ldb_context *ldb, const char *options[]) +#else +struct ldb_module *timestamps_module_init(struct ldb_context *ldb, const char *options[]) +#endif +{ + struct ldb_module *ctx; + + ctx = (struct ldb_module *)malloc(sizeof(struct ldb_module)); + if (!ctx) + return NULL; + + ctx->ldb = ldb; + ctx->prev = ctx->next = NULL; + ctx->private_data = NULL; + ctx->ops = ×tamps_ops; + + return ctx; +} |