check for _hx_customStack on haxe.Exception

Author: Simn

src/main.c

@@ -133,6 +133,26 @@ static void setup_handler() {
 }
 #endif
 
+bool maybe_print_custom_stack(vdynamic* ret) {
+	hl_type* exct = ret->t;
+	while (exct->kind == HOBJ) {
+		if (exct->obj->super == NULL) {
+			if (ucmp(exct->obj->name, USTR("haxe.Exception")) == 0) {
+				hl_field_lookup* f = hl_lookup_find(exct->obj->rt->lookup, exct->obj->rt->nlookup, hl_hash_gen(USTR("_hx_customStack"), true));
+				if (f == NULL || f->field_index < 0) break;
+				vdynamic* customStack = *(vdynamic**)((char*)(ret) + f->field_index);
+				if (customStack != NULL) {
+					uprintf(hl_to_string(customStack));
+					return true;
+				}
+			}
+			break;
+		}
+		exct = exct->obj->super;
+	}
+	return false;
+}
+
 #ifdef HL_WIN
 int wmain(int argc, pchar *argv[]) {
 #else
@@ -240,11 +260,13 @@ int main(int argc, pchar *argv[]) {
 	ctx.ret = hl_dyn_call_safe(&cl,NULL,0,&isExc);
 	hl_profile_end();
 	if( isExc ) {
-		varray *a = hl_exception_stack();
-		int i;
 		uprintf(USTR("Uncaught exception: %s\n"), hl_to_string(ctx.ret));
-		for(i=0;i<a->size;i++)
-			uprintf(USTR("Called from %s\n"), hl_aptr(a,uchar*)[i]);
+		if (!maybe_print_custom_stack(ctx.ret)) {
+			varray* a = hl_exception_stack();
+			int i;
+			for (i = 0; i < a->size; i++)
+				uprintf(USTR("Called from %s\n"), hl_aptr(a, uchar*)[i]);
+		}
 		hl_debug_break();
 		hl_global_free();
 		return 1;