tor-android/external/badvpn_dns/ncd/NCDValCons.c

284 lines
8.1 KiB
C

/**
* @file NCDValCons.c
* @author Ambroz Bizjak <ambrop7@gmail.com>
*
* @section LICENSE
*
* 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 author 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 COPYRIGHT HOLDERS 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 AUTHOR 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 <misc/balloc.h>
#include "NCDValCons.h"
#define GROWARRAY_NAME NCDValCons__Array
#define GROWARRAY_OBJECT_TYPE NCDValCons
#define GROWARRAY_ARRAY_MEMBER elems
#define GROWARRAY_CAPACITY_MEMBER elems_capacity
#define GROWARRAY_MAX_CAPACITY INT_MAX
#include <misc/grow_array.h>
static int alloc_elem (NCDValCons *o)
{
ASSERT(o->elems_size >= 0)
ASSERT(o->elems_size <= o->elems_capacity)
if (o->elems_size == o->elems_capacity && !NCDValCons__Array_DoubleUp(o)) {
return -1;
}
return o->elems_size++;
}
static void assert_cons_val (NCDValCons *o, NCDValConsVal val)
{
#ifndef NDEBUG
switch (val.cons_type) {
case NCDVALCONS_TYPE_COMPLETE: {
ASSERT(!NCDVal_IsInvalid(NCDVal_FromSafe(o->mem, val.u.complete_ref)))
} break;
case NCDVALCONS_TYPE_INCOMPLETE_LIST:
case NCDVALCONS_TYPE_INCOMPLETE_MAP: {
ASSERT(val.u.incomplete.elems_idx >= -1)
ASSERT(val.u.incomplete.elems_idx < o->elems_size)
ASSERT(val.u.incomplete.count >= 0)
} break;
default:
ASSERT(0);
}
#endif
}
static int complete_value (NCDValCons *o, NCDValConsVal val, NCDValSafeRef *out, int *out_error)
{
assert_cons_val(o, val);
ASSERT(out)
ASSERT(out_error)
switch (val.cons_type) {
case NCDVALCONS_TYPE_COMPLETE: {
*out = val.u.complete_ref;
} break;
case NCDVALCONS_TYPE_INCOMPLETE_LIST: {
NCDValRef list = NCDVal_NewList(o->mem, val.u.incomplete.count);
if (NCDVal_IsInvalid(list)) {
goto fail_memory;
}
int elemidx = val.u.incomplete.elems_idx;
while (elemidx != -1) {
ASSERT(elemidx >= 0)
ASSERT(elemidx < o->elems_size)
NCDValRef elem = NCDVal_FromSafe(o->mem, o->elems[elemidx].ref);
if (!NCDVal_ListAppend(list, elem)) {
*out_error = NCDVALCONS_ERROR_DEPTH;
return 0;
}
elemidx = o->elems[elemidx].next;
}
*out = NCDVal_ToSafe(list);
} break;
case NCDVALCONS_TYPE_INCOMPLETE_MAP: {
NCDValRef map = NCDVal_NewMap(o->mem, val.u.incomplete.count);
if (NCDVal_IsInvalid(map)) {
goto fail_memory;
}
int keyidx = val.u.incomplete.elems_idx;
while (keyidx != -1) {
ASSERT(keyidx >= 0)
ASSERT(keyidx < o->elems_size)
int validx = o->elems[keyidx].next;
ASSERT(validx >= 0)
ASSERT(validx < o->elems_size)
NCDValRef key = NCDVal_FromSafe(o->mem, o->elems[keyidx].ref);
NCDValRef value = NCDVal_FromSafe(o->mem, o->elems[validx].ref);
int inserted;
if (!NCDVal_MapInsert(map, key, value, &inserted)) {
*out_error = NCDVALCONS_ERROR_DEPTH;
return 0;
}
if (!inserted) {
*out_error = NCDVALCONS_ERROR_DUPLICATE_KEY;
return 0;
}
keyidx = o->elems[validx].next;
}
*out = NCDVal_ToSafe(map);
} break;
default:
ASSERT(0);
}
return 1;
fail_memory:
*out_error = NCDVALCONS_ERROR_MEMORY;
return 0;
}
int NCDValCons_Init (NCDValCons *o, NCDValMem *mem)
{
ASSERT(mem)
o->mem = mem;
o->elems_size = 0;
if (!NCDValCons__Array_Init(o, 1)) {
return 0;
}
return 1;
}
void NCDValCons_Free (NCDValCons *o)
{
NCDValCons__Array_Free(o);
}
int NCDValCons_NewString (NCDValCons *o, const uint8_t *data, size_t len, NCDValConsVal *out, int *out_error)
{
ASSERT(out)
ASSERT(out_error)
NCDValRef ref = NCDVal_NewStringBin(o->mem, data, len);
if (NCDVal_IsInvalid(ref)) {
*out_error = NCDVALCONS_ERROR_MEMORY;
return 0;
}
out->cons_type = NCDVALCONS_TYPE_COMPLETE;
out->u.complete_ref = NCDVal_ToSafe(ref);
return 1;
}
void NCDValCons_NewList (NCDValCons *o, NCDValConsVal *out)
{
ASSERT(out)
out->cons_type = NCDVALCONS_TYPE_INCOMPLETE_LIST;
out->u.incomplete.elems_idx = -1;
out->u.incomplete.count = 0;
}
void NCDValCons_NewMap (NCDValCons *o, NCDValConsVal *out)
{
ASSERT(out)
out->cons_type = NCDVALCONS_TYPE_INCOMPLETE_MAP;
out->u.incomplete.elems_idx = -1;
out->u.incomplete.count = 0;
}
int NCDValCons_ListPrepend (NCDValCons *o, NCDValConsVal *list, NCDValConsVal elem, int *out_error)
{
assert_cons_val(o, *list);
ASSERT(list->cons_type == NCDVALCONS_TYPE_INCOMPLETE_LIST)
assert_cons_val(o, elem);
ASSERT(out_error)
int elemidx = alloc_elem(o);
if (elemidx < 0) {
*out_error = NCDVALCONS_ERROR_MEMORY;
return 0;
}
o->elems[elemidx].next = list->u.incomplete.elems_idx;
if (!complete_value(o, elem, &o->elems[elemidx].ref, out_error)) {
return 0;
}
list->u.incomplete.elems_idx = elemidx;
list->u.incomplete.count++;
return 1;
}
int NCDValCons_MapInsert (NCDValCons *o, NCDValConsVal *map, NCDValConsVal key, NCDValConsVal value, int *out_error)
{
assert_cons_val(o, *map);
ASSERT(map->cons_type == NCDVALCONS_TYPE_INCOMPLETE_MAP)
assert_cons_val(o, key);
assert_cons_val(o, value);
ASSERT(out_error)
int validx = alloc_elem(o);
if (validx < 0) {
*out_error = NCDVALCONS_ERROR_MEMORY;
return 0;
}
int keyidx = alloc_elem(o);
if (keyidx < 0) {
*out_error = NCDVALCONS_ERROR_MEMORY;
return 0;
}
o->elems[validx].next = map->u.incomplete.elems_idx;
o->elems[keyidx].next = validx;
if (!complete_value(o, value, &o->elems[validx].ref, out_error)) {
return 0;
}
if (!complete_value(o, key, &o->elems[keyidx].ref, out_error)) {
return 0;
}
map->u.incomplete.elems_idx = keyidx;
map->u.incomplete.count++;
return 1;
}
int NCDValCons_Complete (NCDValCons *o, NCDValConsVal val, NCDValRef *out, int *out_error)
{
assert_cons_val(o, val);
ASSERT(out)
ASSERT(out_error)
NCDValSafeRef sref;
if (!complete_value(o, val, &sref, out_error)) {
return 0;
}
*out = NCDVal_FromSafe(o->mem, sref);
return 1;
}