Fix the mess with ldb includes.
[kamenim/samba.git] / source4 / lib / ldb / tools / ldbsearch.c
index 45826f09962d5c6f427073cb8c5f4e5cba22d586..35d4ac70021bea0aea60dbfd05a869ba04b0f41e 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/>.
 */
 
 /*
@@ -32,9 +31,8 @@
  *  Author: Andrew Tridgell
  */
 
-#include "includes.h"
-#include "ldb/include/includes.h"
-#include "ldb/tools/cmdline.h"
+#include "ldb.h"
+#include "tools/cmdline.h"
 
 static void usage(void)
 {
@@ -54,16 +52,17 @@ static int do_compare_msg(struct ldb_message **el1,
                          struct ldb_message **el2,
                          void *opaque)
 {
-       struct ldb_context *ldb = talloc_get_type(opaque, struct ldb_context);
-       return ldb_dn_compare(ldb, (*el1)->dn, (*el2)->dn);
+       return ldb_dn_compare((*el1)->dn, (*el2)->dn);
 }
 
 struct search_context {
+       struct ldb_context *ldb;
        struct ldb_control **req_ctrls;
 
        int sort;
        int num_stored;
        struct ldb_message **store;
+       int refs_stored;
        char **refs_store;
 
        int entries;
@@ -81,12 +80,7 @@ static int store_message(struct ldb_message *msg, struct search_context *sctx) {
                return -1;
        }
 
-       sctx->store[sctx->num_stored] = talloc_steal(sctx->store, msg);
-       if (!sctx->store[sctx->num_stored]) {
-               fprintf(stderr, "talloc_steal failed while storing messages\n");
-               return -1;
-       }
-
+       sctx->store[sctx->num_stored] = talloc_move(sctx->store, &msg);
        sctx->num_stored++;
        sctx->store[sctx->num_stored] = NULL;
 
@@ -95,25 +89,20 @@ static int store_message(struct ldb_message *msg, struct search_context *sctx) {
 
 static int store_referral(char *referral, struct search_context *sctx) {
 
-       sctx->refs_store = talloc_realloc(sctx, sctx->refs_store, char *, sctx->refs + 2);
+       sctx->refs_store = talloc_realloc(sctx, sctx->refs_store, char *, sctx->refs_stored + 2);
        if (!sctx->refs_store) {
                fprintf(stderr, "talloc_realloc failed while storing referrals\n");
                return -1;
        }
 
-       sctx->refs_store[sctx->refs] = talloc_steal(sctx->refs_store, referral);
-       if (!sctx->refs_store[sctx->refs]) {
-               fprintf(stderr, "talloc_steal failed while storing referrals\n");
-               return -1;
-       }
-
-       sctx->refs++;
-       sctx->refs_store[sctx->refs] = NULL;
+       sctx->refs_store[sctx->refs_stored] = talloc_move(sctx->refs_store, &referral);
+       sctx->refs_stored++;
+       sctx->refs_store[sctx->refs_stored] = NULL;
 
        return 0;
 }
 
-static int display_message(struct ldb_context *ldb, struct ldb_message *msg, struct search_context *sctx) {
+static int display_message(struct ldb_message *msg, struct search_context *sctx) {
        struct ldb_ldif ldif;
 
        sctx->entries++;
@@ -131,12 +120,13 @@ static int display_message(struct ldb_context *ldb, struct ldb_message *msg, str
                ldb_msg_sort_elements(ldif.msg);
                }
 
-       ldb_ldif_write_file(ldb, stdout, &ldif);
+       ldb_ldif_write_file(sctx->ldb, stdout, &ldif);
 
        return 0;
 }
 
-static int display_referral(char *referral, struct search_context *sctx) {
+static int display_referral(char *referral, struct search_context *sctx)
+{
 
        sctx->refs++;
        printf("# Referral\nref: %s\n\n", referral);
@@ -144,18 +134,26 @@ static int display_referral(char *referral, struct search_context *sctx) {
        return 0;
 }
 
-static int search_callback(struct ldb_context *ldb, void *context, struct ldb_reply *ares)
+static int search_callback(struct ldb_request *req, struct ldb_reply *ares)
 {
-       struct search_context *sctx = talloc_get_type(context, struct search_context);
+       struct search_context *sctx;
        int ret;
+
+       sctx = talloc_get_type(req->context, struct search_context);
+
+       if (!ares) {
+               return ldb_request_done(req, LDB_ERR_OPERATIONS_ERROR);
+       }
+       if (ares->error != LDB_SUCCESS) {
+               return ldb_request_done(req, ares->error);
+       }
        
        switch (ares->type) {
-
        case LDB_REPLY_ENTRY:
                if (sctx->sort) {
                        ret = store_message(ares->message, sctx);
                } else {
-                       ret = display_message(ldb, ares->message, sctx);
+                       ret = display_message(ares->message, sctx);
                }
                break;
 
@@ -165,6 +163,9 @@ static int search_callback(struct ldb_context *ldb, void *context, struct ldb_re
                } else {
                        ret = display_referral(ares->referral, sctx);
                }
+               if (ret) {
+                       return ldb_request_done(req, LDB_ERR_OPERATIONS_ERROR);
+               }
                break;
 
        case LDB_REPLY_DONE:
@@ -172,29 +173,20 @@ static int search_callback(struct ldb_context *ldb, void *context, struct ldb_re
                        if (handle_controls_reply(ares->controls, sctx->req_ctrls) == 1)
                                sctx->pending = 1;
                }
-               ret = 0;
-               break;
-               
-       default:
-               fprintf(stderr, "unknown Reply Type\n");
-               return LDB_ERR_OTHER;
-       }
-
-       if (talloc_free(ares) == -1) {
-               fprintf(stderr, "talloc_free failed\n");
-               sctx->pending = 0;
-               return LDB_ERR_OPERATIONS_ERROR;
+               talloc_free(ares);
+               return ldb_request_done(req, LDB_SUCCESS);
        }
 
+       talloc_free(ares);
        if (ret) {
-               return LDB_ERR_OPERATIONS_ERROR;
+               return ldb_request_done(req, LDB_ERR_OPERATIONS_ERROR);
        }
 
        return LDB_SUCCESS;
 }
 
 static int do_search(struct ldb_context *ldb,
-                    const struct ldb_dn *basedn,
+                    struct ldb_dn *basedn,
                     struct ldb_cmdline *options,
                     const char *expression,
                     const char * const *attrs)
@@ -203,32 +195,44 @@ static int do_search(struct ldb_context *ldb,
        struct search_context *sctx;
        int ret;
 
-       req = talloc(ldb, struct ldb_request);
-       if (!req) return -1;
+       req = NULL;
        
-       sctx = talloc(req, struct search_context);
+       sctx = talloc(ldb, struct search_context);
        if (!sctx) return -1;
 
+       sctx->ldb = ldb;
        sctx->sort = options->sorted;
        sctx->num_stored = 0;
+       sctx->refs_stored = 0;
        sctx->store = NULL;
-       sctx->req_ctrls = parse_controls(ldb, options->controls);
-       if (options->controls != NULL &&  sctx->req_ctrls== NULL) return -1;
+       sctx->req_ctrls = ldb_parse_control_strings(ldb, sctx, (const char **)options->controls);
+       if (options->controls != NULL &&  sctx->req_ctrls== NULL) {
+               printf("parsing controls failed: %s\n", ldb_errstring(ldb));
+               return -1;
+       }
        sctx->entries = 0;
        sctx->refs = 0;
 
-       req->operation = LDB_SEARCH;
-       req->op.search.base = basedn;
-       req->op.search.scope = options->scope;
-       req->op.search.tree = ldb_parse_tree(ldb, expression);
-       if (req->op.search.tree == NULL) return -1;
-       req->op.search.attrs = attrs;
-       req->controls = sctx->req_ctrls;
-       req->async.context = sctx;
-       req->async.callback = &search_callback;
-       ldb_set_timeout(ldb, req, 0); /* TODO: make this settable by command line */
+       if (basedn == NULL) {
+               basedn = ldb_get_default_basedn(ldb);
+       }
 
 again:
+       /* free any previous requests */
+       if (req) talloc_free(req);
+
+       ret = ldb_build_search_req(&req, ldb, ldb,
+                                  basedn, options->scope,
+                                  expression, attrs,
+                                  sctx->req_ctrls,
+                                  sctx, search_callback,
+                                  NULL);
+       if (ret != LDB_SUCCESS) {
+               talloc_free(sctx);
+               printf("allocating request failed: %s\n", ldb_errstring(ldb));
+               return -1;
+       }
+
        sctx->pending = 0;
 
        ret = ldb_request(ldb, req);
@@ -237,7 +241,7 @@ again:
                return -1;
        }
 
-       ret = ldb_wait(req->async.handle, LDB_WAIT_ALL);
+       ret = ldb_wait(req->handle, LDB_WAIT_ALL);
                if (ret != LDB_SUCCESS) {
                printf("search error - %s\n", ldb_errstring(ldb));
                return -1;
@@ -246,22 +250,18 @@ again:
        if (sctx->pending)
                goto again;
 
-       if (sctx->sort && sctx->num_stored != 0) {
+       if (sctx->sort && (sctx->num_stored != 0 || sctx->refs != 0)) {
                int i;
 
-               ldb_qsort(sctx->store, ret, sizeof(struct ldb_message *),
-                         ldb, (ldb_qsort_cmp_fn_t)do_compare_msg);
-
-               if (ret != 0) {
-                       fprintf(stderr, "An error occurred while sorting messages\n");
-                       exit(1);
+               if (sctx->num_stored) {
+                       ldb_qsort(sctx->store, sctx->num_stored, sizeof(struct ldb_message *),
+                                 ldb, (ldb_qsort_cmp_fn_t)do_compare_msg);
                }
-
                for (i = 0; i < sctx->num_stored; i++) {
-                       display_message(ldb, sctx->store[i], sctx);
+                       display_message(sctx->store[i], sctx);
                }
 
-               for (i = 0; i < sctx->refs; i++) {
+               for (i = 0; i < sctx->refs_stored; i++) {
                        display_referral(sctx->refs_store[i], sctx);
                }
        }
@@ -269,6 +269,7 @@ again:
        printf("# returned %d records\n# %d entries\n# %d referrals\n",
                sctx->entries + sctx->refs, sctx->entries, sctx->refs);
 
+       talloc_free(sctx);
        talloc_free(req);
 
        return 0;
@@ -283,9 +284,10 @@ int main(int argc, const char **argv)
        int ret = -1;
        const char *expression = "(|(objectClass=*)(distinguishedName=*))";
 
-       ldb_global_init();
-
-       ldb = ldb_init(NULL);
+       ldb = ldb_init(NULL, NULL);
+       if (ldb == NULL) {
+               return -1;
+       }
 
        options = ldb_cmdline_process(ldb, argc, argv, usage);
 
@@ -303,8 +305,8 @@ int main(int argc, const char **argv)
        }
 
        if (options->basedn != NULL) {
-               basedn = ldb_dn_explode(ldb, options->basedn);
-               if (basedn == NULL) {
+               basedn = ldb_dn_new(ldb, ldb, options->basedn);
+               if ( ! ldb_dn_validate(basedn)) {
                        fprintf(stderr, "Invalid Base DN format\n");
                        exit(1);
                }