source4/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 #include "lib/ldb/include/ldb_errors.h"
  27
  28 /*
  29   search the sam for the specified attributes - va_list variant
  30 */
  31 int gendb_search_v(struct ldb_context *ldb,
  32                    TALLOC_CTX *mem_ctx,
  33                    const struct ldb_dn *basedn,
  34                    struct ldb_message ***msgs,
  35                    const char * const *attrs,
  36                    const char *format,
  37                    va_list ap)  _PRINTF_ATTRIBUTE(6,0)
  38 {
  39         enum ldb_scope scope = LDB_SCOPE_SUBTREE;
  40         struct ldb_result *res;
  41         char *expr = NULL;
  42         int ret;
  43
  44         if (format) {
  45                 vasprintf(&expr, format, ap);
  46                 if (expr == NULL) {
  47                         return -1;
  48                 }
  49         } else {
  50                 scope = LDB_SCOPE_BASE;
  51         }
  52
  53         res = NULL;
  54
  55         ret = ldb_search(ldb, basedn, scope, expr, attrs, &res);
  56
  57         if (ret == LDB_SUCCESS) {
  58                 talloc_steal(mem_ctx, res);
  59
  60                 DEBUG(4,("gendb_search_v: %s %s -> %d\n",
  61                          basedn?ldb_dn_linearize(mem_ctx,basedn):"NULL",
  62                          expr?expr:"NULL", res->count));
  63
  64                 ret = res->count;
  65                 *msgs = res->msgs;
  66
  67         } else {
  68                 DEBUG(4,("gendb_search_v: search failed: %s", ldb_errstring(ldb)));
  69                 ret = -1;
  70         }
  71
  72         free(expr);
  73
  74         return ret;
  75 }
  76
  77 /*
  78   search the LDB for the specified attributes - varargs variant
  79 */
  80 int gendb_search(struct ldb_context *ldb,
  81                  TALLOC_CTX *mem_ctx,
  82                  const struct ldb_dn *basedn,
  83                  struct ldb_message ***res,
  84                  const char * const *attrs,
  85                  const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
  86 {
  87         va_list ap;
  88         int count;
  89
  90         va_start(ap, format);
  91         count = gendb_search_v(ldb, mem_ctx, basedn, res, attrs, format, ap);
  92         va_end(ap);
  93
  94         return count;
  95 }
  96
  97 int gendb_search_dn(struct ldb_context *ldb,
  98                  TALLOC_CTX *mem_ctx,
  99                  const struct ldb_dn *dn,
 100                  struct ldb_message ***res,
 101                  const char * const *attrs)
 102 {
 103         return gendb_search(ldb, mem_ctx, dn, res, attrs, NULL);
 104 }
 105
 106 /*
 107   setup some initial ldif in a ldb
 108 */
 109 int gendb_add_ldif(struct ldb_context *ldb, const char *ldif_string)
 110 {
 111         struct ldb_ldif *ldif;
 112         int ret;
 113         ldif = ldb_ldif_read_string(ldb, &ldif_string);
 114         if (ldif == NULL) return -1;
 115         ret = ldb_add(ldb, ldif->msg);
 116         talloc_free(ldif);
 117         return ret;
 118 }