source3/lib/dbwrap/dbwrap.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Database interface wrapper
   4    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2006
   5
   6    Major code contributions from Aleksey Fedoseev (fedoseev@ru.ibm.com)
   7
   8    This program is free software; you can redistribute it and/or modify
   9    it under the terms of the GNU General Public License as published by
  10    the Free Software Foundation; either version 3 of the License, or
  11    (at your option) any later version.
  12
  13    This program is distributed in the hope that it will be useful,
  14    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16    GNU General Public License for more details.
  17
  18    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "dbwrap/dbwrap.h"
  24 #include "dbwrap/dbwrap_private.h"
  25 #include "util_tdb.h"
  26
  27 /*
  28  * Fall back using fetch_locked if no genuine fetch operation is provided
  29  */
  30
  31 int dbwrap_fallback_fetch(struct db_context *db, TALLOC_CTX *mem_ctx,
  32                           TDB_DATA key, TDB_DATA *data)
  33 {
  34         struct db_record *rec;
  35
  36         if (!(rec = db->fetch_locked(db, mem_ctx, key))) {
  37                 return -1;
  38         }
  39
  40         data->dsize = rec->value.dsize;
  41         data->dptr = talloc_move(mem_ctx, &rec->value.dptr);
  42         TALLOC_FREE(rec);
  43         return 0;
  44 }
  45
  46 /*
  47  * Fall back using fetch if no genuine parse operation is provided
  48  */
  49
  50 int dbwrap_fallback_parse_record(struct db_context *db, TDB_DATA key,
  51                                  int (*parser)(TDB_DATA key,
  52                                                TDB_DATA data,
  53                                                void *private_data),
  54                                  void *private_data)
  55 {
  56         TDB_DATA data;
  57         int res;
  58
  59         res = db->fetch(db, talloc_tos(), key, &data);
  60         if (res != 0) {
  61                 return res;
  62         }
  63
  64         res = parser(key, data, private_data);
  65         TALLOC_FREE(data.dptr);
  66         return res;
  67 }
  68
  69
  70 TDB_DATA dbwrap_fetch(struct db_context *db, TALLOC_CTX *mem_ctx,
  71                       TDB_DATA key)
  72 {
  73         TDB_DATA result;
  74
  75         if (db->fetch(db, mem_ctx, key, &result) != 0) {
  76                 return make_tdb_data(NULL, 0);
  77         }
  78
  79         return result;
  80 }
  81
  82 NTSTATUS dbwrap_store(struct db_context *db, TDB_DATA key,
  83                       TDB_DATA data, int flags)
  84 {
  85         struct db_record *rec;
  86         NTSTATUS status;
  87
  88         rec = db->fetch_locked(db, talloc_tos(), key);
  89         if (rec == NULL) {
  90                 return NT_STATUS_NO_MEMORY;
  91         }
  92
  93         status = rec->store(rec, data, flags);
  94         TALLOC_FREE(rec);
  95         return status;
  96 }
  97
  98 NTSTATUS dbwrap_delete(struct db_context *db, TDB_DATA key)
  99 {
 100         struct db_record *rec;
 101         NTSTATUS status;
 102
 103         rec = db->fetch_locked(db, talloc_tos(), key);
 104         if (rec == NULL) {
 105                 return NT_STATUS_NO_MEMORY;
 106         }
 107         status = rec->delete_rec(rec);
 108         TALLOC_FREE(rec);
 109         return status;
 110 }
 111
 112 NTSTATUS dbwrap_traverse(struct db_context *db,
 113                          int (*f)(struct db_record*, void*),
 114                          void *private_data,
 115                          int *count)
 116 {
 117         int ret = db->traverse(db, f, private_data);
 118
 119         if (ret < 0) {
 120                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
 121         }
 122
 123         if (count != NULL) {
 124                 *count = ret;
 125         }
 126
 127         return NT_STATUS_OK;
 128 }
 129
 130 NTSTATUS dbwrap_traverse_read(struct db_context *db,
 131                               int (*f)(struct db_record*, void*),
 132                               void *private_data,
 133                               int *count)
 134 {
 135         int ret = db->traverse_read(db, f, private_data);
 136
 137         if (ret < 0) {
 138                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
 139         }
 140
 141         if (count != NULL) {
 142                 *count = ret;
 143         }
 144
 145         return NT_STATUS_OK;
 146 }