summaryrefslogtreecommitdiff
path: root/source4/dsdb/samdb/ldb_modules/rootdse.c
diff options
context:
space:
mode:
Diffstat (limited to 'source4/dsdb/samdb/ldb_modules/rootdse.c')
-rw-r--r--source4/dsdb/samdb/ldb_modules/rootdse.c149
1 files changed, 149 insertions, 0 deletions
diff --git a/source4/dsdb/samdb/ldb_modules/rootdse.c b/source4/dsdb/samdb/ldb_modules/rootdse.c
new file mode 100644
index 0000000000..b0a21efea4
--- /dev/null
+++ b/source4/dsdb/samdb/ldb_modules/rootdse.c
@@ -0,0 +1,149 @@
+/*
+ Unix SMB/CIFS implementation.
+
+ rootDSE ldb module
+
+ Copyright (C) Andrew Tridgell 2005
+
+ 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.
+*/
+
+#include "includes.h"
+#include "lib/ldb/include/ldb.h"
+#include "lib/ldb/include/ldb_errors.h"
+#include "lib/ldb/include/ldb_private.h"
+#include <time.h>
+
+/*
+ return 1 if a specific attribute has been requested
+*/
+static int do_attribute(const char * const *attrs, const char *name)
+{
+ return attrs == NULL ||
+ ldb_attr_in_list(attrs, name) ||
+ ldb_attr_in_list(attrs, "*");
+}
+
+/*
+ add dynamically generated attributes to rootDSE result
+*/
+static int rootdse_add_dynamic(struct ldb_module *module, struct ldb_request *req)
+{
+ struct ldb_search *s = &req->op.search;
+ struct ldb_message *msg;
+
+ /* this is gross, and will be removed when I change ldb_result not
+ to be so pointer crazy :-) */
+ if (s->res[0][0].msgs == NULL) {
+ return LDB_SUCCESS;
+ }
+
+ msg = s->res[0][0].msgs[0];
+
+ msg->dn = ldb_dn_explode(msg, "");
+
+ if (do_attribute(s->attrs, "currentTime")) {
+ if (ldb_msg_add_string(msg, "currentTime",
+ ldb_timestring(msg, time(NULL))) != 0) {
+ goto failed;
+ }
+ }
+
+ /* TODO: lots more dynamic attributes should be added here */
+
+ return 0;
+
+failed:
+ return LDB_ERR_OPERATIONS_ERROR;
+}
+
+/*
+ handle search requests
+*/
+static int rootdse_search_bytree(struct ldb_module *module, struct ldb_request *req)
+{
+// struct ldb_request r = *req;
+ struct ldb_search *s = &req->op.search;
+ int ret;
+ TALLOC_CTX *tmp_ctx;
+
+ /* see if its for the rootDSE */
+ if (s->scope != LDB_SCOPE_BASE ||
+ (s->base && s->base->comp_num != 0)) {
+ return ldb_next_request(module, req);
+ }
+
+ tmp_ctx = talloc_new(module);
+
+ /* in our db we store the rootDSE with a DN of cn=rootDSE */
+ s->base = ldb_dn_explode(tmp_ctx, "cn=rootDSE");
+ s->tree = ldb_parse_tree(tmp_ctx, "dn=*");
+ if (s->base == NULL || s->tree == NULL) {
+ ldb_oom(module->ldb);
+ talloc_free(tmp_ctx);
+ return LDB_ERR_OPERATIONS_ERROR;
+ }
+
+ /* grab the static contents of the record */
+ ret = ldb_next_request(module, req);
+
+ req->op.search.res = s->res;
+
+ if (ret == LDB_SUCCESS) {
+ ret = rootdse_add_dynamic(module, req);
+ }
+
+ talloc_free(tmp_ctx);
+
+ return ret;
+}
+
+
+static int rootdse_request(struct ldb_module *module, struct ldb_request *req)
+{
+ switch (req->operation) {
+ case LDB_REQ_SEARCH:
+ return rootdse_search_bytree(module, req);
+ default:
+ break;
+ }
+ return ldb_next_request(module, req);
+}
+
+static const struct ldb_module_ops rootdse_ops = {
+ .name = "rootdse",
+ .request = rootdse_request
+};
+
+#ifdef HAVE_DLOPEN_DISABLED
+struct ldb_module *init_module(struct ldb_context *ldb, const char *options[])
+#else
+struct ldb_module *rootdse_module_init(struct ldb_context *ldb, const char *options[])
+#endif
+{
+ struct ldb_module *ctx;
+
+ ctx = talloc(ldb, struct ldb_module);
+ if (!ctx)
+ return NULL;
+
+ ctx->ldb = ldb;
+ ctx->prev = ctx->next = NULL;
+ ctx->ops = &rootdse_ops;
+ ctx->private_data = NULL;
+
+ return ctx;
+}
+