Add dbwrap bystring service routines

(This used to be commit 1e214b536b0628db299d701839e62a4ac52727c9)

2 files changed, 47 insertions, 0 deletions

diff --git a/source3/include/dbwrap.h b/source3/include/dbwrap.h
index c175f69c4d..3bb378c841 100644
--- a/source3/include/dbwrap.h
+++ b/source3/include/dbwrap.h
@@ -50,5 +50,10 @@ struct db_context *db_open(TALLOC_CTX *mem_ctx,
 			   int hash_size, int tdb_flags,
 			   int open_flags, mode_t mode);
 
+NTSTATUS dbwrap_delete_bystring(struct db_context *db, const char *key);
+NTSTATUS dbwrap_store_bystring(struct db_context *db, const char *key,
+			       TDB_DATA data, int flags);
+TDB_DATA dbwrap_fetch_bystring(struct db_context *db, TALLOC_CTX *mem_ctx,
+			       const char *key);
 
 #endif /* __DBWRAP_H__ */
diff --git a/source3/lib/dbwrap.c b/source3/lib/dbwrap.c
index 3abd0e2ba5..4e16d18682 100644
--- a/source3/lib/dbwrap.c
+++ b/source3/lib/dbwrap.c
@@ -84,3 +84,45 @@ struct db_context *db_open(TALLOC_CTX *mem_ctx,
 
 	return result;
 }
+
+NTSTATUS dbwrap_delete_bystring(struct db_context *db, const char *key)
+{
+	struct db_record *rec;
+	NTSTATUS status;
+
+	rec = db->fetch_locked(db, talloc_tos(), string_term_tdb_data(key));
+	if (rec == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
+	status = rec->delete_rec(rec);
+	TALLOC_FREE(rec);
+	return status;
+}
+
+NTSTATUS dbwrap_store_bystring(struct db_context *db, const char *key,
+			       TDB_DATA data, int flags)
+{
+	struct db_record *rec;
+	NTSTATUS status;
+
+	rec = db->fetch_locked(db, talloc_tos(), string_term_tdb_data(key));
+	if (rec == NULL) {
+		return NT_STATUS_NO_MEMORY;
+	}
+
+	status = rec->store(rec, data, flags);
+	TALLOC_FREE(rec);
+	return status;
+}
+
+TDB_DATA dbwrap_fetch_bystring(struct db_context *db, TALLOC_CTX *mem_ctx,
+			       const char *key)
+{
+	TDB_DATA result;
+
+	if (db->fetch(db, mem_ctx, string_term_tdb_data(key), &result) == -1) {
+		return make_tdb_data(NULL, 0);
+	}
+
+	return result;
+}