source/lib/util/util_ldb.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    common share info functions
   5
   6    Copyright (C) Andrew Tridgell 2004
   7    Copyright (C) Tim Potter 2004
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 3 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21 */
  22
  23 #include "includes.h"
  24 #include "lib/ldb/include/ldb.h"
  25 #include "lib/ldb/include/ldb_errors.h"
  26 #include "lib/util/util_ldb.h"
  27 /*
  28   search the sam for the specified attributes - va_list variant
  29 */
  30 int gendb_search_v(struct ldb_context *ldb,
  31                    TALLOC_CTX *mem_ctx,
  32                    struct ldb_dn *basedn,
  33                    struct ldb_message ***msgs,
  34                    const char * const *attrs,
  35                    const char *format,
  36                    va_list ap)  _PRINTF_ATTRIBUTE(6,0)
  37 {
  38         enum ldb_scope scope = LDB_SCOPE_SUBTREE;
  39         struct ldb_result *res;
  40         char *expr = NULL;
  41         int ret;
  42
  43         if (format) {
  44                 expr = talloc_vasprintf(mem_ctx, format, ap);
  45                 if (expr == NULL) {
  46                         return -1;
  47                 }
  48         } else {
  49                 scope = LDB_SCOPE_BASE;
  50         }
  51
  52         res = NULL;
  53
  54         ret = ldb_search(ldb, basedn, scope, expr, attrs, &res);
  55
  56         if (ret == LDB_SUCCESS) {
  57                 talloc_steal(mem_ctx, res->msgs);
  58
  59                 DEBUG(6,("gendb_search_v: %s %s -> %d\n",
  60                          basedn?ldb_dn_get_linearized(basedn):"NULL",
  61                          expr?expr:"NULL", res->count));
  62
  63                 ret = res->count;
  64                 *msgs = res->msgs;
  65                 talloc_free(res);
  66         } else if (scope == LDB_SCOPE_BASE && ret == LDB_ERR_NO_SUCH_OBJECT) {
  67                 ret = 0;
  68                 *msgs = NULL;
  69         } else {
  70                 DEBUG(4,("gendb_search_v: search failed: %s", ldb_errstring(ldb)));
  71                 ret = -1;
  72         }
  73
  74         talloc_free(expr);
  75
  76         return ret;
  77 }
  78
  79 /*
  80   search the LDB for the specified attributes - varargs variant
  81 */
  82 int gendb_search(struct ldb_context *ldb,
  83                  TALLOC_CTX *mem_ctx,
  84                  struct ldb_dn *basedn,
  85                  struct ldb_message ***res,
  86                  const char * const *attrs,
  87                  const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
  88 {
  89         va_list ap;
  90         int count;
  91
  92         va_start(ap, format);
  93         count = gendb_search_v(ldb, mem_ctx, basedn, res, attrs, format, ap);
  94         va_end(ap);
  95
  96         return count;
  97 }
  98
  99 /*
 100   search the LDB for a specified record (by DN)
 101 */
 102
 103 int gendb_search_dn(struct ldb_context *ldb,
 104                  TALLOC_CTX *mem_ctx,
 105                  struct ldb_dn *dn,
 106                  struct ldb_message ***res,
 107                  const char * const *attrs)
 108 {
 109         return gendb_search(ldb, mem_ctx, dn, res, attrs, NULL);
 110 }
 111
 112 /*
 113   setup some initial ldif in a ldb
 114 */
 115 int gendb_add_ldif(struct ldb_context *ldb, const char *ldif_string)
 116 {
 117         struct ldb_ldif *ldif;
 118         int ret;
 119         ldif = ldb_ldif_read_string(ldb, &ldif_string);
 120         if (ldif == NULL) return -1;
 121         ret = ldb_add(ldb, ldif->msg);
 122         talloc_free(ldif);
 123         return ret;
 124 }
 125
 126 char *wrap_casefold(void *context, void *mem_ctx, const char *s)
 127 {
 128         return strupper_talloc(mem_ctx, s);
 129 }
 130
 131