altos/lisp: Add incremental collection
[fw/altos] / src / lisp / ao_lisp_save.c
1 /*
2  * Copyright © 2016 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful, but
10  * WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * General Public License for more details.
13  */
14
15 #include <ao_lisp.h>
16
17 ao_poly
18 ao_lisp_save(struct ao_lisp_cons *cons)
19 {
20         if (!ao_lisp_check_argc(_ao_lisp_atom_save, cons, 0, 0))
21                 return AO_LISP_NIL;
22
23 #ifdef AO_LISP_SAVE
24         struct ao_lisp_os_save *os = (struct ao_lisp_os_save *) &ao_lisp_pool[AO_LISP_POOL];
25
26         ao_lisp_collect(AO_LISP_COLLECT_FULL);
27         os->atoms = ao_lisp_atom_poly(ao_lisp_atoms);
28         os->globals = ao_lisp_frame_poly(ao_lisp_frame_global);
29         os->const_checksum = ao_lisp_const_checksum;
30         os->const_checksum_inv = (uint16_t) ~ao_lisp_const_checksum;
31
32         if (ao_lisp_os_save())
33                 return _ao_lisp_atom_t;
34 #endif
35         return AO_LISP_NIL;
36 }
37
38 ao_poly
39 ao_lisp_restore(struct ao_lisp_cons *cons)
40 {
41         if (!ao_lisp_check_argc(_ao_lisp_atom_save, cons, 0, 0))
42                 return AO_LISP_NIL;
43
44 #ifdef AO_LISP_SAVE
45         struct ao_lisp_os_save save;
46         struct ao_lisp_os_save *os = (struct ao_lisp_os_save *) &ao_lisp_pool[AO_LISP_POOL];
47
48         if (!ao_lisp_os_restore_save(&save, AO_LISP_POOL))
49                 return ao_lisp_error(AO_LISP_INVALID, "header restore failed");
50
51         if (save.const_checksum != ao_lisp_const_checksum ||
52             save.const_checksum_inv != (uint16_t) ~ao_lisp_const_checksum)
53         {
54                 return ao_lisp_error(AO_LISP_INVALID, "image is corrupted or stale");
55         }
56
57         if (ao_lisp_os_restore()) {
58
59                 ao_lisp_atoms = ao_lisp_poly_atom(os->atoms);
60                 ao_lisp_frame_global = ao_lisp_poly_frame(os->globals);
61
62                 /* Clear the eval global variabls */
63                 ao_lisp_eval_clear_globals();
64
65                 /* Reset the allocator */
66                 ao_lisp_top = AO_LISP_POOL;
67                 ao_lisp_collect(AO_LISP_COLLECT_FULL);
68
69                 /* Re-create the evaluator stack */
70                 if (!ao_lisp_eval_restart())
71                         return AO_LISP_NIL;
72                 return _ao_lisp_atom_t;
73         }
74 #endif
75         return AO_LISP_NIL;
76 }