altos/lisp: Split out read debug, add memory validation

[fw/altos] / src / lisp / ao_lisp_cons.c

diff --git a/src/lisp/ao_lisp_cons.c b/src/lisp/ao_lisp_cons.c
index b75ffaa0d611ffa5427e7b191d4d68d996e74534..06e9d361e65630bf7b35d66ca2fef0eb4416db4e 100644 (file)
--- a/src/lisp/ao_lisp_cons.c
+++ b/src/lisp/ao_lisp_cons.c
@@ -14,18 +14,22 @@
 
 #include "ao_lisp.h"
 
-#define OFFSET(a)      ((int) ((uint8_t *) (a) - ao_lisp_const))
-
 static void cons_mark(void *addr)
 {
        struct ao_lisp_cons     *cons = addr;
 
        for (;;) {
+               ao_poly cdr = cons->cdr;
+
                ao_lisp_poly_mark(cons->car, 1);
-               cons = ao_lisp_poly_cons(cons->cdr);
-               if (!cons)
+               if (!cdr)
                        break;
-               if (ao_lisp_mark_memory(cons, sizeof (struct ao_lisp_cons)))
+               if (ao_lisp_poly_type(cdr) != AO_LISP_CONS) {
+                       ao_lisp_poly_mark(cdr, 1);
+                       break;
+               }
+               cons = ao_lisp_poly_cons(cdr);
+               if (ao_lisp_mark_memory(&ao_lisp_cons_type, cons))
                        break;
        }
 }
@@ -44,17 +48,29 @@ static void cons_move(void *addr)
                return;
 
        for (;;) {
-               struct ao_lisp_cons     *cdr;
-               int                     ret;
+               ao_poly                 cdr;
+               struct ao_lisp_cons     *c;
+               int     ret;
 
+               MDBG_MOVE("cons_move start %d (%d, %d)\n",
+                         MDBG_OFFSET(cons), MDBG_OFFSET(ao_lisp_ref(cons->car)), MDBG_OFFSET(ao_lisp_ref(cons->cdr)));
                (void) ao_lisp_poly_move(&cons->car, 1);
-               cdr = ao_lisp_poly_cons(cons->cdr);
-               ret = ao_lisp_move_memory((void **) &cdr, sizeof (struct ao_lisp_cons));
-               if (cdr != ao_lisp_poly_cons(cons->cdr))
-                       cons->cdr = ao_lisp_cons_poly(cdr);
+               cdr = cons->cdr;
+               if (!cdr)
+                       break;
+               if (ao_lisp_poly_type(cdr) != AO_LISP_CONS) {
+                       (void) ao_lisp_poly_move(&cons->cdr, 1);
+                       break;
+               }
+               c = ao_lisp_poly_cons(cdr);
+               ret = ao_lisp_move_memory(&ao_lisp_cons_type, (void **) &c);
+               if (c != ao_lisp_poly_cons(cons->cdr))
+                       cons->cdr = ao_lisp_cons_poly(c);
+               MDBG_MOVE("cons_move end %d (%d, %d)\n",
+                         MDBG_OFFSET(cons), MDBG_OFFSET(ao_lisp_ref(cons->car)), MDBG_OFFSET(ao_lisp_ref(cons->cdr)));
                if (ret)
                        break;
-               cons = cdr;
+               cons = c;
        }
 }
 
@@ -62,27 +78,68 @@ const struct ao_lisp_type ao_lisp_cons_type = {
        .mark = cons_mark,
        .size = cons_size,
        .move = cons_move,
+       .name = "cons",
 };
 
+struct ao_lisp_cons *ao_lisp_cons_free_list;
+
 struct ao_lisp_cons *
-ao_lisp_cons_cons(ao_poly car, struct ao_lisp_cons *cdr)
+ao_lisp_cons_cons(ao_poly car, ao_poly cdr)
 {
        struct ao_lisp_cons     *cons;
 
-       ao_lisp_root_add(&ao_lisp_cons_type, &cdr);
-       ao_lisp_root_poly_add(&car);
-       cons = ao_lisp_alloc(sizeof (struct ao_lisp_cons));
-       ao_lisp_root_clear(&car);
-       ao_lisp_root_clear(&cdr);
-       if (!cons)
-               return NULL;
+       if (ao_lisp_cons_free_list) {
+               cons = ao_lisp_cons_free_list;
+               ao_lisp_cons_free_list = ao_lisp_poly_cons(cons->cdr);
+       } else {
+               ao_lisp_poly_stash(0, car);
+               ao_lisp_poly_stash(1, cdr);
+               cons = ao_lisp_alloc(sizeof (struct ao_lisp_cons));
+               car = ao_lisp_poly_fetch(0);
+               cdr = ao_lisp_poly_fetch(1);
+               if (!cons)
+                       return NULL;
+       }
        cons->car = car;
-       cons->cdr = ao_lisp_cons_poly(cdr);
+       cons->cdr = cdr;
        return cons;
 }
 
+struct ao_lisp_cons *
+ao_lisp_cons_cdr(struct ao_lisp_cons *cons)
+{
+       ao_poly cdr = cons->cdr;
+       if (cdr == AO_LISP_NIL)
+               return NULL;
+       if (ao_lisp_poly_type(cdr) != AO_LISP_CONS) {
+               (void) ao_lisp_error(AO_LISP_INVALID, "improper list");
+               return NULL;
+       }
+       return ao_lisp_poly_cons(cdr);
+}
+
+ao_poly
+ao_lisp__cons(ao_poly car, ao_poly cdr)
+{
+       return ao_lisp_cons_poly(ao_lisp_cons_cons(car, cdr));
+}
+
 void
-ao_lisp_cons_print(ao_poly c)
+ao_lisp_cons_free(struct ao_lisp_cons *cons)
+{
+#if DBG_FREE_CONS
+       ao_lisp_cons_check(cons);
+#endif
+       while (cons) {
+               ao_poly cdr = cons->cdr;
+               cons->cdr = ao_lisp_cons_poly(ao_lisp_cons_free_list);
+               ao_lisp_cons_free_list = cons;
+               cons = ao_lisp_poly_cons(cdr);
+       }
+}
+
+void
+ao_lisp_cons_write(ao_poly c)
 {
        struct ao_lisp_cons *cons = ao_lisp_poly_cons(c);
        int     first = 1;
@@ -90,20 +147,27 @@ ao_lisp_cons_print(ao_poly c)
        while (cons) {
                if (!first)
                        printf(" ");
-               ao_lisp_poly_print(cons->car);
-               cons = ao_lisp_poly_cons(cons->cdr);
-               first = 0;
+               ao_lisp_poly_write(cons->car);
+               c = cons->cdr;
+               if (ao_lisp_poly_type(c) == AO_LISP_CONS) {
+                       cons = ao_lisp_poly_cons(c);
+                       first = 0;
+               } else {
+                       printf(" . ");
+                       ao_lisp_poly_write(c);
+                       cons = NULL;
+               }
        }
        printf(")");
 }
 
 void
-ao_lisp_cons_patom(ao_poly c)
+ao_lisp_cons_display(ao_poly c)
 {
        struct ao_lisp_cons *cons = ao_lisp_poly_cons(c);
 
        while (cons) {
-               ao_lisp_poly_patom(cons->car);
+               ao_lisp_poly_display(cons->car);
                cons = ao_lisp_poly_cons(cons->cdr);
        }
 }