source/lib/gendb.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 2 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, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 #include "includes.h"
  25 #include "lib/ldb/include/ldb.h"
  26
  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                    const char *basedn,
  33                    struct ldb_message ***res,
  34                    const char * const *attrs,
  35                    const char *format,
  36                    va_list ap)  _PRINTF_ATTRIBUTE(6,0)
  37 {
  38         char *expr = NULL;
  39         int count;
  40
  41         vasprintf(&expr, format, ap);
  42         if (expr == NULL) {
  43                 return -1;
  44         }
  45
  46         *res = NULL;
  47
  48         count = ldb_search(ldb, basedn, LDB_SCOPE_SUBTREE, expr, attrs, res);
  49
  50         if (*res) talloc_steal(mem_ctx, *res);
  51
  52         DEBUG(4,("gendb_search_v: %s %s -> %d  (%s)\n",
  53                  basedn?basedn:"NULL", expr, count,
  54                  count==-1?ldb_errstring(ldb):"OK"));
  55
  56         free(expr);
  57
  58         return count;
  59 }
  60
  61 /*
  62   search the LDB for the specified attributes - varargs variant
  63 */
  64 int gendb_search(struct ldb_context *ldb,
  65                  TALLOC_CTX *mem_ctx,
  66                  const char *basedn,
  67                  struct ldb_message ***res,
  68                  const char * const *attrs,
  69                  const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
  70 {
  71         va_list ap;
  72         int count;
  73
  74         va_start(ap, format);
  75         count = gendb_search_v(ldb, mem_ctx, basedn, res, attrs, format, ap);
  76         va_end(ap);
  77
  78         return count;
  79 }
  80
  81 int gendb_search_dn(struct ldb_context *ldb,
  82                     TALLOC_CTX *mem_ctx,
  83                     const char *dn,
  84                     struct ldb_message ***res,
  85                     const char * const *attrs)
  86 {
  87         va_list ap;
  88         int count;
  89
  90         *res = NULL;
  91
  92         count = ldb_search(ldb, dn, LDB_SCOPE_BASE, "", attrs, res);
  93
  94         if (count > 1) {
  95                 DEBUG(1, ("DB Corruption ? - Found more then one entry for dn: %s", dn));
  96                 return -1;
  97         }
  98
  99         if (*res) talloc_steal(mem_ctx, *res);
 100
 101         DEBUG(4,("gendb_search_dn: %s -> %d (%s)\n",
 102                  dn, count, count==-1?ldb_errstring(ldb):"OK"));
 103
 104         return count;
 105 }
 106
 107
 108 /*
 109   setup some initial ldif in a ldb
 110 */
 111 int gendb_add_ldif(struct ldb_context *ldb, const char *ldif_string)
 112 {
 113         struct ldb_ldif *ldif;
 114         int ret;
 115         ldif = ldb_ldif_read_string(ldb, ldif_string);
 116         if (ldif == NULL) return -1;
 117         ret = ldb_add(ldb, ldif->msg);
 118         talloc_free(ldif);
 119         return ret;
 120 }