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 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/db_wrap.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                    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                 expr = talloc_vasprintf(mem_ctx, 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->msgs);
  59
  60                 DEBUG(6,("gendb_search_v: %s %s -> %d\n",
  61                          basedn?ldb_dn_get_linearized(basedn):"NULL",
  62                          expr?expr:"NULL", res->count));
  63
  64                 ret = res->count;
  65                 *msgs = res->msgs;
  66                 talloc_free(res);
  67         } else {
  68                 DEBUG(4,("gendb_search_v: search failed: %s", ldb_errstring(ldb)));
  69                 ret = -1;
  70         }
  71
  72         talloc_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                  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 /*
  98   search the LDB for a specified record (by DN)
  99 */
 100
 101 int gendb_search_dn(struct ldb_context *ldb,
 102                  TALLOC_CTX *mem_ctx,
 103                  struct ldb_dn *dn,
 104                  struct ldb_message ***res,
 105                  const char * const *attrs)
 106 {
 107         return gendb_search(ldb, mem_ctx, dn, res, attrs, NULL);
 108 }
 109
 110 /*
 111   setup some initial ldif in a ldb
 112 */
 113 int gendb_add_ldif(struct ldb_context *ldb, const char *ldif_string)
 114 {
 115         struct ldb_ldif *ldif;
 116         int ret;
 117         ldif = ldb_ldif_read_string(ldb, &ldif_string);
 118         if (ldif == NULL) return -1;
 119         ret = ldb_add(ldb, ldif->msg);
 120         talloc_free(ldif);
 121         return ret;
 122 }