Add formatted SDB setters

Author: radare

include/sdb/sdb.h

@@ -235,6 +235,7 @@ SDB_API const char *sdb_const_getf(Sdb*, ut32 *cas, const char *fmt, ...) SDB_PR
 // `vlen` the length of the value string.
 SDB_API const char *sdb_const_get_len(Sdb* s, const char *key, int *vlen, ut32 *cas);
 SDB_API int sdb_set(Sdb*, const char *key, const char *data, ut32 cas);
+SDB_API int sdb_setf(Sdb*, const char *data, ut32 cas, const char *fmt, ...) SDB_PRINTF_CHECK(4, 5);
 SDB_API int sdb_nset(Sdb*, ut64 nkey, const char *data, ut32 cas);
 SDB_API ut64 sdb_num_nget(Sdb *s, ut64 nkey, ut32 *cas);
 SDB_API int sdb_num_nset(Sdb* s, ut64 nkey, ut64 nval, ut32 cas);
@@ -252,6 +253,7 @@ SDB_API int  sdb_num_base(const char *s);
 SDB_API ut64 sdb_num_get(Sdb* s, const char *key, ut32 *cas);
 SDB_API ut64 sdb_num_getf(Sdb* s, ut32 *cas, const char *fmt, ...) SDB_PRINTF_CHECK(3, 4);
 SDB_API int  sdb_num_set(Sdb* s, const char *key, ut64 v, ut32 cas);
+SDB_API int  sdb_num_setf(Sdb* s, ut64 v, ut32 cas, const char *fmt, ...) SDB_PRINTF_CHECK(4, 5);
 SDB_API int  sdb_num_add(Sdb *s, const char *key, ut64 v, ut32 cas);
 SDB_API ut64 sdb_num_inc(Sdb* s, const char *key, ut64 n, ut32 cas);
 SDB_API ut64 sdb_num_dec(Sdb* s, const char *key, ut64 n, ut32 cas);

src/num.c

@@ -38,6 +38,22 @@ SDB_API int sdb_num_set(Sdb *s, const char *key, ut64 v, ut32 cas) {
 	return sdb_set (s, key, val, cas);
 }
 
+SDB_API int sdb_num_setf(Sdb *s, ut64 v, ut32 cas, const char *fmt, ...) {
+	va_list ap;
+	char key[SDB_MAX_KEY];
+	char val[SDB_NUM_BUFSZ];
+	int len, numbase;
+
+	va_start (ap, fmt);
+	len = vsnprintf (key, sizeof (key), fmt, ap);
+	va_end (ap);
+	if (len < 0 || len >= (int)sizeof (key)) {
+		return 0;
+	}
+	numbase = sdb_num_base (sdb_const_get (s, key, NULL));
+	return sdb_set (s, key, sdb_itoa (v, numbase, val, sizeof (val)), cas);
+}
+
 SDB_API ut64 sdb_num_inc(Sdb *s, const char *key, ut64 n2, ut32 cas) {
 	ut32 c;
 	ut64 n = sdb_num_get (s, key, &c);

src/sdb.c

@@ -775,6 +775,16 @@ SDB_API int sdb_set(Sdb* s, const char *key, const char *val, ut32 cas) {
 	return sdb_set_internal (s, key, (char *)val, false, cas);
 }
 
+SDB_API int sdb_setf(Sdb *s, const char *val, ut32 cas, const char *fmt, ...) {
+	va_list ap;
+	char key[SDB_MAX_KEY];
+	int len;
+	va_start (ap, fmt);
+	len = vsnprintf (key, sizeof (key), fmt, ap);
+	va_end (ap);
+	return len >= 0 && len < (int)sizeof (key)? sdb_set (s, key, val, cas): 0;
+}
+
 SDB_API int sdb_nset(Sdb* s, ut64 nkey, const char *val, ut32 cas) {
 	char buf[SDB_NUM_BUFSZ];
 	const char *key = sdb_itoa (nkey, 16, buf, sizeof (buf));

test/unit/test_getf.c

@@ -5,6 +5,10 @@ static bool test_getf(void) {
 	Sdb *db = sdb_new0 ();
 	mu_assert_notnull (db, "database");
 	mu_assert ("set", sdb_set (db, "user.alice.42", "admin", 0));
+	mu_assert ("formatted set", sdb_setf (db, "staff", 0, "group.%s.%d", "alice", 42));
+	mu_assert_streq (sdb_const_get (db, "group.alice.42", NULL), "staff", "formatted set value");
+	mu_assert ("formatted number set", sdb_num_setf (db, 42, 0, "number.%s", "answer"));
+	mu_assert_eq (sdb_num_get (db, "number.answer", NULL), 42, "formatted number set value");
 
 	ut32 const_cas = 0;
 	const char *const_value = sdb_const_getf (db, &const_cas,