#include "lyb.h"
#include <assert.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "compat.h"
#include "context.h"
#include "dict.h"
#include "hash_table.h"
#include "in.h"
#include "in_internal.h"
#include "log.h"
#include "ly_common.h"
#include "parser_data.h"
#include "parser_internal.h"
#include "plugins_exts.h"
#include "plugins_exts/metadata.h"
#include "plugins_internal.h"
#include "set.h"
#include "tree.h"
#include "tree_data.h"
#include "tree_data_internal.h"
#include "tree_edit.h"
#include "tree_schema.h"
#include "tree_schema_internal.h"
#include "validation.h"
#include "xml.h"
static LY_ERR lyb_parse_siblings(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, ly_bool is_root,
struct lyd_node **first_p, struct ly_set *parsed);
static void
lyb_parse_ctx_free(struct lyd_ctx *lydctx)
{
struct lyd_lyb_ctx *ctx = (struct lyd_lyb_ctx *)lydctx;
if (!ctx) {
return;
}
lyd_ctx_free(lydctx);
free(ctx->parse_ctx);
free(ctx);
}
static void
lyb_read(void *buf, uint64_t count_bits, struct lylyb_parse_ctx *lybctx)
{
uint8_t count_buf_bits = 0, count_bits_remainder, new_buf;
uint64_t count_bytes;
assert(lybctx);
if (!count_bits) {
return;
}
if (!lybctx->shrink) {
count_bytes = LYPLG_BITS2BYTES(count_bits);
ly_in_read(lybctx->in, buf, count_bytes);
return;
}
if (lybctx->buf_bits) {
count_buf_bits = (count_bits > lybctx->buf_bits) ? lybctx->buf_bits : count_bits;
count_bits -= count_buf_bits;
}
count_bits_remainder = count_bits % 8;
count_bytes = count_bits / 8 + (count_bits_remainder ? 1 : 0);
if (count_bytes) {
if (buf) {
ly_in_read(lybctx->in, buf, count_bytes);
if (count_bits_remainder) {
new_buf = ((uint8_t *)buf)[count_bytes - 1];
((uint8_t *)buf)[count_bytes - 1] &= lyb_right_bit_mask(count_bits_remainder);
}
} else {
if (count_bits_remainder) {
ly_in_skip(lybctx->in, count_bytes - 1);
ly_in_read(lybctx->in, &new_buf, 1);
} else {
ly_in_skip(lybctx->in, count_bytes);
}
}
}
if (count_buf_bits) {
if (buf) {
lyb_prepend_bits(buf, LYPLG_BITS2BYTES(count_buf_bits + count_bits), lybctx->buf, count_buf_bits);
}
lybctx->buf_bits -= count_buf_bits;
lybctx->buf >>= count_buf_bits;
}
if (count_bits_remainder) {
assert(!lybctx->buf_bits);
lybctx->buf = new_buf >> count_bits_remainder;
lybctx->buf_bits = 8 - count_bits_remainder;
}
}
static void
lyb_read_count(uint32_t *count, struct lylyb_parse_ctx *lybctx)
{
uint8_t prefix = 0, pref_len = 0, byte_len = 0;
uint16_t count16 = 0;
if (!lybctx->shrink) {
byte_len = 2;
lyb_read(&count16, byte_len * 8, lybctx);
*count = le16toh(count16);
return;
}
do {
++pref_len;
prefix <<= 1;
lyb_read(&prefix, 1, lybctx);
} while (prefix & 0x1);
switch (pref_len) {
case 1:
break;
case 2:
lyb_read(count, 4, lybctx);
break;
case 3:
lyb_read(count, 5, lybctx);
break;
case 4:
lyb_read(count, 7, lybctx);
break;
case 5:
lyb_read(count, 11, lybctx);
break;
case 6:
lyb_read(count, 26, lybctx);
break;
default:
LOGINT(lybctx->ctx);
return;
}
*count = le32toh(*count);
}
static void
lyb_read_size(uint32_t *size, struct lylyb_parse_ctx *lybctx)
{
uint8_t prefix = 0, pref_len = 0, byte_len = 0;
if (!lybctx->shrink) {
byte_len = 4;
lyb_read(size, byte_len * 8, lybctx);
*size = le32toh(*size);
return;
}
do {
++pref_len;
prefix <<= 1;
lyb_read(&prefix, 1, lybctx);
} while (prefix & 0x1);
switch (pref_len) {
case 1:
lyb_read(size, 4, lybctx);
break;
case 2:
lyb_read(size, 6, lybctx);
break;
case 3:
lyb_read(size, 12, lybctx);
break;
case 4:
lyb_read(size, 32, lybctx);
break;
default:
LOGINT(lybctx->ctx);
return;
}
*size = le32toh(*size);
}
static LY_ERR
lyb_read_string(char **str, struct lylyb_parse_ctx *lybctx)
{
uint32_t str_len = 0;
*str = NULL;
lyb_read_size(&str_len, lybctx);
*str = malloc((str_len + 1) * sizeof **str);
LY_CHECK_ERR_RET(!*str, LOGMEM(lybctx->ctx), LY_EMEM);
if (str_len) {
lyb_read(*str, str_len * 8, lybctx);
}
(*str)[str_len] = '\0';
return LY_SUCCESS;
}
static LY_ERR
lyb_read_value(const struct lysc_type *type, uint8_t **val, uint64_t *val_size_bits, struct lylyb_parse_ctx *lybctx)
{
enum lyplg_lyb_size_type size_type;
uint32_t lyb_size_bits = 0;
uint64_t fixed_size_bits;
struct lysc_type_leafref *type_lf;
assert(type && val && val_size_bits && lybctx);
*val = NULL;
*val_size_bits = 0;
if (type->basetype == LY_TYPE_LEAFREF) {
type_lf = (struct lysc_type_leafref *)type;
LYSC_GET_TYPE_PLG(type_lf->realtype->plugin_ref)->lyb_size(type_lf->realtype, &size_type, &fixed_size_bits);
} else {
LYSC_GET_TYPE_PLG(type->plugin_ref)->lyb_size(type, &size_type, &fixed_size_bits);
}
if (size_type == LYPLG_LYB_SIZE_FIXED_BITS) {
*val_size_bits = fixed_size_bits;
} else {
lyb_read_size(&lyb_size_bits, lybctx);
*val_size_bits = lyb_size_bits;
if (size_type == LYPLG_LYB_SIZE_VARIABLE_BYTES) {
*val_size_bits *= 8;
}
}
*val = calloc(LYPLG_BITS2BYTES(*val_size_bits) + 1, sizeof **val);
LY_CHECK_ERR_RET(!*val, LOGMEM(lybctx->ctx), LY_EMEM);
if (*val_size_bits > 0) {
lyb_read(*val, *val_size_bits, lybctx);
}
return LY_SUCCESS;
}
static LY_ERR
lyb_check_mod_revision(struct lyd_lyb_ctx *lybctx, const char *mod_name, const char *mod_rev,
const struct lys_module *mod)
{
ly_bool rev;
if (lybctx->parse_opts & LYD_PARSE_LYB_SKIP_MODULE_CHECK) {
return LY_SUCCESS;
}
rev = (mod_rev[0] != '\0');
if ((rev && !mod->revision) || (!rev && mod->revision) || (rev && strcmp(mod->revision, mod_rev))) {
LOGERR(lybctx->parse_ctx->ctx, LY_EDENIED,
"Invalid context for LYB data parsing, module \"%s\" revision mismatch (printer: %s, parser: %s).",
mod_name,
rev ? mod_rev : "<none>",
mod->revision ? mod->revision : "<none>");
return LY_EDENIED;
}
return LY_SUCCESS;
}
static LY_ERR
lyb_check_mod_features(struct lyd_lyb_ctx *lybctx, const char *mod_name, const struct lys_module *mod)
{
LY_ERR rc = LY_SUCCESS;
struct lylyb_parse_ctx *pctx = lybctx->parse_ctx;
char *feature_name = NULL, **features = NULL, *en_feats_printer = NULL, *en_feats_parser = NULL;
uint32_t i, feature_count = 0, ctx_feature_count, len;
lyb_read_count(&feature_count, pctx);
if (feature_count) {
features = calloc(feature_count, sizeof *features);
LY_CHECK_ERR_RET(!features, LOGMEM(pctx->ctx), LY_EMEM);
}
for (i = 0; i < feature_count; i++) {
LY_CHECK_GOTO(rc = lyb_read_string(&feature_name, pctx), cleanup);
features[i] = feature_name;
feature_name = NULL;
}
if (lybctx->parse_opts & LYD_PARSE_LYB_SKIP_MODULE_CHECK) {
goto cleanup;
}
if (feature_count == LY_ARRAY_COUNT(mod->compiled->features)) {
for (i = 0; i < feature_count; i++) {
if (strcmp(mod->compiled->features[i], features[i])) {
break;
}
}
if (i == feature_count) {
goto cleanup;
}
}
len = 0;
for (i = 0; i < feature_count; i++) {
len += strlen(features[i]) + 2;
}
en_feats_printer = calloc(1, len + 1);
LY_CHECK_ERR_GOTO(!en_feats_printer, LOGMEM(pctx->ctx), cleanup);
for (i = 0; i < feature_count; i++) {
strcat(en_feats_printer, features[i]);
if (i + 1 < feature_count) {
strcat(en_feats_printer, ", ");
}
}
len = 0;
ctx_feature_count = LY_ARRAY_COUNT(mod->compiled->features);
for (i = 0; i < ctx_feature_count; i++) {
len += strlen(mod->compiled->features[i]) + 2;
}
en_feats_parser = calloc(1, len + 1);
LY_CHECK_ERR_GOTO(!en_feats_parser, LOGMEM(pctx->ctx), cleanup);
for (i = 0; i < ctx_feature_count; i++) {
strcat(en_feats_parser, mod->compiled->features[i]);
if (i + 1 < ctx_feature_count) {
strcat(en_feats_parser, ", ");
}
}
LOGERR(pctx->ctx, LY_EDENIED,
"Invalid context for LYB data parsing, module \"%s\" enabled feature mismatch (printer: {%s}, parser: {%s}).",
mod_name, en_feats_printer, en_feats_parser);
rc = LY_EDENIED;
cleanup:
free(en_feats_printer);
free(en_feats_parser);
for (i = 0; i < feature_count; i++) {
free(features[i]);
}
free(features);
return rc;
}
static LY_ERR
lyb_parse_module(struct lyd_lyb_ctx *lybctx, const struct lys_module **mod)
{
LY_ERR rc = LY_SUCCESS;
struct lylyb_parse_ctx *pctx = lybctx->parse_ctx;
char *mod_name = NULL, mod_rev[LY_REV_SIZE];
uint16_t rev;
mod_rev[0] = '\0';
LY_CHECK_RET(lyb_read_string(&mod_name, pctx));
assert(mod_name[0]);
*mod = ly_ctx_get_module_implemented(pctx->ctx, mod_name);
if (!*mod) {
LOGERR(pctx->ctx, LY_EDENIED, "Invalid context for LYB data parsing, module \"%s\" not found.", mod_name);
rc = LY_EDENIED;
goto cleanup;
}
lyb_read(&rev, 2 * 8, pctx);
if (rev) {
sprintf(mod_rev, "%04u-%02u-%02u", ((rev & LYB_REV_YEAR_MASK) >> LYB_REV_YEAR_SHIFT) + LYB_REV_YEAR_OFFSET,
(rev & LYB_REV_MONTH_MASK) >> LYB_REV_MONTH_SHIFT, rev & LYB_REV_DAY_MASK);
}
LY_CHECK_GOTO(rc = lyb_check_mod_revision(lybctx, mod_name, mod_rev, *mod), cleanup);
LY_CHECK_GOTO(rc = lyb_check_mod_features(lybctx, mod_name, *mod), cleanup);
cleanup:
free(mod_name);
return rc;
}
static LY_ERR
lyb_parse_module_idx(struct lyd_lyb_ctx *lybctx, const struct lys_module **mod)
{
LY_ERR rc = LY_SUCCESS;
struct lylyb_parse_ctx *pctx = lybctx->parse_ctx;
const struct lys_module *m = NULL;
uint32_t idx = 0;
lyb_read_count(&idx, pctx);
if (idx >= pctx->ctx->modules.count) {
LOGERR(pctx->ctx, LY_EINT, "Invalid context for LYB data parsing, module with index %" PRIu32 " not found.",
idx);
rc = LY_EINT;
goto cleanup;
}
m = pctx->ctx->modules.objs[idx];
if (!m->implemented) {
LOGERR(pctx->ctx, LY_EINT, "Invalid context for LYB data parsing, module \"%s%s%s\" not implemented.",
m->name, m->revision ? "@" : "", m->revision ? m->revision : "");
rc = LY_EINT;
goto cleanup;
}
cleanup:
*mod = m;
return rc;
}
static LY_ERR
lyb_parse_metadata(struct lyd_lyb_ctx *lybctx, const struct lysc_node *sparent, uint32_t metadata_count,
struct lyd_meta **meta)
{
LY_ERR rc = LY_SUCCESS;
ly_bool dynamic;
uint32_t i, count = 0;
uint64_t value_size_bits;
char *meta_name = NULL;
uint8_t *value;
const struct lys_module *mod;
struct lysc_ext_instance *ant;
const struct lysc_type *ant_type;
if (metadata_count == UINT32_MAX) {
lyb_read_count(&count, lybctx->parse_ctx);
} else {
count = metadata_count;
}
for (i = 0; i < count; ++i) {
if (lybctx->parse_ctx->shrink) {
LY_CHECK_GOTO(rc = lyb_parse_module_idx(lybctx, &mod), cleanup);
} else {
LY_CHECK_GOTO(rc = lyb_parse_module(lybctx, &mod), cleanup);
}
rc = lyb_read_string(&meta_name, lybctx->parse_ctx);
LY_CHECK_GOTO(rc, cleanup);
ant = lyd_get_meta_annotation(mod, meta_name, strlen(meta_name));
if (!ant) {
LOGVAL(lybctx->parse_ctx->ctx, NULL, LYVE_REFERENCE, "Annotation definition for metadata \"%s:%s\" not found.",
mod->name, meta_name);
rc = LY_EINT;
goto cleanup;
}
lyplg_ext_get_storage(ant, LY_STMT_TYPE, sizeof ant_type, (const void **)&ant_type);
rc = lyb_read_value(ant_type, &value, &value_size_bits, lybctx->parse_ctx);
LY_CHECK_GOTO(rc, cleanup);
dynamic = 1;
rc = lyd_parser_create_meta((struct lyd_ctx *)lybctx, NULL, meta, mod, meta_name, strlen(meta_name), value,
value_size_bits, &dynamic, LY_VALUE_LYB, NULL, LYD_HINT_DATA, sparent, NULL);
free(meta_name);
meta_name = NULL;
if (dynamic) {
free(value);
dynamic = 0;
}
LY_CHECK_GOTO(rc, cleanup);
}
cleanup:
free(meta_name);
if (rc) {
lyd_free_meta_siblings(*meta);
*meta = NULL;
}
return rc;
}
static LY_ERR
lyb_parse_prefix_data(struct lylyb_parse_ctx *lybctx, LY_VALUE_FORMAT format, void **prefix_data)
{
LY_ERR ret = LY_SUCCESS;
uint32_t count = 0, i;
struct ly_set *set = NULL;
struct lyxml_ns *ns = NULL;
switch (format) {
case LY_VALUE_XML:
lyb_read_count(&count, lybctx);
LY_CHECK_GOTO(ret = ly_set_new(&set), cleanup);
for (i = 0; i < count; ++i) {
ns = calloc(1, sizeof *ns);
LY_CHECK_GOTO(ret = lyb_read_string(&ns->prefix, lybctx), cleanup);
if (!strlen(ns->prefix)) {
free(ns->prefix);
ns->prefix = NULL;
}
LY_CHECK_GOTO(ret = lyb_read_string(&ns->uri, lybctx), cleanup);
LY_CHECK_GOTO(ret = ly_set_add(set, ns, 1, NULL), cleanup);
ns = NULL;
}
*prefix_data = set;
break;
case LY_VALUE_JSON:
case LY_VALUE_LYB:
break;
default:
LOGINT(lybctx->ctx);
ret = LY_EINT;
break;
}
cleanup:
if (ret) {
ly_free_prefix_data(format, set);
if (ns) {
free(ns->prefix);
free(ns->uri);
free(ns);
}
}
return ret;
}
static LY_ERR
lyb_parse_attributes(struct lylyb_parse_ctx *lybctx, struct lyd_attr **attr)
{
LY_ERR ret = LY_SUCCESS;
uint32_t count = 0, i;
struct lyd_attr *attr2 = NULL;
char *prefix = NULL, *module_name = NULL, *name = NULL, *value = NULL;
ly_bool dynamic = 0;
LY_VALUE_FORMAT format = 0;
void *val_prefix_data = NULL;
uint8_t byte = 0;
lyb_read_count(&count, lybctx);
for (i = 0; i < count; ++i) {
ret = lyb_read_string(&prefix, lybctx);
LY_CHECK_GOTO(ret, cleanup);
if (!prefix[0]) {
free(prefix);
prefix = NULL;
}
ret = lyb_read_string(&module_name, lybctx);
LY_CHECK_GOTO(ret, cleanup);
if (!module_name[0]) {
free(module_name);
module_name = NULL;
}
ret = lyb_read_string(&name, lybctx);
LY_CHECK_GOTO(ret, cleanup);
lyb_read(&byte, LYB_OPAQ_FORMAT_BITS, lybctx);
if (byte == LYB_OPAQ_FORMAT_XML) {
format = LY_VALUE_XML;
} else {
assert(byte == LYB_OPAQ_FORMAT_JSON);
format = LY_VALUE_JSON;
}
ret = lyb_parse_prefix_data(lybctx, format, &val_prefix_data);
LY_CHECK_GOTO(ret, cleanup);
ret = lyb_read_string(&value, lybctx);
LY_CHECK_ERR_GOTO(ret, ly_free_prefix_data(format, val_prefix_data), cleanup);
dynamic = 1;
ret = lyd_create_attr(NULL, &attr2, lybctx->ctx, name, strlen(name), prefix, ly_strlen(prefix), module_name,
ly_strlen(module_name), value, ly_strlen(value), &dynamic, format, val_prefix_data, LYD_HINT_DATA);
LY_CHECK_GOTO(ret, cleanup);
free(prefix);
prefix = NULL;
free(module_name);
module_name = NULL;
free(name);
name = NULL;
assert(!dynamic);
value = NULL;
if (!*attr) {
*attr = attr2;
}
LY_CHECK_GOTO(ret, cleanup);
}
cleanup:
free(prefix);
free(module_name);
free(name);
if (dynamic) {
free(value);
}
if (ret) {
lyd_free_attr_siblings(lybctx->ctx, *attr);
*attr = NULL;
}
return ret;
}
static LY_ERR
lyb_read_hashes(struct lylyb_parse_ctx *lybctx, LYB_HASH *hash, uint8_t *hash_count)
{
uint8_t i = 0, j;
lyb_read(&hash[0], sizeof *hash * 8, lybctx);
if (!hash[0]) {
*hash_count = i + 1;
return LY_SUCCESS;
}
for (i = 0; !(hash[0] & (LYB_HASH_COLLISION_ID >> i)); ++i) {
if (i > LYB_HASH_BITS) {
LOGINT_RET(lybctx->ctx);
}
}
hash[i] = hash[0];
for (j = i; j; --j) {
lyb_read(&hash[j - 1], sizeof *hash * 8, lybctx);
assert(hash[j - 1] & (LYB_HASH_COLLISION_ID >> (j - 1)));
assert(!(hash[j - 1] & (LYB_HASH_MASK << (LYB_HASH_BITS - (j - 1)))));
}
*hash_count = i + 1;
return LY_SUCCESS;
}
static int
lyb_is_schema_hash_match(struct lysc_node *sibling, LYB_HASH *hash, uint8_t hash_count)
{
LYB_HASH sibling_hash;
uint8_t i;
for (i = 0; i < hash_count; ++i) {
sibling_hash = lyb_get_hash(sibling, i);
if (sibling_hash != hash[i]) {
return 0;
}
}
return 1;
}
static LY_ERR
lyb_parse_schema_hash(struct lyd_lyb_ctx *lybctx, const struct lysc_node *sparent, const struct lys_module *mod,
const struct lysc_node **snode)
{
LY_ERR r;
const struct lysc_node *sibling;
LYB_HASH hash[LYB_HASH_BITS - 1];
uint32_t getnext_opts;
uint8_t hash_count;
*snode = NULL;
LY_CHECK_RET(lyb_read_hashes(lybctx->parse_ctx, hash, &hash_count));
if (!hash[0]) {
return LY_SUCCESS;
}
getnext_opts = lybctx->int_opts & LYD_INTOPT_REPLY ? LYS_GETNEXT_OUTPUT : 0;
sibling = NULL;
while (1) {
sibling = lys_getnext(sibling, sparent, mod ? mod->compiled : NULL, getnext_opts);
if (!sibling) {
break;
}
if (lyb_is_schema_hash_match((struct lysc_node *)sibling, hash, hash_count)) {
break;
}
}
if (!sibling) {
if (mod) {
LOGERR(lybctx->parse_ctx->ctx, LY_EINT, "Failed to find matching hash for a top-level node from \"%s\".",
mod->name);
} else {
assert(sparent);
LOGERR(lybctx->parse_ctx->ctx, LY_EINT, "Failed to find matching hash for a child node of \"%s\".",
sparent->name);
}
return LY_EINT;
} else if (sibling && (r = lyd_parser_check_schema((struct lyd_ctx *)lybctx, sibling))) {
return r;
}
*snode = sibling;
return LY_SUCCESS;
}
static LY_ERR
lyb_parse_schema_ext(struct lyd_lyb_ctx *lybctx, const struct lyd_node *parent, const char *mod_name,
const struct lysc_node **snode, struct lysc_ext_instance **ext)
{
LY_ERR rc = LY_SUCCESS, r;
char *name = NULL;
LY_CHECK_GOTO(rc = lyb_read_string(&name, lybctx->parse_ctx), cleanup);
r = lys_find_child_node_ext(lybctx->parse_ctx->ctx, NULL, parent, NULL, mod_name, mod_name ? strlen(mod_name) : 0,
LY_VALUE_JSON, NULL, name, strlen(name), 0, snode, ext);
if (r == LY_ENOT) {
LOGERR(lybctx->parse_ctx->ctx, LY_EINVAL, "Failed to parse node \"%s\" as extension instance data.", name);
rc = LY_EINVAL;
goto cleanup;
} else if (r) {
rc = r;
goto cleanup;
}
if (!((*snode)->module->ctx->opts & LY_CTX_LYB_HASHES)) {
LY_CHECK_GOTO(rc = ly_ctx_set_options((*snode)->module->ctx, LY_CTX_LYB_HASHES), cleanup);
}
cleanup:
free(name);
return rc;
}
static void
lyb_insert_node(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, struct lyd_node *node, struct lyd_node **first_p,
struct ly_set *parsed)
{
lyd_insert_node(parent, first_p, node,
lybctx->parse_opts & LYD_PARSE_ORDERED ? LYD_INSERT_NODE_LAST : LYD_INSERT_NODE_DEFAULT);
while (!parent && (*first_p)->prev->next) {
*first_p = (*first_p)->prev;
}
if (parsed) {
ly_set_add(parsed, node, 1, NULL);
}
}
static void
lyb_finish_opaq(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, struct lyd_attr **attr, struct lyd_node **node,
struct lyd_node **first_p, struct ly_set *parsed)
{
struct lyd_attr *iter;
assert(!(*node)->schema);
LY_LIST_FOR(*attr, iter) {
iter->parent = (struct lyd_node_opaq *)*node;
}
((struct lyd_node_opaq *)*node)->attr = *attr;
*attr = NULL;
lyb_insert_node(lybctx, parent, *node, first_p, parsed);
*node = NULL;
}
static LY_ERR
lyb_finish_node(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, uint32_t flags, const struct lysc_ext_instance *ext,
struct lyd_meta **meta, struct lyd_node **node, struct lyd_node **first_p, struct ly_set *parsed)
{
struct lyd_meta *m;
(*node)->flags = flags;
LY_LIST_FOR(*meta, m) {
m->parent = *node;
}
(*node)->meta = *meta;
*meta = NULL;
lyb_insert_node(lybctx, parent, *node, first_p, parsed);
if (!(lybctx->parse_opts & LYD_PARSE_ONLY)) {
if (ext) {
LY_CHECK_RET(lyd_validate_tree_ext(*node, ext, &lybctx->ext_val));
}
LY_CHECK_RET(lyd_validate_node_ext(*node, &lybctx->ext_val));
}
*node = NULL;
return LY_SUCCESS;
}
static LY_ERR
lyb_parse_node_header(struct lyd_lyb_ctx *lybctx, const struct lysc_node *sparent, uint32_t metadata_count,
uint32_t *flags, struct lyd_meta **meta)
{
LY_CHECK_RET(lyb_parse_metadata(lybctx, sparent, metadata_count, meta));
if (!lybctx->parse_ctx->shrink) {
lyb_read(flags, LYB_DATA_NODE_FLAG_BITS, lybctx->parse_ctx);
*flags = le32toh(*flags);
} else {
*flags = LYD_NEW;
}
return LY_SUCCESS;
}
static LY_ERR
lyb_create_term(struct lyd_lyb_ctx *lybctx, const struct lysc_node *snode, const struct lyd_node *lnode,
struct lyd_node **node)
{
LY_ERR rc;
ly_bool dynamic;
uint8_t *value;
uint64_t value_size_bits;
LY_CHECK_RET(lyb_read_value(((struct lysc_node_leaf *)snode)->type, &value, &value_size_bits, lybctx->parse_ctx));
dynamic = 1;
rc = lyd_parser_create_term((struct lyd_ctx *)lybctx, snode, lnode, value, value_size_bits, &dynamic, LY_VALUE_LYB,
NULL, LYD_HINT_DATA, node);
if (dynamic) {
free(value);
}
if (rc) {
lyd_free_tree(*node);
*node = NULL;
}
return rc;
}
static LY_ERR
lyb_validate_node_inner(struct lyd_lyb_ctx *lybctx, struct lyd_node *node)
{
LY_ERR rc = LY_SUCCESS;
if (!(lybctx->parse_opts & LYD_PARSE_ONLY)) {
rc = lyd_parser_validate_new_implicit((struct lyd_ctx *)lybctx, node);
}
return rc;
}
static LY_ERR
lyb_parse_node_opaq(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, struct lyd_node **first_p, struct ly_set *parsed)
{
LY_ERR ret;
struct lyd_node *node = NULL;
struct lyd_attr *attr = NULL;
char *value = NULL, *name = NULL, *prefix = NULL, *module_key = NULL;
ly_bool dynamic = 0;
uint8_t byte = 0;
LY_VALUE_FORMAT format;
void *val_prefix_data = NULL;
const struct ly_ctx *ctx = lybctx->parse_ctx->ctx;
ret = lyb_parse_attributes(lybctx->parse_ctx, &attr);
LY_CHECK_GOTO(ret, cleanup);
ret = lyb_read_string(&prefix, lybctx->parse_ctx);
LY_CHECK_GOTO(ret, cleanup);
ret = lyb_read_string(&module_key, lybctx->parse_ctx);
LY_CHECK_GOTO(ret, cleanup);
ret = lyb_read_string(&name, lybctx->parse_ctx);
LY_CHECK_GOTO(ret, cleanup);
ret = lyb_read_string(&value, lybctx->parse_ctx);
LY_CHECK_GOTO(ret, cleanup);
dynamic = 1;
lyb_read(&byte, LYB_OPAQ_FORMAT_BITS, lybctx->parse_ctx);
if (byte == LYB_OPAQ_FORMAT_XML) {
format = LY_VALUE_XML;
} else {
assert(byte == LYB_OPAQ_FORMAT_JSON);
format = LY_VALUE_JSON;
}
ret = lyb_parse_prefix_data(lybctx->parse_ctx, format, &val_prefix_data);
LY_CHECK_GOTO(ret, cleanup);
ret = lyd_create_opaq(ctx, name, strlen(name), prefix, ly_strlen(prefix), module_key, ly_strlen(module_key),
value, strlen(value), &dynamic, format, val_prefix_data, LYD_HINT_DATA, &node);
LY_CHECK_ERR_GOTO(ret, ly_free_prefix_data(format, val_prefix_data), cleanup);
ret = lyb_parse_siblings(lybctx, node, 0, NULL, NULL);
LY_CHECK_GOTO(ret, cleanup);
if (lybctx->parse_opts & LYD_PARSE_OPAQ) {
lyb_finish_opaq(lybctx, parent, &attr, &node, first_p, parsed);
assert(!attr && !node);
}
cleanup:
free(prefix);
free(module_key);
free(name);
if (dynamic) {
free(value);
}
lyd_free_attr_siblings(ctx, attr);
lyd_free_tree(node);
return ret;
}
static LY_ERR
lyb_parse_node_any(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, const struct lysc_node *snode,
const struct lysc_ext_instance *ext, struct lyd_node **first_p, struct ly_set *parsed)
{
LY_ERR rc = LY_SUCCESS;
struct lyd_node *node = NULL, *tree = NULL;
struct lyd_meta *meta = NULL;
char *value = NULL;
uint32_t flags = 0, value_type = 0, prev_parse_opts, prev_int_opts;
const struct ly_ctx *ctx = lybctx->parse_ctx->ctx;
rc = lyb_parse_node_header(lybctx, snode, UINT32_MAX, &flags, &meta);
LY_CHECK_GOTO(rc, cleanup);
lyb_read_count((uint32_t *)&value_type, lybctx->parse_ctx);
if (value_type == 0) {
prev_parse_opts = lybctx->parse_opts;
prev_int_opts = lybctx->int_opts;
lybctx->parse_opts = LYD_PARSE_ONLY | LYD_PARSE_OPAQ;
lybctx->int_opts = LYD_INTOPT_ANY | LYD_INTOPT_WITH_SIBLINGS;
rc = lyb_parse_siblings(lybctx, NULL, 0, &tree, NULL);
lybctx->parse_opts = prev_parse_opts;
lybctx->int_opts = prev_int_opts;
LY_CHECK_GOTO(rc, cleanup);
rc = lyd_create_any(snode, tree, NULL, 0, 1, 0, &node);
LY_CHECK_GOTO(rc, cleanup);
tree = NULL;
} else if (value_type == 1) {
rc = lyb_read_string(&value, lybctx->parse_ctx);
LY_CHECK_GOTO(rc, cleanup);
rc = lyd_create_any(snode, NULL, value, 0, 1, 0, &node);
LY_CHECK_GOTO(rc, cleanup);
value = NULL;
} else {
LOGINT(ctx);
rc = LY_EINT;
goto cleanup;
}
rc = lyb_finish_node(lybctx, parent, flags, ext, &meta, &node, first_p, parsed);
LY_CHECK_GOTO(rc, cleanup);
cleanup:
lyd_free_meta_siblings(meta);
lyd_free_siblings(tree);
free(value);
lyd_free_tree(node);
return rc;
}
static LY_ERR
lyb_parse_node_inner(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, const struct lysc_node *snode,
const struct lysc_ext_instance *ext, struct lyd_node **first_p, struct ly_set *parsed)
{
LY_ERR rc = LY_SUCCESS;
struct lyd_node *node = NULL;
struct lyd_meta *meta = NULL;
uint32_t flags = 0;
rc = lyb_parse_node_header(lybctx, snode, UINT32_MAX, &flags, &meta);
LY_CHECK_GOTO(rc, cleanup);
rc = lyd_create_inner(snode, &node);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_parse_siblings(lybctx, node, 0, NULL, NULL);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_validate_node_inner(lybctx, node);
LY_CHECK_GOTO(rc, cleanup);
if (snode->nodetype & (LYS_RPC | LYS_ACTION | LYS_NOTIF)) {
lybctx->op_node = node;
}
rc = lyb_finish_node(lybctx, parent, flags, ext, &meta, &node, first_p, parsed);
LY_CHECK_GOTO(rc, cleanup);
cleanup:
lyd_free_meta_siblings(meta);
lyd_free_tree(node);
return rc;
}
static LY_ERR
lyb_parse_node_leaf(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, const struct lysc_node *snode,
const struct lysc_ext_instance *ext, uint32_t metadata_count, struct lyd_node **first_p, struct ly_set *parsed)
{
LY_ERR rc = LY_SUCCESS;
struct lyd_node *node = NULL;
struct lyd_meta *meta = NULL;
uint32_t flags = 0;
rc = lyb_parse_node_header(lybctx, snode, metadata_count, &flags, &meta);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_create_term(lybctx, snode, parent, &node);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_finish_node(lybctx, parent, flags, ext, &meta, &node, first_p, parsed);
LY_CHECK_GOTO(rc, cleanup);
cleanup:
lyd_free_meta_siblings(meta);
lyd_free_tree(node);
return rc;
}
static LY_ERR
lyb_parse_node_leaflist(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, const struct lysc_node *snode,
const struct lysc_ext_instance *ext, struct lyd_node **first_p, struct ly_set *parsed)
{
uint32_t metadata_count;
while (1) {
metadata_count = 0;
lyb_read_count(&metadata_count, lybctx->parse_ctx);
if (metadata_count == LYB_METADATA_END_COUNT) {
break;
}
LY_CHECK_RET(lyb_parse_node_leaf(lybctx, parent, snode, ext, metadata_count, first_p, parsed));
}
return LY_SUCCESS;
}
static LY_ERR
lyb_parse_node_list(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, const struct lysc_node *snode,
const struct lysc_ext_instance *ext, struct lyd_node **first_p, struct ly_set *parsed)
{
LY_ERR rc = LY_SUCCESS;
struct lyd_node *node = NULL;
struct lyd_meta *meta = NULL;
uint32_t flags = 0, metadata_count;
while (1) {
metadata_count = 0;
lyb_read_count(&metadata_count, lybctx->parse_ctx);
if (metadata_count == LYB_METADATA_END_COUNT) {
break;
}
rc = lyb_parse_node_header(lybctx, snode, metadata_count, &flags, &meta);
LY_CHECK_GOTO(rc, cleanup);
rc = lyd_create_inner(snode, &node);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_parse_siblings(lybctx, node, 0, NULL, NULL);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_validate_node_inner(lybctx, node);
LY_CHECK_GOTO(rc, cleanup);
if (snode->nodetype & (LYS_RPC | LYS_ACTION | LYS_NOTIF)) {
lybctx->op_node = node;
}
rc = lyb_finish_node(lybctx, parent, flags, ext, &meta, &node, first_p, parsed);
LY_CHECK_GOTO(rc, cleanup);
}
cleanup:
lyd_free_meta_siblings(meta);
lyd_free_tree(node);
return rc;
}
static LY_ERR
lyb_parse_node(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, struct lyd_node **first_p,
struct ly_set *parsed)
{
LY_ERR rc = LY_SUCCESS;
const struct lysc_node *snode;
struct lysc_ext_instance *ext = NULL;
const struct lys_module *mod;
enum lylyb_node_type lyb_type = 0;
lyb_read(&lyb_type, LYB_NODE_TYPE_BITS, lybctx->parse_ctx);
lyb_type = le32toh(lyb_type);
switch (lyb_type) {
case LYB_NODE_END:
rc = LY_ENOT;
goto cleanup;
case LYB_NODE_TOP:
if (lybctx->parse_ctx->shrink) {
LY_CHECK_GOTO(rc = lyb_parse_module_idx(lybctx, &mod), cleanup);
} else {
LY_CHECK_GOTO(rc = lyb_parse_module(lybctx, &mod), cleanup);
}
LY_CHECK_GOTO(rc = lyb_parse_schema_hash(lybctx, NULL, mod, &snode), cleanup);
break;
case LYB_NODE_CHILD:
LY_CHECK_GOTO(rc = lyb_parse_schema_hash(lybctx, parent ? parent->schema : NULL, NULL, &snode), cleanup);
break;
case LYB_NODE_EXT:
LY_CHECK_GOTO(rc = lyb_parse_module(lybctx, &mod), cleanup);
LY_CHECK_GOTO(rc = lyb_parse_schema_ext(lybctx, parent, mod->name, &snode, &ext), cleanup);
break;
}
if (!snode) {
rc = lyb_parse_node_opaq(lybctx, parent, first_p, parsed);
} else if (snode->nodetype & LYS_LEAFLIST) {
rc = lyb_parse_node_leaflist(lybctx, parent, snode, ext, first_p, parsed);
} else if (snode->nodetype == LYS_LIST) {
rc = lyb_parse_node_list(lybctx, parent, snode, ext, first_p, parsed);
} else if (snode->nodetype & LYD_NODE_ANY) {
rc = lyb_parse_node_any(lybctx, parent, snode, ext, first_p, parsed);
} else if (snode->nodetype & LYD_NODE_INNER) {
rc = lyb_parse_node_inner(lybctx, parent, snode, ext, first_p, parsed);
} else {
rc = lyb_parse_node_leaf(lybctx, parent, snode, ext, UINT32_MAX, first_p, parsed);
}
LY_CHECK_GOTO(rc, cleanup);
cleanup:
return rc;
}
static LY_ERR
lyb_parse_siblings(struct lyd_lyb_ctx *lybctx, struct lyd_node *parent, ly_bool is_root, struct lyd_node **first_p,
struct ly_set *parsed)
{
LY_ERR r;
while (!(r = lyb_parse_node(lybctx, parent, first_p, parsed))) {
assert(!lybctx->parse_ctx->empty_hash);
if (is_root && !(lybctx->int_opts & LYD_INTOPT_WITH_SIBLINGS)) {
break;
}
}
LY_CHECK_RET(r != LY_ENOT, r);
return LY_SUCCESS;
}
static LY_ERR
lyb_parse_context_hash(struct lyd_lyb_ctx *lybctx)
{
uint32_t data_hash, cur_hash;
struct lylyb_parse_ctx *pctx = lybctx->parse_ctx;
data_hash = 0;
lyb_read((uint8_t *)&data_hash, LYB_HEADER_CTX_HASH_BITS, pctx);
data_hash = le32toh(data_hash);
if (!data_hash) {
pctx->empty_hash = 1;
} else if (lybctx->parse_ctx->shrink) {
cur_hash = lyb_truncate_hash_nonzero(ly_ctx_get_modules_hash(pctx->ctx), LYB_HEADER_CTX_HASH_BITS);
if (data_hash != cur_hash) {
LOGERR(pctx->ctx, LY_EINVAL, "Different current LYB context modules hash compared to the one stored in the "
"LYB data (0x%" PRIx32 " != 0x%" PRIx32 ").", data_hash, cur_hash);
return LY_EINVAL;
}
}
return LY_SUCCESS;
}
static LY_ERR
lyb_parse_header(struct lyd_lyb_ctx *lybctx)
{
uint8_t byte, is_shrink, remaining_bit_count;
struct lylyb_parse_ctx *pctx = lybctx->parse_ctx;
pctx->shrink = 1;
byte = 0;
lyb_read(&byte, LYB_HEADER_VERSION_BITS, pctx);
if (byte != LYB_HEADER_VERSION_NUM) {
LOGERR(pctx->ctx, LY_EINVAL, "Invalid LYB format version \"0x%" PRIx8 "\", expected \"0x%x\".",
byte, LYB_HEADER_VERSION_NUM);
return LY_EINVAL;
}
byte = 0;
lyb_read(&byte, LYB_HEADER_HASH_ALG_BITS, pctx);
if (byte != LYB_HEADER_HASH_ALG_NUM) {
LOGERR(pctx->ctx, LY_EINVAL, "Different LYB format hash algorithm \"0x%" PRIx8 "\" used, expected \"0x%x\".",
byte, LYB_HEADER_HASH_ALG_NUM);
return LY_EINVAL;
}
byte = 0;
lyb_read(&byte, LYB_HEADER_SHRINK_FLAG_BITS, pctx);
is_shrink = byte;
remaining_bit_count = 8 - (LYB_HEADER_VERSION_BITS + LYB_HEADER_HASH_ALG_BITS + LYB_HEADER_SHRINK_FLAG_BITS);
byte = 0;
lyb_read(&byte, remaining_bit_count, pctx);
if (byte) {
LOGERR(pctx->ctx, LY_EINVAL, "Invalid reserved bits in LYB header.");
return LY_EINVAL;
}
pctx->shrink = is_shrink;
return LY_SUCCESS;
}
LY_ERR
lyd_parse_lyb(const struct ly_ctx *ctx, struct lyd_node *parent, struct lyd_node **first_p, struct ly_in *in,
uint32_t parse_opts, uint32_t val_opts, uint32_t int_opts, struct ly_set *parsed, struct lyd_ctx **lydctx_p)
{
LY_ERR rc = LY_SUCCESS;
struct lyd_lyb_ctx *lybctx = NULL;
assert(!(parse_opts & ~LYD_PARSE_OPTS_MASK));
assert(!(val_opts & ~LYD_VALIDATE_OPTS_MASK));
if (!(ctx->opts & LY_CTX_LYB_HASHES)) {
LY_CHECK_GOTO(rc = ly_ctx_set_options((struct ly_ctx *)ctx, LY_CTX_LYB_HASHES), cleanup);
}
lybctx = calloc(1, sizeof *lybctx);
LY_CHECK_ERR_GOTO(!lybctx, LOGMEM(ctx); rc = LY_EMEM, cleanup);
lybctx->parse_ctx = calloc(1, sizeof *lybctx->parse_ctx);
LY_CHECK_ERR_GOTO(!lybctx->parse_ctx, LOGMEM(ctx); rc = LY_EMEM, cleanup);
lybctx->parse_ctx->in = in;
lybctx->parse_ctx->ctx = ctx;
lybctx->parse_opts = parse_opts;
lybctx->val_opts = val_opts;
lybctx->int_opts = int_opts;
lybctx->free = lyb_parse_ctx_free;
LY_CHECK_GOTO(rc = lyd_parser_find_operation(parent, int_opts, &lybctx->op_node), cleanup);
rc = lyb_parse_header(lybctx);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_parse_context_hash(lybctx);
LY_CHECK_GOTO(rc, cleanup);
rc = lyb_parse_siblings(lybctx, parent, 1, first_p, parsed);
LY_CHECK_GOTO(rc, cleanup);
if ((int_opts & (LYD_INTOPT_RPC | LYD_INTOPT_ACTION | LYD_INTOPT_NOTIF | LYD_INTOPT_REPLY)) && !lybctx->op_node) {
LOGVAL(ctx, NULL, LYVE_DATA, "Missing the operation node.");
rc = LY_EVALID;
goto cleanup;
}
cleanup:
assert(!lybctx || !(parse_opts & LYD_PARSE_ONLY) || (!lybctx->node_types.count && !lybctx->meta_types.count &&
!lybctx->node_when.count));
if (rc) {
lyb_parse_ctx_free((struct lyd_ctx *)lybctx);
} else {
*lydctx_p = (struct lyd_ctx *)lybctx;
}
return rc;
}