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 "replace.h"
23 #include "lib/util/debug.h"
24 #include "lib/util/fault.h"
25 #include "lib/util/talloc_stack.h"
26 #include "dbwrap/dbwrap.h"
27 #include "dbwrap/dbwrap_private.h"
28 #include "lib/util/util_tdb.h"
29 #include "lib/util/tevent_ntstatus.h"
30
31 #undef DBGC_CLASS
32 #define DBGC_CLASS DBGC_LOCKING
33
34 /*
35 * Fall back using fetch if no genuine exists operation is provided
36 */
37
dbwrap_fallback_exists(struct db_context * db,TDB_DATA key)38 static int dbwrap_fallback_exists(struct db_context *db, TDB_DATA key)
39 {
40 NTSTATUS status = dbwrap_parse_record(db, key, NULL, NULL);
41 return NT_STATUS_IS_OK(status) ? 1 : 0;
42 }
43
delete_record(struct db_record * rec,void * data)44 static int delete_record(struct db_record *rec, void *data)
45 {
46 NTSTATUS status = dbwrap_record_delete(rec);
47 return NT_STATUS_IS_OK(status) ? 0 : -1;
48 }
49
50 /*
51 * Fallback wipe implementation using traverse and delete if no genuine
52 * wipe operation is provided
53 */
dbwrap_fallback_wipe(struct db_context * db)54 static int dbwrap_fallback_wipe(struct db_context *db)
55 {
56 NTSTATUS status = dbwrap_trans_traverse(db, delete_record, NULL);
57 return NT_STATUS_IS_OK(status) ? 0 : -1;
58 }
59
do_nothing(struct db_record * rec,void * unused)60 static int do_nothing(struct db_record *rec, void *unused)
61 {
62 return 0;
63 }
64
65 /*
66 * Fallback check operation: just traverse.
67 */
dbwrap_fallback_check(struct db_context * db)68 static int dbwrap_fallback_check(struct db_context *db)
69 {
70 NTSTATUS status = dbwrap_traverse_read(db, do_nothing, NULL, NULL);
71 return NT_STATUS_IS_OK(status) ? 0 : -1;
72 }
73
74 /*
75 * Wrapper functions for the backend methods
76 */
77
dbwrap_record_get_key(const struct db_record * rec)78 TDB_DATA dbwrap_record_get_key(const struct db_record *rec)
79 {
80 return rec->key;
81 }
82
dbwrap_record_get_value(const struct db_record * rec)83 TDB_DATA dbwrap_record_get_value(const struct db_record *rec)
84 {
85 SMB_ASSERT(rec->value_valid);
86 return rec->value;
87 }
88
dbwrap_record_storev(struct db_record * rec,const TDB_DATA * dbufs,int num_dbufs,int flags)89 NTSTATUS dbwrap_record_storev(struct db_record *rec,
90 const TDB_DATA *dbufs, int num_dbufs, int flags)
91 {
92 NTSTATUS status;
93
94 /*
95 * Invalidate before rec->storev() is called, give
96 * rec->storev() the chance to re-validate rec->value.
97 */
98 rec->value_valid = false;
99
100 status = rec->storev(rec, dbufs, num_dbufs, flags);
101 if (!NT_STATUS_IS_OK(status)) {
102 return status;
103 }
104 return NT_STATUS_OK;
105 }
106
dbwrap_record_store(struct db_record * rec,TDB_DATA data,int flags)107 NTSTATUS dbwrap_record_store(struct db_record *rec, TDB_DATA data, int flags)
108 {
109 return dbwrap_record_storev(rec, &data, 1, flags);
110 }
111
dbwrap_record_delete(struct db_record * rec)112 NTSTATUS dbwrap_record_delete(struct db_record *rec)
113 {
114 NTSTATUS status;
115
116 /*
117 * Invalidate before rec->delete_rec() is called, give
118 * rec->delete_rec() the chance to re-validate rec->value.
119 */
120 rec->value_valid = false;
121
122 status = rec->delete_rec(rec);
123 if (!NT_STATUS_IS_OK(status)) {
124 return status;
125 }
126 return NT_STATUS_OK;
127 }
128
debug_lock_order(int level,struct db_context * dbs[])129 static void debug_lock_order(int level, struct db_context *dbs[])
130 {
131 int i;
132 DEBUG(level, ("lock order: "));
133 for (i=0; i<DBWRAP_LOCK_ORDER_MAX; i++) {
134 DEBUGADD(level, (" %d:%s", i + 1, dbs[i] ? dbs[i]->name : "<none>"));
135 }
136 DEBUGADD(level, ("\n"));
137 }
138
dbwrap_lock_order_lock(struct db_context * db,struct db_context *** lockptr)139 static void dbwrap_lock_order_lock(struct db_context *db,
140 struct db_context ***lockptr)
141 {
142 static struct db_context *locked_dbs[DBWRAP_LOCK_ORDER_MAX];
143 int idx;
144
145 DBG_INFO("check lock order %d for %s\n", (int)db->lock_order,
146 db->name);
147
148 if (!DBWRAP_LOCK_ORDER_VALID(db->lock_order)) {
149 DBG_ERR("Invalid lock order %d of %s\n",
150 (int)db->lock_order, db->name);
151 smb_panic("lock order violation");
152 }
153
154 for (idx=db->lock_order-1; idx<DBWRAP_LOCK_ORDER_MAX; idx++) {
155 if (locked_dbs[idx] != NULL) {
156 DBG_ERR("Lock order violation: Trying %s at %d while "
157 "%s at %d is locked\n",
158 db->name, (int)db->lock_order,
159 locked_dbs[idx]->name, idx + 1);
160 debug_lock_order(0, locked_dbs);
161 smb_panic("lock order violation");
162 }
163 }
164
165 locked_dbs[db->lock_order-1] = db;
166 *lockptr = &locked_dbs[db->lock_order-1];
167
168 debug_lock_order(10, locked_dbs);
169 }
170
dbwrap_lock_order_unlock(struct db_context * db,struct db_context ** lockptr)171 static void dbwrap_lock_order_unlock(struct db_context *db,
172 struct db_context **lockptr)
173 {
174 DBG_INFO("release lock order %d for %s\n",
175 (int)db->lock_order, db->name);
176
177 if (*lockptr == NULL) {
178 DBG_ERR("db %s at order %d unlocked\n", db->name,
179 (int)db->lock_order);
180 smb_panic("lock order violation");
181 }
182
183 if (*lockptr != db) {
184 DBG_ERR("locked db at lock order %d is %s, expected %s\n",
185 (int)(*lockptr)->lock_order, (*lockptr)->name,
186 db->name);
187 smb_panic("lock order violation");
188 }
189
190 *lockptr = NULL;
191 }
192
193 struct dbwrap_lock_order_state {
194 struct db_context *db;
195 struct db_context **lockptr;
196 };
197
dbwrap_lock_order_state_destructor(struct dbwrap_lock_order_state * s)198 static int dbwrap_lock_order_state_destructor(
199 struct dbwrap_lock_order_state *s)
200 {
201 dbwrap_lock_order_unlock(s->db, s->lockptr);
202 return 0;
203 }
204
dbwrap_check_lock_order(struct db_context * db,TALLOC_CTX * mem_ctx)205 static struct dbwrap_lock_order_state *dbwrap_check_lock_order(
206 struct db_context *db, TALLOC_CTX *mem_ctx)
207 {
208 struct dbwrap_lock_order_state *state;
209
210 state = talloc(mem_ctx, struct dbwrap_lock_order_state);
211 if (state == NULL) {
212 DBG_WARNING("talloc failed\n");
213 return NULL;
214 }
215 state->db = db;
216
217 dbwrap_lock_order_lock(db, &state->lockptr);
218 talloc_set_destructor(state, dbwrap_lock_order_state_destructor);
219
220 return state;
221 }
222
dbwrap_fetch_locked_internal(struct db_context * db,TALLOC_CTX * mem_ctx,TDB_DATA key,struct db_record * (* db_fn)(struct db_context * db,TALLOC_CTX * mem_ctx,TDB_DATA key))223 static struct db_record *dbwrap_fetch_locked_internal(
224 struct db_context *db, TALLOC_CTX *mem_ctx, TDB_DATA key,
225 struct db_record *(*db_fn)(struct db_context *db, TALLOC_CTX *mem_ctx,
226 TDB_DATA key))
227 {
228 struct db_record *rec;
229 struct dbwrap_lock_order_state *lock_order = NULL;
230
231 if (db->lock_order != DBWRAP_LOCK_ORDER_NONE) {
232 lock_order = dbwrap_check_lock_order(db, mem_ctx);
233 if (lock_order == NULL) {
234 return NULL;
235 }
236 }
237 rec = db_fn(db, mem_ctx, key);
238 if (rec == NULL) {
239 TALLOC_FREE(lock_order);
240 return NULL;
241 }
242 (void)talloc_steal(rec, lock_order);
243 rec->db = db;
244 return rec;
245 }
246
dbwrap_fetch_locked(struct db_context * db,TALLOC_CTX * mem_ctx,TDB_DATA key)247 struct db_record *dbwrap_fetch_locked(struct db_context *db,
248 TALLOC_CTX *mem_ctx,
249 TDB_DATA key)
250 {
251 return dbwrap_fetch_locked_internal(db, mem_ctx, key,
252 db->fetch_locked);
253 }
254
dbwrap_try_fetch_locked(struct db_context * db,TALLOC_CTX * mem_ctx,TDB_DATA key)255 struct db_record *dbwrap_try_fetch_locked(struct db_context *db,
256 TALLOC_CTX *mem_ctx,
257 TDB_DATA key)
258 {
259 return dbwrap_fetch_locked_internal(
260 db, mem_ctx, key,
261 db->try_fetch_locked
262 ? db->try_fetch_locked : db->fetch_locked);
263 }
264
dbwrap_record_get_db(struct db_record * rec)265 struct db_context *dbwrap_record_get_db(struct db_record *rec)
266 {
267 return rec->db;
268 }
269
270 struct dbwrap_fetch_state {
271 TALLOC_CTX *mem_ctx;
272 TDB_DATA data;
273 };
274
dbwrap_fetch_parser(TDB_DATA key,TDB_DATA data,void * private_data)275 static void dbwrap_fetch_parser(TDB_DATA key, TDB_DATA data,
276 void *private_data)
277 {
278 struct dbwrap_fetch_state *state =
279 (struct dbwrap_fetch_state *)private_data;
280
281 state->data.dsize = data.dsize;
282 state->data.dptr = (uint8_t *)talloc_memdup(state->mem_ctx, data.dptr,
283 data.dsize);
284 }
285
dbwrap_fetch(struct db_context * db,TALLOC_CTX * mem_ctx,TDB_DATA key,TDB_DATA * value)286 NTSTATUS dbwrap_fetch(struct db_context *db, TALLOC_CTX *mem_ctx,
287 TDB_DATA key, TDB_DATA *value)
288 {
289 struct dbwrap_fetch_state state;
290 NTSTATUS status;
291
292 if (value == NULL) {
293 return NT_STATUS_INVALID_PARAMETER;
294 }
295
296 state.mem_ctx = mem_ctx;
297
298 status = dbwrap_parse_record(db, key, dbwrap_fetch_parser, &state);
299 if (!NT_STATUS_IS_OK(status)) {
300 return status;
301 }
302 if ((state.data.dsize != 0) && (state.data.dptr == NULL)) {
303 return NT_STATUS_NO_MEMORY;
304 }
305 *value = state.data;
306 return NT_STATUS_OK;
307 }
308
dbwrap_exists(struct db_context * db,TDB_DATA key)309 bool dbwrap_exists(struct db_context *db, TDB_DATA key)
310 {
311 int result;
312 if (db->exists != NULL) {
313 result = db->exists(db, key);
314 } else {
315 result = dbwrap_fallback_exists(db,key);
316 }
317 return (result == 1);
318 }
319
320 struct dbwrap_store_state {
321 TDB_DATA data;
322 int flags;
323 NTSTATUS status;
324 };
325
dbwrap_store_fn(struct db_record * rec,TDB_DATA value,void * private_data)326 static void dbwrap_store_fn(
327 struct db_record *rec,
328 TDB_DATA value,
329 void *private_data)
330 {
331 struct dbwrap_store_state *state = private_data;
332 state->status = dbwrap_record_store(rec, state->data, state->flags);
333 }
334
dbwrap_store(struct db_context * db,TDB_DATA key,TDB_DATA data,int flags)335 NTSTATUS dbwrap_store(struct db_context *db, TDB_DATA key,
336 TDB_DATA data, int flags)
337 {
338 struct dbwrap_store_state state = { .data = data, .flags = flags };
339 NTSTATUS status;
340
341 status = dbwrap_do_locked(db, key, dbwrap_store_fn, &state);
342 if (!NT_STATUS_IS_OK(status)) {
343 return status;
344 }
345
346 return state.status;
347 }
348
349 struct dbwrap_delete_state {
350 NTSTATUS status;
351 };
352
dbwrap_delete_fn(struct db_record * rec,TDB_DATA value,void * private_data)353 static void dbwrap_delete_fn(
354 struct db_record *rec,
355 TDB_DATA value,
356 void *private_data)
357 {
358 struct dbwrap_delete_state *state = private_data;
359 state->status = dbwrap_record_delete(rec);
360 }
361
dbwrap_delete(struct db_context * db,TDB_DATA key)362 NTSTATUS dbwrap_delete(struct db_context *db, TDB_DATA key)
363 {
364 struct dbwrap_delete_state state = { .status = NT_STATUS_NOT_FOUND };
365 NTSTATUS status;
366
367 status = dbwrap_do_locked(db, key, dbwrap_delete_fn, &state);
368 if (!NT_STATUS_IS_OK(status)) {
369 return status;
370 }
371
372 return state.status;
373 }
374
dbwrap_traverse(struct db_context * db,int (* f)(struct db_record *,void *),void * private_data,int * count)375 NTSTATUS dbwrap_traverse(struct db_context *db,
376 int (*f)(struct db_record*, void*),
377 void *private_data,
378 int *count)
379 {
380 int ret = db->traverse(db, f, private_data);
381
382 if (ret < 0) {
383 return NT_STATUS_INTERNAL_DB_CORRUPTION;
384 }
385
386 if (count != NULL) {
387 *count = ret;
388 }
389
390 return NT_STATUS_OK;
391 }
392
dbwrap_traverse_read(struct db_context * db,int (* f)(struct db_record *,void *),void * private_data,int * count)393 NTSTATUS dbwrap_traverse_read(struct db_context *db,
394 int (*f)(struct db_record*, void*),
395 void *private_data,
396 int *count)
397 {
398 int ret = db->traverse_read(db, f, private_data);
399
400 if (ret < 0) {
401 return NT_STATUS_INTERNAL_DB_CORRUPTION;
402 }
403
404 if (count != NULL) {
405 *count = ret;
406 }
407
408 return NT_STATUS_OK;
409 }
410
dbwrap_null_parser(TDB_DATA key,TDB_DATA val,void * data)411 static void dbwrap_null_parser(TDB_DATA key, TDB_DATA val, void* data)
412 {
413 return;
414 }
415
dbwrap_parse_record(struct db_context * db,TDB_DATA key,void (* parser)(TDB_DATA key,TDB_DATA data,void * private_data),void * private_data)416 NTSTATUS dbwrap_parse_record(struct db_context *db, TDB_DATA key,
417 void (*parser)(TDB_DATA key, TDB_DATA data,
418 void *private_data),
419 void *private_data)
420 {
421 if (parser == NULL) {
422 parser = dbwrap_null_parser;
423 }
424 return db->parse_record(db, key, parser, private_data);
425 }
426
427 struct dbwrap_parse_record_state {
428 struct db_context *db;
429 TDB_DATA key;
430 uint8_t _keybuf[64];
431 };
432
433 static void dbwrap_parse_record_done(struct tevent_req *subreq);
434
dbwrap_parse_record_send(TALLOC_CTX * mem_ctx,struct tevent_context * ev,struct db_context * db,TDB_DATA key,void (* parser)(TDB_DATA key,TDB_DATA data,void * private_data),void * private_data,enum dbwrap_req_state * req_state)435 struct tevent_req *dbwrap_parse_record_send(
436 TALLOC_CTX *mem_ctx,
437 struct tevent_context *ev,
438 struct db_context *db,
439 TDB_DATA key,
440 void (*parser)(TDB_DATA key, TDB_DATA data, void *private_data),
441 void *private_data,
442 enum dbwrap_req_state *req_state)
443 {
444 struct tevent_req *req = NULL;
445 struct tevent_req *subreq = NULL;
446 struct dbwrap_parse_record_state *state = NULL;
447 NTSTATUS status;
448
449 req = tevent_req_create(mem_ctx, &state, struct dbwrap_parse_record_state);
450 if (req == NULL) {
451 *req_state = DBWRAP_REQ_ERROR;
452 return NULL;
453 }
454
455 *state = (struct dbwrap_parse_record_state) {
456 .db = db,
457 };
458
459 if (parser == NULL) {
460 parser = dbwrap_null_parser;
461 }
462
463 *req_state = DBWRAP_REQ_INIT;
464
465 if (db->parse_record_send == NULL) {
466 /*
467 * Backend doesn't implement async version, call sync one
468 */
469 status = db->parse_record(db, key, parser, private_data);
470 if (tevent_req_nterror(req, status)) {
471 *req_state = DBWRAP_REQ_DONE;
472 return tevent_req_post(req, ev);
473 }
474
475 *req_state = DBWRAP_REQ_DONE;
476 tevent_req_done(req);
477 return tevent_req_post(req, ev);
478 }
479
480 /*
481 * Copy the key into our state ensuring the key data buffer is always
482 * available to all the dbwrap backends over the entire lifetime of the
483 * async request. Otherwise the caller might have free'd the key buffer.
484 */
485 if (key.dsize > sizeof(state->_keybuf)) {
486 state->key.dptr = talloc_memdup(state, key.dptr, key.dsize);
487 if (tevent_req_nomem(state->key.dptr, req)) {
488 return tevent_req_post(req, ev);
489 }
490 } else {
491 memcpy(state->_keybuf, key.dptr, key.dsize);
492 state->key.dptr = state->_keybuf;
493 }
494 state->key.dsize = key.dsize;
495
496 subreq = db->parse_record_send(state,
497 ev,
498 db,
499 state->key,
500 parser,
501 private_data,
502 req_state);
503 if (tevent_req_nomem(subreq, req)) {
504 *req_state = DBWRAP_REQ_ERROR;
505 return tevent_req_post(req, ev);
506 }
507
508 tevent_req_set_callback(subreq,
509 dbwrap_parse_record_done,
510 req);
511 return req;
512 }
513
dbwrap_parse_record_done(struct tevent_req * subreq)514 static void dbwrap_parse_record_done(struct tevent_req *subreq)
515 {
516 struct tevent_req *req = tevent_req_callback_data(
517 subreq, struct tevent_req);
518 struct dbwrap_parse_record_state *state = tevent_req_data(
519 req, struct dbwrap_parse_record_state);
520 NTSTATUS status;
521
522 status = state->db->parse_record_recv(subreq);
523 TALLOC_FREE(subreq);
524 if (!NT_STATUS_IS_OK(status)) {
525 tevent_req_nterror(req, status);
526 return;
527 }
528
529 tevent_req_done(req);
530 }
531
dbwrap_parse_record_recv(struct tevent_req * req)532 NTSTATUS dbwrap_parse_record_recv(struct tevent_req *req)
533 {
534 return tevent_req_simple_recv_ntstatus(req);
535 }
536
dbwrap_do_locked(struct db_context * db,TDB_DATA key,void (* fn)(struct db_record * rec,TDB_DATA value,void * private_data),void * private_data)537 NTSTATUS dbwrap_do_locked(struct db_context *db, TDB_DATA key,
538 void (*fn)(struct db_record *rec,
539 TDB_DATA value,
540 void *private_data),
541 void *private_data)
542 {
543 struct db_record *rec;
544
545 if (db->do_locked != NULL) {
546 struct db_context **lockptr = NULL;
547 NTSTATUS status;
548
549 if (db->lock_order != DBWRAP_LOCK_ORDER_NONE) {
550 dbwrap_lock_order_lock(db, &lockptr);
551 }
552
553 status = db->do_locked(db, key, fn, private_data);
554
555 if (db->lock_order != DBWRAP_LOCK_ORDER_NONE &&
556 lockptr != NULL) {
557 dbwrap_lock_order_unlock(db, lockptr);
558 }
559
560 return status;
561 }
562
563 rec = dbwrap_fetch_locked(db, db, key);
564 if (rec == NULL) {
565 return NT_STATUS_NO_MEMORY;
566 }
567
568 /*
569 * Invalidate rec->value, nobody shall assume it's set from
570 * within dbwrap_do_locked().
571 */
572 rec->value_valid = false;
573
574 fn(rec, rec->value, private_data);
575
576 TALLOC_FREE(rec);
577
578 return NT_STATUS_OK;
579 }
580
dbwrap_wipe(struct db_context * db)581 int dbwrap_wipe(struct db_context *db)
582 {
583 if (db->wipe == NULL) {
584 return dbwrap_fallback_wipe(db);
585 }
586 return db->wipe(db);
587 }
588
dbwrap_check(struct db_context * db)589 int dbwrap_check(struct db_context *db)
590 {
591 if (db->check == NULL) {
592 return dbwrap_fallback_check(db);
593 }
594 return db->check(db);
595 }
596
dbwrap_get_seqnum(struct db_context * db)597 int dbwrap_get_seqnum(struct db_context *db)
598 {
599 return db->get_seqnum(db);
600 }
601
dbwrap_transaction_start(struct db_context * db)602 int dbwrap_transaction_start(struct db_context *db)
603 {
604 if (!db->persistent) {
605 /*
606 * dbwrap_ctdb has two different data models for persistent
607 * and non-persistent databases. Transactions are supported
608 * only for the persistent databases. This check is here to
609 * prevent breakages of the cluster case, autobuild at this
610 * point only tests non-clustered Samba. Before removing this
611 * check, please make sure that this facility has also been
612 * added to dbwrap_ctdb.
613 *
614 * Thanks, vl
615 */
616 DEBUG(1, ("transactions not supported on non-persistent "
617 "database %s\n", db->name));
618 return -1;
619 }
620 return db->transaction_start(db);
621 }
622
dbwrap_transaction_start_nonblock(struct db_context * db)623 NTSTATUS dbwrap_transaction_start_nonblock(struct db_context *db)
624 {
625 if (db->transaction_start_nonblock) {
626 return db->transaction_start_nonblock(db);
627 } else {
628 return dbwrap_transaction_start(db) == 0 ? NT_STATUS_OK
629 : NT_STATUS_UNSUCCESSFUL;
630 }
631 }
632
dbwrap_transaction_commit(struct db_context * db)633 int dbwrap_transaction_commit(struct db_context *db)
634 {
635 return db->transaction_commit(db);
636 }
637
dbwrap_transaction_cancel(struct db_context * db)638 int dbwrap_transaction_cancel(struct db_context *db)
639 {
640 return db->transaction_cancel(db);
641 }
642
dbwrap_db_id(struct db_context * db,uint8_t * id,size_t idlen)643 size_t dbwrap_db_id(struct db_context *db, uint8_t *id, size_t idlen)
644 {
645 return db->id(db, id, idlen);
646 }
647
dbwrap_is_persistent(struct db_context * db)648 bool dbwrap_is_persistent(struct db_context *db)
649 {
650 return db->persistent;
651 }
652
dbwrap_name(struct db_context * db)653 const char *dbwrap_name(struct db_context *db)
654 {
655 return db->name;
656 }
657
tdb_data_buf(const TDB_DATA * dbufs,int num_dbufs,uint8_t * buf,size_t buflen)658 static ssize_t tdb_data_buf(const TDB_DATA *dbufs, int num_dbufs,
659 uint8_t *buf, size_t buflen)
660 {
661 size_t needed = 0;
662 uint8_t *p = buf;
663 int i;
664
665 for (i=0; i<num_dbufs; i++) {
666 size_t thislen = dbufs[i].dsize;
667
668 needed += thislen;
669 if (needed < thislen) {
670 /* wrap */
671 return -1;
672 }
673
674 if (p != NULL && (thislen != 0) && (needed <= buflen)) {
675 memcpy(p, dbufs[i].dptr, thislen);
676 p += thislen;
677 }
678 }
679
680 return needed;
681 }
682
683
dbwrap_merge_dbufs(TALLOC_CTX * mem_ctx,const TDB_DATA * dbufs,int num_dbufs)684 TDB_DATA dbwrap_merge_dbufs(TALLOC_CTX *mem_ctx,
685 const TDB_DATA *dbufs, int num_dbufs)
686 {
687 ssize_t len = tdb_data_buf(dbufs, num_dbufs, NULL, 0);
688 uint8_t *buf;
689
690 if (len == -1) {
691 return (TDB_DATA) {0};
692 }
693
694 buf = talloc_array(mem_ctx, uint8_t, len);
695 if (buf == NULL) {
696 return (TDB_DATA) {0};
697 }
698
699 tdb_data_buf(dbufs, num_dbufs, buf, len);
700
701 return (TDB_DATA) { .dptr = buf, .dsize = len };
702 }
703