
[Code reviewed by Love Hörnquist Åstrand <lha@kth.se>] Added heim_db_*() entry points for dealing with databases, and make krb5_aname_to_localname() use it. The following enhancements to libheimbase are included: - Add heim_data_t and heim_string_t "reference" variants to avoid memory copies of potentially large data/strings. See heim_data_ref_create() and heim_string_ref_create(). - Added enhancements to heim_array_t to allow their use for queues and stacks, and to improve performance. See heim_array_insert_value(). - Added XPath-like accessors for heim_object_t. See heim_path_get(), heim_path_copy(), heim_path_create(), and heim_path_delete(). These are used extensively in the DB framework's generic composition of ACID support and in the test_base program - Made libheimbase more consistent with Core Foundation naming conventions. See heim_{dict, array}_{get, copy}_value() and heim_path_{get, copy}(). - Added functionality to and fixed bugs in base/json.c: - heim_serialize(); - depth limit for JSON parsing (for DoS protection); - pretty-printing; - JSON compliance (see below); - flag options for parsing and serializing; these are needed because of impedance mismatches between heim_object_t and JSON (e.g., heim_dict_t allows non-string keys, but JSON does not; heimbase supports binary data, while JSON does not). - Added heim_error_enomem(). - Enhanced the test_base program to test new functionality and to use heim_path*() to better test JSON encoding. This includes some fuzz testing of JSON parsing, and running the test under valgrind. - Started to add doxygen documentation for libheimbase (but doc build for libheimbase is still incomplete). Note that there's still some incomplete JSON support: - JSON string quoting is not fully implemented; - libheimbase lacks support for real numbers, while JSON has it -- otherwise libheimbase is a superset of JSON, specifically in that any heim_object_t can be a key for an associative array. The following DB backends are supported natively: - "sorted-text", a binary search of sorted (in C locale), flat text files; - "json", a backend that stores DB contents serialized as JSON (this is intended for configuration-like contents). The DB framework supports: - multiple key/value tables per-DB - ACID transactions The DB framework also natively implements ACID transactions for any DB backends that a) do not provide transactions natively, b) do provide lock/unlock/sync methods (even on Windows). This includes autocommit of DB updates outside transactions. Future DB enhancements may include: - add backends for various DB types (BDB, CDB, MDB, ...); - make libhdb use heim_db_t; - add a command-line tool for interfacing to databases via libheimbase (e.g., to get/set/delete values, create/copy/ backup DBs, inspect history, check integrity); - framework-level transaction logging (with redo and undo logging), for generic incremental replication; - framework-level DB integrity checking. We could store a MAC of the XOR of a hash function applied to {key, value} for every entry in the DB, then use this to check DB integrity incrementally during incremental replication, as well as for the whole DB.
964 lines
21 KiB
C
964 lines
21 KiB
C
/*
|
|
* Copyright (c) 2010 Kungliga Tekniska Högskolan
|
|
* (Royal Institute of Technology, Stockholm, Sweden).
|
|
* All rights reserved.
|
|
*
|
|
* Portions Copyright (c) 2010 Apple Inc. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
*
|
|
* 1. Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
*
|
|
* 2. Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
*
|
|
* 3. Neither the name of the Institute nor the names of its contributors
|
|
* may be used to endorse or promote products derived from this software
|
|
* without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
|
* SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "baselocl.h"
|
|
#include <syslog.h>
|
|
|
|
static heim_base_atomic_type tidglobal = HEIM_TID_USER;
|
|
|
|
struct heim_base {
|
|
heim_type_t isa;
|
|
heim_base_atomic_type ref_cnt;
|
|
HEIM_TAILQ_ENTRY(heim_base) autorel;
|
|
heim_auto_release_t autorelpool;
|
|
uintptr_t isaextra[3];
|
|
};
|
|
|
|
/* specialized version of base */
|
|
struct heim_base_mem {
|
|
heim_type_t isa;
|
|
heim_base_atomic_type ref_cnt;
|
|
HEIM_TAILQ_ENTRY(heim_base) autorel;
|
|
heim_auto_release_t autorelpool;
|
|
const char *name;
|
|
void (*dealloc)(void *);
|
|
uintptr_t isaextra[1];
|
|
};
|
|
|
|
#define PTR2BASE(ptr) (((struct heim_base *)ptr) - 1)
|
|
#define BASE2PTR(ptr) ((void *)(((struct heim_base *)ptr) + 1))
|
|
|
|
#ifdef HEIM_BASE_NEED_ATOMIC_MUTEX
|
|
HEIMDAL_MUTEX _heim_base_mutex = HEIMDAL_MUTEX_INITIALIZER;
|
|
#endif
|
|
|
|
/*
|
|
* Auto release structure
|
|
*/
|
|
|
|
struct heim_auto_release {
|
|
HEIM_TAILQ_HEAD(, heim_base) pool;
|
|
HEIMDAL_MUTEX pool_mutex;
|
|
struct heim_auto_release *parent;
|
|
};
|
|
|
|
|
|
/**
|
|
* Retain object (i.e., take a reference)
|
|
*
|
|
* @param object to be released, NULL is ok
|
|
*
|
|
* @return the same object as passed in
|
|
*/
|
|
|
|
void *
|
|
heim_retain(void *ptr)
|
|
{
|
|
struct heim_base *p = PTR2BASE(ptr);
|
|
|
|
if (ptr == NULL || heim_base_is_tagged(ptr))
|
|
return ptr;
|
|
|
|
if (p->ref_cnt == heim_base_atomic_max)
|
|
return ptr;
|
|
|
|
if ((heim_base_atomic_inc(&p->ref_cnt) - 1) == 0)
|
|
heim_abort("resurection");
|
|
return ptr;
|
|
}
|
|
|
|
/**
|
|
* Release object, free if reference count reaches zero
|
|
*
|
|
* @param object to be released
|
|
*/
|
|
|
|
void
|
|
heim_release(void *ptr)
|
|
{
|
|
heim_base_atomic_type old;
|
|
struct heim_base *p = PTR2BASE(ptr);
|
|
|
|
if (ptr == NULL || heim_base_is_tagged(ptr))
|
|
return;
|
|
|
|
if (p->ref_cnt == heim_base_atomic_max)
|
|
return;
|
|
|
|
old = heim_base_atomic_dec(&p->ref_cnt) + 1;
|
|
|
|
if (old > 1)
|
|
return;
|
|
|
|
if (old == 1) {
|
|
heim_auto_release_t ar = p->autorelpool;
|
|
/* remove from autorel pool list */
|
|
if (ar) {
|
|
p->autorelpool = NULL;
|
|
HEIMDAL_MUTEX_lock(&ar->pool_mutex);
|
|
HEIM_TAILQ_REMOVE(&ar->pool, p, autorel);
|
|
HEIMDAL_MUTEX_unlock(&ar->pool_mutex);
|
|
}
|
|
if (p->isa->dealloc)
|
|
p->isa->dealloc(ptr);
|
|
free(p);
|
|
} else
|
|
heim_abort("over release");
|
|
}
|
|
|
|
void
|
|
_heim_make_permanent(heim_object_t ptr)
|
|
{
|
|
struct heim_base *p = PTR2BASE(ptr);
|
|
p->ref_cnt = heim_base_atomic_max;
|
|
}
|
|
|
|
|
|
static heim_type_t tagged_isa[9] = {
|
|
&_heim_number_object,
|
|
&_heim_null_object,
|
|
&_heim_bool_object,
|
|
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
heim_type_t
|
|
_heim_get_isa(heim_object_t ptr)
|
|
{
|
|
struct heim_base *p;
|
|
if (heim_base_is_tagged(ptr)) {
|
|
if (heim_base_is_tagged_object(ptr))
|
|
return tagged_isa[heim_base_tagged_object_tid(ptr)];
|
|
heim_abort("not a supported tagged type");
|
|
}
|
|
p = PTR2BASE(ptr);
|
|
return p->isa;
|
|
}
|
|
|
|
/**
|
|
* Get type ID of object
|
|
*
|
|
* @param object object to get type id of
|
|
*
|
|
* @return type id of object
|
|
*/
|
|
|
|
heim_tid_t
|
|
heim_get_tid(heim_object_t ptr)
|
|
{
|
|
heim_type_t isa = _heim_get_isa(ptr);
|
|
return isa->tid;
|
|
}
|
|
|
|
/**
|
|
* Get hash value of object
|
|
*
|
|
* @param object object to get hash value for
|
|
*
|
|
* @return a hash value
|
|
*/
|
|
|
|
unsigned long
|
|
heim_get_hash(heim_object_t ptr)
|
|
{
|
|
heim_type_t isa = _heim_get_isa(ptr);
|
|
if (isa->hash)
|
|
return isa->hash(ptr);
|
|
return (unsigned long)ptr;
|
|
}
|
|
|
|
/**
|
|
* Compare two objects, returns 0 if equal, can use used for qsort()
|
|
* and friends.
|
|
*
|
|
* @param a first object to compare
|
|
* @param b first object to compare
|
|
*
|
|
* @return 0 if objects are equal
|
|
*/
|
|
|
|
int
|
|
heim_cmp(heim_object_t a, heim_object_t b)
|
|
{
|
|
heim_tid_t ta, tb;
|
|
heim_type_t isa;
|
|
|
|
ta = heim_get_tid(a);
|
|
tb = heim_get_tid(b);
|
|
|
|
if (ta != tb)
|
|
return ta - tb;
|
|
|
|
isa = _heim_get_isa(a);
|
|
|
|
if (isa->cmp)
|
|
return isa->cmp(a, b);
|
|
|
|
return (uintptr_t)a - (uintptr_t)b;
|
|
}
|
|
|
|
/*
|
|
* Private - allocates an memory object
|
|
*/
|
|
|
|
static void
|
|
memory_dealloc(void *ptr)
|
|
{
|
|
struct heim_base_mem *p = (struct heim_base_mem *)PTR2BASE(ptr);
|
|
if (p->dealloc)
|
|
p->dealloc(ptr);
|
|
}
|
|
|
|
struct heim_type_data memory_object = {
|
|
HEIM_TID_MEMORY,
|
|
"memory-object",
|
|
NULL,
|
|
memory_dealloc,
|
|
NULL,
|
|
NULL,
|
|
NULL
|
|
};
|
|
|
|
/**
|
|
* Allocate memory for an object of anonymous type
|
|
*
|
|
* @param size size of object to be allocated
|
|
* @param name name of ad-hoc type
|
|
* @param dealloc destructor function
|
|
*
|
|
* Objects allocated with this interface do not serialize.
|
|
*
|
|
* @return allocated object
|
|
*/
|
|
|
|
void *
|
|
heim_alloc(size_t size, const char *name, heim_type_dealloc dealloc)
|
|
{
|
|
/* XXX use posix_memalign */
|
|
|
|
struct heim_base_mem *p = calloc(1, size + sizeof(*p));
|
|
if (p == NULL)
|
|
return NULL;
|
|
p->isa = &memory_object;
|
|
p->ref_cnt = 1;
|
|
p->name = name;
|
|
p->dealloc = dealloc;
|
|
return BASE2PTR(p);
|
|
}
|
|
|
|
heim_type_t
|
|
_heim_create_type(const char *name,
|
|
heim_type_init init,
|
|
heim_type_dealloc dealloc,
|
|
heim_type_copy copy,
|
|
heim_type_cmp cmp,
|
|
heim_type_hash hash)
|
|
{
|
|
heim_type_t type;
|
|
|
|
type = calloc(1, sizeof(*type));
|
|
if (type == NULL)
|
|
return NULL;
|
|
|
|
type->tid = heim_base_atomic_inc(&tidglobal);
|
|
type->name = name;
|
|
type->init = init;
|
|
type->dealloc = dealloc;
|
|
type->copy = copy;
|
|
type->cmp = cmp;
|
|
type->hash = hash;
|
|
|
|
return type;
|
|
}
|
|
|
|
heim_object_t
|
|
_heim_alloc_object(heim_type_t type, size_t size)
|
|
{
|
|
/* XXX should use posix_memalign */
|
|
struct heim_base *p = calloc(1, size + sizeof(*p));
|
|
if (p == NULL)
|
|
return NULL;
|
|
p->isa = type;
|
|
p->ref_cnt = 1;
|
|
|
|
return BASE2PTR(p);
|
|
}
|
|
|
|
void *
|
|
_heim_get_isaextra(heim_object_t ptr, size_t idx)
|
|
{
|
|
struct heim_base *p = (struct heim_base *)PTR2BASE(ptr);
|
|
|
|
heim_assert(ptr != NULL, "internal error");
|
|
if (p->isa == &memory_object)
|
|
return NULL;
|
|
heim_assert(idx < 3, "invalid private heim_base extra data index");
|
|
return &p->isaextra[idx];
|
|
}
|
|
|
|
heim_tid_t
|
|
_heim_type_get_tid(heim_type_t type)
|
|
{
|
|
return type->tid;
|
|
}
|
|
|
|
/**
|
|
* Call func once and only once
|
|
*
|
|
* @param once pointer to a heim_base_once_t
|
|
* @param ctx context passed to func
|
|
* @param func function to be called
|
|
*/
|
|
|
|
void
|
|
heim_base_once_f(heim_base_once_t *once, void *ctx, void (*func)(void *))
|
|
{
|
|
#ifdef HAVE_DISPATCH_DISPATCH_H
|
|
dispatch_once_f(once, ctx, func);
|
|
#else
|
|
static HEIMDAL_MUTEX mutex = HEIMDAL_MUTEX_INITIALIZER;
|
|
HEIMDAL_MUTEX_lock(&mutex);
|
|
if (*once == 0) {
|
|
*once = 1;
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
func(ctx);
|
|
HEIMDAL_MUTEX_lock(&mutex);
|
|
*once = 2;
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
} else if (*once == 2) {
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
} else {
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
while (1) {
|
|
struct timeval tv = { 0, 1000 };
|
|
select(0, NULL, NULL, NULL, &tv);
|
|
HEIMDAL_MUTEX_lock(&mutex);
|
|
if (*once == 2)
|
|
break;
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
}
|
|
HEIMDAL_MUTEX_unlock(&mutex);
|
|
}
|
|
#endif
|
|
}
|
|
|
|
/**
|
|
* Abort and log the failure (using syslog)
|
|
*/
|
|
|
|
void
|
|
heim_abort(const char *fmt, ...)
|
|
{
|
|
va_list ap;
|
|
va_start(ap, fmt);
|
|
heim_abortv(fmt, ap);
|
|
va_end(ap);
|
|
}
|
|
|
|
/**
|
|
* Abort and log the failure (using syslog)
|
|
*/
|
|
|
|
void
|
|
heim_abortv(const char *fmt, va_list ap)
|
|
{
|
|
static char str[1024];
|
|
|
|
vsnprintf(str, sizeof(str), fmt, ap);
|
|
syslog(LOG_ERR, "heim_abort: %s", str);
|
|
abort();
|
|
}
|
|
|
|
/*
|
|
*
|
|
*/
|
|
|
|
static int ar_created = 0;
|
|
static HEIMDAL_thread_key ar_key;
|
|
|
|
struct ar_tls {
|
|
struct heim_auto_release *head;
|
|
struct heim_auto_release *current;
|
|
HEIMDAL_MUTEX tls_mutex;
|
|
};
|
|
|
|
static void
|
|
ar_tls_delete(void *ptr)
|
|
{
|
|
struct ar_tls *tls = ptr;
|
|
if (tls->head)
|
|
heim_release(tls->head);
|
|
free(tls);
|
|
}
|
|
|
|
static void
|
|
init_ar_tls(void *ptr)
|
|
{
|
|
int ret;
|
|
HEIMDAL_key_create(&ar_key, ar_tls_delete, ret);
|
|
if (ret == 0)
|
|
ar_created = 1;
|
|
}
|
|
|
|
static struct ar_tls *
|
|
autorel_tls(void)
|
|
{
|
|
static heim_base_once_t once = HEIM_BASE_ONCE_INIT;
|
|
struct ar_tls *arp;
|
|
int ret;
|
|
|
|
heim_base_once_f(&once, NULL, init_ar_tls);
|
|
if (!ar_created)
|
|
return NULL;
|
|
|
|
arp = HEIMDAL_getspecific(ar_key);
|
|
if (arp == NULL) {
|
|
|
|
arp = calloc(1, sizeof(*arp));
|
|
if (arp == NULL)
|
|
return NULL;
|
|
HEIMDAL_setspecific(ar_key, arp, ret);
|
|
if (ret) {
|
|
free(arp);
|
|
return NULL;
|
|
}
|
|
}
|
|
return arp;
|
|
|
|
}
|
|
|
|
static void
|
|
autorel_dealloc(void *ptr)
|
|
{
|
|
heim_auto_release_t ar = ptr;
|
|
struct ar_tls *tls;
|
|
|
|
tls = autorel_tls();
|
|
if (tls == NULL)
|
|
heim_abort("autorelease pool released on thread w/o autorelease inited");
|
|
|
|
heim_auto_release_drain(ar);
|
|
|
|
if (!HEIM_TAILQ_EMPTY(&ar->pool))
|
|
heim_abort("pool not empty after draining");
|
|
|
|
HEIMDAL_MUTEX_lock(&tls->tls_mutex);
|
|
if (tls->current != ptr)
|
|
heim_abort("autorelease not releaseing top pool");
|
|
|
|
if (tls->current != tls->head)
|
|
tls->current = ar->parent;
|
|
HEIMDAL_MUTEX_unlock(&tls->tls_mutex);
|
|
}
|
|
|
|
static int
|
|
autorel_cmp(void *a, void *b)
|
|
{
|
|
return (a == b);
|
|
}
|
|
|
|
static unsigned long
|
|
autorel_hash(void *ptr)
|
|
{
|
|
return (unsigned long)ptr;
|
|
}
|
|
|
|
|
|
static struct heim_type_data _heim_autorel_object = {
|
|
HEIM_TID_AUTORELEASE,
|
|
"autorelease-pool",
|
|
NULL,
|
|
autorel_dealloc,
|
|
NULL,
|
|
autorel_cmp,
|
|
autorel_hash
|
|
};
|
|
|
|
/**
|
|
* Create thread-specific object auto-release pool
|
|
*
|
|
* Objects placed on the per-thread auto-release pool (with
|
|
* heim_auto_release()) can be released in one fell swoop by calling
|
|
* heim_auto_release_drain().
|
|
*/
|
|
|
|
heim_auto_release_t
|
|
heim_auto_release_create(void)
|
|
{
|
|
struct ar_tls *tls = autorel_tls();
|
|
heim_auto_release_t ar;
|
|
|
|
if (tls == NULL)
|
|
heim_abort("Failed to create/get autorelease head");
|
|
|
|
ar = _heim_alloc_object(&_heim_autorel_object, sizeof(struct heim_auto_release));
|
|
if (ar) {
|
|
HEIMDAL_MUTEX_lock(&tls->tls_mutex);
|
|
if (tls->head == NULL)
|
|
tls->head = ar;
|
|
ar->parent = tls->current;
|
|
tls->current = ar;
|
|
HEIMDAL_MUTEX_unlock(&tls->tls_mutex);
|
|
}
|
|
|
|
return ar;
|
|
}
|
|
|
|
/**
|
|
* Place the current object on the thread's auto-release pool
|
|
*
|
|
* @param ptr object
|
|
*/
|
|
|
|
void
|
|
heim_auto_release(heim_object_t ptr)
|
|
{
|
|
struct heim_base *p = PTR2BASE(ptr);
|
|
struct ar_tls *tls = autorel_tls();
|
|
heim_auto_release_t ar;
|
|
|
|
if (ptr == NULL || heim_base_is_tagged(ptr))
|
|
return;
|
|
|
|
/* drop from old pool */
|
|
if ((ar = p->autorelpool) != NULL) {
|
|
HEIMDAL_MUTEX_lock(&ar->pool_mutex);
|
|
HEIM_TAILQ_REMOVE(&ar->pool, p, autorel);
|
|
p->autorelpool = NULL;
|
|
HEIMDAL_MUTEX_unlock(&ar->pool_mutex);
|
|
}
|
|
|
|
if (tls == NULL || (ar = tls->current) == NULL)
|
|
heim_abort("no auto relase pool in place, would leak");
|
|
|
|
HEIMDAL_MUTEX_lock(&ar->pool_mutex);
|
|
HEIM_TAILQ_INSERT_HEAD(&ar->pool, p, autorel);
|
|
p->autorelpool = ar;
|
|
HEIMDAL_MUTEX_unlock(&ar->pool_mutex);
|
|
}
|
|
|
|
/**
|
|
* Release all objects on the given auto-release pool
|
|
*/
|
|
|
|
void
|
|
heim_auto_release_drain(heim_auto_release_t autorel)
|
|
{
|
|
heim_object_t obj;
|
|
|
|
/* release all elements on the tail queue */
|
|
|
|
HEIMDAL_MUTEX_lock(&autorel->pool_mutex);
|
|
while(!HEIM_TAILQ_EMPTY(&autorel->pool)) {
|
|
obj = HEIM_TAILQ_FIRST(&autorel->pool);
|
|
HEIMDAL_MUTEX_unlock(&autorel->pool_mutex);
|
|
heim_release(BASE2PTR(obj));
|
|
HEIMDAL_MUTEX_lock(&autorel->pool_mutex);
|
|
}
|
|
HEIMDAL_MUTEX_unlock(&autorel->pool_mutex);
|
|
}
|
|
|
|
/*
|
|
* Helper for heim_path_vget() and heim_path_delete(). On success
|
|
* outputs the node named by the path and the parent node and key
|
|
* (useful for heim_path_delete()).
|
|
*/
|
|
|
|
static heim_object_t
|
|
heim_path_vget2(heim_object_t ptr, heim_object_t *parent, heim_object_t *key,
|
|
heim_error_t *error, va_list ap)
|
|
{
|
|
heim_object_t path_element;
|
|
heim_object_t node, next_node;
|
|
heim_tid_t node_type;
|
|
|
|
*parent = NULL;
|
|
*key = NULL;
|
|
if (ptr == NULL)
|
|
return NULL;
|
|
|
|
for (node = ptr; node != NULL; ) {
|
|
path_element = va_arg(ap, heim_object_t);
|
|
if (path_element == NULL) {
|
|
*parent = node;
|
|
*key = path_element;
|
|
return node;
|
|
}
|
|
|
|
node_type = heim_get_tid(node);
|
|
switch (node_type) {
|
|
case HEIM_TID_ARRAY:
|
|
case HEIM_TID_DICT:
|
|
case HEIM_TID_DB:
|
|
break;
|
|
default:
|
|
if (node == ptr)
|
|
heim_abort("heim_path_get() only operates on container types");
|
|
return NULL;
|
|
}
|
|
|
|
if (node_type == HEIM_TID_DICT) {
|
|
next_node = heim_dict_get_value(node, path_element);
|
|
} else if (node_type == HEIM_TID_DB) {
|
|
next_node = _heim_db_get_value(node, NULL, path_element, NULL);
|
|
} else if (node_type == HEIM_TID_ARRAY) {
|
|
int idx = -1;
|
|
|
|
if (heim_get_tid(path_element) == HEIM_TID_NUMBER)
|
|
idx = heim_number_get_int(path_element);
|
|
if (idx < 0) {
|
|
if (error)
|
|
*error = heim_error_create(EINVAL,
|
|
"heim_path_get() path elements "
|
|
"for array nodes must be "
|
|
"numeric and positive");
|
|
return NULL;
|
|
}
|
|
next_node = heim_array_get_value(node, idx);
|
|
} else {
|
|
if (error)
|
|
*error = heim_error_create(EINVAL,
|
|
"heim_path_get() node in path "
|
|
"not a container type");
|
|
return NULL;
|
|
}
|
|
node = next_node;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
/**
|
|
* Get a node in a heim_object tree by path
|
|
*
|
|
* @param ptr tree
|
|
* @param error error (output)
|
|
* @param ap NULL-terminated va_list of heim_object_ts that form a path
|
|
*
|
|
* @return object (not retained) if found
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
heim_object_t
|
|
heim_path_vget(heim_object_t ptr, heim_error_t *error, va_list ap)
|
|
{
|
|
heim_object_t p, k;
|
|
|
|
return heim_path_vget2(ptr, &p, &k, error, ap);
|
|
}
|
|
|
|
/**
|
|
* Get a node in a tree by path, with retained reference
|
|
*
|
|
* @param ptr tree
|
|
* @param error error (output)
|
|
* @param ap NULL-terminated va_list of heim_object_ts that form a path
|
|
*
|
|
* @return retained object if found
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
heim_object_t
|
|
heim_path_vcopy(heim_object_t ptr, heim_error_t *error, va_list ap)
|
|
{
|
|
heim_object_t p, k;
|
|
|
|
return heim_retain(heim_path_vget2(ptr, &p, &k, error, ap));
|
|
}
|
|
|
|
/**
|
|
* Get a node in a tree by path
|
|
*
|
|
* @param ptr tree
|
|
* @param error error (output)
|
|
* @param ... NULL-terminated va_list of heim_object_ts that form a path
|
|
*
|
|
* @return object (not retained) if found
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
heim_object_t
|
|
heim_path_get(heim_object_t ptr, heim_error_t *error, ...)
|
|
{
|
|
heim_object_t o;
|
|
heim_object_t p, k;
|
|
va_list ap;
|
|
|
|
if (ptr == NULL)
|
|
return NULL;
|
|
|
|
va_start(ap, error);
|
|
o = heim_path_vget2(ptr, &p, &k, error, ap);
|
|
va_end(ap);
|
|
return o;
|
|
}
|
|
|
|
/**
|
|
* Get a node in a tree by path, with retained reference
|
|
*
|
|
* @param ptr tree
|
|
* @param error error (output)
|
|
* @param ... NULL-terminated va_list of heim_object_ts that form a path
|
|
*
|
|
* @return retained object if found
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
heim_object_t
|
|
heim_path_copy(heim_object_t ptr, heim_error_t *error, ...)
|
|
{
|
|
heim_object_t o;
|
|
heim_object_t p, k;
|
|
va_list ap;
|
|
|
|
if (ptr == NULL)
|
|
return NULL;
|
|
|
|
va_start(ap, error);
|
|
o = heim_retain(heim_path_vget2(ptr, &p, &k, error, ap));
|
|
va_end(ap);
|
|
return o;
|
|
}
|
|
|
|
/**
|
|
* Create a path in a heim_object_t tree
|
|
*
|
|
* @param ptr the tree
|
|
* @param size the size of the heim_dict_t nodes to be created
|
|
* @param leaf leaf node to be added, if any
|
|
* @param error error (output)
|
|
* @param ap NULL-terminated of path component objects
|
|
*
|
|
* Create a path of heim_dict_t interior nodes in a given heim_object_t
|
|
* tree, as necessary, and set/replace a leaf, if given (if leaf is NULL
|
|
* then the leaf is not deleted).
|
|
*
|
|
* @return 0 on success, else a system error
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
int
|
|
heim_path_vcreate(heim_object_t ptr, size_t size, heim_object_t leaf,
|
|
heim_error_t *error, va_list ap)
|
|
{
|
|
heim_object_t path_element = va_arg(ap, heim_object_t);
|
|
heim_object_t next_path_element = NULL;
|
|
heim_object_t node = ptr;
|
|
heim_object_t next_node = NULL;
|
|
heim_tid_t node_type;
|
|
int ret;
|
|
|
|
if (ptr == NULL)
|
|
heim_abort("heim_path_vcreate() does not create root nodes");
|
|
|
|
while (path_element != NULL) {
|
|
next_path_element = va_arg(ap, heim_object_t);
|
|
node_type = heim_get_tid(node);
|
|
|
|
if (node_type == HEIM_TID_DICT) {
|
|
next_node = heim_dict_get_value(node, path_element);
|
|
} else if (node_type == HEIM_TID_ARRAY) {
|
|
int idx = -1;
|
|
|
|
if (heim_get_tid(path_element) == HEIM_TID_NUMBER)
|
|
idx = heim_number_get_int(path_element);
|
|
if (idx < 0) {
|
|
if (error)
|
|
*error = heim_error_create(EINVAL,
|
|
"heim_path() path elements for "
|
|
"array nodes must be numeric "
|
|
"and positive");
|
|
return EINVAL;
|
|
}
|
|
if (idx < heim_array_get_length(node))
|
|
next_node = heim_array_get_value(node, idx);
|
|
else
|
|
next_node = NULL;
|
|
} else if (node_type == HEIM_TID_DB && next_path_element != NULL) {
|
|
if (error)
|
|
*error = heim_error_create(EINVAL, "Interior node is a DB");
|
|
return EINVAL;
|
|
}
|
|
|
|
if (next_path_element == NULL)
|
|
break;
|
|
|
|
/* Create missing interior node */
|
|
if (next_node == NULL) {
|
|
next_node = heim_dict_create(size); /* no arrays or DBs, just dicts */
|
|
if (next_node == NULL) {
|
|
ret = ENOMEM;
|
|
goto err;
|
|
}
|
|
|
|
if (node_type == HEIM_TID_DICT) {
|
|
ret = heim_dict_set_value(node, path_element, next_node);
|
|
} else if (node_type == HEIM_TID_ARRAY &&
|
|
heim_number_get_int(path_element) <= heim_array_get_length(node)) {
|
|
ret = heim_array_insert_value(node,
|
|
heim_number_get_int(path_element),
|
|
next_node);
|
|
} else {
|
|
ret = EINVAL;
|
|
if (error)
|
|
*error = heim_error_create(ret, "Node in path not a "
|
|
"container");
|
|
goto err;
|
|
}
|
|
heim_release(next_node);
|
|
if (ret)
|
|
goto err;
|
|
}
|
|
|
|
path_element = next_path_element;
|
|
node = next_node;
|
|
next_node = NULL;
|
|
}
|
|
|
|
if (path_element == NULL)
|
|
goto err;
|
|
|
|
/* Add the leaf */
|
|
if (leaf != NULL) {
|
|
if (node_type == HEIM_TID_DICT)
|
|
ret = heim_dict_set_value(node, path_element, leaf);
|
|
else
|
|
ret = heim_array_insert_value(node,
|
|
heim_number_get_int(path_element),
|
|
leaf);
|
|
}
|
|
return 0;
|
|
|
|
err:
|
|
if (error && !*error) {
|
|
if (ret == ENOMEM)
|
|
*error = heim_error_enomem();
|
|
else
|
|
*error = heim_error_create(ret, "Could not set "
|
|
"dict value");
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Create a path in a heim_object_t tree
|
|
*
|
|
* @param ptr the tree
|
|
* @param size the size of the heim_dict_t nodes to be created
|
|
* @param leaf leaf node to be added, if any
|
|
* @param error error (output)
|
|
* @param ... NULL-terminated list of path component objects
|
|
*
|
|
* Create a path of heim_dict_t interior nodes in a given heim_object_t
|
|
* tree, as necessary, and set/replace a leaf, if given (if leaf is NULL
|
|
* then the leaf is not deleted).
|
|
*
|
|
* @return 0 on success, else a system error
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
int
|
|
heim_path_create(heim_object_t ptr, size_t size, heim_object_t leaf,
|
|
heim_error_t *error, ...)
|
|
{
|
|
va_list ap;
|
|
int ret;
|
|
|
|
va_start(ap, error);
|
|
ret = heim_path_vcreate(ptr, size, leaf, error, ap);
|
|
va_end(ap);
|
|
return ret;
|
|
}
|
|
|
|
/**
|
|
* Delete leaf node named by a path in a heim_object_t tree
|
|
*
|
|
* @param ptr the tree
|
|
* @param error error (output)
|
|
* @param ap NULL-terminated list of path component objects
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
void
|
|
heim_path_vdelete(heim_object_t ptr, heim_error_t *error, va_list ap)
|
|
{
|
|
heim_object_t parent, key, child;
|
|
|
|
child = heim_path_vget2(ptr, &parent, &key, error, ap);
|
|
if (child != NULL) {
|
|
if (heim_get_tid(parent) == HEIM_TID_DICT)
|
|
heim_dict_delete_key(parent, key);
|
|
else if (heim_get_tid(parent) == HEIM_TID_DB)
|
|
heim_db_delete_key(parent, NULL, key, error);
|
|
else if (heim_get_tid(parent) == HEIM_TID_ARRAY)
|
|
heim_array_delete_value(parent, heim_number_get_int(key));
|
|
heim_release(child);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Delete leaf node named by a path in a heim_object_t tree
|
|
*
|
|
* @param ptr the tree
|
|
* @param error error (output)
|
|
* @param ap NULL-terminated list of path component objects
|
|
*
|
|
* @addtogroup heimbase
|
|
*/
|
|
|
|
void
|
|
heim_path_delete(heim_object_t ptr, heim_error_t *error, ...)
|
|
{
|
|
va_list ap;
|
|
|
|
va_start(ap, error);
|
|
heim_path_vdelete(ptr, error, ap);
|
|
va_end(ap);
|
|
return;
|
|
}
|
|
|