#include <string.h>
#include "script/expr.h"
-
-static char *string_duplicate(const char *str,
- const char *str_end)
-{
- if (str_end != NULL && str > str_end) {
- return NULL;
- }
-
- const size_t n = str_end == NULL ? strlen(str) : (size_t) (str_end - str);
- char *dup_str = malloc(sizeof(char) * (n + 1));
- if (dup_str == NULL) {
- return NULL;
- }
-
- memcpy(dup_str, str, n);
- return dup_str;
-}
+#include "script/gc.h"
+#include "str.h"
struct Expr atom_as_expr(struct Atom *atom)
{
return expr;
}
-void print_atom_as_sexpr(struct Atom *atom)
+void print_atom_as_sexpr(FILE *stream, struct Atom *atom)
{
assert(atom);
switch (atom->type) {
case ATOM_SYMBOL:
- printf("%s", atom->sym);
+ fprintf(stream, "%s", atom->sym);
break;
case ATOM_NUMBER:
- printf("%f", atom->num);
+ fprintf(stream, "%ld", atom->num);
break;
case ATOM_STRING:
- printf("\"%s\"", atom->str);
+ fprintf(stream, "\"%s\"", atom->str);
+ break;
+
+ case ATOM_NATIVE:
+ fprintf(stream, "<native>");
break;
}
}
-void print_cons_as_sexpr(struct Cons *head)
+void print_cons_as_sexpr(FILE *stream, struct Cons *head)
{
assert(head);
struct Cons *cons = head;
- printf("(");
- print_expr_as_sexpr(cons->car);
+ fprintf(stream, "(");
+ print_expr_as_sexpr(stream, cons->car);
while (cons->cdr.type == EXPR_CONS) {
cons = cons->cdr.cons;
- printf(" ");
- print_expr_as_sexpr(cons->car);
+ fprintf(stream, " ");
+ print_expr_as_sexpr(stream, cons->car);
}
if (cons->cdr.atom->type != ATOM_SYMBOL ||
strcmp("nil", cons->cdr.atom->sym) != 0) {
- printf(" . ");
- print_expr_as_sexpr(cons->cdr);
+ fprintf(stream, " . ");
+ print_expr_as_sexpr(stream, cons->cdr);
}
- printf(")");
+ fprintf(stream, ")");
}
-void print_expr_as_sexpr(struct Expr expr)
+void print_expr_as_sexpr(FILE *stream, struct Expr expr)
{
- /* TODO(#296): print_expr_as_sexpr doesn't support lists */
switch (expr.type) {
case EXPR_ATOM:
- print_atom_as_sexpr(expr.atom);
+ print_atom_as_sexpr(stream, expr.atom);
break;
case EXPR_CONS:
- print_cons_as_sexpr(expr.cons);
+ print_cons_as_sexpr(stream, expr.cons);
break;
case EXPR_VOID:
}
}
-struct Cons *create_cons(struct Expr car, struct Expr cdr)
+struct Cons *create_cons(Gc *gc, struct Expr car, struct Expr cdr)
{
struct Cons *cons = malloc(sizeof(struct Cons));
if (cons == NULL) {
cons->car = car;
cons->cdr = cdr;
+ if (gc_add_expr(gc, cons_as_expr(cons)) < 0) {
+ free(cons);
+ return NULL;
+ }
+
return cons;
}
void destroy_cons(struct Cons *cons)
{
- destroy_expr(cons->car);
- destroy_expr(cons->cdr);
free(cons);
}
-struct Atom *create_number_atom(float num)
+struct Atom *create_number_atom(Gc *gc, long int num)
{
struct Atom *atom = malloc(sizeof(struct Atom));
if (atom == NULL) {
}
atom->type = ATOM_NUMBER;
atom->num = num;
+
+ if (gc_add_expr(gc, atom_as_expr(atom)) < 0) {
+ free(atom);
+ return NULL;
+ }
+
return atom;
}
-struct Atom *create_string_atom(const char *str, const char *str_end)
+struct Atom *create_string_atom(Gc *gc, const char *str, const char *str_end)
{
struct Atom *atom = malloc(sizeof(struct Atom));
+
if (atom == NULL) {
- return NULL;
+ goto error;
}
+
atom->type = ATOM_STRING;
atom->str = string_duplicate(str, str_end);
+
if (atom->str == NULL) {
- free(atom);
- return NULL;
+ goto error;
+ }
+
+ if (gc_add_expr(gc, atom_as_expr(atom)) < 0) {
+ goto error;
}
return atom;
+
+error:
+ if (atom != NULL) {
+ if (atom->str != NULL) {
+ free(atom->str);
+ }
+ free(atom);
+ }
+
+ return NULL;
}
-struct Atom *create_symbol_atom(const char *sym, const char *sym_end)
+struct Atom *create_symbol_atom(Gc *gc, const char *sym, const char *sym_end)
{
struct Atom *atom = malloc(sizeof(struct Atom));
+
if (atom == NULL) {
- return NULL;
+ goto error;
}
+
atom->type = ATOM_SYMBOL;
atom->sym = string_duplicate(sym, sym_end);
+
if (atom->sym == NULL) {
+ goto error;
+ }
+
+ if (gc_add_expr(gc, atom_as_expr(atom)) < 0) {
+ goto error;
+ }
+
+ return atom;
+
+error:
+ if (atom != NULL) {
+ if (atom->sym != NULL) {
+ free(atom->sym);
+ }
free(atom);
- return NULL;
+ }
+
+ return NULL;
+}
+
+struct Atom *create_native_atom(Gc *gc, NativeFunction fun, void *param)
+{
+ struct Atom *atom = malloc(sizeof(struct Atom));
+
+ if (atom == NULL) {
+ goto error;
+ }
+
+ atom->type = ATOM_NATIVE;
+ atom->native.fun = fun;
+ atom->native.param = param;
+
+ if (gc_add_expr(gc, atom_as_expr(atom)) < 0) {
+ goto error;
}
return atom;
+
+error:
+ if (atom != NULL) {
+ free(atom);
+ }
+
+ return NULL;
}
void destroy_atom(struct Atom *atom)
free(atom->str);
} break;
+ case ATOM_NATIVE:
case ATOM_NUMBER: {
/* Nothing */
} break;
free(atom);
}
+
+static int atom_as_sexpr(struct Atom *atom, char *output, size_t n)
+{
+ assert(atom);
+ assert(output);
+
+ switch (atom->type) {
+ case ATOM_SYMBOL:
+ return snprintf(output, n, "%s", atom->sym);
+
+ case ATOM_NUMBER:
+ return snprintf(output, n, "%ld", atom->num);
+
+ case ATOM_STRING:
+ return snprintf(output, n, "\"%s\"", atom->str);
+
+ case ATOM_NATIVE:
+ return snprintf(output, n, "<native>");
+ }
+
+ return 0;
+}
+
+static int cons_as_sexpr(struct Cons *head, char *output, size_t n)
+{
+ assert(head);
+ assert(output);
+
+ /* TODO(#378): cons_as_sexpr does not handle encoding errors of snprintf */
+
+ struct Cons *cons = head;
+
+ int m = (int) n;
+
+ int c = snprintf(output, n, "(");
+ if (m - c <= c) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->car, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+
+ while (cons->cdr.type == EXPR_CONS) {
+ cons = cons->cdr.cons;
+
+ c += snprintf(output + c, (size_t) (m - c), " ");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->car, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+ }
+
+ if (cons->cdr.atom->type != ATOM_SYMBOL ||
+ strcmp("nil", cons->cdr.atom->sym) != 0) {
+
+ c += snprintf(output + c, (size_t) (m - c), " . ");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ c += expr_as_sexpr(cons->cdr, output + c, (size_t) (m - c));
+ if (m - c <= 0) {
+ return c;
+ }
+ }
+
+ c += snprintf(output + c, (size_t) (m - c), ")");
+ if (m - c <= 0) {
+ return c;
+ }
+
+ return c;
+}
+
+int expr_as_sexpr(struct Expr expr, char *output, size_t n)
+{
+ switch(expr.type) {
+ case EXPR_ATOM:
+ return atom_as_sexpr(expr.atom, output, n);
+
+ case EXPR_CONS:
+ return cons_as_sexpr(expr.cons, output, n);
+
+ case EXPR_VOID:
+ return 0;
+ }
+
+ return 0;
+}
projects / nothing.git / blobdiff