sdb.c File Reference

#include <stdio.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <stdlib.h>
#include <sys/stat.h>
#include "sdb.h"
#include "sdb_private.h"

Go to the source code of this file.

Classes
struct	foreach_list_filter_t
struct	_match_sdb_user
struct	UnsetCallbackData
struct	LikeCallbackData

Macros
#define	BUCKET_FOREACH(ht, bt, j, kv)    for ((j) = 0, (kv) = (SdbKv *)(bt)->arr; j < (bt)->count; (j)++, (kv) = next_kv(ht, kv))
#define	BUCKET_FOREACH_SAFE(ht, bt, j, count, kv)

Functions
static SdbKv *	next_kv (HtPP *ht, SdbKv *kv)
static int	nextcas (void)
RZ_API void	sdb_global_hook (SdbHook hook, void *user)
RZ_API Sdb *	sdb_new0 (void)
RZ_API Sdb *	sdb_new (const char *path, const char *name, int lock)
RZ_API void	sdb_file (Sdb *s, const char *dir)
static bool	sdb_merge_cb (void *user, const char *k, const char *v)
RZ_API bool	sdb_merge (Sdb *d, Sdb *s)
RZ_API bool	sdb_isempty (Sdb *s)
RZ_API int	sdb_count (Sdb *s)
static void	sdb_fini (Sdb *s, int donull)
RZ_API bool	sdb_free (Sdb *s)
RZ_API const char *	sdb_const_get_len (Sdb *s, const char *key, int *vlen, ut32 *cas)
RZ_API const char *	sdb_const_get (Sdb *s, const char *key, ut32 *cas)
RZ_API char *	sdb_get_len (Sdb *s, const char *key, int *vlen, ut32 *cas)
RZ_API char *	sdb_get (Sdb *s, const char *key, ut32 *cas)
RZ_API int	sdb_unset (Sdb *s, const char *key, ut32 cas)
RZ_API bool	sdb_remove (Sdb *s, const char *key, ut32 cas)
RZ_API int	sdb_uncat (Sdb *s, const char *key, const char *value, ut32 cas)
RZ_API int	sdb_concat (Sdb *s, const char *key, const char *value, ut32 cas)
RZ_API int	sdb_add (Sdb *s, const char *key, const char *val, ut32 cas)
RZ_API bool	sdb_exists (Sdb *s, const char *key)
RZ_API int	sdb_open (Sdb *s, const char *file)
RZ_API void	sdb_close (Sdb *s)
RZ_API void	sdb_reset (Sdb *s)
static char	lastChar (const char *str)
static bool	match (const char *str, const char *expr)
RZ_API bool	sdbkv_match (SdbKv *kv, const char *expr)
RZ_API SdbKv *	sdbkv_new (const char *k, const char *v)
RZ_API SdbKv *	sdbkv_new2 (const char *k, int kl, const char *v, int vl)
RZ_API void	sdbkv_free (SdbKv *kv)
static ut32	sdb_set_internal (Sdb *s, const char *key, char *val, int owned, ut32 cas)
RZ_API int	sdb_set_owned (Sdb *s, const char *key, char *val, ut32 cas)
RZ_API int	sdb_set (Sdb *s, const char *key, const char *val, ut32 cas)
static bool	sdb_foreach_list_cb (void *user, const char *k, const char *v)
static int	__cmp_asc (const void *a, const void *b)
RZ_API SdbList *	sdb_foreach_list (Sdb *s, bool sorted)
static bool	sdb_foreach_list_filter_cb (void *user, const char *k, const char *v)
RZ_API SdbList *	sdb_foreach_list_filter_user (Sdb *s, SdbForeachCallback filter, bool sorted, void *user)
RZ_API SdbList *	sdb_foreach_list_filter (Sdb *s, SdbForeachCallback filter, bool sorted)
static bool	sdb_foreach_match_cb (void *user, const char *k, const char *v)
RZ_API SdbList *	sdb_foreach_match (Sdb *s, const char *expr, bool single)
static int	getbytes (Sdb *s, char *b, int len)
static bool	sdb_foreach_end (Sdb *s, bool result)
static bool	sdb_foreach_cdb (Sdb *s, SdbForeachCallback cb, SdbForeachCallback cb2, void *user)
RZ_API bool	sdb_foreach (Sdb *s, SdbForeachCallback cb, void *user)
static bool	_insert_into_disk (void *user, const char *key, const char *value)
static bool	_remove_afer_insert (void *user, const char *k, const char *v)
RZ_API bool	sdb_sync (Sdb *s)
RZ_API void	sdb_dump_begin (Sdb *s)
RZ_API bool	sdb_dump_hasnext (Sdb *s)
RZ_API bool	sdb_stats (Sdb *s, ut32 *disk, ut32 *mem)
RZ_API bool	sdb_dump_dupnext (Sdb *s, char *key, char **value, int *_vlen)
static ut64	parse_expire (ut64 e)
RZ_API bool	sdb_expire_set (Sdb *s, const char *key, ut64 expire, ut32 cas)
RZ_API ut64	sdb_expire_get (Sdb *s, const char *key, ut32 *cas)
RZ_API bool	sdb_hook (Sdb *s, SdbHook cb, void *user)
RZ_API bool	sdb_unhook (Sdb *s, SdbHook h)
RZ_API int	sdb_hook_call (Sdb *s, const char *k, const char *v)
RZ_API void	sdb_hook_free (Sdb *s)
RZ_API void	sdb_config (Sdb *s, int options)
RZ_API bool	sdb_unlink (Sdb *s)
RZ_API void	sdb_drain (Sdb *s, Sdb *f)
static bool	copy_foreach_cb (void *user, const char *k, const char *v)
RZ_API void	sdb_copy (Sdb *src, Sdb *dst)
static bool	unset_cb (void *user, const char *k, const char *v)
RZ_API int	sdb_unset_like (Sdb *s, const char *k)
static bool	like_cb (void *user, const char *k, const char *v)
RZ_API char **	sdb_like (Sdb *s, const char *k, const char *v, SdbForeachCallback cb)

Variables
static SdbHook	global_hook = NULL
static void *	global_user = NULL

Macro Definition Documentation

BUCKET_FOREACH

#define BUCKET_FOREACH	(		ht,
			bt,
			j,
			kv
	)		for ((j) = 0, (kv) = (SdbKv *)(bt)->arr; j < (bt)->count; (j)++, (kv) = next_kv(ht, kv))

Definition at line 17 of file sdb.c.

BUCKET_FOREACH_SAFE

#define BUCKET_FOREACH_SAFE	(		ht,
			bt,
			j,
			count,
			kv
	)

Value:

    if ((bt)->arr) \
        for ((j) = 0, (kv) = (SdbKv *)(bt)->arr, (count) = (ht)->count; \
            (j) < (bt)->count; \
            (j) = (count) == (ht)->count ? j + 1 : j, (kv) = (count) == (ht)->count ? next_kv(ht, kv) : kv, (count) = (ht)->count)

Definition at line 20 of file sdb.c.

Function Documentation

__cmp_asc()

static int __cmp_asc ( const void *  a, const void *  b  )

static

Definition at line 625 of file sdb.c.

                                                   {
    const SdbKv *ka = a, *kb = b;
    return strcmp(sdbkv_key(ka), sdbkv_key(kb));
}

References a, b, and sdbkv_key().

Referenced by sdb_foreach_list(), and sdb_foreach_list_filter_user().

_insert_into_disk()

static bool _insert_into_disk ( void *  user, const char *  key, const char *  value  )

static

Definition at line 785 of file sdb.c.

                                                                              {
    Sdb *s = (Sdb *)user;
    if (s) {
        sdb_disk_insert(s, key, value);
        return true;
    }
    return false;
}

References key, s, sdb_disk_insert(), and value.

Referenced by sdb_sync().

_remove_afer_insert()

static bool _remove_afer_insert ( void *  user, const char *  k, const char *  v  )

static

Definition at line 794 of file sdb.c.

                                                                          {
    Sdb *s = (Sdb *)user;
    if (s) {
        sdb_ht_delete(s->ht, k);
        return true;
    }
    return false;
}

References k, s, and sdb_ht_delete().

Referenced by sdb_sync().

copy_foreach_cb()

static bool copy_foreach_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 1078 of file sdb.c.

                                                                      {
    Sdb *dst = user;
    sdb_set(dst, k, v, 0);
    return true;
}

References dst, k, sdb_set(), and v.

Referenced by sdb_copy().

getbytes()

static int getbytes ( Sdb *  s, char *  b, int  len  )

static

Definition at line 717 of file sdb.c.

                                              {
    if (!cdb_read(&s->db, b, len, s->pos)) {
        return -1;
    }
    s->pos += len;
    return len;
}

References b, cdb_read(), len, and s.

Referenced by sdb_dump_dupnext().

lastChar()

static char lastChar ( const char *  str )

static

Definition at line 445 of file sdb.c.

                                      {
    int len = strlen(str);
    return str[(len > 0) ? len - 1 : 0];
}

References len, and cmd_descs_generate::str.

Referenced by match(), and RDG_fillLiteralDistrib().

like_cb()

static bool like_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 1121 of file sdb.c.

                                                              {
    LikeCallbackData *lcd = user;
    if (!user) {
        return false;
    }
    if (k && lcd->key && !sdb_match(k, lcd->key)) {
        return true;
    }
    if (v && lcd->val && !sdb_match(v, lcd->val)) {
        return true;
    }
    if (lcd->array) {
        int idx = lcd->array_index;
        int newsize = lcd->array_size + sizeof(char *) * 2;
        const char **newarray = (const char **)realloc((void *)lcd->array, newsize);
        if (!newarray) {
            return false;
        }
        lcd->array = newarray;
        lcd->array_size = newsize;
        // concatenate in array
        lcd->array[idx] = k;
        lcd->array[idx + 1] = v;
        lcd->array[idx + 2] = NULL;
        lcd->array[idx + 3] = NULL;
        lcd->array_index = idx + 2;
    } else {
        if (lcd->cb) {
            lcd->cb(lcd->sdb, k, v);
        }
    }
    return true;
}

References LikeCallbackData::array, LikeCallbackData::array_index, LikeCallbackData::array_size, LikeCallbackData::cb, setup::idx, k, LikeCallbackData::key, NULL, realloc(), LikeCallbackData::sdb, sdb_match(), v, and LikeCallbackData::val.

Referenced by sdb_like().

match()

static bool match ( const char *  str, const char *  expr  )

static

Definition at line 450 of file sdb.c.

                                                     {
    bool startsWith = *expr == '^';
    bool endsWith = lastChar(expr) == '$';
    if (startsWith && endsWith) {
        return strlen(str) == strlen(expr) - 2 &&
            !strncmp(str, expr + 1, strlen(expr) - 2);
    }
    if (startsWith) {
        return !strncmp(str, expr + 1, strlen(expr) - 1);
    }
    if (endsWith) {
        int alen = strlen(str);
        int blen = strlen(expr) - 1;
        if (alen <= blen) {
            return false;
        }
        const char *a = str + strlen(str) - blen;
        return (!strncmp(a, expr, blen));
    }
    return strstr(str, expr);
}

References a, expr(), lastChar(), and cmd_descs_generate::str.

Referenced by sdbkv_match().

next_kv()

static SdbKv* next_kv ( HtPP *  ht, SdbKv *  kv  )

inlinestatic

Definition at line 13 of file sdb.c.

                                                  {
    return (SdbKv *)((char *)kv + ht->opt.elem_size);
}

Referenced by delete(), and update_key().

nextcas()

static int nextcas ( void  )

inlinestatic

Definition at line 26 of file sdb.c.

                                {
    static ut32 cas = 1;
    if (!cas) {
        cas++;
    }
    return cas++;
}

Referenced by sdb_set_internal(), and sdbkv_new2().

parse_expire()

static ut64 parse_expire ( ut64  e )

inlinestatic

Definition at line 922 of file sdb.c.

                                        {
    const ut64 month = 30 * 24 * 60 * 60;
    if (e > 0 && e < month) {
        e += sdb_now();
    }
    return e;
}

References e, sdb_now(), and ut64().

Referenced by sdb_expire_set().

sdb_add()

RZ_API int sdb_add	(	Sdb *	s,
		const char *	key,
		const char *	val,
		ut32	cas
	)

Definition at line 351 of file sdb.c.

                                                                       {
    if (sdb_exists(s, key)) {
        return 0;
    }
    return sdb_set(s, key, val, cas);
}

References key, s, sdb_exists(), sdb_set(), and val.

Referenced by sdb_num_add().

sdb_close()

RZ_API void sdb_close

(

Sdb *

s

)

Definition at line 416 of file sdb.c.

                              {
    if (s) {
        if (s->fd != -1) {
            if (s->db.fd != -1 && s->db.fd == s->fd) {
                /* close db fd as well */
                s->db.fd = -1;
            }
            close(s->fd);
            s->fd = -1;
        }
        if (s->dir) {
            free(s->dir);
            s->dir = NULL;
        }
    }
}

References close, rz_socket_t::fd, free(), NULL, and s.

Referenced by __func_name_from_ord(), rz_analysis_function_cc_load_handler(), sdb_concat_by_path(), sdb_load_arch_platform_by_path(), sdb_load_arch_profile_by_path(), sdb_load_by_path(), sdb_load_from_string(), sdb_reset(), and session_sdb_save().

sdb_concat()

RZ_API int sdb_concat	(	Sdb *	s,
		const char *	key,
		const char *	value,
		ut32	cas
	)

Definition at line 329 of file sdb.c.

                                                                            {
    int kl, vl;
    const char *p;
    char *o;
    if (!s || !key || !*key || !value || !*value) {
        return 0;
    }
    p = sdb_const_get_len(s, key, &kl, 0);
    if (!p) {
        return sdb_set(s, key, value, cas);
    }
    vl = strlen(value);
    o = malloc(kl + vl + 1);
    if (o) {
        memcpy(o, p, kl);
        memcpy(o + kl, value, vl + 1);
        return sdb_set_owned(s, key, o, cas);
    }
    return 0;
}

References key, malloc(), memcpy(), p, s, sdb_const_get_len(), sdb_set(), sdb_set_owned(), and value.

Referenced by rz_core_search_rop(), and sdb_querys().

sdb_config()

RZ_API void sdb_config	(	Sdb *	s,
		int	options
	)

Definition at line 1041 of file sdb.c.

                                            {
    s->options = options;
    if (options & SDB_OPTION_SYNC) {
        // sync on every query
    }
    if (options & SDB_OPTION_JOURNAL) {
        // sync on every query
        sdb_journal_open(s);
        // load journaling if exists
        sdb_journal_load(s);
        sdb_journal_clear(s);
    } else {
        sdb_journal_close(s);
    }
    if (options & SDB_OPTION_NOSTAMP) {
        // sync on every query
        s->last = 0LL;
    }
    if (options & SDB_OPTION_FS) {
        // have access to fs (handle '.' or not in query)
    }
}

References options, s, sdb_journal_clear(), sdb_journal_close(), sdb_journal_load(), sdb_journal_open(), SDB_OPTION_FS, SDB_OPTION_JOURNAL, SDB_OPTION_NOSTAMP, and SDB_OPTION_SYNC.

Referenced by createdb(), main(), and sdb_grep_dump().

sdb_const_get()

RZ_API const char* sdb_const_get	(	Sdb *	s,
		const char *	key,
		ut32 *	cas
	)

Definition at line 279 of file sdb.c.

                                                                     {
    return sdb_const_get_len(s, key, NULL, cas);
}

References key, NULL, s, and sdb_const_get_len().

Referenced by append_shortcut(), bin_elf_versioninfo_verneed(), bin_elf_versioninfo_versym(), bin_pe_versioninfo(), file_load(), get_atomic_type(), get_callable_type(), get_enum_type(), get_msg_type(), parse_format(), rz_analysis_cc_arg(), rz_analysis_cc_default(), rz_analysis_cc_error(), rz_analysis_cc_exist(), rz_analysis_cc_get(), rz_analysis_cc_max_arg(), rz_analysis_cc_ret(), rz_analysis_cc_self(), rz_analysis_syscc_default(), rz_core_analysis_esil_init_mem(), rz_core_bin_export_info(), rz_core_bin_get_compile_time(), rz_project_load(), rz_serialize_flag_load(), rz_syscall_get(), rz_syscall_get_i(), sdb_array_add_sorted_num(), sdb_array_contains(), sdb_array_get(), sdb_array_get_num(), sdb_array_indexof(), sdb_array_length(), sdb_array_remove(), sdb_array_remove_num(), sdb_array_size(), sdb_bool_get(), sdb_diff_kv_cb(), sdb_num_add(), sdb_num_exists(), sdb_num_get(), sdb_num_max(), sdb_num_min(), sdb_num_set(), and sdb_querys().

sdb_const_get_len()

RZ_API const char* sdb_const_get_len	(	Sdb *	s,
		const char *	key,
		int *	vlen,
		ut32 *	cas
	)

Definition at line 220 of file sdb.c.

                                                                                    {
    ut32 pos, len;
    ut64 now = 0LL;
    bool found;
 
    if (cas) {
        *cas = 0;
    }
    if (vlen) {
        *vlen = 0;
    }
    if (!s || !key) {
        return NULL;
    }
    // TODO: optimize, iterate once
    size_t keylen = strlen(key);
 
    /* search in memory */
    SdbKv *kv = (SdbKv *)sdb_ht_find_kvp(s->ht, key, &found);
    if (found) {
        if (!sdbkv_value(kv) || !*sdbkv_value(kv)) {
            return NULL;
        }
        if (s->timestamped && kv->expire) {
            if (!now) {
                now = sdb_now();
            }
            if (now > kv->expire) {
                sdb_unset(s, key, 0);
                return NULL;
            }
        }
        if (cas) {
            *cas = kv->cas;
        }
        if (vlen) {
            *vlen = sdbkv_value_len(kv);
        }
        return sdbkv_value(kv);
    }
    /* search in disk */
    if (s->fd == -1) {
        return NULL;
    }
    (void)cdb_findstart(&s->db);
    if (cdb_findnext(&s->db, s->ht->opt.hashfn(key), key, keylen) < 1) {
        return NULL;
    }
    len = cdb_datalen(&s->db);
    if (len < SDB_CDB_MIN_VALUE || len >= SDB_CDB_MAX_VALUE) {
        return NULL;
    }
    if (vlen) {
        *vlen = len;
    }
    pos = cdb_datapos(&s->db);
    return s->db.map + pos;
}

References sdb_kv::cas, cdb_datalen, cdb_datapos, cdb_findnext(), cdb_findstart(), sdb_kv::expire, rz_socket_t::fd, found, key, len, NULL, pos, s, SDB_CDB_MAX_VALUE, sdb_ht_find_kvp(), sdb_now(), sdb_unset(), sdbkv_value(), sdbkv_value_len(), and ut64().

Referenced by sdb_array_add_sorted(), sdb_array_append(), sdb_array_insert(), sdb_array_prepend(), sdb_array_set(), sdb_concat(), sdb_const_get(), and sdb_get_len().

sdb_copy()

RZ_API void sdb_copy	(	Sdb *	src,
		Sdb *	dst
	)

Definition at line 1084 of file sdb.c.

                                         {
    sdb_foreach(src, copy_foreach_cb, dst);
    SdbListIter *it;
    SdbNs *ns;
    ls_foreach (src->ns, it, ns) {
        sdb_copy(ns->sdb, sdb_ns(dst, ns->name, true));
    }
}

References copy_foreach_cb(), dst, ls_foreach, sdb_ns_t::name, sdb_ns_t::sdb, sdb_foreach(), sdb_ns(), and src.

Referenced by rz_serialize_analysis_cc_load(), rz_serialize_analysis_cc_save(), rz_serialize_analysis_classes_load(), rz_serialize_analysis_classes_save(), rz_serialize_analysis_function_noreturn_load(), rz_serialize_analysis_function_noreturn_save(), rz_serialize_flag_load(), rz_serialize_flag_save(), and session_sdb_load_ns().

sdb_count()

RZ_API int sdb_count

(

Sdb *

s

)

Definition at line 163 of file sdb.c.

                             {
    int count = 0;
    if (s) {
        if (s->db.fd != -1) {
            sdb_dump_begin(s);
            while (sdb_dump_hasnext(s)) {
                count++;
            }
        }
        if (s->ht) {
            count += s->ht->count;
        }
    }
    return count;
}

References count, rz_socket_t::fd, s, sdb_dump_begin(), and sdb_dump_hasnext().

Referenced by __update_modal(), and resolve_syscalls().

sdb_drain()

RZ_API void sdb_drain	(	Sdb *	s,
		Sdb *	f
	)

Definition at line 1069 of file sdb.c.

                                      {
    if (s && f) {
        f->refs = s->refs;
        sdb_fini(s, 1);
        *s = *f;
        free(f);
    }
}

References f, free(), s, and sdb_fini().

Referenced by rz_cmd_kuery().

sdb_dump_begin()

RZ_API void sdb_dump_begin

(

Sdb *

s

)

Definition at line 835 of file sdb.c.

                                   {
    if (s->fd != -1) {
        s->pos = sizeof(((struct cdb_make *)0)->final);
        seek_set(s->fd, s->pos);
    } else {
        s->pos = 0;
    }
}

References rz_socket_t::fd, s, and seek_set().

Referenced by sdb_count(), sdb_foreach_cdb(), sdb_grep_dump(), sdb_isempty(), and sdb_stats().

sdb_dump_dupnext()

RZ_API bool sdb_dump_dupnext	(	Sdb *	s,
		char *	key,
		char **	value,
		int *	_vlen
	)

Definition at line 877 of file sdb.c.

                                                                          {
    ut32 vlen = 0, klen = 0;
    if (value) {
        *value = NULL;
    }
    if (_vlen) {
        *_vlen = 0;
    }
    if (!cdb_getkvlen(&s->db, &klen, &vlen, s->pos)) {
        return false;
    }
    s->pos += 4;
    if (klen < 1 || vlen < 1) {
        return false;
    }
    if (_vlen) {
        *_vlen = vlen;
    }
    if (key) {
        key[0] = 0;
        if (klen > SDB_CDB_MIN_KEY && klen < SDB_CDB_MAX_KEY) {
            if (getbytes(s, key, klen) == -1) {
                return 0;
            }
            key[klen] = 0;
        }
    }
    if (value) {
        *value = 0;
        if (vlen < SDB_CDB_MAX_VALUE) {
            *value = malloc(vlen + 10);
            if (!*value) {
                return false;
            }
            if (getbytes(s, *value, vlen) == -1) {
                free(*value);
                *value = NULL;
                return false;
            }
            (*value)[vlen] = 0;
        }
    }
    return true;
}

References cdb_getkvlen(), free(), getbytes(), key, malloc(), NULL, s, SDB_CDB_MAX_KEY, SDB_CDB_MAX_VALUE, SDB_CDB_MIN_KEY, and value.

Referenced by sdb_foreach_cdb(), and sdb_grep_dump().

sdb_dump_hasnext()

RZ_API bool sdb_dump_hasnext

(

Sdb *

s

)

Definition at line 844 of file sdb.c.

                                     {
    ut32 k, v;
    if (!cdb_getkvlen(&s->db, &k, &v, s->pos)) {
        return false;
    }
    if (k < 1 || v < 1) {
        return false;
    }
    s->pos += k + v + 4;
    return true;
}

References cdb_getkvlen(), k, s, and v.

Referenced by sdb_count(), sdb_isempty(), and sdb_stats().

sdb_exists()

RZ_API bool sdb_exists	(	Sdb *	s,
		const char *	key
	)

Definition at line 358 of file sdb.c.

                                                {
    ut32 pos;
    char ch;
    bool found;
    int klen = strlen(key) + 1;
    if (!s) {
        return false;
    }
    SdbKv *kv = (SdbKv *)sdb_ht_find_kvp(s->ht, key, &found);
    if (found && kv) {
        char *v = sdbkv_value(kv);
        return v && *v;
    }
    if (s->fd == -1) {
        return false;
    }
    (void)cdb_findstart(&s->db);
    if (cdb_findnext(&s->db, sdb_hash(key), key, klen)) {
        pos = cdb_datapos(&s->db);
        cdb_read(&s->db, &ch, 1, pos);
        return ch != 0;
    }
    return false;
}

References cdb_datapos, cdb_findnext(), cdb_findstart(), cdb_read(), rz_socket_t::fd, found, key, pos, s, sdb_hash(), sdb_ht_find_kvp(), sdbkv_value(), and v.

Referenced by rz_analysis_class_create(), rz_analysis_class_delete_attr_raw(), rz_analysis_class_exists_raw(), and sdb_add().

sdb_expire_get()

RZ_API ut64 sdb_expire_get	(	Sdb *	s,
		const char *	key,
		ut32 *	cas
	)

Definition at line 971 of file sdb.c.

                                                               {
    bool found = false;
    SdbKv *kv = (SdbKv *)sdb_ht_find_kvp(s->ht, key, &found);
    if (found && kv && *sdbkv_value(kv)) {
        if (cas) {
            *cas = kv->cas;
        }
        return kv->expire;
    }
    return 0LL;
}

References sdb_kv::cas, sdb_kv::expire, found, key, s, sdb_ht_find_kvp(), and sdbkv_value().

sdb_expire_set()

RZ_API bool sdb_expire_set	(	Sdb *	s,
		const char *	key,
		ut64	expire,
		ut32	cas
	)

Definition at line 930 of file sdb.c.

                                                                           {
    char *buf;
    ut32 pos, len;
    SdbKv *kv;
    bool found;
    s->timestamped = true;
    if (!key) {
        s->expire = parse_expire(expire);
        return true;
    }
    kv = (SdbKv *)sdb_ht_find_kvp(s->ht, key, &found);
    if (found && kv) {
        if (*sdbkv_value(kv)) {
            if (!cas || cas == kv->cas) {
                kv->expire = parse_expire(expire);
                return true;
            }
        }
        return false;
    }
    if (s->fd == -1) {
        return false;
    }
    (void)cdb_findstart(&s->db);
    if (!cdb_findnext(&s->db, sdb_hash(key), key, strlen(key) + 1)) {
        return false;
    }
    pos = cdb_datapos(&s->db);
    len = cdb_datalen(&s->db);
    if (len < 1 || len >= INT32_MAX) {
        return false;
    }
    if (!(buf = calloc(1, len + 1))) {
        return false;
    }
    cdb_read(&s->db, buf, len, pos);
    buf[len] = 0;
    sdb_set_owned(s, key, buf, cas);
    return sdb_expire_set(s, key, expire, cas); // recursive
}

References calloc(), sdb_kv::cas, cdb_datalen, cdb_datapos, cdb_findnext(), cdb_findstart(), cdb_read(), sdb_kv::expire, rz_socket_t::fd, found, INT32_MAX, key, len, parse_expire(), pos, s, sdb_hash(), sdb_ht_find_kvp(), sdb_set_owned(), and sdbkv_value().

sdb_file()

RZ_API void sdb_file	(	Sdb *	s,
		const char *	dir
	)

Definition at line 128 of file sdb.c.

                                              {
    if (s->lock) {
        sdb_unlock(sdb_lock_file(s->dir));
    }
    free(s->dir);
    s->dir = (dir && *dir) ? strdup(dir) : NULL;
    if (s->lock) {
        sdb_lock(sdb_lock_file(s->dir));
    }
}

References free(), NULL, s, sdb_lock(), sdb_lock_file(), sdb_unlock(), and strdup().

Referenced by rz_cmd_kuery(), and session_sdb_save().

sdb_fini()

static void sdb_fini ( Sdb *  s, int  donull  )

static

Definition at line 179 of file sdb.c.

                                         {
    if (!s) {
        return;
    }
    sdb_hook_free(s);
    cdb_free(&s->db);
    if (s->lock) {
        sdb_unlock(sdb_lock_file(s->dir));
    }
    sdb_ns_free_all(s);
    s->refs = 0;
    free(s->name);
    free(s->path);
    ls_free(s->ns);
    sdb_ht_free(s->ht);
    sdb_journal_close(s);
    if (s->fd != -1) {
        close(s->fd);
        s->fd = -1;
    }
    free(s->ndump);
    free(s->dir);
    if (donull) {
        memset(s, 0, sizeof(Sdb));
    }
}

References cdb_free(), close, rz_socket_t::fd, free(), ls_free(), memset(), s, sdb_hook_free(), sdb_ht_free(), sdb_journal_close(), sdb_lock_file(), sdb_ns_free_all(), and sdb_unlock().

Referenced by sdb_drain(), sdb_free(), and sdb_unlink().

sdb_foreach()

RZ_API bool sdb_foreach	(	Sdb *	s,
		SdbForeachCallback	cb,
		void *	user
	)

Definition at line 758 of file sdb.c.

                                                                   {
    if (!s) {
        return false;
    }
    s->depth++;
    bool result = sdb_foreach_cdb(s, cb, NULL, user);
    if (!result) {
        return sdb_foreach_end(s, false);
    }
 
    ut32 i;
    for (i = 0; i < s->ht->size; ++i) {
        HtPPBucket *bt = &s->ht->table[i];
        SdbKv *kv;
        ut32 j, count;
 
        BUCKET_FOREACH_SAFE(s->ht, bt, j, count, kv) {
            if (kv && sdbkv_value(kv) && *sdbkv_value(kv)) {
                if (!cb(user, sdbkv_key(kv), sdbkv_value(kv))) {
                    return sdb_foreach_end(s, false);
                }
            }
        }
    }
    return sdb_foreach_end(s, true);
}

References BUCKET_FOREACH_SAFE, cb, count, i, NULL, s, sdb_foreach_cdb(), sdb_foreach_end(), sdbkv_key(), and sdbkv_value().

Referenced by deserialize_checkpoints(), deserialize_memory(), deserialize_registers(), load_flags(), rz_analysis_class_foreach(), rz_cmd_kuery(), rz_config_unserialize(), rz_debug_signal_list(), rz_display_opcode_handler(), rz_project_migrate_v1_v2(), rz_project_migrate_v2_v3(), rz_project_migrate_v3_v4(), rz_serialize_analysis_blocks_load(), rz_serialize_analysis_functions_load(), rz_serialize_analysis_global_var_load(), rz_serialize_analysis_hints_load(), rz_serialize_analysis_imports_load(), rz_serialize_analysis_meta_load(), rz_serialize_analysis_xrefs_load(), rz_serialize_bp_load(), rz_serialize_config_load(), rz_serialize_flag_zones_load(), rz_serialize_spaces_load(), rz_syscall_list(), sdb_copy(), sdb_diff_ctx(), sdb_diff_report(), sdb_foreach_list(), sdb_foreach_list_filter_user(), sdb_foreach_match(), sdb_like(), sdb_merge(), sdb_unset_like(), text_save(), and walk_namespace().

sdb_foreach_cdb()

static bool sdb_foreach_cdb ( Sdb *  s, SdbForeachCallback  cb, SdbForeachCallback  cb2, void *  user  )

static

Definition at line 730 of file sdb.c.

                                                                                               {
    char *v = NULL;
    char k[SDB_CDB_MAX_KEY] = { 0 };
    bool found;
    sdb_dump_begin(s);
    while (sdb_dump_dupnext(s, k, &v, NULL)) {
        SdbKv *kv = sdb_ht_find_kvp(s->ht, k, &found);
        if (found) {
            free(v);
            if (kv && sdbkv_key(kv) && sdbkv_value(kv)) {
                if (!cb(user, sdbkv_key(kv), sdbkv_value(kv))) {
                    return false;
                }
                if (cb2 && !cb2(user, k, sdbkv_value(kv))) {
                    return false;
                }
            }
        } else {
            if (!cb(user, k, v)) {
                free(v);
                return false;
            }
            free(v);
        }
    }
    return true;
}

References cb, found, free(), k, NULL, s, SDB_CDB_MAX_KEY, sdb_dump_begin(), sdb_dump_dupnext(), sdb_ht_find_kvp(), sdbkv_key(), sdbkv_value(), and v.

Referenced by sdb_foreach(), and sdb_sync().

sdb_foreach_end()

static bool sdb_foreach_end ( Sdb *  s, bool  result  )

static

Definition at line 725 of file sdb.c.

                                                 {
    s->depth--;
    return result;
}

References s.

Referenced by sdb_foreach().

sdb_foreach_list()

RZ_API SdbList* sdb_foreach_list	(	Sdb *	s,
		bool	sorted
	)

Definition at line 630 of file sdb.c.

                                                      {
    SdbList *list = ls_newf((SdbListFree)sdbkv_free);
    sdb_foreach(s, sdb_foreach_list_cb, list);
    if (sorted) {
        ls_sort(list, __cmp_asc);
    }
    return list;
}

References __cmp_asc(), list(), ls_newf(), ls_sort(), s, sdb_foreach(), sdb_foreach_list_cb(), and sdbkv_free().

Referenced by __delete_almighty(), __exec_almighty(), __init_almighty_db(), __set_rcb(), __update_modal(), rop_kuery(), rz_analysis_calling_conventions(), rz_analysis_class_get_all(), rz_analysis_noreturn_functions(), rz_cmd_search(), rz_core_bin_export_info(), rz_flag_tags_list(), sdb_load_arch_profile(), sdb_load_base_types(), sdb_load_platform_profile(), sdb_load_sysregs(), sdb_querys(), text_save(), and typelinks_load_sdb().

sdb_foreach_list_cb()

static bool sdb_foreach_list_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 615 of file sdb.c.

                                                                          {
    SdbList *list = (SdbList *)user;
    SdbKv *kv = RZ_NEW0(SdbKv);
    /* seems like some k/v are constructed in the stack and cant be used after returning */
    kv->base.key = strdup(k);
    kv->base.value = strdup(v);
    ls_append(list, kv);
    return 1;
}

References sdb_kv::base, k, list(), ls_append(), RZ_NEW0, strdup(), and v.

Referenced by sdb_foreach_list().

sdb_foreach_list_filter()

RZ_API SdbList* sdb_foreach_list_filter	(	Sdb *	s,
		SdbForeachCallback	filter,
		bool	sorted
	)

Definition at line 685 of file sdb.c.

                                                                                        {
    return sdb_foreach_list_filter_user(s, filter, sorted, NULL);
}

References NULL, s, and sdb_foreach_list_filter_user().

Referenced by rz_analysis_dwarf_integrate_functions(), and sdb_load_callables().

sdb_foreach_list_filter_cb()

static bool sdb_foreach_list_filter_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 645 of file sdb.c.

                                                                                 {
    struct foreach_list_filter_t *u = (struct foreach_list_filter_t *)user;
    SdbList *list = u->list;
    SdbKv *kv = NULL;
 
    if (!u->filter || u->filter(u->user, k, v)) {
        kv = RZ_NEW0(SdbKv);
        if (!kv) {
            goto err;
        }
        kv->base.key = strdup(k);
        kv->base.value = strdup(v);
        if (!kv->base.key || !kv->base.value) {
            goto err;
        }
        ls_append(list, kv);
    }
    return true;
err:
    sdbkv_free(kv);
    return false;
}

References sdb_kv::base, err, foreach_list_filter_t::filter, k, list(), foreach_list_filter_t::list, ls_append(), NULL, RZ_NEW0, sdbkv_free(), strdup(), foreach_list_filter_t::user, and v.

Referenced by sdb_foreach_list_filter_user().

sdb_foreach_list_filter_user()

RZ_API SdbList* sdb_foreach_list_filter_user	(	Sdb *	s,
		SdbForeachCallback	filter,
		bool	sorted,
		void *	user
	)

Definition at line 668 of file sdb.c.

                                                                                                         {
    struct foreach_list_filter_t u;
    SdbList *list = ls_newf((SdbListFree)sdbkv_free);
 
    if (!list) {
        return NULL;
    }
    u.filter = filter;
    u.list = list;
    u.user = user;
    sdb_foreach(s, sdb_foreach_list_filter_cb, &u);
    if (sorted) {
        ls_sort(list, __cmp_asc);
    }
    return list;
}

References __cmp_asc(), filter(), foreach_list_filter_t::filter, list(), foreach_list_filter_t::list, ls_newf(), ls_sort(), NULL, s, sdb_foreach(), sdb_foreach_list_filter_cb(), sdbkv_free(), and foreach_list_filter_t::user.

Referenced by sdb_foreach_list_filter().

sdb_foreach_match()

RZ_API SdbList* sdb_foreach_match	(	Sdb *	s,
		const char *	expr,
		bool	single
	)

Definition at line 710 of file sdb.c.

                                                                         {
    SdbList *list = ls_newf((SdbListFree)sdbkv_free);
    _match_sdb_user o = { expr, list, single };
    sdb_foreach(s, sdb_foreach_match_cb, &o);
    return list;
}

References expr(), list(), ls_newf(), s, sdb_foreach(), sdb_foreach_match_cb(), and sdbkv_free().

Referenced by sdb_querys().

sdb_foreach_match_cb()

static bool sdb_foreach_match_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 695 of file sdb.c.

                                                                           {
    _match_sdb_user *o = (_match_sdb_user *)user;
    SdbKv tkv = { .base.key = (char *)k, .base.value = (char *)v };
    if (sdbkv_match(&tkv, o->expr)) {
        SdbKv *kv = RZ_NEW0(SdbKv);
        kv->base.key = strdup(k);
        kv->base.value = strdup(v);
        ls_append(o->list, kv);
        if (o->single) {
            return false;
        }
    }
    return true;
}

References sdb_kv::base, _match_sdb_user::expr, k, _match_sdb_user::list, ls_append(), RZ_NEW0, sdbkv_match(), _match_sdb_user::single, strdup(), and v.

Referenced by sdb_foreach_match().

sdb_free()

RZ_API bool sdb_free

(

Sdb *

s

)

Definition at line 206 of file sdb.c.

                             {
    if (s && s->ht && s->refs) {
        s->refs--;
        if (s->refs < 1) {
            s->refs = 0;
            sdb_fini(s, 0);
            s->ht = NULL;
            free(s);
            return true;
        }
    }
    return false;
}

References free(), NULL, s, and sdb_fini().

Referenced by _parse_resource_directory(), bin_pe_parse_imports(), config_path(), dbdiff(), dot_trace_traverse(), ds_free(), get_gnu_verneed(), get_gnu_versym(), get_vernaux_entry_sdb(), get_verneed_entry_sdb(), get_verneed_entry_sdb_aux(), ns_free_exc_list(), Pe_r_bin_store_string(), Pe_r_bin_store_string_table(), rz_agraph_free(), rz_analysis_esil_free(), rz_analysis_esil_stats(), rz_analysis_free(), rz_analysis_function_cc_load_handler(), rz_asm_free(), rz_asm_use(), rz_bin_elf_free(), rz_bin_elf_get_symbols_info(), rz_bin_file_free(), rz_bin_free(), rz_bin_mdmp_free_pe32_bin(), rz_bin_mdmp_free_pe64_bin(), rz_bin_object_set_items(), rz_core_bin_apply_relocs(), rz_core_fini(), rz_core_sysenv_begin(), rz_debug_free(), rz_debug_session_load(), rz_debug_session_save(), rz_egg_free(), rz_flag_free(), rz_main_rz_bin(), rz_project_free(), rz_project_load_file(), rz_project_load_file_raw(), rz_project_save_file(), rz_syscall_free(), rz_syscall_setup(), sdb_concat_by_path(), sdb_grep_dump(), sdb_load_arch_platform_by_path(), sdb_load_arch_profile_by_path(), sdb_load_by_path(), sdb_load_from_string(), sdb_ns_free(), sdb_ns_set(), session_sdb_load(), session_sdb_load_ns(), set_bin_relocs(), showcount(), and terminate().

sdb_get()

RZ_API char* sdb_get	(	Sdb *	s,
		const char *	key,
		ut32 *	cas
	)

Definition at line 290 of file sdb.c.

                                                         {
    return sdb_get_len(s, key, NULL, cas);
}

References key, NULL, s, and sdb_get_len().

Referenced by __create_panel_db(), __func_name_from_ord(), __search_db(), get_callable_type(), get_enum_type(), get_struct_type(), get_type_data(), get_typedef_type(), get_union_type(), rename_key(), resolveModuleOrdinal(), rz_analysis_class_base_get_all(), rz_analysis_class_delete(), rz_analysis_class_get_attr_raw(), rz_analysis_class_method_delete_class(), rz_analysis_class_method_get_all(), rz_analysis_class_method_rename_class(), rz_analysis_class_rename(), rz_analysis_class_rename_attr_raw(), rz_analysis_class_vtable_delete_class(), rz_analysis_class_vtable_get_all(), rz_analysis_class_vtable_rename_class(), rz_analysis_dwarf_integrate_functions(), rz_asm_describe(), rz_core_bin_apply_resources(), rz_core_bin_info_print(), rz_egg_option_get(), rz_flag_tags_get(), rz_serialize_core_load(), rz_serialize_spaces_load(), sdb_array_delete(), sdb_array_pop_head(), sdb_array_pop_tail(), sdb_array_sort_num(), sdb_load_platform_profile(), sdb_load_sysregs(), and sdb_querys().

sdb_get_len()

RZ_API char* sdb_get_len	(	Sdb *	s,
		const char *	key,
		int *	vlen,
		ut32 *	cas
	)

Definition at line 285 of file sdb.c.

                                                                        {
    const char *value = sdb_const_get_len(s, key, vlen, cas);
    return value ? strdup(value) : NULL;
}

References key, NULL, s, sdb_const_get_len(), strdup(), and value.

Referenced by sdb_array_sort(), sdb_get(), and sdb_uncat().

sdb_global_hook()

RZ_API void sdb_global_hook	(	SdbHook	hook,
		void *	user
	)

Definition at line 37 of file sdb.c.

                                                      {
    global_hook = hook;
    global_user = user;
}

References global_hook, and global_user.

sdb_hook()

RZ_API bool sdb_hook	(	Sdb *	s,
		SdbHook	cb,
		void *	user
	)

Definition at line 983 of file sdb.c.

                                                     {
    int i = 0;
    SdbHook hook;
    SdbListIter *iter;
    if (s->hooks) {
        ls_foreach (s->hooks, iter, hook) {
            if (!(i % 2) && (hook == cb)) {
                return false;
            }
            i++;
        }
    } else {
        s->hooks = ls_new();
        s->hooks->free = NULL;
    }
    ls_append(s->hooks, (void *)cb);
    ls_append(s->hooks, user);
    return true;
}

References cb, i, ls_append(), ls_foreach, ls_new(), NULL, and s.

Referenced by sdb_new().

sdb_hook_call()

RZ_API int sdb_hook_call	(	Sdb *	s,
		const char *	k,
		const char *	v
	)

Definition at line 1019 of file sdb.c.

                                                               {
    SdbListIter *iter;
    SdbHook hook;
    int i = 0;
    if (s->timestamped && s->last) {
        s->last = sdb_now();
    }
    ls_foreach (s->hooks, iter, hook) {
        if (!(i % 2) && k && iter->n) {
            void *u = iter->n->data;
            hook(s, u, k, v);
        }
        i++;
    }
    return i >> 1;
}

References i, k, ls_foreach, s, sdb_now(), and v.

Referenced by sdb_set_internal().

sdb_hook_free()

RZ_API void sdb_hook_free

(

Sdb *

s

)

Definition at line 1036 of file sdb.c.

                                  {
    ls_free(s->hooks);
    s->hooks = NULL;
}

References ls_free(), NULL, and s.

Referenced by sdb_fini().

sdb_isempty()

RZ_API bool sdb_isempty

(

Sdb *

s

)

Definition at line 148 of file sdb.c.

                                {
    if (s) {
        if (s->db.fd != -1) {
            sdb_dump_begin(s);
            while (sdb_dump_hasnext(s)) {
                return false;
            }
        }
        if (s->ht && s->ht->count > 0) {
            return false;
        }
    }
    return true;
}

References rz_socket_t::fd, s, sdb_dump_begin(), and sdb_dump_hasnext().

Referenced by rz_core_analysis_cc_init().

sdb_like()

RZ_API char** sdb_like	(	Sdb *	s,
		const char *	k,
		const char *	v,
		SdbForeachCallback	cb
	)

Definition at line 1155 of file sdb.c.

                                                                                    {
    LikeCallbackData lcd = { s, k, v, cb, NULL, 0, 0 };
    if (cb) {
        sdb_foreach(s, like_cb, &lcd);
        return NULL;
    }
    if (k && !*k) {
        lcd.key = NULL;
    }
    if (v && !*v) {
        lcd.val = NULL;
    }
    lcd.array_size = sizeof(char *) * 2;
    lcd.array = calloc(lcd.array_size, 1);
    if (!lcd.array) {
        return NULL;
    }
    lcd.array_index = 0;
    sdb_foreach(s, like_cb, &lcd);
    if (lcd.array_index == 0) {
        free((void *)lcd.array);
        return NULL;
    }
    return (char **)lcd.array;
}

References LikeCallbackData::array, LikeCallbackData::array_index, LikeCallbackData::array_size, calloc(), cb, free(), k, LikeCallbackData::key, like_cb(), NULL, s, sdb_foreach(), v, and LikeCallbackData::val.

sdb_merge()

RZ_API bool sdb_merge	(	Sdb *	d,
		Sdb *	s
	)

Definition at line 144 of file sdb.c.

                                      {
    return sdb_foreach(s, sdb_merge_cb, d);
}

References d, s, sdb_foreach(), and sdb_merge_cb().

Referenced by rz_analysis_function_cc_load_handler(), and sdb_concat_by_path().

sdb_merge_cb()

static bool sdb_merge_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 139 of file sdb.c.

                                                                   {
    sdb_set(user, k, v, 0);
    return true;
}

References k, sdb_set(), and v.

Referenced by sdb_merge().

sdb_new()

RZ_API Sdb* sdb_new	(	const char *	path,
		const char *	name,
		int	lock
	)

Definition at line 47 of file sdb.c.

                                                                  {
    Sdb *s = RZ_NEW0(Sdb);
    if (!s) {
        return NULL;
    }
    s->db.fd = -1;
    s->fd = -1;
    s->refs = 1;
    if (path && !*path) {
        path = NULL;
    }
    if (name && *name && strcmp(name, "-")) {
        if (path && *path) {
            int plen = strlen(path);
            int nlen = strlen(name);
            s->dir = malloc(plen + nlen + 2);
            if (!s->dir) {
                free(s);
                return NULL;
            }
            memcpy(s->dir, path, plen);
            s->dir[plen] = '/';
            memcpy(s->dir + plen + 1, name, nlen + 1);
            s->path = strdup(path);
        } else {
            s->dir = strdup(name);
        }
        switch (lock) {
        case 1:
            if (!sdb_lock(sdb_lock_file(s->dir))) {
                goto fail;
            }
            break;
        case 2:
            if (!sdb_lock_wait(sdb_lock_file(s->dir))) {
                goto fail;
            }
            break;
        }
        if (sdb_open(s, s->dir) == -1) {
            s->last = s->timestamped ? sdb_now() : 0LL;
            // TODO: must fail if we cant open for write in sync
        }
        s->name = strdup(name);
    } else {
        s->last = s->timestamped ? sdb_now() : 0LL;
        s->fd = -1;
    }
    s->journal = -1;
    s->fdump = -1;
    s->depth = 0;
    s->ndump = NULL;
    s->ns = ls_new(); // TODO: should be NULL
    if (!s->ns) {
        goto fail;
    }
    s->ns->free = NULL;
    if (!s->ns) {
        goto fail;
    }
    s->ht = sdb_ht_new();
    s->lock = lock;
    // if open fails ignore
    if (global_hook) {
        sdb_hook(s, global_hook, global_user);
    }
    cdb_init(&s->db, s->fd);
    return s;
fail:
    if (s->fd != -1) {
        close(s->fd);
        s->fd = -1;
    }
    free(s->dir);
    free(s->name);
    free(s->path);
    free(s);
    return NULL;
}

References cdb_init(), close, fail, rz_socket_t::fd, free(), global_hook, global_user, lock(), ls_new(), malloc(), memcpy(), NULL, path, RZ_NEW0, s, sdb_hook(), sdb_ht_new(), sdb_lock(), sdb_lock_file(), sdb_lock_wait(), sdb_now(), sdb_open(), and strdup().

Referenced by __func_name_from_ord(), bin_pe_parse_imports(), config_path(), createdb(), dbdiff(), load_sdb(), main(), rz_analysis_function_cc_load_handler(), rz_asm_use(), rz_cmd_kuery(), rz_core_init(), rz_core_sysenv_begin(), rz_debug_signal_init(), rz_egg_new(), rz_main_rz_bin(), sdb_concat_by_path(), sdb_grep_dump(), sdb_load_arch_platform_by_path(), sdb_load_arch_profile_by_path(), sdb_load_by_path(), sdb_new0(), set_bin_relocs(), and showcount().

sdb_new0()

RZ_API Sdb* sdb_new0

(

void

)

Definition at line 43 of file sdb.c.

                           {
    return sdb_new(NULL, NULL, 0);
}

References NULL, and sdb_new().

Referenced by __init(), _parse_resource_directory(), _store_resource_sdb(), agraph_init(), dot_trace_traverse(), get_gnu_verneed(), get_gnu_versym(), get_sdb(), get_vernaux_entry_sdb(), get_verneed_entry_sdb_aux(), init_msg_types(), load_buffer(), Pe_r_bin_store_fixed_file_info(), Pe_r_bin_store_resource_version_info(), Pe_r_bin_store_string(), Pe_r_bin_store_string_file_info(), Pe_r_bin_store_string_table(), Pe_r_bin_store_var(), Pe_r_bin_store_var_file_info(), rz_analysis_esil_stats(), rz_analysis_new(), rz_bin_dmp64_new_buf(), rz_bin_elf_get_symbols_info(), rz_bin_elf_new_buf(), rz_bin_file_new(), rz_bin_mdmp_new_buf(), rz_bin_mz_init(), rz_bin_new(), rz_bin_pe_new_buf(), rz_bin_te_new_buf(), rz_debug_session_save(), rz_flag_new(), rz_project_load_file_raw(), rz_project_save_file(), rz_syscall_new(), sdb_load_from_string(), sdb_ns_new(), session_sdb_load(), session_sdb_load_ns(), and ssa_get().

sdb_open()

RZ_API int sdb_open	(	Sdb *	s,
		const char *	file
	)

Definition at line 383 of file sdb.c.

                                              {
    struct stat st;
    if (!s) {
        return -1;
    }
    if (file) {
        if (s->fd != -1) {
            close(s->fd);
            s->fd = -1;
        }
        s->fd = open(file, O_RDONLY | O_BINARY);
        if (file != s->dir) {
            free(s->dir);
            s->dir = strdup(file);
            s->path = NULL; // TODO: path is important
        }
    }
    s->last = 0LL;
    if (s->fd != -1 && fstat(s->fd, &st) != -1) {
        if ((S_IFREG & st.st_mode) != S_IFREG) {
            eprintf("Database must be a file\n");
            close(s->fd);
            s->fd = -1;
            return -1;
        }
        s->last = st.st_mtime;
    }
    if (s->fd != -1) {
        cdb_init(&s->db, s->fd);
    }
    return s->fd;
}

References cdb_init(), close, eprintf, rz_socket_t::fd, free(), fstat, NULL, O_BINARY, O_RDONLY, s, and strdup().

Referenced by load_sdb(), sdb_disk_finish(), sdb_new(), and session_sdb_load_ns().

sdb_remove()

RZ_API bool sdb_remove	(	Sdb *	s,
		const char *	key,
		ut32	cas
	)

Definition at line 299 of file sdb.c.

                                                          {
    return sdb_ht_delete(s->ht, key);
}

References key, s, and sdb_ht_delete().

Referenced by __delete_almighty(), rename_key(), rz_analysis_class_delete(), rz_analysis_class_delete_attr_raw(), rz_analysis_class_rename_attr_raw(), and sdb_sync().

sdb_reset()

RZ_API void sdb_reset

(

Sdb *

s

)

Definition at line 433 of file sdb.c.

                              {
    if (!s) {
        return;
    }
    /* ignore disk cache, file is not removed, but we will ignore
     * its values when syncing again */
    sdb_close(s);
    /* empty memory hashtable */
    sdb_ht_free(s->ht);
    s->ht = sdb_ht_new();
}

References s, sdb_close(), sdb_ht_free(), and sdb_ht_new().

Referenced by cmd_analysis_esil(), load_sdb(), rz_agraph_reset(), rz_analysis_esil_stats(), rz_analysis_purge(), rz_core_analysis_cc_init(), rz_core_analysis_esil_deinit(), rz_flag_tags_reset(), rz_serialize_analysis_classes_load(), rz_serialize_analysis_function_noreturn_load(), and rz_type_cc_del_all_handler().

sdb_set()

RZ_API int sdb_set	(	Sdb *	s,
		const char *	key,
		const char *	val,
		ut32	cas
	)

Definition at line 611 of file sdb.c.

                                                                       {
    return sdb_set_internal(s, key, (char *)val, 0, cas);
}

References key, s, sdb_set_internal(), and val.

Referenced by __init_sdb(), _store_resource_sdb(), art_header_load(), bin_pe_init_hdr(), block_store(), bootimg_header_load(), copy_foreach_cb(), core_analysis_graph_construct_edges(), core_analysis_graph_construct_nodes(), core_analysis_graph_nodes(), file_save(), file_save_cb(), flag_save_cb(), function_store(), get_bbnodes(), get_bbupdate(), get_class_ro_t(), get_elf_intrp(), get_gnu_versym(), get_libs(), get_sdb(), get_vernaux_entry_sdb(), get_verneed_entry_sdb_aux(), hints_acc_store_cb(), info(), init_dt_dynamic_sdb(), init_ehdr_sdb(), init_items(), init_phdr_sdb(), init_sdb_formats(), init_shdr_sdb(), java_set_sdb(), kv_loadlibs(), lmf_header_load(), load_process_line(), parse_segments(), parse_signature(), Pe_r_bin_store_string(), Pe_r_bin_store_string_table(), rename_key(), rop_classify(), rz_agraph_del_node(), rz_agraph_set_curnode(), rz_agraph_set_title(), rz_analysis_cc_set(), rz_analysis_cc_set_error(), rz_analysis_cc_set_self(), rz_analysis_class_create(), rz_analysis_class_rename_attr_raw(), rz_analysis_class_set_attr_raw(), rz_analysis_set_cc_default(), rz_analysis_set_syscc_default(), rz_bin_mdmp_init_directory(), rz_bin_mdmp_init_directory_entry(), rz_bin_mdmp_init_hdr(), rz_bin_mdmp_init_parsing(), rz_bin_mz_init_hdr(), rz_bin_te_init_hdr(), rz_config_serialize(), rz_core_analysis_esil_init_mem(), rz_debug_signal_init(), rz_egg_option_set(), rz_flag_tags_set(), rz_project_migrate_v4_v5(), rz_project_save(), rz_serialize_analysis_global_var_save(), rz_serialize_analysis_imports_save(), rz_serialize_bp_save(), rz_serialize_config_save(), rz_serialize_core_save(), rz_serialize_flag_save(), rz_serialize_flag_zones_save(), rz_serialize_spaces_save(), save_atomic_type(), save_callable(), save_enum(), save_struct(), save_typedef(), save_typelink(), save_union(), sdb_add(), sdb_array_add_sorted_num(), sdb_array_append(), sdb_array_insert(), sdb_array_pop_head(), sdb_array_prepend(), sdb_array_set(), sdb_bool_set(), sdb_concat(), sdb_journal_load(), sdb_merge_cb(), sdb_num_dec(), sdb_num_set(), sdb_querys(), sdb_save_dwarf_function(), sdb_set_enc(), sdb_unset(), serialize_checkpoints(), serialize_memory_cb(), serialize_register_cb(), store_xrefs_list_cb(), v1_v2_types_foreach_cb(), and v2_v3_types_foreach_cb().

sdb_set_internal()

static ut32 sdb_set_internal ( Sdb *  s, const char *  key, char *  val, int  owned, ut32  cas  )

static

Definition at line 532 of file sdb.c.

                                                                                      {
    ut32 vlen, klen;
    SdbKv *kv;
    bool found;
    if (!s || !key) {
        return 0;
    }
    if (!val) {
        if (owned) {
            val = strdup("");
        } else {
            val = "";
        }
    }
    // XXX strlen computed twice.. because of check_*()
    klen = strlen(key);
    vlen = strlen(val);
    if (s->journal != -1) {
        sdb_journal_log(s, key, val);
    }
    cdb_findstart(&s->db);
    kv = sdb_ht_find_kvp(s->ht, key, &found);
    if (found && sdbkv_value(kv)) {
        if (cdb_findnext(&s->db, sdb_hash(key), key, klen)) {
            if (cas && kv->cas != cas) {
                if (owned) {
                    free(val);
                }
                return 0;
            }
            if (vlen == sdbkv_value_len(kv) && !strcmp(sdbkv_value(kv), val)) {
                sdb_hook_call(s, key, val);
                return kv->cas;
            }
            kv->cas = cas = nextcas();
            if (owned) {
                kv->base.value_len = vlen;
                free(kv->base.value);
                kv->base.value = val; // owned
            } else {
                if ((ut32)vlen > kv->base.value_len) {
                    free(kv->base.value);
                    kv->base.value = malloc(vlen + 1);
                }
                memcpy(kv->base.value, val, vlen + 1);
                kv->base.value_len = vlen;
            }
        } else {
            sdb_ht_delete(s->ht, key);
        }
        sdb_hook_call(s, key, val);
        return cas;
    }
    // empty values are also stored
    // TODO store only the ones that are in the CDB
    if (owned) {
        kv = sdbkv_new2(key, klen, NULL, 0);
        if (kv) {
            kv->base.value = val;
            kv->base.value_len = vlen;
        }
    } else {
        kv = sdbkv_new2(key, klen, val, vlen);
    }
    if (kv) {
        ut32 cas = kv->cas = nextcas();
        sdb_ht_insert_kvp(s->ht, kv, true /*update*/);
        free(kv);
        sdb_hook_call(s, key, val);
        return cas;
    }
    // kv set failed, no need to callback   sdb_hook_call (s, key, val);
    return 0;
}

References sdb_kv::base, sdb_kv::cas, cdb_findnext(), cdb_findstart(), found, free(), key, malloc(), memcpy(), nextcas(), NULL, s, sdb_hash(), sdb_hook_call(), sdb_ht_delete(), sdb_ht_find_kvp(), sdb_ht_insert_kvp(), sdb_journal_log(), sdbkv_new2(), sdbkv_value(), sdbkv_value_len(), strdup(), and val.

Referenced by sdb_set(), and sdb_set_owned().

sdb_set_owned()

RZ_API int sdb_set_owned	(	Sdb *	s,
		const char *	key,
		char *	val,
		ut32	cas
	)

Definition at line 607 of file sdb.c.

                                                                       {
    return sdb_set_internal(s, key, val, 1, cas);
}

References key, s, sdb_set_internal(), and val.

Referenced by bin_pe_init_hdr(), get_vernaux_entry_sdb(), get_versym_entry_sdb_from_verdef(), get_versym_entry_sdb_from_verneed(), rz_agraph_add_node_with_color(), sdb_array_add_sorted(), sdb_array_append(), sdb_array_delete(), sdb_array_insert(), sdb_array_pop_tail(), sdb_array_prepend(), sdb_array_set(), sdb_array_sort(), sdb_array_sort_num(), sdb_concat(), sdb_expire_set(), sdb_querys(), and sdb_uncat().

sdb_stats()

RZ_API bool sdb_stats	(	Sdb *	s,
		ut32 *	disk,
		ut32 *	mem
	)

Definition at line 856 of file sdb.c.

                                                     {
    if (!s) {
        return false;
    }
    if (disk) {
        ut32 count = 0;
        if (s->fd != -1) {
            sdb_dump_begin(s);
            while (sdb_dump_hasnext(s)) {
                count++;
            }
        }
        *disk = count;
    }
    if (mem) {
        *mem = s->ht->count;
    }
    return disk || mem;
}

References count, rz_socket_t::fd, mem, s, sdb_dump_begin(), and sdb_dump_hasnext().

Referenced by showcount().

sdb_sync()

RZ_API bool sdb_sync

(

Sdb *

s

)

Definition at line 803 of file sdb.c.

                             {
    bool result;
    ut32 i;
 
    if (!s || !sdb_disk_create(s)) {
        return false;
    }
    result = sdb_foreach_cdb(s, _insert_into_disk, _remove_afer_insert, s);
    if (!result) {
        return false;
    }
 
    /* append new keyvalues */
    for (i = 0; i < s->ht->size; ++i) {
        HtPPBucket *bt = &s->ht->table[i];
        SdbKv *kv;
        ut32 j, count;
 
        BUCKET_FOREACH_SAFE(s->ht, bt, j, count, kv) {
            if (sdbkv_key(kv) && sdbkv_value(kv) && *sdbkv_value(kv) && !kv->expire) {
                if (sdb_disk_insert(s, sdbkv_key(kv), sdbkv_value(kv))) {
                    sdb_remove(s, sdbkv_key(kv), 0);
                }
            }
        }
    }
    sdb_disk_finish(s);
    sdb_journal_clear(s);
    // TODO: sdb_reset memory state?
    return true;
}

References _insert_into_disk(), _remove_afer_insert(), BUCKET_FOREACH_SAFE, count, sdb_kv::expire, i, s, sdb_disk_create(), sdb_disk_finish(), sdb_disk_insert(), sdb_foreach_cdb(), sdb_journal_clear(), sdb_remove(), sdbkv_key(), and sdbkv_value().

Referenced by config_path(), createdb(), ns_sync(), rz_cmd_kuery(), rz_core_sysenv_begin(), session_sdb_save(), and terminate().

sdb_uncat()

RZ_API int sdb_uncat	(	Sdb *	s,
		const char *	key,
		const char *	value,
		ut32	cas
	)

Definition at line 304 of file sdb.c.

                                                                           {
    // remove 'value' from current key value.
    // TODO: cas is ignored here
    int vlen = 0, valen;
    char *p, *v = sdb_get_len(s, key, &vlen, NULL);
    int mod = 0;
    if (!v || !key || !value) {
        free(v);
        return 0;
    }
    valen = strlen(value);
    if (valen > 0) {
        while ((p = strstr(v, value))) {
            memmove(p, p + valen, strlen(p + valen) + 1);
            mod = 1;
        }
    }
    if (mod) {
        sdb_set_owned(s, key, v, 0);
    } else {
        free(v);
    }
    return 0;
}

References free(), key, mod(), NULL, p, s, sdb_get_len(), sdb_set_owned(), v, and value.

Referenced by sdb_querys().

sdb_unhook()

RZ_API bool sdb_unhook	(	Sdb *	s,
		SdbHook	h
	)

Definition at line 1003 of file sdb.c.

                                          {
    int i = 0;
    SdbHook hook;
    SdbListIter *iter, *iter2;
    ls_foreach (s->hooks, iter, hook) {
        if (!(i % 2) && (hook == h)) {
            iter2 = iter->n;
            ls_delete(s->hooks, iter);
            ls_delete(s->hooks, iter2);
            return true;
        }
        i++;
    }
    return false;
}

References h, i, ls_delete(), ls_foreach, and s.

sdb_unlink()

RZ_API bool sdb_unlink

(

Sdb *

s

)

Definition at line 1064 of file sdb.c.

                               {
    sdb_fini(s, 1);
    return sdb_disk_unlink(s);
}

References s, sdb_disk_unlink(), and sdb_fini().

sdb_unset()

RZ_API int sdb_unset	(	Sdb *	s,
		const char *	key,
		ut32	cas
	)

Definition at line 294 of file sdb.c.

                                                        {
    return key ? sdb_set(s, key, "", cas) : 0;
}

References key, s, and sdb_set().

Referenced by rz_analysis_cc_del(), rz_analysis_noreturn_drop(), rz_core_analysis_esil_init_mem_del(), rz_project_migrate_v1_v2(), rz_project_migrate_v2_v3(), rz_project_migrate_v3_v4(), rz_project_migrate_v7_v8(), sdb_array_pop_head(), sdb_const_get_len(), and unset_cb().

sdb_unset_like()

RZ_API int sdb_unset_like	(	Sdb *	s,
		const char *	k
	)

Definition at line 1106 of file sdb.c.

                                                 {
    UnsetCallbackData ucd = { s, k };
    return sdb_foreach(s, unset_cb, &ucd);
}

References k, s, sdb_foreach(), and unset_cb().

Referenced by sdb_querys().

sdbkv_free()

RZ_API void sdbkv_free

(

SdbKv *

kv

)

Definition at line 524 of file sdb.c.

                                  {
    if (kv) {
        free(sdbkv_key(kv));
        free(sdbkv_value(kv));
        RZ_FREE(kv);
    }
}

References free(), RZ_FREE, sdbkv_key(), and sdbkv_value().

Referenced by sdb_foreach_list(), sdb_foreach_list_filter_cb(), sdb_foreach_list_filter_user(), and sdb_foreach_match().

sdbkv_match()

RZ_API bool sdbkv_match	(	SdbKv *	kv,
		const char *	expr
	)

Definition at line 472 of file sdb.c.

                                                     {
    // TODO: add syntax to negate condition
    // TODO: add syntax to OR k/v instead of AND
    // [^]str[$]=[^]str[$]
    const char *eq = strchr(expr, '=');
    if (eq) {
        char *e = strdup(expr);
        char *ep = e + (eq - expr);
        *ep++ = 0;
        bool res = !*e || match(sdbkv_key(kv), e);
        bool res2 = !*ep || match(sdbkv_value(kv), ep);
        free(e);
        return res && res2;
    }
    return match(sdbkv_key(kv), expr);
}

References e, eq, expr(), free(), match(), sdbkv_key(), sdbkv_value(), and strdup().

Referenced by sdb_foreach_match_cb().

sdbkv_new()

RZ_API SdbKv* sdbkv_new	(	const char *	k,
		const char *	v
	)

Definition at line 489 of file sdb.c.

                                                      {
    return sdbkv_new2(k, strlen(k), v, strlen(v));
}

References k, sdbkv_new2(), and v.

sdbkv_new2()

RZ_API SdbKv* sdbkv_new2	(	const char *	k,
		int	kl,
		const char *	v,
		int	vl
	)

Definition at line 493 of file sdb.c.

                                                                       {
    SdbKv *kv;
    if (!v) {
        vl = 0;
    }
    kv = RZ_NEW0(SdbKv);
    kv->base.key_len = kl;
    kv->base.key = malloc(kv->base.key_len + 1);
    if (!kv->base.key) {
        free(kv);
        return NULL;
    }
    memcpy(kv->base.key, k, kv->base.key_len + 1);
    kv->base.value_len = vl;
    if (vl) {
        kv->base.value = malloc(vl + 1);
        if (!kv->base.value) {
            free(kv->base.key);
            free(kv);
            return NULL;
        }
        memcpy(kv->base.value, v, vl + 1);
    } else {
        kv->base.value = NULL;
        kv->base.value_len = 0;
    }
    kv->cas = nextcas();
    kv->expire = 0LL;
    return kv;
}

References sdb_kv::base, sdb_kv::cas, sdb_kv::expire, free(), k, malloc(), memcpy(), nextcas(), NULL, RZ_NEW0, and v.

Referenced by sdb_set_internal(), and sdbkv_new().

unset_cb()

static bool unset_cb ( void *  user, const char *  k, const char *  v  )

static

Definition at line 1098 of file sdb.c.

                                                               {
    UnsetCallbackData *ucd = user;
    if (sdb_match(k, ucd->key)) {
        sdb_unset(ucd->sdb, k, 0);
    }
    return true;
}

References k, UnsetCallbackData::key, UnsetCallbackData::sdb, sdb_match(), and sdb_unset().

Referenced by sdb_unset_like().

Variable Documentation

global_hook

SdbHook global_hook = NULL

static

Definition at line 34 of file sdb.c.

Referenced by sdb_global_hook(), and sdb_new().

global_user

void* global_user = NULL

static

Definition at line 35 of file sdb.c.

Referenced by sdb_global_hook(), and sdb_new().