From c5b10466c30759aabd472bb1baa4a5bd2540215d Mon Sep 17 00:00:00 2001 From: David Disseldorp Date: Tue, 9 Jun 2020 15:46:51 +0200 Subject: dbwrap_rbt: support TDB_INSERT and TDB_MODIFY store flags These flags provide insert-new and overwrite-existing record semantics respectively. Signed-off-by: David Disseldorp Reviewed-by: Jeremy Allison --- lib/dbwrap/dbwrap_rbt.c | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'lib/dbwrap') diff --git a/lib/dbwrap/dbwrap_rbt.c b/lib/dbwrap/dbwrap_rbt.c index d988ca776de..db456dfffba 100644 --- a/lib/dbwrap/dbwrap_rbt.c +++ b/lib/dbwrap/dbwrap_rbt.c @@ -138,6 +138,14 @@ static NTSTATUS db_rbt_storev(struct db_record *rec, return NT_STATUS_MEDIA_WRITE_PROTECTED; } + if ((flag == TDB_INSERT) && (rec_priv->node != NULL)) { + return NT_STATUS_OBJECT_NAME_COLLISION; + } + + if ((flag == TDB_MODIFY) && (rec_priv->node == NULL)) { + return NT_STATUS_OBJECT_NAME_NOT_FOUND; + } + if (num_dbufs == 1) { data = dbufs[0]; } else { -- cgit v1.2.1