lua: use uci_foreach_element_safe in uci_lua_foreach() - patch by jow

[project/uci.git] / list.c

diff --git a/list.c b/list.c
index 0b3a3f2..a336b84 100644 (file)
--- a/list.c
+++ b/list.c
@@ -358,6 +358,18 @@ done:
 }
 
 int
+uci_lookup_next(struct uci_context *ctx, struct uci_element **e, struct uci_list *list, const char *name)
+{
+       UCI_HANDLE_ERR(ctx);
+
+       *e = uci_lookup_list(list, name);
+       if (!*e)
+               UCI_THROW(ctx, UCI_ERR_NOTFOUND);
+
+       return 0;
+}
+
+int
 uci_lookup_ptr(struct uci_context *ctx, struct uci_ptr *ptr, char *str, bool extended)
 {
        struct uci_element *e;
@@ -371,7 +383,11 @@ uci_lookup_ptr(struct uci_context *ctx, struct uci_ptr *ptr, char *str, bool ext
        ptr->flags |= UCI_LOOKUP_DONE;
 
        /* look up the package first */
-       e = uci_lookup_list(&ctx->root, ptr->package);
+       if (ptr->p)
+               e = &ptr->p->e;
+       else
+               e = uci_lookup_list(&ctx->root, ptr->package);
+
        if (!e) {
                UCI_INTERNAL(uci_load, ctx, ptr->package, &ptr->p);
                if (!ptr->p)
@@ -382,15 +398,21 @@ uci_lookup_ptr(struct uci_context *ctx, struct uci_ptr *ptr, char *str, bool ext
                ptr->last = e;
        }
 
-       if (!ptr->section)
+       if (!ptr->section && !ptr->s)
                goto complete;
 
        /* if the section name validates as a regular name, pass through
         * to the regular uci_lookup function call */
-       if (ptr->flags & UCI_LOOKUP_EXTENDED)
-               e = uci_lookup_ext_section(ctx, ptr);
-       else
+       if (ptr->s) {
+               e = &ptr->s->e;
+       } else if (ptr->flags & UCI_LOOKUP_EXTENDED) {
+               if (extended)
+                       e = uci_lookup_ext_section(ctx, ptr);
+               else
+                       UCI_THROW(ctx, UCI_ERR_INVAL);
+       } else {
                e = uci_lookup_list(&ptr->p->sections, ptr->section);
+       }
 
        if (!e)
                goto abort;