X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fcontext_dump.c;h=f1386710aa3eebab1d91cb92a7ad54634b4346ab;hp=93b8f325d4006a05b73ded6aaae7d8974b907ec9;hb=b2200987b147fa850c5b001620cb2535b2546b7d;hpb=b4a3c33fc143fd57288fcfe0878e7a22eeaf61bf

diff --git a/src/context_dump.c b/src/context_dump.c
index 93b8f32..f138671 100644
--- a/src/context_dump.c
+++ b/src/context_dump.c
@@ -4,17 +4,48 @@
 #include "libjit.h"
 #include "libffi.h"
 
-static inline void dump_decl_type(int fd, decl_type *t) {
-	const char *pre;
+static inline void dump_level(int fd, unsigned level) {
+	dprintf(fd, "%.*s", level > 30 ? 30 : level,
+			"\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t\t");
+}
+
+static inline void dump_decl_arg(int fd, decl_arg *darg, unsigned level);
+static inline void dump_struct_args(int fd, decl_args *args, unsigned level);
+static inline void dump_enum_items(int fd, decl_enum_items *items, unsigned level);
 
+#define is_anon_type(name, type) !strncmp(name, type "@", sizeof(type))
+
+static inline void dump_decl_type(int fd, decl_type *t, unsigned level) {
 	switch (t->type) {
+	case PSI_T_POINTER:
+		dprintf(fd, "%s *", t->name);
+		return;
+
 	case PSI_T_STRUCT:
-		pre = "struct ";
+		dprintf(fd, "struct ");
+		if (!strncmp(t->name, "struct@", sizeof("struct"))) {
+			dump_struct_args(fd, t->real.strct->args, level);
+			return;
+		}
+		break;
+
+	case PSI_T_ENUM:
+		dprintf(fd, "enum ");
+		if (!strncmp(t->name, "enum@", sizeof("enum"))) {
+			dump_enum_items(fd, t->real.enm->items, level);
+			return;
+		}
+		break;
+
+	case PSI_T_UNION:
+		dprintf(fd, "union ");
+		if (!strncmp(t->name, "union@", sizeof("union"))) {
+			dump_struct_args(fd, t->real.unn->args, level);
+			return;
+		}
 		break;
-	default:
-		pre = "";
 	}
-	dprintf(fd, "%s%s", pre, t->name);
+	dprintf(fd, "%s", t->name);
 }
 
 static inline void dump_decl_var(int fd, decl_var *v) {
@@ -24,14 +55,31 @@ static inline void dump_decl_var(int fd, decl_var *v) {
 	}
 }
 
-static inline void dump_decl_arg(int fd, decl_arg *a) {
-	dump_decl_type(fd, a->type);
-	dprintf(fd, " ");
-	dump_decl_var(fd, a->var);
-}
-
-static inline void dump_level(int fd, unsigned level) {
-	dprintf(fd, "%.*s", level > 10 ? 10 : level, "\t\t\t\t\t\t\t\t\t\t");
+static inline void dump_decl_arg(int fd, decl_arg *a, unsigned level) {
+	if (a->type->type == PSI_T_FUNCTION) {
+		dump_decl_type(fd, a->type->real.func->func->type, level);
+		dprintf(fd, " (");
+		dump_decl_var(fd, a->var);
+		dprintf(fd, ")(");
+		if (a->type->real.func->args) {
+			size_t j;
+
+			for (j = 0; j < a->type->real.func->args->count; ++j) {
+				if (j) {
+					dprintf(fd, ", ");
+				}
+				dump_decl_arg(fd, a->type->real.func->args->args[j], level+1);
+			}
+			if (a->type->real.func->args->varargs) {
+				dprintf(fd, ", ...");
+			}
+		}
+		dprintf(fd, ")");
+	} else {
+		dump_decl_type(fd, a->type, level);
+		dprintf(fd, " ");
+		dump_decl_var(fd, a->var);
+	}
 }
 
 static inline void dump_num_exp(int fd, num_exp *exp) {
@@ -46,6 +94,9 @@ static inline void dump_num_exp(int fd, num_exp *exp) {
 		case PSI_T_NAME:
 			dump_decl_var(fd, exp->u.dvar);
 			break;
+		case PSI_T_ENUM:
+			dprintf(fd, "%s", exp->u.enm->name);
+			break;
 		EMPTY_SWITCH_DEFAULT_CASE();
 		}
 		if (exp->operand) {
@@ -93,10 +144,10 @@ static inline void dump_impl_set_value(int fd, set_value *set, unsigned level, i
 		dprintf(fd, ", ");
 		dump_num_exp(fd, set->num);
 	}
-	if (set->inner && set->func->type != PSI_T_ELLIPSIS) {
+	if (set->inner && set->inner->vals && set->func->type != PSI_T_ELLIPSIS) {
 		dprintf(fd, ",\n");
-		for (i = 0; i < set->count; ++i) {
-			dump_impl_set_value(fd, set->inner[i], level+1, i == (set->count - 1));
+		for (i = 0; i < set->inner->count; ++i) {
+			dump_impl_set_value(fd, set->inner->vals[i], level+1, i == (set->inner->count - 1));
 		}
 		/* only if inner stmts, i.e. with new lines, were dumped */
 		dump_level(fd, level);
@@ -108,53 +159,131 @@ static inline void dump_impl_set_value(int fd, set_value *set, unsigned level, i
 	}
 }
 
-static inline void dump_typedef(int fd, decl_typedef *tdef) {
+static inline void dump_typedef(int fd, decl_arg *tdef) {
 	dprintf(fd, "typedef ");
-	dump_decl_type(fd, tdef->type);
-	dprintf(fd, " %s%s;", tdef->type->type == PSI_T_POINTER ? "*":"",
-			tdef->alias);
+	dump_decl_arg(fd, tdef, 0);
+	dprintf(fd, ";");
 }
 
 static inline void dump_typedefs(int fd, decl_typedefs *defs) {
 	size_t i;
 
 	for (i = 0; i < defs->count; ++i) {
-		decl_typedef *tdef = defs->list[i];
+		decl_arg *tdef = defs->list[i];
 
 		dump_typedef(fd, tdef);
 		dprintf(fd, "\n");
 	}
 }
 
-static inline void dump_struct(int fd, decl_struct *strct) {
+static inline void dump_struct_args(int fd, decl_args *args, unsigned level) {
 	size_t j;
 
-	dprintf(fd, "struct %s::(%zu)", strct->name, strct->size);
-	if (strct->args && strct->args->count) {
-		dprintf(fd, " {\n");
-		for (j = 0; j < strct->args->count; ++j) {
-			decl_arg *sarg = strct->args->args[j];
+	dprintf(fd, " {\n");
+	if (args) {
+		++level;
+		for (j = 0; j < args->count; ++j) {
+			decl_arg *sarg = args->args[j];
 
-			dprintf(fd, "\t");
-			dump_decl_arg(fd, sarg);
+			dump_level(fd, level);
+			dump_decl_arg(fd, sarg, level);
 			dprintf(fd, "::(%zu, %zu);\n", sarg->layout->pos, sarg->layout->len);
 		}
-		dprintf(fd, "}");
+		--level;
+	}
+	dump_level(fd, level);
+	dprintf(fd, "}");
+}
+
+static inline void dump_struct(int fd, decl_struct *strct) {
+	dprintf(fd, "struct %s::(%zu, %zu)", strct->name, strct->align, strct->size);
+	if (strct->args && strct->args->count) {
+		dump_struct_args(fd, strct->args, 0);
 	} else {
 		dprintf(fd, ";");
 	}
 }
+
 static inline void dump_structs(int fd, decl_structs *structs) {
 	size_t i;
 
 	for (i = 0; i < structs->count; ++i) {
 		decl_struct *strct = structs->list[i];
 
-		dump_struct(fd, strct);
-		dprintf(fd, "\n");
+		if (!is_anon_type(strct->name, "struct")) {
+			dump_struct(fd, strct);
+			dprintf(fd, "\n");
+		}
+	}
+}
+
+static inline void dump_union(int fd, decl_union *unn) {
+	size_t j;
+
+	dprintf(fd, "union %s::(%zu, %zu) {\n", unn->name, unn->align, unn->size);
+	for (j = 0; j < unn->args->count; ++j) {
+		decl_arg *uarg = unn->args->args[j];
+
+		dprintf(fd, "\t");
+		dump_decl_arg(fd, uarg, 0);
+		dprintf(fd, "::(%zu, %zu);\n", uarg->layout->pos, uarg->layout->len);
+	}
+	dprintf(fd, "}");
+}
+
+static inline void dump_unions(int fd, decl_unions *unions) {
+	size_t i;
+
+	for (i = 0; i < unions->count; ++i) {
+		decl_union *unn = unions->list[i];
+
+		if (!is_anon_type(unn->name, "union")) {
+			dump_union(fd, unn);
+			dprintf(fd, "\n");
+		}
+	}
+}
+
+static inline void dump_enum_items(int fd, decl_enum_items *items, unsigned level) {
+	size_t j;
+
+	if (items) {
+		++level;
+		for (j = 0; j < items->count; ++j) {
+			decl_enum_item *i = items->list[j];
+
+			if (j) {
+				dprintf(fd, ",\n");
+			}
+			dump_level(fd, level);
+			dprintf(fd, "%s", i->name);
+			if (i->num && i->num != &i->inc) {
+				dprintf(fd, " = ");
+				dump_num_exp(fd, i->num);
+			}
+		}
+		--level;
 	}
 }
 
+static inline void dump_enum(int fd, decl_enum *e) {
+	dprintf(fd, "enum %s {\n", e->name);
+	dump_enum_items(fd, e->items, 0);
+	dprintf(fd, "\n}");
+}
+
+static inline void dump_enums(int fd, decl_enums *enums) {
+	size_t i;
+
+	for (i = 0; i < enums->count; ++i) {
+		decl_enum *e = enums->list[i];
+
+		if (!is_anon_type(e->name, "enum")) {
+			dump_enum(fd, e);
+			dprintf(fd, "\n");
+		}
+	}
+}
 static inline void dump_constant(int fd, constant *cnst) {
 	dprintf(fd, "const %s %s = ", cnst->type->name, cnst->name);
 	if (cnst->val->type == PSI_T_QUOTED_STRING) {
@@ -179,14 +308,14 @@ static inline void dump_decl(int fd, decl *decl) {
 	size_t j;
 
 	dprintf(fd, "%s ", decl->abi->convention);
-	dump_decl_arg(fd, decl->func);
+	dump_decl_arg(fd, decl->func, 0);
 	dprintf(fd, "(");
 	if (decl->args) {
 		for (j = 0; j < decl->args->count; ++j) {
 			if (j) {
 				dprintf(fd, ", ");
 			}
-			dump_decl_arg(fd, decl->args->args[j]);
+			dump_decl_arg(fd, decl->args->args[j], 0);
 		}
 		if (decl->args->varargs) {
 			dprintf(fd, ", ...");
@@ -214,7 +343,7 @@ static inline void dump_impl_func(int fd, impl_func *func) {
 		for (j = 0; j < func->args->count; ++j) {
 			impl_arg *iarg = func->args->args[j];
 
-			dprintf(fd, "%s%s %s$%s",
+			dprintf(fd, "%s%s %s%s",
 					j ? ", " : "",
 					iarg->type->name,
 					iarg->var->reference ? "&" : "",
@@ -226,7 +355,7 @@ static inline void dump_impl_func(int fd, impl_func *func) {
 		if (func->args->vararg.name) {
 			impl_arg *vararg = func->args->vararg.name;
 
-			dprintf(fd, ", %s %s...$%s",
+			dprintf(fd, ", %s %s...%s",
 					vararg->type->name,
 					vararg->var->reference ? "&" : "",
 					vararg->var->name);
@@ -255,8 +384,23 @@ static inline void dump_impl_let_stmt(int fd, let_stmt *let) {
 			dump_num_exp(fd, let->val->data.alloc->size);
 			dprintf(fd, ")");
 			break;
+		case PSI_LET_CALLBACK:
+			dprintf(fd, "callback %s(%s(", let->val->data.callback->func->name,
+					let->val->data.callback->func->var->name);
+			if (let->val->data.callback->args) {
+				size_t i, c = let->val->data.callback->args->count;
+
+				dprintf(fd, "\n");
+				for (i = 0; i < c; ++i) {
+					set_value *set = let->val->data.callback->args->vals[i];
+					dump_impl_set_value(fd, set, 2, i + 1 == c);
+				}
+				dprintf(fd, "\t");
+			}
+			dprintf(fd, "));");
+			break;
 		case PSI_LET_FUNC:
-			dprintf(fd, "%s($%s)", let->val->data.func->name,
+			dprintf(fd, "%s(%s)", let->val->data.func->name,
 					let->val->data.func->var->name);
 			break;
 		case PSI_LET_NUMEXP:
@@ -275,7 +419,7 @@ static inline void dump_impl_return_stmt(int fd, return_stmt *ret) {
 }
 
 static inline void dump_impl_set_stmt(int fd, set_stmt *set) {
-	dprintf(fd, "\tset $%s = ", set->var->name);
+	dprintf(fd, "\tset %s = ", set->var->name);
 	dump_impl_set_value(fd, set->val, 1, 0);
 }
 
@@ -371,11 +515,18 @@ void PSI_ContextDump(PSI_Context *C, int fd)
 		dump_typedefs(fd, C->defs);
 		dprintf(fd, "\n");
 	}
-
+	if (C->unions) {
+		dump_unions(fd, C->unions);
+		dprintf(fd, "\n");
+	}
 	if (C->structs) {
 		dump_structs(fd, C->structs);
 		dprintf(fd, "\n");
 	}
+	if (C->enums) {
+		dump_enums(fd, C->enums);
+		dprintf(fd, "\n");
+	}
 	if (C->consts) {
 		dump_constants(fd, C->consts);
 		dprintf(fd, "\n");