/* * virjson.c: JSON object parsing/formatting * * Copyright (C) 2009-2010, 2012-2015 Red Hat, Inc. * Copyright (C) 2009 Daniel P. Berrange * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library. If not, see * . * */ #include #include "virjson.h" #include "viralloc.h" #include "virerror.h" #include "virlog.h" #include "virstring.h" #include "virutil.h" #if WITH_YAJL # include # include # ifdef WITH_YAJL2 # define yajl_size_t size_t # define VIR_YAJL_STATUS_OK(status) ((status) == yajl_status_ok) # else # define yajl_size_t unsigned int # define yajl_complete_parse yajl_parse_complete # define VIR_YAJL_STATUS_OK(status) \ ((status) == yajl_status_ok || (status) == yajl_status_insufficient_data) # endif #endif /* XXX fixme */ #define VIR_FROM_THIS VIR_FROM_NONE VIR_LOG_INIT("util.json"); typedef struct _virJSONParserState virJSONParserState; typedef virJSONParserState *virJSONParserStatePtr; struct _virJSONParserState { virJSONValuePtr value; char *key; }; typedef struct _virJSONParser virJSONParser; typedef virJSONParser *virJSONParserPtr; struct _virJSONParser { virJSONValuePtr head; virJSONParserStatePtr state; size_t nstate; int wrap; }; /** * virJSONValueObjectAddVArgs: * @obj: JSON object to add the values to * @args: a key-value argument pairs, terminated by NULL * * Adds the key-value pairs supplied as variable argument list to @obj. * * Keys look like s:name the first letter is a type code: * Explanation of type codes: * s: string value, must be non-null * S: string value, omitted if null * * i: signed integer value * j: signed integer value, error if negative * z: signed integer value, omitted if zero * y: signed integer value, omitted if zero, error if negative * * I: signed long integer value * J: signed long integer value, error if negative * Z: signed long integer value, omitted if zero * Y: signed long integer value, omitted if zero, error if negative * * u: unsigned integer value * p: unsigned integer value, omitted if zero * * U: unsigned long integer value (see below for quirks) * P: unsigned long integer value, omitted if zero * * b: boolean value * B: boolean value, omitted if false * * d: double precision floating point number * n: json null value * * a: json object, must be non-NULL * A: json object, omitted if NULL * m: a bitmap represented as a JSON array, must be non-NULL * M: a bitmap represented as a JSON array, omitted if NULL * * The value corresponds to the selected type. * * Returns -1 on error. 1 on success, if at least one key:pair was valid 0 * in case of no error but nothing was filled. */ int virJSONValueObjectAddVArgs(virJSONValuePtr obj, va_list args) { char type; char *key; int ret = -1; int rc; while ((key = va_arg(args, char *)) != NULL) { if (strlen(key) < 3) { virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' is too short, missing type prefix"), key); goto cleanup; } type = key[0]; key += 2; /* This doesn't support maps, but no command uses those. */ switch (type) { case 'S': case 's': { char *val = va_arg(args, char *); if (!val) { if (type == 'S') continue; virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' must not have null value"), key); goto cleanup; } rc = virJSONValueObjectAppendString(obj, key, val); } break; case 'z': case 'y': case 'j': case 'i': { int val = va_arg(args, int); if (val < 0 && (type == 'j' || type == 'y')) { virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' must not be negative"), key); goto cleanup; } if (!val && (type == 'z' || type == 'y')) continue; rc = virJSONValueObjectAppendNumberInt(obj, key, val); } break; case 'p': case 'u': { unsigned int val = va_arg(args, unsigned int); if (!val && type == 'p') continue; rc = virJSONValueObjectAppendNumberUint(obj, key, val); } break; case 'Z': case 'Y': case 'J': case 'I': { long long val = va_arg(args, long long); if (val < 0 && (type == 'J' || type == 'Y')) { virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' must not be negative"), key); goto cleanup; } if (!val && (type == 'Z' || type == 'Y')) continue; rc = virJSONValueObjectAppendNumberLong(obj, key, val); } break; case 'P': case 'U': { /* qemu silently truncates numbers larger than LLONG_MAX, * so passing the full range of unsigned 64 bit integers * is not safe here. Pass them as signed 64 bit integers * instead. */ long long val = va_arg(args, long long); if (!val && type == 'P') continue; rc = virJSONValueObjectAppendNumberLong(obj, key, val); } break; case 'd': { double val = va_arg(args, double); rc = virJSONValueObjectAppendNumberDouble(obj, key, val); } break; case 'B': case 'b': { int val = va_arg(args, int); if (!val && type == 'B') continue; rc = virJSONValueObjectAppendBoolean(obj, key, val); } break; case 'n': { rc = virJSONValueObjectAppendNull(obj, key); } break; case 'A': case 'a': { virJSONValuePtr val = va_arg(args, virJSONValuePtr); if (!val) { if (type == 'A') continue; virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' must not have null value"), key); goto cleanup; } rc = virJSONValueObjectAppend(obj, key, val); } break; case 'M': case 'm': { virBitmapPtr map = va_arg(args, virBitmapPtr); virJSONValuePtr jsonMap; if (!map) { if (type == 'M') continue; virReportError(VIR_ERR_INTERNAL_ERROR, _("argument key '%s' must not have null value"), key); goto cleanup; } if (!(jsonMap = virJSONValueNewArrayFromBitmap(map))) goto cleanup; if ((rc = virJSONValueObjectAppend(obj, key, jsonMap)) < 0) virJSONValueFree(jsonMap); } break; default: virReportError(VIR_ERR_INTERNAL_ERROR, _("unsupported data type '%c' for arg '%s'"), type, key - 2); goto cleanup; } if (rc < 0) goto cleanup; } /* verify that we added at least one key-value pair */ if (virJSONValueObjectKeysNumber(obj) == 0) { ret = 0; goto cleanup; } ret = 1; cleanup: return ret; } int virJSONValueObjectAdd(virJSONValuePtr obj, ...) { va_list args; int ret; va_start(args, obj); ret = virJSONValueObjectAddVArgs(obj, args); va_end(args); return ret; } int virJSONValueObjectCreateVArgs(virJSONValuePtr *obj, va_list args) { int ret; if (!(*obj = virJSONValueNewObject())) return -1; /* free the object on error, or if no value objects were added */ if ((ret = virJSONValueObjectAddVArgs(*obj, args)) <= 0) { virJSONValueFree(*obj); *obj = NULL; } return ret; } int virJSONValueObjectCreate(virJSONValuePtr *obj, ...) { va_list args; int ret; va_start(args, obj); ret = virJSONValueObjectCreateVArgs(obj, args); va_end(args); return ret; } void virJSONValueFree(virJSONValuePtr value) { size_t i; if (!value || value->protect) return; switch ((virJSONType) value->type) { case VIR_JSON_TYPE_OBJECT: for (i = 0; i < value->data.object.npairs; i++) { VIR_FREE(value->data.object.pairs[i].key); virJSONValueFree(value->data.object.pairs[i].value); } VIR_FREE(value->data.object.pairs); break; case VIR_JSON_TYPE_ARRAY: for (i = 0; i < value->data.array.nvalues; i++) virJSONValueFree(value->data.array.values[i]); VIR_FREE(value->data.array.values); break; case VIR_JSON_TYPE_STRING: VIR_FREE(value->data.string); break; case VIR_JSON_TYPE_NUMBER: VIR_FREE(value->data.number); break; case VIR_JSON_TYPE_BOOLEAN: case VIR_JSON_TYPE_NULL: break; } VIR_FREE(value); } void virJSONValueHashFree(void *opaque, const void *name ATTRIBUTE_UNUSED) { virJSONValueFree(opaque); } virJSONValuePtr virJSONValueNewString(const char *data) { virJSONValuePtr val; if (!data) return virJSONValueNewNull(); if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_STRING; if (VIR_STRDUP(val->data.string, data) < 0) { VIR_FREE(val); return NULL; } return val; } virJSONValuePtr virJSONValueNewStringLen(const char *data, size_t length) { virJSONValuePtr val; if (!data) return virJSONValueNewNull(); if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_STRING; if (VIR_STRNDUP(val->data.string, data, length) < 0) { VIR_FREE(val); return NULL; } return val; } static virJSONValuePtr virJSONValueNewNumber(const char *data) { virJSONValuePtr val; if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_NUMBER; if (VIR_STRDUP(val->data.number, data) < 0) { VIR_FREE(val); return NULL; } return val; } virJSONValuePtr virJSONValueNewNumberInt(int data) { virJSONValuePtr val = NULL; char *str; if (virAsprintf(&str, "%i", data) < 0) return NULL; val = virJSONValueNewNumber(str); VIR_FREE(str); return val; } virJSONValuePtr virJSONValueNewNumberUint(unsigned int data) { virJSONValuePtr val = NULL; char *str; if (virAsprintf(&str, "%u", data) < 0) return NULL; val = virJSONValueNewNumber(str); VIR_FREE(str); return val; } virJSONValuePtr virJSONValueNewNumberLong(long long data) { virJSONValuePtr val = NULL; char *str; if (virAsprintf(&str, "%lld", data) < 0) return NULL; val = virJSONValueNewNumber(str); VIR_FREE(str); return val; } virJSONValuePtr virJSONValueNewNumberUlong(unsigned long long data) { virJSONValuePtr val = NULL; char *str; if (virAsprintf(&str, "%llu", data) < 0) return NULL; val = virJSONValueNewNumber(str); VIR_FREE(str); return val; } virJSONValuePtr virJSONValueNewNumberDouble(double data) { virJSONValuePtr val = NULL; char *str; if (virDoubleToStr(&str, data) < 0) return NULL; val = virJSONValueNewNumber(str); VIR_FREE(str); return val; } virJSONValuePtr virJSONValueNewBoolean(int boolean_) { virJSONValuePtr val; if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_BOOLEAN; val->data.boolean = boolean_; return val; } virJSONValuePtr virJSONValueNewNull(void) { virJSONValuePtr val; if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_NULL; return val; } virJSONValuePtr virJSONValueNewArray(void) { virJSONValuePtr val; if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_ARRAY; return val; } virJSONValuePtr virJSONValueNewObject(void) { virJSONValuePtr val; if (VIR_ALLOC(val) < 0) return NULL; val->type = VIR_JSON_TYPE_OBJECT; return val; } int virJSONValueObjectAppend(virJSONValuePtr object, const char *key, virJSONValuePtr value) { char *newkey; if (object->type != VIR_JSON_TYPE_OBJECT) return -1; if (virJSONValueObjectHasKey(object, key)) return -1; if (VIR_STRDUP(newkey, key) < 0) return -1; if (VIR_REALLOC_N(object->data.object.pairs, object->data.object.npairs + 1) < 0) { VIR_FREE(newkey); return -1; } object->data.object.pairs[object->data.object.npairs].key = newkey; object->data.object.pairs[object->data.object.npairs].value = value; object->data.object.npairs++; return 0; } int virJSONValueObjectAppendString(virJSONValuePtr object, const char *key, const char *value) { virJSONValuePtr jvalue = virJSONValueNewString(value); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNumberInt(virJSONValuePtr object, const char *key, int number) { virJSONValuePtr jvalue = virJSONValueNewNumberInt(number); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNumberUint(virJSONValuePtr object, const char *key, unsigned int number) { virJSONValuePtr jvalue = virJSONValueNewNumberUint(number); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNumberLong(virJSONValuePtr object, const char *key, long long number) { virJSONValuePtr jvalue = virJSONValueNewNumberLong(number); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNumberUlong(virJSONValuePtr object, const char *key, unsigned long long number) { virJSONValuePtr jvalue = virJSONValueNewNumberUlong(number); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNumberDouble(virJSONValuePtr object, const char *key, double number) { virJSONValuePtr jvalue = virJSONValueNewNumberDouble(number); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendBoolean(virJSONValuePtr object, const char *key, int boolean_) { virJSONValuePtr jvalue = virJSONValueNewBoolean(boolean_); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueObjectAppendNull(virJSONValuePtr object, const char *key) { virJSONValuePtr jvalue = virJSONValueNewNull(); if (!jvalue) return -1; if (virJSONValueObjectAppend(object, key, jvalue) < 0) { virJSONValueFree(jvalue); return -1; } return 0; } int virJSONValueArrayAppend(virJSONValuePtr array, virJSONValuePtr value) { if (array->type != VIR_JSON_TYPE_ARRAY) return -1; if (VIR_REALLOC_N(array->data.array.values, array->data.array.nvalues + 1) < 0) return -1; array->data.array.values[array->data.array.nvalues] = value; array->data.array.nvalues++; return 0; } int virJSONValueObjectHasKey(virJSONValuePtr object, const char *key) { size_t i; if (object->type != VIR_JSON_TYPE_OBJECT) return -1; for (i = 0; i < object->data.object.npairs; i++) { if (STREQ(object->data.object.pairs[i].key, key)) return 1; } return 0; } virJSONValuePtr virJSONValueObjectGet(virJSONValuePtr object, const char *key) { size_t i; if (object->type != VIR_JSON_TYPE_OBJECT) return NULL; for (i = 0; i < object->data.object.npairs; i++) { if (STREQ(object->data.object.pairs[i].key, key)) return object->data.object.pairs[i].value; } return NULL; } static virJSONValuePtr virJSONValueObjectSteal(virJSONValuePtr object, const char *key) { size_t i; virJSONValuePtr obj = NULL; if (object->type != VIR_JSON_TYPE_OBJECT) return NULL; for (i = 0; i < object->data.object.npairs; i++) { if (STREQ(object->data.object.pairs[i].key, key)) { VIR_STEAL_PTR(obj, object->data.object.pairs[i].value); VIR_FREE(object->data.object.pairs[i].key); VIR_DELETE_ELEMENT(object->data.object.pairs, i, object->data.object.npairs); break; } } return obj; } /* Return the value associated with KEY within OBJECT, but return NULL * if the key is missing or if value is not the correct TYPE. */ virJSONValuePtr virJSONValueObjectGetByType(virJSONValuePtr object, const char *key, virJSONType type) { virJSONValuePtr value = virJSONValueObjectGet(object, key); if (value && value->type == type) return value; return NULL; } /* Steal the value associated with KEY within OBJECT, but return NULL * if the key is missing or if value is not the correct TYPE. */ static virJSONValuePtr virJSONValueObjectStealByType(virJSONValuePtr object, const char *key, virJSONType type) { virJSONValuePtr value = virJSONValueObjectSteal(object, key); if (value && value->type == type) return value; return NULL; } int virJSONValueObjectKeysNumber(virJSONValuePtr object) { if (object->type != VIR_JSON_TYPE_OBJECT) return -1; return object->data.object.npairs; } const char * virJSONValueObjectGetKey(virJSONValuePtr object, unsigned int n) { if (object->type != VIR_JSON_TYPE_OBJECT) return NULL; if (n >= object->data.object.npairs) return NULL; return object->data.object.pairs[n].key; } /* Remove the key-value pair tied to @key out of @object. If @value is * not NULL, the dropped value object is returned instead of freed. * Returns 1 on success, 0 if no key was found, and -1 on error. */ int virJSONValueObjectRemoveKey(virJSONValuePtr object, const char *key, virJSONValuePtr *value) { size_t i; if (value) *value = NULL; if (object->type != VIR_JSON_TYPE_OBJECT) return -1; for (i = 0; i < object->data.object.npairs; i++) { if (STREQ(object->data.object.pairs[i].key, key)) { if (value) { *value = object->data.object.pairs[i].value; object->data.object.pairs[i].value = NULL; } VIR_FREE(object->data.object.pairs[i].key); virJSONValueFree(object->data.object.pairs[i].value); VIR_DELETE_ELEMENT(object->data.object.pairs, i, object->data.object.npairs); return 1; } } return 0; } virJSONValuePtr virJSONValueObjectGetValue(virJSONValuePtr object, unsigned int n) { if (object->type != VIR_JSON_TYPE_OBJECT) return NULL; if (n >= object->data.object.npairs) return NULL; return object->data.object.pairs[n].value; } bool virJSONValueIsObject(virJSONValuePtr object) { if (object) return object->type == VIR_JSON_TYPE_OBJECT; else return false; } bool virJSONValueIsArray(virJSONValuePtr array) { return array->type == VIR_JSON_TYPE_ARRAY; } ssize_t virJSONValueArraySize(const virJSONValue *array) { if (array->type != VIR_JSON_TYPE_ARRAY) return -1; return array->data.array.nvalues; } virJSONValuePtr virJSONValueArrayGet(virJSONValuePtr array, unsigned int element) { if (array->type != VIR_JSON_TYPE_ARRAY) return NULL; if (element >= array->data.array.nvalues) return NULL; return array->data.array.values[element]; } virJSONValuePtr virJSONValueArraySteal(virJSONValuePtr array, unsigned int element) { virJSONValuePtr ret = NULL; if (array->type != VIR_JSON_TYPE_ARRAY) return NULL; if (element >= array->data.array.nvalues) return NULL; ret = array->data.array.values[element]; VIR_DELETE_ELEMENT(array->data.array.values, element, array->data.array.nvalues); return ret; } /** * virJSONValueArrayForeachSteal: * @array: array to iterate * @cb: callback called on every member of the array * @opaque: custom data for the callback * * Iterates members of the array and calls the callback on every single member. * The return codes of the callback are interpreted as follows: * 0: callback claims ownership of the array element and is responsible for * freeing it * 1: callback doesn't claim ownership of the element * -1: callback doesn't claim ownership of the element and iteration does not * continue * * Returns 0 if all members were iterated and/or stolen by the callback; -1 * on callback failure or if the JSON value object is not an array. * The rest of the members stay in possession of the array and it's condensed. */ int virJSONValueArrayForeachSteal(virJSONValuePtr array, virJSONArrayIteratorFunc cb, void *opaque) { size_t i; size_t j = 0; int ret = 0; int rc; if (array->type != VIR_JSON_TYPE_ARRAY) return -1; for (i = 0; i < array->data.array.nvalues; i++) { if ((rc = cb(i, array->data.array.values[i], opaque)) < 0) { ret = -1; break; } if (rc == 0) array->data.array.values[i] = NULL; } /* condense the remaining entries at the beginning */ for (i = 0; i < array->data.array.nvalues; i++) { if (!array->data.array.values[i]) continue; array->data.array.values[j++] = array->data.array.values[i]; } array->data.array.nvalues = j; return ret; } const char * virJSONValueGetString(virJSONValuePtr string) { if (string->type != VIR_JSON_TYPE_STRING) return NULL; return string->data.string; } int virJSONValueGetNumberInt(virJSONValuePtr number, int *value) { if (number->type != VIR_JSON_TYPE_NUMBER) return -1; return virStrToLong_i(number->data.number, NULL, 10, value); } int virJSONValueGetNumberUint(virJSONValuePtr number, unsigned int *value) { if (number->type != VIR_JSON_TYPE_NUMBER) return -1; return virStrToLong_ui(number->data.number, NULL, 10, value); } int virJSONValueGetNumberLong(virJSONValuePtr number, long long *value) { if (number->type != VIR_JSON_TYPE_NUMBER) return -1; return virStrToLong_ll(number->data.number, NULL, 10, value); } int virJSONValueGetNumberUlong(virJSONValuePtr number, unsigned long long *value) { if (number->type != VIR_JSON_TYPE_NUMBER) return -1; return virStrToLong_ull(number->data.number, NULL, 10, value); } int virJSONValueGetNumberDouble(virJSONValuePtr number, double *value) { if (number->type != VIR_JSON_TYPE_NUMBER) return -1; return virStrToDouble(number->data.number, NULL, value); } int virJSONValueGetBoolean(virJSONValuePtr val, bool *value) { if (val->type != VIR_JSON_TYPE_BOOLEAN) return -1; *value = val->data.boolean; return 0; } /** * virJSONValueGetArrayAsBitmap: * @val: JSON array to convert to bitmap * @bitmap: New bitmap is allocated filled and returned via this argument * * Attempts a conversion of a JSON array to a bitmap. The members of the array * must be non-negative integers for the conversion to succeed. This function * does not report libvirt errors so that it can be used to probe that the * array can be represented as a bitmap. * * Returns 0 on success and fills @bitmap; -1 on error and @bitmap is set to * NULL. */ int virJSONValueGetArrayAsBitmap(const virJSONValue *val, virBitmapPtr *bitmap) { int ret = -1; virJSONValuePtr elem; size_t i; unsigned long long *elems = NULL; unsigned long long maxelem = 0; *bitmap = NULL; if (val->type != VIR_JSON_TYPE_ARRAY) return -1; if (VIR_ALLOC_N_QUIET(elems, val->data.array.nvalues) < 0) return -1; /* first pass converts array members to numbers and finds the maximum */ for (i = 0; i < val->data.array.nvalues; i++) { elem = val->data.array.values[i]; if (elem->type != VIR_JSON_TYPE_NUMBER || virStrToLong_ullp(elem->data.number, NULL, 10, &elems[i]) < 0) goto cleanup; if (elems[i] > maxelem) maxelem = elems[i]; } if (!(*bitmap = virBitmapNewQuiet(maxelem + 1))) goto cleanup; /* second pass sets the correct bits in the map */ for (i = 0; i < val->data.array.nvalues; i++) ignore_value(virBitmapSetBit(*bitmap, elems[i])); ret = 0; cleanup: VIR_FREE(elems); return ret; } virJSONValuePtr virJSONValueNewArrayFromBitmap(virBitmapPtr bitmap) { virJSONValuePtr ret; ssize_t pos = -1; if (!(ret = virJSONValueNewArray())) return NULL; if (!bitmap) return ret; while ((pos = virBitmapNextSetBit(bitmap, pos)) > -1) { virJSONValuePtr newelem; if (!(newelem = virJSONValueNewNumberLong(pos)) || virJSONValueArrayAppend(ret, newelem) < 0) { virJSONValueFree(newelem); goto error; } } return ret; error: virJSONValueFree(ret); return NULL; } bool virJSONValueIsNull(virJSONValuePtr val) { return val->type == VIR_JSON_TYPE_NULL; } const char * virJSONValueObjectGetString(virJSONValuePtr object, const char *key) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return NULL; return virJSONValueGetString(val); } int virJSONValueObjectGetNumberInt(virJSONValuePtr object, const char *key, int *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetNumberInt(val, value); } int virJSONValueObjectGetNumberUint(virJSONValuePtr object, const char *key, unsigned int *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetNumberUint(val, value); } int virJSONValueObjectGetNumberLong(virJSONValuePtr object, const char *key, long long *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetNumberLong(val, value); } int virJSONValueObjectGetNumberUlong(virJSONValuePtr object, const char *key, unsigned long long *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetNumberUlong(val, value); } int virJSONValueObjectGetNumberDouble(virJSONValuePtr object, const char *key, double *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetNumberDouble(val, value); } int virJSONValueObjectGetBoolean(virJSONValuePtr object, const char *key, bool *value) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueGetBoolean(val, value); } virJSONValuePtr virJSONValueObjectGetObject(virJSONValuePtr object, const char *key) { return virJSONValueObjectGetByType(object, key, VIR_JSON_TYPE_OBJECT); } virJSONValuePtr virJSONValueObjectGetArray(virJSONValuePtr object, const char *key) { return virJSONValueObjectGetByType(object, key, VIR_JSON_TYPE_ARRAY); } virJSONValuePtr virJSONValueObjectStealArray(virJSONValuePtr object, const char *key) { return virJSONValueObjectStealByType(object, key, VIR_JSON_TYPE_ARRAY); } int virJSONValueObjectIsNull(virJSONValuePtr object, const char *key) { virJSONValuePtr val = virJSONValueObjectGet(object, key); if (!val) return -1; return virJSONValueIsNull(val); } /** * virJSONValueObjectForeachKeyValue: * @object: JSON object to iterate * @cb: callback to call on key-value pairs contained in the object * @opaque: generic data for the callback * * Iterates all key=value pairs in @object. Iteration breaks if @cb returns * negative value. * * Returns 0 if all elements were iterated, -2 if @cb returned negative value * during iteration and -1 on generic errors. */ int virJSONValueObjectForeachKeyValue(virJSONValuePtr object, virJSONValueObjectIteratorFunc cb, void *opaque) { size_t i; if (object->type != VIR_JSON_TYPE_OBJECT) return -1; for (i = 0; i < object->data.object.npairs; i++) { virJSONObjectPairPtr elem = object->data.object.pairs + i; if (cb(elem->key, elem->value, opaque) < 0) return -2; } return 0; } virJSONValuePtr virJSONValueCopy(const virJSONValue *in) { size_t i; virJSONValuePtr out = NULL; if (!in) return NULL; switch ((virJSONType) in->type) { case VIR_JSON_TYPE_OBJECT: out = virJSONValueNewObject(); if (!out) return NULL; for (i = 0; i < in->data.object.npairs; i++) { virJSONValuePtr val = NULL; if (!(val = virJSONValueCopy(in->data.object.pairs[i].value))) goto error; if (virJSONValueObjectAppend(out, in->data.object.pairs[i].key, val) < 0) { virJSONValueFree(val); goto error; } } break; case VIR_JSON_TYPE_ARRAY: out = virJSONValueNewArray(); if (!out) return NULL; for (i = 0; i < in->data.array.nvalues; i++) { virJSONValuePtr val = NULL; if (!(val = virJSONValueCopy(in->data.array.values[i]))) goto error; if (virJSONValueArrayAppend(out, val) < 0) { virJSONValueFree(val); goto error; } } break; /* No need to error out in the following cases */ case VIR_JSON_TYPE_STRING: out = virJSONValueNewString(in->data.string); break; case VIR_JSON_TYPE_NUMBER: out = virJSONValueNewNumber(in->data.number); break; case VIR_JSON_TYPE_BOOLEAN: out = virJSONValueNewBoolean(in->data.boolean); break; case VIR_JSON_TYPE_NULL: out = virJSONValueNewNull(); break; } return out; error: virJSONValueFree(out); return NULL; } #if WITH_YAJL static int virJSONParserInsertValue(virJSONParserPtr parser, virJSONValuePtr value) { if (!parser->head) { parser->head = value; } else { virJSONParserStatePtr state; if (!parser->nstate) { VIR_DEBUG("got a value to insert without a container"); return -1; } state = &parser->state[parser->nstate-1]; switch (state->value->type) { case VIR_JSON_TYPE_OBJECT: { if (!state->key) { VIR_DEBUG("missing key when inserting object value"); return -1; } if (virJSONValueObjectAppend(state->value, state->key, value) < 0) return -1; VIR_FREE(state->key); } break; case VIR_JSON_TYPE_ARRAY: { if (state->key) { VIR_DEBUG("unexpected key when inserting array value"); return -1; } if (virJSONValueArrayAppend(state->value, value) < 0) return -1; } break; default: VIR_DEBUG("unexpected value type, not a container"); return -1; } } return 0; } static int virJSONParserHandleNull(void *ctx) { virJSONParserPtr parser = ctx; virJSONValuePtr value = virJSONValueNewNull(); VIR_DEBUG("parser=%p", parser); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } return 1; } static int virJSONParserHandleBoolean(void *ctx, int boolean_) { virJSONParserPtr parser = ctx; virJSONValuePtr value = virJSONValueNewBoolean(boolean_); VIR_DEBUG("parser=%p boolean=%d", parser, boolean_); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } return 1; } static int virJSONParserHandleNumber(void *ctx, const char *s, yajl_size_t l) { virJSONParserPtr parser = ctx; char *str; virJSONValuePtr value; if (VIR_STRNDUP(str, s, l) < 0) return -1; value = virJSONValueNewNumber(str); VIR_FREE(str); VIR_DEBUG("parser=%p str=%s", parser, str); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } return 1; } static int virJSONParserHandleString(void *ctx, const unsigned char *stringVal, yajl_size_t stringLen) { virJSONParserPtr parser = ctx; virJSONValuePtr value = virJSONValueNewStringLen((const char *)stringVal, stringLen); VIR_DEBUG("parser=%p str=%p", parser, (const char *)stringVal); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } return 1; } static int virJSONParserHandleMapKey(void *ctx, const unsigned char *stringVal, yajl_size_t stringLen) { virJSONParserPtr parser = ctx; virJSONParserStatePtr state; VIR_DEBUG("parser=%p key=%p", parser, (const char *)stringVal); if (!parser->nstate) return 0; state = &parser->state[parser->nstate-1]; if (state->key) return 0; if (VIR_STRNDUP(state->key, (const char *)stringVal, stringLen) < 0) return 0; return 1; } static int virJSONParserHandleStartMap(void *ctx) { virJSONParserPtr parser = ctx; virJSONValuePtr value = virJSONValueNewObject(); VIR_DEBUG("parser=%p", parser); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } if (VIR_REALLOC_N(parser->state, parser->nstate + 1) < 0) { return 0; } parser->state[parser->nstate].value = value; parser->state[parser->nstate].key = NULL; parser->nstate++; return 1; } static int virJSONParserHandleEndMap(void *ctx) { virJSONParserPtr parser = ctx; virJSONParserStatePtr state; VIR_DEBUG("parser=%p", parser); if (!parser->nstate) return 0; state = &(parser->state[parser->nstate-1]); if (state->key) { VIR_FREE(state->key); return 0; } VIR_DELETE_ELEMENT(parser->state, parser->nstate - 1, parser->nstate); return 1; } static int virJSONParserHandleStartArray(void *ctx) { virJSONParserPtr parser = ctx; virJSONValuePtr value = virJSONValueNewArray(); VIR_DEBUG("parser=%p", parser); if (!value) return 0; if (virJSONParserInsertValue(parser, value) < 0) { virJSONValueFree(value); return 0; } if (VIR_REALLOC_N(parser->state, parser->nstate + 1) < 0) return 0; parser->state[parser->nstate].value = value; parser->state[parser->nstate].key = NULL; parser->nstate++; return 1; } static int virJSONParserHandleEndArray(void *ctx) { virJSONParserPtr parser = ctx; virJSONParserStatePtr state; VIR_DEBUG("parser=%p", parser); if (!(parser->nstate - parser->wrap)) return 0; state = &(parser->state[parser->nstate-1]); if (state->key) { VIR_FREE(state->key); return 0; } VIR_DELETE_ELEMENT(parser->state, parser->nstate - 1, parser->nstate); return 1; } static const yajl_callbacks parserCallbacks = { virJSONParserHandleNull, virJSONParserHandleBoolean, NULL, NULL, virJSONParserHandleNumber, virJSONParserHandleString, virJSONParserHandleStartMap, virJSONParserHandleMapKey, virJSONParserHandleEndMap, virJSONParserHandleStartArray, virJSONParserHandleEndArray }; /* XXX add an incremental streaming parser - yajl trivially supports it */ virJSONValuePtr virJSONValueFromString(const char *jsonstring) { yajl_handle hand; virJSONParser parser = { NULL, NULL, 0, 0 }; virJSONValuePtr ret = NULL; int rc; size_t len = strlen(jsonstring); # ifndef WITH_YAJL2 yajl_parser_config cfg = { 0, 1 }; /* Match yajl 2 default behavior */ virJSONValuePtr tmp; # endif VIR_DEBUG("string=%s", jsonstring); # ifdef WITH_YAJL2 hand = yajl_alloc(&parserCallbacks, NULL, &parser); # else hand = yajl_alloc(&parserCallbacks, &cfg, NULL, &parser); # endif if (!hand) { virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("Unable to create JSON parser")); goto cleanup; } /* Yajl 2 is nice enough to default to rejecting trailing garbage. * Yajl 1.0.12 has yajl_get_bytes_consumed to make that detection * simpler. But we're stuck with yajl 1.0.7 on RHEL 6, which * happily quits parsing at the end of a valid JSON construct, * with no visibility into how much more input remains. Wrapping * things in an array forces yajl to confess the truth. */ # ifdef WITH_YAJL2 rc = yajl_parse(hand, (const unsigned char *)jsonstring, len); # else rc = yajl_parse(hand, (const unsigned char *)"[", 1); parser.wrap = 1; if (VIR_YAJL_STATUS_OK(rc)) rc = yajl_parse(hand, (const unsigned char *)jsonstring, len); parser.wrap = 0; if (VIR_YAJL_STATUS_OK(rc)) rc = yajl_parse(hand, (const unsigned char *)"]", 1); # endif if (!VIR_YAJL_STATUS_OK(rc) || yajl_complete_parse(hand) != yajl_status_ok) { unsigned char *errstr = yajl_get_error(hand, 1, (const unsigned char*)jsonstring, strlen(jsonstring)); virReportError(VIR_ERR_INTERNAL_ERROR, _("cannot parse json %s: %s"), jsonstring, (const char*) errstr); yajl_free_error(hand, errstr); virJSONValueFree(parser.head); goto cleanup; } if (parser.nstate != 0) { virReportError(VIR_ERR_INTERNAL_ERROR, _("cannot parse json %s: unterminated string/map/array"), jsonstring); virJSONValueFree(parser.head); } else { ret = parser.head; # ifndef WITH_YAJL2 /* Undo the array wrapping above */ tmp = ret; ret = NULL; if (virJSONValueArraySize(tmp) > 1) virReportError(VIR_ERR_INTERNAL_ERROR, _("cannot parse json %s: too many items present"), jsonstring); else ret = virJSONValueArraySteal(tmp, 0); virJSONValueFree(tmp); # endif } cleanup: yajl_free(hand); if (parser.nstate) { size_t i; for (i = 0; i < parser.nstate; i++) VIR_FREE(parser.state[i].key); VIR_FREE(parser.state); } VIR_DEBUG("result=%p", ret); return ret; } static int virJSONValueToStringOne(virJSONValuePtr object, yajl_gen g) { size_t i; VIR_DEBUG("object=%p type=%d gen=%p", object, object->type, g); switch (object->type) { case VIR_JSON_TYPE_OBJECT: if (yajl_gen_map_open(g) != yajl_gen_status_ok) return -1; for (i = 0; i < object->data.object.npairs; i++) { if (yajl_gen_string(g, (unsigned char *)object->data.object.pairs[i].key, strlen(object->data.object.pairs[i].key)) != yajl_gen_status_ok) return -1; if (virJSONValueToStringOne(object->data.object.pairs[i].value, g) < 0) return -1; } if (yajl_gen_map_close(g) != yajl_gen_status_ok) return -1; break; case VIR_JSON_TYPE_ARRAY: if (yajl_gen_array_open(g) != yajl_gen_status_ok) return -1; for (i = 0; i < object->data.array.nvalues; i++) { if (virJSONValueToStringOne(object->data.array.values[i], g) < 0) return -1; } if (yajl_gen_array_close(g) != yajl_gen_status_ok) return -1; break; case VIR_JSON_TYPE_STRING: if (yajl_gen_string(g, (unsigned char *)object->data.string, strlen(object->data.string)) != yajl_gen_status_ok) return -1; break; case VIR_JSON_TYPE_NUMBER: if (yajl_gen_number(g, object->data.number, strlen(object->data.number)) != yajl_gen_status_ok) return -1; break; case VIR_JSON_TYPE_BOOLEAN: if (yajl_gen_bool(g, object->data.boolean) != yajl_gen_status_ok) return -1; break; case VIR_JSON_TYPE_NULL: if (yajl_gen_null(g) != yajl_gen_status_ok) return -1; break; default: return -1; } return 0; } char * virJSONValueToString(virJSONValuePtr object, bool pretty) { yajl_gen g; const unsigned char *str; char *ret = NULL; yajl_size_t len; # ifndef WITH_YAJL2 yajl_gen_config conf = { pretty ? 1 : 0, pretty ? " " : " "}; # endif VIR_DEBUG("object=%p", object); # ifdef WITH_YAJL2 g = yajl_gen_alloc(NULL); if (g) { yajl_gen_config(g, yajl_gen_beautify, pretty ? 1 : 0); yajl_gen_config(g, yajl_gen_indent_string, pretty ? " " : " "); yajl_gen_config(g, yajl_gen_validate_utf8, 1); } # else g = yajl_gen_alloc(&conf, NULL); # endif if (!g) { virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("Unable to create JSON formatter")); goto cleanup; } if (virJSONValueToStringOne(object, g) < 0) { virReportOOMError(); goto cleanup; } if (yajl_gen_get_buf(g, &str, &len) != yajl_gen_status_ok) { virReportOOMError(); goto cleanup; } ignore_value(VIR_STRDUP(ret, (const char *)str)); cleanup: yajl_gen_free(g); VIR_DEBUG("result=%s", NULLSTR(ret)); return ret; } #else virJSONValuePtr virJSONValueFromString(const char *jsonstring ATTRIBUTE_UNUSED) { virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("No JSON parser implementation is available")); return NULL; } char * virJSONValueToString(virJSONValuePtr object ATTRIBUTE_UNUSED, bool pretty ATTRIBUTE_UNUSED) { virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("No JSON parser implementation is available")); return NULL; } #endif /** * virJSONStringReformat: * @jsonstr: string to reformat * @pretty: use the pretty formatter * * Reformats a JSON string by passing it to the parser and then to the * formatter. If @pretty is true the JSON is formatted for human eye * compatibility. * * Returns the reformatted JSON string on success; NULL and a libvirt error on * failure. */ char * virJSONStringReformat(const char *jsonstr, bool pretty) { virJSONValuePtr json; char *ret; if (!(json = virJSONValueFromString(jsonstr))) return NULL; ret = virJSONValueToString(json, pretty); virJSONValueFree(json); return ret; } static int virJSONValueObjectDeflattenWorker(const char *key, virJSONValuePtr value, void *opaque) { virJSONValuePtr retobj = opaque; virJSONValuePtr newval = NULL; virJSONValuePtr existobj; char **tokens = NULL; size_t ntokens = 0; int ret = -1; /* non-nested keys only need to be copied */ if (!strchr(key, '.')) { if (virJSONValueIsObject(value)) newval = virJSONValueObjectDeflatten(value); else newval = virJSONValueCopy(value); if (!newval) return -1; if (virJSONValueObjectHasKey(retobj, key)) { virReportError(VIR_ERR_INVALID_ARG, _("can't deflatten colliding key '%s'"), key); goto cleanup; } if (virJSONValueObjectAppend(retobj, key, newval) < 0) goto cleanup; return 0; } if (!(tokens = virStringSplitCount(key, ".", 2, &ntokens))) goto cleanup; if (ntokens != 2) { virReportError(VIR_ERR_INVALID_ARG, _("invalid nested value key '%s'"), key); goto cleanup; } if (!(existobj = virJSONValueObjectGet(retobj, tokens[0]))) { if (!(existobj = virJSONValueNewObject())) goto cleanup; if (virJSONValueObjectAppend(retobj, tokens[0], existobj) < 0) goto cleanup; } else { if (!virJSONValueIsObject(existobj)) { virReportError(VIR_ERR_INVALID_ARG, "%s", _("mixing nested objects and values is forbidden in " "JSON deflattening")); goto cleanup; } } ret = virJSONValueObjectDeflattenWorker(tokens[1], value, existobj); cleanup: virStringListFreeCount(tokens, ntokens); virJSONValueFree(newval); return ret; } /** * virJSONValueObjectDeflatten: * * In some cases it's possible to nest JSON objects by prefixing object members * with the parent object name followed by the dot and then the attribute name * rather than directly using a nested value object (e.g qemu's JSON * pseudo-protocol in backing file definition). * * This function will attempt to reverse the process and provide a nested json * hierarchy so that the parsers can be kept simple and we still can use the * weird syntax some users might use. */ virJSONValuePtr virJSONValueObjectDeflatten(virJSONValuePtr json) { virJSONValuePtr deflattened; virJSONValuePtr ret = NULL; if (!(deflattened = virJSONValueNewObject())) return NULL; if (virJSONValueObjectForeachKeyValue(json, virJSONValueObjectDeflattenWorker, deflattened) < 0) goto cleanup; VIR_STEAL_PTR(ret, deflattened); cleanup: virJSONValueFree(deflattened); return ret; }