commit 9922ea9b647fbcd16d10c0faad98dfa49e35d6c6
author aPiecek <piecek@cesnet.cz> Mon Apr 12 07:59:20 2021 +0200
committer Radek Krejci <rkrejci@cesnet.cz> Mon Apr 12 15:07:32 2021 +0200
tree 5bc67b3640d5e33ba2173c0ddd7b3dce61dd21e9
parent 474ee435b5633efc9d7ff4c85c0aa5780d626c51

context FEATURE added LY_CTX_SET_PRIV_PARSED

With this option, libyang sets ::lysc_node.priv pointers
as a reference to the corresponding ::lysp_node.
This was introduced due to the printer tree module (RFC 8340),
which needs both a compiled and a parsed tree to print.

src/context.c

diff --git a/src/context.c b/src/context.c
index 7ef2cde..de6e4c3 100644
--- a/src/context.c
+++ b/src/context.c
@@ -473,12 +473,30 @@
 API LY_ERR
 ly_ctx_set_options(struct ly_ctx *ctx, uint16_t option)
 {
+    LY_ERR lyrc;
     LY_CHECK_ARG_RET(ctx, ctx, LY_EINVAL);
     LY_CHECK_ERR_RET(option & LY_CTX_NO_YANGLIBRARY, LOGARG(ctx, option), LY_EINVAL);
+    lyrc = LY_SUCCESS;
+
+    if (!(ctx->flags & LY_CTX_SET_PRIV_PARSED) && (option & LY_CTX_SET_PRIV_PARSED)) {
+        ctx->flags |= LY_CTX_SET_PRIV_PARSED;
+        if ((lyrc = lys_recompile(ctx, 0))) {
+            ly_ctx_unset_options(ctx, LY_CTX_SET_PRIV_PARSED);
+        }
+    }
 
     /* set the option(s) */
-    ctx->flags |= option;
+    if (!lyrc) {
+        ctx->flags |= option;
+    }
 
+    return lyrc;
+}
+
+static LY_ERR
+lysc_node_clear_priv_dfs_cb(struct lysc_node *node, void *UNUSED(data), ly_bool *UNUSED(dfs_continue))
+{
+    node->priv = NULL;
     return LY_SUCCESS;
 }
 
@@ -488,6 +506,22 @@
     LY_CHECK_ARG_RET(ctx, ctx, LY_EINVAL);
     LY_CHECK_ERR_RET(option & LY_CTX_NO_YANGLIBRARY, LOGARG(ctx, option), LY_EINVAL);
 
+    if ((ctx->flags & LY_CTX_SET_PRIV_PARSED) && (option & LY_CTX_SET_PRIV_PARSED)) {
+        struct lys_module *mod;
+        uint32_t index;
+
+        index = 0;
+        while ((mod = (struct lys_module *)ly_ctx_get_module_iter(ctx, &index))) {
+            /* ignore modules that didn't compile */
+            if (!mod->compiled) {
+                continue;
+            }
+
+            /* set NULL for all ::lysc_node.priv pointers in module */
+            lysc_module_dfs_full(mod, lysc_node_clear_priv_dfs_cb, NULL);
+        }
+    }
+
     /* unset the option(s) */
     ctx->flags &= ~option;